MpdBoxSetDraw.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                        MpdBoxSetDraw source file                  -----
// -----                  Created 03/01/08  by M. Al-Turany            -----
// -------------------------------------------------------------------------
#include "MpdBoxSetDraw.h"
 
#include "FairRunAna.h"
#include "MpdBoxSet.h"
#include "TEveManager.h"   // for TEveManager, gEve
 
#include <iostream>
using namespace std;
 
// -----   Default constructor   -------------------------------------------
MpdBoxSetDraw::MpdBoxSetDraw()
    : FairTask("MpdBoxSetDraw", 0)
    , fVerbose(0)
    , fList(NULL)
    , fEventManager(NULL)
    , fManager(NULL)
    , fq(NULL)
    , fX(0.3)
    , fY(0.3)
    , fZ(0.3)
    , fTimeWindowPlus(0.)
    , fTimeWindowMinus(0.)
    , fStartTime(0.)
    , fUseEventTime(kTRUE)
    , isRedraw(kFALSE)
    , fStartFunctor()
    , fStopFunctor()
{}
 
// -----   Standard constructor   ------------------------------------------
MpdBoxSetDraw::MpdBoxSetDraw(const char* name, Int_t iVerbose)
    : FairTask(name, iVerbose)
    , fVerbose(iVerbose)
    , fList(NULL)
    , fEventManager(NULL)
    , fManager(NULL)
    , fq(NULL)
    , fX(0.3)
    , fY(0.3)
    , fZ(0.3)
    , fTimeWindowPlus(0.)
    , fTimeWindowMinus(0.)
    , fStartTime(0.)
    , fUseEventTime(kTRUE)
    , isRedraw(kFALSE)
    , fStartFunctor()
    , fStopFunctor()
{}
 
// -----   Destructor   ----------------------------------------------------
MpdBoxSetDraw::~MpdBoxSetDraw() {}
 
InitStatus MpdBoxSetDraw::Init()
{
    if (fVerbose > 1)
        cout << "MpdBoxSetDraw::Init()" << endl;
 
    fManager = FairRootManager::Instance();
 
    fList = (TClonesArray*)FairRootManager::Instance()->GetObject(GetName());
    if (fList == 0) {
        cout << "MpdBoxSetDraw::Init()  branch " << GetName() << " not found! Task will be deactivated" << endl;
        SetActive(kFALSE);
        return kERROR;
    }
    if (fVerbose > 2)
        cout << "MpdBoxSetDraw::Init() get track list " << fList << endl;
 
    fEventManager = MpdEventManager::Instance();
    if (fVerbose > 2)
        cout << "MpdBoxSetDraw::Init() get instance of MpdEventManager" << endl;
 
    fq = 0;
    fStartFunctor = new StopTime();
    fStopFunctor = new StopTime();
 
    return kSUCCESS;
}
// -------------------------------------------------------------------------
void MpdBoxSetDraw::Exec(Option_t* /*option*/)
{
    if (!IsActive())
        return;
 
    Reset();
 
    CreateBoxSet();
    if (FairRunAna::Instance()->IsTimeStamp()) {
        fList->Clear();
        Double_t eventTime = fManager->GetEventTime();
        if (fUseEventTime)
            fStartTime = eventTime - fTimeWindowMinus;
        cout << "EventTime: " << eventTime << " TimeWindow: " << fStartTime << " - " << eventTime + fTimeWindowPlus
             << endl;
 
        fList = fManager->GetData(GetName(), fStartFunctor, fStartTime, fStopFunctor,
                                  eventTime + fTimeWindowPlus);   // FairRootManager::Instance()->GetEventTime() +
    }
    if (fVerbose > 1)
        cout << GetName() << " fList: " << fList->GetEntries() << endl;
 
    for (Int_t i = 0; i < fList->GetEntriesFast(); i++) {
        TObject* p = fList->At(i);
        AddBoxes(fq, p, i);
    }
 
    gEve->AddElement(fq, fEventManager);
    if (isRedraw)
        gEve->Redraw3D();
}
 
void MpdBoxSetDraw::AddBoxes(MpdBoxSet* set, TObject* obj, Int_t i)
{
    TVector3 point = GetVector(obj);
    set->AddBox(point.X(), point.Y(), point.Z());
    set->DigitValue(GetValue(obj, i));
    if (fVerbose > 2)
        cout << "MpdBoxSetDraw::Init() Add point " << i << ": " << point.X() << " " << point.Y() << " " << point.Z()
             << " " << endl;
}
 
Int_t MpdBoxSetDraw::GetValue(TObject* /*obj*/, Int_t i)
{
    return i;
}
 
MpdBoxSet* MpdBoxSetDraw::CreateBoxSet()
{
    MpdBoxSet* aBoxSet = new MpdBoxSet(this, GetName());
    aBoxSet->Reset(MpdBoxSet::kBT_AABoxFixedDim, kFALSE, 64);
    aBoxSet->SetDefWidth(fX);
    aBoxSet->SetDefHeight(fY);
    aBoxSet->SetDefDepth(fZ);
 
    fq = aBoxSet;
 
    return aBoxSet;
}
 
void MpdBoxSetDraw::SetParContainers() {}
 
/** Action after each event**/
void MpdBoxSetDraw::Finish() {}
 
void MpdBoxSetDraw::Reset()
{
    if (fq != 0) {
        fq->Reset();
        gEve->RemoveElement(fq, fEventManager);
    }
}