DisplayManager.h

Go to the documentation of this file.

#ifndef PF_DisplayManager_h
#define PF_DisplayManager_h

#include "RecoParticleFlow/PFRootEvent/interface/DisplayCommon.h"
#include "DataFormats/ParticleFlowReco/interface/PFBrem.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHit.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrack.h"
#include "DataFormats/ParticleFlowReco/interface/PFSimParticle.h"
#include "DataFormats/ParticleFlowReco/interface/PFBrem.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"

#include "DataFormats/HepMCCandidate/interface/GenParticle.h"

#include <TCanvas.h>
#include <TObject.h>
#include <TLine.h>
#include <TBox.h>
#include <TEllipse.h>
#include <string>
#include <map>

class IO;

class GPFRecHit;
class GPFCluster;
class GPFTrack;
class GPFSimParticle;
class GPFBase;
class GPFGenParticle;
class PFRootEventManager;
class TH2F; 

class DisplayManager {
  
 public:
  DisplayManager( PFRootEventManager *em, 
                  const char* optfile );
  virtual ~DisplayManager();
     
  void readOptions( const char* file );

  void display(int ientry);
  void displayEvent( int run, int lumi, int event); 
  void displayAll(bool noRedraw = true);
  void displayNext();
  void displayNextInteresting(int ientry);
  void displayPrevious();
  void displayPFBlock(int blockNb) ;
  void enableDrawPFBlock(bool state);
  void enableDrawBrem(bool state);
  void findAndDraw(int ident);
  //void findAndDrawbis(const int ident);
  void findBlock(int ident) ;
  void lookForGenParticle(unsigned barcode);
  void lookForMaxRecHit(bool ecal);
  void reset();
  void updateDisplay();
  void unZoom();
  void printDisplay(const char* directory="" ) const;
  void printGenParticleInfo(std::string name, int barcode, int barcodeMother);
  void drawWithNewGraphicAttributes();
  void setNewAttrToSimParticles();
     
  //bool getGLoaded() {return isGraphicLoaded_;}  //util to DialogFrame ?
     
  //variables
  //----------------------graphic options variable ---------------------
  
  double clusEnMin_;
  double hitEnMin_;
  double trackPtMin_;
  double gsfPtMin_;
  double particlePtMin_;
  double genParticlePtMin_;
     
  bool drawHits_;
  bool drawTracks_;
  bool drawGsfTracks_;
  bool drawBrems_;
  bool drawClus_;
  bool drawClusterL_;
  bool drawParticles_;
  bool drawGenParticles_;
  bool drawPFBlocks_;
  bool drawHO_;
     
     
  //---------------------- new graphic Container ----------------
  //container of all the graphic Objects of one event 
  std::multimap<int,GPFBase *>  graphicMap_;
  //container of idents/elementNb  of objects within a PFBlock
  std::multimap<int , std::pair <int,int> >   blockIdentsMap_;
  
  //------------- graphic attributes ------------------------------------
  std::vector<float>      trackAttributes_;
  std::vector<float>      gsfAttributes_;
  std::vector<float>      bremAttributes_;
  std::vector<float>      clusterAttributes_;
  
  
  TAttMarker *clusPattern_;
  TAttMarker *clusPSPattern_;
  
  TAttMarker *clusPatternecal_;
  TAttMarker *clusPatternho_;
  TAttMarker *clusPatternhfem_;    

  TAttMarker *trackPatternM_;
  TAttLine   *trackPatternL_;
  TAttMarker *gsfPatternM_;
  TAttLine   *gsfPatternL_;
  TAttMarker *bremPatternM_;
  TAttLine   *bremPatternL_;
  TAttMarker *genPartPattern_;
  
  TAttLine   *simPartPatternL_;
  TAttMarker *simPartPatternPhoton_;
  TAttMarker *simPartPatternElec_ ;
  TAttMarker *simPartPatternMuon_;
  TAttMarker *simPartPatternK_;
  TAttMarker *simPartPatternPi_;
  TAttMarker *simPartPatternProton_;
  TAttMarker *simPartPatternNeutron_;
  TAttMarker *simPartPatternDefault_;
  
  std::vector<TAttMarker *> simPartPatternM_;
  
  
 private:

  PFRootEventManager *em_;
   
  IO*         options_;      

  double maxERecHitEcal_;
  double maxERecHitHcal_;
  double maxERecHitHo_;
  bool   isGraphicLoaded_;
  int    eventNumber_;
  int    maxEvents_;
  double zoomFactor_;
  
  int    shiftId_;

  //-------------- draw Canvas --------------------------------------
  std::vector<TCanvas*> displayView_;

  std::vector<int>      viewSize_; 
        
  std::vector<int>         viewSizeEtaPhi_; 

  std::vector<TH2F*>    displayHist_;

  TEllipse frontFaceECALXY_;

  TBox     frontFaceECALRZ_;

  TEllipse frontFaceHCALXY_;

  TEllipse frontFaceHOXY_;
  
  //----------------  end Draw Canvas ------------------------------------
    
  //std::vector<std::vector<TLine> >            vectGClusterLines_;
  //number of clusterLines by cluster 
  //std::vector<std::vector<int> >              vectClusLNb_;
    
  std::vector<int>                            selectedGObj_;
  std::vector<int>                            badBremsId_;
     
    
  // Display Options read from option file
  void  getDisplayOptions();
     
  //DisplayViews
  void createCanvas();

  void displayCanvas();
     
     
  //create graphicObjects
  void createGRecHit(reco::PFRecHit& rh,int ident, 
                     double maxe, double phi0=0. , int color=4);

  void createGCluster(const reco::PFCluster& cluster,
                      int ident, double phi0 = 0.);

  void createGTrack(reco::PFRecTrack &tr,
                    const std::vector<reco::PFTrajectoryPoint>& points,
                    int ident,double pt,double phi0, double sign, 
                    bool displayInitial, int linestyle, int kfgsfbrem=0);

  void createGPart(const reco::PFSimParticle &ptc, 
                   const std::vector<reco::PFTrajectoryPoint>& points, 
                   int ident,double pt,double phi0, double sign, 
                   bool displayInitial,int markerIndex);
                   
  //  void createGGenParticle(HepMC::GenEvent::particle_const_iterator p);                 
  void createGGenParticle(HepMC::GenParticle* p);                  

  //void createGClusterLines(const reco::PFCluster& cluster,int viewType);                      
  void drawGObject(int ident,int color,bool toInitialColor);

  //fill vectors with graphic Objects
  void loadGraphicObjects();

  void loadGRecHits();

  void loadGClusters();

  void loadGRecTracks();

  void loadGGsfRecTracks();

  void loadGSimParticles();

  void loadGPFBlocks();
  
  void loadGGenParticles();
     
  //void redraw();
  void rubOutGPFBlock();
  
  // 
  void retrieveBadBrems();
  bool findBadBremsId(int ident);
   

 
  // methods
  double getMaxE(int layer) const;
  double getMaxEEcal();
  double getMaxEHcal();
  double getMaxEHo();              
     
     
     
};
#endif