Inheritance diagram for CDCSVGDisplayModule:

Public Member Functions
	__init__ (self, output_folder=tempfile.gettempdir(), interactive=True)

	drawoptions (self)

	all_drawoptions (self)

	draw_wires (self)

	draw_wires (self, draw_wires)

	initialize (self)

	beginRun (self)

	event (self)

	endRun (self)

	terminate (self)

	new_output_basename (self)

	new_output_filename (self)

Public Attributes
	interactive = interactive
	Switch if the module shall show the event to the user and wait to continue or just generate the images.

	output_folder = output_folder
	Folder the images shall be saved to.

bool	animate = False
	Switch to make an animated event display by means of animated SVG.

bool	forward_fade = False
	Switch to make the color of segments and tracks fade out in the forward direction.

bool	use_cpp = True
	Flag to use cpp.

bool	use_python = False
	Flag to use python.

bool	draw_interaction_point = True
	Switch to draw the interaction point.

bool	draw_superlayer_boundaries = False
	Switch to draw the superlayer boundaries.

bool	draw_walls = False
	Switch to draw the superlayer boundaries.

bool	draw_hits = True
	Switch to draw the CDCHits.

bool	draw_takenflag = False
	Switch to draw the CDCHits colored by the associated CDCWireHit taken flag.

bool	draw_mcparticle_id = False
	Switch to draw the MCParticle::getArrayIndex property.

bool	draw_mcparticle_pdgcode = False
	Switch to draw the MCParticle::getPDGCode property.

bool	draw_mcparticle_primary = False
	Switch to draw the MCParticle::hasStatus(c_PrimaryParticle) property.

bool	draw_mcparticle_trajectories = False
	Switch to draw the ideal trajectory of the MCParticle.

bool	draw_simhits = False
	Switch to draw the CDCSimHits with momentum information.

bool	draw_simhit_tof = False
	Switch to draw the CDCSimHits color coded by their time of flight.

bool	draw_simhit_posflag = False
	Switch to draw the CDCSimHits color coded by their getPosFlag() property.

bool	draw_simhit_pdgcode = False
	Switch to draw the CDCSimHits color coded by their getPDGCode() property.

bool	draw_simhit_bkgtag = False
	Switch to draw the CDCSimHits color coded by their getBackgroundTag() property.

bool	draw_simhit_isbkg = False
	Switch to draw the CDCSimHits red for getBackgroundTag() != bg_none.

bool	draw_nloops = False
	Switch to draw the CDCHit colored by the number of loops passed.

bool	draw_connect_tof = False
	Switch to draw the CDCSimHits connected in the order of their getFlightTime for each Monte Carlo particle.

bool	draw_rlinfo = False
	Switch to draw the CDCSimHits color coded by their local right left passage information.

bool	draw_reassigned = False
	Switch to draw the CDCSimHits color coded by their reassignement information to a different MCParticle.

bool	draw_mcsegments = False
	Switch to draw the Monte Carlo segments as generated in CDCMCTrackStore.

bool	draw_mcaxialsegmentpairs = False
	Switch to draw the axial to axial segment pairs from Monte Carlo truth.

bool	draw_mcsegmentpairs = False
	Switch to draw the axial to stereo segment pairs from Monte Carlo truth.

bool	draw_mcsegmenttriples = False
	Switch to draw the segment triples from Monte Carlo truth.

bool	draw_superclusters = False
	Switch to draw the clusters generated by the finder.

bool	draw_clusters = False
	Switch to draw the clusters generated by the finder.

bool	draw_segments = False
	Switch to draw the segments generated by the finder.

bool	draw_segment_trajectories = False
	Switch to draw the trajectories fitted to the segments generated by the finder.

bool	draw_tangentsegments = False
	Switch to draw the tangent segments generated by the finder.

bool	draw_segment_mctrackids = False
	Switch to draw the segments generated by the finder colored with the Monte Carlo track id.

bool	draw_segment_fbinfos = False
	Switch to draw the segments generated by the finder colored by the coalignment information (forward, backward, undetermined)

bool	draw_segment_firstInTrackIds = False
	Switch to draw the segments generated by the finder colored by the first in track hit id.

bool	draw_segment_lastInTrackIds = False
	Switch to draw the segments generated by the finder colored by the second in track hit id.

bool	draw_segment_firstNPassedSuperLayers = False
	Switch to draw the segments generated by the finder colored by the number of passed superlayers associated to the first hit in the segment.

bool	draw_segment_lastNPassedSuperLayers = False
	Switch to draw the segments generated by the finder colored by the number of passed superlayers associated to the last hit in the segment.

bool	draw_axialsegmentpairs = False
	Switch to draw the axial stereo segment pairs generated by the finder.

bool	draw_segmentpairs = False
	Switch to draw the axial stereo segment pairs generated by the finder.

bool	draw_segmenttriples = False
	Switch to draw the segment triples generated by the finder.

bool	draw_segmenttriple_trajectories = False
	Switch to draw the trajectories fitted to the segment triples generated by the finder.

bool	draw_tracks = False
	Switch to draw the tracks generated by the finder.

bool	draw_track_trajectories = False
	Switch to draw the trajectories of the tracks generated by the finder.

bool	draw_recotracks = False
	Draw the output RecoTracks.

bool	draw_mcrecotracks = False
	Draw the MC reference RecoTracks.

bool	draw_recotrack_matching = False
	Draw the output RecoTracks pattern recognition matching status.

bool	draw_mcrecotrack_matching = False
	Draw the MC reference RecoTracks pattern recognition matching status.

bool	draw_recotrack_seed_trajectories = False
	Draw the output track seed trajectories.

bool	draw_recotrack_fit_trajectories = False
	Draw the output trackpoint trajectories.

bool	draw_wrong_rl_infos_in_segments = False
	Draw a red cdc hit of the rl info of the segment reco hits is wrong, else a green one.

bool	draw_wrong_rl_infos_in_tracks = False
	Draw a red cdc hit of the rl info of the track reco hits is wrong, else a green one.

str	cdc_hits_store_array_name = "CDCHits"
	Name of the CDC Hits store array.

str	reco_tracks_store_array_name = "RecoTracks"
	Name of the RecoTracks store array.

str	mc_reco_tracks_store_array_name = "MCRecoTracks"
	Name of the Monte Carlo reference RecoTracks store array.

str	cdc_wire_hit_cluster_store_obj_name = "CDCWireHitClusterVector"
	Name of the CDC Wire Hit Clusters.

str	cdc_segment_vector_store_obj_name = 'CDCSegment2DVector'
	Name of the CDC Reco Segment Vector.

int	file_number = 0
	Current file's number (used for making output filename)

str	filename_prefix = "CDCDisplay"
	Filename prefix.

bool	use_time_in_filename = False
	Use time instead of prefix in filename.

	draw_wires = svgdrawing.CDCSVGPlotter(animate=self.animate)

	prefilled_cppplotter = cppplotter
	prefilled default is to use the C++ plotter

	prefilled_plotter = plotter
	prefilled default is to use the python plotter

Protected Attributes
bool	_draw_wires = True
	Switch to draw the wires.

Detailed Description

