DD4hep_TrackingMaterialPlotter.h

Go to the documentation of this file.

#ifndef DD4hep_TrackingMaterialPlotter_h
#define DD4hep_TrackingMaterialPlotter_h
 
#include <algorithm>
#include <vector>
#include <sstream>
#include <iostream>
#include <iomanip>
 
#include <TH2F.h>
#include <TColor.h>
 
#include "DD4hep_XHistogram.h"
class MaterialAccountingStep;
 
class DD4hep_TrackingMaterialPlotter {
public:
  typedef std::pair<double, double> Range;
 
  DD4hep_TrackingMaterialPlotter(float maxZ, float maxR, float resolution);
  void plotSegmentUnassigned(const MaterialAccountingStep& step);
  void plotSegmentInLayer(const MaterialAccountingStep& step, int layer);
 
  void normalize(void) { m_tracker.normalize(); }
 
  void draw(void);
 
private:
  DD4hep_XHistogram m_tracker;
 
  std::vector<int> m_color;
  std::vector<int> m_gradient;
 
  void fill_color();
  unsigned int fill_gradient(const TColor& first, const TColor& last, unsigned int steps = 100, unsigned int index = 0);
  unsigned int fill_gradient(const unsigned int& first,
                             const unsigned int& last,
                             const unsigned int& steps = 100,
                             const unsigned int& index = 0);
};
 
#endif  //  DD4hep_TrackingMaterialPlotter_h