added Ntuple file as input for runsim_SND

Author: Anushka Durg

shipLHC/run_simSND.py

@@ -60,6 +60,8 @@
 parser.add_argument("--NagoyaEmu","--useNagoyaEmulsions",dest="useNagoyaEmulsions",help="use bricks of 57 Nagoya emulsion films instead of 60 Slavich", required=False,action="store_true")
 parser.add_argument("-y", dest="year", help="specify the year to generate the respective TI18 detector setup", required=False, type=int, default=2024)
 
+parser.add_argument("--dataProfile", dest="dataProfile", help="Path to the ROOT file with Ntuple data", required=False)
+#parser.add_argument("--matrixFile", dest="matrixFile", help="Path to the file with matrix particle counts", required=True, type=str)
 options = parser.parse_args()
 
 # user hook
@@ -166,6 +168,7 @@ def Exec(self,opt):
 primGen = ROOT.FairPrimaryGenerator()
 
 # -----Particle Gun-----------------------
+
 if simEngine == "PG": 
   myPgun = ROOT.FairBoxGenerator(options.pID,1)
   myPgun.SetPRange(options.Estart,options.Eend)
@@ -191,7 +194,52 @@ def Exec(self,opt):
         f'({options.EVx},{options.EVy},{options.EVz})[cm × cm × cm] \n'
         f'with a uniform x-y spread of (Dx,Dy)=({options.Dx},{options.Dy})[cm × cm]'
         f' and {options.nZSlices} z slices in steps of {options.zSliceStep}[cm].')
+
   ROOT.FairLogger.GetLogger().SetLogScreenLevel("WARNING") # otherwise stupid printout for each event
+#________NTUPLE--------
+# Use the Ntuple file if provided
+timerNtuple = ROOT.TStopwatch()
+
+# Start timing
+timerNtuple.Start()
+if options.dataProfile:
+    print(f"Using Ntuple file: {options.dataProfile}")
+    # Open the ROOT file containing the Ntuple with extrapolated muon tracks
+    file = ROOT.TFile(options.dataProfile)
+    ntuple = file.Get("muon_tracks")
+    primGen = ROOT.FairPrimaryGenerator()
+    z_plane = 250.0
+    for i in range(ntuple.GetEntries()):
+        ntuple.GetEntry(i)
+
+        # Extract x, y positions and slopes from the Ntuple
+        x = ntuple.x
+        y = ntuple.y
+        z = 0
+
+        slope_xz = ntuple.slopes_xz
+        slope_yz = ntuple.slopes_yz
+        x0 = x - slope_xz * z_plane
+        y0 = y - slope_yz * z_plane
+        z0 = 0
+        pgun = ROOT.FairBoxGenerator(13, 1)  #  Particle ID 13 for muons
+        pgun.SetPRange(1, 10)  # Set the momentum range, e.g., 1-10 GeV
+        pgun.SetXYZ(x0 * u.cm, y0 * u.cm, z0 * u.cm)
+        primGen.AddGenerator(pgun)
+       # print(f"Configured 1 particle at position ({x}, {y}, {z}) with slopes {slope_xz}, {slope_yz}")
+
+else:
+    print("Please provide a Data Profile")
+    exit(1)
+timerNtuple.Stop()
+
+# Log the time taken for Ntuple processing
+print(f"Time to process Ntuple: {timerNtuple.RealTime()} seconds (Real Time)")
+print(f"CPU Time: {timerNtuple.CpuTime()} seconds")
+
+# Set the primary generator for the run
+run.SetGenerator(primGen)
+
 # -----muon DIS Background------------------------
 if simEngine == "muonDIS":
    ut.checkFileExists(inputFile)