Personal two dimensional event display based on scalable vector graphics

Definition at line 27 of file __init__.py.

Constructor & Destructor Documentation

◆ init()

__init__	(	self,
		output_folder = tempfile.gettempdir(),
		interactive = True )

Constructor method

Parameters
----------
output_folder : str
    Target folder for the output
interactive : bool, optional
    Switch to display each event to the user and ask to continue after each event
    Defaults to True

Definition at line 33 of file __init__.py.

    def __init__(self, output_folder=tempfile.gettempdir(), interactive=True):
        """
        Constructor method
 
        Parameters
        ----------
        output_folder : str
            Target folder for the output
        interactive : bool, optional
            Switch to display each event to the user and ask to continue after each event
            Defaults to True
        """
 
        super().__init__()
        # call constructor of base class, required!
 
        self.interactive = interactive
 
        self.output_folder = output_folder
 
        # List of drawing options
 
        # Animate the display by uncovering the drawn objects in order of their time of flight
        # This can be seen in most standard browsers. Note however that you should switch of
        # the wires in this case to reduce the rendering load.
        
        self.animate = False
 
        self.forward_fade = False
 
        # The following options can be used independent of the track finder
        # to view Monte Carlo information after the simulation is done
 
        self.use_cpp = True
 
        self.use_python = False
 
        self._draw_wires = True
        # This is wrapped by an extra property,
        # because it should always be deactivated in case the scene is animated
 
        self.draw_interaction_point = True
 
        self.draw_superlayer_boundaries = False
 
        self.draw_walls = False
 
        self.draw_hits = True
 
        self.draw_takenflag = False
 
        self.draw_mcparticle_id = False
 
        self.draw_mcparticle_pdgcode = False
 
        self.draw_mcparticle_primary = False
 
        self.draw_mcparticle_trajectories = False
 
        self.draw_simhits = False
 
        self.draw_simhit_tof = False
 
        self.draw_simhit_posflag = False
 
        self.draw_simhit_pdgcode = False
 
        self.draw_simhit_bkgtag = False
 
        self.draw_simhit_isbkg = False
 
        self.draw_nloops = False
 
        self.draw_connect_tof = False
 
        self.draw_rlinfo = False
 
        self.draw_reassigned = False
 
        self.draw_mcsegments = False
 
        self.draw_mcaxialsegmentpairs = False
 
        self.draw_mcsegmentpairs = False  # <- not implemented at the moment
 
        self.draw_mcsegmenttriples = False
 
        # Those are only available if the local track finder is in the module chain
        # and specific compile time flags enable to transportation of this data
 
        self.draw_superclusters = False
 
        self.draw_clusters = False
 
        self.draw_segments = False
 
        self.draw_segment_trajectories = False
 
        self.draw_tangentsegments = False
 
        self.draw_segment_mctrackids = False
 
        self.draw_segment_fbinfos = False
 
        self.draw_segment_firstInTrackIds = False
 
        self.draw_segment_lastInTrackIds = False
 
        self.draw_segment_firstNPassedSuperLayers = False
 
        self.draw_segment_lastNPassedSuperLayers = False
 
        self.draw_axialsegmentpairs = False
 
        self.draw_segmentpairs = False
 
        self.draw_segmenttriples = False
 
        self.draw_segmenttriple_trajectories = False
 
        self.draw_tracks = False
 
        self.draw_track_trajectories = False
 
        # Those are only available, if any track finder is in the module chain (not tested for others than the local track finder)
 
        self.draw_recotracks = False
 
        self.draw_mcrecotracks = False
 
        self.draw_recotrack_matching = False
 
        self.draw_mcrecotrack_matching = False
 
        self.draw_recotrack_seed_trajectories = False
 
        self.draw_recotrack_fit_trajectories = False
 
        self.draw_wrong_rl_infos_in_segments = False
 
        self.draw_wrong_rl_infos_in_tracks = False
 
        self.cdc_hits_store_array_name = "CDCHits"
 
        self.reco_tracks_store_array_name = "RecoTracks"
 
        self.mc_reco_tracks_store_array_name = "MCRecoTracks"
 
        self.cdc_wire_hit_cluster_store_obj_name = "CDCWireHitClusterVector"
 
        self.cdc_segment_vector_store_obj_name = 'CDCSegment2DVector'
 
        self.file_number = 0
 
        self.filename_prefix = "CDCDisplay"
 
        self.use_time_in_filename = False

Member Function Documentation

◆ all_drawoptions()

all_drawoptions ( self )

Property that collects the all names of the draw options to a list.
Note that some draw options only make sense after running the CDC
cellular automaton track finder.
@return list of strings naming the different switches that can be activated.

Definition at line 328 of file __init__.py.

    def all_drawoptions(self):
        """
        Property that collects the all names of the draw options to a list.
        Note that some draw options only make sense after running the CDC
        cellular automaton track finder.
        @return list of strings naming the different switches that can be activated.
        """
        result = self.drawoptions
 
        # Add all attributes that start with draw
        draw_options = [option for option in self.__dict__
                        if option.startswith('draw_')]
        _draw_options = [option[1:] for option in self.__dict__
                         if option.startswith('_draw_')]
 
        result.extend(draw_options)
        result.extend(_draw_options)
 
        return set(result)
 

◆ beginRun()

beginRun ( self )

Begin run method of the module. Empty here.

Definition at line 411 of file __init__.py.

    def beginRun(self):
        """
        Begin run method of the module. Empty here.
        """
 
        pass
 

◆ draw_wires() [1/2]

draw_wires ( self )

Getter for the draw option that indicates if all wires shall be drawn.
Since this has some performance impact in animated events the wires
are prevented from being drawn in this case.

Definition at line 349 of file __init__.py.

    def draw_wires(self):
        """
        Getter for the draw option that indicates if all wires shall be drawn.
        Since this has some performance impact in animated events the wires
        are prevented from being drawn in this case.
        """
 
        return self._draw_wires and not self.animate
 

◆ draw_wires() [2/2]

draw_wires	(		self,
			draw_wires )

Setter for the option to draw all wires.

Definition at line 359 of file __init__.py.

    def draw_wires(self, draw_wires):
        """
        Setter for the option to draw all wires.
        """
 
        self._draw_wires = draw_wires
 

◆ drawoptions()

drawoptions ( self )

Property that collects the various names of the draw options to a list
that are not related to the CDC cellular automaton track finder.
@return list of strings naming the different switches that can be activated.

