class TripExportCubeDialog(wx.Dialog):
def __init__(self):
pre = wx.PreDialog()
self.PostCreate(pre)
def Init(self, plan):
self.plan = plan
self.path = "~/"
self.output_path = ""
self.checkbox_vois = XRCCTRL(self, "checkbox_vois")
wx.EVT_LISTBOX(self, XRCID("checkbox_vois"), self.selected_changed)
wx.EVT_BUTTON(self, XRCID("btn_ok"), self.save_and_close)
wx.EVT_BUTTON(self, XRCID("btn_cancel"), self.close)
wx.EVT_BUTTON(self, XRCID("btn_reset"), self.reset)
self.lbl_path = XRCCTRL(self, "lbl_path")
self.txt_value = XRCCTRL(self, "txt_value")
wx.EVT_TEXT(self, XRCID("txt_value"), self.text_value_changed)
pub.subscribe(self.path_changed, "general.export.cube_export_path")
pub.sendMessage("settings.value.request", "general.export.cube")
pub.subscribe(self.patient_data_updated, "patient.loaded")
pub.sendMessage("patient.request", {})
for voi in plan.get_vois():
self.checkbox_vois.Append(voi.get_name())
def patient_data_updated(self, msg):
self.data = msg.data
def selected_changed(self, evt):
selected = self.checkbox_vois.GetStringSelection()
self.txt_value.SetValue("")
for voi in self.plan.get_vois():
if selected == voi.get_name():
if voi.get_cube_value() is -1:
self.txt_value.SetValue("")
else:
self.txt_value.SetValue("%d" % voi.get_cube_value())
def text_value_changed(self, evt):
selected = self.checkbox_vois.GetStringSelection()
if len(selected) is 0:
return
for voi in self.plan.get_vois():
if selected == voi.get_name():
try:
voi.set_cube_value(int(self.txt_value.GetValue()))
except Exception as e:
pass
def path_changed(self, msg):
if not msg.data is None:
self.path = msg.data
self.lbl_path.SetLabel(self.path)
def reset(self, evt):
for voi in self.plan.get_vois():
voi.set_cube_value(-1)
self.txt_value.SetValue("")
for k, item in enumerate(self.checkbox_vois.GetItems()):
self.checkbox_vois.Check(k, False)
def browse_for_file(self):
dlg = wx.FileDialog(self,
message="Save Picture",
defaultDir=self.path,
style=wx.FD_SAVE)
if dlg.ShowModal() == wx.ID_OK:
path = dlg.GetPath()
a = os.path.splitext(path)
if not a[-1] is "dos":
path = path + ".dos"
self.output_path = path
pub.sendMessage("settings.value.updated",
{"general.export.cube": os.path.dirname(path)})
return True
return False
def save_and_close(self, evt):
selected = self.checkbox_vois.GetCheckedStrings()
vois = []
dos = None
for voi in self.plan.get_vois():
if voi.get_name() in selected:
if dos is None:
dos = voi.get_voi().get_voi_data().get_voi_cube(
) / 1000 * voi.get_cube_value()
else:
dos.cube[dos.cube == 0] = -1
a = voi.get_voi().get_voi_data().get_voi_cube(
) / 1000 * voi.get_cube_value()
dos.cube[dos.cube == -1] = a.cube[dos.cube == -1]
if not dos is None:
if self.browse_for_file():
dos.write(self.output_path)
else:
return
else:
pass
self.Close()
def close(self, evt):
self.Close()
class Panel(ObjectPanel.Panel):
"""
A wx.Panel for displaying and editing Properties
"""
def __init__(self, parent, id=wx.ID_ANY, style=wx.EXPAND):
#load from XRC, need to use two-stage create
pre = wx.PrePanel()
res = gui.XrcUtilities.XmlResource('./gui/xrc/PropertyPanel.xrc')
res.LoadOnPanel(pre, parent, "PropertyPanel")
self.PostCreate(pre)
ObjectPanel.Panel.Setup(self)
self.OnCreate()
def OnCreate(self):
self.name_field = XRCCTRL(self, "name_field")
self.rank_field = XRCCTRL(self, "rank_field")
self.desc_field = XRCCTRL(self, "desc_field")
self.disp_field = XRCCTRL(self, "disp_field")
self.tpcl_disp_stc = XRCCTRL(self, "tpcl_disp_stc")
self.tpcl_req_stc = XRCCTRL(self, "tpcl_req_stc")
self.cat_choice = XRCCTRL(self, "cat_choice")
self.desc_field.Bind(wx.EVT_TEXT,
self.CreateAttributeMonitor('description'))
self.rank_field.Bind(wx.EVT_TEXT, self.CreateAttributeMonitor('rank'))
self.disp_field.Bind(wx.EVT_TEXT,
self.CreateAttributeMonitor('display_text'))
self.tpcl_disp_stc.Bind(wx.stc.EVT_STC_CHANGE,
self.CreateAttributeMonitor('tpcl_display'))
self.tpcl_req_stc.Bind(wx.stc.EVT_STC_CHANGE,
self.CreateAttributeMonitor('tpcl_requires'))
self.cat_choice.Bind(wx.EVT_CHECKLISTBOX,
self.CreateAttributeMonitor('categories'))
self.loaded = False
def LoadObject(self, node):
self.loading = True
self.node = node
self.object = node.GetObject()
self.node.visible = True
self.name_field.SetLabel(str(self.object.name))
self.rank_field.SetValue(str(self.object.rank))
if self.object.errors.has_key('rank'):
print "Error in rank!"
self.SetErrorLabel('rank', self.object.errors['rank'])
self.desc_field.SetValue(str(self.object.description))
if self.object.errors.has_key('description'):
print "Error in description!"
self.SetErrorLabel('description',
self.object.errors['description'])
self.disp_field.SetValue(str(self.object.display_text))
if self.object.errors.has_key('display_text'):
print "Error in display_text!"
self.SetErrorLabel('display_text',
self.object.errors['display_text'])
self.tpcl_disp_stc.SetText(str(self.object.tpcl_display))
if self.object.errors.has_key('tpcl_display'):
print "Error in tpcl_display!"
self.SetErrorLabel('tpcl_display',
self.object.errors['tpcl_display'])
self.tpcl_req_stc.SetText(str(self.object.tpcl_requires))
if self.object.errors.has_key('tpcl_requires'):
print "Error in tpcl_requires!"
self.SetErrorLabel('tpcl_requires',
self.object.errors['tpcl_requires'])
#fill the category choice box
self.cat_choice.Clear()
for catnode in self.node.object_database.getObjectsOfType('Category'):
idx = self.cat_choice.Append(catnode.name)
if catnode.name in self.object.categories:
self.cat_choice.Check(idx)
if self.object.errors.has_key('categories'):
print "Error in categories!"
self.SetErrorLabel('categories', self.object.errors['categories'])
self.loaded = True
self.loading = False
self.Show()
return self
def Destroy(self):
self.Hide()
def ReallyDestroy(self):
wx.Panel.Destroy(self)
def cleanup(self):
self.CleanupErrorLabels()
self.node.visible = False
self.Hide()
self.node.ClearObject()
self.object = None
self.loaded = False
class Panel(ObjectPanel.Panel):
"""
A wx.Panel for displaying and editing Components
"""
def __init__(self, parent, id=wx.ID_ANY, style=wx.EXPAND):
#load from XRC, need to use two-stage create
pre = wx.PrePanel()
res = gui.XrcUtilities.XmlResource('./gui/xrc/ComponentPanel.xrc')
res.LoadOnPanel(pre, parent, "ComponentPanel")
self.PostCreate(pre)
ObjectPanel.Panel.Setup(self)
self.OnCreate()
def OnCreate(self):
self.name_field = XRCCTRL(self, "name_field")
self.desc_field = XRCCTRL(self, "desc_field")
self.tpcl_req_stc = XRCCTRL(self, "tpcl_req_stc")
self.cat_choice = XRCCTRL(self, "cat_choice")
self.prop_list = XRCCTRL(self, "prop_list")
self.tpcl_cost_stc = XRCCTRL(self, "tpcl_cost_stc")
self.tpcl_cost_stc.Enable(False)
add_button = XRCCTRL(self, "add_button")
self.Bind(wx.EVT_BUTTON, self.OnAddProperty, add_button)
remove_button = XRCCTRL(self, "remove_button")
self.Bind(wx.EVT_BUTTON, self.OnRemoveProperty, remove_button)
#bind event handlers
self.desc_field.Bind(wx.EVT_TEXT, self.CreateAttributeMonitor('description'))
self.tpcl_req_stc.Bind(wx.stc.EVT_STC_CHANGE, self.CreateAttributeMonitor('tpcl_requirements'))
self.tpcl_cost_stc.Bind(wx.stc.EVT_STC_CHANGE, self.OnCostEdit)
self.filling_tpcl_cost = False
self.Bind(wx.EVT_LISTBOX, self.OnPropListSelect, self.prop_list)
self.cat_choice.Bind(wx.EVT_CHECKLISTBOX, self.CreateAttributeMonitor('categories'))
#self.BindEditWatchers([self.desc_field, self.tpcl_req_stc])
self.loaded = False
def LoadObject(self, node):
print "ComponentPanel loading %s" % node.name
#VERY IMPORTANT TO MARK OURSELVES AS LOADING HERE
# THIS WAY WE AVOID PROGRAMMATIC CHANGES BEING MARKED AS USER CHANGES
self.loading = True
self.node = node
self.object = node.GetObject()
print "\tErrors:", self.object.errors
self.node.visible = True
self.name_field.SetLabel(str(self.object.name))
self.desc_field.SetValue(str(self.object.description))
if self.object.errors.has_key('description'):
print "Error in description!"
self.SetErrorLabel('description', self.object.errors['description'])
self.tpcl_req_stc.SetText(self.object.tpcl_requirements)
if self.object.errors.has_key('tpcl_requirements'):
print "Error in tpcl_requirements!"
self.SetErrorLabel('tpcl_requirements', self.object.errors['tpcl_requirements'])
self.filling_tpcl_cost = True
self.tpcl_cost_stc.SetText("")
self.filling_tpcl_cost = False
self.tpcl_cost_stc.Enable(False)
if self.object.errors.has_key('properties'):
print "Error in properties!"
self.SetErrorLabel('properties', self.object.errors['properties'])
#fill the category choice box
self.cat_choice.Clear()
for catnode in self.node.object_database.getObjectsOfType('Category'):
idx = self.cat_choice.Append(catnode.name)
if catnode.name in self.object.categories:
self.cat_choice.Check(idx)
if self.object.errors.has_key('categories'):
print "Error in categories!"
self.SetErrorLabel('categories', self.object.errors['categories'])
#create the property list
self.prop_sel = -1
prop_names = [pname for pname in self.object.properties.keys()]
self.prop_list.Set(prop_names)
self.node.object_database.Emphasize(prop_names, "BLUE")
self.loaded = True
self.loading = False
self.Show()
return self
def OnDClickProperty(self, event):
"""\
Should open a window to edit the TPCL cost function here.
"""
pass
def OnPropListSelect(self, event):
sel_idx = self.prop_list.GetSelection()
if sel_idx != wx.NOT_FOUND:
self.tpcl_cost_stc.Enable(True)
self.prop_sel = sel_idx
self.filling_tpcl_cost = True
self.tpcl_cost_stc.SetText(self.object.properties[
self.prop_list.GetString(sel_idx)])
self.filling_tpcl_cost = False
def OnCostEdit(self, event):
"""\
Saves changes made to the TPCL cost functions
"""
print "Handling a cost edit event!"
idx = self.prop_list.GetSelection()
if idx == wx.NOT_FOUND or self.filling_tpcl_cost:
pass
else:
self.object.properties[self.prop_list.GetString(idx)] = \
self.tpcl_cost_stc.GetText()
self.node.SetModified(True)
event.Skip()
def OnAddProperty(self, event):
print "On Add Property"
loose_props = filter(lambda x: not x in self.object.properties.keys(),
[n.name for n in self.node.object_database.getObjectsOfType('Property')])
choice_diag = wx.MultiChoiceDialog(self, "Choose the Properties to add...",
"Add Properties...", loose_props)
choice_diag.ShowModal()
if len(choice_diag.GetSelections()) > 0:
print "Selection OK"
prop_names = []
for i in choice_diag.GetSelections():
print "\t" + loose_props[i]
prop_names.append(loose_props[i])
self.object.properties[loose_props[i]] = "(lambda (design) #)"
self.prop_list.Append(loose_props[i])
self.node.SetModified(True)
self.node.object_database.Emphasize(prop_names, "BLUE")
else:
#cancelled
print "CANCELED!"
print "Selections: ", choice_diag.GetSelections()
pass
choice_diag.Destroy()
def OnRemoveProperty(self, event):
#remove the selected properties
if self.prop_list.GetSelections() != []:
ridx = []
prop_names = []
for idx in self.prop_list.GetSelections():
prop_name = self.prop_list.GetString(idx)
prop_names.append(prop_name)
del self.object.properties[prop_name]
ridx.insert(0, idx)
for i in ridx:
self.prop_list.Delete(i)
self.node.object_database.UnEmphasize(prop_names)
self.tpcl_cost_stc.SetText("")
self.node.SetModified(True)
def Destroy(self):
self.Hide()
def ReallyDestroy(self):
wx.Panel.Destroy(self)
def cleanup(self):
print "Cleaning up Component Panel"
self.CleanupErrorLabels()
self.node.object_database.UnEmphasize(
[self.prop_list.GetString(i) for i in range(0, self.prop_list.GetCount())])
self.node.visible = False
self.Hide()
self.node.ClearObject()
self.object = None
self.loaded = False
