// -------------------------------------------------------------------------
// -----                 CbmFindPrimaryVertex source file              -----
// -----                  Created 28/11/05  by V. Friese               -----
// -------------------------------------------------------------------------
#include "CbmFindPrimaryVertex.h"

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

#include "FairLogger.h"
#include "FairRootManager.h"

#include "TClonesArray.h"

#include <iostream>
#include <iomanip>

using namespace std;

// -----   Default constructor   -------------------------------------------
CbmFindPrimaryVertex::CbmFindPrimaryVertex() 
  : FairTask(),
	fTimer(),
    fFinder(NULL),
    fTracks(NULL),
    fPrimVert(NULL),
	fNofEvents(0),
	fTimeTot(0.)
{
	fName = "FindPrimaryVertex";
}
// -------------------------------------------------------------------------



// -----   Standard constructor   ------------------------------------------
CbmFindPrimaryVertex::CbmFindPrimaryVertex(CbmPrimaryVertexFinder* pvFinder)
  : FairTask(),
	fTimer(),
    fFinder(pvFinder),
    fTracks(NULL),
    fPrimVert(NULL),
	fNofEvents(0),
	fTimeTot(0.)
{
	fName = "FindPrimaryVertex";
}
// -------------------------------------------------------------------------



// -----  Constructor with name and title  ---------------------------------
CbmFindPrimaryVertex::CbmFindPrimaryVertex(const char* name, 
					   const char*, 
					   CbmPrimaryVertexFinder* finder) 
  : FairTask(name),
	fTimer(),
    fFinder(finder),
    fTracks(NULL),
    fPrimVert(NULL),
	fTimeTot(0.),
	fNofEvents(0)
{
}
// -------------------------------------------------------------------------



// -----   Destructor   ----------------------------------------------------
CbmFindPrimaryVertex::~CbmFindPrimaryVertex() { }
// -------------------------------------------------------------------------



// -----   Public method Init   --------------------------------------------
InitStatus CbmFindPrimaryVertex::Init() {

  // Check for vertex finder
  if ( ! fFinder) {
    cout << "-E- CbmFindPrimaryVertex::Init : "
	 << "No vertex finder selected! " << endl;
    return kERROR;
  }

  // Get FairRootManager
  FairRootManager* ioman = FairRootManager::Instance();
  if (! ioman) {
    cout << "-E- CbmFindPrimaryVertex::Init: "
	 << "RootManager not instantised!" << endl;
    return kFATAL;
  }

  // Get CbmStsTrack array
  fTracks = (TClonesArray*) ioman->GetObject("StsTrack");
  if ( ! fTracks) {
    cout << "-W- CbmFindPrimaryVertex::Init: No STSTrack array!"
	 << endl;
    return kERROR;
  }

  // Create and register CbmVertex object
  fPrimVert = new CbmVertex("Primary Vertex", "Global");
  ioman->Register("PrimaryVertex.", "Global", fPrimVert, IsOutputBranchPersistent("PrimaryVertex"));


  // Call the Init method of the vertex finder
  fFinder->Init();

  return kSUCCESS;

}
// -------------------------------------------------------------------------



// -----   Public method Exec   --------------------------------------------
void CbmFindPrimaryVertex::Exec(Option_t*) {

  // Reset primary vertex
  fTimer.Start();
  fPrimVert->Reset();

  // Call find method of vertex finder
  Int_t iFind = fFinder->FindPrimaryVertex(fTracks, fPrimVert);
  if (iFind) cout << "-W- CbmFindPrimaryVertex::Exec: "
		  << "Vertex finder returned " << iFind << endl;

  // --- Event log
  fTimer.Stop();
  fNofEvents++;
  fTimeTot += fTimer.RealTime();
  LOG(INFO) << "+ " << setw(20) << GetName() << ": Event " << setw(6)
  		      << right << fNofEvents
  		      << ", real time " << fixed << setprecision(6)
  		      << fTimer.RealTime() << " s, tracks used: "
			  << fPrimVert->GetNTracks()
  		      << FairLogger::endl;
  LOG(DEBUG) << fPrimVert->ToString() << FairLogger::endl;

  // Print vertex
  /*
  cout << endl;
  cout << "-------------------------------------------------------" << endl;
  cout << "-I-             Primary Vertex Finder               -I-" << endl;
  fPrimVert->Print();
  cout << "-------------------------------------------------------" << endl;
  */

}
// -------------------------------------------------------------------------



// -----   Public method Finish   ------------------------------------------
void CbmFindPrimaryVertex::Finish() {
  fPrimVert->Reset();
  std::cout << std::endl;
  LOG(INFO) << "=====================================" << FairLogger::endl;
  LOG(INFO) << GetName() << ": Run summary" << FairLogger::endl;
  LOG(INFO) << "Events processed   : " << fNofEvents << FairLogger::endl;
  LOG(INFO) << "Time per event     : " << setprecision(6)
		    << fTimeTot / Double_t(fNofEvents) << " s " << FairLogger::endl;
  LOG(INFO) << "=====================================" << FairLogger::endl;

}
// -------------------------------------------------------------------------



ClassImp(CbmFindPrimaryVertex)