Public Member Functions
	__init__ (self, animate=False)

	clone (self)

	styling_to_attribute_map (self, **styling)

	draw_superlayer_boundaries (self, **styling)

	draw_interaction_point (self)

	draw_outer_cdc_wall (self, **styling)

	draw_inner_cdc_wall (self, **styling)

	draw_iterable (self, iterable, **styling)

	draw_storevector (self, storeobj_name, **styling)

	draw_storearray (self, storearray_name, **styling)

	draw (self, obj, **styling)

	saveSVGFile (self, svgFileName='display.svg')

	savePNGFile (self, pngFileName='display.png')

Static Public Member Functions
	unpack_attributes (styling, i_obj=0, obj=None)

Public Attributes
	animate = animate
	Switch to indicating if an animated SVG should be generated.

	eventdata_plotter = Belle2.TrackFindingCDC.EventDataPlotter(self.animate)
	Display the image using the event-data plotter.

Detailed Description

Helper class to generated the svg image from the various tracking objects.
This is a tiny wrapper around the TrackFindingCDC.EventDataPlotter.

Definition at line 21 of file __init__.py.

Constructor & Destructor Documentation

◆ init()

__init__	(		self,
			animate = False )

Constructor method.
@param animate switch indicating if an animated SVG should be generated

Definition at line 28 of file __init__.py.

    def __init__(self, animate=False):
        """
        Constructor method.
        @param animate switch indicating if an animated SVG should be generated
        """
        top = -112
        left = -112
        right = 112
        bottom = 112
 
        default_bound = Belle2.TrackFindingCDC.BoundingBox(left, bottom, right, top)
        default_width = 1120
        default_height = 1120
 
        
        self.animate = animate
        
        self.eventdata_plotter = Belle2.TrackFindingCDC.EventDataPlotter(self.animate)
 
        self.eventdata_plotter.setBoundingBox(default_bound)
        self.eventdata_plotter.setCanvasHeight(default_height)
        self.eventdata_plotter.setCanvasWidth(default_width)
 

Member Function Documentation

◆ clone()

clone ( self )

Make a copy of the current status of the plotter.

Definition at line 51 of file __init__.py.

    def clone(self):
        """
        Make a copy of the current status of the plotter.
        """
        cloned_plotter = CDCSVGPlotter(self.animate)
        cloned_plotter.eventdata_plotter = Belle2.TrackFindingCDC.EventDataPlotter(self.eventdata_plotter)
        return cloned_plotter
 

◆ draw()

draw	(		self,
			obj,
		**	styling )

Draw an object with the specified styling

Definition at line 173 of file __init__.py.

    def draw(self, obj, **styling):
        """
        Draw an object with the specified styling
        """
        attribute_map = self.styling_to_attribute_map(**styling)
        self.eventdata_plotter.draw(obj, attribute_map)
 

◆ draw_inner_cdc_wall()

draw_inner_cdc_wall	(		self,
		**	styling )

Draw the CDC inner wall

Definition at line 90 of file __init__.py.

    def draw_inner_cdc_wall(self, **styling):
        """
        Draw the CDC inner wall
        """
        attribute_map = self.styling_to_attribute_map(**styling)
        self.eventdata_plotter.drawInnerCDCWall(attribute_map)
 

◆ draw_interaction_point()

draw_interaction_point ( self )

Draw the interaction point

Definition at line 77 of file __init__.py.

    def draw_interaction_point(self):
        """
        Draw the interaction point
        """
        self.eventdata_plotter.drawInteractionPoint()
 

◆ draw_iterable()

draw_iterable	(		self,
			iterable,
		**	styling )

Draw one or more items with the specified styling

Definition at line 112 of file __init__.py.

    def draw_iterable(self, iterable, **styling):
        """
        Draw one or more items with the specified styling
        """
        draw = self.draw
        unpack_attributes = self.unpack_attributes
        for (i_obj, obj) in enumerate(iterable):
            obj_styling = unpack_attributes(styling, i_obj, obj)
            draw(obj, **obj_styling)
 

◆ draw_outer_cdc_wall()

draw_outer_cdc_wall	(		self,
		**	styling )

Draw the CDC outer wall

Definition at line 83 of file __init__.py.

    def draw_outer_cdc_wall(self, **styling):
        """
        Draw the CDC outer wall
        """
        attribute_map = self.styling_to_attribute_map(**styling)
        self.eventdata_plotter.drawOuterCDCWall(attribute_map)
 

◆ draw_storearray()

draw_storearray	(		self,
			storearray_name,
		**	styling )

Draw information from a StoreArray with the specified styling

Definition at line 150 of file __init__.py.

    def draw_storearray(self, storearray_name, **styling):
        """
        Draw information from a StoreArray with the specified styling
        """
        print('Drawing StoreArray:', storearray_name, end=' ')
        print()
        storearray = Belle2.PyStoreArray(storearray_name)
        if storearray:
            print('with', storearray.getEntries(), 'entries')
            print('Attributes are')
            for (key, value) in list(styling.items()):
                print(str(key), ':', str(value))
 
            self.draw_iterable(storearray, **styling)
 
        else:
            print("### not present in the DataStore")
            print("Current content of the DataStore")
            print("StoreArrays:")
            Belle2.PyStoreArray.printList()
            print("StoreObjPtr:")
            Belle2.PyStoreObj.printList()
 

◆ draw_storevector()

