Skip to content

Commit

Permalink
[format] Format files with non-trivial print fixes
Browse files Browse the repository at this point in the history 
Format files that still had python2 print statements involving files. In
principle no behavior should change with the fixes, but since there are
no tests, it is hard to say
  • Loading branch information
@tmadlener
tmadlener committed Sep 25, 2024
1 parent 4b59f5c commit f41a9bd
Show file tree
Hide file tree
Showing 4 changed files with 836 additions and 601 deletions.
279 changes: 170 additions & 109 deletions ILD/compact/ILD_common_v02/writeAllILDCompactDescriptions.py
View file
Original file line number Diff line number Diff line change
@@ -1,44 +1,55 @@
def writeTopCompactXml( outfile, version, name, Large, Option, SolenoidMap, AntiDID, FwdFields ):

def writeTopCompactXml(
outfile, version, name, Large, Option, SolenoidMap, AntiDID, FwdFields
):
# defaults for option 0, 1
# si ecal
ecal_sl1=4
ecal_sl2=10
ecal_sl1 = 4
ecal_sl2 = 10
# ahcal
hcal_sl=3
if Option==2 or Option==4: # SDHCAL
hcal_sl=1
elif Option==3 or Option==4: # ScECAL
ecal_sl1=3
ecal_sl2=11
elif Option>1:
print 'ERROR: do not know about Option', Option
hcal_sl = 3
if Option == 2 or Option == 4: # SDHCAL
hcal_sl = 1
elif Option == 3 or Option == 4: # ScECAL
ecal_sl1 = 3
ecal_sl2 = 11
elif Option > 1:
print("ERROR: do not know about Option", Option)
return

if FwdFields!=250 and FwdFields!=500 and FwdFields!=1000 and FwdFields!=0:
print 'ERROR: do not know about FwdFields at energy', FwdFields
if FwdFields != 250 and FwdFields != 500 and FwdFields != 1000 and FwdFields != 0:
print("ERROR: do not know about FwdFields at energy", FwdFields)
return

