BmnBdHitProducer.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                   BmnBdHitProducer source file                -----
// -----                  Created 12/07/18  by A. Zinchenko            -----
// -------------------------------------------------------------------------
 
#include "BmnBdHitProducer.h"
 
#include "BmnBdHit.h"
#include "BmnBdPoint.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
 
#include <iostream>
 
using std::cout;
using std::endl;
 
// -----   Default constructor   -------------------------------------------
BmnBdHitProducer::BmnBdHitProducer()
    : FairTask("Bmn Bd Hit Producer")
    , fPointArray(NULL)
    , fHitArray()
{}
// -------------------------------------------------------------------------
 
// -----   Destructor   ----------------------------------------------------
BmnBdHitProducer::~BmnBdHitProducer() {}
// -------------------------------------------------------------------------
 
// -----   Public method Init   --------------------------------------------
InitStatus BmnBdHitProducer::Init()
{
 
    // Get RootManager
    FairRootManager* ioman = FairRootManager::Instance();
    if (!ioman) {
        cout << "-E- BmnBdHitProducer::Init: "
             << "RootManager not instantiated!" << endl;
        return kFATAL;
    }
 
    // Get input array
    fPointArray = (TClonesArray*)ioman->GetObject("BdPoint");
    if (!fPointArray) {
        cout << "-W- BmnBdHitProducer::Init: "
             << "No BdPoint array!" << endl;
        return kERROR;
    }
 
    // Create and register output array
    fHitArray = new TClonesArray("BmnBdHit");
    ioman->Register("BdHit", "BD", fHitArray, kTRUE);
 
    cout << "-I- BmnBdHitProducer: Intialization successfull" << endl;
    return kSUCCESS;
}
// -------------------------------------------------------------------------
 
// -----   Public method Exec   --------------------------------------------
void BmnBdHitProducer::Exec(Option_t* opt)
{
 
    // Reset output array
    if (!fHitArray)
        Fatal("Exec", "No BdHit array");
 
    fHitArray->Delete();
    fHitMap.clear();
 
    // Declare some variables
    BmnBdPoint* point = NULL;
    Int_t detID = 0;   // Detector ID
    // Int_t trackID = 0;        // Track index
    Double_t x, y, z;       // Position
    Double_t dx = 0.0001;   // Position error
    TVector3 pos, dpos;     // Position and error vectors
 
    // Loop over BdPoints
    Int_t nPoints = fPointArray->GetEntriesFast();
 
    for (Int_t iPoint = 0; iPoint < nPoints; iPoint++) {
        point = (BmnBdPoint*)fPointArray->UncheckedAt(iPoint);
        if (!point)
            continue;
 
        // Detector ID
        detID = point->GetCopy();
 
        // MCTrack ID
        // trackID = point->GetTrackID();
 
        // Hit position
        x = point->GetX();
        y = point->GetY();
        z = point->GetZ();
 
        // Create new hit
        pos.SetXYZ(x, y, z);
        dpos.SetXYZ(dx, dx, 0.);
        // new ((*fHitArray)[iPoint]) BmnBdHit(detID, pos, dpos, iPoint);
        if (fHitMap.find(detID) == fHitMap.end())
            fHitMap[detID] = BmnBdHit(detID, pos, dpos, iPoint);
        // Add point to the hit links
        fHitMap[detID].AddLink(FairLink(0, iPoint));
 
    }   // Loop over MCPoints
 
    // Store hits
    Int_t ihit = 0;
    for (map<Int_t, BmnBdHit>::iterator mit = fHitMap.begin(); mit != fHitMap.end(); ++mit)
        new ((*fHitArray)[ihit++]) BmnBdHit(mit->second);
 
    // Event summary
    cout << "-I- BmnBdHitProducer: " << nPoints << " BdPoints, " << fHitMap.size() << " Hits created." << endl;
}
// -------------------------------------------------------------------------