Combine multiple component experiments. More...

#include "/cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N20-07-07/CAFAna/Experiment/MultiExperiment.h"

Inheritance diagram for ana::MultiExperiment:

Public Member Functions
	MultiExperiment (std::vector< const IExperiment * > expts={})

void	Add (const IExperiment *expt)

virtual double	ChiSq (osc::IOscCalculatorAdjustable *osc, const SystShifts &syst=SystShifts::Nominal()) const override

virtual void	Derivative (osc::IOscCalculator calc, const SystShifts &shift, std::unordered_map< const ISyst , double > &dchi) const override

void	SetSystCorrelations (int idx, const std::vector< std::pair< const ISyst , const ISyst >> &corrs)

virtual void	SaveTo (TDirectory *dir) const override

virtual stan::math::var	LogLikelihood (osc::IOscCalculatorAdjustableStan *osc, const SystShifts &syst=kNoShift) const

Static Public Member Functions
static std::unique_ptr< MultiExperiment >	LoadFrom (TDirectory *dir)

Protected Attributes
std::vector< const IExperiment * >	fExpts

std::vector< std::vector< std::pair< const ISyst , const ISyst > > >	fSystCorrelations

Detailed Description

Combine multiple component experiments.

Definition at line 11 of file MultiExperiment.h.

Constructor & Destructor Documentation

ana::MultiExperiment::MultiExperiment ( std::vector< const IExperiment * > expts = {} )

inline

Definition at line 14 of file MultiExperiment.h.

References fExpts, and fSystCorrelations.

Referenced by LoadFrom().

                                                           {}) : fExpts(expts)
     {
       fSystCorrelations.resize(expts.size());
     }

Member Function Documentation

void ana::MultiExperiment::Add ( const IExperiment * expt )

inline

Definition at line 19 of file MultiExperiment.h.

References calc, ChiSq(), Derivative(), dir, fExpts, fSystCorrelations, compare_h5_caf::idx, LoadFrom(), ana::SystShifts::Nominal(), SaveTo(), SetSystCorrelations(), plotSysts::shift, and plotSysts::syst.

Referenced by compare_pid_cut_sens(), fa_numuContours(), and ana::NumuAnalysis::MakeExperiment().

     {
       fExpts.push_back( expt );
       fSystCorrelations.resize( fExpts.size() );
     }

double ana::MultiExperiment::ChiSq	(	osc::IOscCalculatorAdjustable *	osc,
		const SystShifts &	syst = `SystShifts::Nominal()`
	)		const

overridevirtual

Reimplemented from ana::IExperiment.

Definition at line 14 of file MultiExperiment.cxx.

References Derivative(), fExpts, fSystCorrelations, ana::SystShifts::GetShift(), it, getGoodRuns4SAM::n, runNovaSAM::ret, ana::SystShifts::SetShift(), and plotSysts::syst.

Referenced by Add(), and MakeSurface().

   {
     double ret = 0;
     for(unsigned int n = 0; n < fExpts.size(); ++n){
       // Don't waste time fiddling with the systematics if for sure there's
       // nothing to do.
       if(fSystCorrelations[n].empty()){
         ret += fExpts[n]->ChiSq(osc, syst);
       }
       else{
         // Make a local copy we're going to rewrite into the terms this
         // sub-experiment will accept.
         SystShifts localShifts = syst;
         for(auto it: fSystCorrelations[n]){
           // We're mapping prim -> sec
           const ISyst* prim = it.first;
           const ISyst* sec = it.second;
           if(syst.GetShift(prim) != 0){
             // sec can be unset, which means there's no representation needed
             // of prim in the sub-experiment.
             if(sec) localShifts.SetShift(sec, syst.GetShift(prim));
             // We've either translated or discarded prim, so drop it here.
             localShifts.SetShift(prim, 0);
           }
         }
         ret += fExpts[n]->ChiSq(osc, localShifts);
       }
     }
     return ret;
   }

