ARICHChannelHist.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Luka Santelj                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <TH2Poly.h>
#include <TLine.h>
#include <TText.h>
 
namespace Belle2 {
  class ARICHChannelHist: public TH2Poly {
 
  public:
 
    ARICHChannelHist() {};
 
    ARICHChannelHist(const char* name, const char* title, int type = 0,
                     const std::vector<unsigned>& moduleIDs = std::vector<unsigned>());
 
    void fillBin(unsigned hapdID, unsigned chID, double weight = 1.);
 
    void setBinContent(unsigned hapdID, unsigned chID, double value);
 
    void fillBin(unsigned hapdID, double weight = 1.);
 
    void setBinContent(unsigned hapdID, double value);
 
    void Draw(Option_t* option = "") override;
 
    void fillFromTH1(TH1* hist);
 
    void setPoly(TH2Poly* poly);
 
  protected:
 
    TLine lines[6];
    TText labels[6];
    int m_type = 0; 
    std::vector<unsigned> m_hapd2binMap; 
    ClassDefOverride(ARICHChannelHist, 2); 
  };
 
} // end namespace Belle2