AgataTrackDisplay.h

Go to the documentation of this file.

#ifndef Gw_AgataTrackDisplay
#define Gw_AgataTrackDisplay

#include <map>
#include <vector>

#ifndef ROOT_TArrayF
#include "TArrayF.h"
#endif

#ifndef ROOT_TEveBoxSet
#include "TEveBoxSet.h"
#endif

#ifndef ROOT_TString
#include "TString.h"
#endif

#include "GwLogMessage.h"

namespace Gw {

class AgataTrackDisplay : public TEveBoxSet
{

public:
  AgataTrackDisplay(const Text_t* name);
  virtual ~AgataTrackDisplay();
  
  void SetBoxWidth(Float_t w)  { fBoxWidth = w;}
  void SetBoxHeight(Float_t h) { fBoxHeight = h; }
  
  Float_t GetBoxWidth()  { return fBoxWidth; }
  Float_t GetBoxHeight() { return fBoxHeight; }
  
  Int_t GetSelectedValue() { return fSelectedValue; }
  
  Int_t GetSelectedIdx() { return fSelectedIdx; }
  
  void FirstEnergy() { fTotalEnergyIter = fTotalEnergyArray.begin(); }
  
  Int_t NextEnergy();
  
  Int_t GetNofTracks() { return fNofTracks; }
  
  void DigitSelected(Int_t idx);
  
  void AddNewTrack();
  
  void AddTracklet(Float_t e, Float_t x1, Float_t y1, Float_t z1, Float_t x2, Float_t y2, Float_t z2, Bool_t eTot = false);
  
  void ResetTracks();
  
  void SetMaxEnergy(Float_t e);
  
  Float_t GetMaxEnergy() { return fMaxEnergy; } 
  
  std::pair<Int_t, Int_t> GetIndexes(Int_t index);
  
  void SetStyle(TString s);
  
private:
  
  TEveRGBAPalette *fPalette; // color palette 
  
  Int_t fMaxEnergy;          // max energy

  Float_t fBoxWidth;         // width of track box
  Float_t fBoxHeight;        // height of track box
  
  Int_t fSelectedIdx;        // selected digit index
  Int_t fSelectedValue;      // selected digit value

  TArrayF fBoxVert;          // vertices of track box
  
  Int_t fNofTracks;           // track id
  Int_t fHitIdPerTrack;       // hit id for a given track
  
  TString fStyle;            // flag for display 
    
  std::map<Int_t, std::pair<Int_t,Int_t> > fMapIdx;      // index map for mapping track indexes with BoxSet indexes
  
  std::vector<Int_t>  fTotalEnergyArray;                 // total energy array
  
  std::vector<Int_t>::const_iterator  fTotalEnergyIter;  // total energy iterator

    
  mutable LogMessage fLog;   
  
protected:
  
  Float_t* MakeRecTVert(Float_t x1, Float_t y1, Float_t z1, Float_t x2, Float_t y2, Float_t z2); 

  void MakeSize(Float_t e);
  
  void SetIndexes();

  ClassDef(AgataTrackDisplay, 1);  
};

} // end namespace

#endif