From 7d891d669497e9a37dc8a66c1183e11d19676d04 Mon Sep 17 00:00:00 2001
From: anupama-reghunath <anupamareghu@gmail.com>
Date: Thu, 28 Nov 2024 23:33:26 +0100
Subject: [PATCH] add example script for analysis_toolkit

---
 macro/analysis_example.py | 41 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 macro/analysis_example.py

diff --git a/macro/analysis_example.py b/macro/analysis_example.py
new file mode 100644
index 000000000..95b85503f
--- /dev/null
+++ b/macro/analysis_example.py
@@ -0,0 +1,41 @@
+#!/usr/bin/env python
+"""Example script for usage of the analysis_toolkit for signal selection ."""
+
+import analysis_toolkit
+import ROOT
+
+geo_file = "geofile_full.conical.Pythia8-TGeant4.root"
+input_file = "ship.conical.Pythia8-TGeant4_rec.root"
+
+f = ROOT.TFile.Open(input_file)
+tree = f.cbmsim
+
+selection = analysis_toolkit.selection_check(geo_file)
+
+for event_nr, event in enumerate(tree):
+    if event_nr > 200:
+        break
+
+    selection.access_event(event)
+
+    if len(event.Particles) == 0:
+        continue
+
+    for candidate_id_in_event, signal in enumerate(tree.Particles):
+        print(f"Event:{event_nr} Candidate_index: {candidate_id_in_event}")
+
+        print(f"\t vertex time: {selection.define_candidate_time(signal)} ns")
+        print(f"\t Impact Parameter: {selection.impact_parameter(signal)} cm")
+        print(f"\t is within Fiducial volume: {selection.is_in_fiducial(signal)}")
+        print(f"\t\tDist2InnerWall: {selection.dist_to_innerwall(signal)} cm")
+        print(f"\t\tDist2EntranceLid: {selection.dist_to_entrance_vessel(signal)} cm")
+        print(f"\t Daughter momentum>1 GeV: {selection.daughtermomentum(signal)} GeV")
+        print(f"\t Invariant mass: {selection.invariant_mass(signal)} GeV")
+        print(f"\t Degrees of Freedom ([d1,d2],Cut(>25)): {selection.nDOF(signal)}")
+        print(f"\t Reduced Chi^2  ([d1,d2],Cut(<5): {selection.chi2nDOF(signal)}")
+        print(f"\t DOCA: {selection.DOCA(signal)} cm")
+
+        print(
+            f"\t Preselection Cut passed:{selection.preselection_cut(signal,IP_cut=250)}"
+        )
+        print("------------------------------------------------------------")