BmnTof1HitProducerIdeal.cxx

Go to the documentation of this file.

#include "BmnTof1HitProducerIdeal.h"
 
#include "BmnTOF1Point.h"
#include "BmnTofHit.h"
#include "FairLogger.h"
#include "FairRootManager.h"
#include "TFile.h"
 
#include <assert.h>
 
BmnTof1HitProducerIdeal::BmnTof1HitProducerIdeal(const char* name,
                                                 Bool_t useMCdata,
                                                 Int_t verbose,
                                                 Bool_t test,
                                                 Bool_t merge)
    : BmnTask(name, verbose)
    , aMcPoints(nullptr)
    , aMcTracks(nullptr)
    , aExpDigits(nullptr)
    , aTofHits(nullptr)
    , fDoTest(test)
    , fDoMergeHits(merge)
    , fUseMCData(useMCdata)
    , fOnlyPrimary(false)
    , fTestFlnm("")
{}
 
BmnTof1HitProducerIdeal::~BmnTof1HitProducerIdeal() {}
 
InitStatus BmnTof1HitProducerIdeal::Init()
{
    LOG(info) << "[BmnTof1HitProducerIdeal::Init] Begin initialization.";
 
    FairRootManager* ioman = FairRootManager::Instance();
    assert(ioman);
 
    if (fUseMCData) {
        aMcPoints = (TClonesArray*)FairRootManager::Instance()->GetObject("TOF400Point");
        if (!aMcPoints) {
            cout << "BmnTof1HitProducerIdeal::Init(): branch TOF1Point not found! Task will be deactivated" << endl;
            SetActive(kFALSE);
            return kERROR;
        }
        aMcTracks = (TClonesArray*)FairRootManager::Instance()->GetObject("MCTrack");
        if (!aMcTracks) {
            cout << "BmnTof1HitProducerIdeal::Init(): branch MCTrack not found! Task will be deactivated" << endl;
            SetActive(kFALSE);
            return kERROR;
        }
    } else {
        aExpDigits = (TClonesArray*)FairRootManager::Instance()->GetObject("bmn_tof400");
        if (!aExpDigits) {
            cout << "BmnTof1HitProducerIdeal::Init(): branch bmn_tof400 not found! Task will be deactivated" << endl;
            SetActive(kFALSE);
            return kERROR;
        }
    }
 
    // Create and register output array
    aTofHits = new TClonesArray("BmnTofHit");
    ioman->Register("TOF400Hit", "TOF1", aTofHits, kTRUE);
 
    LOG(info) << "[BmnTof1HitProducerIdeal::Init] Initialization finished succesfully.";
 
    return kSUCCESS;
}
 
void BmnTof1HitProducerIdeal::Exec(Option_t* opt)
{
    if (!IsActive())
        return;
 
    static const TVector3 XYZ_err(0., 0., 0.);   // FIXME:
 
    aTofHits->Clear();
 
    Int_t nSingleHits = 0;
    TVector3 pos;
 
    if (fUseMCData) {
        BmnTOF1Point* pPoint;
        Int_t nTofPoint = aMcPoints->GetEntriesFast();
 
        for (Int_t pointIndex = 0; pointIndex < nTofPoint; pointIndex++)   // cycle by TOF points
        {
            pPoint = (BmnTOF1Point*)aMcPoints->UncheckedAt(pointIndex);
            pPoint->Position(pos);
 
            AddHit(pPoint->GetDetectorID(), pos, XYZ_err, pointIndex, pPoint->GetTrackID(), pPoint->GetTime(), kTOF1);
            nSingleHits++;
        }
    }
 
    int nFinally;
    if (fDoMergeHits) {
        MergeHitsOnStrip();          // save only the fastest hit in the strip
        nFinally = CompressHits();   // remove blank slotes
    } else
        nFinally = aTofHits->GetEntriesFast();
 
    cout << " -I- [BmnTof1HitProducerIdeal::Exec]  single hits= " << nSingleHits << ", final hits= " << nFinally
         << endl;
}
 
void BmnTof1HitProducerIdeal::Finish() {}
 
