RandomUniverseUtility.h

Go to the documentation of this file.

//
// Created by Brian Rebel on 4/20/20.
//
// Functions to help with creating and checking random universes for
// sensitivities
//

#ifndef CMF_UTILITIES_CMFRANDOMUNIVERSEUTILITIES_H
#define CMF_UTILITIES_CMFRANDOMUNIVERSEUTILITIES_H

#include "TRandom3.h"
#include "TH2.h"

#include "fhiclcpp/ParameterSet.h"

#include "CovarianceMatrixFit/dataProducts/FakeUniverse.h"

namespace cmf{
  class RandomUniverseUtility {
  public:

    static RandomUniverseUtility *Instance();

    void CreateLocationWithVariedSystematics(cmf::Location const& nominalLoc,
                                             cmf::Location      & vecLoc,
                                             size_t               universeNum);
    void CreateLocationsWithVariedSystematics(cmf::Location              const& nominalLoc,
                                              std::vector<cmf::Location>      & vecLocs);
    void CreateLocationWithVariedOscParameters(cmf::Location const& nominalPoint,
                                               cmf::Location      & vecPoint,
                                               size_t               universeNum);
    void CreateLocationsWithVariedOscParameters(cmf::Location              const& nominalLoc,
                                                std::vector<cmf::Location>      & vecLoc);

    void Initialize(fhicl::ParameterSet const& pset);

    void FillPoissonSpectrum(std::vector<float>& spectrum,
                             std::vector<float>& poissonSpectrum);

    void MakeUniverseHistograms(cmf::FakeUniverse & fakeU,
                                unsigned int        uNumber);

  private:

    RandomUniverseUtility();

    ~RandomUniverseUtility();

    void PoissonSpectrumBinByBin(std::vector<float>& spectrum,
                                 std::vector<float>& poissonSpectrum);
    bool UniformDrawParameters(std::string const& name);

    TRandom3     fRand;             ///< RNG for picking the variation in parameters
    TH2D*        fSystDrawHist;     ///< keep track of the systematic parameter values
    TH2D*        fOscParamDrawHist; ///< keep track of the oscillation parameter values
    bool         fPoissonFluctuate; ///< do we want to poisson fluctuate the FD prediction?
    bool         fPoissonByHist;    ///< use histogram method for picking poisson numbers
    bool         fPoissonByBin;     ///< use bin by bin method for picking poisson numbers
    unsigned int fNumUniverses;     ///< number of fake universes to create
  };

  //----------------------------------------------------------------------------
  // The calibration and light level systematic uncertainties are
  // really only defined for a 1 sigma shift because their excursions
  // are calculated from special data sets with the shift applied
  // Also pull them from a uniform distribution as they are definitely
  // not Gaussian.
  inline bool RandomUniverseUtility::UniformDrawParameters(const std::string &name)
  {
    return (name.find("lightmodel")  != std::string::npos ||
            name.find("ckv-proton")  != std::string::npos ||
            name.find("Calibration") != std::string::npos ||
            name.find("fluxpca")     != std::string::npos);
    }
}


#endif //CMF_UTILITIES_CMFRANDOMUNIVERSEUTILITIES_H