#===============================================================================
# Copyright (C) 2010 Diego Duclos
#
# This file is part of pyfa.
#
# pyfa is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# pyfa is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with pyfa.  If not, see <http://www.gnu.org/licenses/>.
#===============================================================================

import wx
import service
import config

class ImportDialog(wx.Dialog):
    def __init__(self, parent):
        wx.Dialog.__init__ (self, parent, id=wx.ID_ANY, title=u"Import fitting from ...", pos=wx.DefaultPosition, size=wx.Size(500, 300), style=wx.DEFAULT_DIALOG_STYLE)
        self._toggleEdit = -1
        self.SetSizeHintsSz(wx.DefaultSize, wx.DefaultSize)
        self.SetMinSize((500,300))

        self._fitsFromFile = None
        self._fitsFromEdit = None

        mainSizer = wx.BoxSizer(wx.VERTICAL)

        headerSizer = wx.BoxSizer(wx.HORIZONTAL)

        self.cFilePicker = wx.FilePickerCtrl(self, wx.ID_ANY, wx.EmptyString, u"Select a fit file", u"*.*", style=wx.FLP_DEFAULT_STYLE | wx.FLP_FILE_MUST_EXIST | wx.FLP_CHANGE_DIR)
        headerSizer.Add(self.cFilePicker, 1, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5)

        self.tbtnEdit = wx.ToggleButton( self, wx.ID_ANY, u"Text Edit", wx.DefaultPosition, wx.DefaultSize, 0 )
        headerSizer.Add( self.tbtnEdit, 0, wx.ALIGN_CENTER_VERTICAL|wx.RIGHT, 5 )


        mainSizer.Add(headerSizer, 0, wx.EXPAND, 5)
        self.m_staticline2 = wx.StaticLine( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.LI_HORIZONTAL )
        self.m_staticline2.SetMinSize( wx.Size( 480,1 ) )
        mainSizer.Add( self.m_staticline2, 0, wx.EXPAND, 5 )
        contentSizer = wx.BoxSizer(wx.VERTICAL)

        self.tcEdit = wx.TextCtrl(self, wx.ID_ANY, u"", style=wx.TE_MULTILINE)
        self.tcEdit.SetMinSize( wx.Size( -1,250 ) )
        contentSizer.Add(self.tcEdit, 1, wx.EXPAND, 5)

        mainSizer.Add(contentSizer, 1, wx.EXPAND, 5)

        footerSizer = wx.BoxSizer(wx.HORIZONTAL)


        self.stStatus = wx.StaticText( self, wx.ID_ANY, u"Status: File mode.", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.stStatus.Wrap( -1 )
        footerSizer.Add( self.stStatus, 1, wx.ALIGN_CENTER_VERTICAL|wx.LEFT, 5 )

        self.btnImport = wx.Button( self, wx.ID_ANY, u"Import", wx.DefaultPosition, wx.DefaultSize, 0 )
        footerSizer.Add( self.btnImport, 0, wx.ALIGN_CENTER_VERTICAL|wx.TOP|wx.BOTTOM|wx.RIGHT, 5 )

        self.btnOK = wx.Button(self, wx.ID_OK, u"OK", wx.DefaultPosition, wx.DefaultSize, 0)
        footerSizer.Add(self.btnOK, 0, wx.TOP|wx.BOTTOM|wx.RIGHT|wx.ALIGN_CENTER_VERTICAL, 5)

        mainSizer.Add(footerSizer, 0, wx.EXPAND, 5)

        self.SetSizer(mainSizer)

        self.tbtnEdit.SetValue( True )
        self.tcEdit.Show(False)

        self.Layout()
        self.Fit()

        self.Bind( wx.EVT_CLOSE, self.CloseDlg )
        self.btnOK.Bind( wx.EVT_BUTTON, self.CloseDlg )
        self.cFilePicker.Bind( wx.EVT_FILEPICKER_CHANGED, self.prepareFileFits )
        self.tbtnEdit.Bind( wx.EVT_TOGGLEBUTTON, self.SwitchEditCtrl )
        self.btnImport.Bind( wx.EVT_BUTTON, self.ImportFittings )
        self.Centre(wx.BOTH)


    def ImportFittings( self, event ):
        sFit = service.Fit.getInstance()
        if self._toggleEdit == -1:
            if self._fitsFromFile:
                try:
                    sFit.saveImportedFits(self._fitsFromFile)
                    self.stStatus.SetLabel("Status: %d fit(s) imported" % len(self._fitsFromFile))
                except:
                    self.stStatus.SetLabel("Status: Error importing from file!")
                self._fitsFromFile = None
            else:
                self.stStatus.SetLabel("Status: No fits were specified. Use Browse button.")
        else:
            buffer = self.tcEdit.GetValue()
            if len(buffer) != 0:
                try:
                    self._fitsFromEdit = sFit.importFitFromBuffer(buffer)
                    sFit.saveImportedFits(self._fitsFromEdit)
                    self.stStatus.SetLabel("Status: %d fit(s) imported" % len(self._fitsFromEdit))
                except:
                    self.stStatus.SetLabel("Status: Error importing from text editor!")
            else:
                self.stStatus.SetLabel("Status: Nothing specified.")
        event.Skip()


    def prepareFileFits(self, event):
        sFit = service.Fit.getInstance()
        try:
            self._fitsFromFile = sFit.importFit(event.Path)
            self.stStatus.SetLabel("Status: Found %d fit(s)." % len(self._fitsFromFile))
        except:
            self.stStatus.SetLabel("Status: Invalid fitting file!")

    def ImportFromFile( self, event ):
        print event.Path
        event.Skip()

    def CloseDlg( self, event ):
        event.Skip()

    def SwitchEditCtrl( self, event ):
        self._toggleEdit *= -1
        if self._toggleEdit == -1:
            self.tcEdit.Show(False)
            self.cFilePicker.Enable( True )
            self.stStatus.SetLabel("Status: File mode.")
        else:
            self.tcEdit.Show(True)
            self.cFilePicker.Enable( False )
            self.stStatus.SetLabel("Status: Text edit mode.")
        self.Layout()
        self.Fit()
        event.Skip()

class ExportDialog(wx.Dialog):
    def __init__(self, parent):
        wx.Dialog.__init__ (self, parent, id=wx.ID_ANY, title=u"Export fit as ...", pos=wx.DefaultPosition, size=wx.Size(-1, -1), style=wx.DEFAULT_DIALOG_STYLE)

        self.SetSizeHintsSz(wx.DefaultSize, wx.DefaultSize)

        mainSizer = wx.BoxSizer(wx.VERTICAL)

        fileSizer = wx.BoxSizer(wx.HORIZONTAL)

        self.fitFileName = wx.TextCtrl(self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, 0)
        fileSizer.Add(self.fitFileName, 1, wx.ALIGN_CENTER_VERTICAL | wx.ALL, 5)

        self.cDirPicker = wx.DirPickerCtrl(self, wx.ID_ANY, wx.EmptyString, u"Select a folder", wx.DefaultPosition, wx.DefaultSize, wx.DIRP_DIR_MUST_EXIST)
        fileSizer.Add(self.cDirPicker, 0, wx.TOP | wx.BOTTOM | wx.ALIGN_CENTER_VERTICAL, 5)

        self.btnOK = wx.Button(self, wx.ID_ANY, u"OK", wx.DefaultPosition, wx.DefaultSize, 0)
        fileSizer.Add(self.btnOK, 0, wx.TOP | wx.BOTTOM | wx.ALIGN_CENTER_VERTICAL, 5)

        mainSizer.Add(fileSizer, 0, wx.EXPAND, 5)

        self.m_staticline2 = wx.StaticLine(self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.LI_HORIZONTAL)
        mainSizer.Add(self.m_staticline2, 0, wx.EXPAND, 5)

        choiceSizer = wx.BoxSizer(wx.VERTICAL)

        chCtrlChoices = [ u"EFT file", u"XML file", u"EFT && XML file" ]
        self.chCtrl = wx.RadioBox(self, wx.ID_ANY, u"Choose wisely", wx.DefaultPosition, wx.DefaultSize, chCtrlChoices, 2, wx.RA_SPECIFY_COLS)
        self.chCtrl.SetSelection(0)
        choiceSizer.Add(self.chCtrl, 0, wx.EXPAND | wx.ALL, 5)

        mainSizer.Add(choiceSizer, 1, wx.EXPAND, 5)

        self.SetSizer(mainSizer)
        self.Layout()
        mainSizer.Fit(self)

        self.Centre(wx.BOTH)