def __init__(self, parent = None, graph = None, bgrx = True):
qt.QWidget.__init__(self, parent)
self.setWindowTitle("PyMca RGB Correlator")
self.setWindowIcon(qt.QIcon(qt.QPixmap(RGBCorrelatorGraph.IconDict['gioconda16'])))
self.mainLayout = qt.QVBoxLayout(self)
self.mainLayout.setContentsMargins(0, 0, 0, 0)
self.mainLayout.setSpacing(6)
self.splitter = qt.QSplitter(self)
self.splitter.setOrientation(qt.Qt.Horizontal)
self.controller = RGBCorrelatorWidget.RGBCorrelatorWidget(self.splitter)
self._y1AxisInverted = False
self._imageBuffer = None
self._matplotlibSaveImage = None
standaloneSaving = True
if graph is None:
if MATPLOTLIB:
standaloneSaving = False
if USE_MASK_WIDGET:
self.graphWidgetContainer = MaskImageWidget.MaskImageWidget(self.splitter,
selection=True,
imageicons=True,
standalonesave=standaloneSaving,
profileselection=False,
polygon=True)
self.graphWidget = self.graphWidgetContainer.graphWidget
else:
self.graphWidget = RGBCorrelatorGraph.RGBCorrelatorGraph(self.splitter,
standalonesave=standaloneSaving)
if not standaloneSaving:
self.graphWidget.saveToolButton.clicked.connect( \
self._saveToolButtonSignal)
self._saveMenu = qt.QMenu()
self._saveMenu.addAction(QString("Standard"), self.graphWidget._saveIconSignal)
self._saveMenu.addAction(QString("Matplotlib") , self._saveMatplotlibImage)
self.graph = self.graphWidget.graph
#add flip Icon
self.graphWidget.hFlipToolButton.clicked.connect( \
self._hFlipIconSignal)
self._handleGraph = True
else:
self.graph = graph
self._handleGraph = False
#self.splitter.setStretchFactor(0,1)
#self.splitter.setStretchFactor(1,1)
self.mainLayout.addWidget(self.splitter)
self.reset = self.controller.reset
self.addImage = self.controller.addImage
self.removeImage = self.controller.removeImage
self.addImageSlot = self.controller.addImageSlot
self.removeImageSlot = self.controller.removeImageSlot
self.replaceImageSlot = self.controller.replaceImageSlot
self.setImageShape = self.controller.setImageShape
self.update = self.controller.update
self.transposeImages = self.controller.transposeImages
self.controller.sigRGBCorrelatorWidgetSignal.connect( \
self.correlatorSignalSlot)
def test():
import logging
app = qt.QApplication([])
app.lastWindowClosed.connect(app.quit)
if 0:
graphWidget = RGBCorrelatorGraph.RGBCorrelatorGraph()
graph = graphWidget.graph
w = RGBCorrelator(graph=graph)
else:
w = RGBCorrelator()
w.resize(800, 600)
import getopt
from PyMca5.PyMcaCore.LoggingLevel import getLoggingLevel
options = ''
longoptions = ["logging=", "debug="]
opts, args = getopt.getopt(
sys.argv[1:],
options,
longoptions)
logging.basicConfig(level=getLoggingLevel(opts))
filelist=args
if len(filelist):
try:
import DataSource
DataReader = DataSource.DataSource
except:
import EdfFileDataSource
DataReader = EdfFileDataSource.EdfFileDataSource
for fname in filelist:
source = DataReader(fname)
for key in source.getSourceInfo()['KeyList']:
dataObject = source.getDataObject(key)
w.addImage(dataObject.data, os.path.basename(fname)+" "+key)
else:
print("This is a just test method using 100 x 100 matrices.")
print("Run PyMcaPostBatch to have file loading capabilities.")
array1 = numpy.arange(10000)
array2 = numpy.resize(numpy.arange(10000), (100, 100))
array2 = numpy.transpose(array2)
array3 = array1 * 1
w.addImage(array1)
w.addImage(array2)
w.addImage(array3)
w.setImageShape([100, 100])
w.show()
app.exec_()