Changeset 84 in svn

Timestamp:

Dec 4, 2008, 1:11:27 PM (16 years ago)

Author:

uid677

Message:

add output tree to example

Location:

trunk

Files:

• Examples/Analysis_Ex.cpp (modified) (3 diffs)
• Examples/interface/Analysis_Ex.h (modified) (2 diffs)
• Examples/src/Analysis_Ex.cc (modified) (4 diffs)
• Makefile (modified) (2 diffs)
• genMakefile.tcl (modified) (2 diffs)

trunk/Examples/Analysis_Ex.cpp

@@ -72 +72 @@
   //*********************Output root file of the analysis************************
   //*****************************************************************************
-  TString outputFile(argv[2]);
- 
   ExRootTreeReader *treeReaderGen = new ExRootTreeReader(&chainGen);
   ExRootTreeReader *treeReaderRec = new ExRootTreeReader(&chainRec);
@@ -84 +82 @@
   ofstream f_out(LogName.c_str(),ofstream::app);
 
+  //create the output tree
+  string outputfilename = argv[2];
+  TFile *outputFile = TFile::Open(outputfilename.c_str(), "RECREATE"); // Creates the file, but should be closed just after
+  outputFile->Close();
+
+  ExRootTreeWriter *treeWriter = new ExRootTreeWriter(outputfilename, "Analysis");
+
   //*****************************************************************************
   //***************************Run the analysis**********************************
@@ -89 +94 @@
 
   Analysis_Ex *DefaultOne = new  Analysis_Ex("Examples/Datacard_Analysis_Ex.dat",LogName);
-  DefaultOne->Run(treeReaderGen,treeReaderRec,treeReaderTrig);
+  DefaultOne->Run(treeReaderGen,treeReaderRec,treeReaderTrig,treeWriter);
   DefaultOne->WriteOutput(LogName);
   delete DefaultOne;

trunk/Examples/interface/Analysis_Ex.h

@@ -35 +35 @@
   ~Analysis_Ex();
   
-  void Run(ExRootTreeReader *treeReaderGen, ExRootTreeReader *treeReaderRec, ExRootTreeReader *treeReaderTrig);//Analyze de l'event
+  void Run(ExRootTreeReader *treeReaderGen, ExRootTreeReader *treeReaderRec, ExRootTreeReader *treeReaderTrig, ExRootTreeWriter *treeWriter);//Analyze de l'event
   void WriteOutput(string LogName);
 
@@ -51 +51 @@
 };
 
+//------------------------------------------------------------------------------
+
+class TRootInvm: public TObject
+{
+public:
+
+Float_t M;
+
+  ClassDef(TRootInvm, 1)
+};
+
+
 #endif

trunk/Examples/src/Analysis_Ex.cc

@@ -44 +44 @@
 }
 
-void Analysis_Ex::Run(ExRootTreeReader *treeReaderGen, ExRootTreeReader *treeReaderRec, ExRootTreeReader *treeReaderTrig)
+void Analysis_Ex::Run(ExRootTreeReader *treeReaderGen, ExRootTreeReader *treeReaderRec, ExRootTreeReader *treeReaderTrig, ExRootTreeWriter *treeWriter)
 { 
   total=0;//initialisation of total number of events
@@ -70 +70 @@
   //to get the trigger information
   const TClonesArray *TRIGGER  = treeReaderTrig->UseBranch("TrigResult");
-  
+
+  //Define the branches that will be filled during the analysis
+  ExRootTreeBranch *INVMASS = treeWriter->NewBranch("INVMass", TRootInvm::Class());
+  TRootInvm *inv_mass; 
   //*******************************************
   
@@ -296 +299 @@
      for(Int_t k = numElec; k < (numElec+mu.GetEntries()); k++)Lept[k].SetPxPyPzE(mu[k-numElec]->Px,mu[k-numElec]->Py,mu[k-numElec]->Pz,mu[k-numElec]->E);
     
+     //Example how to white a branch in the output file
+     inv_mass=(TRootInvm*) INVMASS->NewEntry();
+     inv_mass->M=(Lept[0]+Lept[1]).M();
+ 
      if((Lept[0]+Lept[1]).M() > INV_MASS_LL )continue;// the invariant mass should be < INV_MASS_LL
      cut_2++;//event accepted
+
+     treeWriter->Fill(); 
    }
+ treeWriter->Write();
 
 }
@@ -350 +360 @@
 }
 
-

trunk/Makefile

@@ -101 +101 @@
         src/TreeClassesLinkDef.h \
         interface/TreeClasses.h
+tmp/Examples/src/Analysis_ExDict.$(SrcSuf): \
+        Examples/src/Analysis_ExLinkDef.h \
+        Examples/interface/Analysis_Ex.h
 DICT =  \
         tmp/Utilities/ExRootAnalysis/src/BlockClassesDict.$(SrcSuf) \
-        tmp/src/TreeClassesDict.$(SrcSuf)
+        tmp/src/TreeClassesDict.$(SrcSuf) \
+        tmp/Examples/src/Analysis_ExDict.$(SrcSuf)
 
 DICT_OBJ =  \
         tmp/Utilities/ExRootAnalysis/src/BlockClassesDict.$(ObjSuf) \
-        tmp/src/TreeClassesDict.$(ObjSuf)
+        tmp/src/TreeClassesDict.$(ObjSuf) \
+        tmp/Examples/src/Analysis_ExDict.$(ObjSuf)
 
 tmp/src/BFieldProp.$(ObjSuf): \
@@ -541 +546 @@
         @touch $@
 
+Examples/interface/Analysis_Ex.h: \
+        Utilities/ExRootAnalysis/interface/ExRootTreeReader.h \
+        Utilities/ExRootAnalysis/interface/ExRootTreeWriter.h \
+        Utilities/ExRootAnalysis/interface/ExRootTreeBranch.h \
+        Utilities/ExRootAnalysis/interface/TSimpleArray.h \
+        Utilities/ExRootAnalysis/interface/BlockClasses.h
+        @touch $@
+
 interface/SmearUtil.h: \
         Utilities/ExRootAnalysis/interface/BlockClasses.h \

trunk/genMakefile.tcl

@@ -126 +126 @@
   global prefix suffix objSuf exeSuf
    
-  set executable [glob -nocomplain {Delphes.cpp} {Resolutions.cpp} {Examples/Trigger_Only.cpp} ]
+  set executable [glob -nocomplain {Delphes.cpp} {Resolutions.cpp} {Examples/Trigger_Only.cpp} {Exemples/Analysis_Ex.cpp}]
                                 
   set exeFiles {}
@@ -232 +232 @@
 executableDeps {*.cpp} {Examples/*.cpp} 
 
-dictDeps {DICT} {Utilities/ExRootAnalysis/src/*LinkDef.h} {src/*LinkDef.h}
+dictDeps {DICT} {Utilities/ExRootAnalysis/src/*LinkDef.h} {src/*LinkDef.h} {Examples/src/*LinkDef.h}
 
 sourceDeps {SOURCE} {src/*.cc} {Utilities/ExRootAnalysis/src/*.cc} {Utilities/Hector/src/*.cc} {Utilities/CDFCones/src/*cc} {Utilities/Fastjet/src/*.cc} {Utilities/Fastjet/plugins/CDFCones/*.cc} {Utilities/Fastjet/plugins/CDFCones/src/*.cc} {Utilities/Fastjet/plugins/SISCone/*.cc} {Utilities/Fastjet/plugins/SISCone/src/*.cc} {Examples/src/*.cc}