CleanupPrelim.C File Reference

#include <set>

Go to the source code of this file.

Functions
void	CleanupPrelim ()

Function Documentation

void CleanupPrelim

(

)

Definition at line 3 of file CleanupPrelim.C.

References cleanruns, om::cout, allTimeWatchdog::endl, MECModelEnuComparisons::i, makeTrainCVSamples::int, nentries, preliminary, updateRunHistoryTables::run, and getGoodRuns4SAM::subrun.

                    {

   set<int> eventIds; // keep track of already seen event numbers
   int thisId = 0;

   //Get old file, old tree and set top branch address
   TFile *oldfile = new TFile("MetricsTree_All.root","read");
   TTree *oldtree = (TTree*)oldfile->Get("mytree");
   Long64_t nentries = oldtree->GetEntries();
   int run, subrun, preliminary;
   oldtree->SetBranchAddress("run",&run);
   oldtree->SetBranchAddress("subrun",&subrun);
   oldtree->SetBranchAddress("preliminary",&preliminary);

   //Create a new file + a clone of old tree in new file
   TFile *newfile = new TFile("MetricsClean.root","recreate");
   TTree *newtree = oldtree->CloneTree(0);

   set<int> cleanruns;

   for (Long64_t i = 0; i < nentries; i++) {

      if(i%(nentries/10)==0) cout << int(100.0*i/nentries+0.5) << "% done" << endl;

      oldtree->GetEntry(i);

      thisId = run*100 + subrun;
      
      // if we have not seen this subrun yet, add it to the set
      // and to the entry list
      if (eventIds.count(thisId) == 0 && !preliminary){

        eventIds.insert(thisId);

        newtree->Fill();

      }

   }

//   newtree->Print();
   newtree->AutoSave();

   delete oldfile;
   delete newfile;

}