draw_storevector	(		self,
			storeobj_name,
		**	styling )

Draw information in a vector from the DataStore with the specified styling

Definition at line 122 of file __init__.py.

    def draw_storevector(self, storeobj_name, **styling):
        """
        Draw information in a vector from the DataStore with the specified styling
        """
        print('Drawing vector from DataStore:', storeobj_name, end=' ')
        print()
        pystoreobj = Belle2.PyStoreObj(storeobj_name)
 
        if pystoreobj:
            # Wrapper around std::vector like
            wrapped_vector = pystoreobj.obj()
            vector = wrapped_vector.get()
 
            print('with', vector.size(), 'entries')
            print('Attributes are')
            for (key, value) in list(styling.items()):
                print(str(key), ':', str(value))
 
            self.draw_iterable(vector, **styling)
 
        else:
            print("### not present in the DataStore")
            print("Current content of the DataStore")
            print("StoreArrays:")
            Belle2.PyStoreArray.printList()
            print("StoreObjPtr:")
            Belle2.PyStoreObj.printList()
 

◆ draw_superlayer_boundaries()

draw_superlayer_boundaries	(		self,
		**	styling )

Draw the CDC superlayer boundaries

Definition at line 70 of file __init__.py.

    def draw_superlayer_boundaries(self, **styling):
        """
        Draw the CDC superlayer boundaries
        """
        attribute_map = self.styling_to_attribute_map(**styling)
        self.eventdata_plotter.drawSuperLayerBoundaries(attribute_map)
 

◆ savePNGFile()

savePNGFile	(		self,
			pngFileName = 'display.png' )

Save the current dom object representation to disk as a png.

Definition at line 201 of file __init__.py.

    def savePNGFile(self, pngFileName='display.png'):
        """
        Save the current dom object representation to disk as a png.
        """
 
        import cairosvg
        import os
 
        temp_file_name = "tmp.svg"
        eventdata_plotter = self.eventdata_plotter
 
        boundingBox = eventdata_plotter.getBoundingBox()
 
        height = boundingBox.getHeight()
        width = boundingBox.getWidth()
 
        totalPoints = 1120 * 1120
        svgHeight = round(math.sqrt(totalPoints * float(height) / width))
        svgWidth = round(math.sqrt(totalPoints * float(width) / height))
 
        eventdata_plotter.setCanvasHeight(svgHeight)
        eventdata_plotter.setCanvasWidth(svgWidth)
 
        eventdata_plotter.save(temp_file_name)
 
        with open(temp_file_name) as temp_file:
            with open(pngFileName, "w") as output_file:
                svg_code = temp_file.read()
                cairosvg.svg2png(bytestring=svg_code, write_to=output_file)
 
        os.remove(temp_file_name)

◆ saveSVGFile()

saveSVGFile	(		self,
			svgFileName = 'display.svg' )

Save the current dom object representation to disk.

Definition at line 180 of file __init__.py.

    def saveSVGFile(self, svgFileName='display.svg'):
        """
        Save the current dom object representation to disk.
        """
 
        eventdata_plotter = self.eventdata_plotter
 
        boundingBox = eventdata_plotter.getBoundingBox()
 
        height = boundingBox.getHeight()
        width = boundingBox.getWidth()
 
        totalPoints = 1120 * 1120
        svgHeight = round(math.sqrt(totalPoints * float(height) / width))
        svgWidth = round(math.sqrt(totalPoints * float(width) / height))
 
        eventdata_plotter.setCanvasHeight(svgHeight)
        eventdata_plotter.setCanvasWidth(svgWidth)
 
        return eventdata_plotter.save(svgFileName)
 

◆ styling_to_attribute_map()

styling_to_attribute_map	(		self,
		**	styling )

Transfer the styling information to the attribute map

Definition at line 59 of file __init__.py.

    def styling_to_attribute_map(self, **styling):
        """
        Transfer the styling information to the attribute map
        """
        attribute_map = Belle2.TrackFindingCDC.PrimitivePlotter.AttributeMap()
 
        for key, value in list(styling.items()):
            attribute_map[str(key)] = str(value)
 
        return attribute_map
 

◆ unpack_attributes()

unpack_attributes	(	styling,
		i_obj = 0,
		obj = None )

static

Mapping function to unpack the attributes from the attribute maps. Mechanism and interface inspired by d3.js

Definition at line 98 of file __init__.py.

    def unpack_attributes(styling, i_obj=0, obj=None):
        """Mapping function to unpack the attributes from the attribute maps. Mechanism and interface inspired by d3.js"""
        result = {}
        for (key, value) in list(styling.items()):
            if isinstance(value, collections.abc.Callable):
                result[key] = value(i_obj, obj)
            elif isinstance(value, str):
                result[key] = value
            elif hasattr(value, '__getitem__'):
                result[key] = value[i_obj % len(value)]
            else:
                result[key] = value
        return result
 

Member Data Documentation

◆ animate

animate = animate

Switch to indicating if an animated SVG should be generated.

Definition at line 43 of file __init__.py.

◆ eventdata_plotter

eventdata_plotter = Belle2.TrackFindingCDC.EventDataPlotter(self.animate)

Display the image using the event-data plotter.

Definition at line 45 of file __init__.py.

The documentation for this class was generated from the following file:

tracking/trackFindingCDC/scripts/trackfindingcdc/cdcdisplay/svgdrawing/__init__.py

Public Member Functions

Static Public Member Functions

Public Attributes