void ana::MultiExperiment::Derivative	(	osc::IOscCalculator *	calc,
		const SystShifts &	shift,
		std::unordered_map< const ISyst *, double > &	dchi
	)		const

overridevirtual

Reimplemented from ana::IExperiment.

Definition at line 49 of file MultiExperiment.cxx.

References fExpts, fSystCorrelations, ana::SystShifts::GetShift(), it, getGoodRuns4SAM::n, ana::SystShifts::SetShift(), SetSystCorrelations(), and plotSysts::shift.

Referenced by Add(), and ChiSq().

   {
     // Each one should sum into the total so far
     for(unsigned int n = 0; n < fExpts.size(); ++n){
       if(fSystCorrelations[n].empty()){
         // If there are no adjustments needed it's easy
         fExpts[n]->Derivative(calc, shift, dchi);
         if(dchi.empty()) return;
         continue;
       }
 
       auto dchiLocal = dchi;
       for(auto& it: dchiLocal) it.second = 0;
       SystShifts localShifts = shift;
       std::unordered_map<const ISyst*, const ISyst*> reverseMap;
       for(auto it: fSystCorrelations[n]){
         // We're mapping prim -> sec
         const ISyst* prim = it.first;
         const ISyst* sec = it.second;
         if(dchi.count(prim)){
           dchiLocal.erase(dchiLocal.find(prim));
           if(sec){
             dchiLocal.emplace(sec, 0);
             reverseMap[sec] = prim;
           }
         }
         if(sec) localShifts.SetShift(sec, shift.GetShift(prim));
         // We've either translated or discarded prim, so drop it here.
         localShifts.SetShift(prim, 0);
       }
 
       fExpts[n]->Derivative(calc, localShifts, dchiLocal);
 
       // One of the components doesn't support derivatives, don't bother asking
       // the rest.
       if(dchiLocal.empty()){
         dchi.clear();
         return;
       }
 
       // And translate back
       for(auto it: dchiLocal){
         if(reverseMap.count(it.first) > 0){
           dchi[reverseMap[it.first]] += it.second;
         }
         else{
           dchi[it.first] += it.second;
         }
       }
     }
   }

std::unique_ptr< MultiExperiment > ana::MultiExperiment::LoadFrom ( TDirectory * dir )

static

Definition at line 155 of file MultiExperiment.cxx.

References ana::assert(), genie::utils::style::Format(), MECModelEnuComparisons::i, ana::LoadFrom< IExperiment >(), MultiExperiment(), runNovaSAM::release, and cvnie::subdir.

Referenced by Add(), and ana::LoadFrom< IExperiment >().

   {
     TObjString* ptag = (TObjString*)dir->Get("type");
     assert(ptag);
     assert(ptag->GetString() == "MultiExperiment");
 
     std::vector<const IExperiment*> expts;
 
     for(int i = 0; ; ++i){
       TDirectory* subdir = dir->GetDirectory(TString::Format("expt%d", i));
       if(!subdir) break;
 
       expts.push_back(ana::LoadFrom<IExperiment>(subdir).release());
     }
 
     assert(!expts.empty());
 
     return std::unique_ptr<MultiExperiment>(new MultiExperiment(expts));
   }

virtual stan::math::var ana::IExperiment::LogLikelihood	(	osc::IOscCalculatorAdjustableStan *	osc,
		const SystShifts &	syst = `kNoShift`
	)		const

inlinevirtualinherited

Reimplemented in test::GaussQuadExperiment.

Definition at line 38 of file IExperiment.h.

References ana::assert(), dir, and ana::IExperiment::SaveTo().

Referenced by ana::StanFitter::log_prob().

       {
         assert(false && "unimplemented");
         return 0;
       };

void ana::MultiExperiment::SaveTo ( TDirectory * dir ) const

overridevirtual

Reimplemented from ana::IExperiment.

Definition at line 133 of file MultiExperiment.cxx.

