CbmStsHitProducerIdeal.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                CbmStsHitProducerIdeal source file             -----
// -----                  Created 10/01/06  by V. Friese               -----
// -------------------------------------------------------------------------
#include "CbmStsHitProducerIdeal.h"
 
#include "CbmStsHit.h"
#include "CbmStsPoint.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
 
#include <iostream>
 
using std::cout;
using std::endl;
 
// -----   Default constructor   -------------------------------------------
CbmStsHitProducerIdeal::CbmStsHitProducerIdeal()
    : FairTask("Ideal STS Hit Producer")
    , fPointArray(NULL)
    , fHitArray()
{}
// -------------------------------------------------------------------------
 
// -----   Destructor   ----------------------------------------------------
CbmStsHitProducerIdeal::~CbmStsHitProducerIdeal() {}
// -------------------------------------------------------------------------
 
// -----   Public method Init   --------------------------------------------
InitStatus CbmStsHitProducerIdeal::Init()
{
 
    // Get RootManager
    FairRootManager* ioman = FairRootManager::Instance();
    if (!ioman) {
        cout << "-E- CbmStsHitProducerIdeal::Init: "
             << "RootManager not instantised!" << endl;
        return kFATAL;
    }
 
    // Get input array
    fPointArray = (TClonesArray*)ioman->GetObject("StsPoint");
    if (!fPointArray) {
        // cout << "-W- CbmStsHitProducerIdeal::Init: "<< "No STSPoint array!" << endl;
        return kERROR;
    }
 
    // Create and register output array
    fHitArray = new TClonesArray("CbmStsHit");
    ioman->Register("StsHit", "STS", fHitArray, kTRUE);
 
    cout << "-I- CbmStsHitProducerIdeal: Intialisation successfull" << endl;
    return kSUCCESS;
}
// -------------------------------------------------------------------------
 
// -----   Public method Exec   --------------------------------------------
void CbmStsHitProducerIdeal::Exec(Option_t* opt)
{
 
    // Reset output array
    if (!fHitArray)
        Fatal("Exec", "No StsHit array");
 
    //  fHitArray->Clear();
    fHitArray->Delete();
 
    // Declare some variables
    CbmStsPoint* 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 StsPoints
    Int_t nPoints = fPointArray->GetEntriesFast();
    for (Int_t iPoint = 0; iPoint < nPoints; iPoint++) {
        point = (CbmStsPoint*)fPointArray->At(iPoint);
        if (!point)
            continue;
 
        // Detector ID
        detID = point->GetDetectorID();
 
        // MCTrack ID
        trackID = point->GetTrackID();
 
        // Determine hit position (centre plane of station)
        x = 0.5 * (point->GetXOut() + point->GetXIn());
        y = 0.5 * (point->GetYOut() + point->GetYIn());
        z = 0.5 * (point->GetZOut() + point->GetZIn());
 
        // Create new hit
        pos.SetXYZ(x, y, z);
        dpos.SetXYZ(dx, dx, 0.);
        new ((*fHitArray)[iPoint]) CbmStsHit(detID, pos, dpos, 0., iPoint, -1);
 
    }   // Loop over MCPoints
 
    // Event summary
    cout << "-I- CbmStsHitProducerIdeal: " << nPoints << " StsPoints, " << nPoints << " Hits created." << endl;
}
// -------------------------------------------------------------------------