CheatDecomp.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Decomp/IDecomp.h"

class TDirectory;

namespace ana
{
  /// Just return the ND truth spectra as the decomposition
  class CheatDecomp: public IDecomp
  {
  public:
    CheatDecomp(SpectrumLoaderBase& loader,
                std::string label,
                const Binning& bins,
                const Var& var,
                const Cut& cut,
                const SystShifts& shift = kNoShift,
                const Var& wei = kUnweighted);

    CheatDecomp(SpectrumLoaderBase& loader,
                const HistAxis& axis,
                const Cut& cut,
                const SystShifts& shift = kNoShift,
                const Var& wei = kUnweighted);
    //nc
    Spectrum NCTotalComponent()  const override {return fNCTot;}
    Spectrum NCComponent()       const override {return fNC;}
    Spectrum NCAntiComponent()   const override {return fNCAnti;}
    //end nc 
    Spectrum NumuComponent()     const override {return fNumu;}
    Spectrum NueComponent()      const override {return fNue;}
    Spectrum AntiNumuComponent() const override {return fAntiNumu;}
    Spectrum AntiNueComponent()  const override {return fAntiNue;}
    Spectrum TotalMC() {return fNCTot+fNumu+fNue+fAntiNumu+fAntiNue;}

    void SaveTo(TDirectory* dir, const std::string& name) const override;
    static std::unique_ptr<CheatDecomp> LoadFrom(TDirectory* dir, const std::string& name);

  protected:
    CheatDecomp()
      : 
        //nc
        fNCTot   (Spectrum::Uninitialized()),
        fNC      (Spectrum::Uninitialized()),
        fNCAnti  (Spectrum::Uninitialized()),
        //end nc
        fNumu (Spectrum::Uninitialized()),
        fNue  (Spectrum::Uninitialized()),
        fAntiNumu(Spectrum::Uninitialized()),
        fAntiNue (Spectrum::Uninitialized())
    {};
    
    //nc
    Spectrum fNCTot;
    Spectrum fNC;
    Spectrum fNCAnti;
    //end nc
    Spectrum fNumu;
    Spectrum fNue;
    Spectrum fAntiNumu;
    Spectrum fAntiNue;
  };
}