BENDecomp.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Core/Cut.h"
#include "CAFAna/Core/HistAxis.h"
#include "CAFAna/Decomp/IDecomp.h"
#include "CAFAna/Core/Spectrum.h"
#include "3FlavorAna/Cuts/BeamNueCuts.h"
#include "CAFAna/Vars/HistAxes.h"
#include "3FlavorAna/Vars/HistAxes.h"

#include <memory>
#include <string>
#include "TH2.h"
#include "TString.h"

class TDirectory;

#include "StandardRecord/Proxy/FwdDeclare.h" // for Reduce function

namespace ana
{

  class Binning;
  class Loaders;
  class SpectrumLoaderBase;

  class BENDecomp: public IDecomp
  {
    
  public:
    
    enum EBENFixedKa{kFixedKaonScale};
    enum EBENCalcKa{kCalcKaonScale};
    
    // Keep these for backward compatibility
    BENDecomp(SpectrumLoaderBase& loaderMC,
              SpectrumLoaderBase& loaderData,
              const HistAxis& axis,
              const Cut& cut,
              const SystShifts& shiftMC = kNoShift,
              const SystShifts& shiftData = kNoShift,
              const Var& wei = kUnweighted,
              const HistAxis& axisNumuPi = kNumuNonQEBENAxis,
              const Cut& cutNumuPi = kBENPiNumuCut,
              const double kaonNormalization=1.17
              );
    BENDecomp(Loaders& loaders,
              const HistAxis& axis,
              const Cut& cut,
              const SystShifts& shiftMC = kNoShift,
              const SystShifts& shiftData = kNoShift,
              const Var& wei = kUnweighted,
              const HistAxis& axisNumuPi = kNumuNonQEBENAxis,
              const Cut& cutNumuPi = kBENPiNumuCut,
              const double kaonNormalization=1.17
              );

    BENDecomp(EBENCalcKa,
              SpectrumLoaderBase& loaderMC,
              SpectrumLoaderBase& loaderData,
              const HistAxis& axis,
              const Cut& cut,
              const SystShifts& shiftMC = kNoShift,
              const SystShifts& shiftData = kNoShift,
              const Var& wei = kUnweighted,
              const HistAxis& axisNumuPi = kNumuNonQEBENAxis,
              const Cut& cutNumuPi = kBENPiNumuCut,
              const HistAxis& axisNumuKa =  kNumuCCAxisExtended,
              const Cut& cutNumuKa = kBENKaNumuCut
              );
    
    BENDecomp(EBENCalcKa,
              Loaders& loaders,
              const HistAxis& axis,
              const Cut& cut,
              const SystShifts& shiftMC = kNoShift,
              const SystShifts& shiftData = kNoShift,
              const Var& wei = kUnweighted,
              const HistAxis& axisNumuPi = kNumuNonQEBENAxis,
              const Cut& cutNumuPi = kBENPiNumuCut,
              const HistAxis& axisNumuKa =  kNumuCCAxisExtended,
              const Cut& cutNumuKa = kBENKaNumuCut
              );


    virtual ~BENDecomp();

    Spectrum NumuComponent()     const override;
    Spectrum AntiNumuComponent() const override;

    Spectrum NCTotalComponent()  const override;
    Spectrum NCComponent()       const override;
    Spectrum NCAntiComponent()   const override;
    
    Spectrum NueComponent()      const override;
    Spectrum AntiNueComponent()  const override;

    Spectrum Data_Component()  const override;

    // Convenient for making decomposition blessed plots
    Spectrum MC_NCTotalComponent()       const override;
    Spectrum MC_NumuComponent()     const override;
    Spectrum MC_AntiNumuComponent() const override;
    Spectrum MC_NueComponent()      const override;
    Spectrum MC_AntiNueComponent()  const override;
    
