Files for validation pipeline

scripts/FCCee/ALLEGRO/ALLEGRO_o1_v03/ALLEGRO_make_TH1.py

@@ -0,0 +1,129 @@
+import ROOT
+from podio import root_io
+import dd4hep as dd4hepModule
+from ROOT import dd4hep
+
+import argparse
+
+import os
+from dd4hep import Detector
+import DDRec
+
+
+def make_TH1_file(args):
+
+  # create output ROOT file
+  outputFile = ROOT.TFile(args.outputFile, "RECREATE")
+
+  # set file reader
+  podio_reader = root_io.Reader(args.inputFile)
+
+  # set list of directories in ROOT file (one per subsystem to check)
+  dir_list = []
+  # set list of (list of) histograms created (one list per subsystem to check)
+  histo_list = []
+
+
+
+  #############      BEGIN: ECal Barrel Histogram Definition      #############
+
+  dir_ECalBarrel = outputFile.mkdir("ECalBarrel")
+
+  hist_ccE = ROOT.TH1F("h_CaloCluster_E",
+                       "CaloCluster Energy;Energy [MeV];Counts / 0.15 MeV",
+                       100, 0, 15)
+  hist_ctcE = ROOT.TH1F("h_CaloTopoCluster_E",
+                        "CaloTopoCluster Energy;Energy [MeV];Counts / 0.15 MeV",
+                        100, 0, 15)
+  hist_ecal_totE = ROOT.TH1F("h_ECalBarrelModuleThetaMergedPosition_totE",
+                             "ECalBarrelModuleThetaMergedPosition total Energy per evt;Energy [MeV];Counts / 0.15 MeV",
+                             100, 0, 15)
+  hist_ecal_posX = ROOT.TH1F("h_ECalBarrelModuleThetaMergedPosition_posX",
+                             "ECalBarrelModuleThetaMergedPosition position X;X [mm];Counts / 37 mm",
+                             150, -2770, 2770)
+  hist_ecal_posY = ROOT.TH1F("h_ECalBarrelModuleThetaMergedPosition_posY",
+                             "ECalBarrelModuleThetaMergedPosition position Y;Y [mm];Counts / 37 mm",
+                             150, -2770, 2770)
+  hist_ecal_posZ = ROOT.TH1F("h_ECalBarrelModuleThetaMergedPosition_posZ",
+                             "ECalBarrelModuleThetaMergedPosition position Z;Z [mm];Counts / 41 mm",
+                             150, -3100, 3100)
+
+  dir_list.append(dir_ECalBarrel)
+  histo_list.append([hist_ccE, 
+                     hist_ctcE,
+                     hist_ecal_totE,
+                     hist_ecal_posX,
+                     hist_ecal_posY,
+                     hist_ecal_posZ])
+
+  ##############      END: ECal Barrel Histogram Definition      ##############
+
+
+
+
+  # Loop over dataset
+  for event in podio_reader.get("events"):
+
+    ##################     BEGIN: ECal Barrel Event Loop     ##################
+
+    for calo in event.get("CaloClusters"):
+      hist_ccE.Fill(calo.getEnergy())
+
+    for calo in event.get("CaloTopoClusters"):
+      hist_ctcE.Fill(calo.energy())
+
+    energy = 0
+    for ecal in event.get("ECalBarrelModuleThetaMergedPositioned"):
+        energy += ecal.energy()
+        hist_ecal_posX.Fill(ecal.position().x)
+        hist_ecal_posY.Fill(ecal.position().y)
+        hist_ecal_posZ.Fill(ecal.position().z)
+    hist_ecal_totE.Fill(energy)
+
+    ##################      END: ECal Barrel Event Loop      ##################
+
+
+
+  # normalize if desired
+  if args.norm:
+    n_evts = len(podio_reader.get("events"))
+    factor = 1./n_evts
+    for l in histo_list:
+      for h in l:
+        h.Scale(factor)
+
+  # save to file
+  for dir, h_list in zip(dir_list, histo_list):
+    dir.cd()
+    for h in h_list:
+      h.Write()
+
+  outputFile.Close()
+
+  return
+
+
+
+
+
+
+#########################################################################
+
+if __name__ == "__main__":
+  parser = argparse.ArgumentParser(
+        description="Process simulation"
+    )
+  parser.add_argument('-f', "--inputFile",  type=str, 
+                      help='The name of the simulation file to be processed', default='ALLEGRO_sim_digi_reco.root')
+  parser.add_argument('-o', "--outputFile", type=str, 
+                      help='The name of the ROOT file where to save output histograms', default='results.root')
+  parser.add_argument('--norm', action='store_true',
+                      help='Normalize output histograms by number of events')
+  args = parser.parse_args()
+
+  make_TH1_file(args)  
+
+
+
+
+

