AttenProfiles.h

Go to the documentation of this file.

///////////////////////////////////////////////////////////////////////////
/// \brief   Histograms used by attenuation calibration
/// \author  Christopher Backhouse - bckhouse@caltech.edu
/// \date    Apr 2013
///////////////////////////////////////////////////////////////////////////
#ifndef CALDP_ATTENPROFILES_H
#define CALDP_ATTENPROFILES_H
#if defined G__DICTIONARY && ! defined ATTENPROFILES_DEFAULT_CONSTRUCTORS
#define ATTENPROFILES_DEFAULT_CONSTRUCTORS
#endif

#include "TH2.h"

#include "GeometryObjects/OfflineChan.h"
#include "CalibrationDataProducts/AttenHists.h"

#include <cassert>
#include <cmath>
#include <map>

namespace caldp
{
  /// Lightweight TProfile equivalent
  class LiteProfile
  {
  public:

    // No-one in Calibration wants default constructors, they just lead to
    // things accidentally getting the wrong binning, much better to be
    // explicit. But generating reflex dictionaries relies on them, so control
    // them with this #define
    LiteProfile()
      : fMinW(-1)
      , fMaxW(-1)
      {}

#ifndef __GCCXML__
    LiteProfile(float w0, float w1);

    void Fill(float x, float y);
    LiteProfile& operator+=(LiteProfile const& rhs);
    TH1F* ToTH1F() const;
    int TotalHits() const;
    int TotalEntries() const;
    std::vector<int>   const& Count() const;
    std::vector<float> const& Sum()   const;
    std::vector<float> const& SqSum() const;

#endif
  protected:
    float fMinW, fMaxW;

    static const size_t kNumBins = 100;

    std::vector<int>   fCount;
    std::vector<float> fSum;
    std::vector<float> fSqSum;
  };

  /// Profiles used by attenuation calibration
  class AttenProfiles
  {
  public:
    AttenProfiles() {}

#ifndef __GCCXML__
    AttenProfiles(float minW, float maxW);

    AttenProfiles& operator+=(AttenProfiles const& rhs);
    
    LiteProfile & LiteProfileByPathType(caldp::PathType_t const& pt)
    {
      switch (pt){
          case kCalXY:      return WPE_corr_xy;
          case kCalAvg:     return WPE_corr;
          case kCalXYTruth: return WPE_corr_xy_truth;
          case kCalZ:       return WPE_corr_z;
          case kCalTraj:    return WPE_corr_traj;
        default:          return WPE;
      }
      return WPE_corr;
    }
#endif

    LiteProfile WPE, WPE_corr, WPE_corr_xy, WPE_corr_z, WPE_corr_traj, WPE_corr_xy_truth;
  };

  /// AttenProfiles for many channels
  class AttenProfilesMap
  {
  public:

    AttenProfilesMap() : fMinW(-1), fMaxW(-1) {}

#ifndef __GCCXML__
    AttenProfilesMap(float w0, float w1);

    AttenProfilesMap(const AttenProfiles&) = delete; // uncopyable

    bool HasProfiles(const geo::OfflineChan& c) const;
    bool HasProfiles(int plane, int cell) const;

    AttenProfiles & GetProfiles(const geo::OfflineChan& c);
    AttenProfiles & GetProfiles(int plane, int cell);

    /// List of channels with profiles on
    std::vector<geo::OfflineChan> Channels() const;

    /// Intended for writing out/drawing all the histograms
    std::map<int, std::map<int, std::vector<TH1F*>>> GetAllProfilesByPlaneAndCell();

    AttenProfilesMap& operator+=(const AttenProfilesMap& rhs);

    void Clear();

    double MinW() const {return fMinW;}
    double MaxW() const {return fMaxW;}
#endif
  protected:
    float fMinW, fMaxW;

    std::map<geo::OfflineChan, AttenProfiles> fMap;
  };

}

#endif