class JLissajousGenerator(JavaIteratorWrapper): """ Create a LissajousGenerator and wrap the points into java Point objects """ def __init__(self, names, units, box, num_lobes, num_points): super(JLissajousGenerator, self).__init__() self.names = names self.generator = LissajousGenerator(names, units, box, num_lobes, num_points) logging.debug(self.generator.to_dict()) def _iterator(self): x_name = self.names[0] y_name = self.names[1] for point in self.generator.iterator(): index = point.indexes[0] x_position = point.positions[x_name] y_position = point.positions[y_name] java_point = Point(x_name, index, x_position, y_name, index, y_position, False) # Set is2D=False yield java_point
def test_to_dict(self): g = LissajousGenerator(["x", "y"], ["mm", "cm"], [0., 0.], [1., 1.], 1) expected_dict = dict() expected_dict[ 'typeid'] = "scanpointgenerator:generator/LissajousGenerator:1.0" expected_dict['axes'] = ["x", "y"] expected_dict['units'] = ["mm", "cm"] expected_dict['centre'] = [0., 0.] expected_dict['span'] = [1., 1.] expected_dict['lobes'] = 1 expected_dict['size'] = 250 expected_dict['alternate'] = False d = g.to_dict() self.assertEqual(expected_dict, d)
def test_to_dict(self): bounding_box = dict(centre=[0.0, 0.0], width=1.0, height=1.0) g = LissajousGenerator(["x", "y"], ["mm", "cm"], [0., 0.], [1., 1.], 1) expected_dict = dict() box = dict() expected_dict[ 'typeid'] = "scanpointgenerator:generator/LissajousGenerator:1.0" expected_dict['axes'] = ["x", "y"] expected_dict['units'] = ["mm", "cm"] expected_dict['centre'] = [0., 0.] expected_dict['span'] = [1., 1.] expected_dict['lobes'] = 1 expected_dict['size'] = 250 d = g.to_dict() self.assertEqual(expected_dict, d)