Alignment/MuonAlignmentAlgorithms/scripts/convertSQLiteXML.py - master - githubmirror/cmssw

Alignment/MuonAlignmentAlgorithms/scripts/convertSQLiteXML.py
152 строки5 KB
Shahzad Malik Muzaffar
[ALCA-DB] py2/3 compatibility:drop use of __future__
22 ноя 2024, 20:15
22 ноя 2024, 20:15c26340c
#! /usr/bin/env python3

import os, sys, optparse, math

prog = sys.argv[0]

usage = """%(prog)s INPUT_FILE OUTPUT_FILE [--noChambers] [--noLayers] [--ringsOnly] [--relativeTo ideal|none]

performs either sqlite-->xml or xml-->sqlite conversion following the documentation at 
https://twiki.cern.ch/twiki/bin/view/CMS/SWGuideMuonGeometryConversion

Arguments:

INPUT_FILE      is either .db SQLite or .xml file that should be converted
OUTPUT_FILE     is either .xml or .db output file, the result of conversion

Options for sqlite-->xml conversion:

--noChambers    if present, no chambers info would be written into xml
--noLayers      if present, no layers (and no DT superlayers) info would be written into xml
--relativeTo X  by default, xml conversion is done relative to ideal DDD description,
                if "none" is specified, absolute positions would be written into xml
--ringsOnly     special flag for xml dumping of CSC ring structure only, it automatically
                turns off all DTs and also CSC's chambers and layers on output and coordinates
                are relative "to none"
--runNumber N   Use run #N to extract xml from necessary IOV
""" % vars()

if len(sys.argv) < 3:
  print("Too few arguments!\n\n"+usage)
  sys.exit()

parser=optparse.OptionParser(usage)

parser.add_option("--noChambers",
  help="if present, no chambers info would be written into xml",
  action="store_true",
  default=False,
  dest="noChambers")

parser.add_option("--noLayers",
  help="if present, no layers (and no DT superlayers) info would be written into xml",
  action="store_true",
  default=False,
  dest="noLayers")

parser.add_option("-l", "--relativeTo",
  help="by default, xml conversion is done relative to ideal DDD description, if \"none\" is specified, absolute positions would be written into xml",
  type="string",
  default='ideal',
  dest="relativeTo")

parser.add_option("--ringsOnly",
  help="special flag for xml dumping of CSC ring structure only, it automatically turns off all DTs and also CSC's chambers and layers",
  action="store_true",
  default=False,
  dest="ringsOnly")

parser.add_option("-r", "--runNumber",
  help="Use run #N to extract xml from necessary IOV (default is 1)",
  type="int",
  default=1,
  dest="runNumber")

parser.add_option("--gprcdconnect",
  help="connect string for GlobalPositionRcd (frontier://... or sqlite_file:...). The defailt is a trivial/inert GPR",
  type="string",
  default="sqlite_file:inertGlobalPositionRcd.db",
  dest="gprcdconnect")

parser.add_option("--gprcd",
  help="name of GlobalPositionRcd tag",
  type="string",
  default="inertGlobalPositionRcd",
  dest="gprcd")


options, args = parser.parse_args(sys.argv[3:])

supRings="True"
if options.ringsOnly: supRings="False"
supChambers="False"
if options.noChambers or options.ringsOnly: supChambers="True"
supLayers="False"
if options.noLayers or options.ringsOnly: supLayers="True"

relativeTo=options.relativeTo
if options.ringsOnly: relativeTo="none"

runNumber = options.runNumber
gprcdconnect = options.gprcdconnect
gprcd = options.gprcd

theInputFile = sys.argv[1]
theOutputFile = sys.argv[2]

ok = False

if theInputFile[-4:]==".xml" and theOutputFile[-3:]==".db":
  ok = True
  file("tmp_converter_cfg.py","w").write("""# xml2sqlite conversion
from Alignment.MuonAlignment.convertXMLtoSQLite_cfg import *
process.MuonGeometryDBConverter.fileName = "%(theInputFile)s"
process.PoolDBOutputService.connect = "sqlite_file:%(theOutputFile)s"
process.inertGlobalPositionRcd.connect = "%(gprcdconnect)s"
process.inertGlobalPositionRcd.toGet =  cms.VPSet(cms.PSet(record = cms.string('GlobalPositionRcd'), tag = cms.string('%(gprcd)s')))

""" % vars())

if theInputFile[-3:]==".db" and theOutputFile[-4:]==".xml":
  ok = True
  file("tmp_converter_cfg.py","w").write("""# sqlite2xml conversion
from Alignment.MuonAlignment.convertSQLitetoXML_cfg import *

process.source = cms.Source("EmptySource",
    numberEventsInRun = cms.untracked.uint32(1),
    firstRun = cms.untracked.uint32(%(runNumber)d)
)

process.inertGlobalPositionRcd.connect = "%(gprcdconnect)s"
process.inertGlobalPositionRcd.toGet =  cms.VPSet(cms.PSet(record = cms.string('GlobalPositionRcd'), tag = cms.string('%(gprcd)s')))

process.PoolDBESSource.connect = "sqlite_file:%(theInputFile)s"
process.MuonGeometryDBConverter.outputXML.fileName = "%(theOutputFile)s"

process.MuonGeometryDBConverter.outputXML.relativeto = "%(relativeTo)s"

process.MuonGeometryDBConverter.outputXML.suppressDTBarrel = True
process.MuonGeometryDBConverter.outputXML.suppressDTWheels = True
process.MuonGeometryDBConverter.outputXML.suppressDTStations = True
process.MuonGeometryDBConverter.outputXML.suppressDTChambers = %(supChambers)s
process.MuonGeometryDBConverter.outputXML.suppressDTSuperLayers = %(supLayers)s
process.MuonGeometryDBConverter.outputXML.suppressDTLayers = %(supLayers)s

process.MuonGeometryDBConverter.outputXML.suppressCSCEndcaps = True
process.MuonGeometryDBConverter.outputXML.suppressCSCStations = True
process.MuonGeometryDBConverter.outputXML.suppressCSCRings = %(supRings)s
process.MuonGeometryDBConverter.outputXML.suppressCSCChambers = %(supChambers)s
process.MuonGeometryDBConverter.outputXML.suppressCSCLayers = %(supLayers)s

""" % vars())

if not ok:
  print(usage)
  sys.exit()

exit_code = os.system("cmsRun tmp_converter_cfg.py")

if exit_code>0:
  print("problem: cmsRun exited with code:", exit_code)
else: 
  os.system("rm tmp_converter_cfg.py")