#
import sys
sys.path.append("..")
try:
import ROOT
except:
print "It seems you do not have ROOT setup, or PyROOT is not enabled in your ROOT distribution. Sorry."
sys.exit()
from GeometryROOT import GeometryROOT
from GeometryEngine import Geometry,GeometryEngine
import Write_HPS_ecal
gen= GeometryEngine("hps")
Write_HPS_ecal.calculate_ecal_mother_geometry(gen,mother="root")
Write_HPS_ecal.calculate_ecal_geometry(gen)
Write_HPS_ecal.calculate_ecal_vacuum_geometry(gen,style=0)
Write_HPS_ecal.calculate_ecal_crystalbox_geometry(gen)
Write_HPS_ecal.calculate_ecal_coolingsys_geometry(gen)
rr = GeometryROOT()
rr.Create_root_volume()
rr.Build_volumes(gen)
browse = ROOT.TBrowser()
rr.Draw("ogl")
print "You can now access the geometry through the ROOT browser, and with the 'rr' object."
print "ROOT may cause Python to crash upon exit.... no worries, just annoying"
def main(argv=None):
################################################################################################################################
if argv is None:
argv = sys.argv
parser = argparse.ArgumentParser(
description="""Master HPS geometry writer.
This code will create TEXT tables for GEMC, or fill MySQL tables for GEMC.
The options allow you to choose from different geometry combinations, and allow you to choose
for a single file (MySQL table) or one file (table) per system.""",
epilog="""For more information, or errors, please email: [email protected] """)
parser.add_argument('-n','--name',help='Name of the detector (default=hps)',default="hps")
parser.add_argument('-v','--variation',help='Variation for the detector (default=depends on other choices!)',default=None)
parser.add_argument('-i','--identity',type=int,help='Identity or id for the detector variation. Set to -1 to auto-increment. (default=1)',default=1)
parser.add_argument('-f','--fast',action="store_true",help='Fast variation of beamline, use constant magnetic fields.')
parser.add_argument('-S','--nofrascati',action="store_true",help='Do not include the Frascati dipole magnets')
parser.add_argument('-F','--fullmuon',action="store_true",help='Use the Full muon detector instead of the Lite version')
parser.add_argument('-T','--testmuon',action="store_true",help='Use the Test muon detector instead of the Lite version')
parser.add_argument('-s','--shift', action="store_true",help='Shift the geometry so that the target is at (0,0,0)')
parser.add_argument('-a','--align', action="store_true",help='Add in the flux alignment targets.')
parser.add_argument('-t','--txt',action='store_true',help='Produce a TEXT file for output (default)')
parser.add_argument('-m','--mysql',action='store_true',help='Produce a MySQL table for output. You must also specify -host -database -user -password')
parser.add_argument('-H','--host',help='Name of the database host computer',default="localhost")
parser.add_argument('-D','--database',help='Name of the database',default="hps_2014")
parser.add_argument('-u','--user',help='User name for the database',default="clasuser")
parser.add_argument('-p','--passwd',help='Password for connecting to the database',default="")
parser.add_argument('-M','--multi',action='store_true',help='Create separate tables for each detector component.')
# parser.add_argument('-t','--table',help='Table name override for storing the MySQL table.',default=None)
parser.add_argument('-q','--quiet',action="store_true",help='Tries to suppress extra output (depends also on the included Write engines!)')
parser.add_argument('-d','--debug',action="count",help='Increase debug level by one.')
args = parser.parse_args(argv[1:])
if args.debug:
print "Debug level is: "+str(args.debug)
if not args.variation:
if args.fast:
args.variation="fast"
else:
args.variation="original"
if args.nofrascati:
args.variation=args.variation+"-short"
if args.fullmuon:
args.variation=args.variation+"-full"
if args.testmuon:
args.variation=args.variation+"-test"
if args.shift:
args.variation=args.variation+"-shift"
if not args.mysql and not args.txt:
args.txt=True
elif args.mysql:
import MySQLdb
if not args.quiet:
if args.multi:
print "Creating multiple tables for the detectors with variation label = "+args.variation + " and id ="+str(args.identity)
else:
print "Creating tables for detector name = "+args.name + " with variation label = "+args.variation + " and id ="+str(args.identity)
from GeometryEngine import Geometry, GeometryEngine, Sensitive_Detector
sys.path.append("beamline")
sys.path.append("ecal")
sys.path.append("muon")
sys.path.append("svt")
import Write_HPS_beamline
if args.shift:
print "Shifting Geometry."
Write_HPS_beamline.Alignment_Choice=1
import Write_HPS_ecal
import Write_HPS_svt
if args.fullmuon:
import Write_HPS_muon
elif args.testmuon:
import Write_HPS_muon_side_flux_test
else:
import Write_HPS_muon_lite
gen = []
if args.multi:
gen.append(GeometryEngine(Write_HPS_beamline.Standard_Table_Name,variation=args.variation,iden=args.identity))
else:
gen.append(GeometryEngine(args.name,variation=args.variation,iden=args.identity))
gen[len(gen)-1].debug = args.debug
Write_HPS_beamline.calculate_dipole_geometry(gen[len(gen)-1])
if not args.nofrascati:
Write_HPS_beamline.calculate_frascati_magnets(gen[len(gen)-1])
if args.align:
Write_HPS_beamline.calculate_alignment_targets(gen[len(gen)-1])
Write_HPS_beamline.calculate_ps_vacuum(gen[len(gen)-1])
if not args.fullmuon:
Write_HPS_beamline.calculate_muon_lite_vacuum(gen[len(gen)-1])
active=0
if args.align:
active=1
Write_HPS_beamline.calculate_target_geometry(gen[len(gen)-1],radlen=0.125,active=active)
if args.align:
Write_HPS_beamline.calculate_alignment_targets(gen[len(gen)-1])
if args.multi:
gen.append(GeometryEngine(Write_HPS_ecal.Standard_Table_Name,variation=args.variation,iden=args.identity))
gen[len(gen)-1].debug = args.debug
Write_HPS_ecal.calculate_ecal_mother_geometry(gen[len(gen)-1])
Write_HPS_ecal.calculate_ecal_geometry(gen[len(gen)-1])
Write_HPS_ecal.calculate_ecal_vacuum_geometry(gen[len(gen)-1])
# Write_HPS_ecal.calculate_ecal_box_geometry(gen[len(gen)-1])
Write_HPS_ecal.calculate_ecal_crystalbox_geometry(gen[len(gen)-1])
Write_HPS_ecal.calculate_ecal_coolingsys_geometry(gen[len(gen)-1])
Write_HPS_svt.calculate_svt_hit_planes(gen[len(gen)-1])
if args.fullmuon:
if args.multi:
gen.append(GeometryEngine(Write_HPS_muon.Standard_Table_Name,variation=args.variation,iden=args.identity))
gen[len(gen)-1].debug = args.debug
Write_HPS_muon_lite.calculate_muon_lite_geometry(gen[len(gen)-1])
elif args.testmuon:
if args.multi:
gen.append(GeometryEngine(Write_HPS_muon_side_flux_test.Standard_Table_Name,variation=args.variation,iden=args.identity))
gen[len(gen)-1].debug = args.debug
Write_HPS_muon_side_flux_test.calculate_muon_side_test_geometry(gen[len(gen)-1])
else:
if args.multi:
gen.append(GeometryEngine(Write_HPS_muon_lite.Standard_Table_Name,variation=args.variation,iden=args.identity))
gen[len(gen)-1].debug = args.debug
Write_HPS_muon_lite.calculate_muon_lite_geometry(gen[len(gen)-1])
for g in gen:
if args.mysql:
if not args.quiet: print "Writing tables to MySQL for "+g._Detector+" with variation="+args.variation
g.MySQL_OpenDB(args.host,args.user,args.passwd,args.database)
g.MySQL_Write()
if args.txt:
if not args.quiet: print "Writing tables to TEXT for "+g._Detector+" with variation="+args.variation
g.TXT_Write()
import sys
sys.path.append("ecal") # So that Python can find the ECAL geometries.
from GeometryEngine import GeometryEngine, Geometry, Sensitive_Detector
import ROOT
from GeometryROOT import GeometryROOT
import Write_HPS_ecal # import and create the ECAL geometry
geo = GeometryEngine("ecal") # Fire up the GeometryEngine.
Write_HPS_ecal.calculate_ecal_mother_geometry(geo,mother="root")
Write_HPS_ecal.calculate_ecal_box_geometry(geo)
Write_HPS_ecal.calculate_ecal_geometry(geo)
Write_HPS_ecal.calculate_ecal_vacuum_geometry(geo)
rr = GeometryROOT() # Startup the ROOT Renderer.
rr.debug=2 # Show each volume as it is created. Useful for debugging.
rr.Create_root_volume() # Create the "root" volume, or the Hall. This is invisible.
rr.Build_volumes(geo) # Build the "geo" of the ECAL
# You can add additional "geo"s by calling this again.
browse=ROOT.TBrowser() # The browser will show a table of the geometries under "GEMC"
rr.Draw("ogl") # Draw the whole thing using OpenGL (leave out option to draw wireframe)
print "Warning: ROOT will cause a crash when you exit. Blame the ROOT team. "
from GeometryROOT import GeometryROOT
from GeometryEngine import Geometry,GeometryEngine
import Write_HPS_beamline
import Write_HPS_ecal
import Write_HPS_muon_lite
gen= GeometryEngine("hps")
# gen.TXT_Read_Geometry("beamline__geometry_original.txt")
Write_HPS_beamline.calculate_dipole_geometry(gen)
Write_HPS_beamline.calculate_frascati_magnets(gen)
Write_HPS_beamline.calculate_ps_vacuum(gen)
Write_HPS_beamline.calculate_target_geometry(gen,radlen=0.125,active=1)
Write_HPS_beamline.calculate_muon_lite_vacuum(gen)
#Write_HPS_beamline.calculate_alignment_targets(gen)
Write_HPS_ecal.calculate_ecal_mother_geometry(gen)
Write_HPS_ecal.calculate_ecal_geometry(gen)
Write_HPS_ecal.calculate_ecal_vacuum_geometry(gen)
Write_HPS_ecal.calculate_ecal_crystalbox_geometry(gen)
Write_HPS_ecal.calculate_ecal_coolingsys_geometry(gen)
# Write_HPS_ecal.calculate_ecal_box_geometry(gen)
# Write_HPS_muon_lite.calculate_muon_lite_geometry(gen)
rr = GeometryROOT()
rr.Create_root_volume()
rr.Build_volumes(gen)
browse = ROOT.TBrowser()
rr.Draw("ogl")
print("You can now access the geometry through the ROOT browser, and with the 'rr' object.")
print("ROOT may cause Python to crash upon exit.... no worries, just annoying")
def main(argv=None):
################################################################################################################################
if argv is None:
argv = sys.argv
parser = argparse.ArgumentParser(
description="""Master HPS geometry writer.
This code will create TEXT tables for GEMC, or fill MySQL tables for GEMC.
The options allow you to choose from different geometry combinations, and allow you to choose
for a single file (MySQL table) or one file (table) per system.""",
epilog=
"""For more information, or errors, please email: [email protected] """
)
parser.add_argument('-n',
'--name',
help='Name of the detector (default=hps)',
default="hps")
parser.add_argument(
'-v',
'--variation',
help='Variation for the detector (default=depends on other choices!)',
default=None)
parser.add_argument(
'-i',
'--identity',
type=int,
help=
'Identity or id for the detector variation. Set to -1 to auto-increment. (default=1)',
default=1)
parser.add_argument(
'-f',
'--fast',
action="store_true",
help='Fast variation of beamline, use constant magnetic fields.')
parser.add_argument('-S',
'--nofrascati',
action="store_true",
help='Do not include the Frascati dipole magnets')
parser.add_argument(
'-F',
'--fullmuon',
action="store_true",
help='Use the Full muon detector instead of the Lite version')
parser.add_argument(
'-T',
'--testmuon',
action="store_true",
help='Use the Test muon detector instead of the Lite version')
parser.add_argument(
'-s',
'--shift',
action="store_true",
help='Shift the geometry so that the target is at (0,0,0)')
parser.add_argument('-a',
'--align',
action="store_true",
help='Add in the flux alignment targets.')
parser.add_argument('-t',
'--txt',
action='store_true',
help='Produce a TEXT file for output (default)')
parser.add_argument(
'-m',
'--mysql',
action='store_true',
help=
'Produce a MySQL table for output. You must also specify -host -database -user -password'
)
parser.add_argument('-H',
'--host',
help='Name of the database host computer',
default="localhost")
parser.add_argument('-D',
'--database',
help='Name of the database',
default="hps_2014")
parser.add_argument('-u',
'--user',
help='User name for the database',
default="clasuser")
parser.add_argument('-p',
'--passwd',
help='Password for connecting to the database',
default="")
parser.add_argument(
'-M',
'--multi',
action='store_true',
help='Create separate tables for each detector component.')
# parser.add_argument('-t','--table',help='Table name override for storing the MySQL table.',default=None)
parser.add_argument(
'-q',
'--quiet',
action="store_true",
help=
'Tries to suppress extra output (depends also on the included Write engines!)'
)
parser.add_argument('-d',
'--debug',
action="count",
help='Increase debug level by one.')
args = parser.parse_args(argv[1:])
if args.debug:
print "Debug level is: " + str(args.debug)
if not args.variation:
if args.fast:
args.variation = "fast"
else:
args.variation = "original"
if args.nofrascati:
args.variation = args.variation + "-short"
if args.fullmuon:
args.variation = args.variation + "-full"
if args.testmuon:
args.variation = args.variation + "-test"
if args.shift:
args.variation = args.variation + "-shift"
if not args.mysql and not args.txt:
args.txt = True
elif args.mysql:
import MySQLdb
if not args.quiet:
if args.multi:
print "Creating multiple tables for the detectors with variation label = " + args.variation + " and id =" + str(
args.identity)
else:
print "Creating tables for detector name = " + args.name + " with variation label = " + args.variation + " and id =" + str(
args.identity)
from GeometryEngine import Geometry, GeometryEngine, Sensitive_Detector
sys.path.append("beamline")
sys.path.append("ecal")
sys.path.append("muon")
sys.path.append("svt")
import Write_HPS_beamline
if args.shift:
print "Shifting Geometry."
Write_HPS_beamline.Alignment_Choice = 1
import Write_HPS_ecal
import Write_HPS_svt
if args.fullmuon:
import Write_HPS_muon
elif args.testmuon:
import Write_HPS_muon_side_flux_test
else:
import Write_HPS_muon_lite
gen = []
if args.multi:
gen.append(
GeometryEngine(Write_HPS_beamline.Standard_Table_Name,
variation=args.variation,
iden=args.identity))
else:
gen.append(
GeometryEngine(args.name,
variation=args.variation,
iden=args.identity))
gen[len(gen) - 1].debug = args.debug
Write_HPS_beamline.calculate_dipole_geometry(gen[len(gen) - 1])
if not args.nofrascati:
Write_HPS_beamline.calculate_frascati_magnets(gen[len(gen) - 1])
if args.align:
Write_HPS_beamline.calculate_alignment_targets(gen[len(gen) - 1])
Write_HPS_beamline.calculate_ps_vacuum(gen[len(gen) - 1])
if not args.fullmuon:
Write_HPS_beamline.calculate_muon_lite_vacuum(gen[len(gen) - 1])
active = 0
if args.align:
active = 1
Write_HPS_beamline.calculate_target_geometry(gen[len(gen) - 1],
radlen=0.125,
active=active)
if args.align:
Write_HPS_beamline.calculate_alignment_targets(gen[len(gen) - 1])
if args.multi:
gen.append(
GeometryEngine(Write_HPS_ecal.Standard_Table_Name,
variation=args.variation,
iden=args.identity))
gen[len(gen) - 1].debug = args.debug
Write_HPS_ecal.calculate_ecal_mother_geometry(gen[len(gen) - 1])
Write_HPS_ecal.calculate_ecal_geometry(gen[len(gen) - 1])
Write_HPS_ecal.calculate_ecal_vacuum_geometry(gen[len(gen) - 1])
# Write_HPS_ecal.calculate_ecal_box_geometry(gen[len(gen)-1])
Write_HPS_ecal.calculate_ecal_crystalbox_geometry(gen[len(gen) - 1])
Write_HPS_ecal.calculate_ecal_coolingsys_geometry(gen[len(gen) - 1])
Write_HPS_svt.calculate_svt_hit_planes(gen[len(gen) - 1])
if args.fullmuon:
if args.multi:
gen.append(
GeometryEngine(Write_HPS_muon.Standard_Table_Name,
variation=args.variation,
iden=args.identity))
gen[len(gen) - 1].debug = args.debug
Write_HPS_muon_lite.calculate_muon_lite_geometry(gen[len(gen) - 1])
elif args.testmuon:
if args.multi:
gen.append(
GeometryEngine(
Write_HPS_muon_side_flux_test.Standard_Table_Name,
variation=args.variation,
iden=args.identity))
gen[len(gen) - 1].debug = args.debug
Write_HPS_muon_side_flux_test.calculate_muon_side_test_geometry(
gen[len(gen) - 1])
else:
if args.multi:
gen.append(
GeometryEngine(Write_HPS_muon_lite.Standard_Table_Name,
variation=args.variation,
iden=args.identity))
gen[len(gen) - 1].debug = args.debug
Write_HPS_muon_lite.calculate_muon_lite_geometry(gen[len(gen) - 1])
for g in gen:
if args.mysql:
if not args.quiet:
print "Writing tables to MySQL for " + g._Detector + " with variation=" + args.variation
g.MySQL_OpenDB(args.host, args.user, args.passwd, args.database)
g.MySQL_Write()
if args.txt:
if not args.quiet:
print "Writing tables to TEXT for " + g._Detector + " with variation=" + args.variation
g.TXT_Write()
from GeometryROOT import GeometryROOT
from GeometryEngine import Geometry, GeometryEngine
import Write_HPS_beamline
import Write_HPS_ecal
import Write_HPS_muon_lite
gen = GeometryEngine("hps")
# gen.TXT_Read_Geometry("beamline__geometry_original.txt")
Write_HPS_beamline.calculate_dipole_geometry(gen)
Write_HPS_beamline.calculate_frascati_magnets(gen)
Write_HPS_beamline.calculate_ps_vacuum(gen)
Write_HPS_beamline.calculate_target_geometry(gen, radlen=0.125, active=1)
Write_HPS_beamline.calculate_muon_lite_vacuum(gen)
#Write_HPS_beamline.calculate_alignment_targets(gen)
Write_HPS_ecal.calculate_ecal_mother_geometry(gen)
Write_HPS_ecal.calculate_ecal_geometry(gen)
Write_HPS_ecal.calculate_ecal_vacuum_geometry(gen)
Write_HPS_ecal.calculate_ecal_crystalbox_geometry(gen)
Write_HPS_ecal.calculate_ecal_coolingsys_geometry(gen)
# Write_HPS_ecal.calculate_ecal_box_geometry(gen)
# Write_HPS_muon_lite.calculate_muon_lite_geometry(gen)
rr = GeometryROOT()
rr.Create_root_volume()
rr.Build_volumes(gen)
browse = ROOT.TBrowser()
rr.Draw("ogl")
print "You can now access the geometry through the ROOT browser, and with the 'rr' object."
print "ROOT may cause Python to crash upon exit.... no worries, just annoying"