TreeShapeErrorGetter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef LArSamples_TreeShapeErrorGetter_H
#define LArSamples_TreeShapeErrorGetter_H
 
#include "LArSamplesMon/AbsShapeErrorGetter.h"
 
 
#include <vector>
#include "CxxUtils/checker_macros.h"
 
class TH2D;
class TFile;
class TTree;
class TString;
 
namespace LArSamples {
 
  class ShapeErrorData;
  class ResidualCalculator;
  class Interface;
  
  class ATLAS_NOT_THREAD_SAFE TreeShapeErrorGetter : public AbsShapeErrorGetter {
  
    public:
      
      TreeShapeErrorGetter(const TString& fileName, bool recreate = false);
      TreeShapeErrorGetter(const TreeShapeErrorGetter &) = delete;
      TreeShapeErrorGetter& operator = (const TreeShapeErrorGetter &) = delete;
      virtual ~TreeShapeErrorGetter();
 
      virtual std::unique_ptr<ShapeErrorData> shapeErrorData(unsigned int hash, CaloGain::CaloGain gain, const Residual* toExclude = 0) const override;
      virtual std::unique_ptr<ShapeErrorData> phiSymShapeErrorData(short ring, CaloGain::CaloGain gain, const Residual* toExclude = 0) const override;
      
      int addCell(const ResidualCalculator& calc, CaloGain::CaloGain gain);
      int addRing(const ResidualCalculator& calc, CaloGain::CaloGain gain);
 
      void dump(CaloGain::CaloGain gain) const;
      std::unique_ptr<TH2D> correlate(const TreeShapeErrorGetter& other, CaloGain::CaloGain gain, unsigned short sample, bool xip,
                                      unsigned int nBins, double xMin, double xMax) const;
 
      bool compare(const TreeShapeErrorGetter& other, const TString& fileName, const Interface* tmpl = 0) const;
 
      
      const ResidualCalculator* cellCalc() const { return m_cellCalc; }
      const ResidualCalculator* ringCalc() const { return m_ringCalc; }
      
      TTree* cellTree(CaloGain::CaloGain gain) const;
      TTree* ringTree(CaloGain::CaloGain gain) const;
      
      static bool merge(const TString& listFile, const TString& outputFile);
      static bool merge(std::vector<std::unique_ptr<const TreeShapeErrorGetter> >&& getters, const TString& outputFile);
      
      TFile* file() const { return m_file.get(); }
      
    private:
      
      std::unique_ptr<TFile> m_file;
      std::vector<TTree*> m_cellTrees;
      std::vector<TTree*> m_ringTrees;    
      mutable ResidualCalculator* m_cellCalc, *m_ringCalc;
  };
}
#endif