EclPainter2D.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Milkail Remnev, Dmitry Matvienko                         *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 ***************************************************************************/
 
#pragma once
 
//ECL
#include <ecl/modules/eclDisplay/EclPainter.h>
 
class TH2C;
class TH2F;
class TPad;
 
namespace Belle2 {
  class EclPainter2D : public EclPainter {
  public:
    enum Type {
      CHANNEL_2D, 
      SHAPER_2D, 
      NONE
    };
 
    EclPainter2D(EclData* data, Type type);
 
    EclPainter2D(const EclPainter2D& other) : EclPainter(other) { cloneFrom(other); }
    EclPainter2D& operator=(const EclPainter2D& other) { cloneFrom(other); return *this; }
 
    ~EclPainter2D();
 
  private:
    Type m_type;
    TH2F* m_hist;
 
    TH2C* hgrid;
    TPad* grid;
 
    void cloneFrom(const EclPainter2D& other);
    void setTitles();
    int getMaxX();
    int getMaxY();
    int channelToSegIdX(int channel);
    int channelToSegIdY(int channel);
    void initGrid();
    void drawGrid();
 
  public:
    virtual void getInformation(int px, int py, MultilineWidget* panel) override;
 
    Type getType();
 
    virtual void Draw() override;
  };
}