ViewSignal.hh

Go to the documentation of this file.

#ifndef G_VIEW_SIGNAL
#define G_VIEW_SIGNAL
 
#include <Rtypes.h>
 
#include <array>
#include <string>
 
#include "Garfield/ViewBase.hh"
 
class TH1;
class TH1D;
 
namespace Garfield {
 
class Sensor;

 
class ViewSignal : public ViewBase {
 public:
  ViewSignal() : ViewSignal(nullptr) {}
  ViewSignal(Sensor* sensor);
  ~ViewSignal() = default;

  void SetSensor(Sensor* s);

 
  void PlotSignal(const std::string& label, const std::string& optTotal = "t",
                  const std::string& optPrompt = "",
                  const std::string& optDelayed = "", const bool same = false);

  void SetRangeX(const double xmin, const double xmax);
  void UnsetRangeX() { m_userRangeX = false; }

  void SetRangeY(const double ymin, const double ymax);
  void UnsetRangeY() { m_userRangeY = false; }

  void SetLabelY(const std::string& label) { m_labelY = label; }

  void EnableLegend(const bool on = true) { m_legend = on; }

  void SetColourTotal(const short col) { m_colTotal = col; }
  void SetColourElectrons(const short col) { m_colElectrons = col; }
  void SetColourIons(const short col) { m_colIons = col; }
  void SetColourHoles(const short col) { m_colIons = col; }

  void SetColourDelayed(const short colTotal,
                        const short colElectrons = kYellow - 7,
                        const short colIons = kRed - 9) {
    m_colDelayed = {colTotal, colElectrons, colIons};
  }
 
 private:
  // Sensor.
  Sensor* m_sensor = nullptr;
 
  // Axis range.
  double m_xmin = 0.;
  double m_xmax = 0.;
  bool m_userRangeX = false;
  double m_ymin = 0.;
  double m_ymax = 0.;
  bool m_userRangeY = false;
 
  // Axis label.
  std::string m_labelY = "";
 
  bool m_legend = false;
 
  // Colours.
  short m_colTotal = kBlue + 3;
  short m_colElectrons = kOrange - 3;
  short m_colIons = kRed + 1;
  std::array<short, 6> m_colDelayed{
      {kCyan + 2, kYellow - 7, kRed - 9, kGreen + 1, kYellow - 4, kRed - 9}};
 
  std::array<short, 3> m_colPrompt{{kAzure + 10, kRed - 4, kMagenta + 2}};
 
  TH1* DrawHistogram(TH1D& h, const std::string& opt,
                     const std::string& ylabel);
};
}  // namespace Garfield
#endif