def test_to_dict(self):
points = [0., 0., 1., 2., 0.5, 2.7, 1.3, 4.0]
expected = {}
expected["typeid"] = "scanpointgenerator:generator/ArrayGenerator:1.0"
expected["axis"] = "x"
expected["units"] = "cm"
expected["points"] = points
expected["alternate"] = True
g = ArrayGenerator("x", "cm", points, True)
self.assertEqual(expected, g.to_dict())
class JArrayGenerator(JavaIteratorWrapper):
"""
Create an ArrayGenerator and wrap the points into java Scalar objects
"""
def __init__(self, name, units, points):
super(JArrayGenerator, self).__init__()
points = points.tolist() # Convert from array to list
self.name = name
self.generator = ArrayGenerator(name, units, points)
logging.debug(self.generator.to_dict())
def _iterator(self):
for point in self.generator.iterator():
name = self.name[0]
index = point.indexes[0]
position = point.positions[name]
java_point = Scalar(name, index, position)
yield java_point
class JArrayGenerator(JavaIteratorWrapper):
"""
Create an ArrayGenerator and wrap the points into java Scalar objects
"""
def __init__(self, name, units, points):
super(JArrayGenerator, self).__init__()
points = points.tolist() # Convert from array to list
self.name = name
self.generator = ArrayGenerator(name, units, points)
logging.debug(self.generator.to_dict())
def _iterator(self):
for point in self.generator.iterator():
name = self.name[0]
index = point.indexes[0]
position = point.positions[name]
java_point = Scalar(name, index, position)
yield java_point
