class LabeledEntry(Frame):
def __init__(self,
parent,
label,
entry='',
separator=': ',
label_width=20,
entry_width=60,
label_anchor=Tkinter.E,
show='',
returnCallback=None,
tipText=None,
*args,
**kw):
apply(Frame.__init__, (self, parent) + args, kw)
self.grid_columnconfigure(1, weight=1)
self.separator = separator
text = label + separator
self.label = Label(self,
text=text,
width=label_width,
anchor=label_anchor)
self.label.grid(row=0, column=0, sticky=Tkinter.EW)
self.entry = Entry(self,
show=show,
width=entry_width,
returnCallback=returnCallback,
tipText=tipText)
self.entry.insert(0, entry)
self.entry.grid(row=0, column=1, sticky=Tkinter.EW)
def getLabel(self):
text = self.label.get()
n = text.find(self.separator)
if (n >= 0):
text = text[:n]
return text
def setLabel(self, text=''):
text = text + self.separator
self.label.set(text)
def getEntry(self):
return self.entry.get()
def setEntry(self, text=''):
self.entry.set(text)
def body(self, frame):
label = Label(frame, text=self.prompt)
if self.hide:
entry = Entry(frame, text=self.initVal, bg='white', show='*')
else:
entry = Entry(frame, text=self.initVal, bg='white')
label.grid(row=0, column=0, sticky=Tkinter.W)
entry.grid(row=1, column=0, sticky=Tkinter.W)
entry.focus_set() # for the keypresses
return entry
class CloudHomologueAssignPopup(BasePopup):
def __init__(self, parent, *args, **kw):
self.guiParent = parent
self.project = parent.getProject()
self.molSystem = None
self.chain = None
self.assignment = None
self.scores = []
BasePopup.__init__(self, parent, title="Cloud Threader", **kw)
def body(self, guiFrame):
guiFrame.grid_columnconfigure(3, weight=1)
row = 0
label = Label(guiFrame, text='Molecular system: ')
label.grid(row=row, column=0, sticky=Tkinter.NW)
self.molSysPulldown = PulldownMenu(guiFrame, self.changeMolSystem, selected_index=-1, do_initial_callback=0)
self.molSysPulldown.grid(row=row, column=1, sticky=Tkinter.NW)
label = Label(guiFrame, text='Clouds files: ')
label.grid(row=row, column=2, sticky=Tkinter.NW)
self.filenameEntry = Entry(guiFrame,text='perfect00.pdb')
self.filenameEntry.grid(row=row, column=3, sticky=Tkinter.NW)
row += 1
label = Label(guiFrame, text='Chain: ')
label.grid(row=row, column=0, sticky=Tkinter.NW)
self.chainPulldown = PulldownMenu(guiFrame, self.changeChain, selected_index=-1, do_initial_callback=0)
self.chainPulldown.grid(row=row, column=1, sticky=Tkinter.NW)
label = Label(guiFrame, text='Thread steps: ')
label.grid(row=row, column=2, sticky=Tkinter.NW)
self.numStepsEntry = IntEntry(guiFrame,text=3000)
self.numStepsEntry.grid(row=row, column=3, sticky=Tkinter.NW)
row += 1
label = Label(guiFrame, text='Homologue PDB file: ')
label.grid(row=row, column=0, sticky=Tkinter.NW)
self.pdbEntry = Entry(guiFrame,text='')
self.pdbEntry.grid(row=row, column=1, sticky=Tkinter.NW)
label = Label(guiFrame, text='Dist. Threshold: ')
label.grid(row=row, column=2, sticky=Tkinter.NW)
self.distEntry = FloatEntry(guiFrame,text=3.0)
self.distEntry.grid(row=row, column=3, sticky=Tkinter.NW)
row += 1
label = Label(guiFrame, text='Global score: ')
label.grid(row=row, column=0, sticky=Tkinter.NW)
self.globalScoreLabel = Label(guiFrame, text='')
self.globalScoreLabel.grid(row=row, column=1, sticky=Tkinter.NW)
label = Label(guiFrame, text='Assignment Threshold: ')
label.grid(row=row, column=2, sticky=Tkinter.NW)
self.thresholdEntry = FloatEntry(guiFrame,text=-4.5)
self.thresholdEntry.grid(row=row, column=3, sticky=Tkinter.NW)
row += 1
guiFrame.grid_rowconfigure(row, weight=1)
self.graph = ScrolledGraph(guiFrame, width=300, height=200)
self.graph.grid(row=row, column=0, columnspan=4, sticky = Tkinter.NSEW)
row += 1
texts = ['Run','Assign!']
commands = [self.run, self.assignSpinSystems]
bottomButtons = createDismissHelpButtonList(guiFrame,texts=texts,commands=commands,expands=0,help_url=None)
bottomButtons.grid(row=row, column=0, columnspan=4, sticky=Tkinter.EW)
self.assignButton = bottomButtons.buttons[1]
for func in ('__init__','delete'):
Implementation.registerNotify(self.updateMolSystems, 'ccp.molecule.MolSystem.MolSystem', func)
Implementation.registerNotify(self.updateChains, 'ccp.molecule.MolSystem.Chain', func)
self.updateMolSystems()
self.updateChains()
def update(self):
if self.assignment and self.scores:
self.assignButton.enable()
else:
self.assignButton.disable()
def run(self):
if self.chain:
pattern = self.filenameEntry.get()
nSteps = self.numStepsEntry.get() or 4000
pdbFile = self.pdbEntry.get()
dist = self.distEntry.get() or 3.0
pgb = ProgressBar(self, text='Searching', total=nSteps)
files = getFileNamesFromPattern(pattern , '.')
if not files:
return
clouds = getCloudsFromFile(files, self.chain.root)
score, self.scores, self.assignment = cloudHomologueAssign(self.chain, clouds, pdbFile, dist, nSteps, self.graph, pgb)
pgb.destroy()
self.globalScoreLabel.set(str(score))
self.update()
def assignSpinSystems(self):
if self.assignment and self.scores:
if showWarning('Query','Are you sure?'):
threshold = self.thresholdEntry.get() or -4.0
i = 0
for residue in self.assignment.keys():
if self.scores[residue] > threshold:
spinSystem = self.assignment[residue]
assignSpinSystemResidue(spinSystem,residue=None)
for residue in self.assignment.keys():
if self.scores[residue] > threshold:
i += 1
spinSystem = self.assignment[residue]
assignSpinSystemResidue(spinSystem,residue=residue)
showWarning('Done','%d residues assigned' % i)
def getMolSystems(self):
names = []
for molSystem in self.project.molSystems:
if molSystem.chains:
names.append( '%s' % (molSystem.code) )
return names
def changeMolSystem(self, i, name):
self.molSystem = self.project.findFirstMolSystem(code=name)
def updateMolSystems(self, *opt):
names = self.getMolSystems()
if names:
if not self.molSystem:
self.molSystem = self.project.findFirstMolSystem(code=names[0])
self.molSysPulldown.setup(names, names.index(self.molSystem.code))
def getChains(self):
chains = []
if self.molSystem:
for chain in self.molSystem.chains:
chains.append( [chain.code, chain] )
return chains
def changeChain(self, i, name=None):
if not name:
i = self.chainPulldown.selected_index
chains = self.getChains()
if chains:
self.chain = chains[i][1]
def updateChains(self, *chain):
chains = self.getChains()
if chains:
names = [x[0] for x in chains]
if (not self.chain) or (self.chain.code not in names):
self.chain = chains[0][1]
self.chainPulldown.setup(names, names.index(self.chain.code) )
self.update()
def destroy(self):
for func in ('__init__','delete'):
Implementation.unregisterNotify(self.updateMolSystems, 'ccp.molecule.MolSystem.MolSystem', func)
Implementation.unregisterNotify(self.updateChains, 'ccp.molecule.MolSystem.Chain', func)
BasePopup.destroy(self)
class EditPeakPopup(BasePopup):
"""
**Edit Position, Intensity & Details for a Peak**
This popup window provides an means of editing peak information as an
alternative to editing values in the main peak tables. This popup is also used
to specify parameters for when a new peak is explicitly added to a peak list
using a tabular display.
The user can specify the position of the peak's dimensions in ppm, Hz or data
point units. Also, the user can adjust the height and volume peak intensity
values and a textual "Details" field, ,which can carry the user's comments
about the peak.
When editing an existing peak, no changes are made to the peak until the
[Update] button is pressed. Likewise for a new peak the [Add Peak] button
commits the changes. If the popup window is closed before the changes are
committed then the entire editing or peak addition operation is cancelled.
"""
def __init__(self, parent, peak=None, peakList=None, *args, **kw):
self.titleColor = '#000080'
self.numDims = 0
self.peak = peak
kw['borderwidth'] = 6
BasePopup.__init__(self, parent=parent, title='Edit Peak', **kw)
self.registerNotify(self.deletedPeak, 'ccp.nmr.Nmr.Peak', 'delete')
for func in ('setAnnotation', 'setDetails', 'setFigOfMerit'):
self.registerNotify(self.updatePeak, 'ccp.nmr.Nmr.Peak', func)
for func in ('setAnnotation', 'setPosition', 'setNumAliasing'):
self.registerNotify(self.updatePeak, 'ccp.nmr.Nmr.PeakDim', func)
for func in ('__init__', 'delete', 'setValue'):
self.registerNotify(self.updatePeak, 'ccp.nmr.Nmr.PeakIntensity',
func)
self.dimensionLabels = []
self.dimensionEntries = []
self.update(self.peak, peakList)
def body(self, guiParent):
self.geometry("+150+150")
guiParent.grid_columnconfigure(0, weight=1)
self.master_frame = guiParent
units = ('ppm', 'point', 'Hz')
self.unit = 'ppm'
self.specLabel = Label(guiParent,
fg=self.titleColor,
grid=(0, 0),
sticky='ew')
self.peakLabel = Label(guiParent, grid=(0, 1), sticky='ew')
self.unit_frame = frame = Frame(guiParent,
grid=(1, 1),
gridSpan=(1, 2))
self.unitLabel = Label(frame, text='Current units: ', grid=(0, 0))
tipText = 'Selects which unit of measurement to display peak dimension positions with'
self.unitSelect = PulldownList(frame,
callback=self.changeUnit,
texts=units,
grid=(0, 1),
tipText=tipText)
self.heightLabel = Label(guiParent,
text='Height',
borderwidth=2,
relief='groove')
tipText = 'Sets the peak height; the value of the spectrum point intensity (albeit often interpolated)'
self.heightEntry = FloatEntry(guiParent,
borderwidth=1,
tipText=tipText)
self.volumeLabel = Label(guiParent,
text='Volume',
borderwidth=2,
relief='groove')
tipText = 'Sets the peak volume integral; normally a summation of data point values'
self.volumeEntry = FloatEntry(guiParent,
borderwidth=1,
tipText=tipText)
self.detailLabel = Label(guiParent,
text='Details',
borderwidth=2,
relief='groove')
tipText = 'A user-configurable textual comment for the peak, which appears an tables and occasionally on spectrum displays'
self.detailEntry = Entry(guiParent, borderwidth=1, tipText=tipText)
tipTexts = [
'Commits the specified values to update the peak and closes the popup',
]
texts = ['Update']
commands = [self.commit]
self.buttons = UtilityButtonList(guiParent,
texts=texts,
commands=commands,
doClone=False,
helpUrl=self.help_url,
tipTexts=tipTexts)
def open(self):
self.updatePeak()
BasePopup.open(self)
def updatePeak(self, object=None):
peak = None
if object:
if object.className == 'Peak':
peak = object
elif object.className == 'PeakDim':
peak = object.peak
elif object.className == 'PeakIntensity':
peak = object.peak
if (peak is None) or (peak is self.peak):
self.update(peak=self.peak)
def update(self, peak=None, peakList=None):
# first destroy old labels and entries (saves grid hassles)
for label in self.dimensionLabels:
label.destroy()
for entry in self.dimensionEntries:
entry.destroy()
# now setup required data
if peak:
title = 'Edit Peak'
self.buttons.buttons[0].config(text='Update')
else:
title = 'Add Peak'
self.buttons.buttons[0].config(text='Add Peak')
self.setTitle(title)
self.peak = peak
self.peakList = peakList
if not peakList:
if peak:
self.peakList = peak.peakList
else:
return
peakList = self.peakList
spectrum = peakList.dataSource.name
self.numDims = peakList.dataSource.numDim
self.posn = self.numDims * [0]
self.dataDims = peakList.dataSource.sortedDataDims()
if self.peak:
serial = self.peak.serial
dims = self.peak.sortedPeakDims()
details = self.peak.details
if not details:
details = ''
if self.peak.annotation:
annotn = '%0.16s' % self.peak.annotation
else:
annotn = ''
heightIntensity = self.peak.findFirstPeakIntensity(
intensityType='height')
volumeIntensity = self.peak.findFirstPeakIntensity(
intensityType='volume')
if heightIntensity:
height = heightIntensity.value
else:
height = 0.0
if volumeIntensity:
volume = volumeIntensity.value
else:
volume = 0.0
for i in range(self.numDims):
peakDim = dims[i]
dataDimRef = peakDim.dataDimRef
if dataDimRef:
self.posn[i] = peakDim.position + (
peakDim.numAliasing * dataDimRef.dataDim.numPointsOrig)
else:
self.posn[i] = peakDim.position
else:
dict = peakList.__dict__.get('serialDict')
if dict is None:
serial = 1
else:
serial = dict.get('peaks', 0) + 1
height = 0.0
volume = 0.0
details = ''
annotn = ''
self.specLabel.set(
text='Experiment: %s Spectrum: %s PeakList: %d' %
(peakList.dataSource.experiment.name, spectrum, peakList.serial))
self.peakLabel.set(text='Peak: %d' % serial)
self.dimensionLabels = self.numDims * ['']
self.dimensionEntries = self.numDims * ['']
for i in range(self.numDims):
pos = self.posn[i]
if self.unit != 'point':
dataDim = self.dataDims[i]
if dataDim.className == 'FreqDataDim':
pos = unit_converter[('point', self.unit)](
pos, getPrimaryDataDimRef(dataDim))
self.dimensionLabels[i] = Label(self.master_frame,
text='F%d' % (i + 1),
borderwidth=2,
relief='groove')
tipText = 'The peak position in dimension %d, in the specified units' % (
i + 1)
self.dimensionEntries[i] = FloatEntry(self.master_frame,
borderwidth=1,
text='%8.4f' % pos,
tipText=tipText)
self.heightEntry.set(text='%f' % height)
self.volumeEntry.set(text='%f' % volume)
self.detailEntry.set(text=details)
row = 0
self.specLabel.grid(row=row, column=0, columnspan=2, sticky='nsew')
row = row + 1
self.peakLabel.grid(row=row, column=0, sticky='nsew')
self.unit_frame.grid(row=row, column=1, columnspan=2, sticky='nsew')
for i in range(self.numDims):
row = row + 1
self.dimensionLabels[i].grid(row=row, column=0, sticky='nsew')
self.dimensionEntries[i].grid(row=row,
column=1,
columnspan=3,
sticky='e')
row = row + 1
self.heightLabel.grid(row=row, column=0, sticky='nsew')
self.heightEntry.grid(row=row, column=1, columnspan=3, sticky='e')
row = row + 1
self.volumeLabel.grid(row=row, column=0, sticky='nsew')
self.volumeEntry.grid(row=row, column=1, columnspan=3, sticky='e')
row = row + 1
self.detailLabel.grid(row=row, column=0, sticky='nsew')
self.detailEntry.grid(row=row, column=1, columnspan=3, sticky='e')
row = row + 1
self.buttons.grid(row=row, column=0, columnspan=4, sticky='nsew')
def changeUnit(self, unit):
posDisp = self.numDims * [None]
for i in range(self.numDims):
posDisp[i] = float(self.dimensionEntries[i].get())
if self.unit != 'point':
dataDim = self.dataDims[i]
if dataDim.className == 'FreqDataDim':
posDisp[i] = unit_converter[(self.unit, 'point')](
posDisp[i], getPrimaryDataDimRef(dataDim))
self.unit = unit
if self.unit != 'point':
for i in range(self.numDims):
dataDim = self.dataDims[i]
if dataDim.className == 'FreqDataDim':
posDisp[i] = unit_converter[('point', self.unit)](
posDisp[i], getPrimaryDataDimRef(dataDim))
for i in range(self.numDims):
value = posDisp[i]
if value is None:
self.dimensionEntries[i].set('None')
else:
self.dimensionEntries[i].set('%8.4f' % posDisp[i])
def commit(self):
posDisp = self.numDims * [0]
for i in range(self.numDims):
posDisp[i] = float(self.dimensionEntries[i].get())
if self.unit != 'point':
dataDim = self.dataDims[i]
if dataDim.className == 'FreqDataDim':
self.posn[i] = unit_converter[(self.unit, 'point')](
posDisp[i], getPrimaryDataDimRef(dataDim))
else:
self.posn[i] = posDisp[i]
if self.peak:
movePeak(self.peak, self.posn)
else:
self.peak = pickPeak(self.peakList, self.posn)
height = self.heightEntry.get()
volume = self.volumeEntry.get()
setManualPeakIntensity(self.peak, height, intensityType='height')
setManualPeakIntensity(self.peak, volume, intensityType='volume')
details = self.detailEntry.get() or None
self.peak.setDetails(details)
self.close()
def deletedPeak(self, peak):
if self.peak is peak:
self.close()
def destroy(self):
self.unregisterNotify(self.deletedPeak, 'ccp.nmr.Nmr.Peak', 'delete')
for func in ('setAnnotation', 'setDetails', 'setFigOfMerit'):
self.unregisterNotify(self.updatePeak, 'ccp.nmr.Nmr.Peak', func)
for func in ('setAnnotation', 'setPosition', 'setNumAliasing'):
self.unregisterNotify(self.updatePeak, 'ccp.nmr.Nmr.PeakDim', func)
for func in ('__init__', 'delete', 'setValue'):
self.unregisterNotify(self.updatePeak, 'ccp.nmr.Nmr.PeakIntensity',
func)
BasePopup.destroy(self)
class NmrPipePseudoPopup(BasePopup):
pseudoEntries = ('Is Pseudo Expt', 'Is Not Pseudo Expt')
def __init__(self, parent, params, dim, fileName='', *args, **kw):
self.dim = dim
self.params = params
self.fileName = fileName
m = template_re.match(fileName)
if m:
n = len(m.groups(2))
ss = '%%0%dd' % n
template = re.sub(template_re, r'\1%s\3' % ss, fileName)
else:
template = fileName
self.template = template
BasePopup.__init__(self,
parent=parent,
title='NMRPipe Pseudo Data',
modal=True,
**kw)
def body(self, master):
fileName = self.fileName
directory = os.path.dirname(fileName)
if not directory:
directory = os.getcwd()
fileName = os.path.basename(fileName)
m = template_re.match(fileName)
if m:
n = len(m.groups(2))
ss = '%%0%dd' % n
template = re.sub(template_re, r'\1%s\3' % ss, fileName)
else:
template = fileName
master.rowconfigure(0, weight=1)
master.rowconfigure(1, weight=1)
master.columnconfigure(1, weight=1)
tipTexts = [
'The experiment is pseudo-N dimensional, with a sampled axis',
'The experiment is the regular kind with only NMR frequency axes'
]
self.pseudoButton = RadioButtons(
master,
entries=self.pseudoEntries,
select_callback=self.changedPseudoMode,
grid=(0, 0),
sticky='nw',
tipTexts=tipTexts)
frame = self.pseudoFrame = Frame(master)
self.pseudoFrame.grid(row=1, column=0, sticky='nsew')
row = 0
npts = self.params.npts[self.dim]
tipText = 'Number of data points (planes) along sampled axis'
label = Label(frame, text='Number of points: ')
label.grid(row=row, column=0, sticky='e')
self.nptsEntry = IntEntry(frame,
text=npts,
tipText=tipText,
width=8,
grid=(row, 1))
tipText = 'Load the values for the sampled axis from a text file containing a list of numeric values'
Button(frame,
text='Load values from text file',
command=self.loadValues,
tipText=tipText,
grid=(row, 2),
sticky='ew')
row = row + 1
tipText = 'The values (e.g. T1, T2) corresponding to each data point (plane) along sampled axis'
label = Label(frame, text='Point values: ')
label.grid(row=row, column=0, sticky='e')
self.valueEntry = FloatEntry(frame, isArray=True, tipText=tipText)
self.valueEntry.grid(row=row, column=1, columnspan=2, sticky='ew')
row = row + 1
tipText = 'Fetch the Point values from the files given by the NMRPipe template'
button = Button(
frame,
text='Fetch values from file(s) specified by template below',
command=self.fetchValues,
tipText=tipText)
button.grid(row=row, column=1, columnspan=2, sticky='w')
row = row + 1
tipText = 'The directory where the data files reside'
button = Button(frame,
text='Data directory: ',
command=self.chooseDirectory)
button.grid(row=row, column=0, sticky='e')
self.directoryEntry = Entry(frame,
text=directory,
width=40,
tipText=tipText)
self.directoryEntry.grid(row=row, column=1, columnspan=2, sticky='ew')
row = row + 1
tipText = 'The NMRPipe template for the data files, if you want to use these to fetch the point values from'
button = Button(frame,
text='NMRPipe template: ',
command=self.chooseFile)
button.grid(row=row, column=0, sticky='e')
self.templateEntry = Entry(frame, text=template, tipText=tipText)
self.templateEntry.grid(row=row, column=1, columnspan=2, sticky='ew')
for n in range(row):
frame.rowconfigure(n, weight=1)
frame.columnconfigure(1, weight=1)
buttons = UtilityButtonList(master,
closeText='Ok',
doClone=False,
closeCmd=self.updateParams,
helpUrl=self.help_url)
buttons.grid(row=2, column=0, sticky='ew')
def loadValues(self):
directory = self.parent.fileSelect.getDirectory()
fileSelectPopup = FileSelectPopup(self,
title='Select Sampled Data File',
dismiss_text='Cancel',
selected_file_must_exist=True,
multiSelect=False,
directory=directory)
fileName = fileSelectPopup.file_select.getFile()
if not fileName:
return
fileObj = open(fileName, 'rU')
data = ''
line = fileObj.readline()
while line:
data += line
line = fileObj.readline()
fileObj.close()
data = re.sub(',\s+', ',', data)
data = re.sub('\s+', ',', data)
data = re.sub(',,', ',', data)
data = re.sub('[^0-9,.\-+eE]', '', data)
self.valueEntry.set(data)
def chooseDirectory(self):
directory = os.path.dirname(self.fileName)
if not directory:
directory = os.getcwd()
popup = FileSelectPopup(self, directory=directory, show_file=False)
directory = popup.getDirectory()
popup.destroy()
if directory:
self.directoryEntry.set(directory)
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 updateParams(self):
params = self.params
if self.pseudoButton.get() == self.pseudoEntries[0]:
npts = self.nptsEntry.get()
params.npts[self.dim] = npts
values = self.valueEntry.get()
try:
params.setSampledDim(self.dim, values)
except ApiError, e:
showError('Set Sampled Dim', e.error_msg, parent=self)
return
params.fixNullDims(ignoreDim=self.dim)
else:
class AddContourFilePopup(BasePopup):
"""
**Add Existing Contour Files to Project**
The purpose of this dialog is to allow the user to add pre-existing
contour files to the project. Contour files only depend on the spectrum
data so the same contour files can be used across multiple projects,
and that is the reason this dialog might be used.
See also: `Spectrum Contour Files`_, `Creating Contour Files`_.
.. _`Spectrum Contour Files`: EditContourFilesPopup.html
.. _`Creating Contour Files`: CreateContourFilePopup.html
"""
def __init__(self, parent, *args, **kw):
self.spectrum = None
BasePopup.__init__(self,
parent=parent,
title='Add existing contour file',
**kw)
def body(self, master):
self.geometry('600x130')
master.grid_columnconfigure(1, weight=1)
for n in range(5):
master.grid_rowconfigure(n, weight=1)
row = 0
label = Label(master, text='Spectrum: ')
label.grid(row=row, column=0, sticky='e')
tipText = 'The spectrum for which the contour file is being added'
self.expt_spectrum = PulldownList(master,
callback=self.updateContourDir,
tipText=tipText)
self.expt_spectrum.grid(row=row, column=1, sticky='w')
row = row + 1
tipText = 'The location of the directory where contour files are stored on disk'
label = Label(master, text='Contour dir: ')
label.grid(row=row, column=0, sticky='e')
self.dir_label = Label(master, text='', tipText=tipText)
self.dir_label.grid(row=row, column=1, sticky='w')
row = row + 1
label = Label(
master,
text=
'(file will be copied into Contour dir if it is not already in there)'
)
label.grid(row=row, column=1, sticky='w')
row = row + 1
tipText = 'Browse for a file store contour data'
button = Button(master,
text='File name: ',
command=self.selectFile,
tipText=tipText)
button.grid(row=row, column=0, sticky='e')
tipText = 'Enter the name of the file to store contour data'
self.file_entry = Entry(master, tipText=tipText)
self.file_entry.grid(row=row, column=1, sticky='ew')
row = row + 1
texts = ['Add File']
commands = [self.addFile]
tipTexts = [
'Use the selected contour file in the current project, copying it to the contour directory if required',
]
self.buttons = UtilityButtonList(master,
texts=texts,
doClone=False,
tipTexts=tipTexts,
commands=commands,
helpUrl=self.help_url)
self.buttons.grid(row=row, column=0, columnspan=2, sticky='ew')
self.curateNotifiers(self.registerNotify)
self.updateSpectrum()
def destroy(self):
self.curateNotifiers(self.unregisterNotify)
BasePopup.destroy(self)
def curateNotifiers(self, notifyFunc):
for clazz in ('Experiment', 'DataSource'):
for func in ('__init__', 'delete', 'setName'):
notifyFunc(self.updateNotifier, 'ccp.nmr.Nmr.%s' % clazz, func)
def updateSpectrum(self, spectrum=None):
if not spectrum:
spectrum = self.spectrum
spectra = self.parent.getSpectra()
if spectra:
if spectrum not in spectra:
spectrum = spectra[0]
index = spectra.index(spectrum)
names = ['%s:%s' % (x.experiment.name, x.name) for x in spectra]
else:
index = 0
names = []
self.expt_spectrum.setup(names, spectra, index)
self.updateContourDir(spectrum)
def updateNotifier(self, *extra):
self.updateSpectrum()
def updateContourDir(self, spectrum):
if spectrum is self.spectrum:
return
self.spectrum = spectrum
if spectrum:
path = spectrum.analysisSpectrum.contourDir.dataLocation
else:
path = ''
self.dir_label.set(path)
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 addFile(self):
spectrum = self.spectrum
if not spectrum:
return
dataStore = spectrum.dataStore
if not dataStore:
showError('No dataStore',
'Spectrum does not have associated dataStore',
parent=self)
return
if not isinstance(dataStore, BlockedBinaryMatrix):
showError('No blockedBinaryMatrix',
'Spectrum dataStore is not a blockedBinaryMatrix',
parent=self)
return
if not dataStore.blockSizes:
showError('No blockSize',
'Spectrum dataStore does not have blockSize set',
parent=self)
return
blockSize = list(dataStore.blockSizes)
fileName = self.file_entry.get()
if not fileName:
showError('No filename', 'No filename given', parent=self)
return
fileName = normalisePath(fileName, makeAbsolute=True)
contourDir = spectrum.analysisSpectrum.contourDir.dataLocation
if fileName.startswith(contourDir):
path = fileName[len(contourDir) + 1:]
else:
path = os.path.basename(fileName)
if not os.path.exists(contourDir):
os.makedirs(contourDir)
print 'Copying %s to %s' % (fileName, contourDir)
shutil.copy(fileName, contourDir)
try:
header = getStoredContourHeader(fileName)
except Exception, e:
showError('File error', str(e), parent=self)
return
if header['ndim'] != spectrum.numDim:
showError(
'Number of dimensions',
'Number of dimensions in file (%d) does not match spectrum (%d)'
% (header['ndim'], spectrum.numDim),
parent=self)
return
dataDims = spectrum.sortedDataDims()
npoints = [x.numPoints for x in dataDims]
if header['npoints'] != npoints:
showError(
'Number of points',
'Number of points in file (%s) does not match spectrum (%s)' %
(header['npoints'], npoints),
parent=self)
return
if header['blockSize'] != blockSize:
showError('Block size',
'Block size in file (%s) does not match spectrum (%s)' %
(header['blockSize'], blockSize),
parent=self)
return
createStoredContour(spectrum, path, header['xdim'], header['ydim'])
showInfo('Added file',
'Successfully added stored contour file',
parent=self)
class HelpFrame(Frame):
def __init__(self,
parent,
width=70,
height=20,
xscroll=False,
yscroll=True,
*args,
**kw):
apply(Frame.__init__, (self, parent) + args, kw)
self.grid_columnconfigure(1, weight=1)
row = 0
texts = ('back', 'forward', 'load')
commands = (self.prevUrl, self.nextUrl, self.loadUrl)
self.buttons = ButtonList(self, texts=texts, commands=commands)
self.buttons.grid(row=row, column=0)
self.url_entry = Entry(self, returnCallback=self.loadUrl)
self.url_entry.grid(row=row, column=1, columnspan=2, sticky=Tkinter.EW)
row = row + 1
frame = Frame(self)
frame.grid(row=row, column=0, columnspan=3, sticky=Tkinter.W)
self.createFindFrame(frame)
row = row + 1
self.grid_rowconfigure(row, weight=1)
self.help_text = ScrolledHtml(self,
width=width,
height=height,
xscroll=xscroll,
yscroll=yscroll,
startUrlCallback=self.startOpenUrl,
endUrlCallback=self.endOpenUrl,
enterLinkCallback=self.setLinkLabel,
leaveLinkCallback=self.setLinkLabel)
self.help_text.grid(row=row,
column=0,
columnspan=3,
sticky=Tkinter.NSEW)
row = row + 1
self.link_label = Label(self)
self.link_label.grid(row=row, column=0, columnspan=2, sticky=Tkinter.W)
button = memops.gui.Util.createDismissButton(self,
dismiss_cmd=self.close)
button.grid(row=row, column=2)
self.urls = []
self.current = -1
self.updateUrlList = True
self.setButtonState()
def close(self):
if (self.popup):
if (self.popup.modal):
self.popup.do_grab()
popup = self.parent
while (not hasattr(popup, 'top')):
popup = popup.parent
popup.close()
def createFindFrame(self, master):
frame = Frame(master)
frame.grid(row=0, column=0, sticky=Tkinter.W)
arrow = ToggleArrow(frame, callback=self.toggleFindFrame)
arrow.grid(row=0, column=0)
button = Button(frame, text='find:', command=self.findPhrase)
button.grid(row=0, column=1)
self.find_entry = Entry(frame,
width=20,
returnCallback=self.findPhrase)
self.find_entry.grid(row=0, column=2)
self.find_frame = frame = Frame(master)
entries = ('search forwards', 'search backwards')
self.direction_buttons = PulldownMenu(frame, entries=entries)
self.direction_buttons.grid(row=0, column=0, sticky=Tkinter.W, padx=5)
self.forwards_entries = entries = ('wrap search', 'stop at end')
self.backwards_entries = ('wrap search', 'stop at beginning')
self.wrap_buttons = PulldownMenu(frame, entries=entries)
self.wrap_buttons.grid(row=0, column=1, sticky=Tkinter.W, padx=5)
self.direction_buttons.callback = self.setWrapText
entries = ('case insensitive', 'case sensitive')
self.case_buttons = PulldownMenu(frame, entries=entries)
self.case_buttons.grid(row=0, column=2, sticky=Tkinter.W, padx=5)
entries = ('literal search', 'regex search')
self.pattern_buttons = PulldownMenu(frame, entries=entries)
self.pattern_buttons.grid(row=0, column=3, sticky=Tkinter.W, padx=5)
self.countVar = Tkinter.IntVar()
def setWrapText(self, entry_ind, entry):
if (entry_ind == 0):
entries = self.forwards_entries
else:
entries = self.backwards_entries
self.wrap_buttons.replace(
entries, selected_index=self.wrap_buttons.getSelectedIndex())
def findPhrase(self, *extra):
phrase = self.find_entry.get()
if (not phrase):
showError('No find phrase', 'Enter phrase in entry box.')
return
if (self.direction_buttons.getSelectedIndex() == 0):
forwards = 1
backwards = 0
else:
forwards = 0
backwards = 1
if (self.case_buttons.getSelectedIndex() == 0):
nocase = 1
else:
nocase = 0
if (self.pattern_buttons.getSelectedIndex() == 0):
exact = 1
regexp = 0
else:
exact = 0
regexp = 1
start = self.help_text.tag_nextrange(Tkinter.SEL, '1.0')
if (start):
start = start[0]
if (forwards):
start = '%s+1c' % start
else:
start = '%s-1c' % start
else:
start = Tkinter.CURRENT
if (self.wrap_buttons.getSelectedIndex() == 0):
match = self.help_text.search(phrase,
start,
count=self.countVar,
forwards=forwards,
backwards=backwards,
nocase=nocase,
exact=exact,
regexp=regexp)
elif (forwards):
match = self.help_text.search(phrase,
start,
count=self.countVar,
forwards=forwards,
backwards=backwards,
nocase=nocase,
exact=exact,
regexp=regexp,
stopindex=Tkinter.END)
else:
match = self.help_text.search(phrase,
start,
count=self.countVar,
forwards=forwards,
backwards=backwards,
nocase=nocase,
exact=exact,
regexp=regexp,
stopindex='1.0')
if (match):
self.help_text.see(match)
self.help_text.setSelection(
match, '%s+%dc' % (match, self.countVar.get()))
self.help_text.tag_config(Tkinter.SEL, background='gray70')
else:
showInfo('No match', 'No match found for phrase.')
def toggleFindFrame(self, isClosed):
if (isClosed):
self.find_frame.grid_forget()
else:
self.find_frame.grid(row=1, column=0, sticky=Tkinter.W)
def setButtonState(self):
n = len(self.urls)
if ((n >= 2) and (self.current > 0)):
state = Tkinter.NORMAL
else:
state = Tkinter.DISABLED
self.buttons.buttons[0].config(state=state)
if ((n >= 2) and (self.current < (n - 1))):
state = Tkinter.NORMAL
else:
state = Tkinter.DISABLED
self.buttons.buttons[1].config(state=state)
def setLinkLabel(self, url=''):
self.link_label.set(url)
def loadUrl(self, *extra):
url = self.url_entry.get()
if (url):
if (url.find(':') == -1):
if (url[:2] != '//'):
url = '//' + url
url = 'http:' + url
self.showUrl(url, forceLoad=True)
def newUrl(self, url):
self.current = self.current + 1
self.urls[self.current:] = [[url, 0.0]]
self.setButtonState()
#print 'newUrl', self.current, self.urls
def prevUrl(self):
if (self.current > 0): # True if called via button
(url, offset) = self.urls[self.current - 1]
self.updateUrlList = False
self.showUrl(url, offset=offset, allowModifyPath=False)
self.updateUrlList = True
self.current = self.current - 1
self.setButtonState()
#print 'prevUrl', self.current, self.urls
def nextUrl(self):
if (self.current < (len(self.urls) - 1)): # True if called via button
(url, offset) = self.urls[self.current + 1]
self.updateUrlList = False
self.showUrl(url, offset=offset, allowModifyPath=False)
self.updateUrlList = True
self.current = self.current + 1
self.setButtonState()
#print 'nextUrl', self.current, self.urls
def showText(self, message, popup=None):
self.popup = popup
if (popup):
self.parent.modal = popup.modal
self.help_text.setState(Tkinter.NORMAL) # so that can add text
self.help_text.clear() # clear existing text
self.help_text.append(message)
self.help_text.setState(Tkinter.DISABLED) # make text read-only
self.open()
def startOpenUrl(self, yview):
if (self.urls):
self.urls[self.current][1] = yview[0]
def endOpenUrl(self, url):
#print 'endOpenUrl', url, self.updateUrlList
self.url_entry.set(url)
if (self.updateUrlList):
self.newUrl(url)
self.setLinkLabel()
self.open()
def showUrl(self,
url,
offset=None,
forceLoad=False,
allowModifyPath=True,
popup=None):
self.popup = popup
if (popup):
self.parent.modal = popup.modal
self.help_text.openUrl(url,
forceLoad=forceLoad,
allowModifyPath=allowModifyPath)
if (offset is not None):
self.help_text.yview(Tkinter.MOVETO, str(offset))
class CingGui(BasePopup):
def __init__(self, parent, options, *args, **kw):
# Fill in below variable once run generates some results
self.haveResults = None
# store the options
self.options = options
BasePopup.__init__(self, parent=parent, title='CING Setup', **kw)
# self.setGeometry(850, 750, 50, 50)
self.project = None
# self.tk_strictMotif( True)
self.updateGui()
# end def __init__
def body(self, guiFrame):
row = 0
col =0
# frame = Frame( guiFrame )
# frame.grid(row=row, column=col, sticky='news')
self.menuBar = Menu( guiFrame)
self.menuBar.grid( row=row, column=col, sticky='ew')
#----------------------------------------------------------------------------------
# Project frame
#----------------------------------------------------------------------------------
# guiFrame.grid_columnconfigure(row, weight=1)
# frame = LabelFrame(guiFrame, text='Project', font=medFont)
row = +1
col =0
frame = LabelFrame(guiFrame, text='Project', **labelFrameAttributes )
print '>', frame.keys()
frame.grid(row=row, column=col, sticky='nsew' )
frame.grid_columnconfigure(2, weight=1)
# frame.grid_rowconfigure(0, weight=1)
srow = 0
self.projectOptions = ['old','new from PDB','new from CCPN','new from CYANA']
self.projOptionsSelect = RadioButtons(frame, selected_index=0, entries=self.projectOptions, direction='vertical',
select_callback=self.updateGui
)
self.projOptionsSelect.grid(row=srow,column=0,rowspan=len(self.projectOptions),columnspan=2, sticky='w')
if self.options.name:
text = self.options.name
else:
text=''
# end if
self.projEntry = Entry(frame, bd=1, text=text, returnCallback=self.updateGui)
self.projEntry.grid(row=srow,column=2,columnspan=2,sticky='ew')
# self.projEntry.bind('<Key>', self.updateGui)
self.projEntry.bind('<Leave>', self.updateGui)
projButton = Button(frame, bd=1,command=self.chooseOldProjectFile, text='browse')
projButton.grid(row=srow,column=3,sticky='ew')
srow += 1
self.pdbEntry = Entry(frame, bd=1, text='')
self.pdbEntry.grid(row=srow,column=2,sticky='ew')
self.pdbEntry.bind('<Leave>', self.updateGui)
pdbButton = Button(frame, bd=1,command=self.choosePdbFile, text='browse')
pdbButton.grid(row=srow,column=3,sticky='ew')
srow += 1
self.ccpnEntry = Entry(frame, bd=1, text='')
self.ccpnEntry.grid(row=srow,column=2,sticky='ew')
self.ccpnEntry.bind('<Leave>', self.updateGui)
ccpnButton = Button(frame, bd=1,command=self.chooseCcpnFile, text='browse')
ccpnButton.grid(row=srow,column=3,sticky='ew')
srow += 1
self.cyanaEntry = Entry(frame, bd=1, text='')
self.cyanaEntry.grid(row=srow,column=2,sticky='ew')
self.cyanaEntry.bind('<Leave>', self.updateGui)
cyanaButton = Button(frame, bd=1,command=self.chooseCyanaFile, text='browse')
cyanaButton.grid(row=srow,column=3,sticky='ew')
#Empty row
srow += 1
label = Label(frame, text='')
label.grid(row=srow,column=0,sticky='nw')
srow += 1
label = Label(frame, text='Project name:')
label.grid(row=srow,column=0,sticky='nw')
self.nameEntry = Entry(frame, bd=1, text='')
self.nameEntry.grid(row=srow,column=2,sticky='w')
#Empty row
srow += 1
label = Label(frame, text='')
label.grid(row=srow,column=0,sticky='nw')
srow += 1
self.openProjectButton = Button(frame, command=self.openProject, text='Open Project', **actionButtonAttributes )
self.openProjectButton.grid(row=srow,column=0, columnspan=4, sticky='ew')
#----------------------------------------------------------------------------------
# status
#----------------------------------------------------------------------------------
# guiFrame.grid_columnconfigure(1, weight=0)
srow = 0
frame = LabelFrame(guiFrame, text='Status', **labelFrameAttributes)
frame.grid( row=srow, column=1, sticky='wnes')
self.projectStatus = Text(frame, height=11, width=70, borderwidth=0, relief='flat')
self.projectStatus.grid(row=0, column=0, sticky='wen')
#Empty row
srow += 1
label = Label(frame, text='')
label.grid(row=srow,column=0,sticky='nw')
srow += 1
self.closeProjectButton = Button(frame, command=self.closeProject, text='Close Project', **actionButtonAttributes)
self.closeProjectButton.grid(row=srow,column=0, columnspan=4, sticky='ew')
#----------------------------------------------------------------------------------
# Validate frame
#----------------------------------------------------------------------------------
row +=1
col=0
frame = LabelFrame(guiFrame, text='Validate', **labelFrameAttributes)
# frame = LabelFrame(guiFrame, text='Validate', font=medFont)
frame.grid(row=row, column=col, sticky='nsew')
# frame.grid_columnconfigure(2, weight=1)
frame.grid_rowconfigure(0, weight=1)
srow = 0
# label = Label(frame, text='validation')
# label.grid(row=srow,column=0,sticky='nw')
#
# self.selectDoValidation = CheckButton(frame)
# self.selectDoValidation.grid(row=srow, column=1,sticky='nw' )
# self.selectDoValidation.set(True)
#
# srow += 1
# label = Label(frame, text='')
# label.grid(row=srow,column=0,sticky='nw')
#
# srow += 1
label = Label(frame, text='checks')
label.grid(row=srow,column=0,sticky='nw')
self.selectCheckAssign = CheckButton(frame)
self.selectCheckAssign.grid(row=srow, column=1,sticky='nw' )
self.selectCheckAssign.set(True)
label = Label(frame, text='assignments and shifts')
label.grid(row=srow,column=2,sticky='nw')
# srow += 1
# self.selectCheckQueen = CheckButton(frame)
# self.selectCheckQueen.grid(row=srow, column=4,sticky='nw' )
# self.selectCheckQueen.set(False)
# label = Label(frame, text='QUEEN')
# label.grid(row=srow,column=5,sticky='nw')
#
# queenButton = Button(frame, bd=1,command=None, text='setup')
# queenButton.grid(row=srow,column=6,sticky='ew')
srow += 1
self.selectCheckResraint = CheckButton(frame)
self.selectCheckResraint.grid(row=srow, column=1,sticky='nw' )
self.selectCheckResraint.set(True)
label = Label(frame, text='restraints')
label.grid(row=srow,column=2,sticky='nw')
srow += 1
self.selectCheckStructure = CheckButton(frame)
self.selectCheckStructure.grid(row=srow, column=1,sticky='nw' )
self.selectCheckStructure.set(True)
label = Label(frame, text='structural')
label.grid(row=srow,column=2,sticky='nw')
srow += 1
self.selectMakeHtml = CheckButton(frame)
self.selectMakeHtml.grid(row=srow, column=1,sticky='nw' )
self.selectMakeHtml.set(True)
label = Label(frame, text='generate HTML')
label.grid(row=srow,column=2,sticky='nw')
srow += 1
self.selectCheckScript = CheckButton(frame)
self.selectCheckScript.grid(row=srow, column=1,sticky='nw' )
self.selectCheckScript.set(False)
label = Label(frame, text='user script')
label.grid(row=srow,column=0,sticky='nw')
self.validScriptEntry = Entry(frame, bd=1, text='')
self.validScriptEntry.grid(row=srow,column=2,columnspan=3, sticky='ew')
scriptButton = Button(frame, bd=1,command=self.chooseValidScript, text='browse')
scriptButton.grid(row=srow,column=5,sticky='ew')
srow += 1
label = Label(frame, text='ranges')
label.grid(row=srow,column=0,sticky='nw')
self.rangesEntry = Entry( frame, text='' )
self.rangesEntry.grid( row=srow, column=2, columnspan=3, sticky='ew')
# self.validScriptEntry = Entry(frame, bd=1, text='')
# self.validScriptEntry.grid(row=srow,column=3,sticky='ew')
#
# scriptButton = Button(frame, bd=1,command=self.chooseValidScript, text='browse')
# scriptButton.grid(row=srow,column=4,sticky='ew')
srow += 1
texts = ['Run Validation','View Results','Setup QUEEN']
commands = [self.runCing, None, None]
buttonBar = ButtonList(frame, texts=texts, commands=commands,expands=True)
buttonBar.grid(row=srow, column=0, columnspan=6, sticky='ew')
for button in buttonBar.buttons:
button.config(**actionButtonAttributes)
# end for
self.runButton = buttonBar.buttons[0]
self.viewResultButton = buttonBar.buttons[1]
self.queenButton = buttonBar.buttons[2]
#----------------------------------------------------------------------------------
# Miscellaneous frame
#----------------------------------------------------------------------------------
row +=0
col=1
# frame = LabelFrame(guiFrame, text='Miscellaneous', font=medFont)
frame = LabelFrame(guiFrame, text='Miscellaneous', **labelFrameAttributes)
frame.grid(row=row, column=col, sticky='news')
frame.grid_columnconfigure(2, weight=1)
frame.grid_columnconfigure(4, weight=1,minsize=30)
frame.grid_rowconfigure(0, weight=1)
# Exports
srow = 0
label = Label(frame, text='export to')
label.grid(row=srow,column=0,sticky='nw')
self.selectExportXeasy = CheckButton(frame)
self.selectExportXeasy.grid(row=srow, column=1,sticky='nw' )
self.selectExportXeasy.set(True)
label = Label(frame, text='Xeasy, Sparky, TALOS, ...')
label.grid(row=srow,column=2,sticky='nw')
srow += 1
self.selectExportCcpn = CheckButton(frame)
self.selectExportCcpn.grid(row=srow, column=1,sticky='nw' )
self.selectExportCcpn.set(True)
label = Label(frame, text='CCPN')
label.grid(row=srow,column=2,sticky='nw')
srow += 1
self.selectExportQueen = CheckButton(frame)
self.selectExportQueen.grid(row=srow, column=1,sticky='nw' )
self.selectExportQueen.set(True)
label = Label(frame, text='QUEEN')
label.grid(row=srow,column=2,sticky='nw')
srow += 1
self.selectExportRefine = CheckButton(frame)
self.selectExportRefine.grid(row=srow, column=1,sticky='nw' )
self.selectExportRefine.set(True)
label = Label(frame, text='refine')
label.grid(row=srow,column=2,sticky='nw')
srow += 1
label = Label(frame, text='')
label.grid(row=srow,column=0,sticky='nw')
# User script
srow += 1
label = Label(frame, text='user script')
label.grid(row=srow,column=0,sticky='nw')
self.selectMiscScript = CheckButton(frame)
self.selectMiscScript.grid(row=srow, column=1,sticky='nw' )
self.selectMiscScript.set(False)
self.miscScriptEntry = Entry(frame, bd=1, text='')
self.miscScriptEntry.grid(row=srow,column=3,sticky='ew')
script2Button = Button(frame, bd=1,command=self.chooseMiscScript, text='browse')
script2Button.grid(row=srow,column=4,sticky='ew')
srow += 1
texts = ['Export','Run Script']
commands = [None, None]
buttonBar = ButtonList(frame, texts=texts, commands=commands,expands=True)
buttonBar.grid(row=srow, column=0, columnspan=5, sticky='ew')
for button in buttonBar.buttons:
button.config(**actionButtonAttributes)
# end for
self.exportButton = buttonBar.buttons[0]
self.scriptButton = buttonBar.buttons[1]
#----------------------------------------------------------------------------------
# Textarea
#----------------------------------------------------------------------------------
row +=1
guiFrame.grid_rowconfigure(row, weight=1)
self.outputTextBox = ScrolledText(guiFrame)
self.outputTextBox.grid(row=row, column=0, columnspan=2, sticky='nsew')
self.redirectConsole()
#----------------------------------------------------------------------------------
# Buttons
#----------------------------------------------------------------------------------
row +=1
col=0
texts = ['Quit', 'Help']
commands = [self.close, None]
self.buttonBar = ButtonList(guiFrame, texts=texts, commands=commands,expands=True)
self.buttonBar.grid(row=row, column=col, columnspan=2, sticky='ew')
# self.openProjectButton = self.buttonBar.buttons[0]
# self.closeProjectButton = self.buttonBar.buttons[1]
# self.runButton = self.buttonBar.buttons[0]
# self.viewResultButton = self.buttonBar.buttons[1]
for button in self.buttonBar.buttons:
button.config(**actionButtonAttributes)
# end for
# end def body
def getGuiOptions(self):
projectName = self.projEntry.get()
index = self.projOptionsSelect.getIndex()
if index > 0:
makeNewProject = True
projectImport = None
if index > 1:
i = index-2
format = ['PDB','CCPN','CYANA'][i]
file = [self.pdbEntry, self.ccpnEntry, self.cyanaEntry][i].get()
if not file:
showWarning('Failure','No %s file selected' % format)
return
# end if
projectImport = (format, file)
# end if
else:
# Chould also check that any old project file exists
makeNewProject = False
projectImport = None
# end if
doValidation = self.selectDoValidation.get()
checks = []
if doValidation:
if self.selectCheckAssign.get():
checks.append('assignments')
# end if
if self.selectCheckResraint.get():
checks.append('restraints')
# end if
if self.selectCheckStructure.get():
checks.append('structural')
# end if
if self.selectMakeHtml.get():
checks.append('HTML')
# end if
if self.selectCheckScript.get():
script = self.validScriptEntry.get()
if script:
checks.append( ('script',script) )
# end if
# end if
if self.selectCheckQueen.get():
checks.append('queen')
# end if
# end if
exports = []
if self.selectExportXeasy.get():
exports.append('Xeasy')
# end if
if self.selectExportCcpn.get():
exports.append('CCPN')
# end if
if self.selectExportQueen.get():
exports.append('QUEEN')
# end if
if self.selectExportRefine.get():
exports.append('refine')
# end if
miscScript = None
if self.selectMiscScript.get():
script = self.miscScriptEntry.get()
if script:
miscScript = script
# end if
# end if
return projectName, makeNewProject, projectImport, doValidation, checks, exports, miscScript
# end def getGuiOptions
def runCing(self):
options = self.getGuiOptions()
if options:
projectName, makeNewProject, projectImport, doValidation, checks, exports, miscScript = options
print 'Project name:', projectName
print 'Make new project?', makeNewProject
print 'Import source:', projectImport
print 'Do vailidation?', doValidation
print 'Validation checks:', ','.join(checks)
print 'Export to:', ','.join(exports)
print 'User script:', miscScript
# end if
# end def runCing
# else there was already an error message
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()
# end def chooseOldProjectFile
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()
# end def choosePdbFile
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()
# end def chooseCcpnFile
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()
# end def chooseCyanaFile
def chooseValidScript(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.validScriptEntry.set(fileName)
popup.destroy()
# end def chooseValidScript
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()
# end def chooseMiscScript
def openProject(self ):
projOption = self.projOptionsSelect.get()
if projOption == self.projectOptions[0]:
self.openOldProject()
elif projOption == self.projectOptions[1]:
self.initPdb()
# end if
if self.project:
self.project.gui = self
# end if
self.updateGui()
#end def
def openOldProject(self ):
fName = self.projEntry.get()
if not os.path.exists( fName ):
nTerror('Error: file "%s" does not exist\n', fName)
#end if
if self.project:
self.closeProject()
# end if
self.project = cing.Project.open( name=fName, status='old', verbose=False )
#end def
def initPdb(self ):
fName = self.pdbEntry.get()
if not os.path.exists( fName ):
nTerror('Error: file "%s" does not exist\n', fName)
#end if
self.project = cing.Project.open( self.nameEntry.get(), status='new' )
self.project.initPDB( pdbFile=fName, convention = 'PDB' )
#end def
def closeProject(self):
if self.project:
self.project.close()
# end if
self.project = None
self.updateGui()
#end def
def updateGui(self, event=None):
projOption = self.projOptionsSelect.get()
buttons = self.buttonBar.buttons
# Disable entries
for e in [self.projEntry, self.pdbEntry, self.ccpnEntry, self.cyanaEntry, self.nameEntry]:
e.configure(state='disabled')
#end for
if projOption == self.projectOptions[0]:
# Enable entries
self.projEntry.configure(state='normal')
if (len(self.projEntry.get()) > 0):
self.openProjectButton.enable()
self.runButton.enable()
else:
self.openProjectButton.disable()
self.runButton.disable()
#end if
elif projOption == self.projectOptions[1]:
# Enable entries
self.pdbEntry.configure(state='normal')
self.nameEntry.configure(state='normal')
if (len(self.pdbEntry.get()) > 0 and len(self.nameEntry.get()) > 0):
buttons[0].enable()
buttons[1].enable()
else:
buttons[0].disable()
buttons[1].disable()
#end if
#end if
elif projOption == self.projectOptions[2]:
# Enable entries
self.ccpnEntry.configure(state='normal')
self.nameEntry.configure(state='normal')
if (len(self.ccpnEntry.get()) > 0 and len(self.nameEntry.get()) > 0):
buttons[0].enable()
buttons[1].enable()
else:
buttons[0].disable()
buttons[1].disable()
#end if
#end if
elif projOption == self.projectOptions[3]:
# Enable entries
self.cyanaEntry.configure(state='normal')
self.nameEntry.configure(state='normal')
if (len(self.cyanaEntry.get()) > 0 and len(self.nameEntry.get()) > 0):
buttons[0].enable()
buttons[1].enable()
else:
buttons[0].disable()
buttons[1].disable()
#end if
#end if
self.projectStatus.clear()
if not self.project:
self.projectStatus.setText('No open project')
self.closeProjectButton.setText('Close Project')
self.closeProjectButton.disable()
else:
self.projectStatus.setText(self.project.format())
self.closeProjectButton.enable()
self.closeProjectButton.setText(sprintf('Close Project "%s"', self.project.name))
#end if
#end def
def redirectConsole(self):
#pipe = TextPipe(self.inputTextBox.text_area)
#sys.stdin = pipe
pipe = TextPipe(self.outputTextBox.text_area)
sys.stdout = pipe
nTmessage.stream = pipe
# end def redirectConsole
# sys.stderr = pipe
def resetConsole(self):
#sys.stdin = stdin
nTmessage.stream = stdout
sys.stdout = stdout
sys.stderr = stderr
# end def resetConsole
def open(self):
self.redirectConsole()
BasePopup.open(self)
# end def open
def close(self):
geometry = self.getGeometry()
self.resetConsole()
BasePopup.close(self)
print 'close:',geometry
sys.exit(0) # remove later
# end def close
def destroy(self):
geometry = self.getGeometry()
self.resetConsole()
BasePopup.destroy(self)
print 'destroy:',geometry
sys.exit(0) # remove later
class TableExportPopup(BasePopup):
def __init__(self,
parent,
headings,
entries,
title='',
file='',
*args,
**kw):
headings = [heading.replace('\n', ' ') for heading in headings]
if (headings[0].lower() == 'number'):
self.addNumber = False
else:
self.addNumber = True
headings.insert(0, 'Number')
if (not title):
title = 'Export data'
self.exportSelection = parent.tableExportSelection
self.headings = headings
self.entries = entries
self.file = file
kw['title'] = title
kw['transient'] = True
kw['modal'] = True
BasePopup.__init__(self, parent=parent, *args, **kw)
def body(self, master):
master.grid_columnconfigure(2, weight=1)
row = 0
label = Label(master, text='Data to export:')
label.grid(row=row, column=0, columnspan=3, sticky=Tkinter.W)
self.check_buttons = {}
i = 0
for heading in self.headings:
row = row + 1
isSelected = self.exportSelection.get(row, True)
self.check_buttons[i] = c = CheckButton(
master, selected=isSelected, callback=self.toggleCheckButton)
c.grid(row=row, column=1)
label = Label(master, text=heading)
label.grid(row=row, column=2, sticky=Tkinter.W)
i += 1
row = row + 1
button = Button(master, text='File:', command=self.findFile)
button.grid(row=row, column=0, sticky=Tkinter.W)
self.file_entry = Entry(master, text=self.file, width=30)
self.file_entry.grid(row=row,
column=1,
columnspan=2,
sticky=Tkinter.EW)
row = row + 1
label = Label(master, text='Format:')
label.grid(row=row, column=0, sticky=Tkinter.W)
self.format_menu = PulldownMenu(master, entries=exportFormats)
self.format_menu.grid(row=row,
column=1,
columnspan=2,
sticky=Tkinter.W)
row = row + 1
master.grid_rowconfigure(row, weight=1)
texts = ['Save']
commands = [self.ok]
buttons = createDismissHelpButtonList(master,
texts=texts,
commands=commands,
dismiss_text='Cancel')
buttons.grid(row=row, column=0, columnspan=3, sticky=Tkinter.EW)
def toggleCheckButton(self, isSelected):
for i in self.check_buttons.keys():
self.exportSelection[i] = self.check_buttons[i].get()
def apply(self):
file = self.file_entry.get()
format = self.format_menu.getSelected()
join = exportJoin[format]
n = 0
headings = []
for heading in self.headings:
if self.check_buttons[n].get():
headings.append(n)
n += 1
if (not headings):
showError('Nothing selected',
'No data selected to export.',
parent=self)
return False
try:
fp = open(file, 'w')
except IOError, e:
showError('File error', str(e))
return False
for n in headings:
if (n != headings[0]):
fp.write(join)
fp.write(self.headings[n].encode('utf-8'))
fp.write('\n')
m = 0
for data in self.entries:
if data:
for n in headings:
if (n != headings[0]):
fp.write(join)
if (self.addNumber):
if (n == 0):
x = m + 1
else:
x = data[n - 1]
else:
x = data[n]
if (type(x) == type(float(1))):
if (x == 0):
t = '0'
elif ((abs(x) > 10000) or (abs(x) < 0.001)):
t = '%6.5e' % x
else:
t = '%6.5f' % x
else:
t = str(x)
fp.write(t)
fp.write('\n')
m = m + 1
fp.close()
return True
class EntryScale(Frame):
def __init__(self,
parent,
values,
initial_index=0,
orient=Tkinter.HORIZONTAL,
scale_length=200,
entry_width=5,
entry_format='%2.1f',
set_callback=None,
*args,
**kw):
assert len(values) > 0
self.entry_format = entry_format
self.values = tuple(values)
self.set_callback = set_callback
apply(Frame.__init__, (self, parent) + args, kw)
if (orient == Tkinter.HORIZONTAL):
self.grid_columnconfigure(1, weight=1)
sticky = Tkinter.EW
row = 0
col = 1
else:
self.grid_rowconfigure(1, weight=1)
sticky = Tkinter.NS
row = 1
col = 0
self.entry = Entry(self, width=entry_width)
self.entry.grid(row=0, column=0, sticky=sticky)
self.entry.bind('<KeyPress>', self.keyPress)
self.scale = Tkinter.Scale(self,
orient=orient,
length=scale_length,
from_=0,
to=len(values) - 1,
label='',
showvalue=0,
command=self.setCallback)
self.scale.grid(row=row, column=col, sticky=sticky)
self.set(initial_index)
def get(self):
return self.scale.get()
def getValue(self):
return self.values[self.get()]
def set(self, index):
assert index >= 0 and index < len(
self.values), 'index = %s, len(self.values) = %s' % (
index, len(self.values))
self.scale.set(index)
self.setEntry()
def setEntry(self):
index = self.get()
text = self.entry_format % self.values[index]
self.entry.set(text)
def setCallback(self, ignore):
self.setEntry()
if (self.set_callback):
self.set_callback(self.get())
def keyPress(self, event):
if (event.keysym == 'Return'):
text = self.entry.get()
try:
value = text.atof()
except:
return
min_index = 0
min_diff = abs(self.values[0] - value)
for i in range(1, len(self.values)):
diff = abs(self.values[i] - value)
if (diff < min_diff):
min_index = i
min_diff = diff
self.set(min_index)
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
class PrintFrame(Frame):
def __init__(self, parent, getOption = None, setOption = None,
haveTicks = False, doOutlineBox = True, *args, **kw):
self.getOption = getOption
self.setOption = setOption
self.haveTicks = haveTicks
self.doOutlineBox = doOutlineBox
Frame.__init__(self, parent=parent, *args, **kw)
self.file_select_popup = None
self.getOptionValues()
try:
size_index = paper_types.index(self.paper_type)
except:
size_index = 0
try:
other_unit_index = paper_units.index(self.other_unit)
except:
other_unit_index = 0
try:
orientation_index = paper_orientations.index(self.paper_orientation)
except:
orientation_index = 0
try:
style_index = style_choices.index(self.output_style)
except:
style_index = 0
try:
format_index = format_choices.index(self.output_format)
except:
format_index = 0
if haveTicks:
try:
tick_location_index = tick_locations.index(self.tick_location)
except:
tick_location_index = 0
self.grid_columnconfigure(1, weight=1)
row = 0
button = Button(self, text='File:', command=self.findFile,
tipText='Select location to save print file')
button.grid(row=row, column=0, sticky='e')
self.file_entry = Entry(self, width=40, text=self.file_name,
tipText='Location where file is saved on disk')
self.file_entry.grid(row=row, column=1, sticky='ew')
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,
tipText='Title of the printout, displayed at top')
self.title_entry.grid(row=row, column=1, sticky='ew')
row += 1
label = Label(self, text='X axis label:')
label.grid(row=row, column=0, sticky='e')
self.x_axis_entry = Entry(self, width=40, text=self.x_axis_label,
tipText='X axis label for the printout')
self.x_axis_entry.grid(row=row, column=1, sticky='ew')
row += 1
label = Label(self, text='Y axis label:')
label.grid(row=row, column=0, sticky='e')
self.y_axis_entry = Entry(self, width=40, text=self.y_axis_label,
tipText='Y axis label for the printout')
self.y_axis_entry.grid(row=row, column=1, sticky='ew')
row += 1
frame = Frame(self)
frame.grid(row=row, column=0, columnspan=2, 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,
tipText='The paper size for the printout')
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,
tipText='The size of the Other paper in both dimensions; this requires two values, space or comma separated')
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,
tipText='The unit for the Other paper size')
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,
tipText='Whether the paper should be set in Portrait or Landscape mode')
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,
tipText='Whether the printout should be in colour or black and white')
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,
tipText='Whether to save as PS, EPS or PDF')
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,
tipText='Whether the tick marks appear on the inside or outside of the frame')
self.tick_menu.grid(row=0, column=1, sticky='w')
label = Label(frame, text=' Tick Placement:')
label.grid(row=0, column=2, sticky='e')
if self.tick_placement is None:
selected = None
else:
selected = [(x in self.tick_placement) for x in tick_placements]
self.tick_buttons = CheckButtons(frame, entries=tick_placements,
selected=selected,
tipTexts=('Whether the tick marks appear on the top and/or bottom and/or left and/or right',))
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(1, weight=1)
frame.grid_columnconfigure(3, weight=1)
label = Label(frame, text='Tick Font:')
label.grid(row=0, column=0, sticky='e')
self.tick_font_list = FontList(frame, mode='Print',
selected=self.tick_font,
extraTexts=[PrintTicks.no_tick_text],
tipText='The font used for the tick mark labels')
self.tick_font_list.grid(row=0, column=1, sticky='w')
label = Label(frame, text='Tick Spacing:')
label.grid(row=0, column=2, sticky='e')
# TBD: put preferred choice in data model
self.spacing_menu = PulldownList(frame, texts=spacing_choices, index=0,
callback=self.changedSpacing,
tipText='Whether the program should automatically calculate the major/minor tick spacings and how many decimal places are used for the ticks, or whether the these are specified manually')
self.spacing_menu.grid(row=0, column=3, sticky='w')
ff = self.spacing_frame = Frame(frame)
ff.grid_columnconfigure(1, weight=1)
ff.grid_columnconfigure(2, weight=1)
label = Label(ff, text='Tick Spacing')
label.grid(row=0, column=0, sticky='w')
label = Label(ff, text='Major')
label.grid(row=0, column=1, sticky='ew')
label = Label(ff, text='Minor')
label.grid(row=0, column=2, sticky='ew')
label = Label(ff, text='Decimals')
label.grid(row=0, column=3, sticky='ew')
label = Label(ff, text='X:')
label.grid(row=1, column=0, sticky='w')
self.x_major_entry = FloatEntry(ff, tipText='The spacing in display units of the major tick marks in the X dimension')
self.x_major_entry.grid(row=1, column=1, sticky='ew')
self.x_minor_entry = FloatEntry(ff, tipText='The spacing in display units of the minor tick marks in the X dimension (not printed if left blank)')
self.x_minor_entry.grid(row=1, column=2, sticky='ew')
self.x_decimal_entry = IntEntry(ff, tipText='The number of decimal places for the tick numbers in the X dimension')
self.x_decimal_entry.grid(row=1, column=3, sticky='ew')
label = Label(ff, text='Y:')
label.grid(row=2, column=0, sticky='w')
self.y_major_entry = FloatEntry(ff, tipText='The spacing in display units of the major tick marks in the Y dimension')
self.y_major_entry.grid(row=2, column=1, sticky='ew')
self.y_minor_entry = FloatEntry(ff, tipText='The spacing in display units of the minor tick marks in the Y dimension (not printed if left blank)')
self.y_minor_entry.grid(row=2, column=2, sticky='ew')
self.y_decimal_entry = IntEntry(ff, tipText='The number of decimal places for the tick numbers in the Y dimension')
self.y_decimal_entry.grid(row=2, column=3, sticky='ew')
row += 1
frame = Frame(self)
frame.grid(row=row, column=0, columnspan=4, sticky='ew')
frame.grid_columnconfigure(1, weight=1)
label = Label(frame, text='Tick Length:')
label.grid(row=0, column=0, sticky='e')
# TBD: put preferred choice in data model
self.tick_length_menu = PulldownList(frame, texts=tick_length_choices, index=0,
callback=self.changedLength,
tipText='Whether the program should automatically calculate the major/minor tick lengths, or whether the these are specified manually')
self.tick_length_menu.grid(row=0, column=1, sticky='w')
ff = self.length_frame = Frame(frame)
ff.grid_columnconfigure(1, weight=1)
label = Label(ff, text=' Major length:')
label.grid(row=0, column=0, sticky='w')
self.length_major_entry = FloatEntry(ff, tipText='The length in points of the major tick marks')
self.length_major_entry.grid(row=0, column=1, sticky='w')
label = Label(ff, text='Minor length:')
label.grid(row=0, column=2, sticky='w')
self.length_minor_entry = FloatEntry(ff, tipText='The length in points of the minor tick marks')
self.length_minor_entry.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)
frame.grid_columnconfigure(4, weight=1)
label = Label(frame, text='Scaling:')
label.grid(row=0, column=0, sticky='e')
# TBD: put preferred choice in data model
self.scaling_menu = PulldownList(frame, texts=scaling_choices, index=0,
callback=self.changedScaling,
tipText='Whether the plot should be scaled as a percentage of the maximum size that would fit on the paper, or instead should be specified by the number of cms or inches per unit')
self.scaling_menu.grid(row=0, column=1, sticky='ew')
self.scaling_scale = Scale(frame, orient=Tkinter.HORIZONTAL, value=self.scaling, tipText='The percentage of the maximum size that would fit on the paper that the plot is scaled by')
self.scaling_scale.grid(row=0, column=2, columnspan=3, sticky='ew')
self.x_scaling_label = Label(frame, text='X:')
self.x_scaling_entry = FloatEntry(frame, tipText='The scaling that should be used in the X dimension as cms or inches per unit')
self.y_scaling_label = Label(frame, text='Y:')
self.y_scaling_entry = FloatEntry(frame, tipText='The scaling that should be used in the Y dimension as cms or inches per unit')
row += 1
frame = Frame(self)
frame.grid(row=row, column=0, columnspan=4, sticky='w')
frame.grid_columnconfigure(2, weight=1)
label = Label(frame, text='Include:')
label.grid(row=0, column=0, sticky='e')
tipTexts = ('Whether the time and date should be included in the printout',
'Whether the file name should be included in the printout')
if self.border_decoration is None:
selected = None
else:
selected = [(x in self.border_decoration) for x in border_decorations]
self.border_buttons = CheckButtons(frame, entries=border_decorations,
selected=selected,
tipTexts=tipTexts)
self.border_buttons.grid(row=0, column=1, sticky='w')
label = Label(frame, text=' Using Font:')
label.grid(row=0, column=2, sticky='e')
self.border_font_list = FontList(frame, mode='Print',
selected=self.border_font,
tipText='The font used for the border texts')
self.border_font_list.grid(row=0, column=3, sticky='w')
row += 1
label = Label(self, text='Line width:')
label.grid(row=row, column=0, sticky='w')
self.linewidth_entry = FloatEntry(self, width=10,
text=self.linewidth,
tipText='Line width for drawing')
self.linewidth_entry.grid(row=row, column=1, sticky='w')
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='')
x_axis_label = getOption('XAxisLabel', defaultValue='')
y_axis_label = getOption('YAxisLabel', 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])
paper_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_font = getOption('BorderFont', defaultValue='Helvetica 10')
border_decoration = []
if dateTime:
border_decoration.append(border_decorations[0])
if fileName:
border_decoration.append(border_decorations[1])
if self.haveTicks:
spacing_choice = getOption('SpacingChoice', defaultValue=spacing_choices[0])
x_major = getOption('XMajor', defaultValue=1.0)
x_minor = getOption('XMinor', defaultValue=1.0)
x_decimal = getOption('XDecimal', defaultValue=3)
y_major = getOption('YMajor', defaultValue=1.0)
y_minor = getOption('YMinor', defaultValue=1.0)
y_decimal = getOption('YDecimal', defaultValue=3)
tick_length_choice = getOption('TickLengthChoice', defaultValue=tick_length_choices[0])
tick_major = getOption('TickMajor', defaultValue=10)
tick_minor = getOption('TickMinor', defaultValue=5)
scaling_choice = getOption('ScalingChoice', defaultValue=scaling_choices[0])
scaling = getOption('Scaling', defaultValue=0.7)
scaling = int(round(100.0 * scaling))
x_scaling = getOption('XScaling', defaultValue=1.0)
y_scaling = getOption('YScaling', defaultValue=1.0)
if self.haveTicks:
tick_font = getOption('TickFont', defaultValue='Helvetica 10')
linewidth = getOption('LineWidth', defaultValue=Output.default_linewidth)
else:
file_name = ''
title = ''
x_axis_label = ''
y_axis_label = ''
paper_type = paper_types[0]
other_unit = paper_units[0]
other_size = ''
paper_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
border_font = 'Helvetica 10'
if self.haveTicks:
spacing_choice = spacing_choices[0]
x_major = 1.0
x_minor = 1.0
x_decimal = 3
y_major = 1.0
y_minor = 1.0
y_decimal = 3
tick_length_choice = tick_length_choices[0]
tick_major = 10
tick_minor = 5
scaling_choice = scaling_choices[0]
scaling = 70
x_scaling = 1.0
y_scaling = 1.0
if self.haveTicks:
tick_font = 'Helvetica 10'
linewidth = Output.default_linewidth
if not self.haveTicks:
tick_location = None
tick_placement = None
spacing_choice = spacing_choices[0]
x_major = 1.0
x_minor = 1.0
x_decimal = 3
y_major = 1.0
y_minor = 1.0
y_decimal = 3
tick_font = 'Helvetica 10'
tick_length_choice = tick_length_choices[0]
tick_major = 10
tick_minor = 5
self.file_name = file_name
self.title = title
self.x_axis_label = x_axis_label
self.y_axis_label = y_axis_label
self.paper_type = paper_type
self.other_unit = other_unit
self.other_size = other_size
self.paper_orientation = paper_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.border_font = border_font
self.spacing_choice = spacing_choice
self.x_major = x_major
self.x_minor = x_minor
self.x_decimal = x_decimal
self.y_major = y_major
self.y_minor = y_minor
self.y_decimal = y_decimal
self.scaling_choice = scaling_choices[0]
self.scaling = scaling
self.x_scaling = x_scaling
self.y_scaling = y_scaling
self.tick_font = tick_font
self.linewidth = linewidth
self.tick_length_choice = tick_length_choice
self.tick_major = tick_major
self.tick_minor = tick_minor
def setOptionValues(self):
if not hasattr(self, 'file_entry'):
# it looks like on destroy can have function called but file_entry deleted already
return
self.file_name = file_name = self.file_entry.get()
self.title = title = self.title_entry.get()
self.x_axis_label = x_axis_label = self.x_axis_entry.get()
self.y_axis_label = y_axis_label = self.y_axis_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()
self.border_font = border_font = self.border_font_list.getText()
if self.haveTicks:
self.spacing_choice = spacing_choice = self.spacing_menu.getText()
if spacing_choice != spacing_choices[0]:
self.x_major = self.x_major_entry.get()
self.x_minor = self.x_minor_entry.get()
self.x_decimal = self.x_decimal_entry.get()
self.y_major = self.y_major_entry.get()
self.y_minor = self.y_minor_entry.get()
self.y_decimal = self.y_decimal_entry.get()
self.tick_length_choice = tick_length_choice = self.tick_length_menu.getText()
if tick_length_choice != tick_length_choices[0]:
self.tick_major = self.length_major_entry.get()
self.tick_minor = self.length_minor_entry.get()
self.scaling_choice = scaling_choice = self.scaling_menu.getText()
if self.scaling_choice == scaling_choices[0]:
scaling = self.scaling_scale.get()
self.scaling = int(round(scaling))
else:
self.x_scaling = self.x_scaling_entry.get()
self.y_scaling = self.y_scaling_entry.get()
if self.haveTicks:
self.tick_font = self.tick_font_list.getText()
self.linewidth = self.linewidth_entry.get()
setOption = self.setOption
if setOption:
setOption('FileName', value=file_name)
setOption('Title', value=title)
setOption('XAxisLabel', value=x_axis_label)
setOption('YAxisLabel', value=y_axis_label)
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('BorderFont', value=border_font)
if self.haveTicks:
setOption('SpacingChoice', value=spacing_choice)
if spacing_choice != spacing_choices[0]:
setOption('XMajor', self.x_major)
setOption('XMinor', self.x_minor)
setOption('XDecimal', self.x_decimal)
setOption('YMajor', self.y_major)
setOption('YMinor', self.y_minor)
setOption('YDecimal', self.y_decimal)
setOption('TickLengthChoice', value=tick_length_choice)
if tick_length_choice != tick_length_choices[0]:
setOption('TickMajor', self.tick_major)
setOption('TickMinor', self.tick_minor)
setOption('ScalingChoice', value=scaling_choice)
if scaling_choice == scaling_choices[0]:
setOption('Scaling', value=0.01*self.scaling)
else:
setOption('XScaling', value=self.x_scaling)
setOption('YScaling', value=self.y_scaling)
if self.haveTicks:
setOption('TickFont', self.tick_font)
setOption('LineWidth', self.linewidth)
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)
def changedScaling(self, choice):
if choice == scaling_choices[0]:
self.scaling_scale.grid(row=0, column=2, columnspan=3, sticky='ew')
self.x_scaling_label.grid_forget()
self.x_scaling_entry.grid_forget()
self.y_scaling_label.grid_forget()
self.y_scaling_entry.grid_forget()
else:
self.scaling_scale.grid_forget()
self.x_scaling_label.grid(row=0, column=2, sticky='w')
self.x_scaling_entry.grid(row=0, column=3, columnspan=2, sticky='ew')
self.y_scaling_label.grid(row=1, column=2, sticky='w')
self.y_scaling_entry.grid(row=1, column=3, columnspan=2, sticky='ew')
self.setOptionValues()
def changedSpacing(self, choice):
if choice == spacing_choices[0]:
self.spacing_frame.grid_forget()
else:
self.spacing_frame.grid(row=1, column=1, columnspan=3, sticky='ew')
self.setOptionValues()
def changedLength(self, choice):
if choice == tick_length_choices[0]:
self.length_frame.grid_forget()
else:
self.length_frame.grid(row=1, column=0, columnspan=4, sticky='ew')
self.setOptionValues()
# width and height are of plot, in pixels
def getOutputHandler(self, pixel_width, pixel_height, unit_width=1.0, unit_height=1.0, 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
x_scaling = y_scaling = 1.0
if self.scaling_choice != scaling_choices[0]:
try:
x_scaling = float(self.x_scaling)
except:
showError('Bad X Scaling', 'Specified X Scaling must be floating point', parent=self)
return None
try:
y_scaling = float(self.y_scaling)
except:
showError('Bad Y Scaling', 'Specified Y Scaling must be floating point', 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
border_font = self.border_font
(font, size) = border_font.split()
size = int(size)
border_text = {}
for decoration in self.border_decoration:
if decoration == 'Time and 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, size)
if self.title:
location = 'n'
border_text[location] = (self.title, font, size+6)
if font not in fonts:
fonts.append(font)
if self.haveTicks and self.tick_location == PrintTicks.Outside:
axis_label_offset = 2
else:
axis_label_offset = 0
outputHandler = PrintHandler.getOutputHandler(self.file_name,
pixel_width, pixel_height,
unit_width, unit_height,
scaling_choice=self.scaling_choice,
output_scaling=output_scaling,
w_scaling=x_scaling,
h_scaling=y_scaling,
paper_size=paper_size,
paper_orientation=self.paper_orientation,
output_style=self.output_style,
output_format=self.output_format,
border_text=border_text,
x_axis_label=self.x_axis_label,
y_axis_label=self.y_axis_label,
axis_label_font=font,
axis_label_size=size,
axis_label_offset=axis_label_offset,
fonts=fonts,
linewidth=self.linewidth,
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
class UpdateAdministratorPopup(BasePopup, UpdateAgent):
def __init__(self,
parent,
serverLocation=UPDATE_SERVER_LOCATION,
serverDirectory=UPDATE_DIRECTORY,
dataFile=UPDATE_DATABASE_FILE):
UpdateAgent.__init__(self,
serverLocation,
serverDirectory,
dataFile,
admin=1)
self.fileTypes = [
FileType('Python', ['*.py']),
FileType('C', ['*.c']),
FileType('All', ['*'])
]
self.fileUpdate = None
BasePopup.__init__(self,
parent=parent,
title='CcpNmr Update Administrator',
quitFunc=self.quit)
def body(self, guiParent):
guiParent.grid_columnconfigure(3, weight=1)
self.commentsEntry = Entry(self)
self.priorityEntry = IntEntry(self)
row = 0
label = Label(guiParent, text='Server location:')
label.grid(row=row, column=0, sticky='w')
location = ''
uid = ''
httpDir = ''
subDir = ''
version = 'None'
if self.server:
location, uid, httpDir, subDir = self.server.identity
version = self.server.version or 'None'
self.serverEntry = Entry(guiParent, text=location)
self.serverEntry.grid(row=row, column=1, stick='w')
label = Label(guiParent, text='User ID:')
label.grid(row=row, column=2, sticky='w')
self.uidEntry = Entry(guiParent, text=uid)
self.uidEntry.grid(row=row, column=3, stick='w')
row += 1
label = Label(guiParent, text='HTTP directory:')
label.grid(row=row, column=0, sticky='w')
self.httpDirEntry = Entry(guiParent, text=httpDir)
self.httpDirEntry.grid(row=row, column=1, stick='w')
label = Label(guiParent, text='Sub-directory:')
label.grid(row=row, column=2, sticky='w')
self.subDirEntry = Entry(guiParent, text=subDir)
self.subDirEntry.grid(row=row, column=3, stick='w')
row += 1
self.localVerLabel = Label(guiParent,
text='Local version: %s' % self.version)
self.localVerLabel.grid(row=row, column=0, sticky='w')
self.serverLabel = Label(guiParent,
text='Server version: %s' % version)
self.serverLabel.grid(row=row, column=2, sticky='w')
row += 1
guiParent.grid_rowconfigure(row, weight=1)
headingList = [
'File',
'Location',
'Date',
'Priority',
'Comments',
'StoredAs',
]
editWidgets = [
None, None, None, self.priorityEntry, self.commentsEntry
]
editGetCallbacks = [
None, None, None, self.getPriority, self.getComments
]
editSetCallbacks = [
None, None, None, self.setPriority, self.setComments
]
self.scrolledMatrix = ScrolledMatrix(guiParent,
headingList=headingList,
multiSelect=True,
editWidgets=editWidgets,
callback=self.selectCell,
editGetCallbacks=editGetCallbacks,
editSetCallbacks=editSetCallbacks)
self.scrolledMatrix.grid(row=row,
column=0,
columnspan=4,
sticky='nsew')
row += 1
texts = [
'Add\nFiles', 'Remove\nFiles', 'Remove\nAll', 'Query\nServer',
'Commit\nSelected', 'Synchronise\nAll', 'Commit\nNew', 'Quit'
]
commands = [
self.addFile, self.removeFile, self.removeAll, self.queryFiles,
self.synchroniseSelected, self.synchroniseServer,
self.updateServer, self.quit
]
self.buttonList = ButtonList(guiParent,
texts=texts,
commands=commands,
expands=1)
self.buttonList.grid(row=row, column=0, columnspan=4, sticky='ew')
self.update()
def getPriority(self, fileUpdate):
if fileUpdate:
self.priorityEntry.set(fileUpdate.priority)
def setPriority(self, event):
i = self.priorityEntry.get()
if self.fileUpdate:
self.fileUpdate.priority = i
self.update()
def getComments(self, fileUpdate):
if fileUpdate:
self.commentsEntry.set(fileUpdate.details)
def setComments(self, event):
text = self.commentsEntry.get()
if self.fileUpdate:
self.fileUpdate.details = text
self.update()
def quit(self):
self.close()
self.destroy()
sys.exit()
def updateServer(self):
if self.server:
# Unly upload new updates
success = self.server.setFileUpdates()
self.serverLabel.set('Server version: %s' % self.server.version)
if success:
self.queryFiles()
def synchroniseSelected(self):
if self.server:
selectedUpdates = self.scrolledMatrix.currentObjects
if selectedUpdates:
# Refresh all selected updates
success = self.server.setFileUpdates(
fileUpdates=selectedUpdates, refresh=True)
self.serverLabel.set('Server version: %s' %
self.server.version)
if success:
self.queryFiles()
def synchroniseServer(self):
if self.server:
# Refresh all avaiable updates
success = self.server.setFileUpdates(refresh=True)
self.serverLabel.set('Server version: %s' % self.server.version)
if success:
self.queryFiles()
def queryFiles(self):
if self.server:
self.server.getFileUpdates()
self.update()
def addFile(self):
if self.server:
fileSelectPopup = FileSelectPopup(self,
title='Select Source File',
dismiss_text='Cancel',
file_types=self.fileTypes,
multiSelect=True,
selected_file_must_exist=True)
filePaths = fileSelectPopup.file_select.getFiles()
n = len(self.installRoot)
for filePath in filePaths:
if self.installRoot != filePath[:n]:
showWarning(
'Warning',
'Install root %s not found in file path %s' %
(self.installRoot, filePath))
continue
filePath = filePath[n + 1:]
if filePath:
dirName, fileName = splitPath(filePath)
if fileName[-3:] == '.py':
language = 'python'
else:
language = 'None'
fileUpdate = FileUpdate(self.server,
fileName,
dirName,
language,
isNew=True)
self.update()
def removeFile(self):
if self.fileUpdate:
self.fileUpdate.delete()
self.update()
def removeAll(self):
if self.server:
for fileUpdate in list(self.server.fileUpdates):
fileUpdate.delete()
self.update()
def selectAll(self):
if self.server:
for fileUpdate in self.server.fileUpdates:
fileUpdate.isSelected = 1
self.update()
def selectCell(self, object, row, col):
self.fileUpdate = object
self.updateButtons()
def updateButtons(self):
buttons = self.buttonList.buttons
if self.server:
buttons[0].enable()
buttons[3].enable()
buttons[4].enable()
else:
buttons[0].disable()
buttons[3].disable()
buttons[4].disable()
if self.server and self.server.fileUpdates:
buttons[2].enable()
else:
buttons[2].disable()
if self.fileUpdate:
buttons[1].enable()
else:
buttons[1].disable()
def update(self):
location = self.serverEntry.get()
uid = self.uidEntry.get()
httpDir = self.httpDirEntry.get()
subDir = self.subDirEntry.get()
if self.server:
if (location, uid, httpDir, subDir) != self.server.identity:
self.setServer(location)
self.updateButtons()
self.fileUpdate = None
textMatrix = []
objectList = []
colorMatrix = []
if self.server:
for fileUpdate in self.server.fileUpdates:
datum = []
datum.append(fileUpdate.fileName)
datum.append(fileUpdate.filePath)
datum.append(fileUpdate.date)
datum.append(fileUpdate.priority)
datum.append(fileUpdate.details)
datum.append(fileUpdate.storedAs)
textMatrix.append(datum)
objectList.append(fileUpdate)
if fileUpdate.isNew:
colorMatrix.append(5 * ['#B0FFB0'])
elif not fileUpdate.getIsUpToDate():
colorMatrix.append(5 * ['#FFB0B0'])
else:
colorMatrix.append(5 * [None])
self.scrolledMatrix.update(textMatrix=textMatrix,
objectList=objectList,
colorMatrix=colorMatrix)
class BacusPopup(BasePopup):
def __init__(self, parent, *args, **kw):
self.project = parent.project
self.hNoesy = None
self.nNoesy = None
self.cNoesy = None
self.shiftList = None
BasePopup.__init__(self, parent, title="BACUS control module", **kw)
def body(self, guiParent):
guiParent.grid_columnconfigure(0, weight=1)
row = 0
guiParent.grid_rowconfigure(row, weight=1)
inputFrame = LabelFrame(guiParent, text='Inputs:')
inputFrame.grid(row=row, column=0, sticky=Tkinter.NSEW)
label = Label(inputFrame, text='NOESY:')
label.grid(row=0, column=0, sticky=Tkinter.NW)
self.hNoesyPulldown = PulldownMenu(inputFrame,callback=self.setHNoesy)
self.hNoesyPulldown.grid(row=0, column=1, sticky=Tkinter.NW)
label = Label(inputFrame, text='15N NOESY:')
label.grid(row=1, column=0, sticky=Tkinter.NW)
self.nNoesyPulldown = PulldownMenu(inputFrame,callback=self.setNNoesy)
self.nNoesyPulldown.grid(row=1, column=1, sticky=Tkinter.NW)
label = Label(inputFrame, text='13C NOESY:')
label.grid(row=2, column=0, sticky=Tkinter.NW)
self.cNoesyPulldown = PulldownMenu(inputFrame,callback=self.setCNoesy)
self.cNoesyPulldown.grid(row=2, column=1, sticky=Tkinter.NW)
label = Label(inputFrame, text='Shift List:')
label.grid(row=3, column=0, sticky=Tkinter.NW)
self.shiftListPulldown = PulldownMenu(inputFrame,callback=self.setShiftList)
self.shiftListPulldown.grid(row=3, column=1, sticky=Tkinter.NW)
label = Label(inputFrame, text='2d BACUS executable:')
label.grid(row=4, column=0, sticky=Tkinter.NW)
self.executableEntry = Entry(inputFrame, text='/home/tjs23/clouds/justin/SpiBacusMidge/bacus/bacus_tjs.exe')
self.executableEntry.grid(row=4, column=1, sticky=Tkinter.NW)
self.executableButton = Button(inputFrame, text='Choose file', command=self.chooseExecutable)
self.executableButton.grid(row=5, column=1, sticky=Tkinter.EW)
row += 1
outputFrame = LabelFrame(guiParent, text='Output:')
outputFrame.grid(row=row, column=0, sticky=Tkinter.NSEW)
row += 1
texts = ['Run BACUS 2d','Run BACUS 3d']
commands = [self.runBacus,self.runBacus3d]
self.bottomButtons = createDismissHelpButtonList(guiParent,texts=texts,commands=commands,expands=0,help_url=None)
self.bottomButtons.grid(row=row, column=0, sticky=Tkinter.EW)
self.update()
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 update(self):
# set defaults via pulldown callbacks
names = [self.getPeakListlabel(x) for x in self.gethNoesys()]
if names:
self.hNoesyPulldown.setup(names,0)
else:
self.hNoesyPulldown.setup([],-1)
names = [self.getPeakListlabel(x) for x in self.getNNoesys()]
if names:
self.nNoesyPulldown.setup(names,0)
else:
self.nNoesyPulldown.setup([],-1)
names = [self.getPeakListlabel(x) for x in self.getCNoesys()]
if names:
self.cNoesyPulldown.setup(names,0)
else:
self.cNoesyPulldown.setup([],-1)
names = ['Shift List %d' % x.serial for x in self.getShiftLists()]
if names:
self.shiftListPulldown.setup(names,0)
else:
self.shiftListPulldown.setup([],-1)
self.updateButtons()
def updateButtons(self):
if ( self.cNoesy or self.nNoesy ) and self.shiftList:
self.bottomButtons.buttons[1].enable()
else:
self.bottomButtons.buttons[1].disable()
if self.hNoesy and self.executableEntry.get() and self.shiftList:
self.bottomButtons.buttons[0].enable()
else:
self.bottomButtons.buttons[0].disable()
def runBacus3d(self, fileRoot='CCPN1'):
if ( self.cNoesy or self.nNoesy ) and self.shiftList:
run3dBacus(fileRoot,self.shiftList,self.cNoesy,self.nNoesy)
def runBacus(self, fileRoot='CCPN1'):
executable = self.executableEntry.get()
if self.hNoesy and self.shiftList and executable:
run2dBacus(fileRoot, executable, self.hNoesy, self.shiftList)
def getPeakListlabel(self, peakList):
return '%s:%s:%d' % (peakList.dataSource.experiment.name, peakList.dataSource.name, peakList.serial)
def getShiftLists(self):
shiftLists = []
if self.project:
nmrProject = self.project.currentNmrProject
shiftLists = nmrProject.findAllMeasurementLists(className='ShiftList')
return list(shiftLists)
def setShiftList(self, index, name=None):
shiftLists = self.getShiftLists()
if shiftLists:
self.shiftList = shiftLists[index]
else:
self.shiftList = None
def gethNoesys(self):
peakLists = []
if self.project:
for experiment in self.project.currentNmrProject.experiments:
for spectrum in experiment.dataSources:
isotopes = getSpectrumIsotopes(spectrum)
if isotopes == ['1H', '1H']:
for peakList in spectrum.peakLists:
#if 'NOE' in experiment.name.upper():
# peakLists.insert(0, peakList)
#else:
peakLists.append(peakList)
return peakLists
def getNNoesys(self):
peakLists = []
if self.project:
for experiment in self.project.currentNmrProject.experiments:
for spectrum in experiment.dataSources:
isotopes = getSpectrumIsotopes(spectrum)
isotopes.sort()
if isotopes == ['15N', '1H', '1H']:
for peakList in spectrum.peakLists:
#if 'NOE' in experiment.name.upper():
# peakLists.insert(0, peakList)
#else:
peakLists.append(peakList)
return peakLists
def getCNoesys(self):
peakLists = []
if self.project:
for experiment in self.project.currentNmrProject.experiments:
for spectrum in experiment.dataSources:
isotopes = getSpectrumIsotopes(spectrum)
isotopes.sort()
if isotopes == ['13C', '1H', '1H']:
for peakList in spectrum.peakLists:
if 'NOE' in experiment.name.upper():
peakLists.insert(0, peakList)
else:
peakLists.append(peakList)
return peakLists
def setHNoesy(self, index, name=None):
peakLists = self.gethNoesys()
if peakLists:
self.hNoesy = peakLists[index]
else:
self.hNoesy = None
def setNNoesy(self, index, name=None):
peakLists = self.getNNoesys()
if peakLists:
self.nNoesy = peakLists[index]
else:
self.nNoesy = None
def setCNoesy(self, index, name=None):
peakLists = self.getCNoesys()
if peakLists:
self.cNoesy = peakLists[index]
else:
self.cNoesy = None
def destroy(self):
BasePopup.destroy(self)
class OpenMacroPopup(BasePopup):
"""
**Locate Python Macro Scripts On Disk**
This popup is used to locate a Python file and function definition on disk
so that it can be loaded into Analysis as a "Macro". The top table
is a file browser that the user can navigate to the location of the Python
file (typically sending in ".py"). When a Python file is selected in the
upper table the system will look through the contents of the file to find
any macro functions that are defined within. Note that a macro function
is defined by the normal Python "def" keyword but must take "argSever"
as its first input argument.
Any macro functions in the selected file will be displayed in the
lower table. Clicking on the name of a function in this table
selects it for opening. The user may edit the name of the macro for
display purposes within Analysis. Finally clicking [Load Macro]
will add it to the CCPN project, stored in a user-specific way
inside the current Analysis profile. Once loaded into the
project the new macro will appear in the main macro table
(the "Macros" tab of `User Options`_)
.. _`User Options`: EditProfilesPopup.html
"""
def __init__(self, parent, **kw):
# TBD: properly
transient = True
BasePopup.__init__(self, parent=parent, title='Open macro', transient=transient, **kw)
def body(self, guiParent):
guiParent.grid_columnconfigure(1,weight=1)
row = 0
file_types = [ FileType('Python', ['*.py']), FileType('All', ['*']) ]
self.file_select = FileSelect(guiParent, file_types=file_types,
single_callback=self.chooseFile,
double_callback=self.chooseFile)
self.file_select.grid(row=row, column=0, columnspan=2, sticky='nsew')
row = row + 1
headingList=('Function',)
self.scrolledMatrix = ScrolledMatrix(guiParent, initialRows=4,
headingList=headingList, callback=self.selectFunction)
self.scrolledMatrix.grid(row=row, column=0, columnspan=2, sticky='nsew')
guiParent.grid_rowconfigure(row,weight=1)
row = row + 1
self.moduleLabel1 = Label(guiParent, text='Module: ')
self.moduleLabel1.grid(row=row, column=0, sticky='nw')
self.moduleLabel2 = Label(guiParent, text=' ')
self.moduleLabel2.grid(row=row, column=1, sticky='nw')
row = row + 1
self.functionLabel1 = Label(guiParent, text='Function: ')
self.functionLabel1.grid(row=row, column=0, sticky='nw')
self.functionLabel2 = Label(guiParent, text=' ')
self.functionLabel2.grid(row=row, column=1, sticky='nw')
row = row + 1
self.nameLabel = Label(guiParent, text='Name: ')
self.nameLabel.grid(row=row, column=0, sticky='nw')
self.nameEntry = Entry(guiParent, text=' ', width=40)
self.nameEntry.grid(row=row, column=1, sticky='nw')
row = row + 1
texts = [ 'Load Macro' ]
commands = [ self.loadMacro ]
buttons = UtilityButtonList(guiParent, texts=texts,
commands=commands, helpUrl=self.help_url)
buttons.grid(row=row, column=0, columnspan=2, sticky='ew')
self.loadButton = buttons.buttons[0]
self.loadButton.disable()
self.path = None
self.module = None
self.function = None
def selectFunction(self, function, row, col):
if function:
self.loadButton.enable()
self.function = function
self.functionLabel2.set(function)
self.nameEntry.set(function)
def updateFunctions(self, file):
self.loadButton.disable()
functions = []
fileHandle = open(file, 'r')
line = fileHandle.readline()
textMatrix = []
while line :
match = re.match('\s*def\s+(\w+)\s*\(.*argServer',line)
if match:
function = match.group(1)
functions.append(function)
textMatrix.append( [function,] )
line = fileHandle.readline()
self.scrolledMatrix.update(objectList=functions,textMatrix=textMatrix)
def chooseFile(self, file):
fullPath =self.file_select.getFile()
if not os.path.isdir(fullPath) and file:
(self.path, module) = splitPath(fullPath)
module = re.sub('\..+?$','',module)
self.module = module
self.moduleLabel2.set(module)
self.updateFunctions(file)
def loadMacro(self):
if self.module and self.function and self.path:
name = self.nameEntry.get()
if not name:
name = self.function
analysisProfile = self.analysisProfile
m1 = analysisProfile.newMacro(name=name,path=self.path,
function=self.function,
module=self.module,ordering=1)
self.close()
class ArchiveProjectPopup(BasePopup):
"""
**Archive the Project**
This popup window enables the user to archive the CCPN Project into a gzipped
tar file. By default it only includes the *.xml files which reside in the
project directory. If the "Include *.xml.bak files" check button is checked
then it will also include the *.xml.bak files which reside in the project
directory. If the "Include data files which are in the project directory"
check button is checked then it will also include the binary data files
which reside in the project directory.
**Caveats & Tips**
The archive excludes the reference data *.xml files (quite sensibly).
"""
def __init__(self,
parent,
project,
title='Project : Archive',
callback=None,
help_msg='',
help_url='',
dismiss_text='',
*args,
**kw):
self.callback = callback
self.help_msg = help_msg
self.help_url = help_url
self.dismiss_text = dismiss_text
BasePopup.__init__(self,
parent=parent,
project=project,
title=title,
*args,
**kw)
def body(self, guiParent):
now = datetime.date.today().strftime('%y%m%d')
filePrefix = '%s_%s' % (self.project.name, now)
projDir = getUserDataPath(self.project)
directory = os.path.dirname(projDir)
guiParent.grid_rowconfigure(0, weight=1)
guiParent.grid_columnconfigure(1, weight=1)
row = 0
label = Label(guiParent, text='Archive File:')
label.grid(row=row, column=0, sticky=Tkinter.E)
tipText = 'File name (excluding .tgz ending) for archive'
self.fileEntry = Entry(guiParent, text=filePrefix, tipText=tipText)
self.fileEntry.grid(row=row, column=1, sticky=Tkinter.EW)
label = Label(guiParent, text='.tgz (automatically appended)')
label.grid(row=row, column=2, sticky=Tkinter.W)
row = row + 1
self.backupCheck = CheckButton(
guiParent,
text='Include *.xml.bak files',
tipText='If checked include *.xml.bak files')
self.backupCheck.grid(row=row,
column=1,
columnspan=2,
sticky=Tkinter.W)
row = row + 1
self.dataCheck = CheckButton(
guiParent,
text='Include data files which are in project directory',
tipText=
'If checked include data files if they are located in project directory'
)
self.dataCheck.grid(row=row, column=1, columnspan=2, sticky=Tkinter.W)
row = row + 1
labelFrame = LabelFrame(guiParent, text='Archive Location')
labelFrame.grid(row=row, column=0, columnspan=3, sticky=Tkinter.NSEW)
labelFrame.grid_rowconfigure(0, weight=1)
labelFrame.grid_columnconfigure(0, weight=1)
self.dirSelect = FileSelect(labelFrame,
directory=directory,
show_file=False)
self.dirSelect.grid(row=0, column=0, sticky=Tkinter.NSEW)
guiParent.grid_rowconfigure(row, weight=1)
row = row + 1
texts = ['Save']
tipTexts = ['Create archive file']
commands = [self.save]
buttons = createDismissHelpButtonList(guiParent,
texts=texts,
tipTexts=tipTexts,
commands=commands,
help_msg=self.help_msg,
help_url=self.help_url,
dismiss_text=self.dismiss_text,
expands=True)
buttons.grid(row=row, column=0, columnspan=3, sticky=Tkinter.EW)
def save(self):
filePrefix = self.fileEntry.get()
directory = self.dirSelect.getDirectory()
filePrefix = joinPath(directory, filePrefix)
fileName = filePrefix + '.tgz'
if os.path.exists(fileName):
if not showYesNo('File exists',
'File "%s" exists, overwrite?' % fileName,
parent=self):
return
includeBackups = self.backupCheck.isSelected()
includeData = self.dataCheck.isSelected()
packageProject(self.project, filePrefix, includeBackups, includeData)
class SaveProjectFrame(Frame):
def __init__(self,
guiParent,
project,
callback=None,
help_msg='',
help_url='',
dismiss_text='',
modal=False,
*args,
**kw):
self.project = project
self.callback = callback
self.help_msg = help_msg
self.help_url = help_url
self.dismiss_text = dismiss_text
self.modal = modal
self.did_save = False
Frame.__init__(self, guiParent, *args, **kw)
projDir = Io.getUserDataPath(self.project)
guiParent.grid_columnconfigure(1, weight=1)
row = 0
label = Label(guiParent, text='Project Name:')
label.grid(row=row, column=0, sticky=Tkinter.E)
self.proj_name_entry = Entry(
guiParent,
text=self.project.name,
returnCallback=self.updateInfo,
leaveCallback=self.updateInfo,
tipText='The name used for the project save directory')
self.proj_name_entry.grid(row=row, column=1, sticky=Tkinter.EW)
row = row + 1
label = Label(guiParent, text='Project Directory:')
label.grid(row=row, column=0, sticky=Tkinter.E)
label = self.proj_dir_label = Label(guiParent, text=projDir)
label.grid(row=row, column=1, sticky=Tkinter.W)
row = row + 1
label = Label(
guiParent,
text='Note: Project Directory = Save Location + Project Name')
label.grid(row=row, column=1, sticky=Tkinter.W)
text = 'Save binary data with project'
tipText = 'Copy data files (e.g. for spectra) into new project directory if not already in current project directory: careful, this can take some time'
row = row + 1
self.dataCheckButton = CheckButton(guiParent,
text=text,
tipText=tipText)
self.dataCheckButton.grid(row=row,
column=0,
columnspan=2,
sticky=Tkinter.W)
row = row + 1
guiParent.grid_rowconfigure(row, weight=1)
labelFrame = LabelFrame(guiParent, text='Save Location')
labelFrame.grid(row=row, column=0, columnspan=2, sticky=Tkinter.NSEW)
labelFrame.grid_rowconfigure(0, weight=1)
labelFrame.grid_columnconfigure(0, weight=1)
directory = os.path.dirname(projDir)
self.proj_dir_select = FileSelect(
labelFrame,
directory=directory,
select_dir_callback=self.selectDir,
change_dir_callback=self.updateInfo,
should_change_dir_callback=self.shouldChangeDir,
getRowColor=self.getEntryColor,
show_file=False)
self.proj_dir_select.grid(row=0, column=0, sticky=Tkinter.NSEW)
row = row + 1
texts = ['Save']
tipTexts = ['Save project with specified name in specified directory']
commands = [self.save]
buttons = createDismissHelpButtonList(guiParent,
texts=texts,
tipTexts=tipTexts,
commands=commands,
help_msg=self.help_msg,
help_url=self.help_url,
dismiss_text=self.dismiss_text,
expands=True)
buttons.grid(row=row, column=0, columnspan=2, sticky=Tkinter.EW)
def save(self):
projName = self.proj_name_entry.get()
directory = self.proj_dir_select.getDirectory()
directory = joinPath(directory, projName)
if self.isProjectDirectory(directory):
if not showOkCancel('Overwrite directory',
'Overwrite existing project directory?',
parent=self):
return
self.updateInfo()
self.did_save = False
changeDataLocations = self.dataCheckButton.isSelected()
done = False
try:
done = Io.saveProject(self.project,
newPath=directory,
newProjectName=projName,
createFallback=True,
showYesNo=showYesNo,
changeDataLocations=changeDataLocations,
showWarning=showWarning)
if done:
showInfo('Project saved', 'Project saved successfully')
self.did_save = True
if self.callback:
self.callback(self.project)
elif self.modal:
return # give another chance
except Implementation.ApiError, e:
showError('Save project', e.error_msg)
except IOError, e:
showError('Save project', str(e))
class CloudsPopup(BasePopup):
def __init__(self, parent, *args, **kw):
self.guiParent = parent
self.project = parent.getProject()
self.waiting = 0
self.specFreq = 800.13
self.maxIter = 50
self.mixTime = 60
self.corrTime = 11.5
self.leakRate = 2.0
self.peakListDict = {}
self.noesyPeakList = None
self.tocsyPeakList = None
self.noesy3dPeakList = None
self.hsqcPeakList = None
self.maxIntens = 37000000
self.resonances = None
self.origResonances = None
self.noesyPeaks = None
self.distanceConstraintList = None
self.antiDistConstraintList = None
self.numClouds = 100
self.filePrefix = 'cloud_'
self.cloudsFiles = []
self.adcAtomTypes = 'HN'
self.structure = None
# step num, initial temp, final temp, cooling steps, MD steps, MD tau, rep scale
self.coolingScheme = []
self.coolingScheme.append([1, 1, 1, 3, 500, 0.001, 0])
self.coolingScheme.append([2, 80000, 4000, 19, 1000, 0.001, 0])
self.coolingScheme.append([3, 4000, 1, 5, 500, 0.001, 0])
self.coolingScheme.append([4, 15000, 1, 3, 1000, 0.001, 0])
self.coolingScheme.append([5, 1, 1, 5, 500, 0.001, 0])
self.coolingScheme.append([6, 8000, 1, 3, 1000, 0.001, 0])
self.coolingScheme.append([7, 1, 1, 5, 500, 0.001, 0])
self.coolingScheme.append([8, 3000, 25, 60, 2500, 0.001, 1])
self.coolingScheme.append([9, 25, 25, 1, 7500, 0.001, 1])
self.coolingScheme.append([10, 10, 10, 1, 7500, 0.001, 1])
self.coolingScheme.append([11, 0.01, 0.01, 1, 7500, 0.0005, 1])
self.coolingStep = None
BasePopup.__init__(self,
parent,
title="Resonance Clouds Analysis",
**kw)
def body(self, guiFrame):
self.specFreqEntry = IntEntry(self,
text=self.specFreq,
width=8,
returnCallback=self.setSpecFreq)
self.maxIterEntry = IntEntry(self,
text=self.maxIter,
width=8,
returnCallback=self.setMaxIter)
self.mixTimeEntry = FloatEntry(self,
text=self.mixTime,
width=8,
returnCallback=self.setMixTime)
self.corrTimeEntry = FloatEntry(self,
text=self.corrTime,
width=8,
returnCallback=self.setCorrTime)
self.leakRateEntry = FloatEntry(self,
text=self.leakRate,
width=8,
returnCallback=self.setLeakRate)
self.maxIntensEntry = IntEntry(self,
text=self.maxIntens,
width=8,
returnCallback=self.setMaxIntens)
self.mdInitTempEntry = FloatEntry(self,
text='',
returnCallback=self.setMdInitTemp)
self.mdFinTempEntry = FloatEntry(self,
text='',
returnCallback=self.setMdFinTemp)
self.mdCoolStepsEntry = IntEntry(self,
text='',
returnCallback=self.setMdCoolSteps)
self.mdSimStepsEntry = IntEntry(self,
text='',
returnCallback=self.setMdSimSteps)
self.mdTauEntry = FloatEntry(self,
text='',
returnCallback=self.setMdTau)
self.mdRepScaleEntry = FloatEntry(self,
text='',
returnCallback=self.setMdRepScale)
guiFrame.grid_columnconfigure(0, weight=1)
row = 0
frame0 = LabelFrame(guiFrame, text='Setup peak lists')
frame0.grid(row=row, column=0, sticky=Tkinter.NSEW)
frame0.grid(row=row, column=0, sticky=Tkinter.NSEW)
frame0.grid_columnconfigure(1, weight=1)
f0row = 0
label00 = Label(frame0, text='1H-1H NOESY spectrum')
label00.grid(row=f0row, column=0, sticky=Tkinter.NW)
self.noesyPulldown = PulldownMenu(frame0,
entries=self.getNoesys(),
callback=self.setNoesy,
selected_index=0,
do_initial_callback=0)
self.noesyPulldown.grid(row=f0row, column=1, sticky=Tkinter.NW)
f0row += 1
label01 = Label(frame0, text='15N HSQC spectrum')
label01.grid(row=f0row, column=0, sticky=Tkinter.NW)
self.hsqcPulldown = PulldownMenu(frame0,
entries=self.getHsqcs(),
callback=self.setHsqc,
selected_index=0,
do_initial_callback=0)
self.hsqcPulldown.grid(row=f0row, column=1, sticky=Tkinter.NW)
f0row += 1
label02 = Label(frame0, text='15N HSQC TOCSY spectrum')
label02.grid(row=f0row, column=0, sticky=Tkinter.NW)
self.tocsyPulldown = PulldownMenu(frame0,
entries=self.getTocsys(),
callback=self.setTocsy,
selected_index=0,
do_initial_callback=0)
self.tocsyPulldown.grid(row=f0row, column=1, sticky=Tkinter.NW)
f0row += 1
label02 = Label(frame0, text='15N HSQC NOESY spectrum')
label02.grid(row=f0row, column=0, sticky=Tkinter.NW)
self.noesy3dPulldown = PulldownMenu(frame0,
entries=self.getNoesy3ds(),
callback=self.setNoesy3d,
selected_index=0,
do_initial_callback=0)
self.noesy3dPulldown.grid(row=f0row, column=1, sticky=Tkinter.NW)
f0row += 1
texts = ['Setup resonances & peaks', 'Show Peaks', 'Show resonances']
commands = [self.setupResonances, self.showPeaks, self.showResonances]
self.setupButtons = ButtonList(frame0,
expands=1,
texts=texts,
commands=commands)
self.setupButtons.grid(row=f0row,
column=0,
columnspan=2,
sticky=Tkinter.NSEW)
f0row += 1
self.label03a = Label(frame0, text='Resonances found: 0')
self.label03a.grid(row=f0row, column=0, sticky=Tkinter.NW)
self.label03b = Label(frame0, text='NOESY peaks found: 0')
self.label03b.grid(row=f0row, column=1, sticky=Tkinter.NW)
row += 1
frame1 = LabelFrame(guiFrame, text='Calculate distance constraints')
frame1.grid(row=row, column=0, sticky=Tkinter.NSEW)
frame1.grid_columnconfigure(3, weight=1)
f1row = 0
frame1.grid_rowconfigure(f1row, weight=1)
data = [
self.specFreq, self.maxIter, self.mixTime, self.corrTime,
self.leakRate, self.maxIntens
]
colHeadings = [
'Spectrometer\nfrequency', 'Max\niterations', 'Mixing\ntime (ms)',
'Correl.\ntime (ns)', 'Leak\nrate', 'Max\nintensity'
]
editWidgets = [
self.specFreqEntry,
self.maxIterEntry,
self.mixTimeEntry,
self.corrTimeEntry,
self.leakRateEntry,
self.maxIntensEntry,
]
editGetCallbacks = [
self.getSpecFreq,
self.getMaxIter,
self.getMixTime,
self.getCorrTime,
self.getLeakRate,
self.getMaxIntens,
]
editSetCallbacks = [
self.setSpecFreq,
self.setMaxIter,
self.setMixTime,
self.setCorrTime,
self.setLeakRate,
self.setMaxIntens,
]
self.midgeParamsMatrix = ScrolledMatrix(
frame1,
editSetCallbacks=editSetCallbacks,
editGetCallbacks=editGetCallbacks,
editWidgets=editWidgets,
maxRows=1,
initialCols=5,
headingList=colHeadings,
callback=None,
objectList=[
'None',
],
textMatrix=[
data,
])
self.midgeParamsMatrix.grid(row=f1row,
column=0,
columnspan=4,
sticky=Tkinter.NSEW)
f1row += 1
label10 = Label(frame1, text='Benchmark structure')
label10.grid(row=f1row, column=0, sticky=Tkinter.NW)
self.structurePulldown = PulldownMenu(frame1,
entries=self.getStructures(),
callback=self.setStructure,
selected_index=0,
do_initial_callback=0)
self.structurePulldown.grid(row=f1row, column=1, sticky=Tkinter.NW)
label11 = Label(frame1, text='ADC atom types:')
label11.grid(row=f1row, column=2, sticky=Tkinter.NW)
self.adcAtomsPulldown = PulldownMenu(frame1,
entries=self.getAdcAtomTypes(),
callback=self.setAdcAtomTypes,
selected_index=0,
do_initial_callback=0)
self.adcAtomsPulldown.grid(row=f1row, column=3, sticky=Tkinter.NW)
f1row += 1
texts = [
'Calculate distances', 'Show distance\nconstraints',
'Show anti-distance\nconstraints'
]
commands = [
self.calculateDistances, self.showConstraints,
self.showAntiConstraints
]
self.midgeButtons = ButtonList(frame1,
expands=1,
texts=texts,
commands=commands)
self.midgeButtons.grid(row=f1row,
column=0,
columnspan=4,
sticky=Tkinter.NSEW)
f1row += 1
self.distConstrLabel = Label(frame1, text='Distance constraints:')
self.distConstrLabel.grid(row=f1row,
column=0,
columnspan=2,
sticky=Tkinter.NW)
self.antiConstrLabel = Label(frame1, text='Anti-distance constraints:')
self.antiConstrLabel.grid(row=f1row,
column=2,
columnspan=2,
sticky=Tkinter.NW)
row += 1
guiFrame.grid_rowconfigure(row, weight=1)
frame2 = LabelFrame(guiFrame, text='Proton cloud molecular dynamics')
frame2.grid(row=row, column=0, sticky=Tkinter.NSEW)
frame2.grid_columnconfigure(1, weight=1)
f2row = 0
frame2.grid_rowconfigure(f2row, weight=1)
data = [
self.specFreq, self.maxIter, self.mixTime, self.corrTime,
self.leakRate
]
colHeadings = [
'Step', 'Initial temp.', 'Final temp.', 'Cooling steps',
'MD steps', 'MD tau', 'Rep. scale'
]
editWidgets = [
None, self.mdInitTempEntry, self.mdFinTempEntry,
self.mdCoolStepsEntry, self.mdSimStepsEntry, self.mdTauEntry,
self.mdRepScaleEntry
]
editGetCallbacks = [
None, self.getMdInitTemp, self.getMdFinTemp, self.getMdCoolSteps,
self.getMdSimSteps, self.getMdTau, self.getMdRepScale
]
editSetCallbacks = [
None, self.setMdInitTemp, self.setMdFinTemp, self.setMdCoolSteps,
self.setMdSimSteps, self.setMdTau, self.setMdRepScale
]
self.coolingSchemeMatrix = ScrolledMatrix(
frame2,
editSetCallbacks=editSetCallbacks,
editGetCallbacks=editGetCallbacks,
editWidgets=editWidgets,
maxRows=9,
initialRows=12,
headingList=colHeadings,
callback=self.selectCoolingStep,
objectList=self.coolingScheme,
textMatrix=self.coolingScheme)
self.coolingSchemeMatrix.grid(row=f2row,
column=0,
columnspan=4,
sticky=Tkinter.NSEW)
f2row += 1
texts = ['Move earlier', 'Move later', 'Add step', 'Remove step']
commands = [
self.moveStepEarlier, self.moveStepLater, self.addCoolingStep,
self.removeCoolingStep
]
self.coolingSchemeButtons = ButtonList(frame2,
expands=1,
commands=commands,
texts=texts)
self.coolingSchemeButtons.grid(row=f2row,
column=0,
columnspan=4,
sticky=Tkinter.EW)
f2row += 1
label20 = Label(frame2, text='Number of clouds:')
label20.grid(row=f2row, column=0, sticky=Tkinter.NW)
self.numCloudsEntry = FloatEntry(frame2,
text=100,
returnCallback=self.setNumClouds,
width=10)
self.numCloudsEntry.grid(row=f2row, column=1, sticky=Tkinter.NW)
label21 = Label(frame2, text='Cloud file prefix:')
label21.grid(row=f2row, column=2, sticky=Tkinter.NW)
self.filePrefixEntry = Entry(frame2,
text='cloud_',
returnCallback=self.setFilePrefix,
width=10)
self.filePrefixEntry.grid(row=f2row, column=3, sticky=Tkinter.NW)
f2row += 1
texts = ['Start molecular dynamics', 'Show dynamics progress']
commands = [self.startMd, self.showMdProgress]
self.mdButtons = ButtonList(frame2,
expands=1,
commands=commands,
texts=texts)
self.mdButtons.grid(row=f2row,
column=0,
columnspan=4,
sticky=Tkinter.NSEW)
row += 1
self.bottomButtons = createDismissHelpButtonList(guiFrame,
expands=0,
help_url=None)
self.bottomButtons.grid(row=row, column=0, sticky=Tkinter.EW)
self.setButtonStates()
def getStructures(self):
names = [
'<None>',
]
for molSystem in self.project.sortedMolSystems():
for structure in molSystem.sortedStructureEnsembles():
names.append('%s:%d' % (molSystem.name, structure.ensembleId))
return names
def setStructure(self, index, name=None):
if index < 1:
self.structure = None
else:
structures = []
for molSystem in self.project.molSystems:
for structure in molSystem.structureEnsembles:
structures.append(structure)
self.structure = structures[index - 1]
def getAdcAtomTypes(self):
return ['HN', 'HN HA', 'HN HA HB']
def setAdcAtomTypes(self, index, name=None):
if name is None:
name = self.adcAtomsPulldown.getSelected()
self.adcAtomTypes = name
def startMd(self):
self.setNumClouds()
self.setFilePrefix()
if (self.distanceConstraintList and self.antiDistConstraintList
and (self.numClouds > 0) and self.filePrefix):
resDict = {}
for resonance in self.guiParent.project.currentNmrProject.resonances:
resDict[resonance.serial] = resonance
resonances = []
for constraint in self.distanceConstraintList.constraints:
for item in constraint.items:
for fixedResonance in item.resonances:
if resDict.get(
fixedResonance.resonanceSerial) is not None:
resonances.append(
resDict[fixedResonance.resonanceSerial])
resDict[fixedResonance.resonanceSerial] = None
startMdProcess(self.numClouds, self.distanceConstraintList,
resonances, self.coolingScheme, self.filePrefix)
#structGen = self.distanceConstraintList.structureGeneration
serials = []
for resonance in resonances:
serials.append(resonance.serial)
clouds = []
for i in range(self.numClouds):
clouds.append('%s%3.3d.pdb' % (self.filePrefix, i))
self.guiParent.application.setValues(
self.distanceConstraintList.nmrConstraintStore,
'clouds',
values=clouds)
self.guiParent.application.setValues(
self.distanceConstraintList.nmrConstraintStore,
'cloudsResonances',
values=serials)
# do better than this check for creation
def showMdProgress(self):
n = 0
m = self.numClouds
for i in range(m):
pdbFileName = '%s%3.3d.pdb' % (self.filePrefix, i)
if os.path.exists(pdbFileName):
n += 1
p = n * 100 / float(m)
text = 'Done %d of %d clouds (%1.2f)%%' % (n, m, p)
showInfo('MD Progress', text)
def setFilePrefix(self, text=None):
if not text:
text = self.filePrefixEntry.get()
if text:
self.filePrefix = text
def setNumClouds(self, n=None, *event):
if not n:
n = self.numCloudsEntry.get()
if n:
self.numClouds = int(n)
def calculateDistances(self):
# setup normalisation factor intensityMax
# self.maxIter
# what if failure ?
resDict = {}
for resonance in self.project.currentNmrProject.resonances:
resDict[resonance.serial] = resonance
self.resonances = self.origResonances
intensityFactors = [1.0 for x in range(len(self.resonances))]
# optimiseRelaxation will remove unconstrained resonances
self.distanceConstraintList = optimiseRelaxation(
self.resonances,
self.noesyPeaks,
intensityMax=self.maxIntens,
intensityFactors=intensityFactors,
tmix=self.mixTime,
sf=self.specFreq,
tcor=self.corrTime,
rleak=self.leakRate)
constrainSpinSystems(self.distanceConstraintList)
# for testing calculate distances from structure overrides any resonances: uses assigned ones
#(self.distanceConstraintList, self.resonances) = self.cheatForTesting()
#self.antiDistConstraintList = self.distanceConstraintList
protonNumbs = {'CH3': 3, 'Haro': 2, 'HN': 1, 'H': 1}
PI = 3.1415926535897931
GH = 2.6752e4
HB = 1.05459e-27
CONST = GH * GH * GH * GH * HB * HB
tc = 1.0e-9 * self.corrTime
wh = 2.0 * PI * self.specFreq * 1.0e6
j0 = CONST * tc
j1 = CONST * tc / (1.0 + wh * wh * tc * tc)
j2 = CONST * tc / (1.0 + 4.0 * wh * wh * tc * tc)
#jself = 6.0*j2 + 3.0*j1 + j0
jcross = 6.0 * j2 - j0
if self.distanceConstraintList:
constraintStore = self.distanceConstraintList.nmrConstraintStore
dict = {
'HN': ['H'],
'HN HA': ['H', 'HA', 'HA1', 'HA2'],
'HN HA HB': ['H', 'HA', 'HA1', 'HA2', 'HB', 'HB2', 'HB3']
}
self.antiDistConstraintList = makeNoeAdcs(
self.resonances,
self.noesyPeakList.dataSource,
constraintStore,
allowedAtomTypes=dict[self.adcAtomTypes])
if self.structure:
N = len(self.resonances)
sigmas = [[] for i in range(N)]
for i in range(N):
sigmas[i] = [0.0 for j in range(N)]
for constraint in self.distanceConstraintList.constraints:
resonances = list(constraint, findFirstItem().resonances)
ri = resDict[resonances[0].resonanceSerial]
rj = resDict[resonances[1].resonanceSerial]
i = self.resonances.index(ri)
j = self.resonances.index(rj)
atomSets1 = list(ri.resonanceSet.atomSets)
atomSets2 = list(rj.resonanceSet.atomSets)
if atomSets1 == atomSets2:
ass = list(atomSets1)
atomSets1 = [
ass[0],
]
atomSets2 = [
ass[-1],
]
distance = getAtomSetsDistance(atomSets1, atomSets2,
self.structure)
r = distance * 1e-8
nhs = protonNumbs[rj.name]
sigma = 0.1 * jcross * nhs / (r**6)
sigmas[i][j] = sigma
constraint.setDetails('Known Dist: %4.3f' % (distance))
#for constraint in self.antiDistConstraintList.constraints:
# atomSets1 = list(resonances[0].resonanceSet.atomSets)
# atomSets2 = list(resonances[1].resonanceSet.atomSets)
# distance = getAtomSetsDistance(atomSets1, atomSets2, self.structure)
# constraint.setDetails('Known Dist: %4.3f' % (distance))
fp = open('sigmas.out', 'w')
for i in range(N - 1):
for j in range(i + 1, N):
if sigmas[i][j] != 0.0:
fp.write('%3.1d %3.1d %9.2e\n' %
(i, j, sigmas[i][j]))
#fp.write('\n')
fp.close()
self.setButtonStates()
def cheatForTesting(self, atomSelection='H'):
""" Makes a perfect cloud from a structure. """
project = self.project
structure = self.guiParent.argumentServer.getStructure()
constraintStore = makeNmrConstraintStore(project)
distConstraintList = NmrConstraint.DistanceConstraintList(
constraintStore)
chain = structure.findFirstCoodChain()
structureGeneration.hydrogenResonances = []
molSystem = structure.molSystem
atomSets = []
resonances = []
i = 0
for resonance in project.currentNmrProject.resonances:
if resonance.isotopeCode == '1H':
if resonance.resonanceSet:
atomSet = resonance.resonanceSet.findFirstAtomSet()
atom = atomSet.findFirstAtom()
seqId = atom.residue.seqId
if (seqId < 9) or (seqId > 78):
continue
if atom.residue.chain.molSystem is molSystem:
if atomSelection == 'methyl':
if len(atomSet.atoms) == 3:
if atom.residue.ccpCode not in ('Ala', 'Val',
'Ile', 'Leu',
'Thr', 'Met'):
continue
elif atom.name != 'H':
continue
elif atomSelection == 'amide':
if atom.name != 'H':
continue
if atom.name == 'H':
resonance.name = 'HN'
else:
resonance.name = 'H'
resonances.append(resonance)
atomSets.append(list(resonance.resonanceSet.atomSets))
i += 1
print "Found %d atomSets" % (len(atomSets))
weight = 1
adcWeight = 1
constrDict = {}
N = len(atomSets)
for i in range(N - 1):
atomSets0 = atomSets[i]
residue0 = atomSets0[0].findFirstAtom().residue.seqId
print "R", residue0
for j in range(i + 1, N):
if j == i:
continue
atomSets1 = atomSets[j]
dist = getAtomSetsDistance(atomSets0, atomSets1, structure)
if not dist:
continue
if dist < 5.5:
fixedResonance0 = getFixedResonance(
constraintStore, resonances[i])
fixedResonance1 = getFixedResonance(
constraintStore, resonances[j])
constrDict[i] = 1
constrDict[j] = 1
constraint = NmrConstraint.DistanceConstraint(
distConstraintList,
weight=weight,
targetValue=dist,
upperLimit=dist + (dist / 10),
lowerLimit=dist - (dist / 10),
error=dist / 5)
item = NmrConstraint.DistanceConstraintItem(
constraint,
resonances=[fixedResonance0, fixedResonance1])
elif (atomSets1[0].findFirstAtom().name
== 'H') and (atomSets0[0].findFirstAtom().name
== 'H') and (dist > 7):
#else:
fixedResonance0 = getFixedResonance(
constraintStore, resonances[i])
fixedResonance1 = getFixedResonance(
constraintStore, resonances[j])
constrDict[i] = 1
constrDict[j] = 1
constraint = NmrConstraint.DistanceConstraint(
distConstraintList,
weight=adcWeight,
targetValue=75,
upperLimit=175,
lowerLimit=5.0,
error=94.5)
item = NmrConstraint.DistanceConstraintItem(
constraint,
resonances=[fixedResonance0, fixedResonance1])
return (distConstraintList, resonances)
def showConstraints(self):
if self.distanceConstraintList:
self.guiParent.browseConstraints(
constraintList=self.distanceConstraintList)
def showAntiConstraints(self):
if self.antiDistConstraintList:
self.guiParent.browseConstraints(
constraintList=self.antiDistConstraintList)
def showPeaks(self):
self.guiParent.viewPeaks(peaks=self.noesyPeaks)
def showResonances(self):
pass
#self.guiParent.viewResonances(resonances=self.resonances)
def setupResonances(self):
if self.noesyPeakList and self.noesy3dPeakList and self.tocsyPeakList and self.hsqcPeakList:
disambiguateNoesyPeaks(self.noesyPeakList, self.noesy3dPeakList,
self.tocsyPeakList, self.hsqcPeakList)
(self.origResonances, self.noesyPeaks,
null) = getCloudsResonanceList(self.guiParent.argumentServer,
hsqcPeakList=self.hsqcPeakList,
tocsy3dPeakList=self.tocsyPeakList,
noesy2dPeakList=self.noesyPeakList)
self.setButtonStates()
def setButtonStates(self):
if self.origResonances:
self.label03a.set('Resonances found: %d' %
(len(self.origResonances)))
if self.noesyPeaks:
self.label03b.set('NOESY peaks found: %d' % (len(self.noesyPeaks)))
if self.noesyPeakList and self.tocsyPeakList and self.hsqcPeakList:
self.setupButtons.buttons[0].enable()
else:
self.setupButtons.buttons[0].disable()
if self.noesyPeaks:
self.setupButtons.buttons[1].enable()
else:
self.setupButtons.buttons[1].disable()
if self.origResonances:
self.setupButtons.buttons[2].enable()
else:
self.setupButtons.buttons[2].disable()
if self.noesyPeaks and self.origResonances:
self.midgeButtons.buttons[0].enable()
else:
self.midgeButtons.buttons[0].disable()
if self.distanceConstraintList:
self.midgeButtons.buttons[1].enable()
self.distConstrLabel.set(
'Distance constraints: %d' %
len(self.distanceConstraintList.constraints))
else:
self.distConstrLabel.set('Distance constraints:')
self.midgeButtons.buttons[1].disable()
if self.antiDistConstraintList:
self.antiConstrLabel.set(
'Anti-distance constraints: %d' %
len(self.antiDistConstraintList.constraints))
self.midgeButtons.buttons[2].enable()
else:
self.antiConstrLabel.set('Anti-distance constraints:')
self.midgeButtons.buttons[2].disable()
if (self.distanceConstraintList and self.antiDistConstraintList
and (self.numClouds > 0) and self.filePrefix):
self.mdButtons.buttons[0].enable()
self.mdButtons.buttons[1].enable()
else:
self.mdButtons.buttons[0].disable()
self.mdButtons.buttons[1].disable()
def getNoesys(self):
names = []
spectra = getSpectraByType(self.project, '2dNOESY')
for spectrum in spectra:
for peakList in spectrum.peakLists:
name = '%s:%s:%s' % (spectrum.experiment.name, spectrum.name,
peakList.serial)
names.append(name)
self.peakListDict[name] = peakList
if not self.noesyPeakList:
self.noesyPeakList = peakList
return names
def setNoesy(self, index, name=None):
if not name:
name = self.noesyPulldown.getSelected()
self.noesyPeakList = self.peakListDict[name]
self.setButtonStates()
def getTocsys(self):
names = []
spectra = getSpectraByType(self.project, '3dTOCSY')
for spectrum in spectra:
for peakList in spectrum.peakLists:
name = '%s:%s:%s' % (spectrum.experiment.name, spectrum.name,
peakList.serial)
names.append(name)
self.peakListDict[name] = peakList
if not self.tocsyPeakList:
self.tocsyPeakList = peakList
return names
def getNoesy3ds(self):
names = []
spectra = getSpectraByType(self.project, '3dNOESY')
for spectrum in spectra:
for peakList in spectrum.peakLists:
name = '%s:%s:%s' % (spectrum.experiment.name, spectrum.name,
peakList.serial)
names.append(name)
self.peakListDict[name] = peakList
if not self.noesy3dPeakList:
self.noesy3dPeakList = peakList
return names
def setTocsy(self, index, name=None):
if not name:
name = self.tocsyPulldown.getSelected()
self.tocsyPeakList = self.peakListDict[name]
self.setButtonStates()
def setNoesy3d(self, index, name=None):
if not name:
name = self.noesy3dPulldown.getSelected()
self.noesy3dPeakList = self.peakListDict[name]
self.setButtonStates()
def getHsqcs(self):
names = []
spectra = getSpectraByType(self.project, 'HSQC')
for spectrum in spectra:
for peakList in spectrum.peakLists:
name = '%s:%s:%s' % (spectrum.experiment.name, spectrum.name,
peakList.serial)
names.append(name)
self.peakListDict[name] = peakList
if not self.hsqcPeakList:
self.hsqcPeakList = peakList
return names
def setHsqc(self, index, name=None):
if not name:
name = self.hsqcPulldown.getSelected()
self.hsqcPeakList = self.peakListDict[name]
self.setButtonStates()
def getMdInitTemp(self, coolingStep):
self.mdInitTempEntry.set(coolingStep[1])
def getMdFinTemp(self, coolingStep):
self.mdFinTempEntry.set(coolingStep[2])
def getMdCoolSteps(self, coolingStep):
self.mdCoolStepsEntry.set(coolingStep[3])
def getMdSimSteps(self, coolingStep):
self.mdSimStepsEntry.set(coolingStep[4])
def getMdTau(self, coolingStep):
self.mdTauEntry.set(coolingStep[5])
def getMdRepScale(self, coolingStep):
self.mdRepScaleEntry.set(coolingStep[6])
def setMdInitTemp(self, event):
value = self.mdInitTempEntry.get()
if value is not None:
self.coolingStep[1] = value
self.updateCoolingScheme()
def setMdFinTemp(self, event):
value = self.mdFinTempEntry.get()
if value is not None:
self.coolingStep[2] = value
self.updateCoolingScheme()
def setMdCoolSteps(self, event):
value = self.mdCoolStepsEntry.get()
if value is not None:
self.coolingStep[3] = value
self.updateCoolingScheme()
def setMdSimSteps(self, event):
value = self.mdSimStepsEntry.get()
if value is not None:
self.coolingStep[4] = value
self.updateCoolingScheme()
def setMdTau(self, event):
value = self.mdTauEntry.get()
if value is not None:
self.coolingStep[5] = value
self.updateCoolingScheme()
def setMdRepScale(self, event):
value = self.mdRepScaleEntry.get()
if value is not None:
self.coolingStep[6] = value
self.updateCoolingScheme()
def selectCoolingStep(self, object, row, col):
self.coolingStep = object
def moveStepEarlier(self):
if self.coolingStep:
i = self.coolingStep[0] - 1
if i > 0:
coolingStep = self.coolingScheme[i - 1]
coolingStep[0] = i + 1
self.coolingStep[0] = i
self.coolingScheme[i - 1] = self.coolingStep
self.coolingScheme[i] = coolingStep
self.updateCoolingScheme()
self.coolingSchemeMatrix.hilightObject(self.coolingStep)
def moveStepLater(self):
if self.coolingStep:
i = self.coolingStep[0] - 1
if i < len(self.coolingScheme) - 1:
coolingStep = self.coolingScheme[i + 1]
coolingStep[0] = i + 1
self.coolingStep[0] = i + 2
self.coolingScheme[i + 1] = self.coolingStep
self.coolingScheme[i] = coolingStep
self.updateCoolingScheme()
self.coolingSchemeMatrix.hilightObject(self.coolingStep)
def addCoolingStep(self):
i = len(self.coolingScheme) + 1
datum = [i, 3000, 100, 10, 2500, 0.001, 1]
self.coolingScheme.append(datum)
self.updateCoolingScheme()
def removeCoolingStep(self):
if self.coolingStep:
coolingScheme = []
i = 0
for coolingStep in self.coolingScheme:
if coolingStep is not self.coolingStep:
i += 1
coolingStep[0] = i
coolingScheme.append(coolingStep)
self.coolingScheme = coolingScheme
self.updateCoolingScheme()
def updateCoolingScheme(self):
objectList = self.coolingScheme
textMatrix = self.coolingScheme
self.coolingSchemeMatrix.update(objectList=objectList,
textMatrix=textMatrix)
def updateMidgeParams(self):
data = [
self.specFreq, self.maxIter, self.mixTime, self.corrTime,
self.leakRate, self.maxIntens
]
self.midgeParamsMatrix.update(textMatrix=[
data,
])
def getSpecFreq(self, obj):
self.specFreqEntry.set(self.specFreq)
def getMaxIter(self, obj):
self.maxIterEntry.set(self.maxIter)
def getMixTime(self, obj):
self.mixTimeEntry.set(self.mixTime)
def getCorrTime(self, obj):
self.corrTimeEntry.set(self.corrTime)
def getLeakRate(self, obj):
self.leakRateEntry.set(self.leakRate)
def getMaxIntens(self, obj):
self.maxIntensEntry.set(self.maxIntens)
def setSpecFreq(self, event):
value = self.specFreqEntry.get()
if value is not None:
self.specFreq = value
self.updateMidgeParams()
def setMaxIter(self, event):
value = self.maxIterEntry.get()
if value is not None:
self.maxIter = value
self.updateMidgeParams()
def setMixTime(self, event):
value = self.mixTimeEntry.get()
if value is not None:
self.mixTime = value
self.updateMidgeParams()
def setCorrTime(self, event):
value = self.corrTimeEntry.get()
if value is not None:
self.corrTime = value
self.updateMidgeParams()
def setLeakRate(self, event):
value = self.leakRateEntry.get()
if value is not None:
self.leakRate = value
self.updateMidgeParams()
def setMaxIntens(self, event):
value = self.maxIntensEntry.get()
if value is not None:
self.maxIntens = value
self.updateMidgeParams()
def destroy(self):
BasePopup.destroy(self)
class FilterCloudsPopup(BasePopup):
def __init__(self, parent, *args, **kw):
self.guiParent = parent
self.structure = None
self.name = None
self.clouds = []
self.rmsds = []
self.names = []
self.atomTypes = None
self.waiting = 0
BasePopup.__init__(self, parent=parent, title="Filter Clouds", **kw)
def body(self, guiFrame):
row = 0
guiFrame.grid_columnconfigure(3, weight=1)
label = Label(guiFrame, text='Cloud file names:')
label.grid(row=row, column=0, sticky=Tkinter.W)
self.fileNameEntry = Entry(guiFrame,
text='testHistone\d+.pdb',
returnCallback=self.loadClouds)
self.fileNameEntry.grid(row=row, column=1, sticky=Tkinter.W)
strucLabel = Label(guiFrame, text='Comparison structure')
strucLabel.grid(row=row, column=2, sticky=Tkinter.W)
self.strucPulldown = PulldownMenu(guiFrame,
entries=self.getStructures(),
callback=self.setStructure,
selected_index=0,
do_initial_callback=0)
self.strucPulldown.grid(row=row, column=3, sticky=Tkinter.W)
row += 1
sdTolLabel = Label(guiFrame, text='Tolerance (SDs):')
sdTolLabel.grid(row=row, column=0, sticky=Tkinter.W)
self.sdToleranceEntry = FloatEntry(guiFrame, text=2.0, width=6)
self.sdToleranceEntry.grid(row=row, column=1, stick=Tkinter.W)
atomTypes = ['All', 'H', 'H HA', 'H HA HB']
label = Label(guiFrame, text='RMSD Atom Types:')
label.grid(row=row, column=2, sticky=Tkinter.W)
self.atomsPulldown = PulldownMenu(guiFrame,
entries=atomTypes,
callback=self.setAtomTypes,
selected_index=0,
do_initial_callback=0)
self.atomsPulldown.grid(row=row, column=3, sticky=Tkinter.W)
row += 1
guiFrame.grid_rowconfigure(row, weight=1)
colHeadings = ['#', 'File name', 'RMSD to mean']
self.scrolledMatrix = ScrolledMatrix(guiFrame,
initialRows=10,
headingList=colHeadings,
callback=self.selectCell,
objectList=[],
textMatrix=[
[],
],
multiSelect=1)
self.scrolledMatrix.grid(row=row,
column=0,
columnspan=4,
sticky=Tkinter.NSEW)
row += 1
texts = [
'Load\nClouds', 'Align\nClouds', 'Calc\nRMSD',
'Make Cloud\nfrom structure', 'Remove', 'Remove\nbad'
]
commands = [
self.loadClouds, self.alignClouds, self.calcRmsd,
self.makeStrucCloud, self.deleteClouds, self.filterClouds
]
self.bottomButtons = createDismissHelpButtonList(
guiFrame,
texts=texts,
expands=1,
commands=commands,
help_url=self.help_url)
self.bottomButtons.grid(row=row,
column=0,
columnspan=4,
sticky=Tkinter.NSEW)
self.update()
def alignClouds(self):
pattern = self.fileNameEntry.get()
self.names = getFileNamesFromPattern(pattern, '.')
self.clouds = getCloudsFromFile(self.names, self.guiParent.project)
alignClouds(self.clouds, self.names)
def loadClouds(self):
pattern = self.fileNameEntry.get()
self.names = getFileNamesFromPattern(pattern, '.')
self.clouds = getCloudsFromFile(self.names, self.guiParent.project)
self.name = None
self.rmsds = [None for x in range(len(self.clouds))]
self.updateAfter()
def getStructures(self):
names = [
'<None>',
]
for molSystem in self.project.sortedMolSystems():
for structure in molSystem.sortedStructureEnsembles():
names.append('%s:%d' % (molSystem.name, structure.ensembleId))
return names
def setStructure(self, index, name=None):
if index < 1:
self.structure = None
else:
structures = []
for molSystem in self.project.molSystems:
for structure in molSystem.structures:
structures.append(structure)
self.structure = structures[index - 1]
self.updateButtons()
def setAtomTypes(self, index, name=None):
self.atomTypes = atomTypeList[index]
def filterClouds(self):
if self.clouds:
sdTolerance = self.sdToleranceEntry.get() or 2.0
keptClouds = []
meanRmsd = 0.0
N = 0
for rmsd in self.rmsds:
meanRmsd += rmsd or 0.0
N += 1
if N > 0:
meanRmsd /= float(N)
sd = 0.0
for r in self.rmsds:
rmsd = r or 0.0
sd += (rmsd - meanRmsd) * (rmsd - meanRmsd)
if N > 0:
sd /= float(N - 1)
sd = sqrt(sd)
print meanRmsd, '+/-', sd
for i in range(len(self.clouds), 0, -1):
rmsd = self.rmsds[i]
if abs(rmsd - meanRmsd) > (sdTolerance * sd):
self.rmsds.pop(i)
self.names.pop(i)
self.clouds.pop(i)
#print 'Cloud %s is bad' % (cloud)
self.updateAfter()
def makeStrucCloud(self):
if self.structure and self.clouds:
pdbFileName = 'CloudForStructure.pdb'
atomCoordList = []
atomCoordList0 = []
resDict = {}
hmass = 25
resonances = self.clouds[0].keys()
resonances2 = []
C = 0
for resonance in resonances:
if resonance == 'rmsd':
continue
resonanceSet = resonance.resonanceSet
if resonanceSet:
i = list(resonanceSet.resonances).index(resonance)
atomSet = list(resonance.resonanceSet.atomSets)[i]
coords = getAtomSetCoords(atomSet, self.structure)
coord = coords[0]
atomCoordList.append([coord.x, coord.y, coord.z])
atomCoordList0.append([coord.x, coord.y, coord.z])
resonances2.append(resonance)
C += 1
print len(atomCoordList)
print len(resonances), len(resonances2)
print "Generating Mean"
cloudsList = []
for cloud in self.clouds:
orderCloud = []
for resonance in resonances2:
x, y, z = cloud.get(resonance) or (0.0, 0.0, 0.0)
orderCloud.append([-x, -y, -z])
cloudsList.append(orderCloud)
(meanCloud, cloudsList) = alignToMeanCloud(cloudsList)
weights = [1.0 for x in atomCoordList]
centerCoords(atomCoordList)
print "init cen", getMeanCoords(atomCoordList)
print "mean cen", getMeanCoords(meanCloud)
print "Print aligning struct clouds to mean", len(meanCloud), len(
atomCoordList), len(weights)
atomCoordsList, error, rotMat = alignCoordinates(
meanCloud, atomCoordList, weights)
print " Rotation", rotMat
writeTypedPdbCloud(atomCoordList, pdbFileName, resonances2)
print "Getting centres"
oldCentre = getMeanCoords(atomCoordList0)
newCentre = getMeanCoords(atomCoordList)
delta = [
newCentre[i] - oldCentre[i] for i in range(len(oldCentre))
]
print " New centre", newCentre
print " Old centre", oldCentre
print " Delta", delta
#inverseRot = inverseMatrix(rotMat)
model = self.structure.findFirstModel()
coordinates = model.coordinates
offset = 0
iis = (0, 1, 2)
for atom in self.structure.orderedAtoms:
next = offset + 3
coords = [coordinates[offset + ii] + delta[ii] for ii in iis]
coords = matrixVecMultiply(rotMat, coords)
coordinates[offset:next] = coords
offset = next
model.setSubmatrixData('coordinates', coordinates)
clouds = getCloudsFromFile([
pdbFileName,
], self.structure.root)
self.clouds.append(clouds[0])
self.rmsds.append(None)
self.names.append(pdbFileName)
self.updateAfter()
def calcRmsd(self):
if self.clouds:
if len(self.scrolledMatrix.currentObjects) < 2:
clouds = self.clouds
else:
clouds = []
for name in self.scrolledMatrix.currentObjects:
clouds.append(self.clouds[self.names.index(name)])
self.rmsds = filterClouds(clouds, atomTypes=self.atomTypes)
self.updateAfter()
def deleteClouds(self):
if self.names and self.name and showOkCancel(
'Confirm', 'Really remove selected clouds?'):
indices = []
for name in self.scrolledMatrix.currentObjects:
i = self.names.index(name)
indices.append(i)
indices.sort()
indices.reverse()
for i in indices:
self.clouds.pop(i)
self.rmsds.pop(i)
self.names.pop(i)
self.name = None
self.updateAfter()
def selectCell(self, name, row, col):
self.name = name
self.updateButtons()
def updateAfter(self, *opt):
if self.waiting:
return
else:
self.waiting = 1
self.after_idle(self.update)
def destroy(self):
BasePopup.destroy(self)
def updateButtons(self):
if self.names:
self.bottomButtons.buttons[1].enable()
self.bottomButtons.buttons[2].enable()
self.bottomButtons.buttons[5].enable()
else:
self.bottomButtons.buttons[1].disable()
self.bottomButtons.buttons[2].enable()
self.bottomButtons.buttons[5].disable()
if self.name:
self.bottomButtons.buttons[4].enable()
else:
self.bottomButtons.buttons[4].disable()
if self.structure and self.clouds:
self.bottomButtons.buttons[3].enable()
else:
self.bottomButtons.buttons[3].disable()
def update(self):
textMatrix = []
objectList = self.names
self.updateButtons()
i = 0
for name in objectList:
datum = []
datum.append(i + 1)
datum.append(name)
datum.append(self.rmsds[i])
textMatrix.append(datum)
i += 1
if not objectList:
textMatrix = [
[],
]
self.scrolledMatrix.update(objectList=objectList,
textMatrix=textMatrix)
self.waiting = 0
class MeccanoPopup(BasePopup):
def __init__(self, parent, project, *args, **kw):
self.alignMedium = None
self.chain = None
self.constraint = None
self.constraintSet = None
self.molSystem = None
self.project = project
self.run = None
self.shiftList = None
self.tensor = None
BasePopup.__init__(self, parent=parent, title='MECCANO', *args, **kw)
self.curateNotifiers(self.registerNotify)
def body(self, guiFrame):
guiFrame.grid_columnconfigure(0, weight=1)
guiFrame.grid_rowconfigure(0, weight=1)
options = ['Parameters','Restraints','Alignment Media & Tensors','About Meccano']
tabbedFrame = TabbedFrame(guiFrame, options=options)
tabbedFrame.grid(row=0, column=0, sticky='nsew')
frameA, frameB, frameC, frameD = tabbedFrame.frames
frameA.grid_columnconfigure(1, weight=1)
frameA.grid_rowconfigure(13, weight=1)
frameB.grid_columnconfigure(1, weight=1)
frameB.grid_rowconfigure(1, weight=1)
frameC.grid_columnconfigure(0, weight=1)
frameC.grid_rowconfigure(1, weight=1)
frameD.grid_columnconfigure(0, weight=1)
frameD.grid_rowconfigure(0, weight=1)
texts = ['Run MECCANO!']
commands = [self.runMeccano]
bottomButtons = createDismissHelpButtonList(guiFrame, texts=texts,
commands=commands, expands=True)
bottomButtons.grid(row=1, column=0, sticky='ew')
if not Meccano:
bottomButtons.buttons[0].disable()
# Parameters
row = 0
label = Label(frameA, text='Calculation Run:')
label.grid(row=row,column=0,sticky='w')
self.runPulldown = PulldownList(frameA, callback=self.selectRun)
self.runPulldown.grid(row=row,column=1,sticky='w')
row += 1
label = Label(frameA, text='Shift List (for CO):')
label.grid(row=row,column=0,sticky='w')
self.shiftListPulldown = PulldownList(frameA, callback=self.selectShiftList)
self.shiftListPulldown.grid(row=row,column=1,sticky='w')
row += 1
label = Label(frameA, text='Keep Copy of Used Shifts:')
label.grid(row=row,column=0,sticky='w')
self.toggleCopyShifts = CheckButton(frameA)
self.toggleCopyShifts.grid(row=row,column=1,sticky='w')
self.toggleCopyShifts.set(True)
row += 1
label = Label(frameA, text='Molecular System:')
label.grid(row=row,column=0,sticky='w')
self.molSystemPulldown = PulldownList(frameA, callback=self.selectMolSystem)
self.molSystemPulldown.grid(row=row,column=1,sticky='w')
row += 1
label = Label(frameA, text='Chain:')
label.grid(row=row,column=0,sticky='w')
self.chainPulldown = PulldownList(frameA, callback=self.selectChain)
self.chainPulldown.grid(row=row,column=1,sticky='w')
self.chainPulldown.bind('<Leave>', self.updateRunParams)
row += 1
label = Label(frameA, text='First Peptide Plane:')
label.grid(row=row,column=0,sticky='w')
self.firstResEntry = IntEntry(frameA, text=None, width=8)
self.firstResEntry.grid(row=row,column=1,sticky='w')
self.firstResEntry.bind('<Leave>', self.updateRunParams)
row += 1
label = Label(frameA, text='Last Peptide Plane:')
label.grid(row=row,column=0,sticky='w')
self.lastResEntry = IntEntry(frameA, text=None, width=8)
self.lastResEntry.grid(row=row,column=1,sticky='w')
self.lastResEntry.bind('<Leave>', self.updateRunParams)
row += 1
label = Label(frameA, text='Max Num Optimisation Steps:')
label.grid(row=row,column=0,sticky='w')
self.maxOptStepEntry = IntEntry(frameA, text=500, width=8)
self.maxOptStepEntry.grid(row=row,column=1,sticky='w')
self.maxOptStepEntry.bind('<Leave>', self.updateRunParams)
row += 1
label = Label(frameA, text='Num Optimisation Peptide Planes:')
label.grid(row=row,column=0,sticky='w')
self.numOptPlaneEntry = IntEntry(frameA, text=2, width=8)
self.numOptPlaneEntry.grid(row=row,column=1,sticky='w')
self.numOptPlaneEntry.bind('<Leave>', self.updateRunParams)
row += 1
label = Label(frameA, text='Min Num Optimisation Hits:')
label.grid(row=row,column=0,sticky='w')
self.numOptHitsEntry = IntEntry(frameA, text=5, width=8)
self.numOptHitsEntry.grid(row=row,column=1,sticky='w')
self.numOptHitsEntry.bind('<Leave>', self.updateRunParams)
row += 1
label = Label(frameA, text='File Name Prefix:')
label.grid(row=row,column=0,sticky='w')
self.pdbFileEntry = Entry(frameA, text='Meccano', width=8)
self.pdbFileEntry.grid(row=row,column=1,sticky='w')
self.pdbFileEntry.bind('<Leave>', self.updateRunParams)
row += 1
label = Label(frameA, text='Write Output File (.out):')
label.grid(row=row,column=0,sticky='w')
self.toggleWriteOutFile = CheckButton(frameA)
self.toggleWriteOutFile.grid(row=row,column=1,sticky='w')
self.toggleWriteOutFile.set(False)
self.toggleWriteOutFile.bind('<Leave>', self.updateRunParams)
row += 1
label = Label(frameA, text='Write PDB File (.pdb):')
label.grid(row=row,column=0,sticky='w')
self.toggleWritePdbFile = CheckButton(frameA)
self.toggleWritePdbFile.grid(row=row,column=1,sticky='w')
self.toggleWritePdbFile.set(True)
self.toggleWritePdbFile.bind('<Leave>', self.updateRunParams)
if not Meccano:
row += 1
label = Label(frameA, text='The Meccano executable is not available (it needs to be compiled)', fg='red')
label.grid(row=row,column=0,columnspan=2,sticky='w')
# Restraints
label = Label(frameB, text='Constraint Set:')
label.grid(row=0,column=0,sticky='w')
self.constraintSetPulldown = PulldownList(frameB, callback=self.selectConstraintSet)
self.constraintSetPulldown.grid(row=0,column=1,sticky='w')
self.alignMediumPulldown= PulldownList(self, callback=self.setAlignMedium)
headingList = ['#','List Type','Use?','Alignment\nMedium','Num\nRestraints']
editWidgets = [None,None,None,self.alignMediumPulldown,None]
editGetCallbacks = [None,None,self.toggleUseRestraints,self.getAlignMedium,None]
editSetCallbacks = [None,None,None,self.setAlignMedium,None]
self.restraintMatrix = ScrolledMatrix(frameB,
headingList=headingList,
editSetCallbacks=editSetCallbacks,
editGetCallbacks=editGetCallbacks,
editWidgets=editWidgets,
callback=None,
multiSelect=True)
self.restraintMatrix.grid(row=1,column=0,columnspan=2,sticky='nsew')
# Alignment Media
div = LabelDivider(frameC,text='Alignment Media')
div.grid(row=0,column=0,sticky='ew')
self.mediumNameEntry = Entry(self, returnCallback=self.setMediumName)
self.mediumDetailsEntry = Entry(self, returnCallback=self.setMediumDetails)
headingList = ['#','Name','Details','Static Tensor','Dynamic Tensor']
editWidgets = [None, self.mediumNameEntry, self.mediumDetailsEntry, None, None]
editGetCallbacks = [None, self.getMediumName, self.getMediumDetails, None, None]
editSetCallbacks = [None, self.setMediumName, self.setMediumDetails, None, None]
self.mediaMatrix = ScrolledMatrix(frameC,
headingList=headingList,
editSetCallbacks=editSetCallbacks,
editGetCallbacks=editGetCallbacks,
editWidgets=editWidgets,
callback=self.selectAlignMedium,
multiSelect=True)
self.mediaMatrix.grid(row=1,column=0,sticky='nsew')
texts = ['Add Alignment medium','Remove Alignment Medium']
commands = [self.addAlignMedium,self.removeAlignMedium]
buttonList = ButtonList(frameC, texts=texts, commands=commands, expands=True)
buttonList.grid(row=2,column=0,sticky='nsew')
self.editAxialEntry = FloatEntry(self, returnCallback=self.setAxial)
self.editRhombicEntry = FloatEntry(self, returnCallback=self.setRhombic)
self.editAlphaEulerEntry = FloatEntry(self, returnCallback=self.setEulerAlpha)
self.editBetaEulerEntry = FloatEntry(self, returnCallback=self.setEulerBeta)
self.editGammaEulerEntry = FloatEntry(self, returnCallback=self.setEulerGamma)
div = LabelDivider(frameC,text='Alignment Tensors')
div.grid(row=3,column=0,sticky='ew')
headingList = ['Type', u'Axial (\u03B6)',u'Rhombic (\u03B7)',
u'Euler \u03B1',u'Euler \u03B2',u'Euler \u03B3']
editWidgets = [None,self.editAxialEntry,
self.editRhombicEntry,self.editAlphaEulerEntry,
self.editBetaEulerEntry,self.editGammaEulerEntry]
editSetCallbacks = [None,self.setAxial,self.setRhombic,
self.setEulerAlpha,self.setEulerBeta,self.setEulerGamma]
editGetCallbacks = [None,self.getAxial,self.getRhombic,
self.getEulerAlpha,self.getEulerBeta,self.getEulerGamma]
self.tensorMatrix = ScrolledMatrix(frameC, maxRows=2,
headingList=headingList,
editSetCallbacks=editSetCallbacks,
editGetCallbacks=editGetCallbacks,
editWidgets=editWidgets,
callback=self.selectTensor,
multiSelect=True)
self.tensorMatrix.grid(row=4,column=0,sticky='nsew')
texts = ['Add Static Tensor','Add Dynamic Tensor','Remove Tensor']
commands = [self.addStaticTensor,self.addDynamicTensor,self.removeTensor]
buttonList = ButtonList(frameC,texts=texts, commands=commands, expands=True)
buttonList.grid(row=5,column=0,sticky='ew')
# About
label = Label(frameD, text='About Meccano...')
label.grid(row=0,column=0,sticky='w')
#
self.geometry('500x400')
self.updateShiftLists()
self.updateMolSystems()
self.updateResidueRanges()
self.updateConstraintSets()
self.updateAlignMedia()
self.updateRuns()
def close(self):
self.updateRunParams()
BasePopup.close(self)
def destroy(self):
self.updateRunParams()
self.curateNotifiers(self.unregisterNotify)
BasePopup.destroy(self)
def curateNotifiers(self, notifyFunc):
for func in ('__init__', 'delete'):
notifyFunc(self.updateConstraintSetsAfter, 'ccp.nmr.NmrConstraint.NmrConstraintStore', func)
for func in ('__init__', 'delete','setName','setConditionState'):
for clazz in ('ccp.nmr.NmrConstraint.CsaConstraintList',
'ccp.nmr.NmrConstraint.DihedralConstraintList',
'ccp.nmr.NmrConstraint.DistanceConstraintList',
'ccp.nmr.NmrConstraint.HBondConstraintList',
'ccp.nmr.NmrConstraint.JCouplingConstraintList',
'ccp.nmr.NmrConstraint.RdcConstraintList'):
notifyFunc(self.updateConstraintListsAfter, clazz, func)
for func in ('__init__', 'delete',):
for clazz in ('ccp.nmr.NmrConstraint.CsaConstraint',
'ccp.nmr.NmrConstraint.DihedralConstraint',
'ccp.nmr.NmrConstraint.DistanceConstraint',
'ccp.nmr.NmrConstraint.HBondConstraint',
'ccp.nmr.NmrConstraint.JCouplingConstraint',
'ccp.nmr.NmrConstraint.RdcConstraint'):
notifyFunc(self.updateConstraintsAfter, clazz, func)
for func in ('__init__', 'delete'):
notifyFunc(self.updateShiftListsAfter,'ccp.nmr.Nmr.ShiftList', func)
for func in ('__init__', 'delete'):
notifyFunc(self.updateMolSystemsAfter,'ccp.molecule.MolSystem.MolSystem', func)
for func in ('__init__', 'delete'):
notifyFunc(self.updateChainsAfter,'ccp.molecule.MolSystem.Chain', func)
for func in ('__init__', 'delete','setDynamicAlignment',
'setStaticAlignment','setName','setDetails'):
notifyFunc(self.updateAlignMediaAfter,'ccp.nmr.NmrConstraint.ConditionState', func)
def updateAlignMediaAfter(self, alignMedium):
if alignMedium.nmrConstraintStore is self.constraintSet:
self.after_idle(self.updateAlignMedia)
if alignMedium is self.alignMedium:
self.after_idle(self.updateTensors)
def updateConstraintSetsAfter(self, constraintSet):
self.after_idle(self.updateConstraintSets)
def updateShiftListsAfter(self, shiftList):
self.after_idle(self.updateShiftLists)
def updateMolSystemsAfter(self, molSystem):
self.after_idle(self.updateMolSystems)
def updateChainsAfter(self, chain):
self.after_idle(self.updateChains)
def updateConstraintsAfter(self, constraint):
if constraint.parent.parent is self.constraintSet:
self.after_idle(self.updateConstraintLists)
def updateConstraintListsAfter(self, constraintList):
if constraintList.parent is self.constraintSet:
self.after_idle(self.updateConstraintLists)
def runMeccano(self):
#
#
# Input checks first
#
#
warning = ''
if not self.molSystem:
warning += 'No molecular system selected\n'
if not self.chain:
warning += 'No chain selected\n'
if not self.constraintSet:
warning += 'No selected constraint set\n'
else:
constraintLists = [cl for cl in self.constraintSet.constraintLists if cl.useForMeccano]
if not constraintLists:
warning += 'No constraint lists selected for use\n'
first, last = self.updateResidueRanges()
if (last-first) < 2:
warning += 'Too few peptide planes selected\n'
if warning:
showWarning('Cannot run MECCANO','Encountered the following problems:\n' + warning)
return
if not self.run:
self.run = self.makeSimRun()
self.updateRunParams()
if self.toggleCopyShifts.get() and self.shiftList:
shiftList = self.run.findFirstOutputMeasurementList(className='ShiftList')
if not shiftList:
shiftList = self.project.currentNmrProject.newShiftList(name='Meccano Input')
self.run.setOutputMeasurementLists([shiftList,])
shiftDict = {}
for shift in shiftList.shifts:
shiftDict[shift.resonance] = shift
for shift in self.shiftList.shifts:
resonance = shift.resonance
resonanceSet = resonance.resonanceSet
if resonanceSet:
atom = resonanceSet.findFirstAtomSet().findFirstAtom()
if (atom.name == 'C') and (atom.residue.molResidue.molType == 'protein'):
shift2 = shiftDict.get(resonance)
if shift2:
shift2.value = shift.value
shift2.error = shift.error
else:
shiftList.newShift(resonance=resonance, value=shift.value, error=shift.error)
#
#
# Accumulate data from CCPN data model & GUI
#
#
# Sequence
residues = self.chain.sortedResidues()
residueDict = {}
seqData = []
for residue in residues:
molResidue = residue.molResidue
code1Letter = molResidue.chemComp.code1Letter
if not code1Letter:
msg = 'Encountered non-standard residue type: %s'
showWarning('Cannot run MECCANO', msg % residue.ccpCode)
return
seqCode = residue.seqCode
seqData.append((seqCode, code1Letter))
residueDict[seqCode] = residue.chemCompVar.chemComp.code3Letter
# Media, RDCs & Dihedrals
rdcLists = []
dihedralLists = []
for constraintList in constraintLists:
if constraintList.className == 'RdcConsraintList':
if constraintList.conditionState:
rdcLists.append(constraintList)
elif constraintList.className == 'DihedralConstraintList':
dihedralLists.append(dihedralLists)
f = PI_OVER_180
mediaData = []
for constraintList in rdcLists:
medium = constraintList.conditionState
dynamicTensor = medium.dynamicAlignment
staticTensor = medium.staticAlignment
if not (dynamicTensor or staticTensor):
continue
if dynamicTensor:
dynamicTensorData = ['Dynamic', dynamicTensor.aAxial, dynamicTensor.aRhombic,
f*dynamicTensor.alpha, f*dynamicTensor.beta, f*dynamicTensor.gamma]
if staticTensor:
staticTensorData = ['Static', staticTensor.aAxial, staticTensor.aRhombic,
f*staticTensor.alpha, f*staticTensor.beta, f*staticTensor.gamma]
if not dynamicTensor:
dynamicTensorData = staticTensorData
elif not staticTensor:
staticTensorData = dynamicTensorData
rdcData = []
for restraint in constraintList.constraints:
items = list(restraint.items)
if len(items) != 1:
continue
resonanceA, resonanceB = [fr.resonance for fr in items[0].resonances]
resonanceSetA = resonanceA.resonanceSet
if not resonanceSetA:
continue
resonanceSetB = resonanceB.resonanceSet
if not resonanceSetB:
continue
resNameA = getResonanceName(resonanceA)
resNameB = getResonanceName(resonanceB)
residueA = resonanceSetA.findFirstAtomSet().findFirstAtom().residue
residueB = resonanceSetB.findFirstAtomSet().findFirstAtom().residue
seqA = residueA.seqCode
seqB = residueB.seqCode
value = restraint.targetValue
error = restraint.error
if error is None:
key = [resNameA,resNameB]
key.sort()
sigma = DEFAULT_ERRORS.get(tuple(key), 1.0)
rdcData.append([seqA, resNameA, seqB, resNameB, value, error])
mediaData.append((dynamicTensorData,staticTensorData,rdcData))
oneTurn = 360.0
dihedralDict = {}
for constraintList in dihedralLists:
for restraint in constraintList.constraints:
items = list(restraint.items)
if len(items) != 1:
continue
item = items[0]
resonances = [fr.resonance for fr in item.resonances]
resonanceSets = [r.resonanceSet for r in resonances]
if None in resonanceSets:
continue
atoms = [rs.findFirstAtomSet().findFirstAtom() for rs in resonanceSets]
atomNames = [a.name for a in atoms]
if atomNames == PHI_ATOM_NAMES:
isPhi = True
elif atomNames == PSI_ATOM_NAMES:
isPhi = False
else:
continue
residue = atoms[2].residue
if residue.chain is not self.chain:
continue
if isPhi:
residuePrev = getLinkedResidue(residue, linkCode='prev')
if not residuePrev:
continue
atomC0 = residuePrev.findFirstAtom(name='C')
atomN = residue.findFirstAtom(name='N')
atomCa = residue.findFirstAtom(name='CA')
atomC = residue.findFirstAtom(name='C')
atoms2 = [atomC0, atomN, atomCa, atomC]
else:
residueNext = getLinkedResidue(residue, linkCode='next')
if not residueNext:
continue
atomN = residue.findFirstAtom(name='N')
atomCa = residue.findFirstAtom(name='CA')
atomC = residue.findFirstAtom(name='C')
atomN2 = residueNext.findFirstAtom(name='N')
atoms2 = [atomN, atomCa, atomC, atomN2]
if atoms != atoms2:
continue
value = item.targetValue
error = item.error
if error is None:
upper = item.upperLimit
lower = item.lowerLimit
if (upper is not None) and (lower is not None):
dUpper = angleDifference(value, lower, oneTurn)
dLower = angleDifference(upper, value, oneTurn)
error = max(dUpper, dLower)
elif lower is not None:
error = angleDifference(value, lower, oneTurn)
elif upper is not None:
error = angleDifference(upper, value, oneTurn)
else:
error = 30.0 # Arbitrary, but sensible for TALOS, DANGLE
seqCode = residue.seqCode
if not dihedralDict.has_key(seqCode):
dihedralDict[seqCode] = [None, None, None, None] # Phi, Psi, PhiErr, PsiErr
if isPhi:
dihedralDict[seqCode][0] = value
dihedralDict[seqCode][2] = error
else:
dihedralDict[seqCode][1] = value
dihedralDict[seqCode][3] = error
phipsiData = []
seqCodes = dihedralDict.keys()
seqCodes.sort()
for seqCode in seqCodes:
data = dihedralDict[seqCode]
if None not in data:
phi, psi, phiErr, psiErr = data
phipsiData.append((seqCode, phi, psi, phiErr, psiErr))
# User options
firstPPlaneFrag = self.firstResEntry.get() or 1
lastPPlaneFrag = self.lastResEntry.get() or 1
ppNbMin = self.numOptPlaneEntry.get() or 1
minValueBest = self.numOptHitsEntry.get() or 2
maxValueBest = self.maxOptStepEntry.get() or 5
strucData = Meccano.runFwd(firstPPlaneFrag, lastPPlaneFrag, ppNbMin,
minValueBest, maxValueBest, RAMACHANDRAN_DATABASE,
seqData, mediaData, phipsiData)
if strucData:
fileName = 'CcpnMeccanoPdb%f.pdb' % time.time()
fileObj = open(fileName, 'w')
ch = self.chain.pdbOneLetterCode.strip()
if not ch:
ch = self.chain.code[0].upper()
i = 1
for atomType, resNb, x, y, z in strucData:
resType = residueDict.get(resNb, '???')
line = PDB_FORMAT % ('ATOM',i,'%-3s' % atomType,'',resType,ch,resNb,'',x,y,z,1.0,1.0)
i += 1
fileObj.close()
ensemble = getStructureFromFile(self.molSystem, fileName)
if not self.toggleWritePdbFile.get():
os.unlink(fileName)
self.run.outputEnsemble = ensemble
self.run = None
self.updateRuns()
def getMediumName(self, alignMedium):
self.mediumNameEntry.set(alignMedium.name)
def getMediumDetails(self, alignMedium):
self.mediumDetailsEntry.set(alignMedium.details)
def setMediumName(self, event):
value = self.mediumNameEntry.get()
self.alignMedium.name = value or None
def setMediumDetails(self, event):
value = self.mediumDetailsEntry.get()
self.alignMedium.details = value or None
def setAlignMedium(self, alignMedium):
if self.constraintSet:
self.constraintSet.conditionState = alignMedium
def getAlignMedium(self, constraintList):
media = self.getAlignmentMedia()
names = [am.name for am in media]
if constraintList.conditionState in media:
index = media.index(constraintList.conditionState)
else:
index = 0
self.alignMediumPulldown.setup(names, media, index)
def toggleUseRestraints(self, constraintList):
bool = constraintList.useForMeccano
bool = not bool
if bool and (not constraintList.conditionState) \
and (constraintList.className == 'RdcConsraintList'):
msg = 'Cannot use RDC restraint list for Meccano '
msg += 'unless it is first associated with an amigment medium'
showWarning('Warning', msg, parent=self)
else:
constraintList.useForMeccano = bool
self.updateConstraintLists()
def addStaticTensor(self):
if self.alignMedium:
tensor = Implementation.SymmTracelessMatrix(aAxial=0.0,aRhombic=0.0,
alpha=0.0,beta=0.0,
gamma=0.0)
self.alignMedium.staticAlignment = tensor
self.updateAlignMediaAfter(self.alignMedium)
def addDynamicTensor(self):
if self.alignMedium:
tensor = Implementation.SymmTracelessMatrix(aAxial=0.0,aRhombic=0.0,
alpha=0.0,beta=0.0,
gamma=0.0)
self.alignMedium.dynamicAlignment = tensor
self.updateAlignMediaAfter(self.alignMedium)
def removeTensor(self):
if self.alignMedium and self.tensor:
if self.tensor is self.alignMedium.dynamicAlignment:
self.alignMedium.dynamicAlignment = None
elif self.tensor is self.alignMedium.staticAlignment:
self.alignMedium.staticAlignment = None
self.updateAlignMediaAfter(self.alignMedium)
def addAlignMedium(self):
if self.constraintSet:
medium = self.constraintSet.newConditionState()
medium.name = 'Align Medium %d' % medium.serial
def removeAlignMedium(self):
if self.alignMedium:
self.alignMedium.delete()
def updateTensor(self, aAxial=None, aRhombic=None, alpha=None, beta=None, gamma=None):
aAxial = aAxial or self.tensor.aAxial
aRhombic = aRhombic or self.tensor.aRhombic
alpha = alpha or self.tensor.alpha
beta = beta or self.tensor.beta
gamma = gamma or self.tensor.gamma
tensor = Implementation.SymmTracelessMatrix(aAxial=aAxial,
aRhombic=aRhombic,
alpha=alpha,beta=beta,
gamma=gamma)
if self.alignMedium:
if self.tensor is self.alignMedium.dynamicAlignment:
self.alignMedium.dynamicAlignment = tensor
elif self.tensor is self.alignMedium.staticAlignment:
self.alignMedium.staticAlignment = tensor
self.tensor = tensor
def setAxial(self, event):
value = self.editAxialEntry.get()
self.updateTensor(aAxial=value)
self.updateTensors()
def setRhombic(self, event):
value = self.editRhombicEntry.get()
self.updateTensor(aRhombic=value)
self.updateTensors()
def setEulerAlpha(self, event):
value = self.editAlphaEulerEntry.get()
self.updateTensor(alpha=value)
self.updateTensors()
def setEulerBeta(self, event):
value = self.editBetaEulerEntry.get()
self.updateTensor(beta=value)
self.updateTensors()
def setEulerGamma(self, event):
value = self.editGammaEulerEntry.get()
self.updateTensor(gamma=value)
self.updateTensors()
def getAxial(self, tensor):
value = tensor.aAxial
self.editAxialEntry.set(value)
def getRhombic(self, tensor):
value = tensor.aRhombic
self.editRhombicEntry.set(value)
def getEulerAlpha(self, tensor):
value = tensor.alpha
self.editAlphaEulerEntry.set(value)
def getEulerBeta(self, tensor):
value = tensor.beta
self.editBetaEulerEntry.set(value)
def getEulerGamma(self, tensor):
value = tensor.gamma
self.editGammaEulerEntry.set(value)
def selectTensor(self, tensor, row, col):
self.tensor = tensor
def selectAlignMedium(self, alignMedium, row, col):
self.alignMedium = alignMedium
self.updateTensors()
def getAlignmentMedia(self):
if self.constraintSet:
return self.constraintSet.sortedConditionStates()
else:
return []
def updateAlignMedia(self):
textMatrix = []
objectList = []
if self.constraintSet:
objectList = self.getAlignmentMedia()
for conditionState in objectList:
staticTensor = None
dyamicTensor = None
tensor = conditionState.dynamicAlignment
if tensor:
vals = (tensor.aAxial, tensor.aRhombic, tensor.alpha, tensor.beta, tensor.gamma)
dyamicTensor = u'\u03B6:%.3f \u03B7:%.3f \u03B1:%.3f \u03B2:%.3f \u03B3:%.3f ' % vals
tensor = conditionState.staticAlignment
if tensor:
vals = (tensor.aAxial, tensor.aRhombic, tensor.alpha, tensor.beta, tensor.gamma)
staticTensor = u'\u03B6:%.3f \u03B7:%.3f \u03B1:%.3f \u03B2:%.3f \u03B3:%.3f ' % vals
datum = [conditionState.serial,
conditionState.name,
conditionState.details,
dyamicTensor,
staticTensor]
textMatrix.append(datum)
if dyamicTensor or staticTensor:
if not self.alignMedium:
self.alignMedium = conditionState
self.mediaMatrix.update(textMatrix=textMatrix, objectList=objectList)
if self.alignMedium:
self.mediaMatrix.selectObject(self.alignMedium)
def updateTensors(self):
textMatrix = []
objectList = []
conditionState = self.alignMedium
if conditionState:
tensor = conditionState.dynamicAlignment
if tensor:
datum = ['Dynamic', tensor.aAxial, tensor.aRhombic,
tensor.alpha, tensor.beta, tensor.gamma]
textMatrix.append(datum)
objectList.append(tensor)
tensor = conditionState.staticAlignment
if tensor:
datum = ['Static', tensor.aAxial, tensor.aRhombic,
tensor.alpha, tensor.beta, tensor.gamma]
textMatrix.append(datum)
objectList.append(tensor)
self.tensorMatrix.update(textMatrix=textMatrix, objectList=objectList)
def getMolSystems(self):
molSystems = []
for molSystem in self.project.sortedMolSystems():
if molSystem.chains:
molSystems.append(molSystem)
return molSystems
def updateMolSystems(self, *notifyObj):
index = 0
names = []
molSystems = self.getMolSystems()
molSystem = self.molSystem
if molSystems:
if molSystem not in molSystems:
molSystem = molSystems[0]
index = molSystems.index(molSystem)
names = [ms.code for ms in molSystems]
else:
self.molSystem = None
if self.molSystem is not molSystem:
if self.run:
self.run.molSystem = molSystem
self.molSystem = molSystem
self.updateChains()
self.molSystemPulldown.setup(texts=names, objects=molSystems, index=index)
def selectMolSystem(self, molSystem):
if self.molSystem is not molSystem:
if self.run:
self.run.molSystem = molSystem
self.molSystem = molSystem
self.updateChains()
def updateChains(self, *notifyObj):
index = 0
names = []
chains = []
chain = self.chain
if self.molSystem:
chains = [c for c in self.molSystem.sortedChains() if c.residues]
if chains:
if chain not in chains:
chain = chains[0]
index = chains.index(chain)
names = [c.code for c in chains]
if chain is not self.chain:
self.chain = chain
self.updateResidueRanges()
self.chainPulldown.setup(texts=names, objects=chains, index=index)
def selectChain(self, chain):
if chain is not self.chain:
self.chain = chain
self.updateResidueRanges()
def updateResidueRanges(self):
first = self.firstResEntry.get()
last = self.lastResEntry.get()
if self.chain:
residues = self.chain.sortedResidues()
firstSeq = residues[0].seqCode
lastSeq = residues[-2].seqCode
if first < firstSeq:
first = firstSeq
if last == first:
last = lastSeq
elif last > lastSeq:
last = lastSeq
if first > last:
last, first = first, last
self.firstResEntry.set(first)
self.lastResEntry.set(last)
return first, last
def getConstraintSets(self):
constraintSets = []
nmrProject = self.project.currentNmrProject
for constraintSet in nmrProject.sortedNmrConstraintStores():
for constraintList in constraintSet.constraintLists:
if constraintList.className not in ('ChemShiftConstraintList','somethingElse'):
constraintSets.append(constraintSet)
break
return constraintSets
def updateConstraintSets(self, *notifyObj):
index = 0
names = []
constraintSets = self.getConstraintSets()
constraintSet = self.constraintSet
if constraintSets:
if constraintSet not in constraintSets:
constraintSet = constraintSets[0]
index = constraintSets.index(constraintSet)
names = ['%d' % cs.serial for cs in constraintSets]
if constraintSet is not self.constraintSet:
if self.run:
self.run.inputConstraintStore = constraintSet
self.constraintSet = constraintSet
self.updateConstraintLists()
self.constraintSetPulldown.setup(texts=names, objects=constraintSets, index=index)
def selectConstraintSet(self, constraintSet):
if self.constraintSet is not constraintSet:
if self.run:
self.run.inputConstraintStore = constraintSet
self.constraintSet = constraintSet
self.updateConstraintLists()
def getConstraintLists(self):
constraintLists = []
if self.constraintSet:
for constraintList in self.constraintSet.sortedConstraintLists():
if constraintList.className not in ('ChemShiftConstraintList','somethingElse'):
constraintLists.append(constraintList)
return constraintLists
def updateConstraintLists(self, *notifyObj):
textMatrix = []
objectList = self.getConstraintLists()
for constraintList in objectList:
if not hasattr(constraintList, 'useForMeccano'):
if constraintList.conditionState \
or (constraintList.className != 'RdcConstraintList'):
bool = True
else:
bool = False
constraintList.useForMeccano = bool
if constraintList.conditionState:
alignMedium = constraintList.conditionState.name
else:
alignMedium = None
datum = [constraintList.serial,
constraintList.className[:-14],
constraintList.useForMeccano and 'Yes' or 'No',
alignMedium,
len(constraintList.constraints)]
textMatrix.append(datum)
self.restraintMatrix.update(textMatrix=textMatrix, objectList=objectList)
def selectConstraint(self, obj, row, column):
if self.constraint is not obj:
self.constraint = obj
def getSimStore(self):
simStore = self.project.findFirstNmrSimStore(name='meccano')
if not simStore:
simStore = self.project.newNmrSimStore(name='meccano')
return simStore
def getRuns(self):
runs = [None, ]
if self.molSystem and self.constraintSet:
simStore = self.getSimStore()
runs += simStore.sortedRuns()
return runs
def updateRuns(self, *notifyObj):
index = 0
names = ['<New>']
runs = self.getRuns()
run = self.run
if runs:
if run not in runs:
run = runs[0]
index = runs.index(run)
names += [r.serial for r in runs if r]
if run is not self.run:
self.updateConstraintSets()
self.updateMolSystems()
self.updateShiftLists()
self.runPulldown.setup(names, runs, index)
def updateRunParams(self, event=None):
if self.run and self.molSystem and self.constraintSet:
simRun = self.run
simRun.inputConstraintStore = self.constraintSet
simRun.molSystem = self.molSystem
if self.shiftList:
simRun.setInputMeasurementLists([self.shiftList,])
simRun.newRunParameter(code='FirstPepPlane',id=1,
intValue=self.firstResEntry.get() or 0)
simRun.newRunParameter(code='LastPepPlane' ,id=1,
intValue=self.lastResEntry.get() or 0)
simRun.newRunParameter(code='MaxOptSteps', id=1,
intValue=self.maxOptStepEntry.get() or 0)
simRun.newRunParameter(code='NumOptPlanes', id=1,
intValue=self.numOptPlaneEntry.get() or 0)
simRun.newRunParameter(code='MinOptHits', id=1,
intValue=self.numOptHitsEntry.get() or 0)
def makeSimRun(self, template=None):
simStore = self.getSimStore()
if template:
molSystem = template.molSystem
constraintSet = template.inputConstraintStore
shiftList = template.findFirstInputMeasurementList(className='ShiftList')
protocol = template.annealProtocol
else:
molSystem = self.molSystem
constraintSet = self.constraintSet
shiftList = self.shiftList
protocol = self.annealProtocol
simRun = simStore.newRun(annealProtocol=protocol,
molSystem=molSystem,
inputConstraintStore=protocol)
if shiftList:
simRun.addInputMeasurementList(shiftList)
if template:
for param in template.runParameters:
simRun.newRunParameter(code=param.code,
id=param.id,
booleanValue=param.booleanValue,
floatValue=param.floatValue,
intValue=param.intValue,
textValue=param.textValue)
else:
if self.chain:
chainCode = self.chain.code
else:
chaincode = 'A'
simRun.newRunParameter(code='FirstPepPlane',id=1,
intValue=self.firstResEntry.get() or 0)
simRun.newRunParameter(code='LastPepPlane' ,id=1,
intValue=self.lastResEntry.get() or 0)
simRun.newRunParameter(code='MaxOptSteps', id=1,
intValue=self.maxOptStepEntry.get() or 0)
simRun.newRunParameter(code='NumOptPlanes', id=1,
intValue=self.numOptPlaneEntry.get() or 0)
simRun.newRunParameter(code='MinOptHits', id=1,
intValue=self.numOptHitsEntry.get() or 0)
simRun.newRunParameter(code='ChainCode', id=1,
textValue=chainCode)
return simRun
def selectRun(self, simRun):
if self.run is not simRun:
if simRun:
if simRun.outputEnsemble:
msg = 'Selected run has already been used to generate a structure.'
msg += 'A new run will be setup based on the selection.'
showWarning('Warning',msg)
simRun = self.makeSimRun(template=simRun)
molSystem = simRun.molSystem
constraintSet = simRun.inputConstraintStore
shiftList = simRun.findFirstInputMeasurementList(className='ShiftList')
if molSystem and (self.molSystem is not molSystem):
self.molSystem = molSystem
self.updateMolSystems()
if constraintSet and (self.constraintSet is not constraintSet):
self.constraintSet = constraintSet
self.updateConstraintSets()
if shiftList and (self.shiftList is not shiftList):
self.shiftList = shiftList
self.updateShiftLists()
firstPepPlane = simRun.findFirstrunParameter(code='FirstPepPlane')
lastPepPlane = simRun.findFirstrunParameter(code='LastPepPlane')
maxOptSteps = simRun.findFirstrunParameter(code='MaxOptSteps')
numOptPlanes = simRun.findFirstrunParameter(code='NumOptPlanes')
minOptHits = simRun.findFirstrunParameter(code='MinOptHits')
chainCode = simRun.findFirstrunParameter(code='ChainCode')
if firstPepPlane is not None:
self.firstResEntry.set(firstPepPlane.intValue or 0)
if lastPepPlane is not None:
self.lastResEntry.set(lastPepPlane.intValue or 0)
if maxOptSteps is not None:
self.maxOptStepEntry.set(maxOptSteps.intValue or 0)
if numOptPlanes is not None:
self.numOptPlaneEntry.set(numOptPlanes.intValue or 0)
if minOptHits is not None:
self.numOptHitsEntry.set(minOptHits.intValue or 0)
if chainCode is not None:
chainCode = chainCode.textValue or 'A'
if self.molSystem:
chain = self.molSystem.findFirsChain(code=chainCode)
if chain:
self.selectChain(chain)
self.run = simRun
def updateShiftLists(self, *notifyObj):
index = 0
names = []
nmrProject = self.project.currentNmrProject
shiftLists = nmrProject.findAllMeasurementLists(className='ShiftList')
shiftLists = [(sl.serial, sl) for sl in shiftLists]
shiftLists.sort()
shiftLists = [x[1] for x in shiftLists]
shiftList = self.shiftList
if shiftLists:
if shiftList not in shiftLists:
shiftList = shiftLists[0]
index = shiftLists.index(shiftList)
names = ['%d'% sl.serial for sl in shiftLists]
if shiftList is not self.shiftList:
if self.run:
self.run.setInputMeasurementLists([shiftList])
self.shiftListPulldown.setup(texts=names, objects=shiftLists, index=index)
def selectShiftList(self, shiftList):
if shiftList is not self.shiftList:
if self.run:
self.run.setInputMeasurementLists([shiftList])
self.shiftList = shiftList
class CingGui(BasePopup):
def __init__(self, parent, options, *args, **kw):
# Fill in below variable once run generates some results
self.haveResults = None
# store the options
self.options = options
BasePopup.__init__(self, parent=parent, title='CING Setup', **kw)
# self.setGeometry(850, 750, 50, 50)
self.project = None
# self.tk_strictMotif( True)
self.updateGui()
# end def __init__
def body(self, guiFrame):
row = 0
col = 0
# frame = Frame( guiFrame )
# frame.grid(row=row, column=col, sticky='news')
self.menuBar = Menu(guiFrame)
self.menuBar.grid(row=row, column=col, sticky='ew')
#----------------------------------------------------------------------------------
# Project frame
#----------------------------------------------------------------------------------
# guiFrame.grid_columnconfigure(row, weight=1)
# frame = LabelFrame(guiFrame, text='Project', font=medFont)
row = +1
col = 0
frame = LabelFrame(guiFrame, text='Project', **labelFrameAttributes)
print '>', frame.keys()
frame.grid(row=row, column=col, sticky='nsew')
frame.grid_columnconfigure(2, weight=1)
# frame.grid_rowconfigure(0, weight=1)
srow = 0
self.projectOptions = [
'old', 'new from PDB', 'new from CCPN', 'new from CYANA'
]
self.projOptionsSelect = RadioButtons(frame,
selected_index=0,
entries=self.projectOptions,
direction='vertical',
select_callback=self.updateGui)
self.projOptionsSelect.grid(row=srow,
column=0,
rowspan=len(self.projectOptions),
columnspan=2,
sticky='w')
if self.options.name:
text = self.options.name
else:
text = ''
# end if
self.projEntry = Entry(frame,
bd=1,
text=text,
returnCallback=self.updateGui)
self.projEntry.grid(row=srow, column=2, columnspan=2, sticky='ew')
# self.projEntry.bind('<Key>', self.updateGui)
self.projEntry.bind('<Leave>', self.updateGui)
projButton = Button(frame,
bd=1,
command=self.chooseOldProjectFile,
text='browse')
projButton.grid(row=srow, column=3, sticky='ew')
srow += 1
self.pdbEntry = Entry(frame, bd=1, text='')
self.pdbEntry.grid(row=srow, column=2, sticky='ew')
self.pdbEntry.bind('<Leave>', self.updateGui)
pdbButton = Button(frame,
bd=1,
command=self.choosePdbFile,
text='browse')
pdbButton.grid(row=srow, column=3, sticky='ew')
srow += 1
self.ccpnEntry = Entry(frame, bd=1, text='')
self.ccpnEntry.grid(row=srow, column=2, sticky='ew')
self.ccpnEntry.bind('<Leave>', self.updateGui)
ccpnButton = Button(frame,
bd=1,
command=self.chooseCcpnFile,
text='browse')
ccpnButton.grid(row=srow, column=3, sticky='ew')
srow += 1
self.cyanaEntry = Entry(frame, bd=1, text='')
self.cyanaEntry.grid(row=srow, column=2, sticky='ew')
self.cyanaEntry.bind('<Leave>', self.updateGui)
cyanaButton = Button(frame,
bd=1,
command=self.chooseCyanaFile,
text='browse')
cyanaButton.grid(row=srow, column=3, sticky='ew')
#Empty row
srow += 1
label = Label(frame, text='')
label.grid(row=srow, column=0, sticky='nw')
srow += 1
label = Label(frame, text='Project name:')
label.grid(row=srow, column=0, sticky='nw')
self.nameEntry = Entry(frame, bd=1, text='')
self.nameEntry.grid(row=srow, column=2, sticky='w')
#Empty row
srow += 1
label = Label(frame, text='')
label.grid(row=srow, column=0, sticky='nw')
srow += 1
self.openProjectButton = Button(frame,
command=self.openProject,
text='Open Project',
**actionButtonAttributes)
self.openProjectButton.grid(row=srow,
column=0,
columnspan=4,
sticky='ew')
#----------------------------------------------------------------------------------
# status
#----------------------------------------------------------------------------------
# guiFrame.grid_columnconfigure(1, weight=0)
srow = 0
frame = LabelFrame(guiFrame, text='Status', **labelFrameAttributes)
frame.grid(row=srow, column=1, sticky='wnes')
self.projectStatus = Text(frame,
height=11,
width=70,
borderwidth=0,
relief='flat')
self.projectStatus.grid(row=0, column=0, sticky='wen')
#Empty row
srow += 1
label = Label(frame, text='')
label.grid(row=srow, column=0, sticky='nw')
srow += 1
self.closeProjectButton = Button(frame,
command=self.closeProject,
text='Close Project',
**actionButtonAttributes)
self.closeProjectButton.grid(row=srow,
column=0,
columnspan=4,
sticky='ew')
#----------------------------------------------------------------------------------
# Validate frame
#----------------------------------------------------------------------------------
row += 1
col = 0
frame = LabelFrame(guiFrame, text='Validate', **labelFrameAttributes)
# frame = LabelFrame(guiFrame, text='Validate', font=medFont)
frame.grid(row=row, column=col, sticky='nsew')
# frame.grid_columnconfigure(2, weight=1)
frame.grid_rowconfigure(0, weight=1)
srow = 0
# label = Label(frame, text='validation')
# label.grid(row=srow,column=0,sticky='nw')
#
# self.selectDoValidation = CheckButton(frame)
# self.selectDoValidation.grid(row=srow, column=1,sticky='nw' )
# self.selectDoValidation.set(True)
#
# srow += 1
# label = Label(frame, text='')
# label.grid(row=srow,column=0,sticky='nw')
#
# srow += 1
label = Label(frame, text='checks')
label.grid(row=srow, column=0, sticky='nw')
self.selectCheckAssign = CheckButton(frame)
self.selectCheckAssign.grid(row=srow, column=1, sticky='nw')
self.selectCheckAssign.set(True)
label = Label(frame, text='assignments and shifts')
label.grid(row=srow, column=2, sticky='nw')
# srow += 1
# self.selectCheckQueen = CheckButton(frame)
# self.selectCheckQueen.grid(row=srow, column=4,sticky='nw' )
# self.selectCheckQueen.set(False)
# label = Label(frame, text='QUEEN')
# label.grid(row=srow,column=5,sticky='nw')
#
# queenButton = Button(frame, bd=1,command=None, text='setup')
# queenButton.grid(row=srow,column=6,sticky='ew')
srow += 1
self.selectCheckResraint = CheckButton(frame)
self.selectCheckResraint.grid(row=srow, column=1, sticky='nw')
self.selectCheckResraint.set(True)
label = Label(frame, text='restraints')
label.grid(row=srow, column=2, sticky='nw')
srow += 1
self.selectCheckStructure = CheckButton(frame)
self.selectCheckStructure.grid(row=srow, column=1, sticky='nw')
self.selectCheckStructure.set(True)
label = Label(frame, text='structural')
label.grid(row=srow, column=2, sticky='nw')
srow += 1
self.selectMakeHtml = CheckButton(frame)
self.selectMakeHtml.grid(row=srow, column=1, sticky='nw')
self.selectMakeHtml.set(True)
label = Label(frame, text='generate HTML')
label.grid(row=srow, column=2, sticky='nw')
srow += 1
self.selectCheckScript = CheckButton(frame)
self.selectCheckScript.grid(row=srow, column=1, sticky='nw')
self.selectCheckScript.set(False)
label = Label(frame, text='user script')
label.grid(row=srow, column=0, sticky='nw')
self.validScriptEntry = Entry(frame, bd=1, text='')
self.validScriptEntry.grid(row=srow,
column=2,
columnspan=3,
sticky='ew')
scriptButton = Button(frame,
bd=1,
command=self.chooseValidScript,
text='browse')
scriptButton.grid(row=srow, column=5, sticky='ew')
srow += 1
label = Label(frame, text='ranges')
label.grid(row=srow, column=0, sticky='nw')
self.rangesEntry = Entry(frame, text='')
self.rangesEntry.grid(row=srow, column=2, columnspan=3, sticky='ew')
# self.validScriptEntry = Entry(frame, bd=1, text='')
# self.validScriptEntry.grid(row=srow,column=3,sticky='ew')
#
# scriptButton = Button(frame, bd=1,command=self.chooseValidScript, text='browse')
# scriptButton.grid(row=srow,column=4,sticky='ew')
srow += 1
texts = ['Run Validation', 'View Results', 'Setup QUEEN']
commands = [self.runCing, None, None]
buttonBar = ButtonList(frame,
texts=texts,
commands=commands,
expands=True)
buttonBar.grid(row=srow, column=0, columnspan=6, sticky='ew')
for button in buttonBar.buttons:
button.config(**actionButtonAttributes)
# end for
self.runButton = buttonBar.buttons[0]
self.viewResultButton = buttonBar.buttons[1]
self.queenButton = buttonBar.buttons[2]
#----------------------------------------------------------------------------------
# Miscellaneous frame
#----------------------------------------------------------------------------------
row += 0
col = 1
# frame = LabelFrame(guiFrame, text='Miscellaneous', font=medFont)
frame = LabelFrame(guiFrame,
text='Miscellaneous',
**labelFrameAttributes)
frame.grid(row=row, column=col, sticky='news')
frame.grid_columnconfigure(2, weight=1)
frame.grid_columnconfigure(4, weight=1, minsize=30)
frame.grid_rowconfigure(0, weight=1)
# Exports
srow = 0
label = Label(frame, text='export to')
label.grid(row=srow, column=0, sticky='nw')
self.selectExportXeasy = CheckButton(frame)
self.selectExportXeasy.grid(row=srow, column=1, sticky='nw')
self.selectExportXeasy.set(True)
label = Label(frame, text='Xeasy, Sparky, TALOS, ...')
label.grid(row=srow, column=2, sticky='nw')
srow += 1
self.selectExportCcpn = CheckButton(frame)
self.selectExportCcpn.grid(row=srow, column=1, sticky='nw')
self.selectExportCcpn.set(True)
label = Label(frame, text='CCPN')
label.grid(row=srow, column=2, sticky='nw')
srow += 1
self.selectExportQueen = CheckButton(frame)
self.selectExportQueen.grid(row=srow, column=1, sticky='nw')
self.selectExportQueen.set(True)
label = Label(frame, text='QUEEN')
label.grid(row=srow, column=2, sticky='nw')
srow += 1
self.selectExportRefine = CheckButton(frame)
self.selectExportRefine.grid(row=srow, column=1, sticky='nw')
self.selectExportRefine.set(True)
label = Label(frame, text='refine')
label.grid(row=srow, column=2, sticky='nw')
srow += 1
label = Label(frame, text='')
label.grid(row=srow, column=0, sticky='nw')
# User script
srow += 1
label = Label(frame, text='user script')
label.grid(row=srow, column=0, sticky='nw')
self.selectMiscScript = CheckButton(frame)
self.selectMiscScript.grid(row=srow, column=1, sticky='nw')
self.selectMiscScript.set(False)
self.miscScriptEntry = Entry(frame, bd=1, text='')
self.miscScriptEntry.grid(row=srow, column=3, sticky='ew')
script2Button = Button(frame,
bd=1,
command=self.chooseMiscScript,
text='browse')
script2Button.grid(row=srow, column=4, sticky='ew')
srow += 1
texts = ['Export', 'Run Script']
commands = [None, None]
buttonBar = ButtonList(frame,
texts=texts,
commands=commands,
expands=True)
buttonBar.grid(row=srow, column=0, columnspan=5, sticky='ew')
for button in buttonBar.buttons:
button.config(**actionButtonAttributes)
# end for
self.exportButton = buttonBar.buttons[0]
self.scriptButton = buttonBar.buttons[1]
#----------------------------------------------------------------------------------
# Textarea
#----------------------------------------------------------------------------------
row += 1
guiFrame.grid_rowconfigure(row, weight=1)
self.outputTextBox = ScrolledText(guiFrame)
self.outputTextBox.grid(row=row, column=0, columnspan=2, sticky='nsew')
self.redirectConsole()
#----------------------------------------------------------------------------------
# Buttons
#----------------------------------------------------------------------------------
row += 1
col = 0
texts = ['Quit', 'Help']
commands = [self.close, None]
self.buttonBar = ButtonList(guiFrame,
texts=texts,
commands=commands,
expands=True)
self.buttonBar.grid(row=row, column=col, columnspan=2, sticky='ew')
# self.openProjectButton = self.buttonBar.buttons[0]
# self.closeProjectButton = self.buttonBar.buttons[1]
# self.runButton = self.buttonBar.buttons[0]
# self.viewResultButton = self.buttonBar.buttons[1]
for button in self.buttonBar.buttons:
button.config(**actionButtonAttributes)
# end for
# end def body
def getGuiOptions(self):
projectName = self.projEntry.get()
index = self.projOptionsSelect.getIndex()
if index > 0:
makeNewProject = True
projectImport = None
if index > 1:
i = index - 2
format = ['PDB', 'CCPN', 'CYANA'][i]
file = [self.pdbEntry, self.ccpnEntry,
self.cyanaEntry][i].get()
if not file:
showWarning('Failure', 'No %s file selected' % format)
return
# end if
projectImport = (format, file)
# end if
else:
# Chould also check that any old project file exists
makeNewProject = False
projectImport = None
# end if
doValidation = self.selectDoValidation.get()
checks = []
if doValidation:
if self.selectCheckAssign.get():
checks.append('assignments')
# end if
if self.selectCheckResraint.get():
checks.append('restraints')
# end if
if self.selectCheckStructure.get():
checks.append('structural')
# end if
if self.selectMakeHtml.get():
checks.append('HTML')
# end if
if self.selectCheckScript.get():
script = self.validScriptEntry.get()
if script:
checks.append(('script', script))
# end if
# end if
if self.selectCheckQueen.get():
checks.append('queen')
# end if
# end if
exports = []
if self.selectExportXeasy.get():
exports.append('Xeasy')
# end if
if self.selectExportCcpn.get():
exports.append('CCPN')
# end if
if self.selectExportQueen.get():
exports.append('QUEEN')
# end if
if self.selectExportRefine.get():
exports.append('refine')
# end if
miscScript = None
if self.selectMiscScript.get():
script = self.miscScriptEntry.get()
if script:
miscScript = script
# end if
# end if
return projectName, makeNewProject, projectImport, doValidation, checks, exports, miscScript
# end def getGuiOptions
def runCing(self):
options = self.getGuiOptions()
if options:
projectName, makeNewProject, projectImport, doValidation, checks, exports, miscScript = options
print 'Project name:', projectName
print 'Make new project?', makeNewProject
print 'Import source:', projectImport
print 'Do vailidation?', doValidation
print 'Validation checks:', ','.join(checks)
print 'Export to:', ','.join(exports)
print 'User script:', miscScript
# end if
# end def runCing
# else there was already an error message
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()
# end def chooseOldProjectFile
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()
# end def choosePdbFile
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()
# end def chooseCcpnFile
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()
# end def chooseCyanaFile
def chooseValidScript(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.validScriptEntry.set(fileName)
popup.destroy()
# end def chooseValidScript
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()
# end def chooseMiscScript
def openProject(self):
projOption = self.projOptionsSelect.get()
if projOption == self.projectOptions[0]:
self.openOldProject()
elif projOption == self.projectOptions[1]:
self.initPdb()
# end if
if self.project:
self.project.gui = self
# end if
self.updateGui()
#end def
def openOldProject(self):
fName = self.projEntry.get()
if not os.path.exists(fName):
nTerror('Error: file "%s" does not exist\n', fName)
#end if
if self.project:
self.closeProject()
# end if
self.project = cing.Project.open(name=fName,
status='old',
verbose=False)
#end def
def initPdb(self):
fName = self.pdbEntry.get()
if not os.path.exists(fName):
nTerror('Error: file "%s" does not exist\n', fName)
#end if
self.project = cing.Project.open(self.nameEntry.get(), status='new')
self.project.initPDB(pdbFile=fName, convention='PDB')
#end def
def closeProject(self):
if self.project:
self.project.close()
# end if
self.project = None
self.updateGui()
#end def
def updateGui(self, event=None):
projOption = self.projOptionsSelect.get()
buttons = self.buttonBar.buttons
# Disable entries
for e in [
self.projEntry, self.pdbEntry, self.ccpnEntry, self.cyanaEntry,
self.nameEntry
]:
e.configure(state='disabled')
#end for
if projOption == self.projectOptions[0]:
# Enable entries
self.projEntry.configure(state='normal')
if (len(self.projEntry.get()) > 0):
self.openProjectButton.enable()
self.runButton.enable()
else:
self.openProjectButton.disable()
self.runButton.disable()
#end if
elif projOption == self.projectOptions[1]:
# Enable entries
self.pdbEntry.configure(state='normal')
self.nameEntry.configure(state='normal')
if (len(self.pdbEntry.get()) > 0
and len(self.nameEntry.get()) > 0):
buttons[0].enable()
buttons[1].enable()
else:
buttons[0].disable()
buttons[1].disable()
#end if
#end if
elif projOption == self.projectOptions[2]:
# Enable entries
self.ccpnEntry.configure(state='normal')
self.nameEntry.configure(state='normal')
if (len(self.ccpnEntry.get()) > 0
and len(self.nameEntry.get()) > 0):
buttons[0].enable()
buttons[1].enable()
else:
buttons[0].disable()
buttons[1].disable()
#end if
#end if
elif projOption == self.projectOptions[3]:
# Enable entries
self.cyanaEntry.configure(state='normal')
self.nameEntry.configure(state='normal')
if (len(self.cyanaEntry.get()) > 0
and len(self.nameEntry.get()) > 0):
buttons[0].enable()
buttons[1].enable()
else:
buttons[0].disable()
buttons[1].disable()
#end if
#end if
self.projectStatus.clear()
if not self.project:
self.projectStatus.setText('No open project')
self.closeProjectButton.setText('Close Project')
self.closeProjectButton.disable()
else:
self.projectStatus.setText(self.project.format())
self.closeProjectButton.enable()
self.closeProjectButton.setText(
sprintf('Close Project "%s"', self.project.name))
#end if
#end def
def redirectConsole(self):
#pipe = TextPipe(self.inputTextBox.text_area)
#sys.stdin = pipe
pipe = TextPipe(self.outputTextBox.text_area)
sys.stdout = pipe
nTmessage.stream = pipe
# end def redirectConsole
# sys.stderr = pipe
def resetConsole(self):
#sys.stdin = stdin
nTmessage.stream = stdout
sys.stdout = stdout
sys.stderr = stderr
# end def resetConsole
def open(self):
self.redirectConsole()
BasePopup.open(self)
# end def open
def close(self):
geometry = self.getGeometry()
self.resetConsole()
BasePopup.close(self)
print 'close:', geometry
sys.exit(0) # remove later
# end def close
def destroy(self):
geometry = self.getGeometry()
self.resetConsole()
BasePopup.destroy(self)
print 'destroy:', geometry
sys.exit(0) # remove later
def body(self, mainFrame):
mainFrame.grid_columnconfigure(1, weight=1, minsize=100)
mainFrame.config(borderwidth=5, relief='solid')
row = 0
label = Label(mainFrame, text="Frame (with sub-widgets):")
label.grid(row=row, column=0, sticky=Tkinter.E)
frame = Frame(mainFrame, relief='raised', border=2, background='#8080D0')
# Frame expands East-West
frame.grid(row=row, column=1, sticky=Tkinter.EW)
# Last column expands => Widgets pusted to the West
frame.grid_columnconfigure(3, weight=1)
# Label is within the sub frame
label = Label(frame, text='label ')
label.grid(row=0, column=0, sticky=Tkinter.W)
entry = Entry(frame, text='Entry', returnCallback=self.showWarning)
entry.grid(row=0, column=1, sticky=Tkinter.W)
self.check = CheckButton(frame, text='Checkbutton', selected=True, callback=self.updateObjects)
self.check.grid(row=0, column=2, sticky=Tkinter.W)
# stick a button to the East wall
button = Button(frame, text='Button', command=self.pressButton)
button.grid(row=0, column=3, sticky=Tkinter.E)
row += 1
label = Label(mainFrame, text="Text:")
label.grid(row=row, column=0, sticky=Tkinter.E)
self.textWindow = Text(mainFrame, text='Initial Text\n', width=60, height=5)
self.textWindow.grid(row=row, column=1, sticky=Tkinter.NSEW)
row += 1
label = Label(mainFrame, text="CheckButtons:")
label.grid(row=row, column=0, sticky=Tkinter.E)
entries = ['Alpha','Beta','Gamma','Delta']
selected = entries[2:]
self.checkButtons = CheckButtons(mainFrame, entries, selected=selected,select_callback=self.changedCheckButtons)
self.checkButtons.grid(row=row, column=1, sticky=Tkinter.W)
row += 1
label = Label(mainFrame, text="PartitionedSelector:")
label.grid(row=row, column=0, sticky=Tkinter.E)
labels = ['Bool','Int','Float','String']
objects = [type(0),type(1),type(1.0),type('a')]
selected = [type('a')]
self.partitionedSelector= PartitionedSelector(mainFrame, labels=labels,
objects=objects,
colors = ['red','yellow','green','#000080'],
callback=self.toggleSelector,selected=selected)
self.partitionedSelector.grid(row=row, column=1, sticky=Tkinter.EW)
row += 1
label = Label(mainFrame, text="PulldownMenu")
label.grid(row=row, column=0, sticky=Tkinter.E)
entries = ['Frodo','Pipin','Merry','Sam','Bill','Gandalf','Strider','Gimli','Legolas']
self.pulldownMenu = PulldownMenu(mainFrame, callback=self.selectPulldown,
entries=entries, selected_index=2,
do_initial_callback=False)
self.pulldownMenu.grid(row=row, column=1, sticky=Tkinter.W)
row += 1
label = Label(mainFrame, text="RadioButtons in a\nScrolledFrame.frame:")
label.grid(row=row, column=0, sticky=Tkinter.EW)
frame = ScrolledFrame(mainFrame, yscroll = False, doExtraConfig = True, width=100)
frame.grid(row=row, column=1, sticky=Tkinter.EW)
frame.grid_columnconfigure(0, weight=1)
self.radioButtons = RadioButtons(frame.frame, entries=entries,
select_callback=self.checkRadioButtons,
selected_index=1, relief='groove')
self.radioButtons.grid(row=0, column=0, sticky=Tkinter.W)
row += 1
label = Label(mainFrame, text="LabelFrame with\nToggleLabels inside:")
label.grid(row=row, column=0, sticky=Tkinter.E)
labelFrame = LabelFrame(mainFrame, text='Frame Title')
labelFrame.grid(row=row, column=1, sticky=Tkinter.NSEW)
labelFrame.grid_rowconfigure(0, weight=1)
labelFrame.grid_columnconfigure(3, weight=1)
self.toggleLabel1 = ToggleLabel(labelFrame, text='ScrolledMatrix', callback=self.toggleFrame1)
self.toggleLabel1.grid(row=0, column=0, sticky=Tkinter.W)
self.toggleLabel1.arrowOn()
self.toggleLabel2 = ToggleLabel(labelFrame, text='ScrolledGraph', callback=self.toggleFrame2)
self.toggleLabel2.grid(row=0, column=1, sticky=Tkinter.W)
self.toggleLabel3 = ToggleLabel(labelFrame, text='ScrolledCanvas', callback=self.toggleFrame3)
self.toggleLabel3.grid(row=0, column=2, sticky=Tkinter.W)
row += 1
mainFrame.grid_rowconfigure(row, weight=1)
label = Label(mainFrame, text="changing/shrinking frames:")
label.grid(row=row, column=0, sticky=Tkinter.E)
self.toggleRow = row
self.toggleFrame = Frame(mainFrame)
self.toggleFrame.grid(row=row, column=1, sticky=Tkinter.NSEW)
self.toggleFrame.grid_rowconfigure(0, weight=1)
self.toggleFrame.grid_columnconfigure(0, weight=1)
# option 1
self.intEntry = IntEntry(self, returnCallback = self.setNumber, width=8)
self.multiWidget = MultiWidget(self, Entry, options=None,
values=None, callback=self.setKeywords,
minRows=3, maxRows=5)
editWidgets = [None, None, self.intEntry, self.multiWidget]
editGetCallbacks = [None, None, self.getNumber, self.getKeywords]
editSetCallbacks = [None, None, self.setNumber, self.setKeywords]
headingList = ['Name','Color','Number','Keywords']
self.scrolledMatrix = ScrolledMatrix(self.toggleFrame, headingList=headingList,
editSetCallbacks=editSetCallbacks,
editGetCallbacks=editGetCallbacks,
editWidgets=editWidgets,
callback=self.selectObject,
multiSelect=False)
self.scrolledMatrix.grid(row=0, column=0, sticky=Tkinter.NSEW)
# option 2
self.scrolledGraph = ScrolledGraph(self.toggleFrame, width=400,
height=300, symbolSize=5,
symbols=['square','circle'],
dataColors=['#000080','#800000'],
lineWidths=[0,1] )
self.scrolledGraph.setZoom(1.3)
dataSet1 = [[0,0],[1,1],[2,4],[3,9],[4,16],[5,25]]
dataSet2 = [[0,0],[1,3],[2,6],[3,9],[4,12],[5,15]]
self.scrolledGraph.update(dataSets=[dataSet1,dataSet2],
xLabel = 'X axis label',
yLabel = 'Y axis label',
title = 'Main Title')
self.scrolledGraph.draw()
# option 3
self.scrolledCanvas = ScrolledCanvas(self.toggleFrame,relief = 'groove', borderwidth = 2, resizeCallback=None)
canvas = self.scrolledCanvas.canvas
font = 'Helvetica 10'
box = canvas.create_rectangle(10,10,150,200, outline='grey', fill='grey90')
line = canvas.create_line(0,0,200,200,fill='#800000', width=2)
text = canvas.create_text(120,50, text='Text', font=font, fill='black')
circle = canvas.create_oval(30,30,50,50,outline='#008000',fill='#404040',width=3)
row += 1
label = Label(mainFrame, text="FloatEntry:")
label.grid(row=row, column=0, sticky=Tkinter.E)
self.floatEntry = FloatEntry(mainFrame, text=3.14159265, returnCallback=self.floatEntryReturn)
self.floatEntry.grid(row=row, column=1, sticky=Tkinter.W)
row += 1
label = Label(mainFrame, text="Scale:")
label.grid(row=row, column=0, sticky=Tkinter.E)
self.scale = Scale(mainFrame, from_=10, to=90, value=50, orient=Tkinter.HORIZONTAL)
self.scale.grid(row=row, column=1, sticky=Tkinter.W)
row += 1
label = Label(mainFrame, text="Value Ramp:")
label.grid(row=row, column=0, sticky=Tkinter.E)
self.valueRamp = ValueRamp(mainFrame, self.valueRampCallback, speed = 1.5, delay = 50)
self.valueRamp.grid(row=row, column=1, sticky=Tkinter.W)
row += 1
label = Label(mainFrame, text="ButtonList:")
label.grid(row=row, column=0, sticky=Tkinter.E)
texts = ['Select File','Close','Quit']
commands = [self.selectFile, self.close, self.quit]
bottomButtons = ButtonList(mainFrame, texts=texts, commands=commands, expands=True)
bottomButtons.grid(row=row, column=1, sticky=Tkinter.EW)
self.protocol('WM_DELETE_WINDOW', self.quit)
class HcloudsMdPopup(BasePopup):
def __init__(self, parent, *args, **kw):
self.guiParent = parent
self.project = parent.getProject()
self.waiting = 0
self.constraintSet = None
self.constrLists = [None] * 4
self.numClouds = 100
self.filePrefix = 't_intra_'
self.cloudsFiles = []
self.adcAtomTypes = 'HN'
# step num, initial temp, final temp, cooling steps, MD steps, MD tau, rep scale
self.coolingScheme = []
self.coolingScheme.append([1, 1, 1, 3, 500, 0.001, 0])
self.coolingScheme.append([2, 80000, 4000, 19, 1000, 0.001, 0])
self.coolingScheme.append([3, 4000, 1, 5, 500, 0.001, 0])
self.coolingScheme.append([4, 15000, 1, 3, 1000, 0.001, 0])
self.coolingScheme.append([5, 1, 1, 5, 500, 0.001, 0])
self.coolingScheme.append([6, 8000, 1, 3, 1000, 0.001, 0])
self.coolingScheme.append([7, 1, 1, 5, 500, 0.001, 0])
self.coolingScheme.append([8, 3000, 25, 60, 2500, 0.001, 1])
self.coolingScheme.append([9, 25, 25, 1, 7500, 0.001, 1])
self.coolingScheme.append([10, 10, 10, 1, 7500, 0.001, 1])
self.coolingScheme.append([11, 0.01, 0.01, 1, 7500, 0.0005, 1])
self.coolingStep = None
BasePopup.__init__(self,
parent,
title="Hydrogen Cloud Molecular Dynamics",
**kw)
def body(self, guiFrame):
self.mdInitTempEntry = FloatEntry(self,
text='',
returnCallback=self.setMdInitTemp)
self.mdFinTempEntry = FloatEntry(self,
text='',
returnCallback=self.setMdFinTemp)
self.mdCoolStepsEntry = IntEntry(self,
text='',
returnCallback=self.setMdCoolSteps)
self.mdSimStepsEntry = IntEntry(self,
text='',
returnCallback=self.setMdSimSteps)
self.mdTauEntry = FloatEntry(self,
text='',
returnCallback=self.setMdTau)
self.mdRepScaleEntry = FloatEntry(self,
text='',
returnCallback=self.setMdRepScale)
guiFrame.grid_columnconfigure(0, weight=1)
row = 0
guiFrame.grid_rowconfigure(row, weight=1)
frame = LabelFrame(guiFrame, text='Input constraints')
frame.grid(row=row, column=0, sticky=Tkinter.NSEW)
frame.grid_columnconfigure(2, weight=1)
srow = 0
label = Label(frame, text='Constraint set:')
label.grid(row=srow, column=0, sticky=Tkinter.W)
self.constraintSetPulldown = PulldownMenu(
frame,
callback=self.changeConstraintSet,
selected_index=0,
do_initial_callback=0)
self.constraintSetPulldown.grid(row=srow, column=1, sticky=Tkinter.W)
srow += 1
label = Label(frame, text='Dist constraint list 1:')
label.grid(row=srow, column=0, sticky=Tkinter.W)
self.distance1Pulldown = PulldownMenu(
frame,
callback=self.changeDistance1ConstraintList,
selected_index=0,
do_initial_callback=0)
self.distance1Pulldown.grid(row=srow, column=1, sticky=Tkinter.W)
self.numConstr1Label = Label(frame, text='Constraints: 0')
self.numConstr1Label.grid(row=srow, column=2, sticky=Tkinter.W)
srow += 1
label = Label(frame, text='Dist constraint list 2:')
label.grid(row=srow, column=0, sticky=Tkinter.W)
self.distance2Pulldown = PulldownMenu(
frame,
callback=self.changeDistance2ConstraintList,
selected_index=0,
do_initial_callback=0)
self.distance2Pulldown.grid(row=srow, column=1, sticky=Tkinter.W)
self.numConstr2Label = Label(frame, text='Constraints: 0')
self.numConstr2Label.grid(row=srow, column=2, sticky=Tkinter.W)
srow += 1
label = Label(frame, text='Dist constraint list 3:')
label.grid(row=srow, column=0, sticky=Tkinter.W)
self.distance3Pulldown = PulldownMenu(
frame,
callback=self.changeDistance3ConstraintList,
selected_index=0,
do_initial_callback=0)
self.distance3Pulldown.grid(row=srow, column=1, sticky=Tkinter.W)
self.numConstr3Label = Label(frame, text='Constraints: 0')
self.numConstr3Label.grid(row=srow, column=2, sticky=Tkinter.W)
srow += 1
label = Label(frame, text='Dist constraint list 4:')
label.grid(row=srow, column=0, sticky=Tkinter.W)
self.distance4Pulldown = PulldownMenu(
frame,
callback=self.changeDistance4ConstraintList,
selected_index=0,
do_initial_callback=0)
self.distance4Pulldown.grid(row=srow, column=1, sticky=Tkinter.W)
self.numConstr4Label = Label(frame, text='Constraints: 0')
self.numConstr4Label.grid(row=srow, column=2, sticky=Tkinter.W)
row += 1
frame0 = LabelFrame(guiFrame, text='Cooling scheme')
frame0.grid(row=row, column=0, sticky=Tkinter.NSEW)
frame0.grid_columnconfigure(1, weight=1)
f0row = 0
frame0.grid_rowconfigure(f0row, weight=1)
colHeadings = [
'Step', 'Initial\nTemp.', 'Final\nTemp.', 'Cooling\nSteps',
'MD Steps', 'MD Tau', 'Rep.\nScale'
]
editWidgets = [
None, self.mdInitTempEntry, self.mdFinTempEntry,
self.mdCoolStepsEntry, self.mdSimStepsEntry, self.mdTauEntry,
self.mdRepScaleEntry
]
editGetCallbacks = [
None, self.getMdInitTemp, self.getMdFinTemp, self.getMdCoolSteps,
self.getMdSimSteps, self.getMdTau, self.getMdRepScale
]
editSetCallbacks = [
None, self.setMdInitTemp, self.setMdFinTemp, self.setMdCoolSteps,
self.setMdSimSteps, self.setMdTau, self.setMdRepScale
]
self.coolingSchemeMatrix = ScrolledMatrix(
frame0,
editSetCallbacks=editSetCallbacks,
editGetCallbacks=editGetCallbacks,
editWidgets=editWidgets,
maxRows=9,
initialRows=12,
headingList=colHeadings,
callback=self.selectCoolingStep,
objectList=self.coolingScheme,
textMatrix=self.coolingScheme)
self.coolingSchemeMatrix.grid(row=f0row,
column=0,
columnspan=4,
sticky=Tkinter.NSEW)
f0row += 1
texts = ['Move earlier', 'Move later', 'Add step', 'Remove step']
commands = [
self.moveStepEarlier, self.moveStepLater, self.addCoolingStep,
self.removeCoolingStep
]
self.coolingSchemeButtons = ButtonList(frame0,
expands=1,
commands=commands,
texts=texts)
self.coolingSchemeButtons.grid(row=f0row,
column=0,
columnspan=4,
sticky=Tkinter.EW)
row += 1
guiFrame.grid_rowconfigure(row, weight=1)
frame1 = LabelFrame(guiFrame, text='Dynamics control')
frame1.grid(row=row, column=0, sticky=Tkinter.NSEW)
frame1.grid_columnconfigure(1, weight=1)
f1row = 0
label20 = Label(frame1, text='Number of clouds:')
label20.grid(row=f1row, column=0, sticky=Tkinter.NW)
self.numCloudsEntry = IntEntry(frame1,
text=1,
returnCallback=self.setNumClouds,
width=10)
self.numCloudsEntry.grid(row=f1row, column=1, sticky=Tkinter.NW)
label21 = Label(frame1, text='Cloud file prefix:')
label21.grid(row=f1row, column=2, sticky=Tkinter.NW)
self.filePrefixEntry = Entry(frame1,
text='t_intra_',
returnCallback=self.setFilePrefix,
width=10)
self.filePrefixEntry.grid(row=f1row, column=3, sticky=Tkinter.NW)
f1row += 1
texts = ['Start molecular dynamics', 'Show dynamics progress']
commands = [self.startMd, self.showMdProgress]
self.mdButtons = ButtonList(frame1,
expands=1,
commands=commands,
texts=texts)
self.mdButtons.grid(row=f1row,
column=0,
columnspan=4,
sticky=Tkinter.NSEW)
row += 1
self.bottomButtons = createDismissHelpButtonList(guiFrame,
expands=0,
help_url=None)
self.bottomButtons.grid(row=row, column=0, sticky=Tkinter.EW)
self.update()
for func in ('__init__', 'delete', 'setName'):
for clazz in ('ccp.nmr.NmrConstraint.DistanceConstraintList', ):
Implementation.registerNotify(self.updateConstraintLists,
clazz, func)
for func in ('__init__', 'delete'):
Implementation.registerNotify(
self.updateConstraintSets,
'ccp.nmr.NmrConstraint.NmrConstraintStore', func)
def getContraintSetNames(self):
names = []
constraintSets = self.project.currentNmrProject.nmrConstraintStores
for set in constraintSets:
names.append('%d' % set.serial)
return names
def changeConstraintSet(self, i, name):
project = self.project
if project.currentNmrProject.nmrConstraintStores:
constraintSet = project.currentNmrProject.sortedNmrConstraintStores(
)[i]
else:
constraintSet = None
if constraintSet is not self.constraintSet:
self.constraintSet = constraintSet
self.updateConstraintLists()
self.update()
def updateConstraintLists(self, *opt):
constrListData = self.getConstraintLists()
constrListNames = [x[0] for x in constrListData]
constraintLists = [x[1] for x in constrListData]
# copes with self.constraintSet being None
if constrListNames:
i = 0
for constraintList in self.constrLists:
if constraintList not in constraintLists:
if i == 0:
self.constrLists[i] = constraintLists[0]
else:
self.constrLists[i] = None
i += 1
constraintLists.append(None)
constrListNames.append('<None>')
self.distance1Pulldown.setup(
constrListNames, constraintLists.index(self.constrLists[0]))
self.distance2Pulldown.setup(
constrListNames, constraintLists.index(self.constrLists[1]))
self.distance3Pulldown.setup(
constrListNames, constraintLists.index(self.constrLists[2]))
self.distance4Pulldown.setup(
constrListNames, constraintLists.index(self.constrLists[3]))
else:
self.constrLists = [None] * 4
self.distance1Pulldown.setup([], -1)
self.distance2Pulldown.setup([], -1)
self.distance3Pulldown.setup([], -1)
self.distance4Pulldown.setup([], -1)
def updateConstraintSets(self, *opt):
project = self.project
constraintSets = list(project.currentNmrProject.nmrConstraintStores)
if constraintSets:
constraintSetNames = self.getContraintSetNames()
# set defaults
if self.constraintSet not in constraintSets:
self.constraintSet = constraintSets[0]
if self.constraintSet:
j = 0
for constraintList in self.constrLists:
if constraintList and (constraintList.nmrConstraintStore
is not self.constraintSet):
if self.constraintSet.constraintLists and j == 0:
self.constrLists[
j] = self.constraintSet.constraintLists[0]
else:
self.constrLists[j] = None
j += 1
else:
self.constrLists = [None] * 4
i = constraintSets.index(self.constraintSet)
self.constraintSetPulldown.setup(constraintSetNames, i)
else:
self.constraintSet = None
self.constrLists = [None] * 4
self.constraintSetPulldown.setup([], -1)
def getConstraintListName(self, constraintList):
if constraintList.name:
listName = ':%s' % (constraintList.name)
else:
listName = ''
name = '%d:%d:%s%s' % (constraintList.nmrConstraintStore.serial,
constraintList.serial,
constraintList.className[:-14], listName)
return name
def getConstraintLists(self):
constraintLists = []
if self.constraintSet:
for constraintList in self.constraintSet.constraintLists:
if constraintList.className == 'DistanceConstraintList':
name = self.getConstraintListName(constraintList)
constraintLists.append([name, constraintList])
return constraintLists
def changeDistance1ConstraintList(self, i, name):
self.changeDistanceConstraintList(i, name, 0)
def changeDistance2ConstraintList(self, i, name):
self.changeDistanceConstraintList(i, name, 1)
def changeDistance3ConstraintList(self, i, name):
self.changeDistanceConstraintList(i, name, 2)
def changeDistance4ConstraintList(self, i, name):
self.changeDistanceConstraintList(i, name, 3)
def changeDistanceConstraintList(self, i, name, listNum):
project = self.project
constraintLists = self.getConstraintLists()
if constraintLists and (i < 4):
self.constrLists[listNum] = constraintLists[i][1]
else:
self.constrLists[listNum] = None
self.update()
def startMd(self):
self.setNumClouds()
self.setFilePrefix()
if ((self.constrLists != [None] * 4) and (self.numClouds > 0)
and self.filePrefix):
resDict = {}
for resonance in self.guiParent.project.currentNmrProject.resonances:
resDict[resonance.serial] = resonance
constraints = []
constraintStore = None
for dcl in self.constrLists:
if dcl:
constraintStore = dcl.nmrConstraintStore
constraints.extend(list(dcl.constraints))
resonances = []
for constraint in constraints:
for item in constraint.items:
for fixedResonance in item.resonances:
if fixedResonance.resonanceSerial is None:
resonance = newResonance(
self.guiParent.project,
isotopeCode=fixedResonance.isotopeCode)
resonance.setName(fixedResonance.name)
fixedResonance.setResonanceSerial(resonance.serial)
resDict[resonance.serial] = resonance
if fixedResonance.resonanceSet:
atomSets = list(
fixedResonance.resonanceSet.atomSets)
assignAtomsToRes(atomSets, resonance)
if resDict.get(
fixedResonance.resonanceSerial) is not None:
resonances.append(
resDict[fixedResonance.resonanceSerial])
resDict[fixedResonance.resonanceSerial] = None
resonances, intraConstraintList = self.makeIntraConstraints(
resonances, constraintStore)
constraints.extend(list(intraConstraintList.constraints))
resonances, interConstraintList = self.makeInterConstraints(
resonances, constraintStore)
constraints.extend(list(interConstraintList.constraints))
startMdProcess(self.numClouds, constraints, resonances,
self.coolingScheme, self.filePrefix)
#structGen = self.distanceConstraintList.structureGeneration
serials = []
for resonance in resonances:
serials.append(resonance.serial)
clouds = []
for i in range(self.numClouds):
clouds.append('%s%3.3d.pdb' % (self.filePrefix, i))
self.guiParent.application.setValues(constraintStore,
'clouds',
values=clouds)
self.guiParent.application.setValues(constraintStore,
'cloudsResonances',
values=serials)
# do better than this check for creation
def makeInterConstraints(self, resonances, constraintSet):
from ccpnmr.analysis.core.ConstraintBasic import getFixedResonance
from ccpnmr.analysis.core.AssignmentBasic import findConnectedSpinSystem
project = constraintSet.root
constraintList = constraintSet.newDistanceConstraintList()
constraintList.name = 'Seq connections'
resDict = {}
spinSystemDict = {}
for resonance in resonances:
resDict[resonance] = True
spinSystem = resonance.resonanceGroup
if spinSystem:
spinSystemDict[spinSystem] = None
spinSystems = spinSystemDict.keys()
for spinSystem in spinSystems:
nextSpinSystem = findConnectedSpinSystem(spinSystem, delta=1)
if nextSpinSystem:
ca = spinSystem.newAtoms.get('CA')
c = spinSystem.newAtoms.get('C')
n = nextSpinSystem.newAtoms.get('N')
if ca and c and n:
if resDict.get(ca) is None:
resonances.append(ca)
if resDict.get(c) is None:
resonances.append(c)
if resDict.get(n) is None:
resonances.append(n)
c_n = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=1.32,
upperLimit=1.35,
lowerLimit=1.29,
error=0.06)
# below based on angle constraints
ca_n = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=2.415,
upperLimit=2.513,
lowerLimit=2.316,
error=0.197)
frCa = getFixedResonance(constraintSet, ca)
frC = getFixedResonance(constraintSet, c)
frN = getFixedResonance(constraintSet, n)
item = ca_n.newDistanceConstraintItem(
resonances=[frCa, frN])
item = c_n.newDistanceConstraintItem(resonances=[frC, frN])
return resonances, constraintList
def makeIntraConstraints(self, resonances, constraintSet):
from ccpnmr.analysis.core.ConstraintBasic import getFixedResonance
project = constraintSet.root
constraintList = constraintSet.newDistanceConstraintList()
constraintList.name = 'Backbone intra'
dict = {}
for resonance in resonances:
if resonance.resonanceGroup and resonance.assignNames:
dict[resonance] = 1
resonance.resonanceGroup.newAtoms = {}
for resonance in dict.keys():
ss = resonance.resonanceGroup
if resonance.assignNames[0] == 'H':
for resonance2 in ss.resonances:
if dict.get(resonance2
) and resonance2.assignNames[0][:2] == 'HA':
ca = ss.newAtoms.get('CA')
if ca is None:
ca = project.newResonance(isotopeCode='13C',
assignNames=[
'CA',
])
resonances.append(ca)
c = ss.newAtoms.get('C')
if c is None:
c = project.newResonance(isotopeCode='13C',
assignNames=[
'C',
])
resonances.append(c)
n = ss.newAtoms.get('N')
if n is None:
n = project.newResonance(isotopeCode='15N',
assignNames=[
'N',
])
resonances.append(n)
ss.newAtoms['C'] = c
ss.newAtoms['CA'] = ca
ss.newAtoms['N'] = n
frCa = getFixedResonance(constraintSet, ca)
frC = getFixedResonance(constraintSet, c)
frN = getFixedResonance(constraintSet, n)
frH = getFixedResonance(constraintSet, resonance)
frha = getFixedResonance(constraintSet, resonance2)
h_n = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=1.01,
upperLimit=1.1,
lowerLimit=0.9,
error=0.2)
n_ca = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=1.465,
upperLimit=1.50,
lowerLimit=1.43,
error=0.07)
ca_c = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=1.525,
upperLimit=1.55,
lowerLimit=1.50,
error=0.05)
# below based on angle constraints
n_c = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=2.464,
upperLimit=2.572,
lowerLimit=2.356,
error=0.216)
item = h_n.newDistanceConstraintItem(
resonances=[frH, frN])
item = n_ca.newDistanceConstraintItem(
resonances=[frN, frCa])
item = ca_c.newDistanceConstraintItem(
resonances=[frCa, frC])
item = n_c.newDistanceConstraintItem(
resonances=[frN, frC])
if ss.newAtoms.get('CAHA') is None:
ca_ha = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=1.09,
upperLimit=1.19,
lowerLimit=0.99,
error=0.2)
item = ca_ha.newDistanceConstraintItem(
resonances=[frC, frha])
ss.newAtoms['CAHA'] = 1
if resonance.assignNames[0][:2] == 'HA':
for resonance2 in ss.resonances:
if dict.get(resonance2
) and resonance2.assignNames[0][:2] == 'HB':
ca = ss.newAtoms.get('CA')
if ca is None:
ca = project.newResonance(isotopeCode='13C',
assignNames=[
'CA',
])
resonances.append(ca)
cb = ss.newAtoms.get('CB')
if cb is None:
cb = project.newResonance(isotopeCode='13C',
assignNames=[
'CB',
])
resonances.append(cb)
ss.newAtoms['CA'] = cb
ss.newAtoms['CB'] = ca
ss.newAtoms['CAHA'] = 1
frCA = getFixedResonance(constraintSet, ca)
frCB = getFixedResonance(constraintSet, cb)
frHA = getFixedResonance(constraintSet, resonance)
frHB = getFixedResonance(constraintSet, resonance2)
c_b = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=1.09,
upperLimit=1.19,
lowerLimit=0.99,
error=0.2)
c_c = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=1.53,
upperLimit=1.56,
lowerLimit=1.51,
error=0.05)
item = c_b.newDistanceConstraintItem(
resonances=[frCB, frHB])
item = c_c.newDistanceConstraintItem(
resonances=[frCA, frCB])
if ss.newAtoms.get('CAHA') is None:
c_a = constraintList.newDistanceConstraint(
weight=1.0,
origData=1.0,
targetValue=1.09,
upperLimit=1.19,
lowerLimit=0.99,
error=0.2)
item = c_a.newDistanceConstraintItem(
resonances=[frCA, frHA])
ss.newAtoms['CAHA'] = 1
return resonances, constraintList
def showMdProgress(self):
n = 0
m = self.numClouds
for i in range(m):
pdbFileName = '%s%3.3d.pdb' % (self.filePrefix, i)
if os.path.exists(pdbFileName):
n += 1
p = n * 100 / float(m)
text = 'Done %d of %d clouds (%1.2f)%%' % (n, m, p)
showInfo('MD Progress', text)
def setFilePrefix(self, text=None):
if not text:
text = self.filePrefixEntry.get()
if text:
self.filePrefix = text
def setNumClouds(self, n=None, *event):
if not n:
n = self.numCloudsEntry.get()
if n:
self.numClouds = int(n)
def update(self):
self.updateConstraintSets()
self.updateConstraintLists()
if (self.constrLists != [None] * 4) and self.coolingScheme:
self.mdButtons.buttons[0].enable()
self.mdButtons.buttons[1].enable()
else:
self.mdButtons.buttons[0].disable()
self.mdButtons.buttons[1].disable()
if self.constrLists[0]:
self.numConstr1Label.set('Constraints: ' +
str(len(self.constrLists[0].constraints)))
else:
self.numConstr1Label.set('Constraints: 0')
if self.constrLists[1]:
self.numConstr2Label.set('Constraints: ' +
str(len(self.constrLists[1].constraints)))
else:
self.numConstr2Label.set('Constraints: 0')
if self.constrLists[2]:
self.numConstr3Label.set('Constraints: ' +
str(len(self.constrLists[2].constraints)))
else:
self.numConstr3Label.set('Constraints: 0')
if self.constrLists[3]:
self.numConstr4Label.set('Constraints: ' +
str(len(self.constrLists[3].constraints)))
else:
self.numConstr4Label.set('Constraints: 0')
def getMdInitTemp(self, coolingStep):
self.mdInitTempEntry.set(coolingStep[1])
def getMdFinTemp(self, coolingStep):
self.mdFinTempEntry.set(coolingStep[2])
def getMdCoolSteps(self, coolingStep):
self.mdCoolStepsEntry.set(coolingStep[3])
def getMdSimSteps(self, coolingStep):
self.mdSimStepsEntry.set(coolingStep[4])
def getMdTau(self, coolingStep):
self.mdTauEntry.set(coolingStep[5])
def getMdRepScale(self, coolingStep):
self.mdRepScaleEntry.set(coolingStep[6])
def setMdInitTemp(self, event):
value = self.mdInitTempEntry.get()
if value is not None:
self.coolingStep[1] = value
self.updateCoolingScheme()
def setMdFinTemp(self, event):
value = self.mdFinTempEntry.get()
if value is not None:
self.coolingStep[2] = value
self.updateCoolingScheme()
def setMdCoolSteps(self, event):
value = self.mdCoolStepsEntry.get()
if value is not None:
self.coolingStep[3] = value
self.updateCoolingScheme()
def setMdSimSteps(self, event):
value = self.mdSimStepsEntry.get()
if value is not None:
self.coolingStep[4] = value
self.updateCoolingScheme()
def setMdTau(self, event):
value = self.mdTauEntry.get()
if value is not None:
self.coolingStep[5] = value
self.updateCoolingScheme()
def setMdRepScale(self, event):
value = self.mdRepScaleEntry.get()
if value is not None:
self.coolingStep[6] = value
self.updateCoolingScheme()
def selectCoolingStep(self, object, row, col):
self.coolingStep = object
def moveStepEarlier(self):
if self.coolingStep:
i = self.coolingStep[0] - 1
if i > 0:
coolingStep = self.coolingScheme[i - 1]
coolingStep[0] = i + 1
self.coolingStep[0] = i
self.coolingScheme[i - 1] = self.coolingStep
self.coolingScheme[i] = coolingStep
self.updateCoolingScheme()
self.coolingSchemeMatrix.hilightObject(self.coolingStep)
def moveStepLater(self):
if self.coolingStep:
i = self.coolingStep[0] - 1
if i < len(self.coolingScheme) - 1:
coolingStep = self.coolingScheme[i + 1]
coolingStep[0] = i + 1
self.coolingStep[0] = i + 2
self.coolingScheme[i + 1] = self.coolingStep
self.coolingScheme[i] = coolingStep
self.updateCoolingScheme()
self.coolingSchemeMatrix.hilightObject(self.coolingStep)
def addCoolingStep(self):
i = len(self.coolingScheme) + 1
datum = [i, 3000, 100, 10, 2500, 0.001, 1]
self.coolingScheme.append(datum)
self.updateCoolingScheme()
def removeCoolingStep(self):
if self.coolingStep:
coolingScheme = []
i = 0
for coolingStep in self.coolingScheme:
if coolingStep is not self.coolingStep:
i += 1
coolingStep[0] = i
coolingScheme.append(coolingStep)
self.coolingScheme = coolingScheme
self.updateCoolingScheme()
def updateCoolingScheme(self):
objectList = self.coolingScheme
textMatrix = self.coolingScheme
self.coolingSchemeMatrix.update(objectList=objectList,
textMatrix=textMatrix)
def updateMidgeParams(self):
data = [
self.specFreq, self.maxIter, self.mixTime, self.corrTime,
self.leakRate, self.maxIntens
]
self.midgeParamsMatrix.update(textMatrix=[
data,
])
def destroy(self):
for func in ('__init__', 'delete', 'setName'):
for clazz in ('ccp.nmr.NmrConstraint.DistanceConstraintList', ):
Implementation.unregisterNotify(self.updateConstraintLists,
clazz, func)
for func in ('__init__', 'delete'):
Implementation.unregisterNotify(
self.updateConstraintSets,
'ccp.nmr.NmrConstraint.NmrConstraintStore', func)
BasePopup.destroy(self)
