def exportText(self, *event):
from memops.gui.FileSelect import FileType
from memops.gui.FileSelectPopup import FileSelectPopup
if self.textOut:
fileTypes = [
FileType('Text', ['*.txt']),
FileType('CSV', ['*.csv']),
FileType('All', ['*'])
]
fileSelectPopup = FileSelectPopup(self,
file_types=fileTypes,
title='Save table as text',
dismiss_text='Cancel',
selected_file_must_exist=False)
fileName = fileSelectPopup.getFile()
if fileName:
file = open(fileName, 'w')
if fileName.endswith('.csv'):
for textRow in self.textMatrix:
file.write(','.join(textRow) + '\n')
else:
file.write(self.textOut)
def importNmrStar31(self):
if not self.project:
showWarning('Failure','Please create a new CCPN project.', parent=self)
return
fileTypes = [ FileType('STAR', ['*.str']),
FileType('All', ['*'])]
fileSelectPopup = FileSelectPopup(self, file_types=fileTypes,
title='Import NMR-STAR 3.1 file', dismiss_text='Cancel',
selected_file_must_exist=True, multiSelect=False,)
fileName = fileSelectPopup.getFile()
if not os.path.exists(fileName):
showWarning('Failure','No such file.', parent=self)
return
if fileName:
nmrStarObj = NmrStarFormat(self.project, self, verbose=True)
nmrStarObj.readProject(fileName, minimalPrompts=True, version='3.1')
self.eciFrame.updateAll()
return nmrStarObj
def importFastaSeq(self):
if not self.project:
showWarning('Failure','Please create a new CCPN project.', parent=self)
return
fileTypes = [ FileType('Fasta', ['*.fsa']),
FileType('All', ['*']) ]
fileSelectPopup = FileSelectPopup(self, file_types=fileTypes,
title='Import Fasta sequence file', dismiss_text='Cancel',
selected_file_must_exist=True, multiSelect=False,)
fileName = fileSelectPopup.getFile()
if not os.path.exists(fileName):
showWarning('Failure','No such file.', parent=self)
return
if fileName:
name = os.path.split(fileName)[1]
if '.' in name:
name = name.split('.')[0]
fastaObj = FastaFormat(self.project, self, verbose=True)
fastaObj.readSequence(fileName, minimalPrompts=True)
self.eciFrame.updateAll()
return fastaObj
def importPdb(self):
if not self.project:
showWarning('Failure','Please create a new CCPN project.', parent=self)
return
fileTypes = [ FileType('PDB', ['*.pdb']),
FileType('PDB Entry', ['*.ent']),
FileType('All', ['*']) ]
fileSelectPopup = FileSelectPopup(self, file_types=fileTypes,
title='Import PDB 3.20 file', dismiss_text='Cancel',
selected_file_must_exist=True, multiSelect=False,)
fileName = fileSelectPopup.getFile()
if not os.path.exists(fileName):
showWarning('Failure','No such file.', parent=self)
return
if fileName:
name = os.path.split(fileName)[1]
if '.' in name:
name = name.split('.')[0]
pdbObj = ReadPdb(fileName, self.project, 'PDB_'+ name, self)
self.eciFrame.updateAll()
return pdbObj
def importPdbCoords(self):
if not self.project:
showWarning('Failure','Please create a new CCPN project.', parent=self)
return
fileTypes = [ FileType('PDB', ['*.pdb']),
FileType('PDB Entry', ['*.ent']),
FileType('All', ['*']) ]
fileSelectPopup = FileSelectPopup(self, file_types=fileTypes,
title='Import PDB coordinates file', dismiss_text='Cancel',
selected_file_must_exist=True, multiSelect=False,)
fileName = fileSelectPopup.getFile()
if not os.path.exists(fileName):
showWarning('Failure','No such file.', parent=self)
return
if fileName:
name = os.path.split(fileName)[1]
if '.' in name:
name = name.split('.')[0]
pdbObj = PseudoPdbFormat(self.project, self, verbose=True)
pdbObj.readCoordinates(fileName, minimalPrompts=True)
self.eciFrame.updateAll()
return pdbObj
def chooseOldProjectFile(self):
fileTypes = [ FileType('CING', ['project.xml']),
FileType('All', ['*'])
]
popup = FileSelectPopup(self, file_types = fileTypes,
title = 'Select CING project file', dismiss_text = 'Cancel',
selected_file_must_exist = True)
fileName = popup.getFile()
# dirName = popup.getDirectory()
if len(fileName) > 0:
# Put text into entry,name widgets
dummy,name = cing.Project.rootPath(fileName)
self.projEntry.configure(state='normal')
self.projEntry.set(fileName)
self.nameEntry.configure(state='normal')
self.nameEntry.set(name)
self.nameEntry.configure(state='disabled')
# choose the correct radiobutton
self.projOptionsSelect.setIndex(0)
self.updateGui()
# end if
#nd if
popup.destroy()
def chooseOldProjectFile(self):
fileTypes = [FileType('CING', ['project.xml']), FileType('All', ['*'])]
popup = FileSelectPopup(self,
file_types=fileTypes,
title='Select CING project file',
dismiss_text='Cancel',
selected_file_must_exist=True)
fileName = popup.getFile()
# dirName = popup.getDirectory()
if len(fileName) > 0:
# Put text into entry,name widgets
dummy, name = cing.Project.rootPath(fileName)
self.projEntry.configure(state='normal')
self.projEntry.set(fileName)
self.nameEntry.configure(state='normal')
self.nameEntry.set(name)
self.nameEntry.configure(state='disabled')
# choose the correct radiobutton
self.projOptionsSelect.setIndex(0)
self.updateGui()
# end if
#nd if
popup.destroy()
def selectFile(self):
fileSelectPopup = FileSelectPopup(self, title = 'Choose file', dismiss_text = 'Cancel',
selected_file_must_exist = True)
fileName = fileSelectPopup.getFile()
self.textWindow.setText('File Selected: %s\n' % fileName)
def selectExecutable(self):
""" Choose the MODULE executable appropriate for your system. """
popup = FileSelectPopup(self, show_file=True)
executable = popup.getFile()
if executable:
self.moduleExeEntry.set(executable)
popup.destroy()
def chooseExecutable(self):
fileTypes = [FileType('Table', ['*.exe']), FileType('All', ['*'])]
fileSelectPopup = FileSelectPopup(self, file_types = fileTypes,
title = 'Choose 2d BACUS executable', dismiss_text = 'Cancel',
selected_file_must_exist = False)
fileName = fileSelectPopup.getFile()
if fileName:
self.executableEntry.set(fileName)
def chooseMiscScript(self):
# Prepend default Cyana file extension below
fileTypes = [ FileType('All', ['*']), ]
popup = FileSelectPopup(self, file_types = fileTypes,
title = 'Script file', dismiss_text = 'Cancel',
selected_file_must_exist = True)
fileName = popup.getFile()
self.miscScriptEntry.set(fileName)
popup.destroy()
def chooseValidScript(self):
# Prepend default Cyana file extension below
fileTypes = [ FileType('Python', ['*.py']), ]
popup = FileSelectPopup(self, file_types = fileTypes,
title='Python file', dismiss_text='Cancel',
selected_file_must_exist = True)
fileName = popup.getFile()
self.validScriptEntry.set(fileName)
popup.destroy()
def selectFile(self):
spectrum = self.spectrum
if spectrum:
directory = spectrum.analysisSpectrum.contourDir.dataLocation
else:
directory = os.getcwd()
popup = FileSelectPopup(self, directory=directory)
fileName = popup.getFile()
popup.destroy()
if fileName:
self.file_entry.set(fileName)
def chooseZipFile(self):
fileTypes = [ FileType('Zip', ['*.zip']), ]
popup = FileSelectPopup(self, file_types=fileTypes, file=self.resultFileEntry.get(),
title='Results zip file location', dismiss_text='Cancel',
selected_file_must_exist=False)
fileName = popup.getFile()
if fileName:
self.resultFileEntry.set(fileName)
popup.destroy()
def chooseFile(self):
directory = self.directoryEntry.get()
if not directory:
directory = os.getcwd()
popup = FileSelectPopup(self, directory=directory)
file = popup.getFile()
popup.destroy()
if file:
template = os.path.basename(file)
self.templateEntry.set(template)
def chooseCyanaFile(self):
# Prepend default Cyana file extension below
fileTypes = [ FileType('All', ['*']), ]
popup = FileSelectPopup(self, file_types = fileTypes,
title = 'CYANA fproject file', dismiss_text = 'Cancel',
selected_file_must_exist = True)
fileName = popup.getFile()
self.cyanaEntry.set(fileName)
self.projOptionsSelect.setIndex(3)
popup.destroy()
def selectCnsExe(self):
fileTypes = [ FileType("All", ["*"]), FileType("EXE", ["*.exe"]) ]
popup = FileSelectPopup(self, fileTypes)
file = popup.getFile()
if file:
self.cnsExeEntry.set( file )
popup.destroy()
self.updateEntryParams()
def askFile(title, prompt, initial_value = '', parent = None,
dismiss_text='Cancel', extra_dismiss_text = ''):
if (parent):
popup = FileSelectPopup(parent, title=title, prompt=prompt, show_file=True,
dismiss_text=dismiss_text,
extra_dismiss_text=extra_dismiss_text,
file=initial_value)
file = popup.getFile()
popup.destroy()
return file
else:
return askString(title, prompt, initial_value)
def readPCSFileChange(self):
fileTypes = [FileType('PCS', ['*.npc']), FileType('All', ['*'])]
fileSelectPopup = FileSelectPopup(
self,
file=os.path.basename(self.calPCSEntry.get()),
directory=os.path.dirname(self.calPCSEntry.get()),
file_types=fileTypes,
title='Save PCS values to file',
dismiss_text='Cancel',
selected_file_must_exist=False,
multiSelect=False,
)
self.calPCSEntry.set(fileSelectPopup.getFile())
def writeScriptFileChange(self):
fileTypes = [FileType('python', ['*.py']), FileType('All', ['*'])]
fileSelectPopup = FileSelectPopup(
self,
file=os.path.basename(self.scriptEntry.get()),
directory=os.path.dirname(self.scriptEntry.get()),
file_types=fileTypes,
title='Paramagpy script',
dismiss_text='Cancel',
selected_file_must_exist=False,
multiSelect=False,
)
self.scriptEntry.set(fileSelectPopup.getFile())
def printCanvas(self, *event):
fileTypes = [ FileType('PostScript', ['*.ps']), FileType('All', ['*'])]
fileSelectPopup = FileSelectPopup(self, file_types = fileTypes,
title = 'Print canvas to file', dismiss_text = 'Cancel',
selected_file_must_exist = False)
fileName = fileSelectPopup.getFile()
self.bbox = bbox = self.canvas.bbox('all')
w = bbox[2] - bbox[0]
h = bbox[3] - bbox[1]
self.canvas.postscript(colormode='color',file=fileName,
x=bbox[0], y=bbox[1], width=w+2,
pagewidth='21.c', height=h+2, fontmap='fontmap')
def printCanvas(self, *event):
from memops.gui.FileSelect import FileType
from memops.gui.FileSelectPopup import FileSelectPopup
fileTypes = [FileType('PostScript', ['*.ps']), FileType('All', ['*'])]
fileSelectPopup = FileSelectPopup(self,
file_types=fileTypes,
title='Print canvas to file',
dismiss_text='Cancel',
selected_file_must_exist=False)
fileName = fileSelectPopup.getFile()
self.postscript(colormode='color', file=fileName)
def findFile(self):
format = self.format_menu.getSelected()
if (format == tabFormat):
file_types = [FileType('All', ['*']), FileType('Text', ['*.txt'])]
else:
file_types = [FileType('All', ['*']), FileType('CSV', ['*.csv'])]
popup = FileSelectPopup(self, file_types=file_types)
file = popup.getFile()
if (file):
self.file_entry.set(file)
popup.destroy()
def chooseCyanaFile(self):
# Prepend default Cyana file extension below
fileTypes = [
FileType('All', ['*']),
]
popup = FileSelectPopup(self,
file_types=fileTypes,
title='CYANA fproject file',
dismiss_text='Cancel',
selected_file_must_exist=True)
fileName = popup.getFile()
self.cyanaEntry.set(fileName)
self.projOptionsSelect.setIndex(3)
popup.destroy()
def getFile(self, widget, obj, attrName):
# No widget because table placement is irrelevent and setting process not interruptable
file_types = [
FileType("All", ["*"]),
]
value = getattr(obj, attrName)
if not os.path.exists(value):
value = None
popup = FileSelectPopup(self,
file_types,
file=value,
dismiss_text='Cancel')
value = popup.getFile()
if value:
setattr(obj, attrName, value)
def exportText(self, *event):
from memops.gui.FileSelect import FileType
from memops.gui.FileSelectPopup import FileSelectPopup
if self.textOut:
fileTypes = [FileType('Text', ['*.txt']), FileType('All', ['*'])]
fileSelectPopup = FileSelectPopup(self,
file_types=fileTypes,
title='Save table as plain text',
dismiss_text='Cancel',
selected_file_must_exist=False)
fileName = fileSelectPopup.getFile()
if fileName:
file = open(fileName, 'w')
file.write(self.textOut)
def getFileName(self, title, fileTypes):
fileSelectPopup = FileSelectPopup(
self,
file_types=fileTypes,
title=title,
dismiss_text='Cancel',
selected_file_must_exist=True,
multiSelect=False,
)
self.fileName = fileSelectPopup.getFile()
if not self.fileName:
showWarning('Failure',
'Please select an existing file.',
parent=self)
return False
return True
def chooseCcpnFile(self):
fileTypes = [ FileType('XML', ['*.xml']), FileType('All', ['*'])]
popup = FileSelectPopup(self, file_types = fileTypes,
title = 'CCPN project XML file', dismiss_text = 'Cancel',
selected_file_must_exist = True)
fileName = popup.getFile()
if len(fileName)>0:
self.pdbEntry.configure(state='normal')
self.pdbEntry.set(fileName)
self.projOptionsSelect.setIndex(1)
_dir,name,dummy = nTpath( fileName )
self.nameEntry.set(name)
#end if
self.ccpnEntry.set(fileName)
self.projOptionsSelect.setIndex(2)
popup.destroy()
def choosePdbFile(self):
fileTypes = [ FileType('PDB', ['*.pdb']), FileType('All', ['*'])]
popup = FileSelectPopup(self, file_types = fileTypes,
title = 'PDB file', dismiss_text = 'Cancel',
selected_file_must_exist = True)
fileName = popup.getFile()
if len(fileName)>0:
# Put text into entry widget
self.pdbEntry.configure(state='normal')
self.pdbEntry.set(fileName)
# Put text into name widget
_dir,name,dummy = nTpath( fileName )
self.nameEntry.configure(state='normal')
self.nameEntry.set(name)
# choose the correct radiobutton
self.projOptionsSelect.setIndex(1)
self.updateGui()
#end if
popup.destroy()
def getDataPath(self, *extra):
if self.dataStore:
dataStore = self.dataStore
oldPath = dataStore.fullPath
directory = file = ''
if os.path.exists(oldPath):
file = oldPath
elif os.path.exists(os.path.dirname(oldPath)):
directory = os.path.dirname(oldPath)
elif os.path.exists(dataStore.dataUrl.url.dataLocation):
directory = dataStore.dataUrl.url.dataLocation
popup = FileSelectPopup(self, file=file, directory=directory)
newPath = popup.getFile()
popup.destroy()
if newPath and oldPath != newPath:
pp = dataStore.dataUrl.url.dataLocation
if newPath.startswith(pp):
newPath = newPath[len(pp) + 1:]
dataStore.path = newPath
else:
setDataStoreFileName(dataStore, newPath)
def chooseCcpnFile(self):
fileTypes = [FileType('XML', ['*.xml']), FileType('All', ['*'])]
popup = FileSelectPopup(self,
file_types=fileTypes,
title='CCPN project XML file',
dismiss_text='Cancel',
selected_file_must_exist=True)
fileName = popup.getFile()
if len(fileName) > 0:
self.pdbEntry.configure(state='normal')
self.pdbEntry.set(fileName)
self.projOptionsSelect.setIndex(1)
_dir, name, dummy = nTpath(fileName)
self.nameEntry.set(name)
#end if
self.ccpnEntry.set(fileName)
self.projOptionsSelect.setIndex(2)
popup.destroy()
def saveText(self):
fileSelectPopup = FileSelectPopup(
self,
file_types=[
FileType('Text', ['*.txt', '*.out', '*.text']),
FileType('All', ['*'])
],
title='Select text output save file')
fileName = fileSelectPopup.getFile()
if fileName:
writeFile = 1
if os.path.exists(fileName):
if not showYesNo('Overwrite file',
'Overwrite existing file %s?' % fileName):
writeFile = 0
if writeFile:
fout = open(fileName, 'w')
fout.write(self.text.text_area.getText())
fout.close()
def selectModuleBvExport(self):
""" Choose the Back Value file that was exported from MODULE. """
file_types = [
FileType("Back Value files", ["*.back"]),
FileType("All files", ["*"])
]
popup = FileSelectPopup(self,
file_types,
dismiss_text='Cancel',
show_file=True)
chosenBvFile = popup.getFile()
if os.path.isfile(chosenBvFile):
moduleBackValueFile = chosenBvFile
else:
warnPopup = showWarning('MODULE',
'File %s not found.' % chosenBvFile,
parent=self)
return None
popup.destroy()
return moduleBackValueFile
def choosePdbFile(self):
fileTypes = [FileType('PDB', ['*.pdb']), FileType('All', ['*'])]
popup = FileSelectPopup(self,
file_types=fileTypes,
title='PDB file',
dismiss_text='Cancel',
selected_file_must_exist=True)
fileName = popup.getFile()
if len(fileName) > 0:
# Put text into entry widget
self.pdbEntry.configure(state='normal')
self.pdbEntry.set(fileName)
# Put text into name widget
_dir, name, dummy = nTpath(fileName)
self.nameEntry.configure(state='normal')
self.nameEntry.set(name)
# choose the correct radiobutton
self.projOptionsSelect.setIndex(1)
self.updateGui()
#end if
popup.destroy()
class PrintFrame(LabelFrame):
def __init__(self,
parent,
getOption=None,
setOption=None,
text='Print Options',
haveTicks=False,
doOutlineBox=True,
*args,
**kw):
self.getOption = getOption
self.setOption = setOption
self.haveTicks = haveTicks
self.doOutlineBox = doOutlineBox
LabelFrame.__init__(self, parent=parent, text=text, *args, **kw)
self.file_select_popup = None
self.getOptionValues()
try:
size_index = self.paper_types.index(self.paper_type)
except:
size_index = 0
try:
other_unit_index = self.paper_units.index(self.other_unit)
except:
other_unit_index = 0
try:
orientation_index = self.paper_orientations.index(self.orientation)
except:
orientation_index = 0
try:
style_index = self.style_choices.index(self.output_style)
except:
style_index = 0
try:
format_index = self.format_choices.index(self.output_format)
except:
format_index = 0
if haveTicks:
try:
tick_location_index = self.tick_locations.index(
self.tick_location)
except:
tick_location_index = 0
self.grid_columnconfigure(2, weight=1)
row = 0
label = Label(self, text='File:')
label.grid(row=row, column=0, sticky='e')
self.file_entry = Entry(self, width=40, text=self.file_name)
self.file_entry.grid(row=row, column=1, columnspan=2, sticky='ew')
button = Button(self, text='Choose File', command=self.findFile)
button.grid(row=row, column=3, rowspan=2, sticky='nsew')
row += 1
label = Label(self, text='Title:')
label.grid(row=row, column=0, sticky='e')
self.title_entry = Entry(self, width=40, text=self.title)
self.title_entry.grid(row=row, column=1, columnspan=2, sticky='ew')
row += 1
frame = Frame(self)
frame.grid(row=row, column=0, columnspan=4, sticky='ew')
frame.grid_columnconfigure(4, weight=1)
label = Label(frame, text='Paper size:')
label.grid(row=0, column=0, sticky='e')
entries = []
for t in paper_types:
if t == Output.other_paper_type:
entry = t
else:
(w, h, u) = paper_sizes[t]
entry = t + ' (%2.1f %s x %2.1f %s)' % (w, u, h, u)
entries.append(entry)
self.size_menu = PulldownList(frame,
callback=self.changedSize,
texts=entries,
index=size_index)
self.size_menu.grid(row=0, column=1, sticky='w')
self.other_frame = Frame(frame)
self.other_frame.grid_columnconfigure(0, weight=1)
self.other_entry = FloatEntry(self.other_frame,
text=self.other_size,
isArray=True)
self.other_entry.grid(row=0, column=0, sticky='ew')
self.other_unit_menu = PulldownList(self.other_frame,
texts=paper_units,
index=other_unit_index)
self.other_unit_menu.grid(row=0, column=1, sticky='ew')
row += 1
frame = Frame(self)
frame.grid(row=row, column=0, columnspan=4, sticky='ew')
frame.grid_columnconfigure(1, weight=1)
frame.grid_columnconfigure(3, weight=1)
frame.grid_columnconfigure(5, weight=1)
label = Label(frame, text='Orientation:')
label.grid(row=0, column=0, sticky='e')
self.orientation_menu = PulldownList(frame,
texts=paper_orientations,
index=orientation_index)
self.orientation_menu.grid(row=0, column=1, sticky='w')
label = Label(frame, text=' Style:')
label.grid(row=0, column=2, sticky='e')
self.style_menu = PulldownList(frame,
texts=style_choices,
index=style_index)
self.style_menu.grid(row=0, column=3, sticky='w')
label = Label(frame, text=' Format:')
label.grid(row=0, column=4, sticky='e')
self.format_menu = PulldownList(frame,
callback=self.changedFormat,
texts=format_choices,
index=format_index)
self.format_menu.grid(row=0, column=5, sticky='w')
if haveTicks:
row += 1
frame = Frame(self)
frame.grid(row=row, column=0, columnspan=4, sticky='ew')
frame.grid_columnconfigure(1, weight=1)
frame.grid_columnconfigure(3, weight=1)
label = Label(frame, text='Tick Location:')
label.grid(row=0, column=0, sticky='e')
self.tick_menu = PulldownList(frame,
texts=tick_locations,
index=tick_location_index)
self.tick_menu.grid(row=0, column=1, sticky='w')
label = Label(frame, text=' Tick Placement:')
label.grid(row=0, column=2, sticky='e')
self.tick_buttons = CheckButtons(frame,
entries=tick_placements,
selected=self.tick_placement)
self.tick_buttons.grid(row=0, column=3, sticky='w')
row += 1
frame = Frame(self)
frame.grid(row=row, column=0, columnspan=4, sticky='ew')
frame.grid_columnconfigure(3, weight=1)
label = Label(frame, text='Include:')
label.grid(row=0, column=0, sticky='e')
self.border_buttons = CheckButtons(frame,
entries=border_decorations,
selected=self.border_decoration)
self.border_buttons.grid(row=0, column=1, sticky='w')
label = Label(frame, text=' Scaling:')
label.grid(row=0, column=2, sticky='e')
self.scaling_scale = Scale(frame,
orient=Tkinter.HORIZONTAL,
value=self.scaling)
self.scaling_scale.grid(row=0, column=3, sticky='ew')
def destroy(self):
self.setOptionValues()
if self.file_select_popup:
self.file_select_popup.destroy()
Frame.destroy(self)
def getOptionValues(self):
getOption = self.getOption
if getOption:
file_name = getOption('FileName', defaultValue='')
title = getOption('Title', defaultValue='')
paper_type = getOption('PaperSize', defaultValue=paper_types[0])
paper_type = paper_type_dict.get(paper_type, paper_types[0])
other_height = getOption('OtherHeight', defaultValue=10)
other_width = getOption('OtherWidth', defaultValue=10)
other_size = [other_height, other_width]
other_unit = getOption('OtherUnit', defaultValue=paper_units[0])
orientation = getOption('Orientation',
defaultValue=paper_orientations[0])
in_color = getOption('InColor', defaultValue=True)
if in_color:
output_style = style_choices[0]
else:
output_style = style_choices[1]
format_option = getOption('OutputFormat',
defaultValue=format_options[0])
output_format = format_choices[format_options.index(format_option)]
if self.haveTicks:
tick_outside = getOption('TickOutside',
defaultValue=tick_locations[0])
if tick_outside:
tick_location = tick_locations.index(PrintTicks.Outside)
else:
tick_location = tick_locations.index(PrintTicks.Inside)
tick_placement = getTickPlacement1(
getOption('TickPlacement', defaultValue='nsew'))
dateTime = getOption('ShowsDateTime', defaultValue=True)
fileName = getOption('ShowsFileName', defaultValue=True)
border_decoration = []
if dateTime:
border_decoration.append(border_decorations[0])
if fileName:
border_decoration.append(border_decorations[1])
scaling = getOption('Scaling', defaultValue=0.9)
scaling = int(round(100.0 * scaling))
else:
file_name = ''
title = ''
paper_type = paper_types[0]
other_unit = paper_units[0]
other_size = ''
orientation = paper_orientations[0]
output_style = style_choices[0]
output_format = format_choices[0]
if self.haveTicks:
tick_location = tick_locations[0]
tick_placement = tick_placements
border_decoration = border_decorations
scaling = 90
if not self.haveTicks:
tick_location = None
tick_placement = None
self.file_name = file_name
self.title = title
self.paper_type = paper_type
self.other_unit = other_unit
self.other_size = other_size
self.orientation = orientation
self.output_style = output_style
self.output_format = output_format
self.tick_location = tick_location
self.tick_placement = tick_placement
self.border_decoration = border_decoration
self.scaling = scaling
def setOptionValues(self):
self.file_name = file_name = self.file_entry.get()
self.title = title = self.title_entry.get()
n = self.size_menu.getSelectedIndex()
self.paper_type = paper_type = paper_types[n]
if paper_type == Output.other_paper_type:
other_size = self.other_entry.get()
other_unit = self.other_unit_menu.getText()
else:
other_size = None
other_unit = None
self.other_size = other_size
self.other_unit = other_unit
self.paper_orientation = paper_orientation = self.orientation_menu.getText(
)
self.output_style = output_style = self.style_menu.getText()
self.output_format = output_format = self.format_menu.getText()
if self.haveTicks:
tick_location = self.tick_menu.getText()
tick_placement = self.tick_buttons.getSelected()
else:
tick_location = tick_placement = None
self.tick_location = tick_location
self.tick_placement = tick_placement
self.border_decoration = border_decoration = self.border_buttons.getSelected(
)
scaling = self.scaling_scale.get()
self.scaling = scaling = int(round(scaling))
setOption = self.setOption
if setOption:
setOption('FileName', value=file_name)
setOption('Title', value=title)
if paper_type == Output.other_paper_type:
setOption('OtherHeight', value=other_size[0])
setOption('OtherWidth', value=other_size[1])
setOption('OtherUnit', value=other_unit)
else:
paper_type = paper_type_inverse_dict[paper_type]
setOption('PaperSize', value=paper_type)
setOption('Orientation', value=paper_orientation)
in_color = (output_style == style_choices[0])
setOption('InColor', value=in_color)
output_format = format_options[format_choices.index(output_format)]
setOption('OutputFormat', value=output_format)
if self.haveTicks:
tick_outside = (tick_location == PrintTicks.Outside)
setOption('TickOutside', value=tick_outside)
tick_placement = getTickPlacement2(tick_placement)
setOption('TickPlacement', value=tick_placement)
dateTime = (border_decorations[0] in border_decoration)
fileName = (border_decorations[1] in border_decoration)
setOption('ShowsDateTime', value=dateTime)
setOption('ShowsFileName', value=fileName)
setOption('Scaling', value=0.01 * scaling)
def findFile(self):
if self.file_select_popup:
self.file_select_popup.open()
else:
file_types = [
FileType('All', ['*']),
FileType('PostScript', ['*.ps', '*.eps']),
FileType('PDF', ['*.pdf', '*.ai'])
]
self.file_select_popup = FileSelectPopup(self,
file_types=file_types)
file = self.file_select_popup.getFile()
if file:
self.file_entry.set(file)
def changedSize(self, entry):
if entry == Output.other_paper_type:
self.other_frame.grid(row=0, column=2, columnspan=2, sticky='w')
else:
self.other_frame.grid_forget()
def changedFormat(self, entry):
file_suffix = file_suffixes.get(entry)
if not file_suffix:
return
file_name = self.file_entry.get()
if not file_name:
return
for suffix in format_suffixes:
if file_name.endswith(suffix):
if suffix != file_suffix:
n = len(suffix)
file_name = file_name[:-n] + file_suffix
self.file_entry.set(file_name)
break
else:
file_name = file_name + file_suffix
self.file_entry.set(file_name)
# width and height are of plot, in pixels
def getOutputHandler(self, width, height, fonts=None):
if not fonts:
fonts = []
else:
fonts = list(fonts)
for n in range(len(fonts)):
if fonts[n] == 'Times':
fonts[n] = 'Times-Roman'
self.setOptionValues()
if not self.file_name:
showError('No file', 'No file specified', parent=self)
return None
if os.path.exists(self.file_name):
if not showYesNo('File exists',
'File "%s" exists, overwrite?' % self.file_name,
parent=self):
return None
if (self.paper_type == Output.other_paper_type):
paper_size = self.other_size + [self.other_unit]
else:
paper_size = paper_sizes[self.paper_type]
output_scaling = self.scaling / 100.0
font = 'Times-Roman'
border_text = {}
for decoration in self.border_decoration:
if (decoration == 'Time & date'):
location = 'se'
text = time.ctime(time.time())
elif (decoration == 'File name'):
location = 'sw'
text = self.file_name
else:
continue # should not be here
border_text[location] = (text, font, 12)
if (self.title):
location = 'n'
border_text[location] = (self.title, font, 18)
if font not in fonts:
fonts.append(font)
outputHandler = PrintHandler.getOutputHandler(
self.file_name,
width,
height,
output_scaling=output_scaling,
paper_size=paper_size,
paper_orientation=self.paper_orientation,
output_style=self.output_style,
output_format=self.output_format,
border_text=border_text,
fonts=fonts,
do_outline_box=self.doOutlineBox)
return outputHandler
def getAspectRatio(self):
self.setOptionValues()
if self.paper_type == Output.other_paper_type:
paper_size = self.other_size
else:
paper_size = paper_sizes[self.paper_type]
r = paper_size[1] / paper_size[0]
if self.paper_orientation == 'Landscape':
r = 1.0 / r
return r