342 lines
12 KiB
Python
342 lines
12 KiB
Python
#===============================================================================
|
|
# 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 wx.lib.mixins.listctrl as listmix
|
|
import gui.display as d
|
|
from gui.cachingImageList import CachingImageList
|
|
from gui.contextMenu import ContextMenu
|
|
|
|
|
|
ItemSelected, ITEM_SELECTED = wx.lib.newevent.NewEvent()
|
|
|
|
class MarketBrowser(wx.Panel):
|
|
def __init__(self, parent):
|
|
wx.Panel.__init__(self, parent)
|
|
vbox = wx.BoxSizer(wx.VERTICAL)
|
|
self.SetSizer(vbox)
|
|
|
|
#Add a search button on top
|
|
|
|
#Add a WHOLE panel for ONE SINGLE search button
|
|
#We have to be able to give the search more size, which can't be done in another way.
|
|
#(That I found)
|
|
p = wx.Panel(self)
|
|
sizer = wx.BoxSizer(wx.HORIZONTAL)
|
|
p.SetSizer(sizer)
|
|
|
|
vbox.Add(p, 0, wx.EXPAND)
|
|
self.search = SearchBox(p)
|
|
sizer.Add(self.search, 1, wx.ALIGN_CENTER_VERTICAL | wx.ALL, 2)
|
|
p.SetMinSize((wx.SIZE_AUTO_WIDTH, 33))
|
|
|
|
self.splitter = wx.SplitterWindow(self, style = wx.SP_LIVE_UPDATE)
|
|
vbox.Add(self.splitter, 1, wx.EXPAND)
|
|
|
|
self.marketView = MarketTree(self.splitter, self)
|
|
self.itemView = ItemView(self.splitter, self)
|
|
|
|
self.splitter.SplitHorizontally(self.marketView, self.itemView)
|
|
self.splitter.SetMinimumPaneSize(250)
|
|
|
|
cMarket = service.Market.getInstance()
|
|
# Setup our buttons for metaGroup selection
|
|
# Same fix as for search box on macs,
|
|
# need some pixels of extra space or everything clips and is ugly
|
|
p = wx.Panel(self)
|
|
box = wx.BoxSizer(wx.HORIZONTAL)
|
|
p.SetSizer(box)
|
|
vbox.Add(p, 0, wx.EXPAND)
|
|
for name in ("normal", "faction", "complex", "officer"):
|
|
btn = wx.ToggleButton(p, wx.ID_ANY, name.capitalize(), style=wx.BU_EXACTFIT)
|
|
setattr(self, name, btn)
|
|
box.Add(btn, 1, wx.ALIGN_CENTER)
|
|
btn.Bind(wx.EVT_TOGGLEBUTTON, self.toggleMetagroup)
|
|
btn.metaName = name
|
|
|
|
self.normal.SetValue(True)
|
|
cMarket.activateMetaGroup("normal")
|
|
p.SetMinSize((wx.SIZE_AUTO_WIDTH, btn.GetSize()[1] + 5))
|
|
|
|
def toggleMetagroup(self, event):
|
|
ctrl = wx.GetMouseState().ControlDown()
|
|
cMarket = service.Market.getInstance()
|
|
btn = event.EventObject
|
|
if not ctrl:
|
|
for name in ("normal", "faction", "complex", "officer"):
|
|
button = getattr(self, name)
|
|
button.SetValue(False)
|
|
button.Enable(True)
|
|
|
|
cMarket.disableMetaGroup(name)
|
|
|
|
btn.SetValue(True)
|
|
cMarket.activateMetaGroup(btn.metaName)
|
|
else:
|
|
# Note: using the old value might seem weird,
|
|
# But the button hasn't been toggled by wx yet
|
|
target = btn.GetValue()
|
|
btn.SetValue(target)
|
|
if target:
|
|
cMarket.activateMetaGroup(btn.metaName)
|
|
else:
|
|
cMarket.disableMetaGroup(btn.metaName)
|
|
|
|
if self.itemView.searching:
|
|
self.itemView.filteredSearchAdd()
|
|
else:
|
|
self.itemView.selectionMade(event)
|
|
|
|
def jump(self, item):
|
|
self.marketView.jump(item)
|
|
|
|
|
|
class SearchBox(wx.SearchCtrl):
|
|
def __init__(self, parent):
|
|
wx.SearchCtrl.__init__(self, parent, wx.ID_ANY, style=wx.TE_PROCESS_ENTER)
|
|
self.ShowCancelButton(True)
|
|
|
|
class MarketTree(wx.TreeCtrl):
|
|
def __init__(self, parent, marketBrowser):
|
|
wx.TreeCtrl.__init__(self, parent, style=wx.TR_DEFAULT_STYLE | wx.TR_HIDE_ROOT)
|
|
self.root = self.AddRoot("CHRISTMAS TREE!")
|
|
self.marketBrowser = marketBrowser
|
|
|
|
self.imageList = CachingImageList(16, 16)
|
|
self.SetImageList(self.imageList)
|
|
|
|
cMarket = service.Market.getInstance()
|
|
|
|
root = cMarket.getMarketRoot()
|
|
for id, name, iconFile in root:
|
|
iconId = self.addImage(iconFile)
|
|
childId = self.AppendItem(self.root, name, iconId, data=wx.TreeItemData(id))
|
|
self.AppendItem(childId, "dummy")
|
|
|
|
self.SortChildren(self.root)
|
|
|
|
#Bind our lookup method to when the tree gets expanded
|
|
self.Bind(wx.EVT_TREE_ITEM_EXPANDING, self.expandLookup)
|
|
|
|
def addImage(self, iconFile):
|
|
if iconFile is None:
|
|
return -1
|
|
|
|
return self.imageList.Add(iconFile, "pack")
|
|
|
|
def expandLookup(self, event):
|
|
root = event.Item
|
|
child, cookie = self.GetFirstChild(root)
|
|
if self.GetItemText(child) == "dummy":
|
|
cMarket = service.Market.getInstance()
|
|
#A DUMMY! Keeeel!!! EBUL DUMMY MUST DIAF!
|
|
self.Delete(child)
|
|
|
|
#Add 'real stoof!' instead
|
|
for id, name, iconFile, more in cMarket.getChildren(self.GetPyData(root)):
|
|
iconId = self.addImage(iconFile)
|
|
childId = self.AppendItem(root, name, iconId, data=wx.TreeItemData(id))
|
|
if more:
|
|
self.AppendItem(childId, "dummy")
|
|
|
|
self.SortChildren(root)
|
|
|
|
def jump(self, item):
|
|
cMarket = service.Market.getInstance()
|
|
mg = item.marketGroup
|
|
if mg is None and item.metaGroup is not None:
|
|
mg = item.metaGroup.parent.marketGroup
|
|
for btn in ("normal", "faction", "complex", "officer"):
|
|
getattr(self.marketBrowser, btn).SetValue(False)
|
|
cMarket.disableMetaGroup(btn)
|
|
|
|
metaGroup = cMarket.getMetaName(item.metaGroup.ID)
|
|
|
|
getattr(self.marketBrowser, metaGroup).SetValue(True)
|
|
cMarket.activateMetaGroup(metaGroup)
|
|
self.marketBrowser.itemView.searching = False
|
|
|
|
if mg is None:
|
|
return
|
|
|
|
jumpList = []
|
|
while mg is not None:
|
|
jumpList.append(mg.ID)
|
|
mg = mg.parent
|
|
|
|
cMarket.MARKET_GROUPS
|
|
for id in cMarket.MARKET_GROUPS:
|
|
if id in jumpList:
|
|
jumpList = jumpList[:jumpList.index(id)+1]
|
|
|
|
item = self.root
|
|
for i in range(len(jumpList) -1, -1, -1):
|
|
target = jumpList[i]
|
|
child, cookie = self.GetFirstChild(item)
|
|
while self.GetItemPyData(child) != target:
|
|
child, cookie = self.GetNextChild(item, cookie)
|
|
|
|
item = child
|
|
self.Expand(item)
|
|
|
|
self.SelectItem(item)
|
|
self.marketBrowser.itemView.searching = False
|
|
|
|
class ItemView(d.Display):
|
|
DEFAULT_COLS = ["Base Icon",
|
|
"Base Name",
|
|
"attr:power,,,True",
|
|
"attr:cpu,,,True"]
|
|
|
|
def __init__(self, parent, marketBrowser):
|
|
d.Display.__init__(self, parent)
|
|
marketBrowser.Bind(wx.EVT_TREE_SEL_CHANGED, self.selectionMade)
|
|
|
|
self.searching = False
|
|
self.marketBrowser = marketBrowser
|
|
self.marketView = marketBrowser.marketView
|
|
|
|
#Make sure our search actualy does intresting stuff
|
|
self.marketBrowser.search.Bind(wx.EVT_TEXT_ENTER, self.scheduleSearch)
|
|
self.marketBrowser.search.Bind(wx.EVT_SEARCHCTRL_SEARCH_BTN, self.scheduleSearch)
|
|
self.marketBrowser.search.Bind(wx.EVT_SEARCHCTRL_CANCEL_BTN, self.clearSearch)
|
|
self.marketBrowser.search.Bind(wx.EVT_TEXT, self.scheduleSearch)
|
|
|
|
#Make sure WE do intresting stuff TOO
|
|
self.Bind(wx.EVT_CONTEXT_MENU, self.contextMenu)
|
|
self.Bind(wx.EVT_LIST_ITEM_ACTIVATED, self.itemActivated)
|
|
|
|
def itemActivated(self, event):
|
|
#Check if something is selected, if so, spawn the menu for it
|
|
sel = self.GetFirstSelected()
|
|
if sel == -1:
|
|
return
|
|
|
|
wx.PostEvent(self.mainFrame, ItemSelected(itemID=self.active[sel].ID))
|
|
|
|
def selectionMade(self, event):
|
|
if self.searching:
|
|
self.clearSearch(None, False)
|
|
|
|
root = self.marketView.GetSelection()
|
|
if root.IsOk():
|
|
if self.marketView.GetChildrenCount(root) != 0:
|
|
return
|
|
|
|
cMarket = service.Market.getInstance()
|
|
items, usedMetas = cMarket.getVariations(self.marketView.GetPyData(root))
|
|
self.update(items)
|
|
self.toggleButtons(usedMetas)
|
|
|
|
def clearSearch(self, event=None, clear=True):
|
|
self.DeleteAllItems()
|
|
if clear:
|
|
self.marketBrowser.search.Clear()
|
|
|
|
cMarket = service.Market.getInstance()
|
|
for name in ("normal", "faction", "complex", "officer"):
|
|
btn = getattr(self.marketBrowser, name)
|
|
btn.Enable(True)
|
|
btn.SetValue(False)
|
|
cMarket.disableMetaGroup(btn.metaName)
|
|
|
|
cMarket.activateMetaGroup("normal")
|
|
self.marketBrowser.normal.SetValue(True)
|
|
|
|
self.searching = False
|
|
|
|
def scheduleSearch(self, event):
|
|
search = self.marketBrowser.search.GetLineText(0)
|
|
if len(search) < 3:
|
|
self.clearSearch(event, False)
|
|
return
|
|
|
|
cMarket = service.Market.getInstance()
|
|
if not self.searching:
|
|
for name in ("normal", "faction", "complex", "officer"):
|
|
getattr(self.marketBrowser, name).SetValue(True)
|
|
cMarket.activateMetaGroup(name)
|
|
|
|
self.searching = True
|
|
cMarket.searchItems(search, self.populateSearch)
|
|
|
|
def populateSearch(self, results):
|
|
self.filteredSearchAdd(*results)
|
|
|
|
def itemFilter(self, item):
|
|
pass
|
|
|
|
def filteredSearchAdd(self, items=None, usedMetas=None):
|
|
if self.searching is False:
|
|
return
|
|
|
|
self.items = items if items is not None else self.items
|
|
self.usedMetas = usedMetas if usedMetas is not None else self.usedMetas
|
|
sMarket = service.Market.getInstance()
|
|
|
|
self.update(sMarket.filterItems(self.items))
|
|
|
|
#Gray out empty toggles
|
|
self.toggleButtons(self.usedMetas)
|
|
|
|
def toggleButtons(self, usedMetas):
|
|
cMarket = service.Market.getInstance()
|
|
for name in ("normal", "faction", "complex", "officer"):
|
|
btn = getattr(self.marketBrowser, name)
|
|
btn.SetValue(False)
|
|
btn.Enable(False)
|
|
|
|
for meta in usedMetas:
|
|
btn = getattr(self.marketBrowser, cMarket.getMetaName(meta))
|
|
btn.SetValue(cMarket.isMetaIdActive(meta))
|
|
btn.Enable(True)
|
|
|
|
def contextMenu(self, event):
|
|
#Check if something is selected, if so, spawn the menu for it
|
|
sel = self.GetFirstSelected()
|
|
if sel == -1:
|
|
return
|
|
|
|
item = self.active[sel]
|
|
# We were searching, this means that our results come from the worker thread
|
|
# Refetch items, else it'll try to reuse the object fetched in the other thread
|
|
# Which makes it go BOOM CRACK DOOM
|
|
if self.searching:
|
|
sMarket = service.Market.getInstance()
|
|
item = sMarket.getItem(item.ID)
|
|
|
|
menu = ContextMenu.getMenu((item,), "item" if self.searching is False else "itemSearch")
|
|
self.PopupMenu(menu)
|
|
|
|
def itemSort(self, item):
|
|
if item.metaGroup is None:
|
|
return (item.name, 0, item.name)
|
|
else:
|
|
return (item.metaGroup.parent.name, item.metaGroup.ID , item.name)
|
|
|
|
def populate(self, stuff):
|
|
stuff.sort(key=self.itemSort)
|
|
self.active = stuff
|
|
d.Display.populate(self, stuff)
|
|
|
|
def refresh(self, stuff):
|
|
stuff.sort(key=self.itemSort)
|
|
d.Display.refresh(self, stuff)
|