MuonAlignmentAlgorithms/scripts/alignCSCRings.py

0001 #! /usr/bin/env python3
0002
0003 from __future__ import print_function
0004 import re,os,sys,shutil
0005 import optparse
0006
0007 from mutypes import *
0008
0009 execfile("plotscripts.py")
0010
0011 ROOT.gROOT.SetBatch(1);
0012
0013 ######################################################
0014 # To parse commandline args
0015
0016 usage='%prog [options]\n'+\
0017   'a script to run CSC ring alignment procedure'+\
0018   'It is designed to run right after alignmentValidation.py script that was run with --map option. '
0019
0020 parser=optparse.OptionParser(usage)
0021
0022 parser.add_option("-l", "--runLabel",
0023   help="[REQUIRED] label to use for a run",
0024   type="string",
0025   default='',
0026   dest="runLabel")
0027
0028 parser.add_option("-d", "--dir",
0029   help="[REQUIRED] directory where tmp_test_results_map__{runLabel}.pkl fit results file is located",
0030   type="string",
0031   default='',
0032   dest="dir")
0033
0034 parser.add_option("-x", "--xml",
0035   help="[REQUIRED] xml file with input geometry",
0036   type="string",
0037   default='',
0038   dest="xml")
0039
0040 parser.add_option("--ring2only",
0041   help="if invoked, use only ring 2 results to align all rings in corresponding disks",
0042   action="store_true",
0043   dest="ring2only")
0044
0045 options,args=parser.parse_args()
0046
0047
0048 if options.runLabel=='' or options.dir=='' or options.xml=='':
0049   print("\nOne or more of REQUIRED options is missing!\n")
0050   parser.print_help()
0051   sys.exit()
0052
0053 allOptions = "-l "+options.runLabel+" -i "+options.dir+" -x "+options.xml
0054 #if options.diagnostic: allOptions += " --diagnostic"
0055 print(sys.argv[0]+" "+allOptions)
0056
0057 pwd = str(os.getcwd())
0058
0059 if not os.access(options.dir,os.F_OK):
0060   print("Directory " + options.dir + " does not exist. Exiting...")
0061   sys.exit()
0062 os.chdir(options.dir)
0063
0064 if not loadTestResultsMap(options.runLabel):
0065   print("Cant open pickle file with mapplots fit results. Exiting...")
0066   sys.exit()
0067
0068
0069 xml_corr = {}
0070
0071 print("       \tdX(mm)   \t dY(mm)   \tdPhiZ(mrad)")
0072 for endcap in CSC_TYPES:
0073   for station in endcap[2]:
0074     for ring in station[2]:
0075       if ring[1]=="ALL": continue
0076       # skip ME4/2 for now
0077       if station[1]=="4" and ring[1]=="2": continue
0078
0079       ring_id = "%s%s/%s" % (endcap[0], station[1],ring[1])
0080
0081       if ring_id in MAP_RESULTS_FITSIN:
0082         postal_address = idToPostalAddress(ring_id+'/01')
0083
0084         fits = MAP_RESULTS_FITSIN[ring_id]
0085         d_x, de_x = fits['sin'][0]/10., fits['sin'][1]/10.
0086         d_y, de_y = -fits['cos'][0]/10., fits['cos'][1]/10.
0087         d_phiz, de_phiz = -fits['a'][0]/10./signConventions[postal_address][3], fits['a'][1]/10./signConventions[postal_address][3]
0088
0089         print("%s \t%+.2f+-%.2f  \t%+.2f+-%.2f \t%+.2f+-%.2f" % (ring_id, d_x*10 , de_x*10, d_y*10 , de_y*10 , d_phiz*1000, de_phiz*1000))
0090
0091         e = endcap[3]
0092         s = station[1]
0093         r = ring[1]
0094         xml_corr[ring_id] = "<setposition relativeto=\"none\" x=\"%(d_x)s\" y=\"%(d_y)s\" phiz=\"%(d_phiz)s\" />" % vars()
0095
0096 xml_str = """
0097 """
0098 for endcap in CSC_TYPES:
0099   for station in endcap[2]:
0100     for ring in station[2]:
0101       if ring[1]=="ALL": continue
0102       # skip ME4/2 for now
0103       #if station[1]=="4" and ring[1]=="2": continue
0104
0105       r_with_corr = ring[1]
0106       s_with_corr = station[1]
0107       # use ME4/1 for aligning ME4/2
0108       if station[1]=="4" and ring[1]=="2": r_with_corr = "1"
0109       # ring 2 only option
0110       if options.ring2only : r_with_corr = "2"
0111       if options.ring2only and station[1]=="3": s_with_corr = "2"
0112       # no matter what, always use ME11 to correct ME11
0113       if station[1]=="1" and ring[1]=="1": r_with_corr = "1"
0114
0115       # for jim's BH cross-check
0116       #if station[1]=="1" and ring[1]=="3": r_with_corr = "2"
0117       #if station[1]=="2" or  station[1]=="3":
0118       #   r_with_corr = "1"
0119       #   s_with_corr = "2"
0120
0121       ring_id = "%s%s/%s" % (endcap[0], s_with_corr, r_with_corr)
0122
0123       if ring_id in xml_corr:
0124         corr = xml_corr[ring_id]
0125         e = endcap[3]
0126         s = station[1]
0127         r = ring[1]
0128         xml_str += """<operation>
0129   <CSCRing endcap=\"%(e)s\" station=\"%(s)s\" ring=\"%(r)s\" />
0130   %(corr)s
0131 </operation>
0132
0133 """ % vars()
0134         # if it's ME1/1, move ME1/4 the same amount as well
0135         if s=="1" and r=="1":
0136           xml_str += """<operation>
0137   <CSCRing endcap=\"%(e)s\" station=\"%(s)s\" ring=\"4\" />
0138   %(corr)s
0139 </operation>
0140
0141 """ % vars()
0142
0143 print(xml_str)
0144 xml_str += "</MuonAlignment>"
0145
0146 os.chdir(pwd)
0147
0148 ff = open(options.xml+".ring",mode="w")
0149 print(xml_str, file=ff)
0150 ff.close()
0151
0152 os.system('grep -v "</MuonAlignment>" %s > %s' % (options.xml, options.xml+".tmp"))
0153 os.system('cat %s %s > %s' % (options.xml+".tmp", options.xml+".ring", options.xml+".ring.xml") )
0154 os.system('rm %s %s' % (options.xml+".tmp", options.xml+".ring") )