PndCADisplay.h

Go to the documentation of this file.

#ifndef PNDCADISPLAY_H
#define PNDCADISPLAY_H

#ifdef CATRACKER_DISPLAY

#include "PndCAParameters.h"
#include "PndCAPerformanceBase.h"
#include "PndCANPletsV.h"

class PndCAGBTracker;
class PndCATracker;
class PndCATrack;
class PndCATrackParam;
class PndCAParam;
class PndCAPerformance;
class PndCAMCTrack;
class PndCALocalMCPoint;
class TCanvas;
class TPad;
class PndCAHit;
class PndCAHitV;
class PndCAHitsV;
class PndCATracks;

#include "TArc.h"
#include "TLine.h"
#include "TPolyLine.h"
#include "TArrow.h"
#include "TBox.h"
#include "TCrown.h"
#include "TMarker.h"
#include "TLatex.h"

#include <vector>
using std::vector;


class PndCADisplay
{
    
public:
    
    class PndCADisplayTmpHit;
    
    static PndCADisplay &Instance();
    
    PndCADisplay();
    
    virtual ~PndCADisplay();
    
    void Init();
    void Update();
    void ClearView();
    void Ask();
    void SetTPC( const PndCAParam& tpcParam);  // iklm
    void SetGB( const PndCAGBTracker * GBTracker );
    
    const PndCAGBTracker * GetGB(){return fGB;};
    int GetColor( int i ) const;
    int GetColorZ( double z ) const ;
    int GetColorY( double y ) const ;
    int GetColorK( double k ) const ;
    int GetTrackMC( const PndCADisplayTmpHit *vHits, int NHits );
    
    
    void DrawTPC();
    void DrawFwdMvd();
    void DrawFwdGBHits( const PndCAGBTracker &tracker);
    void DrawSlice( PndCATracker *slice, bool DrawRows = 0, bool DrawGrid = 1 );
    
    void DrawArc(float x, float y, float r, int Start = 1, Size_t width = 1 );
    void DrawPoint(float x, float y, float z, int Start = 1, Size_t width = 1 );
    
    void DrawGBPoint(float x, float y, float z, int Start = 1, Size_t width = 1 );
    void DrawGBPoint(float x, float y, float z, float angle = 0, int Start = 1, Size_t width = 1 );
    void DrawSliceOutTrackParam( int itr, int color, Size_t width );
    void DrawSliceOutTrack1( int itr, int color, Size_t width );
    void DrawHelix(float p0, float c, float z, float zStart, float z0, float xc, float yc, float r, float b, int color, Size_t width);
    void DrawParticleGlobal(float *param, float q, float tStart, float tEnd, float b, int color=kOrange, Size_t width = 1);
    void DrawParticleGlobal(float *param, float q, float n[4], float b, int color=kOrange, Size_t width = 1);
    void DrawGBLine(float x, float y, float z, float x2, float y2, float z2, int Start=1, Size_t width=1, int projection = -1 );
    
    bool DrawTrack( PndCATrackParam t, double Alpha, const PndCADisplayTmpHit *vHits,
                   int NHits, int color = -1, Size_t width = -1, bool pPoint = 0 );
    
    void DrawGBTrack( int itr, int color = -1, int width = -1 );
    void DrawRecoTrack( int itr, int color = -1, int width = -1 );
    void DrawMCTrack( int itr, int color = -1, int width = -1 );
    
    void HitToGlobal( const PndCAHit& hit, float& x, float& y, float &z );
    void HitToGlobal( const PndCAHitV& hit, int iV, float& x, float& y, float &z );
    
    void DrawGBHits    (const PndCAHits&     all);
    void DrawGBPoints();
    void DrawPVHisto(const vector<float>& pvHist, const PndCAParam& param);
    void DrawGBNPlets(const PndCANPletsV& all);
    void DrawGBTracks(const PndCATracks&  all);
    
    void DrawGBTrackFast( const PndCAGBTracker &tracker, int itr, int color = -1 );
    bool DrawTracklet( PndCATrackParam &track, const int *hitstore, int color = -1, int width = -1, bool pPoint = 0 );
    
    //     void DrawGBLinks( const PndCAGBTracker &tracker, int color = -1, Size_t width = -1 );
    
    void DrawGBHit( const PndCAGBTracker &tracker, int iHit, int color = -1, Size_t width = -1 );
    void DrawGBHits( const PndCAGBTracker &tracker, int color = -1, Size_t width = -1, int hitsType = -1 );
    
    void DrawTrackParam( PndCATrackParam t, int color = 1 );
    
    void SaveCanvasToFile( TString fileName);
    
    TPad *CanvasYX() { return fYX; }
    TPad *CanvasZX() { return fZX; }
    
protected:
    
    TCanvas *fCanvas;              // the canvas
    TPad *fYX, *fZX, *fZR;               // two views
    TCanvas *fFwdMvdCanvas;//, *fFwdMvdXYCanvas;              // the canvas
    TPad *fFwdMvdZY, *fFwdMvdZX, *fFwdMvdZR; //*fFwdMvdXY;               // two views
    bool fAsk;                      // flag to ask for the pressing key
    const PndCAGBTracker *fGB;      // the Aliglobal tracker
    PndCAPerformance *fPerf; // Performance class (mc labels etc)
    double fZMin, fZMax, fYMin, fYMax;// view parameters
    double fRInnerMin, fRInnerMax, fROuterMin, fROuterMax, fTPCZMin, fTPCZMax; // view parameters
    
    TArc fArc;       // parameters of drawing objects are copied from this members
    TLine fLine;     
    TPolyLine fPLine;
    TMarker fMarker; 
    TBox fBox;       
    TCrown fCrown;   
    TLatex fLatex;   
    TArrow fArrow;
    
    bool fDrawOnlyRef; // draw only clusters from ref. mc tracks
    
private:
    PndCADisplay( const PndCADisplay& );
    PndCADisplay& operator=( const PndCADisplay& );
};

#endif //CATRACKER_DISPLAY

#endif