MpdPointSetDraw.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                     MpdPointSetDraw source file               -----
// -----                  Created 03/01/08  by M. Al-Turany            -----
// -------------------------------------------------------------------------
#include "MpdPointSetDraw.h"
 
#include "FairLogger.h"
#include "FairRootManager.h"
#include "TEveManager.h"
#include "TEveTreeTools.h"   // for TEvePointSelectorConsumer
#include "TString.h"         // for Form()
 
#include <iostream>
using namespace std;
 
// -----   Default constructor   -------------------------------------------
MpdPointSetDraw::MpdPointSetDraw()
    : FairTask("MpdPointSetDraw", 0)
    , fVerbose(0)
    , fEventManager(NULL)
    , fPointList(NULL)
    , fq(NULL)
    , fColor(0)
    , fStyle(0)
{}
 
// -----   Standard constructor   ------------------------------------------
MpdPointSetDraw::MpdPointSetDraw(const char* name, Color_t color, Style_t mstyle, Int_t iVerbose)
    : FairTask(name, iVerbose)
    , fVerbose(iVerbose)
    , fEventManager(NULL)
    , fPointList(NULL)
    , fq(NULL)
    , fColor(color)
    , fStyle(mstyle)
{}
 
// -----   Destructor   ----------------------------------------------------
MpdPointSetDraw::~MpdPointSetDraw() {}
 
// -------------------------------------------------------------------------
InitStatus MpdPointSetDraw::Init()
{
    if (fVerbose > 0)
        cout << "MpdPointSetDraw::Init()" << endl;
 
    FairRootManager* fManager = FairRootManager::Instance();
 
    fPointList = (TClonesArray*)fManager->GetObject(GetName());
    if (fPointList == 0) {
        LOG(error) << "MpdPointSetDraw::Init() branch " << GetName() << " not found! Task will be deactivated";
        SetActive(kFALSE);
    }
    if (fVerbose > 1)
        cout << "MpdPointSetDraw::Init() get point list " << fPointList << endl;
 
    fEventManager = MpdEventManager::Instance();
    if (fVerbose > 1)
        cout << "MpdPointSetDraw::Init() get instance of MpdEventManager" << endl;
 
    fq = 0;
 
    return kSUCCESS;
}
 
void MpdPointSetDraw::Exec(Option_t* /*option*/)
{
    if (fVerbose > 0)
        cout << "MpdPointSetDraw::Exec()" << endl;
    if (!IsActive())
        return;
 
    Reset();
 
    Int_t npoints = fPointList->GetEntriesFast();
    if (fVerbose > 0)
        cout << "MpdPointSetDraw::Exec() the number of points is " << fPointList->GetEntries() << endl;
 
    TEvePointSet* q = new TEvePointSet(GetName(), npoints, TEvePointSelectorConsumer::kTVT_XYZ);
    q->SetOwnIds(kTRUE);
    q->SetMarkerColor(fColor);
    q->SetMarkerSize(1.5);
    q->SetMarkerStyle(fStyle);
 
    for (Int_t i = 0; i < npoints; i++) {
        TObject* p = (TObject*)fPointList->At(i);
        if (p != 0) {
            TVector3 vec(GetVector(p));
            q->SetNextPoint(vec.X(), vec.Y(), vec.Z());
            q->SetPointId(GetValue(p, i));
            if (fVerbose > 2)
                cout << "MpdPointSetDraw::Exec() add point to EVE set: " << i << endl;
        } else
            cout << "CRITICAL ERROR: MpdPointSetDraw::Exec() point is not TObject" << endl;
    }
 
    fq = q;
 
    AddEveElementList();
 
    // gEve->Redraw3D(kFALSE);
}
 
TObject* MpdPointSetDraw::GetValue(TObject* /*obj*/, Int_t i)
{
    return new TNamed(Form("Point %d", i), "");
}
 
// -------------------------------------------------------------------------
void MpdPointSetDraw::SetParContainers() {}
 
/** Action after each event**/
void MpdPointSetDraw::Finish() {}
 
// -------------------------------------------------------------------------
void MpdPointSetDraw::Reset()
{
    if (fq != 0) {
        fq->Reset();
        RemoveEveElementList();
    }
}