CbmFindPrimaryVertexEvents.cxx

/* Copyright (C) 2014-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer], Florian Uhlig */

/** @file CbmFindPrimaryVertexEvents.h
 ** @author Volker Friese <v.friese@gsi.de>
 ** @since 16.06.2014
 ** @date 02.11.2016
 **/

#include "CbmFindPrimaryVertexEvents.h"

#include "CbmEvent.h"
#include "CbmPrimaryVertexFinder.h"
#include "CbmVertex.h"

#include "FairRootManager.h"
#include <Logger.h>

#include "TClonesArray.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)