CloverGainDriftAna.h

Go to the documentation of this file.

#include "PostProcBase.h"
 
 
#include "TH2.h"
#include "TGraph.h"
 
namespace NSDAna
{
struct clovGainTreeEntry
{
public:
  int16_t  detID;
  int16_t  elementID;
  uint16_t amp;
  int32_t wallTime;
};
 
class CloverGainDriftAna : public PostProcBase<clovGainTreeEntry>
{
public:
  CloverGainDriftAna();
  int addFile(string a_filename);
  std::vector<TH2*> buildTimeHistory();
  void setStartDate(int a_year,
                    int a_month,
                    int a_day
                    );
  void setStopDate(int a_year,
                   int a_month,
                   int a_day
                   );
  void setStartBin(int a_bin);
  void setStopBin(int a_bin);
  void setTStep(int a_tStepS);
  void setSearchStart(double m_searchStart);
  void setSearchStop(double m_searchStop );
  void setSearchStep(double m_searchStep );
 
  void readTreeIntoMemory(double a_dsFactor = 1,
                          double a_minAmp = -1
                          );
  // allows for loading single leaf into memory
  void readTreeIntoMemory(int a_cloverID, int a_leafID);
  std::ostream& printTimeChunkList(std::ostream& a_stream,
                                   int           a_detID,
                                   int           a_elementID
                                  );
  TH1* getHistForTimeChunk(int    a_detID,
                           int    a_elementID,
                           int    a_mapLocation,
                           double a_ampScalar=1.0
                          );
  TH2* getHistFromMap(int    a_detID,
                      int    a_elementID);
  void writeMapHistsToFile(std::string a_fileName);
  std::map<double,double> getHistShapeTest(int    a_detID,
                               int    a_elementID,
                               int    a_mapLocation1,
                               int    a_mapLocation2,
                               double a_searchStart = -1.0,
                               double a_searchStop = -1.0,
                               double a_searchStep = -1.0
                               );
 
  TGraph* getHistShapeTestPlot(int    a_detID,
                               int    a_elementID,
                               int    a_mapLocation1,
                               int    a_mapLocation2
                               );
 
  std::pair<double,double> getGainCorrectionFactor (int    a_detID,
                                                      int    a_elementID,
                                                      int    a_mapLocation1,
                                                      int    a_mapLocation2
                                                      );
  std::map<uint32_t,double> getGainCorrectionFactors(int    a_detID,
                                                     int    a_elementID,
                                                     int    a_mapLocation
                                                     );
  std::map<uint32_t,double> recalcGainCorrectionFactors(int    a_detID,
                                                        int    a_elementID,
                                                        int    a_mapLocation,
                                                        int    a_mapRegionStart,
                                                        int    a_mapRegionStop,
                                                        double a_searchStart,
                                                        double a_searchStop,
                                                        double a_searchStep
                                                        );
  TGraph* plotGCState(int    a_detID,
                      int    a_elementID
                      );
  virtual std::ostream& printJSONFormat(int           a_detID,
                                        int           a_elementID,
                                        std::ostream& a_stream = std::cout
                                        ) const;
  void saveGCFactorJSON(int         a_detID,
                        int         a_elementID,
                        std::string a_fileName
                        );
 
  TH2* getCorrectedHist(int a_detID,
                        int a_elementID
                        );
 
  TH2* getCorrectedMap(int a_detID,
                       int a_elementID
                       );
  void setGCFactors(int    a_detID,
                    int    a_elementID,
                    int    a_startMapElement,
                    int    a_stopMapElement,
                    double a_gcFactor
                    );
private:
  void setTreeName();
  //not useful for this overload
  void setBranchName(){};
  int32_t m_startTime;
  int32_t m_stopTime;
  int m_ampBins;
  double m_searchStart;
  double m_searchStop ;
  double m_searchStep ;
  TH2* buildEmptyTimeHistoryHist(int a_detID, int a_elementID);
 
  TH2* buildEmptyMapHist(int a_detID, int a_elementID,int a_mapLength);
 
  typedef std::map<uint32_t, std::vector<uint16_t> > timeHistogram;
  std::map<std::pair<uint16_t,uint16_t>,timeHistogram > m_detTimeHistory;
  bool m_treeRead;
  std::map<string,int> m_histNames;
  std::map<std::pair<uint16_t,uint16_t>,
           std::map<uint32_t,double> > m_gainCorrection;
  //once a correction has been determined, this should be true.
  std::map<std::pair<uint16_t,uint16_t>,bool> m_hasCorrection;
  //used to track when hitting boundary conditions
  std::map<std::pair<uint16_t,uint16_t>,std::map<uint32_t,bool> > m_nearBoundary;
 
 
  int m_startBin;
  int m_stopBin;
 
  int m_tStep;
 
 
};
 
}