Definition at line 263 of file __init__.py.

    def drawoptions(self):
        """
        Property that collects the various names of the draw options to a list
        that are not related to the CDC cellular automaton track finder.
        @return list of strings naming the different switches that can be activated.
        """
        result = [
            'animate',
            'forward_fade',
            'draw_superlayer_boundaries',
            'draw_walls',
            'draw_interaction_point',
            'draw_mcparticle_id',
            'draw_mcparticle_pdgcode',
            'draw_mcparticle_primary',
            'draw_mcparticle_trajectories',
            'draw_mcsegments',
            'draw_simhits',
            'draw_simhit_tof',
            'draw_simhit_posflag',
            'draw_simhit_pdgcode',
            'draw_simhit_bkgtag',
            'draw_simhit_isbkg',
            'draw_nloops',
            'draw_connect_tof',
            'draw_rlinfo',
            'draw_reassigned',
            'draw_recotracks',
            'draw_mcrecotracks',
            'draw_recotrack_matching',
            'draw_mcrecotrack_matching',
            'draw_recotrack_seed_trajectories',
            'draw_recotrack_fit_trajectories',
            # Specialised options to be used in the CDC local tracking context
            # obtain them from the all_drawoptions property
            # 'draw_takenflag',
            # 'draw_superclusters',
            # 'draw_clusters',
            # 'draw_tangentsegments',
            # 'draw_segment_trajectories',
            # 'draw_segments',
            # 'draw_segment_mctrackids',
            # 'draw_segment_fbinfos',
            # 'draw_segment_firstInTrackIds',
            # 'draw_segment_lastInTrackIds',
            # 'draw_segment_firstNPassedSuperLayers',
            # 'draw_segment_lastNPassedSuperLayers',
            # 'draw_segmentpairs'
            # 'draw_mcsegmentpairs',
            # 'draw_axialsegmentpairs'
            # 'draw_mcaxialsegmentpairs',
            # 'draw_segmenttriples',
            # 'draw_segmenttriple_trajectories',
            # 'draw_mcsegmenttriples',
            # 'draw_tracks',
            # 'draw_track_trajectories',
        ]
        for name in result:
            if not hasattr(self, name):
                raise NameError('%s is not a valid draw option. Fix the misspelling.'
                                % name)
 
        return result
 

◆ endRun()

endRun ( self )

endRun method of the module. Empty here.

Definition at line 1049 of file __init__.py.

    def endRun(self):
        """
        endRun method of the module. Empty here.
        """
 
        pass
 

◆ event()

event ( self )

Event method of the module. Draws the event into a new svg file.

