def _graph_add_topological_block(self, graph, graphnodes, pfblock):
'''this adds the block links (distance, is_linked) onto the DAG in red'''
for edge in pfblock.edges.itervalues():
if edge.linked:
label = "{:.1E}".format(edge.distance)
if edge.distance == 0:
label = "0"
graph.add_edge(pydot.Edge(graphnodes[Identifier.pretty(edge.id1)],
graphnodes[Identifier.pretty(edge.id2)],
label=label, style="dashed", color="red", arrowhead="none", arrowtail="none", fontsize='7'))
def short_elements_string(self):
''' Construct a string description of each of the elements in a block.
The elements are given a short name E/H/T according to ecal/hcal/track
and then sequential numbering starting from 0, this naming is also used to index the
matrix of distances. The full unique id is also given.
For example:-
elements: {
E0:1104134446736:SmearedCluster : ecal_in 0.57 0.33 -2.78
H1:2203643940048:SmearedCluster : hcal_in 6.78 0.35 -2.86
T2:3303155568016:SmearedTrack : 5.23 4.92 0.34 -2.63
}
'''
count = 0
elemdetails = " elements:\n"
for uid in self.element_uniqueids:
elemdetails += "{shortname:>7}{count} = {strdescrip:9} value={val:5.1f} ({uid})\n".format(
shortname=Identifier.type_letter(uid),
count=count,
strdescrip=Identifier.pretty(uid),
val=Identifier.get_value(uid),
uid=uid)
count = count + 1
return elemdetails
def short_elements_string(self):
''' Construct a string description of each of the elements in a block.
The elements are given a short name E/H/T according to ecal/hcal/track
and then sequential numbering starting from 0, this naming is also used to index the
matrix of distances. The full unique id is also given.
For example:-
elements: {
E0:1104134446736:SmearedCluster : ecal_in 0.57 0.33 -2.78
H1:2203643940048:SmearedCluster : hcal_in 6.78 0.35 -2.86
T2:3303155568016:SmearedTrack : 5.23 4.92 0.34 -2.63
}
'''
count = 0
elemdetails = " elements:\n"
for uid in self.element_uniqueids:
elemdetails += "{shortname:>7}{count} = {strdescrip:9} value={val:5.1f} ({uid})\n".format(
shortname=Identifier.type_letter(uid),
count=count,
strdescrip=Identifier.pretty(uid),
val=Identifier.get_value(uid),
uid=uid)
count = count + 1
return elemdetails
def test_papasevent(self):
papasevent = PapasEvent(0)
ecals = dict()
tracks = dict()
mixed = dict()
for i in range(0, 2):
uid = Identifier.make_id(Identifier.PFOBJECTTYPE.ECALCLUSTER, i,'t', 4.5)
ecals[uid] = uid
for i in range(0, 2):
uid = Identifier.make_id(Identifier.PFOBJECTTYPE.TRACK, i, 's', 4.5)
tracks[uid] = uid
lastid = Identifier.make_id(Identifier.PFOBJECTTYPE.ECALCLUSTER, 3, 't', 3)
ecals[lastid] = lastid
papasevent.add_collection(ecals)
papasevent.add_collection(tracks)
#check that adding the same collection twice fails
self.assertRaises(ValueError, papasevent.add_collection, ecals)
#check that adding a mixed collection fails
mixed = ecals.copy()
mixed.update(tracks)
self.assertRaises(ValueError, papasevent.add_collection, mixed)
#get we can get back collections OK
self.assertTrue( papasevent.get_collection('zz') is None)
self.assertTrue( len(papasevent.get_collection('et')) == 3 )
#check get_object
self.assertTrue( Identifier.pretty(papasevent.get_object(lastid)) == 'et3' )
self.assertTrue( papasevent.get_object(499) is None )
def __str__(self):
mainstr = super(Particle, self).__str__()
idstr = '{pretty:6}:{id}'.format(
pretty=Identifier.pretty(self.uniqueid),
id=self.uniqueid)
fields = mainstr.split(':')
fields.insert(1, idstr)
return ':'.join(fields)
def __str__(self):
mainstr = super(Particle, self).__str__()
idstr = '{pretty:6}:{uid}'.format(pretty=Identifier.pretty(
self.uniqueid),
uid=self.uniqueid)
fields = mainstr.split(':')
fields.insert(1, idstr)
return ':'.join(fields)
def info(self):
subclusterstr = str('sub(')
for s in self.subclusters:
subclusterstr += str('{:}, '.format(Identifier.pretty(s.uniqueid)))
subclusterstr += ")"
return '{energy:7.2f} {theta:5.2f} {phi:5.2f} {sub}'.format(
energy=self.energy,
theta=math.pi / 2. - self.position.Theta(),
phi=self.position.Phi(),
sub=subclusterstr)
def id_from_pretty(self, pretty):
''' Searches to find the true id given a pretty id string
Not super efficient but OK for occasional use
eg uid = self.id_from_pretty('et103')
@param: pretty is the easily readable name from the Identifier class which is shown in prints and plots eg 'et103'
'''
for uid in self.history.keys():
if Identifier.pretty(uid) == pretty:
return uid
return None
def id_from_pretty(self, pretty):
''' Searches to find the true id given a pretty id string
Not super efficient but OK for occasional use
eg uid = self.id_from_pretty('et103')
@param: pretty is the easily readable name from the Identifier class which is shown in prints and plots eg 'et103'
'''
for uid in self.history.keys():
if Identifier.pretty(uid) == pretty:
return uid
return None
def info(self):
subclusterstr = str('sub(')
for s in self.subclusters:
subclusterstr += str('{:}, '.format(Identifier.pretty(s.uniqueid)))
subclusterstr += ")"
return '{energy:7.2f} {theta:5.2f} {phi:5.2f} {sub}'.format(
energy=self.energy,
theta=math.pi/2. - self.position.Theta(),
phi=self.position.Phi(),
sub=subclusterstr
)
def __repr__(self):
''' Short Block description
'''
description = "block:"
description += str(
'{shortname:8} :{prettyid:6}: ecals = {count_ecal} hcals = {count_hcal} tracks = {count_tracks}'
.format(shortname=self.short_info(),
prettyid=Identifier.pretty(self.uniqueid),
count_ecal=self.count_ecal(),
count_hcal=self.count_hcal(),
count_tracks=self.count_tracks()))
return description
def __repr__(self):
''' Short Block description
'''
description = "block:"
description += str('{shortname:8} :{prettyid:6}: ecals = {count_ecal} hcals = {count_hcal} tracks = {count_tracks}'.format(
shortname=self.short_info(),
prettyid=Identifier.pretty(self.uniqueid),
count_ecal=self.count_ecal(),
count_hcal=self.count_hcal(),
count_tracks=self.count_tracks())
)
return description
def test_papasevent(self):
papasevent = PapasEvent(0)
ecals = dict()
tracks = dict()
mixed = dict()
for i in range(0, 2):
uid = Identifier.make_id(Identifier.PFOBJECTTYPE.ECALCLUSTER, i,
't', 4.5)
ecals[uid] = uid
for i in range(0, 2):
uid = Identifier.make_id(Identifier.PFOBJECTTYPE.TRACK, i, 's',
4.5)
tracks[uid] = uid
lastid = Identifier.make_id(Identifier.PFOBJECTTYPE.ECALCLUSTER, 3,
't', 3)
ecals[lastid] = lastid
papasevent.add_collection(ecals)
papasevent.add_collection(tracks)
#check that adding the same collection twice fails
self.assertRaises(ValueError, papasevent.add_collection, ecals)
#check that adding a mixed collection fails
mixed = ecals.copy()
mixed.update(tracks)
self.assertRaises(ValueError, papasevent.add_collection, mixed)
#get we can get back collections OK
self.assertTrue(len(
papasevent.get_collection('zz')) == 0) # this one does not exist
self.assertTrue(len(papasevent.get_collection('et')) == 3)
#check get_object
self.assertTrue(
Identifier.pretty(papasevent.get_object(lastid)) == 'et3')
self.assertTrue(papasevent.get_object(499) is None)
def __str__(self):
return '{classname}: {pretty:6}:{uid}: {info}'.format(
classname=self.__class__.__name__,
pretty=Identifier.pretty(self.uniqueid),
uid=self.uniqueid,
info=self.info())
def short_info(self, node):
'''used to label plotted dag nodes
@param node: a node in the DAG history'''
obj = self.object(node)
return Identifier.pretty(obj.uniqueid) + "\n " +obj.short_info()
def __str__(self):
return '{classname}: {pretty:6}:{id}: {info}'.format(
classname=self.__class__.__name__,
pretty=Identifier.pretty(self.uniqueid),
id=self.uniqueid,
info=self.info())
def pretty(self, node):
''' pretty form of the unique identifier
@param node: a node in the DAG history'''
return Identifier.pretty(node.get_value())