References om::cerr, dir, allTimeWatchdog::endl, fExpts, genie::utils::style::Format(), fSystCorrelations, MECModelEnuComparisons::i, it, and tmp.

Referenced by Add().

   {
     bool hasCorr = false;
     for(auto it: fSystCorrelations) if(!it.empty()) hasCorr = true;
 
     if(hasCorr){
       std::cerr << "Warning in MultiExperiment: systematic correlations are set and will not be serialized by this call to SaveTo(). You will have to re-set them once you load the experiment back in." << std::endl;
     }
 
     TDirectory* tmp = dir;
 
     dir->cd();
     TObjString("MultiExperiment").Write("type");
 
     for(unsigned int i = 0; i < fExpts.size(); ++i){
       fExpts[i]->SaveTo(dir->mkdir(TString::Format("expt%d", i)));
     }
 
     tmp->cd();
   }

void ana::MultiExperiment::SetSystCorrelations	(	int	idx,
		const std::vector< std::pair< const ISyst , const ISyst >> &	corrs
	)

For the subexperiment idx, set up a mapping between systematics

Each element in the vector is a pair from a "primary" systematic to a "secondary". When this MultiExperiment is called with a primary systematic shifted, the sub-experiment will be called with the secondary systematic set to the same value (and the primary unset).

You can pass NULL for a secondary to indicate that the systematic simply has no effect on the experiment in question and should be filtered out.

Definition at line 105 of file MultiExperiment.cxx.

References ana::assert(), om::cout, allTimeWatchdog::endl, fSystCorrelations, compare_h5_caf::idx, and it.

Referenced by Add(), Derivative(), futureSig_reach_singlePOTcombo_syst(), joint_fit_2017_make_fc_slice(), joint_fit_2017_make_fc_surf(), make_fc_mass_and_oct_nersc_2018(), make_fc_mass_and_oct_nersc_2019(), make_fc_mh_nersc_2018(), make_fc_mh_nersc_2019(), make_fc_oct_nersc_2018(), make_fc_oct_nersc_2019(), make_fc_slices_nersc_2018(), make_fc_slices_nersc_2018_stats(), make_fc_slices_nersc_2019(), make_fc_surfaces_2020(), make_fc_surfaces_nersc_2018(), make_fc_surfaces_nersc_2018_stats(), and make_fc_surfaces_nersc_2019().

   {
     // Sanity-check the mapping
     std::map<const ISyst*, const ISyst*> already;
     for(auto it: corrs){
       assert(it.first != it.second);
 
       // Don't worry if second element is null pointer
       if (!it.second) continue;
       if(already.find(it.second) == already.end()){
         already[it.second] = it.first;
       }
       else{
         std::cout << "MultiExperiment::SetSystCorrelations(): Warning!\n"
                   << "In experiment " << idx << " both "
                   << already[it.second]->ShortName() << " and "
                   << it.first->ShortName()
                   << " are configured to map to " << it.second->ShortName()
                   << ". That's probably not what you want." << std::endl;
       }
     }
 
     // Apply it
     fSystCorrelations[idx] = corrs;
   }

Member Data Documentation

std::vector<const IExperiment*> ana::MultiExperiment::fExpts

protected

Definition at line 50 of file MultiExperiment.h.

Referenced by Add(), ChiSq(), Derivative(), MultiExperiment(), and SaveTo().

std::vector<std::vector<std::pair<const ISyst*, const ISyst*> > > ana::MultiExperiment::fSystCorrelations

protected

Definition at line 52 of file MultiExperiment.h.

Referenced by Add(), ChiSq(), Derivative(), MultiExperiment(), SaveTo(), and SetSystCorrelations().

The documentation for this class was generated from the following files:

/cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N20-07-07/CAFAna/Experiment/MultiExperiment.h
/cvmfs/nova-development.opensciencegrid.org/novasoft/releases/N20-07-07/CAFAna/Experiment/MultiExperiment.cxx

Public Member Functions

Static Public Member Functions

Protected Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation