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 15 of file geometry.py.

Constructor & Destructor Documentation

def geometry.Alignables.__init__	(		self,
			config
	)

Definition at line 20 of file geometry.py.

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

Member Function Documentation

def geometry.Alignables.create_children_list

(

self

)

Definition at line 69 of file geometry.py.

References cmsPerfStripChart.dict, geometry.Alignables.get_detids(), join(), genParticles_cff.map, geometry.Alignables.structures, and ComparisonHelper.zip().

    def create_children_list(self):
        for struct in self.structures:
            # loop over discriminators -> create patterns
            # pattern {"half": 2, "side": 2, "layer": 6, ...}
            ranges = struct.ndiscriminator
            pranges = [range(1, x+1) for x in ranges]
            # loop over all possible combinations of the values of the
            # discriminators
            for number in itertools.product(*pranges):
                # create pattern dict
                pattern = dict(zip(map(lambda x: x.lower(), struct.discriminator), number))
                # name out of pattern
                name = " ".join("{0} {1}".format(key.lower(), value)
                                for (key, value) in pattern.items())
                # get detids of child
                detids = self.get_detids(struct.subdetid, pattern)
                # create child and add it to parent
                child = Structure(name, struct.subdetid, detids=detids)
                struct.children.append(child)

def geometry.Alignables.create_list	(		self,
			MillePedeUser
	)

Definition at line 51 of file geometry.py.

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

    def create_list(self, MillePedeUser):
        # loop over output TTree
        for entry in MillePedeUser:
            # check which structures were aligned
            if (entry.ObjId != 1 and 999999 not in map(abs, entry.Par)):
                # check if structure is not yet in the list
                if not any(x.name == str(entry.Name) for x in self.structures):
                    # create new structure object
                    name = str(entry.Name)
                    subdetid = self.get_subdetid(entry.ObjId)
                    discriminator = self.get_discriminator(entry.ObjId)
                    ndiscriminator = self.get_ndiscriminator(entry.ObjId)
                    # create structure
                    self.structures.append(
                        Structure(name, subdetid, discriminator, ndiscriminator))
                    # add detids which belong to this structure
                    self.structures[-1].detids = self.get_detids(subdetid)

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

Definition at line 90 of file geometry.py.

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

    def get_detids(self, subdetid, pattern={}):
        # list of all detids in the structure
        detids = []
        # open TrackerTree.root file
        treeFile = ROOT.TFile(os.path.join(self.config.jobDataPath,
                                           ".TrackerTree.root"))
        tree = treeFile.Get("TrackerTreeGenerator/TrackerTree/TrackerTree")

        for entry in tree:
            # check if entry is part of the structure
            if (entry.SubdetId == subdetid):
                # to create a child also check the pattern
                structure_found = False
                for structure in ("Half", "Side", "Layer", "Rod", "Ring",
                                  "Petal", "Blade", "Panel", "OuterInner",
                                  "Module"):
                    if structure.lower() in pattern:
                        if getattr(entry, structure) != pattern[structure.lower()]:
                            structure_found = True
                            break
                if structure_found: continue

                detids.append(entry.RawId)
        return detids

def geometry.Alignables.get_discriminator	(		self,
			objid
	)

Definition at line 28 of file geometry.py.

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

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

def geometry.Alignables.get_ndiscriminator	(		self,
			objid
	)

Definition at line 31 of file geometry.py.

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

Referenced by geometry.Alignables.create_list().

    def get_ndiscriminator(self, objid):
        subdetid = self.get_subdetid(objid)
        discriminator = self.get_discriminator(objid)
        ndiscriminator = {key: [] for key in discriminator}
        # open TrackerTree.root file
        treeFile = ROOT.TFile(os.path.join(self.config.jobDataPath,
                                           ".TrackerTree.root"))
        tree = treeFile.Get("TrackerTreeGenerator/TrackerTree/TrackerTree")

        for entry in tree:
            # check if entry is part of the structure
            if (entry.SubdetId == subdetid):
                for structure in discriminator:
                    ndiscriminator[structure].append(getattr(entry, structure))
        for structure in discriminator:
            ndiscriminator[structure] = [x for x in ndiscriminator[structure] if x != 0]

        return [len(set(ndiscriminator[structure]))
                for structure in discriminator]

def geometry.Alignables.get_subdetid	(		self,
			objid
	)

Definition at line 25 of file geometry.py.

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

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

Member Data Documentation

geometry.Alignables.config

Definition at line 23 of file geometry.py.

Referenced by preexistingValidation.PreexistingValidation.getRepMap(), genericValidation.GenericValidation.getRepMap(), and zMuMuValidation.ZMuMuValidation.trackcollection().

geometry.Alignables.structures

Definition at line 22 of file geometry.py.

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