def write_test(self):
filename = r'./radpy/plugins/BeamAnalysis/view/RFB/Unit Tests/test1.rfb'
data = load_rfb_data(filename)
bdml_export(data, self.temp_file.name)
data2 = load_xml_data(self.temp_file.name)
for j,k in zip(data, data2):
for i in [x[0] for x in TRAITS_TO_XML]:
if isinstance(getattr(j, i), numpy.ndarray):
npt.assert_equal(getattr(j, i), getattr(k, i))
else:
self.assertEqual(getattr(j, i), getattr(k, i))
def perform(self, event):
""" Perform the action. """
widget = self.window.active_view.control
file_root = widget.model().nodeFromIndex(widget.currentIndex()).getFileBranch()
filename = file_root.filename
#extension = os.path.basename(filename).split('.')[1]
filename = unicode(QFileDialog.getSaveFileName(self.window.control,
"Choose Save Filename", USERHOME,
"XML Files *.xml"))
beam_list = file_root.asRecord()
progress = QProgressBar()
progress.setWindowTitle('Saving...')
progress.setMinimum(0)
progress.setMaximum(len(beam_list))
progress.show()
bdml_export(beam_list, filename, progress)
def perform(self, event):
""" Perform the action. """
widget = self.window.active_view.control
file_root = widget.model().nodeFromIndex(
widget.currentIndex()).getFileBranch()
filename = file_root.filename
#extension = os.path.basename(filename).split('.')[1]
filename = unicode(
QFileDialog.getSaveFileName(self.window.control,
"Choose Save Filename", USERHOME,
"XML Files *.xml"))
beam_list = file_root.asRecord()
progress = QProgressBar()
progress.setWindowTitle('Saving...')
progress.setMinimum(0)
progress.setMaximum(len(beam_list))
progress.show()
bdml_export(beam_list, filename, progress)
