Changeset 84 in svn for trunk/Examples

Timestamp:

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

Author:

uid677

Message:

add output tree to example

Location:

trunk/Examples

Files:

• Analysis_Ex.cpp (modified) (3 diffs)
• interface/Analysis_Ex.h (modified) (2 diffs)
• src/Analysis_Ex.cc (modified) (4 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 @@
 }
 
-