class clip(QtGui.QMainWindow):
"""An application called clip."""
def __init__(self):
QtGui.QMainWindow.__init__(self)
self.appTitle = "CLIP 3 - Cologne Laue Indexation Program"
self.setWindowTitle(self.appTitle)
self.setBackgroundRole(QtGui.QPalette.Window)
self.MdiArea=QtGui.QMdiArea(self)
self.setCentralWidget(self.MdiArea)
self.windowMapper = QtCore.QSignalMapper()
self.setWindowIcon(QtGui.QIcon(':/Clip.png'))
self.connect(self.windowMapper, QtCore.SIGNAL('mapped(QWidget *)'), self.MdiArea.setActiveSubWindow)
self.transferCurveMapper=QtCore.QSignalMapper()
#self.connect(self.TransferCurveMapper, QtCore.SIGNAL('mapped(QWidget*)'), self.)
self.lastOpenDir='.'
self.statusBar().showMessage(self.appTitle+" ready", 2000)
self.crystalStore=ObjectStore()
self.tools=[ImgTransferCurve(self), RotateCrystal(self), Reorient(self), ReflexInfo(self), Fit(self)]
self.initActions()
self.initMenu()
self.initToolbar()
#FIXME: Remove Debug Code
w=self.slotNewCrystal()
w.crystal.setCell(5, 5, 5, 90, 90, 90)
self.slotNewStereoProjector()
self.slotNewLauePlaneProjector()
def initMenu(self):
menudef = [
('&File',
[('New Crystal', self.slotNewCrystal),
('New Stereographic Projection', self.slotNewStereoProjector),
('New Laue Projection', self.slotNewLauePlaneProjector)]),
('&Tools',
[]),
('&Windows',
[]),
('&Help',
[('&About', self.slotAbout),
('About &QT', self.slotAboutQT),
(None,),
('What\'s this', self.slotWhatsThis)])
]
menus=[]
for menuName, subMenu in menudef:
menu=self.menuBar().addMenu( menuName )
for menuOption in subMenu:
if len(menuOption)==1:
menu.addSeparator()
elif len(menuOption)==2:
menu.addAction(menuOption[0], menuOption[1])
elif len(menuOption)==3:
act=menu.addAction(menuOption[0])
act.setCheckable(True)
act.setChecked(menuOption[2])
self.connect(act, QtCore.SIGNAL('toggled(bool)'), menuOption[1])
menus.append(menu)
toolMenu=menus[1]
for t in self.tools:
toolMenu.addAction(t.menuName, t.showWindow)
self.windowMenu=menus[2]
self.slotUpdateWindowMenu()
self.connect(self.windowMenu, QtCore.SIGNAL('aboutToShow()'), self.slotUpdateWindowMenu)
def initActions(self):
self.closeAct = QtGui.QAction("Cl&ose", self)
self.closeAct.setShortcut("Ctrl+F4")
self.closeAct.setStatusTip("Close the active window")
self.connect(self.closeAct, QtCore.SIGNAL('triggered()'), self.MdiArea.closeActiveSubWindow)
self.closeAllAct = QtGui.QAction("Close &All", self);
self.closeAllAct.setStatusTip("Close all the windows");
self.connect(self.closeAllAct, QtCore.SIGNAL('triggered()'), self.MdiArea.closeAllSubWindows)
self.tileAct = QtGui.QAction("&Tile", self)
self.tileAct.setStatusTip("Tile the windows");
self.connect(self.tileAct, QtCore.SIGNAL('triggered()'), self.MdiArea.tileSubWindows);
self.cascadeAct = QtGui.QAction("&Cascade", self)
self.cascadeAct.setStatusTip("Cascade the windows")
self.connect(self.cascadeAct, QtCore.SIGNAL('triggered()'), self.MdiArea.cascadeSubWindows)
self.nextAct = QtGui.QAction("Ne&xt", self)
self.nextAct.setStatusTip("Move the focus to the next window")
self.connect(self.nextAct, QtCore.SIGNAL('triggered()'), self.MdiArea.activateNextSubWindow)
self.previousAct = QtGui.QAction("Pre&vious", self)
self.previousAct.setStatusTip("Move the focus to the previous window")
self.connect(self.previousAct, QtCore.SIGNAL('triggered()'), self.MdiArea.activatePreviousSubWindow)
def initToolbar(self):
pass
def slotUpdateWindowMenu(self):
self.windowMenu.clear()
self.windowMenu.addAction(self.closeAct)
self.windowMenu.addAction(self.closeAllAct)
self.windowMenu.addSeparator()
self.windowMenu.addAction(self.tileAct)
self.windowMenu.addAction(self.cascadeAct)
self.windowMenu.addSeparator()
self.windowMenu.addAction(self.nextAct)
self.windowMenu.addAction(self.previousAct)
windows = self.MdiArea.subWindowList();
if len(windows)>0:
self.windowMenu.addSeparator()
for i,w in enumerate(windows):
if i < 9:
text = "&%i %s"%(i+1, w.windowTitle())
else:
text = "%i %s"%(i+1, w.windowTitle())
action = self.windowMenu.addAction(text)
action.setCheckable(True)
action.setChecked(w==self.MdiArea.activeSubWindow())
self.connect(action, QtCore.SIGNAL('triggered()'), self.windowMapper, QtCore.SLOT('map()'))
self.windowMapper.setMapping(action, w)
def slotWhatsThis(self):
self.whatsThis()
def slotAbout(self):
QtGui.QMessageBox.about(self, self.appTitle, \
"This is the Cologne Laue Indexation Program (CLIP)\n" + \
"a program for indexing Laue exposures")
def slotAboutQT(self):
QtGui.QMessageBox.aboutQt(self, self.appTitle)
def slotNewCrystal(self):
wid = Crystal(self)
self.crystalStore.addObject(wid.crystal)
for t in self.tools:
self.connect(wid.crystal, QtCore.SIGNAL('rotationAxisChanged()'), t.rotAxisChanged)
self.connect(wid.crystal, QtCore.SIGNAL('orientationChanged()'), t.orientationChanged)
mdi=self.MdiArea.addSubWindow(wid)
mdi.setWindowIcon(wid.windowIcon())
wid.show()
return wid
def slotNewStereoProjector(self):
wid = ProjectionPlaneWidget(0, self)
for t in self.tools:
self.connect(wid, QtCore.SIGNAL('reflexInfo(int,int,int)'), t.reflexInfo)
self.connect(wid, QtCore.SIGNAL('projectorAddedRotation(double)'), t.addedRotation)
if self.crystalStore.size()>0:
wid.projector.connectToCrystal(self.crystalStore.at(0))
wid.setWindowTitle('Stereographic Projection')
mdi=self.MdiArea.addSubWindow(wid)
mdi.setWindowIcon(wid.windowIcon())
wid.show()
return wid
def slotNewLauePlaneProjector(self):
wid = ProjectionPlaneWidget(1, self)
for t in self.tools:
self.connect(wid, QtCore.SIGNAL('reflexInfo(int,int,int)'), t.reflexInfo)
self.connect(wid, QtCore.SIGNAL('projectorAddedRotation(double)'), t.addedRotation)
if self.crystalStore.size()>0:
wid.projector.connectToCrystal(self.crystalStore.at(0))
wid.setWindowTitle('LauePlane')
mdi=self.MdiArea.addSubWindow(wid)
mdi.setWindowIcon(wid.windowIcon())
wid.show()
return wid
class clip(QtGui.QMainWindow):
"""An application called clip."""
def __init__(self):
QtGui.QMainWindow.__init__(self)
self.appTitle = "CLIP 3 - Cologne Laue Indexation Program"
self.setWindowTitle(self.appTitle)
self.setBackgroundRole(QtGui.QPalette.Window)
self.MdiArea=QtGui.QMdiArea(self)
self.setCentralWidget(self.MdiArea)
self.windowMapper = QtCore.QSignalMapper()
self.setWindowIcon(QtGui.QIcon(':/Clip.png'))
self.connect(self.windowMapper, QtCore.SIGNAL('mapped(QWidget *)'), self.MdiArea.setActiveSubWindow)
self.transferCurveMapper=QtCore.QSignalMapper()
#self.connect(self.TransferCurveMapper, QtCore.SIGNAL('mapped(QWidget*)'), self.)
self.statusBar().showMessage(self.appTitle+" ready", 2000)
self.crystalStore=ObjectStore()
self.tools=[ImgTransferCurve(self), RotateCrystal(self), Reorient(self), ReflexInfo(self), Fit(self)]
self.initActions()
self.initMenu()
self.initToolbar()
self.loadWorkspaceFromFile('DefaultWorkspace.cws')
def initMenu(self):
menudef = [
('&File',
[('Load Workspace', self.slotLoadWorkspace),
('Save Workspace', self.slotSaveWorkspace),
('New Crystal', self.slotNewCrystal),
('New Stereographic Projection', self.slotNewStereoProjector),
('New Laue Projection', self.slotNewLauePlaneProjector)]),
('&Tools',
[]),
('&Windows',
[]),
('&Help',
[('&About', self.slotAbout),
('About &QT', self.slotAboutQT),
(None,),
('What\'s this', self.slotWhatsThis)])
]
menus=[]
for menuName, subMenu in menudef:
menu=self.menuBar().addMenu( menuName )
for menuOption in subMenu:
if len(menuOption)==1:
menu.addSeparator()
elif len(menuOption)==2:
menu.addAction(menuOption[0], menuOption[1])
elif len(menuOption)==3:
act=menu.addAction(menuOption[0])
act.setCheckable(True)
act.setChecked(menuOption[2])
self.connect(act, QtCore.SIGNAL('toggled(bool)'), menuOption[1])
menus.append(menu)
toolMenu=menus[1]
for t in self.tools:
toolMenu.addAction(t.menuName, t.showWindow)
self.windowMenu=menus[2]
self.slotUpdateWindowMenu()
self.connect(self.windowMenu, QtCore.SIGNAL('aboutToShow()'), self.slotUpdateWindowMenu)
def initActions(self):
self.closeAct = QtGui.QAction("Cl&ose", self)
self.closeAct.setShortcut("Ctrl+F4")
self.closeAct.setStatusTip("Close the active window")
self.connect(self.closeAct, QtCore.SIGNAL('triggered()'), self.MdiArea.closeActiveSubWindow)
self.closeAllAct = QtGui.QAction("Close &All", self);
self.closeAllAct.setStatusTip("Close all the windows");
self.connect(self.closeAllAct, QtCore.SIGNAL('triggered()'), self.MdiArea.closeAllSubWindows)
self.tileAct = QtGui.QAction("&Tile", self)
self.tileAct.setStatusTip("Tile the windows");
self.connect(self.tileAct, QtCore.SIGNAL('triggered()'), self.MdiArea.tileSubWindows);
self.cascadeAct = QtGui.QAction("&Cascade", self)
self.cascadeAct.setStatusTip("Cascade the windows")
self.connect(self.cascadeAct, QtCore.SIGNAL('triggered()'), self.MdiArea.cascadeSubWindows)
self.nextAct = QtGui.QAction("Ne&xt", self)
self.nextAct.setStatusTip("Move the focus to the next window")
self.connect(self.nextAct, QtCore.SIGNAL('triggered()'), self.MdiArea.activateNextSubWindow)
self.previousAct = QtGui.QAction("Pre&vious", self)
self.previousAct.setStatusTip("Move the focus to the previous window")
self.connect(self.previousAct, QtCore.SIGNAL('triggered()'), self.MdiArea.activatePreviousSubWindow)
def initToolbar(self):
pass
def slotUpdateWindowMenu(self):
self.windowMenu.clear()
self.windowMenu.addAction(self.closeAct)
self.windowMenu.addAction(self.closeAllAct)
self.windowMenu.addSeparator()
self.windowMenu.addAction(self.tileAct)
self.windowMenu.addAction(self.cascadeAct)
self.windowMenu.addSeparator()
self.windowMenu.addAction(self.nextAct)
self.windowMenu.addAction(self.previousAct)
windows = self.MdiArea.subWindowList();
if len(windows)>0:
self.windowMenu.addSeparator()
for i,w in enumerate(windows):
if i < 9:
text = "&%i %s"%(i+1, w.windowTitle())
else:
text = "%i %s"%(i+1, w.windowTitle())
action = self.windowMenu.addAction(text)
action.setCheckable(True)
action.setChecked(w==self.MdiArea.activeSubWindow())
self.connect(action, QtCore.SIGNAL('triggered()'), self.windowMapper, QtCore.SLOT('map()'))
self.windowMapper.setMapping(action, w)
def slotWhatsThis(self):
self.whatsThis()
def slotAbout(self):
QtGui.QMessageBox.about(self, self.appTitle, \
"This is the Cologne Laue Indexation Program (CLIP)\n" + \
"a program for indexing Laue exposures")
def slotAboutQT(self):
QtGui.QMessageBox.aboutQt(self, self.appTitle)
def slotNewCrystal(self):
wid = Crystal(self)
self.crystalStore.addObject(wid.crystal)
for t in self.tools:
self.connect(wid.crystal, QtCore.SIGNAL('rotationAxisChanged()'), t.rotAxisChanged)
self.connect(wid.crystal, QtCore.SIGNAL('orientationChanged()'), t.orientationChanged)
self.connect(wid.crystal, QtCore.SIGNAL('constrainsChanged()'), t.crystalConstrainChanged)
mdi=self.MdiArea.addSubWindow(wid)
mdi.setWindowIcon(wid.windowIcon())
wid.show()
return wid
def slotNewStereoProjector(self):
self.newProjector(StereoProjector())
def slotNewLauePlaneProjector(self):
self.newProjector(LauePlaneProjector())
def newProjector(self, projector):
wid = ProjectionPlaneWidget(projector, self)
for t in self.tools:
self.connect(wid, QtCore.SIGNAL('reflexInfo(int,int,int)'), t.reflexInfo)
self.connect(wid, QtCore.SIGNAL('projectorAddedRotation(double)'), t.addedRotation)
#if self.crystalStore.size()>0:
# wid.projector.connectToCrystal(self.crystalStore.at(0))
mdi=self.MdiArea.addSubWindow(wid)
mdi.setWindowIcon(wid.windowIcon())
wid.show()
return wid
def slotSaveWorkspace(self):
xmlString=QtCore.QString()
w=QtCore.QXmlStreamWriter(xmlString)
w.setAutoFormatting(True)
w.setAutoFormattingIndent(2)
w.writeStartElement('ClipWorkspace')
for n in range(self.crystalStore.size()):
c=self.crystalStore.at(n)
w.writeStartElement('CrystalConnection')
c.parent().crystaldata2xml(w)
for p in c.getConnectedProjectors():
p.parent().projector2xml(w)
w.writeEndElement()
w.writeEndElement()
fileName = QtGui.QFileDialog.getSaveFileName(self, 'Choose File to save Cell', '', 'Clip Workspace files (*.cws);;All Files (*)')
if fileName!="":
f=open(str(fileName), 'w')
f.write(str(xmlString))
f.close()
def slotLoadWorkspace(self):
fileName = str(QtGui.QFileDialog.getOpenFileName(self, 'Choose Cell to load from File', '', 'Clip Cell files (*.cws);;All Files (*)'))
self.loadWorkspaceFromFile(fileName)
def loadWorkspaceFromFile(self, fileName):
try:
f=open(fileName)
s=''.join(f.readlines())
except:
return
else:
r=QtCore.QXmlStreamReader(s)
while not r.atEnd():
r.readNext()
if r.name()=='CrystalConnection' and r.isStartElement():
self.loadCrystalConnection(r)
def loadCrystalConnection(self, r):
if r.name()!='CrystalConnection' or not r.isStartElement():
return
cryst=None
while not r.atEnd() and not (r.isEndElement() and r.name()=="CrystalConnection"):
if r.readNext()==QtCore.QXmlStreamReader.StartElement:
if r.name()=="Crystal" and not cryst:
cryst=self.slotNewCrystal()
cryst.loadFromXML(r)
elif r.name()=='ProjectionPlane' and cryst:
s=r.attributes().value('projectorType')
if not s.isNull():
pr=eval(str(s.toString())+'()')
proj=self.newProjector(pr)
proj.projector.connectToCrystal(cryst.crystal)
proj.loadFromXML(r)