example.cc

/*******************************************************************************
*
*  Filename    : example.cc
*  Description : Simple example for reading contents of bprime Kit ntuple
*  Author      : Yi-Mu "Enoch" Chen [ ensc@hep1.phys.ntu.edu.tw ]
*
*******************************************************************************/


#include <iostream>
#include <string.h>
#include "TTree.h"
#include "TChain.h"

// Files that must be manually added!
#include "format.h"
#include "checkEvtTool.h"

using namespace std;

void example()
{
   bool debug = true;
   checkEvtTool checkEvt; //or checkEvtTool checkEvt( debug );
   // Supports multiple File input
   checkEvt.addJson( "Cert_254833_13TeV_PromptReco_Collisions15_JSON.txt" );
   checkEvt.addJson( "Cert_246908-254879_13TeV_PromptReco_Collisions15_JSON.txt" );
   checkEvt.makeJsonMap();

   TChain* root = new TChain( "bprimeKit/root" );
   root->Add( "bprimeKit_ntuples.root" );

   EvtInfoBranches EvtInfo ;
   EvtInfo.Register( root );
   JetInfoBranches jetInfo ;
   jetInfo.Register( root, "JetInfo" );
   JetInfoBranches wjetInfo;
   wjetInfo.Register( root, "AK8BosonJetInfo" );

   for( int entry = 0; entry < root->GetEntries() ; ++entry ) { //Loop over all events
      root->GetEntry( entry );

      // Golden JSON File check!
      if( !checkEvt.isGoodEvt( EvtInfo.RunNo, EvtInfo.LumiNo ) ) {
         continue;
      }
      for( int i = 0 ; i < jetInfo.Size ; i++ ) { // Loop over all jets in "JetInfo" collection
         cout << jetInfo.Pt[i] << endl;
      }
      for( int i = 0 ; i < wjetInfo.Size; ++i ) { // Loop over all jets in "AK8BosonJetInfo" collections
         for( int j = 0 ; j < wjetInfo.NSubjets[i] ; ++j ) { // Loop over all subjets for jet i
            auto SubjetMass = *( wjetInfo.SubjetMass );
            cout << SubjetMass[ wjetInfo.SubjetsIdxStart[i] + j  ] << endl ;
         }
      }
   }

   checkEvt.saveJson( "Cert_246908-254833-254879_13TeV_PromptReco_Collisions15_JSON.txt" );
}