TriggerPlot.h

Go to the documentation of this file.

 
#ifndef __TriggerPlot__
#define __TriggerPlot__
 
#include "GwollumPlot.h"
#include "ReadTriggers.h"
#include "TriggerSelect.h"
#include "Date.h"
#include "TF1.h"
#include "TH1.h"
#include "TH2.h"
#include "TGraph.h"
 
using namespace std;
 
class TriggerPlot: public GwollumPlot, public ReadTriggers {
 
 public:
 
  TriggerPlot(const unsigned int aNcoll, const string aPattern, const string aDirectory="",
              const string aStyleName="GWOLLUM", const unsigned int aVerbose=0);
  
  virtual ~TriggerPlot(void);
  void MakeCollection(const int aCollIndex=-1, Segments *aInSeg=NULL);
 
  inline void PrintPlot(TObject *aObj){
    GwollumPlot::ResetLegend(); GwollumPlot::Wpad->Clear(); GwollumPlot::Draw(aObj);
  };
 
  inline void PrintPlot(const string aPlotType,
                        const unsigned int aCollIndex,
                        const bool aFirst=true){
    vector <unsigned int> ind(1, aCollIndex);
    return PrintPlot(aPlotType, ind, aFirst);
  };
  
  inline void PrintPlot(const string aPlotType){
    vector <unsigned int> ind;
    for(unsigned int i=0; i<Ncoll; i++) ind.push_back(i);
    return PrintPlot(aPlotType, ind, true);
    
  };
  
  void PrintPlot(const string aPlotType,
                 const vector<unsigned int> aCollIndex,
                 const bool aFirst);
 
  inline TriggerSelect* GetCollectionSelection(const unsigned int aCollIndex){
    return Collection[aCollIndex];
  };
 
  void SetSelectionSegments(Segments *SelectionSegments);
 
  inline void SetCollectionLegend(const unsigned int aCollIndex,
                                  const string aLegend){
    legend[aCollIndex]=aLegend;
  };
 
  inline void SetCollectionColor(const unsigned int aCollIndex,
                                 const unsigned int aColorIndex){
    Hfreqtime[aCollIndex]->SetMarkerColor(aColorIndex);
  };
 
  inline void SetCollectionMarker(const unsigned int aCollIndex,
                                  const unsigned int aMarkerStyleIndex,
                                  const double aMarkerSize=1.0){
    Hfreqtime[aCollIndex]->SetMarkerStyle(aMarkerStyleIndex);
    Hfreqtime[aCollIndex]->SetMarkerSize(aMarkerSize);
  };
 
  inline unsigned int GetCollectionN(const unsigned int aCollIndex){
    return (unsigned int)Gfreqtime[aCollIndex]->GetN();
  };
 
  TH1D* GetTH1D(const string aPlotType, const unsigned int aCollIndex);
 
  TGraph* GetTGraph(TH1D** aFrame, const string aPlotType, const unsigned int aCollIndex);
 
  inline unsigned int GetN(void){ return Ncoll; };
  
  inline double GetYmax(const string aPlotType, const int aCollIndex){
    if(!aPlotType.compare("snrtime"))
      return Hsnrtime[aCollIndex]->GetYaxis()->GetXmax();
    else if(!aPlotType.compare("snrfreq"))
      return Hsnrfreq[aCollIndex]->GetYaxis()->GetXmax();
    else if(!aPlotType.compare("rate"))
      return Htime[aCollIndex]->GetYaxis()->GetXmax();
    else if(!aPlotType.compare("frequency"))
      return Hfreq[aCollIndex]->GetYaxis()->GetXmax();
    else if(!aPlotType.compare("snr"))
      return Hsnr[aCollIndex]->GetYaxis()->GetXmax();
    else
      return Hfreqtime[aCollIndex]->GetYaxis()->GetXmax();
  };
  
  inline double GetYmin(const string aPlotType, const int aCollIndex){
    if(!aPlotType.compare("snrtime"))
      return Hsnrtime[aCollIndex]->GetYaxis()->GetXmin();
    else if(!aPlotType.compare("snrfreq"))
      return Hsnrfreq[aCollIndex]->GetYaxis()->GetXmin();
    else if(!aPlotType.compare("rate"))
      return Htime[aCollIndex]->GetYaxis()->GetXmin();
    else if(!aPlotType.compare("frequency"))
      return Hfreq[aCollIndex]->GetYaxis()->GetXmin();
    else if(!aPlotType.compare("snr"))
      return Hsnr[aCollIndex]->GetYaxis()->GetXmin();
    else
      return Hfreqtime[aCollIndex]->GetYaxis()->GetXmin();
  };
  
  inline void PlotLoudestEvent(const bool aFlag=true){ plotstar=aFlag; };
  
  inline void SetRateRange(const double aRateMin, const double aRateMax){
    rate_min = aRateMin;
    rate_max = aRateMax;
  };
  
 
 private:
  
  // PLOTS
  unsigned int Ncoll;        
  TH2D   **Hfreqtime;        
  TGraph **Gfreqtime;        
  TGraph *Gfreqtimeloud;     
  TGraph *Gfreqtimeloud2;    
  TH2D   **Hsnrtime;         
  TGraph **Gsnrtime;         
  TH2D   **Hsnrfreq;         
  TGraph **Gsnrfreq;         
  TH1D   **Htime;            
  TH1D   **Hfreq;            
  TH1D   **Hsnr;             
 
   // STYLE
  string *legend;            
  bool plotstar;             
  double rate_min;           
  double rate_max;           
 
  // SELECTION
  TriggerSelect **Collection;
  Segments *selseg;          
  
  // BOX
  TPave *SegBox;             
  TPave **OnBox;             
  TPave **SelBox;            
  unsigned int nonbox;       
  unsigned int nselbox;      
 
  void ResetCollection(const unsigned int aCollIndex);
 
  void ResetParameters(const unsigned int aCollIndex);
 
 
  ClassDef(TriggerPlot,0)  
};
 
#endif