scripts/FCCee/ALLEGRO/ALLEGRO_o1_v03/ALLEGRO_o1_v03_script.sh

@@ -0,0 +1,26 @@
+# enable exit on error to check correct script execution from within pipeline
+set -e
+
+# setup phase
+echo "SETUP PHASE:"
+
+[ -z "$KEY4HEP_STACK" ] && source /cvmfs/sw-nightlies.hsf.org/key4hep/setup.sh
+
+
+# simulation phase
+echo "SIM-DIGI-RECO PHASE:"
+
+echo "Starting script..."
+source $FCCCONFIG/share/FCC-config/FullSim/ALLEGRO/ALLEGRO_o1_v03/ctest_sim_digi_reco.sh
+
+
+# analyze simulation file
+echo "ANALYSIS PHASE:"
+
+echo "Starting analysis script..."
+python $WORKAREA/key4hep-reco-validation/scripts/FCCee/ALLEGRO/ALLEGRO_o1_v03/ALLEGRO_make_TH1.py \
+       -f ALLEGRO_sim_digi_reco.root -o results.root
+echo "Script executed successfully"
+
+
+

scripts/FCCee/CLD/CLD_o3_v01/ARC_make_TH1.py

@@ -0,0 +1,149 @@
+import ROOT
+from podio import root_io
+import dd4hep as dd4hepModule
+from ROOT import dd4hep
+import argparse
+import numpy as np
+import os
+from dd4hep import Detector
+import DDRec
+
+def find_directory_upwards(start_dir, target_dir_name):
+    """
+    Recursively searches upwards from the start directory for a directory with the specified name.
+
+    :param start_dir: The starting directory path.
+    :param target_dir_name: The name of the directory to search for.
+    :return: The path to the found directory or None if not found.
+    """
+    current_dir = os.path.abspath(start_dir)
+
+    while True:
+        # Check if the target directory exists in the current directory
+        if target_dir_name in os.listdir(current_dir):
+            possible_match = os.path.join(current_dir, target_dir_name)
+            if os.path.isdir(possible_match):
+                return possible_match
+
+        # Move up one level in the directory tree
+        parent_dir = os.path.dirname(current_dir)
+
+        # If we have reached the root directory and haven't found the directory, return None
+        if current_dir == parent_dir:
+            return None
+
+        current_dir = parent_dir
+
+
+def make_TH1_file(args):
+
+  # create output ROOT file
+  outputFile = ROOT.TFile(args.outputFile, "UPDATE")
+
+  # set file reader
+  podio_reader = root_io.Reader(args.inputFile)
+
+  # get detector description for cell id decoding
+  k4geo = find_directory_upwards('./', 'k4geo')
+  theDetector = Detector.getInstance()
+  theDetector.fromXML(k4geo+"/test/compact/ARC_standalone_o1_v01.xml")
+  idposConv = DDRec.CellIDPositionConverter(theDetector)
+
+  # set list of directories in ROOT file (one per subsystem to check)
+  dir_list = []
+  # set list of (list of) histograms created (one list per subsystem to check)
+  histo_list = []
+
+
+
+  ###########      BEGIN: ARC standalone Histogram Definition      ############
+
+  dir_ARC = outputFile.mkdir("ARC_standalone")
+
+  hist_ARC_nPh = ROOT.TH1F("h_ARC_nPh",
+                           "Total number of photon counts per event;Number of Photons;Photon count / 5",
+                           50, 0, 250)
+  hist_ARC_theta = ROOT.TH1F("h_ARC_theta",
+                             "Photons counts vs. #theta;Polar angle #theta;Photon count / 35 mrad",
+                              90, 0, np.pi)
+  hist_ARC_1stHit = ROOT.TH1F("h_ARC_1stHit",
+                              "Photons counts vs. #theta of incoming particle;Polar angle #theta;Photon count / 35 mrad",
+                              90, 0, np.pi)
+
+  dir_list.append(dir_ARC)
+  histo_list.append([hist_ARC_nPh, 
+                     hist_ARC_theta,
+                     hist_ARC_1stHit])
+
+  ############      END: ARC standalone Histogram Definition      #############
+
+
+
+
+  # loop over dataset
+  for event in podio_reader.get("events"):
+
+    ###############      BEGIN: ARC standalone Event Loop      ################
+
+    n_ph = 0
+
+    p = (event.get("MCParticles"))[0]
+    mom = ROOT.TVector3(p.getMomentum().x, p.getMomentum().y, p.getMomentum().z)
+    theta_1stHit = mom.Theta()
+
+    for arc_hit in event.get("ArcCollection"):
+      particle =  arc_hit.getMCParticle()
+      pdgID = particle.getPDG()
+      if pdgID == 22 or pdgID == -22:
+        n_ph += 1
+        cellID = arc_hit.getCellID()
+        x = idposConv.position(cellID) 
+        theta = x.theta()
+
+        hist_ARC_theta.Fill(theta)
+
+    hist_ARC_nPh.Fill(n_ph)
+    hist_ARC_1stHit.Fill(theta_1stHit, n_ph)
+
+    ################      END: ARC standalone Event Loop      #################
+
+
+  if args.norm:
+    n_evts = len(podio_reader.get("events"))
+    factor = 1./n_evts
+    for l in histo_list:
+      for h in l:
+        h.Scale(factor)
+
+  # save to file
+  for dir, h_list in zip(dir_list, histo_list):
+    dir.cd()
+    for h in h_list:
+      h.Write()
+
+  outputFile.Close()
+
+  return 
+
+
+#########################################################################
+
+if __name__ == "__main__":
+  parser = argparse.ArgumentParser(
+        description="Process simulation"
+    )
+  parser.add_argument('-f', "--inputFile",  type=str, 
+                      help='The name of the simulation file to be processed', default='ARC_sim.root')
+  parser.add_argument('-o', "--outputFile", type=str, 
+                      help='The name of the ROOT file where to save output histograms', default='results.root')
+  parser.add_argument('--norm', action='store_true',
+                      help='Normalize output histograms by number of events')
+  args = parser.parse_args()
+
+  ph_presence = make_TH1_file(args)
+  print(ph_presence)  
+
+
+
+
+

