def loadFile(self):
dataHolder = DataHolder()
xDataSet = DatasetFactory.zeros(1,len(self.xData),self.xData)
xDataSet.setName(self.xName)
dataHolder.addDataSet(xDataSet.getName(), xDataSet);
yDataSet = DatasetFactory.zeros(1,len(self.yData),self.yData)
yDataSet.setName(self.yName)
dataHolder.addDataSet(yDataSet.getName(), yDataSet);
return dataHolder
def loadFile(self):
dataHolder = DataHolder()
xDataSet = DatasetFactory.zeros(1, len(self.xData), self.xData)
xDataSet.setName(self.xName)
dataHolder.addDataSet(xDataSet.getName(), xDataSet)
yDataSet = DatasetFactory.zeros(1, len(self.yData), self.yData)
yDataSet.setName(self.yName)
dataHolder.addDataSet(yDataSet.getName(), yDataSet)
return dataHolder
def testRenderShapes(self):
self.testAddShape()
ds = DatasetFactory.zeros(4, 5)
result = self.renderer.renderShapes(ds)
expected = [[1., 1., 0., 0., 0.], [0., 1., 1., 1., 1.],
[0., 1., 0., 0., 1.], [0., 1., 1., 1., 1.]]
self.assertEquals(ds2lofl(result), expected)
def testPainter(self): line = LinePainter(0, 0 , 0, 1) ds = line.paint(DatasetFactory.zeros(2, 3)) expected = [ [1., 1., 0.], [0., 0., 0.]] self.assertEquals(ds2lofl(ds), expected)
def renderShapes(self, targetDataset):
# Make a blank data set
# beware bizarre Jython bug where it cannot call correct method with only a shape argument
image = DatasetFactory.zeros(targetDataset.getShape(), Dataset.FLOAT64)
for shapeDict in self.shapesToPaint.values():
for shape in shapeDict.values():
image = shape.paint(image)
return image
def __init__(self, shape=None, dtype=None, buffer=None, copy=False):
# check what buffer is and convert if necessary
if buffer is not None:
self.__dataset = __cvt_jobj(_jinput(buffer), dtype=dtype, copy=copy, force=True)
if shape is not None:
self.__dataset.setShape(asIterable(shape))
else:
dtype = _translatenativetype(dtype)
self.__dataset = _df.zeros(dtype.elements, asIterable(shape), dtype.value)
def testPainter(self): rect = RectPainter(1, 1, 3, 4) ds = rect.paint(DatasetFactory.zeros(4, 5)) expected = [ [0., 0., 0., 0., 0.], [0., 1., 1., 1., 1.], [0., 1., 0., 0., 1.], [0., 1., 1., 1., 1.]] self.assertEquals(ds2lofl(ds), expected)
def testRenderShapesOntoDataset(self):
self.testAddShape()
ds = DatasetFactory.zeros(4, 5)
ds.set(10, (2, 2))
result = self.renderer.renderShapesOntoDataset(ds)
expected = [[-10.0, -10.0, 0.0, 0.0, 0.0],
[0.0, -10.0, -10.0, -10.0, -10.0],
[0.0, -10.0, 10.0, 0.0, -10.0],
[0.0, -10.0, -10.0, -10.0, -10.0]]
self.assertEquals(ds2lofl(result), expected)
def testRenderShapes(self): self.testAddShape() ds = DatasetFactory.zeros(4, 5) result = self.renderer.renderShapes(ds) expected = [ [1., 1., 0., 0., 0.], [0., 1., 1., 1., 1.], [0., 1., 0., 0., 1.], [0., 1., 1., 1., 1.]] self.assertEquals(ds2lofl(result), expected)
def testRenderShapesOntoDataset(self): self.testAddShape() ds = DatasetFactory.zeros(4, 5) ds.set(10, (2, 2)) result = self.renderer.renderShapesOntoDataset(ds) expected = [ [-10.0, -10.0, 0.0, 0.0, 0.0], [0.0, -10.0, -10.0, -10.0, -10.0], [0.0, -10.0, 10.0, 0.0, -10.0], [0.0, -10.0, -10.0, -10.0, -10.0]] self.assertEquals(ds2lofl(result), expected)
def zeros(shape, dtype=float64, elements=None):
'''Create a dataset filled with 0'''
dtype = _translatenativetype(dtype)
if elements is not None:
if type(dtype) is _types.FunctionType:
dtype = dtype(elements)
else:
dtype.elements = elements
elif type(dtype) is _types.FunctionType:
raise ValueError, "Given data-type is a function and needs elements defining"
return _df.zeros(dtype.elements, asIterable(shape), dtype.value)
def collectData(self): #rawdata = self.pvs['DATA'].cagetArray() if self.determine_data_pv_based_on_zoom: zoom_ordinal = int(float(self.pvs['ZOOM'].caget())) rawdata = self.pv_data[zoom_ordinal].get() else: rawdata = self.pv_data[0].get() data = map(unsign2, rawdata ) self.ds = DatasetFactory.zeros(int(float(self.pvs['HEIGHT'].caget())), int(float(self.pvs['WIDTH'].caget())), data) self.last_image_number += 1 self.last_filename = self._generateCurrentFilename() if self.filepath is not None: self.saveImage(self.last_filename);
def full(shape, fill_value, dtype=None, elements=None):
'''Create a dataset filled with fill_value'''
dtype = _translatenativetype(dtype)
if dtype is None:
dtype = _getdtypefromobj(fill_value)
if elements is not None:
if type(dtype) is _types.FunctionType:
dtype = dtype(elements)
else:
dtype.elements = elements
elif type(dtype) is _types.FunctionType:
raise ValueError, "Given data-type is a function and needs elements defining"
return _df.zeros(dtype.elements, asIterable(shape), dtype.value).fill(fill_value)
def plotWholeData(self, numberOfPoints):
arrayEnergyPGM = self.energyPGM.cagetArrayDouble()
arrayEnergyIDGAP = self.energyIDGAP.cagetArrayDouble()
arrayChannel01 = self.channel01.cagetArrayDouble()
arrayChannel02 = self.channel02.cagetArrayDouble()
arrayChannel03 = self.channel03.cagetArrayDouble()
arrayChannel04 = self.channel04.cagetArrayDouble()
dataSetPGM = DatasetFactory.zeros(numberOfPoints)
for i in range(numberOfPoints):
dataSetPGM.set(arrayEnergyPGM[i], i)
dvp = Plotter()
dvp.plotOver("Data Vector", dataSetPGM.getIndexDataSet(), dataSetPGM)
def collectData(self):
#rawdata = self.pvs['DATA'].cagetArray()
if self.determine_data_pv_based_on_zoom:
zoom_ordinal = int(float(self.pvs['ZOOM'].caget()))
rawdata = self.pv_data[zoom_ordinal].get()
else:
rawdata = self.pv_data[0].get()
data = map(unsign2, rawdata)
self.ds = DatasetFactory.zeros(int(float(self.pvs['HEIGHT'].caget())),
int(float(self.pvs['WIDTH'].caget())),
data)
self.last_image_number += 1
self.last_filename = self._generateCurrentFilename()
if self.filepath is not None:
self.saveImage(self.last_filename)
def plotData(self):
newHead = self.getDataNumbers()
if self.arrayHead >= newHead:
print "No new data added for plotting"
return
self.arrayHead = newHead
#to get new data
arrayEnergyPGM = self.energyPGM.cagetArrayDouble()
arrayEnergyIDGAP = self.energyIDGAP.cagetArrayDouble()
arrayChannel01 = self.channel01.cagetArrayDouble()
arrayChannel02 = self.channel02.cagetArrayDouble()
arrayChannel03 = self.channel03.cagetArrayDouble()
arrayChannel04 = self.channel04.cagetArrayDouble()
dataSetEnergyPGM = DatasetFactory.zeros(newHead)
dataSetEnergyPGM.setName("PGM Energy")
dataSetEnergyIDGAP = DatasetFactory.zeros(newHead)
dataSetEnergyIDGAP.setName("ID Gap Energy")
dataSetChannel01 = DatasetFactory.zeros(newHead)
dataSetChannel01.setName("Channel 1")
dataSetChannel02 = DatasetFactory.zeros(newHead)
dataSetChannel02.setName("Channel 2")
dataSetChannel03 = DatasetFactory.zeros(newHead)
dataSetChannel03.setName("Channel 3")
dataSetChannel04 = DatasetFactory.zeros(newHead)
dataSetChannel04.setName("Channel 4")
for i in range(0, newHead):
#print i, arrayEnergyPGM[i], arrayEnergyIDGAP[i], arrayChannel01[i], arrayChannel02[i], arrayChannel03[i], arrayChannel04[i];
dataSetEnergyPGM[i] = arrayEnergyPGM[i]
dataSetEnergyIDGAP[i] = arrayEnergyIDGAP[i]
dataSetChannel01[i] = arrayChannel01[i]
dataSetChannel02[i] = arrayChannel02[i]
dataSetChannel03[i] = arrayChannel03[i]
dataSetChannel04[i] = arrayChannel04[i]
#print i, arrayEnergyPGM[i], arrayEnergyIDGAP[i], arrayChannel01[i], arrayChannel02[i], arrayChannel03[i], arrayChannel04[i];
dvp = Plotter()
indexDataSet = dataSetEnergyPGM.getIndexDataSet()
#dvp.plot("Data Vector", indexDataSet, [dataSetChannel01, dataSetChannel02, dataSetChannel03, dataSetChannel04]);
dvp.plot("Data Vector", dataSetEnergyPGM, [
dataSetChannel01, dataSetChannel02, dataSetChannel03,
dataSetChannel04
])
def zeros_like(a, dtype=None):
z = _df.zeros(a)
if dtype is not None:
dtype = _translatenativetype(dtype)
z = z.cast(dtype.value)
return z
def testPainter(self):
rect = RectPainter(1, 1, 3, 4)
ds = rect.paint(DatasetFactory.zeros(4, 5))
expected = [[0., 0., 0., 0., 0.], [0., 1., 1., 1., 1.],
[0., 1., 0., 0., 1.], [0., 1., 1., 1., 1.]]
self.assertEquals(ds2lofl(ds), expected)
def testPainter(self):
line = LinePainter(0, 0, 0, 1)
ds = line.paint(DatasetFactory.zeros(2, 3))
expected = [[1., 1., 0.], [0., 0., 0.]]
self.assertEquals(ds2lofl(ds), expected)