outfile.write('<lccdd xmlns:compact="http://www.lcsim.org/schemas/compact/1.0"\n')
outfile.write(' xmlns:xs="http://www.w3.org/2001/XMLSchema"\n')
outfile.write(' xs:noNamespaceSchemaLocation="http://www.lcsim.org/schemas/compact/1.0/compact.xsd">\n')
outfile.write(' <info name="'+name+'"\n')
outfile.write(' title="ILD multi-technology model used for the optimisation"\n')
outfile.write(
' xs:noNamespaceSchemaLocation="http://www.lcsim.org/schemas/compact/1.0/compact.xsd">\n'
)
outfile.write(' <info name="' + name + '"\n')
outfile.write(
' title="ILD multi-technology model used for the optimisation"\n'
)
outfile.write(' author="F. Gaede, S.Lu, D.Jeans"\n')
outfile.write(' url="http://ilcsoft.desy.de"\n')
outfile.write(' status="experimental"\n')
outfile.write(' version="v'+version+'">\n')
outfile.write(' <comment>ILD simulation models used for detector optimisation </comment>\n')
outfile.write(' </info>\n')
outfile.write(' <includes>\n')
outfile.write(' version="v' + version + '">\n')
outfile.write(
" <comment>ILD simulation models used for detector optimisation </comment>\n"
)
outfile.write(" </info>\n")
outfile.write(" <includes>\n")
outfile.write(' <gdmlFile ref="../ILD_common_v02/elements.xml"/>\n')
outfile.write(' <gdmlFile ref="../ILD_common_v02/materials.xml"/>\n')
outfile.write(' </includes>\n')
outfile.write(' <define>\n')
outfile.write(" </includes>\n")
outfile.write(" <define>\n")
if Large:
outfile.write(' <include ref="../ILD_common_v02/top_defs_ILD_l5_v02.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/top_defs_ILD_l5_v02.xml"/>\n'
)
else:
outfile.write(' <include ref="../ILD_common_v02/top_defs_ILD_s5_v02.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/top_defs_ILD_s5_v02.xml"/>\n'
)
outfile.write(' <include ref="../ILD_common_v02/top_defs_common_v02.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/basic_defs.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/envelope_defs.xml"/>\n')
Expand All @@ -50,41 +61,69 @@ def writeTopCompactXml( outfile, version, name, Large, Option, SolenoidMap, Anti
outfile.write(' <include ref="../ILD_common_v02/hcal_defs.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/yoke_defs.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/services_defs.xml"/>\n')
outfile.write(' <include ref="${DD4hepINSTALL}/DDDetectors/compact/detector_types.xml"/>\n')
outfile.write(
' <include ref="${DD4hepINSTALL}/DDDetectors/compact/detector_types.xml"/>\n'
)
outfile.write(' <include ref="../ILD_common_v02/limits.xml"/>\n')
outfile.write(' <!-- Readout slice in ecal for reconstruction -->\n')
outfile.write(' <constant name="Ecal_readout_segmentation_slice0" value="'+str(ecal_sl1)+'"/>\n')
outfile.write(' <constant name="Ecal_readout_segmentation_slice1" value="'+str(ecal_sl2)+'"/>\n')
outfile.write(' <!-- Readout slice in hcal for reconstruction -->\n')
outfile.write(' <constant name="Hcal_readout_segmentation_slice" value="'+str(hcal_sl)+'"/>\n')
outfile.write(' </define>\n')
outfile.write(' <limits>\n')
outfile.write(" <!-- Readout slice in ecal for reconstruction -->\n")
outfile.write(
' <constant name="Ecal_readout_segmentation_slice0" value="'
+ str(ecal_sl1)
+ '"/>\n'
)
outfile.write(
' <constant name="Ecal_readout_segmentation_slice1" value="'
+ str(ecal_sl2)
+ '"/>\n'
)
outfile.write(" <!-- Readout slice in hcal for reconstruction -->\n")
outfile.write(
' <constant name="Hcal_readout_segmentation_slice" value="'
+ str(hcal_sl)
+ '"/>\n'
)
outfile.write(" </define>\n")
outfile.write(" <limits>\n")
outfile.write(' <limitset name="cal_limits">\n')
outfile.write(' <limit name="step_length_max" particles="*" value="cal_steplimit_val" unit="cal_steplimit_unit" />\n')
outfile.write(' </limitset>\n')
outfile.write(
' <limit name="step_length_max" particles="*" value="cal_steplimit_val" unit="cal_steplimit_unit" />\n'
)
outfile.write(" </limitset>\n")
outfile.write(' <limitset name="TPC_limits">\n')
outfile.write(' <limit name="step_length_max" particles="*" value="tpc_steplimit_val" unit="tpc_steplimit_unit" />\n')
outfile.write(' </limitset>\n')
outfile.write(
' <limit name="step_length_max" particles="*" value="tpc_steplimit_val" unit="tpc_steplimit_unit" />\n'
)
outfile.write(" </limitset>\n")
outfile.write(' <limitset name="Tracker_limits">\n')
outfile.write(' <limit name="step_length_max" particles="*" value="tracker_steplimit_val" unit="tracker_steplimit_unit" />\n')
outfile.write(' </limitset>\n')
outfile.write(' </limits>\n')
outfile.write(
' <limit name="step_length_max" particles="*" value="tracker_steplimit_val" unit="tracker_steplimit_unit" />\n'
)
outfile.write(" </limitset>\n")
outfile.write(" </limits>\n")
outfile.write(' <include ref="../ILD_common_v02/display.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/Beampipe_o1_v01_01.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/vxd07.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/ftd_simple_staggered_02.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/sit_simple_pixel_sensors_01.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/sit_simple_pixel_sensors_01.xml"/>\n'
)
outfile.write(' <include ref="../ILD_common_v02/tpc10_01.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/set_simple_planar_sensors_01.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/set_simple_planar_sensors_01.xml"/>\n'
)
outfile.write(' <include ref="../ILD_common_v02/SEcal06_hybrid_Barrel.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/SEcal06_hybrid_Endcaps.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/SEcal05_siw_ECRing.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/SHcalSc04_Barrel_v04.xml"/>\n')

if Large:
outfile.write(' <include ref="../ILD_common_v02/SHcalSc04_Endcaps_v01_LARGE.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/SHcalSc04_Endcaps_v01_LARGE.xml"/>\n'
)
else:
outfile.write(' <include ref="../ILD_common_v02/SHcalSc04_Endcaps_v01_SMALL.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/SHcalSc04_Endcaps_v01_SMALL.xml"/>\n'
)

outfile.write(' <include ref="../ILD_common_v02/SHcalSc04_EndcapRing_v01.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/Yoke05_Barrel.xml"/>\n')
Expand All @@ -94,108 +133,130 @@ def writeTopCompactXml( outfile, version, name, Large, Option, SolenoidMap, Anti
outfile.write(' <include ref="../ILD_common_v02/BeamCal08.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/coil03.xml"/>\n')
outfile.write(' <include ref="../ILD_common_v02/SServices00.xml"/>\n')
outfile.write(' <plugins>\n')
outfile.write(" <plugins>\n")
outfile.write(' <plugin name="DD4hepVolumeManager"/>\n')
outfile.write(' <plugin name="InstallSurfaceManager"/>\n')
outfile.write(' </plugins>\n')
outfile.write(" </plugins>\n")

if SolenoidMap:
if Large:
outfile.write(' <include ref="../ILD_common_v02/Field_Solenoid_Map_l_3.5T.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/Field_Solenoid_Map_l_3.5T.xml"/>\n'
)
else:
outfile.write(' <include ref="../ILD_common_v02/Field_Solenoid_Map_s_4.0T.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/Field_Solenoid_Map_s_4.0T.xml"/>\n'
)
else:
outfile.write(' <include ref="../ILD_common_v02/Field_Solenoid_Ideal.xml"/>\n')

if AntiDID:
if Large:
outfile.write(' <include ref="../ILD_common_v02/Field_AntiDID_Map_l.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/Field_AntiDID_Map_l.xml"/>\n'
)
else:
outfile.write(' <include ref="../ILD_common_v02/Field_AntiDID_Map_s.xml"/>\n')
outfile.write(
' <include ref="../ILD_common_v02/Field_AntiDID_Map_s.xml"/>\n'
)

if FwdFields > 0:
outfile.write(
' <include ref="../ILD_common_v02/Field_FwdMagnets_Ideal_'
+ str(FwdFields)
+ 'GeV.xml"/>\n'
)

outfile.write("</lccdd>\n")

if FwdFields>0:
outfile.write(' <include ref="../ILD_common_v02/Field_FwdMagnets_Ideal_'+str(FwdFields)+'GeV.xml"/>\n')

outfile.write('</lccdd>\n')
# -----------------------------------------------------

#-----------------------------------------------------

def getVersionParameters(version):
vparams={}
if version==2:
vparams['SolenoidMap']=False
vparams['AntiDID']=False
vparams['FwdFields']=0
elif version==3:
vparams['SolenoidMap']=True
vparams['AntiDID']=False
vparams['FwdFields']=250
elif version==4:
vparams['SolenoidMap']=True
vparams['AntiDID']=False
vparams['FwdFields']=500
elif version==5:
vparams['SolenoidMap']=True
vparams['AntiDID']=True
vparams['FwdFields']=250
elif version==6:
vparams['SolenoidMap']=True
vparams['AntiDID']=True
vparams['FwdFields']=500
elif version==7:
vparams['SolenoidMap']=True
vparams['AntiDID']=False
vparams['FwdFields']=1000
elif version==8:
vparams['SolenoidMap']=True
vparams['AntiDID']=True
vparams['FwdFields']=1000
vparams = {}
if version == 2:
vparams["SolenoidMap"] = False
vparams["AntiDID"] = False
vparams["FwdFields"] = 0
elif version == 3:
vparams["SolenoidMap"] = True
vparams["AntiDID"] = False
vparams["FwdFields"] = 250
elif version == 4:
vparams["SolenoidMap"] = True
vparams["AntiDID"] = False
vparams["FwdFields"] = 500
elif version == 5:
vparams["SolenoidMap"] = True
vparams["AntiDID"] = True
vparams["FwdFields"] = 250
elif version == 6:
vparams["SolenoidMap"] = True
vparams["AntiDID"] = True
vparams["FwdFields"] = 500
elif version == 7:
vparams["SolenoidMap"] = True
vparams["AntiDID"] = False
vparams["FwdFields"] = 1000
elif version == 8:
vparams["SolenoidMap"] = True
vparams["AntiDID"] = True
vparams["FwdFields"] = 1000
else:
print 'ERROR: unknown version requested:',version,'!!'
print("ERROR: unknown version requested:", version, "!!")
return vparams

if version<10:
vparams['vString']='0'+str(version)
if version < 10:
vparams["vString"] = "0" + str(version)
else:
vparams['vString']=str(version)
vparams["vString"] = str(version)

return vparams


#-----------------------------------------------------
# -----------------------------------------------------

import os

prename='ILD' # for the real thing
prename = "ILD" # for the real thing
# prename='test' # for testing

topdir='../'
mainoutdirname=prename+'_sl5_v02'
mainoutdir=topdir+mainoutdirname
topdir = "../"
mainoutdirname = prename + "_sl5_v02"
mainoutdir = topdir + mainoutdirname

if not os.path.exists(mainoutdir):
os.makedirs(mainoutdir)

for version in range(2, 9):
vparams=getVersionParameters(version)
vparams = getVersionParameters(version)
for Large in (True, False):
for Option in range(0,5):

modelname=prename+'_'
for Option in range(0, 5):
modelname = prename + "_"
if Large:
modelname=modelname+'l'
modelname = modelname + "l"
else:
modelname=modelname+'s'
modelname=modelname+'5_'
modelname = modelname + "s"
modelname = modelname + "5_"

if Option > 0:
modelname = modelname + "o" + str(Option) + "_"

if Option>0:
modelname=modelname+'o'+str(Option)+'_'
modelname = modelname + "v" + vparams["vString"]

modelname=modelname+'v'+vparams['vString']

outfile=open(mainoutdir+'/'+modelname+'.xml','w')
writeTopCompactXml( outfile, version=vparams['vString'], name=modelname, Large=Large, Option=Option, SolenoidMap=vparams['SolenoidMap'], AntiDID=vparams['AntiDID'], FwdFields=vparams['FwdFields'] )
outfile = open(mainoutdir + "/" + modelname + ".xml", "w")
writeTopCompactXml(
outfile,
version=vparams["vString"],
name=modelname,
Large=Large,
Option=Option,
SolenoidMap=vparams["SolenoidMap"],
AntiDID=vparams["AntiDID"],
FwdFields=vparams["FwdFields"],
)
outfile.close()
if not os.path.exists( topdir+modelname ):
os.symlink( mainoutdirname, topdir+modelname )

if not os.path.exists(topdir + modelname):
os.symlink(mainoutdirname, topdir + modelname)
Loading

0 comments on commit f41a9bd

Please sign in to comment.