scripts/FCCee/CLD/CLD_o3_v01/CLD_o3_v01_script.sh

@@ -0,0 +1,42 @@
+# enable exit on error to check correct script execution from within pipeline
+set -e
+
+# setup phase
+echo "SETUP PHASE:"
+
+[ -z "$KEY4HEP_STACK" ] && source /cvmfs/sw-nightlies.hsf.org/key4hep/setup.sh
+
+echo "Downloading necessary Github repos..."
+git clone https://github.com/key4hep/k4geo.git
+K4GEO_PATH=$(realpath k4geo)
+echo "Download terminated - setup stage successful"
+
+
+# simulation phase
+echo "SIMULATION PHASE:"
+
+cd $CLDCONFIG/share/CLDConfig
+echo "Starting simulation..."
+ddsim --steeringFile cld_arc_steer.py \
+      --compactFile $K4GEO_PATH/test/compact/ARC_standalone_o1_v01.xml \
+      --enableGun --gun.distribution "cos(theta)" \
+      --gun.energy "20*GeV" --gun.particle proton \
+      --numberOfEvents $NUMBER_OF_EVENTS \
+      --outputFile $WORKAREA/$GEOMETRY/$VERSION/ARC_sim.root \
+      --random.enableEventSeed --random.seed 42 \
+      --part.userParticleHandler='' > $WORKAREA/$GEOMETRY/$VERSION/ddsim_log.txt
+echo "Simulation ended successfully"
+cd $WORKAREA/$GEOMETRY/$VERSION
+
+
+# analyze simulation file
+echo "ANALYSIS PHASE:"
+
+echo "Starting analysis script..."
+python $WORKAREA/key4hep-reco-validation/scripts/FCCee/CLD/CLD_o3_v01/ARC_make_TH1.py \
+       -f ARC_sim.root -o results.root
+echo "Script executed successfully"
+
+
+
+