class CircuitEditor(EditorInterface, ElementMap):
def __init__(self, parent, ID, uri):
EditorInterface.__init__(self, parent, False, uri)
ElementMap.__init__(self, parent, ID)
self._controller = Controller()
self._logger = self._controller.getLogger()
self._model = Model()
self._circtree = ComponentManager().getComponent("circuit").getCircuitTree()
self._importer = edef.Importer()
self._mainframe = self._controller.getMainFrame()
self.setTitle(uri)
if not uri == "circ://":
self._loadCircuit(uri)
self.redraw()
self.setModified(False)
self._target = CircuitDropTarget(self)
self.SetDropTarget( self._target )
def _loadCircuit(self, uri):
wire_list = []
mod_table = {}
self._logger.debug("Load %s"%uri)
xml = self._model.openURI(uri)
meta = CircuitMeta(xml)
mods = meta.getModules()
for (ID, (name, label, (x,y), params)) in mods.items():
self._logger.debug("Load module %s as %s @ %s,%s"%(name, label, x,y))
moduri = "mod://"+"/".join(name.split("."))
mod_table[ID] = self.loadModule(moduri, x,y, params)
mod_table[ID].setLabel(label)
wire_list += meta.getWires(ID)
for ( frm, to ) in wire_list:
(frm_id, frm_pin) = frm
(to_id, to_pin) = to
frm = mod_table[frm_id].getPin(frm_pin)
to = mod_table[to_id].getPin(to_pin)
self.connect(frm,to)
def OnSelected(self, evt=None):
self._updateMainFrame()
def _updateMainFrame(self):
self._mainframe.bindCopy()
self._mainframe.bindCut()
self._mainframe.bindPaste()
self._mainframe.bindRedo()
self._mainframe.bindUndo()
self._mainframe.bindSaveAs(self.OnSaveAs)
if self.isModified() and self.getURI() != "circ://":
self._mainframe.bindSave(self.OnSave)
else: self._mainframe.bindSave()
def OnModified(self):
self.setModified(True)
self._updateMainFrame()
def OnSave(self, evt):
self._logger.debug("On save...")
if self.getURI() == "circ://":
self.OnSaveAs(evt)
return
txt = self._to_xml().toprettyxml(" ")
self._model.saveURI(self.getURI(), txt)
self.setModified(False)
self._updateMainFrame()
def OnSaveAs(self, evt):
dlg = SaveAsDialog(self, -1)
if dlg.ShowModal()==wx.ID_CANCEL:
dlg.Destroy()
return
uri = dlg.getSelection()
dlg.Destroy()
try:
txt = self._to_xml().toprettyxml(" ")
self._model.saveURI(uri, txt)
except:
showExceptionDialog(self, -1, "Unable to save file to %s"%uri)
self.setURI(uri)
self.setTitle(uri)
self._circtree.addURI(uri)
self.setModified(False)
self._updateMainFrame()
def _to_xml(self):
dom_impl = getDOMImplementation()
doc = dom_impl.createDocument(None, "Circuit", None)
root = doc.documentElement
mods = self.getObjects(gModule)
for idx in range(len(mods)):
mod = mods[idx]
mod_node = mod._to_xml(doc, idx)
cons = self.getConnectionsFrom(mod)
for con in cons:
to_pin = con.getTo().getName()
frm_pin = con.getFrom().getName()
to_idx = mods.index(con.getTo().getModule())
wire = doc.createElement("Wire")
frm = doc.createElement("From")
frm.setAttribute("id", str(idx))
frm.setAttribute("pin",frm_pin)
wire.appendChild(frm)
to = doc.createElement("To")
to.setAttribute("id", str(to_idx))
to.setAttribute("pin", to_pin)
wire.appendChild(to)
root.appendChild(wire)
root.appendChild(mod_node)
return doc
class ArchiveTree(wx.TreeCtrl):
def __init__(self, parent, ID):
wx.TreeCtrl.__init__(self, parent, ID, size=wx.Size(-1,-1),
style=wx.TR_HAS_BUTTONS|wx.TR_HIDE_ROOT|wx.TR_NO_LINES|wx.TR_FULL_ROW_HIGHLIGHT)
self._d_imgs = wx.ImageList(16,16)
self._bmp_class = self._d_imgs.Add( getArchiveBitmap() )
self._bmp_class_open = self._d_imgs.Add( getArchiveBitmap() )
self._bmp_module = self._d_imgs.Add( getPythonBitmap() )
self.SetImageList(self._d_imgs)
self._d_model = Model()
self._d_controller = Controller()
self._d_mainframe = self._d_controller.getMainFrame()
self._d_root = self.AddRoot("root")
self.SetPyData(self._d_root, (None,None))
self._logger = self._d_controller.getLogger()
#Add all archives:
archives = self._d_model.openURI("zip://")
for archive_uri in archives:
uri_list = self._d_model.openURI(archive_uri)
archive_name = getArchive(archive_uri)
arch_item = self.AppendItem(self._d_root, archive_name)
self.SetPyData(arch_item, ("Archive", archive_uri) )
self.SetItemImage(arch_item, self._bmp_class, wx.TreeItemIcon_Normal)
self.SetItemImage(arch_item, self._bmp_class_open, wx.TreeItemIcon_Expanded)
for file_uri in uri_list:
file_name = getPyFile(file_uri)
file_item = self.AppendItem(arch_item, file_name)
self.SetPyData(file_item, ("File", file_uri))
self.SetItemImage(file_item, self._bmp_module, wx.TreeItemIcon_Normal)
self.SetItemImage(file_item, self._bmp_module, wx.TreeItemIcon_Expanded)
# FIXME self._d_mainframe.bindNewArch(self.NewArchive)
# Events:
self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelection)
self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnActivate)
self.Bind(wx.EVT_SET_FOCUS, self.OnFocus)
def removeURI(self, uri):
item = self.getItemByURI(uri,self._d_root)
if not item: raise Exception("Item %s not found"%uri)
self.Delete(item)
def addURI(self, uri):
if self.getItemByURI(uri, self._d_root): return
if isPyFileURI(uri):
(ar, py) = splitPyFile(uri)
item = self.getItemByURI("zip://"+ar, self._d_root)
fitem = self.AppendItem(item, py)
self.SetPyData(fitem, ("File", uri))
self.SetItemImage(fitem, self._bmp_module, wx.TreeItemIcon_Normal)
self.SetItemImage(fitem, self._bmp_module, wx.TreeItemIcon_Expanded)
elif isArchiveURI(uri):
path = getArchive(uri)
item = self.AppendItem(self._d_root, path)
self.SetPyData(item, ("Archive",uri))
self.SetItemImage(item, self._bmp_class, wx.TreeItemIcon_Normal)
self.SetItemImage(item, self._bmp_class_open, wx.TreeItemIcon_Expanded)
else: raise Exception("Unkonw URI %s"%uri)
def getItemByURI(self, uri, item):
(typ, iuri) = self.GetPyData(item)
if iuri==uri: return item
(citem, cookie) = self.GetFirstChild(item)
while citem:
ret = self.getItemByURI(uri, citem)
if ret != None: return ret
(citem, cookie) = self.GetNextChild(item,cookie)
#
# CONTROLLER PART
#
def OnFocus(self, evt):
self._updateMainFrame()
evt.Skip()
def OnSelection(self, event):
item = event.GetItem()
self._updateMainFrame(item)
event.Skip()
def _updateMainFrame(self, item=None):
if not item: item = self.GetSelection()
self._d_mainframe.bindNew()
self._d_mainframe.bindOpen()
self._d_mainframe.bindDelete()
if not item: return
(typ, uri) = self.GetPyData(item)
if typ is "File":
self._d_mainframe.bindOpen(self.OpenFile, "&Open file")
self._d_mainframe.bindDelete(self.DeleteFile, "&Delete file")
elif typ is "Archive":
self._d_mainframe.bindNew(self.NewFile, "&New file")
self._d_mainframe.bindOpen(self.OpenArchive, "&Open archive")
self._d_mainframe.bindDelete(self.DeleteArchive, "&Delete archive")
def OnActivate(self, event):
item = event.GetItem()
(typ, uri) = self.GetPyData(item)
if typ == "File":
self._d_controller.DocumentOpen(uri)
event.Skip()
def OpenFile(self, evt=None):
item = self.GetSelection()
if not item: return
(typ, uri) = self.GetPyData(item)
if not typ == "File": return
self._d_controller.DocumentOpen(uri)
def DeleteFile(self, evt=None):
item = self.GetSelection()
if not item:
self._logger.warning("No item selected to delete")
return
(typ, uri) = self.GetPyData(item)
if not typ == "File": return
try:
self._d_controller.DocumentDelete(uri)
self.removeURI(uri)
except:
self._logger.exception("Unable to remove file %s"%uri)
def NewFile(self, evt=None):
self._d_controller.DocumentOpen("zip://")
def OpenArchive(self, evt=None):
item = self.GetSelection()
if not item: return
(typ, uri) = self.GetPyData(item)
if not typ=="Archive": return
self.Expand(item)
def NewArchive(self, evt=None):
selected = False
while not selected:
dlg = Dialogs.eDevNewArchiveDialog(self, -1)
if dlg.ShowModal() == wx.ID_CANCEL: return
uri = dlg.getSelection()
dlg.Destroy()
if self._d_model.checkURI(uri):
# FIXME ask for overwrite
continue
selected = True
self._d_model.saveURI(uri,"")
self.addURI(uri)
def DeleteArchive(self, evt=None):
item = self.GetSelection()
(typ, uri) = self.GetPyData(item)
py_list = self._d_model.openURI(uri)
for py in py_list:
self._d_controller.DocumentDelete(py)
self._d_model.deleteURI(uri)
self.Delete(item)