Definition at line 418 of file __init__.py.

    def event(self):
        """
        Event method of the module. Draws the event into a new svg file.
        """
 
        # Clone the plotter that contains the wires already
        if self.use_cpp:
            cppplotter = self.prefilled_cppplotter.clone()
        if self.use_python:
            plotter = self.prefilled_plotter.clone()
 
        self.file_number += 1
 
        # if self.draw_wires:
        #    theCDCWireTopology = \
        #        Belle2.TrackFindingCDC.CDCWireTopology.getInstance()
        #    cppplotter.draw(theCDCWireTopology)
 
        # Plotter instance receiving drawable tracking objects.
        # self.plotter = plotter
 
        # Construct additional information from basic Monte Carlo data, if it is available from the DataStore
        # Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
 
        # Skip empty events
        if Belle2.PyStoreArray(self.cdc_hits_store_array_name).getEntries() == 0:
            basf2.B2INFO("Skip empty event")
            return
 
        # ######### CDCHits ##########
        # Draw the raw CDCHits
        if self.draw_hits:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'ZeroDriftLengthColorMap', 'ZeroDriftLengthStrokeWidthMap')
            if self.use_python:
                styleDict = {'stroke': attributemaps.ZeroDriftLengthColorMap(),
                             'stroke-width': attributemaps.ZeroDriftLengthStrokeWidthMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw the CDCHits colored by the taken flag from the CDCWireHit.
        if self.draw_takenflag:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'TakenFlagColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.TakenFlagColorMap(), }
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw mcparticle id
        if self.draw_mcparticle_id:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'MCParticleColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.MCParticleColorMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw monte carlo pdg codes
        if self.draw_mcparticle_pdgcode:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'MCPDGCodeColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.MCPDGCodeColorMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw monte carlo pdg codes
        if self.draw_mcparticle_primary:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'MCPrimaryColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.MCPrimaryColorMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw SimHits
        if self.draw_simhits:
            if self.use_cpp:
                cppplotter.drawSimHits(self.cdc_hits_store_array_name, '', '.2')
            if self.use_python:
                hit_storearray = Belle2.PyStoreArray(self.cdc_hits_store_array_name)
                if hit_storearray:
                    simHits_related_to_hits = [hit.getRelated('CDCSimHits')
                                               for hit in hit_storearray]
                    styleDict = {'stroke-width': '0.2'}
                    plotter.draw_iterable(simHits_related_to_hits, **styleDict)
 
        # Draw RL MC info
        if self.draw_simhit_posflag:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'PosFlagColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.PosFlagColorMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw local RL info
        if self.draw_rlinfo:
            Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'RLColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.RLColorMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw tof info
        if self.draw_simhit_tof:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'TOFColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.TOFColorMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw pdg code of simhits
        if self.draw_simhit_pdgcode:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, "SimHitPDGCodeColorMap", "")
            if self.use_python:
                def color_map(iHit, hit):
                    simHit = hit.getRelated('CDCSimHits')
                    pdgCode = simHit.getPDGCode()
                    color = \
                        attributemaps.MCPDGCodeColorMap.color_by_pdgcode.get(pdgCode,
                                                                             'orange')
                    return color
 
                styleDict = {'stroke': color_map}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw background tag of related simhits
        if self.draw_simhit_bkgtag:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'BackgroundTagColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.BackgroundTagColorMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw background tag != bg_none of related simhits
        if self.draw_simhit_isbkg:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'SimHitIsBkgColorMap', '')
            if self.use_python:
                def color_map(iHit, hit):
                    simHit = hit.getRelated('CDCSimHits')
                    bkgTag = simHit.getBackgroundTag()
                    color = ('gray' if bkgTag else 'red')
                    return color
 
                styleDict = {'stroke': color_map}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw background tag != bg_none of related simhits
        if self.draw_nloops:
            if self.use_cpp:
                Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'NLoops', '')
            if self.use_python:
                print('No Python-function defined')
 
        if self.draw_connect_tof:
            if self.use_cpp:
                cppplotter.drawSimHitsConnectByToF(self.cdc_hits_store_array_name, "black", ".2")
            if self.use_python:
                cdchits_storearray = Belle2.PyStoreArray(self.cdc_hits_store_array_name)
                if cdchits_storearray:
                    simhits_related_to_cdchit = [cdchit.getRelated('CDCSimHits')
                                                 for cdchit in cdchits_storearray]
 
                    # group them by their mcparticle id
                    simhits_by_mcparticle = {}
                    for simhit in simhits_related_to_cdchit:
                        mcparticle = simhit.getRelated('MCParticles')
                        if not mcparticle == None:  # noqa
                            mcTrackId = mcparticle.getArrayIndex()
                            simhits_by_mcparticle.setdefault(mcTrackId, [])
                            simhits_by_mcparticle[mcTrackId].append(simhit)
 
                    for simhits_for_mcparticle in list(simhits_by_mcparticle.values()):
                        simhits_for_mcparticle.sort(key=lambda simhit:
                                                    simhit.getFlightTime())
 
                        nSimHits = len(simhits_for_mcparticle)
                        for iSimHit in range(nSimHits - 1):
                            fromSimHit = simhits_for_mcparticle[iSimHit]
                            toSimHit = simhits_for_mcparticle[iSimHit + 1]
 
                            styleDict = {'stroke-width': '0.2', "stroke": "black"}
 
                            fromHit = fromSimHit.getRelated(self.cdc_hits_store_array_name)
                            toHit = toSimHit.getRelated(self.cdc_hits_store_array_name)
 
                            fromWireHit = Belle2.TrackFindingCDC.CDCWireHit(fromHit)
                            toWireHit = Belle2.TrackFindingCDC.CDCWireHit(toHit)
 
                            fromRLWireHit = Belle2.TrackFindingCDC.CDCRLWireHit(fromWireHit, 0)
                            toRLWireHit = Belle2.TrackFindingCDC.CDCRLWireHit(toWireHit, 0)
 
                            fromDisplacement = Belle2.TrackFindingCDC.Vector3D(fromSimHit.getPosTrack() - fromSimHit.getPosWire())
                            toDisplacement = Belle2.TrackFindingCDC.Vector3D(toSimHit.getPosTrack() - toSimHit.getPosWire())
 
                            fromRecoHit2D = Belle2.TrackFindingCDC.CDCRecoHit2D(fromRLWireHit, fromDisplacement.xy())
                            toRecoHit2D = Belle2.TrackFindingCDC.CDCRecoHit2D(toRLWireHit, toDisplacement.xy())
 
                            tangent = Belle2.TrackFindingCDC.CDCTangent(fromRecoHit2D, toRecoHit2D)
                            plotter.draw(tangent, **styleDict)
 
        # Draw the reassignment information of hits
        if self.draw_reassigned:
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'ReassignedSecondaryMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.ReassignedSecondaryMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw the in track segment id
        if self.draw_mcsegments:
            Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
            if self.use_cpp:
                cppplotter.drawHits(self.cdc_hits_store_array_name, 'MCSegmentIdColorMap', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.MCSegmentIdColorMap()}
                plotter.draw_storearray(self.cdc_hits_store_array_name, **styleDict)
 
        # Draw superclusters
        if self.draw_superclusters:
            if self.use_cpp:
                cppplotter.drawClusters('CDCWireHitSuperClusterVector',
                                        '', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                plotter.draw_storevector('CDCWireHitSuperClusterVector', **styleDict)
 
        # Draw clusters
        if self.draw_clusters:
            if self.use_cpp:
                cppplotter.drawClusters(self.cdc_wire_hit_cluster_store_obj_name,
                                        '', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                plotter.draw_storevector(self.cdc_wire_hit_cluster_store_obj_name, **styleDict)
 
        # ######### CDCSegments2D ##########
        # Draw Segments
        if self.draw_segments:
            if self.use_cpp:
                cppplotter.drawSegments(self.cdc_segment_vector_store_obj_name,
                                        "ListColors", "")
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                plotter.draw_storevector(self.cdc_segment_vector_store_obj_name, **styleDict)
 
        if self.draw_segment_mctrackids:
            Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
            if self.use_cpp:
                cppplotter.drawSegments(self.cdc_segment_vector_store_obj_name,
                                        "SegmentMCTrackIdColorMap", "")
            if self.use_python:
                styleDict = {'stroke': attributemaps.SegmentMCTrackIdColorMap()}
                plotter.draw_storevector(self.cdc_segment_vector_store_obj_name, **styleDict)
 
        if self.draw_segment_fbinfos:
            if self.use_cpp:
                cppplotter.drawSegments(self.cdc_segment_vector_store_obj_name,
                                        "SegmentFBInfoColorMap", "")
            if self.use_python:
                styleDict = {'stroke': attributemaps.SegmentFBInfoColorMap()}
                plotter.draw_storevector(self.cdc_segment_vector_store_obj_name, **styleDict)
 
        if self.draw_segment_firstInTrackIds:
            if self.use_cpp:
                cppplotter.drawSegments(self.cdc_segment_vector_store_obj_name,
                                        "SegmentFirstInTrackIdColorMap", "")
            if self.use_python:
                styleDict = \
                    {'stroke': attributemaps.SegmentFirstInTrackIdColorMap()}
                plotter.draw_storevector(self.cdc_segment_vector_store_obj_name, **styleDict)
 
        if self.draw_segment_lastInTrackIds:
            if self.use_cpp:
                cppplotter.drawSegments(self.cdc_segment_vector_store_obj_name,
                                        "SegmentLastInTrackIdColorMap", "")
            if self.use_python:
                styleDict = \
                    {'stroke': attributemaps.SegmentLastInTrackIdColorMap()}
                plotter.draw_storevector(self.cdc_segment_vector_store_obj_name, **styleDict)
 
        if self.draw_segment_firstNPassedSuperLayers:
            Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
            if self.use_cpp:
                cppplotter.drawSegments(self.cdc_segment_vector_store_obj_name,
                                        "SegmentFirstNPassedSuperLayersColorMap", "")
            if self.use_python:
                styleDict = \
                    {'stroke': attributemaps.SegmentFirstNPassedSuperLayersColorMap()}
                plotter.draw_storevector(self.cdc_segment_vector_store_obj_name, **styleDict)
 
        if self.draw_segment_lastNPassedSuperLayers:
            Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
            if self.use_cpp:
                cppplotter.drawSegments(self.cdc_segment_vector_store_obj_name,
                                        "SegmentLastNPassedSuperLayersColorMap", "")
            if self.use_python:
                styleDict = \
                    {'stroke': attributemaps.SegmentLastNPassedSuperLayersColorMap()}
                plotter.draw_storevector(self.cdc_segment_vector_store_obj_name, **styleDict)
 
        if self.draw_segmentpairs:
            if self.use_cpp:
                cppplotter.drawSegmentPairs('CDCSegmentPairs', 'black', '')
            if self.use_python:
                styleDict = {"stroke": "black"}
                plotter.draw_storearray('CDCSegmentPairs', **styleDict)
 
        # Mimic axial to axial pair selection
        if self.draw_mcaxialsegmentpairs:
            if self.use_cpp:
                cppplotter.drawMCAxialSegmentPairs(self.cdc_segment_vector_store_obj_name, "black", '')
            if self.use_python:
                segment_storevector = Belle2.PyStoreObj(self.cdc_segment_vector_store_obj_name)
                if segment_storevector:
                    segments = segment_storevector.obj().unwrap()
                    axial_segments = [segment for segment in segments
                                      if segment.getStereoType() == 0]
 
                    mc_axial_segment_pair_segment_filter = \
                        Belle2.TrackFindingCDC.MCAxialSegmentPairFilter()
                    axial_segment_pair_relations = \
                        (Belle2.TrackFindingCDC.CDCAxialSegmentPair(startSegment, endSegment)
                         for startSegment in axial_segments
                         for endSegment in axial_segments)
 
                    def is_good_pair(pair):
                        weight = mc_axial_segment_pair_segment_filter(pair)
                        return weight == weight  # not nan
 
                    good_axial_segment_pair_relations = [pair for pair in
                                                         axial_segment_pair_relations if is_good_pair(pair)]
                    styleDict = {"stroke": "black"}
                    plotter.draw_iterable(good_axial_segment_pair_relations, **styleDict)
 
        if self.draw_mcsegmentpairs:
            if self.use_cpp:
                cppplotter.drawMCSegmentPairs(self.cdc_segment_vector_store_obj_name, "black", '')
            if self.use_python:
                print('No Python-function defined')
 
        if self.draw_mcsegmenttriples:
            if self.use_cpp:
                cppplotter.drawMCSegmentTriples(self.cdc_segment_vector_store_obj_name, '', '')
            if self.use_python:
                segment_storevector = Belle2.PyStoreObj(self.cdc_segment_vector_store_obj_name)
                if segment_storevector:
                    segments = segment_storevector.obj().unwrap()
                    axial_segments = [segment for segment in segments
                                      if segment.getStereoType() == 0]
 
                    stereo_segments = [segment for segment in segments
                                       if segment.getStereoType() != 0]
 
                    # Misuse this a bit but still does what we want
                    mc_axial_segment_pair_segment_filter = \
                        Belle2.TrackFindingCDC.MCAxialSegmentPairFilter()
                    mc_segment_lookup = \
                        Belle2.TrackFindingCDC.CDCMCSegment2DLookUp.getInstance()
 
                    segment_triples = \
                        (Belle2.TrackFindingCDC.CDCSegmentTriple(startSegment,
                                                                 middleSegment, endSegment) for startSegment in
                         axial_segments for middleSegment in stereo_segments
                         for endSegment in axial_segments)
 
                    def is_good_triple(triple):
                        start = triple.getStartSegment()
                        middle = triple.getMiddleSegment()
                        end = triple.getEndSegment()
 
                        pairWeight = \
                            mc_axial_segment_pair_segment_filter(triple)
 
                        if not pairWeight == pairWeight:
                            return False
 
                        startToMiddleFBInfo = \
                            mc_segment_lookup.areAlignedInMCTrack(start, middle)
                        if abs(startToMiddleFBInfo) > 1:
                            return False
 
                        middleToEndFBInfo = \
                            mc_segment_lookup.areAlignedInMCTrack(middle, end)
                        if abs(middleToEndFBInfo) > 1:
                            return False
 
                        if startToMiddleFBInfo == middleToEndFBInfo:
                            return True
                        else:
                            return False
 
                    good_segment_triples = [triple for triple in segment_triples
                                            if is_good_triple(triple)]
 
                    styleDict = {"stroke": "black"}
                    plotter.draw_iterable(good_segment_triples, **styleDict)
 
        # Draw Tangent segments
        if self.draw_tangentsegments:
            if self.use_cpp:
                print('No CPP-function defined')
            if self.use_python:
                styleDict = {'stroke-width': '0.2'}
                plotter.draw_storearray('CDCTangentSegments', **styleDict)
 
        # Draw axial axial segment pairs
        if self.draw_axialsegmentpairs:
            if self.use_cpp:
                cppplotter.drawAxialSegmentPairs('CDCAxialSegmentPairVector', '', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                plotter.draw_storevector('CDCAxialSegmentPairVector', **styleDict)
 
        # Draw segment triples
        if self.draw_segmenttriples:
            if self.use_cpp:
                cppplotter.drawSegmentTriples('CDCSegmentTripleVector', '', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                plotter.draw_storevector('CDCSegmentTriples', **styleDict)
 
        # Draw Tracks
        if self.draw_tracks:
            if self.use_cpp:
                cppplotter.drawTracks('CDCTrackVector', '', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                plotter.draw_storevector('CDCTrackVector', **styleDict)
 
        # Wrong RL Info
        if self.draw_wrong_rl_infos_in_tracks:
            if self.use_cpp:
                Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
                cppplotter.drawWrongRLHitsInTracks('CDCTrackVector')
            if self.use_python:
                styleDict = {'stroke': attributemaps.WrongRLColorMap()}
                pystoreobj = Belle2.PyStoreObj('CDCTrackVector')
 
                if pystoreobj:
                    # Wrapper around std::vector like
                    wrapped_vector = pystoreobj.obj()
                    vector = wrapped_vector.get()
 
                    for track in vector:
                        plotter.draw_iterable(list(track.items()), **styleDict)
 
        if self.draw_wrong_rl_infos_in_segments:
            if self.use_cpp:
                Belle2.TrackFindingCDC.CDCMCHitLookUp.getInstance().fill()
                cppplotter.drawWrongRLHitsInSegments(self.cdc_segment_vector_store_obj_name)
            if self.use_python:
                styleDict = {'stroke': attributemaps.WrongRLColorMap()}
                pystoreobj = Belle2.PyStoreObj(self.cdc_segment_vector_store_obj_name)
 
                if pystoreobj:
                    # Wrapper around std::vector like
                    wrapped_vector = pystoreobj.obj()
                    vector = wrapped_vector.get()
 
                    for track in vector:
                        plotter.draw_iterable(list(track.items()), **styleDict)
 
        # Draw the RecoTracks
        if self.draw_recotracks:
            if self.use_cpp:
                cppplotter.drawRecoTracks(self.reco_tracks_store_array_name, 'ListColors', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                plotter.draw_storearray(self.reco_tracks_store_array_name, **styleDict)
 
        # Draw the MCRecoTracks
        if self.draw_mcrecotracks:
            if self.use_cpp:
                cppplotter.drawRecoTracks(self.mc_reco_tracks_store_array_name, 'ListColors', '')
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                plotter.draw_storearray(self.mc_reco_tracks_store_array_name, **styleDict)
 
        # Draw the RecoTracks matching status
        if self.draw_recotrack_matching:
            if self.use_cpp:
                cppplotter.drawRecoTracks(self.reco_tracks_store_array_name, 'MatchingStatus', '')
            if self.use_python:
                print('No Python-function defined')
 
        # Draw the Monte Carlo reference RecoTracks matching status
        if self.draw_mcrecotrack_matching:
            if self.use_cpp:
                cppplotter.drawRecoTracks(self.mc_reco_tracks_store_array_name, 'MCMatchingStatus', '')
            if self.use_python:
                print('No Python-function defined')
 
        # Draw interaction point
        if self.draw_interaction_point:
            if self.use_cpp:
                cppplotter.drawInteractionPoint()
            if self.use_python:
                plotter.draw_interaction_point()
 
        # Draw the superlayer boundaries
        if self.draw_superlayer_boundaries:
            if self.use_cpp:
                cppplotter.drawSuperLayerBoundaries()
            if self.use_python:
                plotter.draw_superlayer_boundaries()
 
        # Draw the outer and inner wall of the cdc.
        if self.draw_walls:
            if self.use_cpp:
                cppplotter.drawOuterCDCWall('black')
                cppplotter.drawInnerCDCWall('black')
            if self.use_python:
                styleDict = {'stroke': 'black'}
                plotter.draw_outer_cdc_wall(**styleDict)
                plotter.draw_inner_cdc_wall(**styleDict)
 
        # Draw the trajectories of the reco tracks
        if self.draw_mcparticle_trajectories:
            if self.use_cpp:
                cppplotter.drawMCParticleTrajectories("MCParticles", 'black', '')
            if self.use_python:
                print("Python backend can not draw mc particles")
 
        # Draw the fits to the segments
        if self.draw_segment_trajectories:
            if self.use_cpp:
                cppplotter.drawSegmentTrajectories(self.cdc_segment_vector_store_obj_name,
                                                   "ListColors", "")
            if self.use_python:
                segment_storevector = Belle2.PyStoreObj(self.cdc_segment_vector_store_obj_name)
                if segment_storevector:
                    segments = segment_storevector.obj().unwrap()
 
                    iterTrajectories = (segment.getTrajectory2D() for segment in segments)
                    plotter.draw_iterable(iterTrajectories)
 
        # Draw segment triples fits
        if self.draw_segmenttriple_trajectories:
            cppplotter.drawSegmentTripleTrajectories("CDCSegmentTriples",
                                                     "ListColors", "")
 
            if self.use_python:
                segmentTriple_storearray = Belle2.PyStoreArray('CDCSegmentTriples')
                if segmentTriple_storearray:
                    iterSegmentTriples = iter(segmentTriple_storearray)
                    iterTrajectories = (segmentTriple.getTrajectory2D()
                                        for segmentTriple in iterSegmentTriples)
                    plotter.draw_iterable(iterTrajectories)
 
        # Draw Track Trajectories
        if self.draw_track_trajectories:
            if self.use_cpp:
                cppplotter.drawTrackTrajectories("CDCTrackVector",
                                                 "ListColors", "")
            if self.use_python:
                styleDict = {'stroke': attributemaps.listColors}
                track_storevector = Belle2.PyStoreObj('CDCTrackVector')
                if track_storevector:
                    tracks = track_storevector.obj().unwrap()
                    iterTrajectories = (cdcTrack.getStartTrajectory3D().getTrajectory2D()
                                        for cdcTrack in tracks)
                    plotter.draw_iterable(iterTrajectories, **styleDict)
 
        # Draw the trajectories of the reco tracks
        if self.draw_recotrack_seed_trajectories:
            if self.use_python:
                recotrack_storearray = Belle2.PyStoreArray(self.reco_tracks_store_array_name)
                if recotrack_storearray:
                    def color_map(iTrajectory, trajectory):
                        # return "black"
                        return attributemaps.listColors[iTrajectory
                                                        % len(attributemaps.listColors)]
 
                    styleDict = {'stroke': color_map}
 
                    trajectories = []
                    for recotrack in recotrack_storearray:
                        tMomentum = recotrack.getMomentumSeed()
                        charge = recotrack.getChargeSeed()
                        tPosition = recotrack.getPositionSeed()
                        time = recotrack.getTimeSeed()
 
                        momentum = Belle2.TrackFindingCDC.Vector2D(tMomentum.X(),
                                                                   tMomentum.Y())
                        position = Belle2.TrackFindingCDC.Vector2D(tPosition.X(),
                                                                   tPosition.Y())
 
                        trajectory = \
                            Belle2.TrackFindingCDC.CDCTrajectory2D(position, time,
                                                                   momentum, charge)
                        trajectories.append(trajectory)
 
                    plotter.draw_iterable(trajectories, **styleDict)
 
            if self.use_cpp:
                raise NotImplementedError
 
        if self.draw_recotrack_fit_trajectories:
            if self.use_cpp:
                cppplotter.drawRecoTrackTrajectories(self.reco_tracks_store_array_name, '', '')
 
            if self.use_python:
                raise NotImplementedError
 
        fileName = self.new_output_filename()
        cppfileName = self.new_output_filename()
 
        if self.use_cpp:
            cppplotter.saveFile(cppfileName)
        if self.use_python:
            plotter.saveSVGFile(fileName)
 
        if self.interactive:
            if self.use_python:
                print(" Use the 'display' command to show the svg file", fileName,
                      'generated for the last event')
            if self.use_cpp:
                print(" Use the 'display' command to show the svg file", cppfileName,
                      'generated for the last event with cpp')
 
            # 'display' is part of the ImageMagic package commonly installed in linux
            if self.use_python:
                subprocess.Popen(['eog', fileName])
            if self.use_cpp:
                subprocess.Popen(['eog', cppfileName])
            # procDisplay = subprocess.Popen(['display','-background','white',
            # '-flatten',fileName])
            # procConverter = subprocess.Popen(['rsvg', root + '.svg', root + '.png'])
            input('Hit enter for next event')
 

◆ initialize()

initialize ( self )

Initialisation method of the module.
Creates the output folder if it is not present yet.

Definition at line 366 of file __init__.py.

    def initialize(self):
        """
        Initialisation method of the module.
        Creates the output folder if it is not present yet.
        """
 
        output_folder = self.output_folder
        if not os.path.exists(output_folder):
            print("CDCSVGDisplay.__init__ : Output folder '", output_folder,
                  "' does not exist.")
            answer = input('Create it? (y or n)')
            if answer == 'y':
                os.makedirs(output_folder)
 
        # Make sure at least one backend is available
        if not self.use_cpp and not self.use_python:
            self.use_cpp = True
 
        if self.use_cpp:
            cppplotter = Belle2.TrackFindingCDC.CDCSVGPlotter(self.animate, self.forward_fade)
        if self.use_python:
            plotter = svgdrawing.CDCSVGPlotter(animate=self.animate)
 
        # ######### CDCWires ##########
        # Draw wires from cdcwire objects
        # Now preferred way of plotting the wires
        if self.draw_wires:
            theCDCWireTopology = \
                Belle2.TrackFindingCDC.CDCWireTopology.getInstance()
 
            if self.use_cpp:
                cppplotter.drawWires(theCDCWireTopology)
            if self.use_python:
                plotter.draw(theCDCWireTopology)
 
        if self.use_cpp:
            
            self.prefilled_cppplotter = cppplotter
        if self.use_python:
            
            self.prefilled_plotter = plotter
 
        segment_relation_filter = Belle2.TrackFindingCDC.MVAFeasibleSegmentRelationFilter()
        segment_relation_filter.initialize()
 

◆ new_output_basename()

new_output_basename ( self )

Generates a new unique name for the current event without the folder prefix

Definition at line 1063 of file __init__.py.

    def new_output_basename(self):
        """
        Generates a new unique name for the current event without the folder prefix
        """
 
        if self.use_time_in_filename:
            output_basename = datetime.now().isoformat() + '.svg'
        else:
            output_basename = self.filename_prefix + str(self.file_number).zfill(4) + '.svg'
        return output_basename
 

◆ new_output_filename()

new_output_filename ( self )

Generates a new unique name for the current event with the folder prefix

Reimplemented in QueueDrawer.

Definition at line 1074 of file __init__.py.

    def new_output_filename(self):
        """
        Generates a new unique name for the current event with the folder prefix
        """
 
        return os.path.join(self.output_folder, self.new_output_basename())

◆ terminate()

terminate ( self )

terminate method of the module. Empty here.

Reimplemented in QueueDrawer.

Definition at line 1056 of file __init__.py.

    def terminate(self):
        """
        terminate method of the module. Empty here.
        """
 
        pass
 

Member Data Documentation

◆ _draw_wires

bool _draw_wires = True

protected

Switch to draw the wires.

Definition at line 76 of file __init__.py.

◆ animate

bool animate = False

Switch to make an animated event display by means of animated SVG.

Definition at line 61 of file __init__.py.

◆ cdc_hits_store_array_name

cdc_hits_store_array_name = "CDCHits"

Name of the CDC Hits store array.

Definition at line 239 of file __init__.py.

◆ cdc_segment_vector_store_obj_name

cdc_segment_vector_store_obj_name = 'CDCSegment2DVector'

Name of the CDC Reco Segment Vector.

Definition at line 251 of file __init__.py.

◆ cdc_wire_hit_cluster_store_obj_name

cdc_wire_hit_cluster_store_obj_name = "CDCWireHitClusterVector"

Name of the CDC Wire Hit Clusters.

Definition at line 248 of file __init__.py.

◆ draw_axialsegmentpairs

bool draw_axialsegmentpairs = False

Switch to draw the axial stereo segment pairs generated by the finder.

Definition at line 195 of file __init__.py.

◆ draw_clusters

bool draw_clusters = False

Switch to draw the clusters generated by the finder.

Definition at line 162 of file __init__.py.

◆ draw_connect_tof

bool draw_connect_tof = False

Switch to draw the CDCSimHits connected in the order of their getFlightTime for each Monte Carlo particle.

Default: inactive

Definition at line 132 of file __init__.py.

◆ draw_hits

bool draw_hits = True

Switch to draw the CDCHits.

Default: active

Definition at line 90 of file __init__.py.

◆ draw_interaction_point

bool draw_interaction_point = True

Switch to draw the interaction point.

Default: active

Definition at line 81 of file __init__.py.

◆ draw_mcaxialsegmentpairs

bool draw_mcaxialsegmentpairs = False

Switch to draw the axial to axial segment pairs from Monte Carlo truth.

Default: inactive

Definition at line 147 of file __init__.py.

◆ draw_mcparticle_id

bool draw_mcparticle_id = False

Switch to draw the MCParticle::getArrayIndex property.

Default: inactive

Definition at line 96 of file __init__.py.

◆ draw_mcparticle_pdgcode

bool draw_mcparticle_pdgcode = False

Switch to draw the MCParticle::getPDGCode property.

Default: inactive

Definition at line 99 of file __init__.py.

◆ draw_mcparticle_primary

bool draw_mcparticle_primary = False

Switch to draw the MCParticle::hasStatus(c_PrimaryParticle) property.

Default: inactive

Definition at line 102 of file __init__.py.

◆ draw_mcparticle_trajectories

bool draw_mcparticle_trajectories = False

Switch to draw the ideal trajectory of the MCParticle.

Default: inactive

Definition at line 105 of file __init__.py.

◆ draw_mcrecotrack_matching

bool draw_mcrecotrack_matching = False

Draw the MC reference RecoTracks pattern recognition matching status.

Definition at line 224 of file __init__.py.

◆ draw_mcrecotracks

bool draw_mcrecotracks = False

Draw the MC reference RecoTracks.

Definition at line 218 of file __init__.py.

◆ draw_mcsegmentpairs

bool draw_mcsegmentpairs = False

Switch to draw the axial to stereo segment pairs from Monte Carlo truth.

Default: inactive

Definition at line 150 of file __init__.py.

◆ draw_mcsegments

bool draw_mcsegments = False

Switch to draw the Monte Carlo segments as generated in CDCMCTrackStore.

Default: inactive

Definition at line 144 of file __init__.py.

◆ draw_mcsegmenttriples

bool draw_mcsegmenttriples = False

Switch to draw the segment triples from Monte Carlo truth.

Default: inactive

Definition at line 153 of file __init__.py.

◆ draw_nloops

bool draw_nloops = False

Switch to draw the CDCHit colored by the number of loops passed.

Definition at line 127 of file __init__.py.

◆ draw_reassigned

bool draw_reassigned = False

Switch to draw the CDCSimHits color coded by their reassignement information to a different MCParticle.

Default: inactive

Definition at line 141 of file __init__.py.

◆ draw_recotrack_fit_trajectories

bool draw_recotrack_fit_trajectories = False

Draw the output trackpoint trajectories.

Definition at line 230 of file __init__.py.

◆ draw_recotrack_matching

bool draw_recotrack_matching = False

Draw the output RecoTracks pattern recognition matching status.

Definition at line 221 of file __init__.py.

◆ draw_recotrack_seed_trajectories

bool draw_recotrack_seed_trajectories = False

Draw the output track seed trajectories.

Definition at line 227 of file __init__.py.

◆ draw_recotracks

bool draw_recotracks = False

Draw the output RecoTracks.

Definition at line 215 of file __init__.py.

◆ draw_rlinfo

bool draw_rlinfo = False

Switch to draw the CDCSimHits color coded by their local right left passage information.

Default: inactive

Definition at line 136 of file __init__.py.

◆ draw_segment_fbinfos

bool draw_segment_fbinfos = False

Switch to draw the segments generated by the finder colored by the coalignment information (forward, backward, undetermined)

Definition at line 178 of file __init__.py.

◆ draw_segment_firstInTrackIds

bool draw_segment_firstInTrackIds = False

Switch to draw the segments generated by the finder colored by the first in track hit id.

Definition at line 181 of file __init__.py.

◆ draw_segment_firstNPassedSuperLayers

bool draw_segment_firstNPassedSuperLayers = False

Switch to draw the segments generated by the finder colored by the number of passed superlayers associated to the first hit in the segment.

Definition at line 188 of file __init__.py.

◆ draw_segment_lastInTrackIds

bool draw_segment_lastInTrackIds = False

Switch to draw the segments generated by the finder colored by the second in track hit id.

Definition at line 184 of file __init__.py.

◆ draw_segment_lastNPassedSuperLayers

bool draw_segment_lastNPassedSuperLayers = False

Switch to draw the segments generated by the finder colored by the number of passed superlayers associated to the last hit in the segment.

Definition at line 192 of file __init__.py.

◆ draw_segment_mctrackids

bool draw_segment_mctrackids = False

Switch to draw the segments generated by the finder colored with the Monte Carlo track id.

Definition at line 174 of file __init__.py.

◆ draw_segment_trajectories

bool draw_segment_trajectories = False

Switch to draw the trajectories fitted to the segments generated by the finder.

Definition at line 168 of file __init__.py.

◆ draw_segmentpairs

bool draw_segmentpairs = False

Switch to draw the axial stereo segment pairs generated by the finder.

Definition at line 198 of file __init__.py.

◆ draw_segments

bool draw_segments = False

Switch to draw the segments generated by the finder.

Definition at line 165 of file __init__.py.

◆ draw_segmenttriple_trajectories

bool draw_segmenttriple_trajectories = False

Switch to draw the trajectories fitted to the segment triples generated by the finder.

Definition at line 204 of file __init__.py.

◆ draw_segmenttriples

bool draw_segmenttriples = False

Switch to draw the segment triples generated by the finder.

Definition at line 201 of file __init__.py.

◆ draw_simhit_bkgtag

bool draw_simhit_bkgtag = False

Switch to draw the CDCSimHits color coded by their getBackgroundTag() property.

Default: inactive

Definition at line 121 of file __init__.py.

◆ draw_simhit_isbkg

bool draw_simhit_isbkg = False

Switch to draw the CDCSimHits red for getBackgroundTag() != bg_none.

Default: inactive

Definition at line 124 of file __init__.py.

◆ draw_simhit_pdgcode

bool draw_simhit_pdgcode = False

Switch to draw the CDCSimHits color coded by their getPDGCode() property.

Default: inactive

Definition at line 117 of file __init__.py.

◆ draw_simhit_posflag

bool draw_simhit_posflag = False

Switch to draw the CDCSimHits color coded by their getPosFlag() property.

Default: inactive

Definition at line 114 of file __init__.py.

◆ draw_simhit_tof

bool draw_simhit_tof = False

Switch to draw the CDCSimHits color coded by their time of flight.

Default: inactive

Definition at line 111 of file __init__.py.

◆ draw_simhits

bool draw_simhits = False

Switch to draw the CDCSimHits with momentum information.

Default: inactive

Definition at line 108 of file __init__.py.

◆ draw_superclusters

bool draw_superclusters = False

Switch to draw the clusters generated by the finder.

Definition at line 159 of file __init__.py.

◆ draw_superlayer_boundaries

bool draw_superlayer_boundaries = False

Switch to draw the superlayer boundaries.

Default: inactive

Definition at line 84 of file __init__.py.

◆ draw_takenflag

bool draw_takenflag = False

Switch to draw the CDCHits colored by the associated CDCWireHit taken flag.

Default: active

Definition at line 93 of file __init__.py.

◆ draw_tangentsegments

bool draw_tangentsegments = False

Switch to draw the tangent segments generated by the finder.

Definition at line 171 of file __init__.py.

◆ draw_track_trajectories

bool draw_track_trajectories = False

Switch to draw the trajectories of the tracks generated by the finder.

Definition at line 210 of file __init__.py.

◆ draw_tracks

bool draw_tracks = False

Switch to draw the tracks generated by the finder.

Definition at line 207 of file __init__.py.

◆ draw_walls

bool draw_walls = False

Switch to draw the superlayer boundaries.

Default: inactive

Definition at line 87 of file __init__.py.

◆ draw_wires

draw_wires = svgdrawing.CDCSVGPlotter(animate=self.animate)

Definition at line 392 of file __init__.py.

◆ draw_wrong_rl_infos_in_segments

bool draw_wrong_rl_infos_in_segments = False

Draw a red cdc hit of the rl info of the segment reco hits is wrong, else a green one.

Definition at line 233 of file __init__.py.

◆ draw_wrong_rl_infos_in_tracks

bool draw_wrong_rl_infos_in_tracks = False

Draw a red cdc hit of the rl info of the track reco hits is wrong, else a green one.

Definition at line 236 of file __init__.py.

◆ file_number

int file_number = 0

Current file's number (used for making output filename)

Definition at line 254 of file __init__.py.

◆ filename_prefix

str filename_prefix = "CDCDisplay"

Filename prefix.

Definition at line 257 of file __init__.py.

◆ forward_fade

bool forward_fade = False

Switch to make the color of segments and tracks fade out in the forward direction.

Definition at line 64 of file __init__.py.

◆ interactive

interactive = interactive

Switch if the module shall show the event to the user and wait to continue or just generate the images.

Definition at line 50 of file __init__.py.

◆ mc_reco_tracks_store_array_name

mc_reco_tracks_store_array_name = "MCRecoTracks"

Name of the Monte Carlo reference RecoTracks store array.

Definition at line 245 of file __init__.py.

◆ output_folder

output_folder = output_folder

Folder the images shall be saved to.

Definition at line 53 of file __init__.py.

◆ prefilled_cppplotter

prefilled_cppplotter = cppplotter

prefilled default is to use the C++ plotter

Definition at line 403 of file __init__.py.

◆ prefilled_plotter

prefilled_plotter = plotter

prefilled default is to use the python plotter

Definition at line 406 of file __init__.py.

◆ reco_tracks_store_array_name

reco_tracks_store_array_name = "RecoTracks"

Name of the RecoTracks store array.

Definition at line 242 of file __init__.py.

◆ use_cpp

bool use_cpp = True

Flag to use cpp.

Definition at line 70 of file __init__.py.

◆ use_python

bool use_python = False

Flag to use python.

Definition at line 73 of file __init__.py.

◆ use_time_in_filename

bool use_time_in_filename = False

Use time instead of prefix in filename.

Definition at line 260 of file __init__.py.

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

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

Public Member Functions

Public Attributes

Protected Attributes

Detailed Description

Constructor & Destructor Documentation

◆ __init__()

Member Function Documentation

◆ all_drawoptions()

◆ beginRun()

◆ draw_wires() [1/2]

◆ draw_wires() [2/2]

◆ drawoptions()

◆ endRun()

◆ event()

◆ initialize()

◆ new_output_basename()

◆ new_output_filename()

◆ terminate()

Member Data Documentation

◆ _draw_wires

◆ animate

◆ cdc_hits_store_array_name

◆ cdc_segment_vector_store_obj_name

◆ cdc_wire_hit_cluster_store_obj_name

◆ draw_axialsegmentpairs

◆ draw_clusters

◆ draw_connect_tof

◆ draw_hits

◆ draw_interaction_point

◆ draw_mcaxialsegmentpairs

◆ draw_mcparticle_id

◆ draw_mcparticle_pdgcode

◆ draw_mcparticle_primary

◆ draw_mcparticle_trajectories

◆ draw_mcrecotrack_matching

◆ draw_mcrecotracks

◆ draw_mcsegmentpairs

◆ draw_mcsegments

◆ draw_mcsegmenttriples

◆ draw_nloops

◆ draw_reassigned

◆ draw_recotrack_fit_trajectories

◆ draw_recotrack_matching

◆ draw_recotrack_seed_trajectories

◆ draw_recotracks

◆ draw_rlinfo

◆ draw_segment_fbinfos

◆ draw_segment_firstInTrackIds

◆ draw_segment_firstNPassedSuperLayers

◆ draw_segment_lastInTrackIds

◆ draw_segment_lastNPassedSuperLayers

◆ draw_segment_mctrackids

◆ draw_segment_trajectories

◆ draw_segmentpairs

◆ draw_segments

◆ draw_segmenttriple_trajectories

◆ draw_segmenttriples

◆ draw_simhit_bkgtag

◆ draw_simhit_isbkg

◆ draw_simhit_pdgcode

◆ draw_simhit_posflag

◆ draw_simhit_tof

◆ draw_simhits

◆ draw_superclusters

◆ draw_superlayer_boundaries

◆ draw_takenflag

◆ draw_tangentsegments

◆ draw_track_trajectories

◆ draw_tracks

◆ draw_walls

◆ draw_wires

◆ draw_wrong_rl_infos_in_segments

◆ draw_wrong_rl_infos_in_tracks

◆ file_number

◆ filename_prefix

◆ forward_fade

◆ interactive

◆ mc_reco_tracks_store_array_name

◆ output_folder

◆ prefilled_cppplotter

◆ init()