LArMissingFeb2Ascii.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
from AthenaConfiguration.ComponentFactory import CompFactory
 
def LArMissingFeb2AsciiCfg(flags,OutputFile,dbname="LAR_OFL",folder=None,tag=None,summaryfile=""):
    from LArGeoAlgsNV.LArGMConfig import LArGMCfg
    result=LArGMCfg(flags)
 
    #Setup regular cabling
    from LArCabling.LArCablingConfig import LArOnOffIdMappingCfg
    result.merge(LArOnOffIdMappingCfg(flags))
 
    
    if folder is None:
        if dbname in ("LAR","LAR_ONL") or flags.Input.isMC:
            folder="/LAR/BadChannels/MissingFEBs"
        else: 
            folder="/LAR/BadChannelsOfl/MissingFEBs"
 
    if tag is not None:
        if not tag.startswith("LAR"):
            if not tag.startswith("-"): tag= "-"+tag
            tag="".join(folder.split("/"))+tag
   
        print("Tag=",tag)
 
    if 'MissingFEBs' in folder:
       from LArBadChannelTool.LArBadChannelConfig import LArBadFebCfg
       result.merge(LArBadFebCfg(flags, tag=tag, dbname=dbname))
       ReadKey='LArBadFeb'
    elif 'KnownBADFEBs' in folder:
       from LArBadChannelTool.LArBadFebsConfig import LArKnownBadFebCfg
       result.merge(LArKnownBadFebCfg(flags,tag=tag, dbname=dbname))
       ReadKey='LArKnownBadFEBs'
    elif 'KnownMNBFEBs' in folder:   
       from LArBadChannelTool.LArBadFebsConfig import LArKnownMNBFebCfg
       result.merge(LArKnownMNBFebCfg(flags,tag=tag, dbname=dbname))
       ReadKey='LArKnownMNBFEBs'
    else:
       print('Unknown folder: ',folder,' exiting !!!')
       import sys
       sys.exit(-1)
 
    theLArMF2Ascii=CompFactory.LArBadFeb2Ascii(FileName=OutputFile,BFKey=ReadKey)
    result.addEventAlgo(theLArMF2Ascii)
 
    return result
 
 
 
if __name__=="__main__":
    import sys,argparse
    parser= argparse.ArgumentParser()
    parser.add_argument("--loglevel", default=None, help="logging level (ALL, VERBOSE, DEBUG,INFO, WARNING, ERROR, or FATAL")
    parser.add_argument("-r","--runnumber",default=0x7fffffff, type=int, help="run number to query the DB")
    parser.add_argument("-l","--lbnumber",default=1, type=int, help="LB number to query the DB")
    parser.add_argument("-d","--database",default="LAR_OFL", help="Database name or sqlite file name")
    parser.add_argument("-o","--output",default="bc_output.txt", help="output file name")
    parser.add_argument("-f","--folder",default=None, help="database folder to read")
    parser.add_argument("-t","--tag",default=None, help="folder-level tag to read")
    parser.add_argument("-s","--summary",default="", help="Executive summary file")
    parser.add_argument("--MC", action='store_true', default=False, help="Work on MC DB")
 
    (args,leftover)=parser.parse_known_args(sys.argv[1:])
 
    if len(leftover)>0:
        print("ERROR, unhandled argument(s):",leftover)
        sys.exit(-1)
    
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from LArCalibProcessing.LArCalibConfigFlags import addLArCalibFlags
    flags=initConfigFlags()
    addLArCalibFlags(flags)
    flags.Input.Files = []
    flags.Input.isMC = args.MC
    flags.IOVDb.DatabaseInstance="OFLP200" if args.MC else "CONDBR2"
    flags.LAr.doAlign=False
    flags.Input.RunNumbers=[args.runnumber]
    flags.IOVDb.GlobalTag="OFLCOND-MC21-SDR-RUN3-11" if args.MC else "CONDBR2-ES1PA-2023-02"
    if args.MC:
       from Campaigns.Utils import Campaign
       flags.Input.MCCampaign = Campaign.MC21a
    from AthenaConfiguration.TestDefaults import defaultGeometryTags
    flags.GeoModel.AtlasVersion=defaultGeometryTags.RUN3
    
    if args.loglevel:
        from AthenaCommon import Constants
        if hasattr(Constants,args.loglevel):
            flags.Exec.OutputLevel=getattr(Constants,args.loglevel)
        else:
            raise ValueError("Unknown log-level, allowed values are ALL, VERBOSE, DEBUG,INFO, WARNING, ERROR, FATAL")
 
    if ".db" in args.database:
      flags.IOVDb.SqliteInput=args.database
      if args.folder is None:
         flags.IOVDb.SqliteFolders=("/LAR/BadChannelsOfl/MissingFEBs")    
      else:   
         flags.IOVDb.SqliteFolders=(args.folder)    
    flags.lock()
 
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    cfg=MainServicesCfg(flags)
    #MC Event selector since we have no input data file
    from McEventSelector.McEventSelectorConfig import McEventSelectorCfg
    cfg.merge(McEventSelectorCfg(flags,
                                 FirstLB=args.lbnumber,
                                 EventsPerRun      = 1,
                                 FirstEvent        = 1,
                                 InitialTimeStamp  = 0,
                                 TimeStampInterval = 1))
 
    cfg.merge(LArMissingFeb2AsciiCfg(flags,args.output, 
                                     dbname=args.database,
                                     folder=args.folder,
                                     tag=args.tag,
                                     summaryfile=args.summary))
    
 
 
    sc=cfg.run(1)
    if sc.isSuccess():
        sys.exit(0)
    else:
        sys.exit(1)