#===============================================================================
# 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 wx.lib.newevent
import controller
import gui.builtinViewColumns
import gui.mainFrame
from gui.builtinViewColumns import *
import sys
from eos.types import Slot

FitChanged, FIT_CHANGED = wx.lib.newevent.NewEvent()

class FittingView(wx.ListCtrl):
    DEFAULT_COLS = ["Module state",
                    "Module name/slot",
                    "attr:power",
                    "attr:cpu",
                    "attr:capacitorNeed",
                    "attr:trackingSpeed",
                    "Max range"]

    def __init__(self, parent):
        listStyle = wx.LC_REPORT | wx.BORDER_NONE
        wx.ListCtrl.__init__(self, parent, wx.ID_ANY, style=listStyle)

        self.imageList = wx.ImageList(16, 16)
        self.SetImageList(self.imageList, wx.IMAGE_LIST_SMALL)
        self.activeColumns = []
        self.Bind(wx.EVT_LIST_COL_BEGIN_DRAG, self.resizeChecker)

        mainFrame = gui.mainFrame.MainFrame.getInstance()
        mainFrame.Bind(FIT_CHANGED, self.fitChanged)

        self.shipBrowser = mainFrame.shipBrowser
        self.shipView = mainFrame.shipBrowser.shipView
        self.searchView = mainFrame.shipBrowser.shipView
        self.switch = mainFrame.fitMultiSwitch

        i = 0
        for colName in FittingView.DEFAULT_COLS:
            if colName[0:5] == "attr:":
                attrName = colName[5:]
                params = {"showIcon": True,
                          "displayName": False,
                          "attribute": attrName}
                col = gui.builtinViewColumns.getColumn("Attribute Display")(self, params)
            else:
                col = gui.builtinViewColumns.getColumn(colName)(self, None)

            self.addColumn(i, col)
            i += 1

        self.imageListBase = self.imageList.ImageCount
        self.activeFitID = None

        self.Bind(wx.EVT_LEFT_DCLICK, self.removeItem)

        self.Hide() #Don't show ourselves at start

        self.mainFrame = gui.mainFrame.MainFrame.getInstance()

    def OnEraseBackGround(self, event):
        #Prevent flicker by not letting the parent's method get called.
        pass

    def addColumn(self, i, col):
        self.activeColumns.append(col)
        info = wx.ListItem()
        info.m_mask = col.mask
        info.m_image = col.imageId
        info.m_text = col.columnText
        self.InsertColumnInfo(i, info)
        col.resized = False
        self.SetColumnWidth(i, wx.LIST_AUTOSIZE_USEHEADER if col.size is wx.LIST_AUTOSIZE else col.size)

    def resizeChecker(self, event):
        if self.activeColumns[event.Column].resizable is False:
            event.Veto()
        else:
            self.activeColumns[event.Column].resized = True

    def clearItemImages(self):
        for i in xrange(self.imageList.ImageCount - 1, self.imageListBase, -1):
            self.imageList.Remove(i)

    #Gets called from the fitMultiSwitch when it decides its time
    def changeFit(self, fitID):
        self.activeFitID = fitID
        if fitID == None:
            self.Hide()
        else:
            self.Show()

        wx.PostEvent(self.mainFrame, FitChanged(fitID=fitID))

    def appendItem(self, itemID):
        fitID = self.activeFitID
        if fitID != None:
            cFit = controller.Fit.getInstance()
            cFit.appendItem(fitID, itemID)
            wx.PostEvent(self.mainFrame, FitChanged(fitID=fitID))

    def removeItem(self, event):
        row, _ = self.HitTest(event.Position)
        if row != -1:
            cFit = controller.Fit.getInstance()
            cFit.removeItem(self.activeFitID, self.GetItemData(row))

        wx.PostEvent(self.mainFrame, FitChanged(fitID=self.activeFitID))

    def fitChanged(self, event):
        self.Hide()
        cFit = controller.Fit.getInstance()
        fit = cFit.getFit(event.fitID)
        selection = []
        sel = self.GetFirstSelected()
        while sel != -1:
            selection.append(sel)
            sel = self.GetNextSelected(sel)

        self.DeleteAllItems()
        self.clearItemImages()
        modSlotMap = {}
        slotOrder = [Slot.SUBSYSTEM, Slot.HIGH, Slot.MED, Slot.LOW, Slot.RIG]

        if fit is not None:
            for modid, mod in enumerate(fit.modules):
                index = self.InsertStringItem(sys.maxint, "")
                for i, col in enumerate(self.activeColumns):
                    self.SetStringItem(index, i, col.getText(mod), col.getImageId(mod))
                    self.SetItemData(index, modid)
                    modSlotMap[modid] = mod.slot

        for i, col in enumerate(self.activeColumns):
            if not col.resized:
                self.SetColumnWidth(i, wx.LIST_AUTOSIZE)
                if self.GetColumnWidth(i) < 40:
                    self.SetColumnWidth(i, 40)

        self.SortItems(lambda id1, id2: cmp(slotOrder.index(modSlotMap[id1]), slotOrder.index(modSlotMap[id2])))

        for sel in selection:
            self.Select(sel)

        self.Show()
        event.Skip()