run_evtselection_tool_test.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifdef XAOD_STANDALONE
#include <memory>
#include <string>
#include "xAODRootAccess/Init.h"
#include "xAODRootAccess/TEvent.h"
#include "AsgMessaging/Check.h"
#include "CxxUtils/make_unique.h"
#include "HIEventUtils/HIVertexSelectionTool.h"
#include "HIEventUtils/HIEventSelectionTool.h"
#include "xAODTracking/VertexContainer.h"
#include "InDetTrackSelectionTool/InDetTrackSelectionTool.h"
#include "TError.h"
#include "TFile.h"
 
using std::string;
using std::unique_ptr;
using CxxUtils::make_unique;
 
int main(int argc, char* argv[])
{
  string appName = argv[0];
  if( !xAOD::Init(appName.data()).isSuccess() ) {
    Error( appName.data(), "Could not call xAOD::Init()" );
    return 1;
  }
 
  string fileName;
  // Check if we received a file name:
  if( argc < 2 ) {
    fileName = getenv("ROOTCORE_TEST_FILE");
    if (fileName.empty()) {
      Error( appName.data(), "No file name received!" );
      Error( appName.data(), "  Usage: %s [xAOD file name]", appName.data() );
      return 1;
    }
  } else {
    fileName = argv[ 1 ];
  }
 
  unique_ptr<TFile> fin( TFile::Open( fileName.data(), "READ" ) );
 
  auto event = make_unique<xAOD::TEvent>( fin.get(), xAOD::TEvent::kClassAccess );
  Long64_t entries = event->getEntries();
  Info( appName.data(), "Number of events in file: %llu", entries );
 
  auto eventSelTool = make_unique< HI::HIEventSelectionTool >("EventSelectionTool");
 
  auto vertexSelTool = make_unique< HI::HIVertexSelectionTool >("vertexSelectionTool");
 
  auto vtxTrkTool = make_unique<InDet::InDetTrackSelectionTool>("vtxTrkTool", "HILoose");
  ToolHandle< InDet::IInDetTrackSelectionTool > vtxTrkToolHandle = vtxTrkTool.get();
  ASG_CHECK_SA( appName, vtxTrkTool->initialize() );
  ASG_CHECK_SA( appName, vertexSelTool->setProperty( "TrackSelectionTool", vtxTrkToolHandle ) );
 
  ToolHandle< HI::IHIVertexSelectionTool > vertexSelToolHandle = vertexSelTool.get();
 
  ASG_CHECK_SA( appName, vertexSelTool->initialize() );
  ASG_CHECK_SA( appName, eventSelTool->setProperty( "VertexSelectionTool", vertexSelToolHandle ) );
  //ASG_CHECK_SA( appName, eventSelTool->setProperty( "TimeCut", 5 ) );
 
  ASG_CHECK_SA( appName, eventSelTool->initialize() );
 
  Long64_t nVtxPass = 0;
  Long64_t nVtxFail = 0;
  Long64_t nEvtPass = 0;
  Long64_t nEvtFail = 0;
  Long64_t nEvtTotal = 0;
  Long64_t nVtxTotal = 0;
 
  for (Long64_t i_evt=0; i_evt<entries; ++i_evt) {
    event->getEntry( i_evt );
 
    if( eventSelTool->accept()) {nEvtPass++; nEvtTotal++; }
    else { nEvtFail++; nEvtTotal++; }
 
    const xAOD::VertexContainer* vertices = nullptr;
    if ( !event->retrieve( vertices, "PrimaryVertices" ).isSuccess() ) {
      Error( appName.data(), "Could not retrieve PrimaryVertices" );
      return 1;
    }
 
    for (const auto* vtx : *vertices) {
      if (vertexSelTool->accept( *vtx )) { nVtxPass++; nVtxTotal++; }
      else { nVtxFail++; nVtxTotal++; }
    }
 
  }
  Info( appName.data(), "%llu evt passed.", nEvtPass );
  Info( appName.data(), "%llu evt failed.", nEvtFail );
  Info( appName.data(), "%llu evt Total.",  nEvtTotal );
  Info( appName.data(), "%llu vtx passed.", nVtxPass );
  Info( appName.data(), "%llu vtx failed.", nVtxFail );
  Info( appName.data(), "%llu vtx Total.",  nVtxTotal );
 
  ASG_CHECK_SA( appName, eventSelTool->finalize() );
  ASG_CHECK_SA( appName, vertexSelTool->finalize() );
  ASG_CHECK_SA( appName, vtxTrkTool->finalize() );
  return 0;
}
#endif // XAOD_STANDALONE