Int_t BmnTof1HitProducerIdeal::MergeHitsOnStrip(void)
{
    typedef map<Int_t, BmnTofHit*> hitsMapType;
    hitsMapType fHits;   // pair<detectorUID, BmnTofHit*> fastest hits map
    hitsMapType::iterator it;
    Int_t mergedNmb = 0;
 
    typedef multiset<Int_t> msUIDsType;   // detectorUID for Hits
    msUIDsType UIDs;
 
    BmnTofHit *fastHit, *slowHit;
    for (Int_t hitIndex = 0, nHits = aTofHits->GetEntriesFast(); hitIndex < nHits; hitIndex++)   // cycle by hits
    {
        BmnTofHit* pHit = (BmnTofHit*)aTofHits->UncheckedAt(hitIndex);
        assert(nullptr != pHit);
 
        Int_t UID = pHit->GetDetectorID();
 
        if (fDoTest)
            UIDs.insert(UID);
 
        it = fHits.find(UID);
        if (it != fHits.end())   // hit for this detectorUID already exist
        {
            mergedNmb++;
 
            if (pHit->GetTimeStamp() < it->second->GetTimeStamp())   //  faster hit  found
            {
                fastHit = pHit;
                slowHit = it->second;
            } else {
                fastHit = it->second;
                slowHit = pHit;
            }
 
            fastHit->AddLinks(slowHit->GetLinks());   // copy links
            aTofHits->Remove(slowHit);                // remove old hit   --> make blank slote !!
            it->second = fastHit;                     // change pair value to current UID
        } else
            fHits.insert(make_pair(UID, pHit));   // insert new detectorUID pair
 
    }   // cycle by hits
 
    return mergedNmb;
}
 
void BmnTof1HitProducerIdeal::AddHit(Int_t detUID,
                                     const TVector3& posHit,
                                     const TVector3& posHitErr,
                                     Int_t pointIndex,
                                     Int_t trackIndex,
                                     Double_t time,
                                     Int_t Idd = kTOF1)
{
    BmnTofHit* pHit = new ((*aTofHits)[aTofHits->GetEntriesFast()]) BmnTofHit(detUID, posHit, posHitErr, pointIndex);
 
    FairRootManager::Instance()->SetUseFairLinks(kTRUE);
    pHit->SetTimeStamp(time);
    pHit->AddLink(FairLink(0x1, pointIndex));
    pHit->AddLink(FairLink(0x2, trackIndex));
    pHit->AddLink(FairLink(0x4, detUID));
    Int_t str = BmnTOF1Point::GetStrip(detUID) - 1;
    Int_t plane = BmnTOF1Point::GetModule(detUID) - 1;
    pHit->SetModule(plane);
    pHit->SetStation(str);
    pHit->SetDetId(DetectorId(Idd));
    FairRootManager::Instance()->SetUseFairLinks(kFALSE);
}
 
Int_t BmnTof1HitProducerIdeal::CompressHits()
{
    aTofHits->Compress();
    return aTofHits->GetEntriesFast();
}
 
void BmnTof1HitProducerIdeal::Dump(const char* title, ostream& out) const
{
    out << "\n [BmnTof1HitProducerIdeal::Dump]   ";
    if (title)
        out << title;
    out << ", size= " << aTofHits->GetEntriesFast();
 
    BmnTOF1Point* point;
    BmnTofHit* pHit;
    TVector3 hitPos, pointPos;
 
    TIterator* iter = aTofHits->MakeIterator();
    while ((pHit = (BmnTofHit*)iter->Next())) {
        pHit->Position(hitPos);
        out << "\n    hit detUID = " << pHit->GetDetectorID() << ", hit pos(" << hitPos.X() << "," << hitPos.Y() << ","
            << hitPos.Z() << "), ";
        if (aMcPoints) {
            point = (BmnTOF1Point*)aMcPoints->UncheckedAt(pHit->GetRefIndex());
            point->Position(pointPos);
            out << "\n point detUID = " << point->GetDetectorID() << ", point pos(" << pointPos.X() << ","
                << pointPos.Y() << "," << pointPos.Z() << "), dev=" << (hitPos - pointPos).Mag();
        }
    }
 
    delete iter;
}