CMS 3D CMS Logo

List of all members | Public Member Functions | Public Attributes
geometry.Alignables Class Reference

Public Member Functions

def __init__ (self, config)
 
def create_children_list (self)
 
def create_list (self, MillePedeUser)
 
def get_detids (self, subdetid, pattern={})
 
def get_discriminator (self, objid)
 
def get_ndiscriminator (self, objid)
 
def get_subdetid (self, objid)
 

Public Attributes

 config
 
 structures
 

Detailed Description

Creates a list of the aligned strucutres. Get the fields out of the
TrackerTree.root file.

Definition at line 16 of file geometry.py.

Constructor & Destructor Documentation

◆ __init__()

def geometry.Alignables.__init__ (   self,
  config 
)

Definition at line 21 of file geometry.py.

21  def __init__(self, config):
22  # list of Structure objects, contains structures which were aligned
23  self.structures = []
24  self.config = config
25 

Member Function Documentation

◆ create_children_list()

def geometry.Alignables.create_children_list (   self)

Definition at line 70 of file geometry.py.

70  def create_children_list(self):
71  for struct in self.structures:
72  # loop over discriminators -> create patterns
73  # pattern {"half": 2, "side": 2, "layer": 6, ...}
74  ranges = struct.ndiscriminator
75  pranges = [list(range(1, x+1)) for x in ranges]
76  # loop over all possible combinations of the values of the
77  # discriminators
78  for number in itertools.product(*pranges):
79  # create pattern dict
80  pattern = dict(zip(map(lambda x: x.lower(), struct.discriminator), number))
81  # name out of pattern
82  name = " ".join("{0} {1}".format(key.lower(), value)
83  for (key, value) in pattern.items())
84  # get detids of child
85  detids = self.get_detids(struct.subdetid, pattern)
86  # create child and add it to parent
87  child = Structure(name, struct.subdetid, detids=detids)
88  struct.children.append(child)
89 
90 

References geometry.Alignables.get_detids(), join(), genParticles_cff.map, FastTimerService_cff.range, geometry.Alignables.structures, and ComparisonHelper.zip().

◆ create_list()

def geometry.Alignables.create_list (   self,
  MillePedeUser 
)

Definition at line 52 of file geometry.py.

52  def create_list(self, MillePedeUser):
53  # loop over output TTree
54  for entry in MillePedeUser:
55  # check which structures were aligned
56  if (entry.ObjId != 1 and 999999 not in map(abs, entry.Par)):
57  # check if structure is not yet in the list
58  if not any(x.name == str(entry.Name) for x in self.structures):
59  # create new structure object
60  name = str(entry.Name)
61  subdetid = self.get_subdetid(entry.ObjId)
62  discriminator = self.get_discriminator(entry.ObjId)
63  ndiscriminator = self.get_ndiscriminator(entry.ObjId)
64  # create structure
65  self.structures.append(
66  Structure(name, subdetid, discriminator, ndiscriminator))
67  # add detids which belong to this structure
68  self.structures[-1].detids = self.get_detids(subdetid)
69 

References any(), mps_setup.append, geometry.Alignables.get_detids(), geometry.Alignables.get_discriminator(), geometry.Alignables.get_ndiscriminator(), geometry.Alignables.get_subdetid(), genParticles_cff.map, str, and geometry.Alignables.structures.

◆ get_detids()

def geometry.Alignables.get_detids (   self,
  subdetid,
  pattern = {} 
)

Definition at line 91 of file geometry.py.

91  def get_detids(self, subdetid, pattern={}):
92  # list of all detids in the structure
93  detids = []
94  # open TrackerTree.root file
95  treeFile = ROOT.TFile(os.path.join(self.config.jobDataPath,
96  ".TrackerTree.root"))
97  tree = treeFile.Get("TrackerTreeGenerator/TrackerTree/TrackerTree")
98 
99  for entry in tree:
100  # check if entry is part of the structure
101  if (entry.SubdetId == subdetid):
102  # to create a child also check the pattern
103  structure_found = False
104  for structure in ("Half", "Side", "Layer", "Rod", "Ring",
105  "Petal", "Blade", "Panel", "OuterInner",
106  "Module"):
107  if structure.lower() in pattern:
108  if getattr(entry, structure) != pattern[structure.lower()]:
109  structure_found = True
110  break
111  if structure_found: continue
112 
113  detids.append(entry.RawId)
114  return detids
115 
116 

References geometry.Alignables.config.

Referenced by geometry.Alignables.create_children_list(), and geometry.Alignables.create_list().

◆ get_discriminator()

def geometry.Alignables.get_discriminator (   self,
  objid 
)

Definition at line 29 of file geometry.py.

29  def get_discriminator(self, objid):
30  return mpsv_geometrydata.data[objid].discriminator
31 

Referenced by geometry.Alignables.create_list(), and geometry.Alignables.get_ndiscriminator().

◆ get_ndiscriminator()

def geometry.Alignables.get_ndiscriminator (   self,
  objid 
)

Definition at line 32 of file geometry.py.

32  def get_ndiscriminator(self, objid):
33  subdetid = self.get_subdetid(objid)
34  discriminator = self.get_discriminator(objid)
35  ndiscriminator = {key: [] for key in discriminator}
36  # open TrackerTree.root file
37  treeFile = ROOT.TFile(os.path.join(self.config.jobDataPath,
38  ".TrackerTree.root"))
39  tree = treeFile.Get("TrackerTreeGenerator/TrackerTree/TrackerTree")
40 
41  for entry in tree:
42  # check if entry is part of the structure
43  if (entry.SubdetId == subdetid):
44  for structure in discriminator:
45  ndiscriminator[structure].append(getattr(entry, structure))
46  for structure in discriminator:
47  ndiscriminator[structure] = [x for x in ndiscriminator[structure] if x != 0]
48 
49  return [len(set(ndiscriminator[structure]))
50  for structure in discriminator]
51 

References mps_setup.append, geometry.Alignables.config, geometry.Alignables.get_discriminator(), and geometry.Alignables.get_subdetid().

Referenced by geometry.Alignables.create_list().

◆ get_subdetid()

def geometry.Alignables.get_subdetid (   self,
  objid 
)

Definition at line 26 of file geometry.py.

26  def get_subdetid(self, objid):
27  return mpsv_geometrydata.data[objid].subdetid
28 

Referenced by geometry.Alignables.create_list(), and geometry.Alignables.get_ndiscriminator().

Member Data Documentation

◆ config

geometry.Alignables.config

◆ structures

geometry.Alignables.structures
FastTimerService_cff.range
range
Definition: FastTimerService_cff.py:34
join
static std::string join(char **cmd)
Definition: RemoteFile.cc:17
any
bool any(const std::vector< T > &v, const T &what)
Definition: ECalSD.cc:38
str
#define str(s)
Definition: TestProcessor.cc:51
mps_setup.append
append
Definition: mps_setup.py:85
ComparisonHelper::zip
OutputIterator zip(InputIterator1 first1, InputIterator1 last1, InputIterator2 first2, InputIterator2 last2, OutputIterator result, Compare comp)
Definition: L1TStage2CaloLayer1.h:41
format
genParticles_cff.map
map
Definition: genParticles_cff.py:11