dumpNswErrorDb.py

Go to the documentation of this file.

# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
import logging
import json
from itertools import product
def setupArgParser():
    from argparse import ArgumentParser
 
    parser = ArgumentParser()
    parser.add_argument("--outFilePreFix", help="Output JSON file",
                                     default="NswUncerts")
    parser.add_argument("--scaleError", help="Scale the uncertainties of the MM",type=float, default = 5.)
    parser.add_argument("--addErrorMMClusterTime", help="add fixed error to the uncertainties of the MM",type=float, default = 0.)
    return parser
 
if __name__ == "__main__":
    args = setupArgParser().parse_args()
 
    dbEntriesMMASide = []
    dbEntriesMMCSide = []
    
    for stationName, stationEta, stationPhi, multilayer, gasGap, author in product([ "MMS", "MML"], [-2,-1,1,2], range(1,9), [1,2],range(1,5), [0,65]):
        constant = args.scaleError * 0.0815
        linear =  args.scaleError * 0.0
        quad =  args.scaleError * 1.56
 
        minStrip = 1
        maxStrip = (5 if abs(stationEta)==1 else 3)*1024
        
        calibEntry = dict({"station": stationName, "eta":stationEta, "phi": stationPhi, "multilayer": multilayer, "gasGap": gasGap, "minStrip": minStrip,"maxStrip": maxStrip, "modelName":"tanThetaPolynomial" ,"modelPars": [constant, linear, quad], "clusterAuthor": author})
        logging.debug(calibEntry)
        if(stationEta<0):
            dbEntriesMMCSide.append(calibEntry)
        else:
            dbEntriesMMASide.append(calibEntry)
 
 
    for stationName, stationEta, stationPhi, multilayer, gasGap, author in product([ "MMS", "MML"], [-2,-1,1,2], range(1,9), [1,2],range(1,5), [66]):
        constant = args.scaleError* 0.0615 + args.addErrorMMClusterTime
        linear =  args.scaleError * 0
        quad =  args.scaleError * 0.275
 
        minStrip = 1
        maxStrip = (5 if abs(stationEta)==1 else 3)*1024
        
        calibEntry = dict({"station": stationName, "eta":stationEta, "phi": stationPhi, "multilayer": multilayer, "gasGap": gasGap, "minStrip": minStrip,"maxStrip": maxStrip, "modelName":"tanThetaPolynomial" ,"modelPars": [constant, linear, quad], "clusterAuthor": author})
        logging.debug(calibEntry)
        if(stationEta<0):
            dbEntriesMMCSide.append(calibEntry)
        else:
            dbEntriesMMCSide.append(calibEntry)
    
    
    dbEntriessTGCASide = []
    dbEntriessTGCCSide = []
    maxStripStgc = dict({"QL1": 408 ,"QL2": 366,"QL3": 353,"QS1": 406, "QS2": 365, "QS3": 307})
    for stationName, stationEta, stationPhi, multilayer, gasGap in product([ "STL", "STS"], [-3, -2,-1,1,2,3], range(1,9), [1,2],range(1,5)):
        chamberKey = f"Q{stationName[2]}{abs(stationEta)}"
        minStrip = 1
        maxStrip = maxStripStgc[chamberKey]
 
        const = args.scaleError * 0.151
        linear = args.scaleError * 0
        quad = args.scaleError * 0.201
        
        author = 3 # centroid
        calibEntry = dict({"station": stationName, "eta":stationEta, "phi": stationPhi, "multilayer": multilayer, "gasGap": gasGap, "minStrip": minStrip,"maxStrip": maxStrip, "modelName":"tanThetaPolynomial" ,"modelPars": [const, linear, quad], "clusterAuthor": author})
        logging.debug(calibEntry)
        if(stationEta<0):
            dbEntriessTGCCSide.append(calibEntry)
        else:
            dbEntriessTGCASide.append(calibEntry)
        
        author = 4 #caruana 
        const = args.scaleError * 0.175
        linear = args.scaleError * 0
        quad = args.scaleError * 0.201
        calibEntry = dict({"station": stationName, "eta":stationEta, "phi": stationPhi, "multilayer": multilayer, "gasGap": gasGap, "minStrip": minStrip,"maxStrip": maxStrip, "modelName": "tanThetaPolynomial" ,"modelPars": [const, linear, quad], "clusterAuthor": author})
        logging.debug(calibEntry)
        if(stationEta<0):
            dbEntriessTGCCSide.append(calibEntry)
        else:
            dbEntriessTGCASide.append(calibEntry)
 
 
 
    outFileName = f"{args.outFilePreFix}_scale_x{args.scaleError}_mmConstOffset_{args.addErrorMMClusterTime}.json"
    with open(outFileName,'w') as outStream:
        json.dump(dbEntriesMMASide + dbEntriesMMCSide  +dbEntriessTGCASide + dbEntriessTGCCSide, outStream)
    
    
    outFileName = f"{args.outFilePreFix}_MMonly_ASide_scale_x{args.scaleError}_mmConstOffset_{args.addErrorMMClusterTime}.json"
    with open(outFileName,'w') as outStream:
        json.dump(dbEntriesMMASide, outStream)
    
    outFileName = f"{args.outFilePreFix}_MMonly_CSide_scale_x{args.scaleError}_mmConstOffset_{args.addErrorMMClusterTime}.json"
    with open(outFileName,'w') as outStream:
        json.dump(dbEntriesMMCSide, outStream)
    
    outFileName = f"{args.outFilePreFix}_sTGConly_ASide_scale_x{args.scaleError}.json"
    with open(outFileName,'w') as outStream:
        json.dump(dbEntriessTGCASide, outStream)
    
    outFileName = f"{args.outFilePreFix}_sTGConly_CSide_scale_x{args.scaleError}.json"
    with open(outFileName,'w') as outStream:
        json.dump(dbEntriessTGCCSide, outStream)