    void SaveTo(TDirectory* dir, const std::string& name) const override;
    void SavePlotsPi(TDirectory * dir);
    void SavePlotsKa(TDirectory * dir);
    void SavePlots  (TDirectory * dir);
    static std::unique_ptr<BENDecomp> LoadFrom(TDirectory* dir, const std::string& name);
    TH2 * GetPionWeights() const {Decompose(); return fNumuPiWeights;} //for debugging
    double GetKaonScale()  const {Decompose(); return fKaonNormalization;}
    Spectrum NuePiEstimate() {Decompose();return fNueCCFromPiEstim;}
    Spectrum NueKaEstimate() {Decompose();return fNueCCFromKaEstim;}

    // Reduction function for BEN 2017
    static void ReduceForBEN2020Decaf(caf::StandardRecord* sr,
                                      const caf::SRProxy* srProxy);


  protected:
  BENDecomp()
    : fData(Spectrum::Uninitialized()),
      fNCTot(Spectrum::Uninitialized()),
      fNue(Spectrum::Uninitialized()),
      fAntiNue(Spectrum::Uninitialized()),
      fNumu(Spectrum::Uninitialized()),
      fAntiNumu(Spectrum::Uninitialized()),
      fTotal(Spectrum::Uninitialized()),
      fNotNue(Spectrum::Uninitialized()),

      fNueCCFromPi(Spectrum::Uninitialized()),
      fNueCCFromKa(Spectrum::Uninitialized()),
      fNueCCFromOther(Spectrum::Uninitialized()),
      fNueCCFromPiPtPz(Spectrum::Uninitialized()),

      fNumuSelData(Spectrum::Uninitialized()),
      fNumuSelBkg(Spectrum::Uninitialized()),
      fNumuSelCCFromPi(Spectrum::Uninitialized()),
      fNumuSelCCFromKa(Spectrum::Uninitialized()),
      fNumuSelCCFromOther(Spectrum::Uninitialized()),
      fNumuSelCCFromPiPtPz(Spectrum::Uninitialized()),

      fNumuUncontainData(Spectrum::Uninitialized()),
      fNumuUncontainBkg(Spectrum::Uninitialized()),
      fNumuUncontainCCFromPi(Spectrum::Uninitialized()),
      fNumuUncontainCCFromKa(Spectrum::Uninitialized()),
      fNumuUncontainCCFromOther(Spectrum::Uninitialized()),

      fNueCCFromPiEstim(Spectrum::Uninitialized()),
      fNueCCFromKaEstim(Spectrum::Uninitialized()),
      fIsDecomposed(0),
      fIsFixedKaScale(0),
      fKaonNormalization(0)
    {};

    void Decompose () const;
    void MakeWeightsNumuFromPion() const;
    void MakeWeightsNumuFromKaon() const;

    virtual Spectrum NueEstimate() const;
    void NueEstimateFromPi() const;
    void NueEstimateFromKa() const;
    void NueEstimateFromOther() const;    

    Spectrum fData;
    Spectrum fNCTot;
    Spectrum fNue;
    Spectrum fAntiNue;
    Spectrum fNumu;
    Spectrum fAntiNumu;
    Spectrum fTotal;
    Spectrum fNotNue;

    Spectrum fNueCCFromPi;
    Spectrum fNueCCFromKa;
    Spectrum fNueCCFromOther;
    Spectrum fNueCCFromPiPtPz;
    
    Spectrum fNumuSelData;
    Spectrum fNumuSelBkg;
    Spectrum fNumuSelCCFromPi;
    Spectrum fNumuSelCCFromKa;
    Spectrum fNumuSelCCFromOther;
    Spectrum fNumuSelCCFromPiPtPz;

    Spectrum fNumuUncontainData;
    Spectrum fNumuUncontainBkg;
    Spectrum fNumuUncontainCCFromPi;
    Spectrum fNumuUncontainCCFromKa;
    Spectrum fNumuUncontainCCFromOther;
    
    mutable Spectrum fNueCCFromPiEstim;
    mutable Spectrum fNueCCFromKaEstim;

    mutable bool fIsDecomposed; 
    mutable TH2 * fNumuPiWeights;
    bool fIsFixedKaScale;
    mutable double fKaonNormalization;
  };

}