// -------------------------------------------------------------------------
// -----                CbmStsHitProducerIdeal source file             -----
// -----                  Created 10/01/06  by V. Friese               -----
// -------------------------------------------------------------------------


#include "TClonesArray.h"

#include "CbmRootManager.h"
#include "MvdHitProducerIdeal.h"
#include "MvdHit.h"
#include "MvdHitInfo.h"
#include "MvdPoint.h"
#include "CbmRunAna.h"
#include "CbmRuntimeDb.h"


// -----   Default constructor   -------------------------------------------
MvdHitProducerIdeal::MvdHitProducerIdeal() :
  CbmTask("Ideal MVD Hit Producer") 
{
	fBranchName 	= "MVDPoint";
}
// -------------------------------------------------------------------------



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

// -----   Public method Init   --------------------------------------------
InitStatus MvdHitProducerIdeal::Init() 
{
  // Get RootManager
  CbmRootManager* ioman = CbmRootManager::Instance();
  
  if ( ! ioman ) 
    {
      std::cout << "-E- MvdHitProducerIdeal::Init: "
	   << "RootManager not instantiated!" << std::endl;
      return kFATAL;
    }

  // Get input array
  fPointArray = (TClonesArray*) ioman->GetObject(fBranchName);

  if ( ! fPointArray ) 
    {
      std::cout << "-W- MvdHitProducerIdeal::Init: "
	   << "No MVDPoint array!" << std::endl;
      return kERROR;
  }

  // Create and register output array
  fHitArray = new TClonesArray("MvdHit");
  ioman->Register("MVDHit", "MVD", fHitArray, kTRUE);

  std::cout << "-I- MvdHitProducerIdeal: Intialisation successfull" << std::endl;
  return kSUCCESS;
}
// -------------------------------------------------------------------------
void MvdHitProducerIdeal::SetParContainers()
{
  // Get Base Container
  CbmRunAna* ana = CbmRunAna::Instance();
  CbmRuntimeDb* rtdb=ana->GetRuntimeDb();
  fGeoPar = (MvdGeoPar*)(rtdb->getContainer("MvdGeoPar"));

}


// -----   Public method Exec   --------------------------------------------
void MvdHitProducerIdeal::Exec(Option_t* opt) 
{
  // Reset output array
  if ( ! fHitArray ) 
    Fatal("Exec", "No HitArray");
  
  fHitArray->Clear();
  
  // Declare some variables
  MvdPoint *point = 0;

  Int_t 
    detID = 0,       // Detector ID
    trackID = 0;     // Track index

  TVector3 
    pos, dpos;       // Position and error vectors

  // Loop over MvdPoints
  Int_t 
    nPoints = fPointArray->GetEntriesFast();

  for (Int_t iPoint = 0; iPoint < nPoints; iPoint++) 
    {
      point = (MvdPoint*) fPointArray->At(iPoint);
	std::cout << " Ideal Hit Producer -Point-: " << point << std::endl;
      if ( ! point) 
	continue;

      // Detector ID
      detID = point->GetDetectorID();

      // MCTrack ID
      trackID = point->GetTrackID();

     TVector3
	position(point->GetX(), point->GetY(), point->GetZ());
	dpos.SetXYZ(0.005,0.005,0.005);
 
 
      // Create new hit
      new ((*fHitArray)[iPoint]) MvdHit(trackID, detID, point->GetDetName(), position, dpos, iPoint, point->GetEnergyLoss(),1);
	std::cout << "Hit created for module: " << point->GetDetName() << std::endl;
      
      
      
    }   // Loop over MCPoints

  // Event summary
  std::cout << "-I- MvdHitProducerIdeal: " << nPoints << " MvdPoints, "
       << nPoints << " Hits created." << std::endl;

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


ClassImp(MvdHitProducerIdeal)