CbmFindPrimaryVertexEvents.cxx

Go to the documentation of this file.

/* Copyright (C) 2014-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer], Florian Uhlig */
 
#include "CbmFindPrimaryVertexEvents.h"
 
#include "CbmEvent.h"
#include "CbmPrimaryVertexFinder.h"
#include "CbmVertex.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
 
#include <Logger.h>
 
#include <cassert>
#include <iomanip>
#include <iostream>
 
using namespace std;
 
 
// -----   Default constructor   -------------------------------------------
CbmFindPrimaryVertexEvents::CbmFindPrimaryVertexEvents()
  : FairTask()
  , fTimer()
  , fFinder(NULL)
  , fEvents(NULL)
  , fTracks(NULL)
  , fPrimVert(NULL)
  , fNofEvents(0)
  , fTimeTot(0.)
{
  fName = "FindPrimaryVertex";
}
// -------------------------------------------------------------------------
 
 
// -----   Standard constructor   ------------------------------------------
CbmFindPrimaryVertexEvents::CbmFindPrimaryVertexEvents(CbmPrimaryVertexFinder* pvFinder)
  : FairTask()
  , fTimer()
  , fFinder(pvFinder)
  , fEvents(NULL)
  , fTracks(NULL)
  , fPrimVert(NULL)
  , fNofEvents(0)
  , fTimeTot(0.)
{
  fName = "FindPrimaryVertex";
}
// -------------------------------------------------------------------------
 
 
// -----  Constructor with name and title  ---------------------------------
CbmFindPrimaryVertexEvents::CbmFindPrimaryVertexEvents(const char* name, const char*, CbmPrimaryVertexFinder* finder)
  : FairTask(name)
  , fTimer()
  , fFinder(finder)
  , fEvents(NULL)
  , fTracks(NULL)
  , fPrimVert(NULL)
  , fNofEvents(0)
  , fTimeTot(0.)
{
}
// -------------------------------------------------------------------------
 
 
// -----   Destructor   ----------------------------------------------------
CbmFindPrimaryVertexEvents::~CbmFindPrimaryVertexEvents() {}
// -------------------------------------------------------------------------
 
 
// -----   Initialisation  --------------------------------------------------
InitStatus CbmFindPrimaryVertexEvents::Init()
{
 
  assert(fFinder);
 
  // Get FairRootManager
  FairRootManager* ioman = FairRootManager::Instance();
  assert(ioman);
 
  // Get event array
  fEvents = dynamic_cast<TClonesArray*>(ioman->GetObject("CbmEvent"));
  if (nullptr == fEvents) {
    LOG(fatal) << GetName() << "No CbmEvent TClonesArray found!";
  }
 
  // Get CbmStsTrack array
  fTracks = (TClonesArray*) ioman->GetObject("StsTrack");
  assert(fTracks);
 
  // Create and register CbmVertex object
  fPrimVert = new CbmVertex("Primary Vertex", "Global");
  ioman->Register("PrimaryVertex.", "Global", fPrimVert, IsOutputBranchPersistent("PrimaryVertex"));
 
 
  // Initialise vertex finder
  fFinder->Init();
 
  return kSUCCESS;
}
// -------------------------------------------------------------------------
 
 
// -----   Task execution   ------------------------------------------------
void CbmFindPrimaryVertexEvents::Exec(Option_t*)
{
 
  // --- Event loop
  Int_t nEvents = fEvents->GetEntriesFast();
  LOG(debug) << GetName() << ": reading time slice with " << nEvents << " events ";
  for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) {
 
    fTimer.Start();
    CbmEvent* event = static_cast<CbmEvent*>(fEvents->At(iEvent));
 
    // --- Call find method of vertex finder
    //    Int_t status = fFinder->FindEventVertex(event, fTracks);
    fFinder->FindEventVertex(event, fTracks);
 
    // --- Event log
    LOG(info) << "+ " << setw(20) << GetName() << ": Event " << setw(6) << right << event->GetNumber() << ", real time "
              << fixed << setprecision(6) << fTimer.RealTime()
              << " s, tracks used: " << event->GetVertex()->GetNTracks();
    LOG(debug) << fPrimVert->ToString();
 
    // --- Counters
    fNofEvents++;
    fTimeTot += fTimer.RealTime();
 
  }  //# events
}
// -------------------------------------------------------------------------
 
 
// -----   End-of-run action   ---------------------------------------------
void CbmFindPrimaryVertexEvents::Finish()
{
 
  std::cout << std::endl;
  LOG(info) << "=====================================";
  LOG(info) << GetName() << ": Run summary";
  LOG(info) << "Events processed   : " << fNofEvents;
  LOG(info) << "Time per event     : " << fTimeTot / Double_t(fNofEvents) << " s ";
  LOG(info) << "=====================================";
}
// -------------------------------------------------------------------------
 
 
ClassImp(CbmFindPrimaryVertexEvents)