PndFTSCADisplay.h

Go to the documentation of this file.

//-*- Mode: C++ -*-
// @(#) $Id: PndFTSCADisplay.h,v 1.8 2016/12/16 19:39:31 mpugach Exp $
//  *************************************************************************
//  This file is property of and copyright by the ALICE HLT Project         *
//  ALICE Experiment at CERN, All rights reserved.                          *
//  See cxx source for full Copyright notice                                *
//                                                                          *
//  PndFTSCADisplay class is a debug utility.                            *
//  It is not used in the normal data processing.                           *
//                                                                          *
//***************************************************************************

#ifndef PNDFTSCADISPLAY_H
#define PNDFTSCADISPLAY_H

#include "PndFTSCAParameters.h"
#include "PndFTSPerformanceBase.h"
#include "PndFTSArray.h"
#include "FTSCANPletsV.h"

class PndFTSCAGBTracker;
class PndFTSCATracker;
class PndFTSCATrack;
class PndFTSCATrackParam;
class PndFTSCAParam;
class PndFTSCAPerformance;
// class PndFTSCAMCTrack;
// class PndFTSCALocalMCPoint;
class TCanvas;
class TPad;
class FTSCAHit;
class FTSCAHitV;
class FTSCAHitsV;
class FTSCATracks;

#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 PndFTSCADisplay {

 public:
  class PndFTSCADisplayTmpHit;

  static PndFTSCADisplay &Instance();

  PndFTSCADisplay();

  virtual ~PndFTSCADisplay();

  void Init();
  void Update();
  void ClearView();
  void Ask();
  void SetTPC(const PndFTSCAParam &tpcParam);
  void SetGB(const PndFTSCAGBTracker *GBTracker);

  const PndFTSCAGBTracker *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 PndFTSCADisplayTmpHit *vHits, int NHits);

  void DrawTPC();
  void DrawSlice(PndFTSCATracker *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(PndFTSCATrackParam t, double Alpha, const PndFTSCADisplayTmpHit *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 FTSCAHit &hit, float &x, float &y, float &z);
  void HitToGlobal(const FTSCAHitV &hit, int iV, float &x, float &y, float &z);

  void DrawGBHits(const FTSCAHitsV &all);

  void DrawGBHits(const FTSCAHits &all);

  void DrawGBPoints();
  void DrawPVHisto(const vector<float> &pvHist, const PndFTSCAParam &param);
  void DrawGBNPlets(const FTSCANPletsV &all);

  void DrawGBNPlets(const FTSCAElementsOnStation<FTSCANPletV> &s);

  void DrawGBTracks(const FTSCATracks &all);

#if !defined(PANDA_FTS)
  void DrawGBTrackFast(const PndFTSCAGBTracker &tracker, int itr, int color = -1);
#endif
  bool DrawTracklet(PndFTSCATrackParam &track, const int *hitstore, int color = -1, int width = -1, bool pPoint = 0);

  //     void DrawGBLinks( const PndFTSCAGBTracker &tracker, int color = -1, Size_t width = -1 );

  void DrawGBHit(const PndFTSCAGBTracker &tracker, int iHit, int color = -1, Size_t width = -1);
  void DrawGBHits(const PndFTSCAGBTracker &tracker, int color = -1, Size_t width = -1, int hitsType = -1);

#if !defined(PANDA_FTS)
  void DrawTrackParam(PndFTSCATrackParam t, int color = 1);
#endif

  void SaveCanvasToFile(TString fileName);

  TPad *CanvasYX() { return fYX; }
  TPad *CanvasZX() { return fZX; }

 protected:
  TCanvas *fCanvas;                  // the canvas
  TPad *fYX,                         // in case of PANDA_FTS fYX states for XY plane
    *fZX, *fZR;                      // two views
  bool fAsk;                         // flag to ask for the pressing key
  const PndFTSCAGBTracker *fGB;      // the Aliglobal tracker
  PndFTSCAPerformance *fPerf;        // Performance class (mc labels etc)
  double fZMin, fZMax, fYMin, fYMax; // view parameters
#ifdef PANDA_FTS
  double fXMin, fXMax;
#else
  double fRInnerMin, fRInnerMax, fROuterMin, fROuterMax, fTPCZMin, fTPCZMax; // view parameters
#endif

  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:
  PndFTSCADisplay(const PndFTSCADisplay &);
  PndFTSCADisplay &operator=(const PndFTSCADisplay &);
};

#endif