AFPAlignMCDBCreate.AFPDBDict Class Reference

Collaboration diagram for AFPAlignMCDBCreate.AFPDBDict:

Public Member Functions
	__init__ (self, folderBlk)
	append (self, stationID, layerID=-1, alignType="None", shiftX=0.0, shiftY=0.0, shiftZ=0.0, alpha=0.0, beta=0.0, gamma=0.0)
	savePayload (self, folderBlk)

Public Attributes
dict	mydict

Detailed Description

A class to create a dictionary, fill it with zeros in the constructor and to overwrite zeros later

Definition at line 58 of file AFPAlignMCDBCreate.py.

Constructor & Destructor Documentation

__init__()

AFPAlignMCDBCreate.AFPDBDict.__init__	(		self,
			folderBlk )

Definition at line 60 of file AFPAlignMCDBCreate.py.

    def __init__ (self, folderBlk):
        if(folderBlk.folderName=="/FWD/AFP/Align/Local"):
            emptydict={"stationID":0, "layerID":-1, "shiftX":0.0, "shiftY":0.0, "shiftZ":0.0, "alpha":0.0, "beta":0.0, "gamma":0.0}
        elif(folderBlk.folderName=="/FWD/AFP/Align/Global"):
            emptydict={"stationID":0, "alignType":"None", "shiftX":0.0, "shiftY":0.0, "shiftZ":0.0, "alpha":0.0, "beta":0.0, "gamma":0.0}
        else:
            print ("unknown folder %s, please edit \"AFPDBDict\" class constructor, exiting now" % folderBlk.folderName)
            sys.exit(1)
        
        self.mydict={"author":"Petr Balek", 
                     "version":"AFP_DB_v2",
                     "nchannels":16,
                     "data": dict.fromkeys(range(0, 16))}
        
        for i in range(0, 16):
            self.mydict["data"][i]=copy.deepcopy(emptydict)
            self.mydict["data"][i]["stationID"]=i//4
            if(folderBlk.folderName=="/FWD/AFP/Align/Local"):
                self.mydict["data"][i]["layerID"]=i%4
            if(folderBlk.folderName=="/FWD/AFP/Align/Global"):
                if i%4==0: self.mydict["data"][i]["alignType"]="tracker"
                if i%4==1: self.mydict["data"][i]["alignType"]="beam"
                if i%4==2: self.mydict["data"][i]["alignType"]="RP"
                if i%4==3: self.mydict["data"][i]["alignType"]="correction"
    

Member Function Documentation

append()

AFPAlignMCDBCreate.AFPDBDict.append	(		self,
			stationID,
			layerID = -1,
			alignType = "None",
			shiftX = 0.0,
			shiftY = 0.0,
			shiftZ = 0.0,
			alpha = 0.0,
			beta = 0.0,
			gamma = 0.0 )

A function that overwrites one slice of the alignment constants in the dictionary. Local constants have to have layerID defined, while alignType is undefined. Global constants have to have alignType defined (tracker/RP/beam/correction) and layerID undefined.

Definition at line 85 of file AFPAlignMCDBCreate.py.

    def append(self, stationID, layerID=-1, alignType="None", shiftX=0.0, shiftY=0.0, shiftZ=0.0, alpha=0.0, beta=0.0, gamma=0.0):
        """A function that overwrites one slice of the alignment constants in the dictionary. Local constants have to have layerID defined, while alignType is undefined. Global constants have to have alignType defined (tracker/RP/beam/correction) and layerID undefined."""
 
        # perform some simple check so basic mistakes are avoided
        if(layerID==-1 and alignType=="None"):
            print ("cannot save payload, got layerID=%d and alignType=%s; one of them has to be specified" % layerID, alignType)
            sys.exit(1)
        elif(layerID!=-1 and alignType!="None"):
            print ("cannot save payload, got layerID=%d and alignType=%s; one of them should not be specified" % layerID, alignType)
            sys.exit(1)
 
        if(stationID<0 or stationID>=4):
            print ("cannot save payload, got stationID=%d, unknown" % stationID)
            sys.exit(1)
 
        channel=0
        # station 0 occupies channels 0-3, station 1 occupies channels 4-7, ..., and station 3 occupies channels 12-15
        # if you are insterested in e.g. "beam" constants, look at channels 1, 5, 9, and 13
        if(alignType!="None"):
            channel=stationID*4
            if(alignType=="tracker"):
                channel+=0
            elif(alignType=="beam"):
                channel+=1
            elif(alignType=="RP"):
                channel+=2
            elif(alignType=="correction"):
                channel+=3
            else:
                print ("cannot save payload, got alignType=%s, unknown" % alignType)
                sys.exit(1)
            
            # overwrite it
            self.mydict["data"][channel]['alignType']=alignType
 
        # station 0 again occupies channels 0-3 etc.
        if(layerID!=-1):
            channel=stationID*4
            if(0<=layerID and layerID<4):
                channel+=layerID
            else:
                print ("cannot save payload, got layerID=%d, unknown" % layerID)
                sys.exit(1)
 
            # overwrite it
            self.mydict["data"][channel]['layerID']=layerID
 
        # overwrite it
        mydict_helper=self.mydict["data"][channel]
        mydict_helper['shiftX'], mydict_helper['shiftY'], mydict_helper['shiftZ'] = shiftX, shiftY, shiftZ
        mydict_helper['alpha'], mydict_helper['beta'], mydict_helper['gamma'] = alpha, beta, gamma    
        mydict_helper['stationID']=stationID
        
 

savePayload()

AFPAlignMCDBCreate.AFPDBDict.savePayload	(		self,
			folderBlk )

A function to transform the dictionary to JSON and save it in IOV from 0 to infinity; technically, upper limits are undefined and the maximum value (until beginning of a next entry) would be used.

Definition at line 139 of file AFPAlignMCDBCreate.py.

    def savePayload(self, folderBlk):
        """A function to transform the dictionary to JSON and save it in IOV from 0 to infinity; technically, upper limits are undefined and the maximum value (until beginning of a next entry) would be used."""
        
        # transform run nr. and LB to the right integers
        since=runLBtoDB(0,0)
        until=cool.ValidityKeyMax
 
        # provided set of constants have to be the right one for the provided specification
        payload=cool.Record(folderBlk.spec)
        # transform dictionary to JSON
        payload["data"]=json.dumps(self.mydict, indent = 1)
        # save everything (that "0" stands for channel nr. 0, there is only 1 real channel in the DB)
        folderBlk.folder.storeObject(since,until,payload,0,folderBlk.tag)
 
 

Member Data Documentation

mydict

dict AFPAlignMCDBCreate.AFPDBDict.mydict

Initial value:

= {"author":"Petr Balek", 
                     "version":"AFP_DB_v2",
                     "nchannels":16,
                     "data": dict.fromkeys(range(0, 16))}

Definition at line 69 of file AFPAlignMCDBCreate.py.

---

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

• AFPAlignMCDBCreate.py