MpdPointSetDraw.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                     MpdPointSetDraw source file               -----
// -----                  Created 03/01/08  by M. Al-Turany            -----
// -------------------------------------------------------------------------
#include "MpdPointSetDraw.h"
 
#include "FairRootManager.h"
#include "FairLogger.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()
{}
 
void MpdPointSetDraw::Finish()
{}
 
// -------------------------------------------------------------------------
void MpdPointSetDraw::Reset()
{
    if (fq != 0)
    {
        fq->Reset();
        RemoveEveElementList();
    }
}