Updated STARLIGHT from r176 ro r188 (http://starlight.hepforge.org/svn/)

[u/mrichter/AliRoot.git] / STARLIGHT / starlight / utils / convertStarlightAsciiToTree.C

diff --git a/STARLIGHT/starlight/utils/convertStarlightAsciiToTree.C b/STARLIGHT/starlight/utils/convertStarlightAsciiToTree.C
index 795dbe23b26c535b4df5e9f93ef6a41ebcc2695e..52d00309831f6504d91e1ad1594deebd775bb388 100644 (file)
--- a/STARLIGHT/starlight/utils/convertStarlightAsciiToTree.C
+++ b/STARLIGHT/starlight/utils/convertStarlightAsciiToTree.C
@@ -1,3 +1,19 @@
+// This macro reads a starlight output file (default name slight.out) and creates a root file 
+// with  TLorentzVectors for the parent and a TClonesArray of TLorentzVectors for the daughter 
+// particles.  The output is stored in a root file (default name starlight.root) with one branch 
+// labeled "parent" and the other labeled "daughters". Any number of daughter tracks can be 
+// accomodated.  Daughter species currently accomodated are:  electrons, muons, charged or neutral 
+// pions, charged or neutral kaons, and protons.  
+//
+// To use this macro, open root and then 
+// type .x convertStarlightAsciiToTree.C("inputfilename", "outputfilename")
+// 
+// The root file produced can be examined in a root TBrowser.
+//
+// A macro to read this root file and make some standard plots is also provided.  This macro is 
+// called AnalyzeTree.cxx; it can be compiled and run with the anaTree.C macro by opening root 
+// and typing .x anaTree.C()
+
 #include <iostream>
 #include <fstream>
 #include <sstream>
@@ -11,10 +27,9 @@
 using namespace std;
 
 
-void readStarlightAscii(const char* inFileName  = "slight.out",
+void convertStarlightAsciiToTree(const char* inFileName  = "slight.out",
                         const char* outFileName = "starlight.root")
 {
-       const double daughterMass = 0.13957018; // charged pion [GeV]
 
        // create output tree
        TFile* outFile = new TFile(outFileName, "RECREATE");
@@ -24,7 +39,7 @@ void readStarlightAscii(const char* inFileName  = "slight.out",
        }
        TTree*          outTree           = new TTree("starlightTree", "starlightTree");
        TLorentzVector* parentParticle    = new TLorentzVector();
-       TClonesArray*   daughterParticles = new TClonesArray("TLorentzVector");
+       TClonesArray*   daughterParticles = new TClonesArray("TLorentzVector");
        outTree->Branch("parent",    "TLorentzVector", &parentParticle,    32000, -1);
        outTree->Branch("daughters", "TClonesArray",   &daughterParticles, 32000, -1);
 
@@ -65,12 +80,13 @@ void readStarlightAscii(const char* inFileName  = "slight.out",
                        lineStream.str(line);
                        assert(lineStream >> label >> particleCode >> momentum[0] >> momentum[1] >> momentum[2]);
                        assert(label == "TRACK:");
+                       Double_t daughterMass = IDtoMass(particleCode);
+                       if (daughterMass < 0) {break;}
                        const double E = sqrt(  momentum[0] * momentum[0] + momentum[1] * momentum[1]
                                              + momentum[2] * momentum[2] + daughterMass * daughterMass);
                        new ( (*daughterParticles)[i] ) TLorentzVector(momentum[0], momentum[1], momentum[2], E);
                        *parentParticle += *(static_cast<TLorentzVector*>(daughterParticles->At(i)));
                }
-
                daughterParticles->Compress();
                outTree->Fill();
        }
@@ -81,3 +97,21 @@ void readStarlightAscii(const char* inFileName  = "slight.out",
                delete outFile;
        }
 }
+
+       double IDtoMass(int particleCode){
+               double mass;
+               if (particleCode == 2 || particleCode==3) {mass = 0.0051099907;} // electron
+               else if (particleCode == 5 || particleCode==6) {mass = 0.105658389;} // muon
+               else if (particleCode == 8 || particleCode==9)  {mass = 0.13956995;} // charged pion
+               else if (particleCode == 7) {mass = 0.1345766;} // neutral pion
+               else if (particleCode == 11|| particleCode==12) {mass = 0.493677;} // charged kaon
+               else if (particleCode == 10 || particleCode == 16)  {mass = 0.497614;} // neutral kaon
+               else if (particleCode == 14)    {mass = 0.93827231;} // proton
+               else {
+                       cout << "unknown daughter particle (ID = " << particleCode << "), please modify code to accomodate" << endl;
+                       mass = -1.0;
+//                     exit(0); 
+                    } 
+
+               return mass;
+       }