FitVars.cxx

Go to the documentation of this file.

#include "CAFAna/Vars/FitVars.h"

#include "OscLib/IOscCalc.h"
#include "Utilities/func/MathUtil.h"

#include <cassert>
#include <cmath>

namespace ana
{
  // declared as 'extern' in FitVars.h
  const FitTheta13 kFitTheta13;
  const FitSinSq2Theta13 kFitSinSq2Theta13;
  const FitDeltaInPiUnits kFitDeltaInPiUnits;
  const FitSinSqTheta23 kFitSinSqTheta23;
  const FitSinSqTheta23UpperOctant kFitSinSqTheta23UpperOctant;
  const FitSinSqTheta23LowerOctant kFitSinSqTheta23LowerOctant;
  const FitDmSq32 kFitDmSq32;
  const FitDmSq32Scaled kFitDmSq32Scaled;
  const FitDmSq32NormalHierarchy kFitDmSq32NormalHierarchy;
  const FitDmSq32ScaledNH kFitDmSq32ScaledNH;
  const FitDmSq32ScaledIH kFitDmSq32ScaledIH;
  const FitDmSq32InvertedHierarchy kFitDmSq32InvertedHierarchy;
  const FitTanSqTheta12 kFitTanSqTheta12;
  const FitSinSq2Theta12 kFitSinSq2Theta12;
  const FitDmSq21 kFitDmSq21;


  //----------------------------------------------------------------------
  double FitTheta13::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return osc->GetTh13();
  }

  //----------------------------------------------------------------------
  void FitTheta13::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    return osc->SetTh13(val);
  }

//----------------------------------------------------------------------
  double FitSinSq2Theta13::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return util::sqr(sin(2*osc->GetTh13()));
  }

  //----------------------------------------------------------------------
  void FitSinSq2Theta13::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetTh13(asin(sqrt(Clamp(val)))/2);
  }

  //----------------------------------------------------------------------
  // helper function so the code doesn't need to be written twice
  namespace
  {
    template<typename T>
    T dcP_GetVal_Helper(const osc::_IOscCalcAdjustable<T>* osc)
    {
      T ret = osc->GetdCP()/M_PI;

    // convert to the range 0-2
      auto a = floor(ret/2+1);
    ret -= 2*a;
    // Instead of figuring out all the rounding just nudge the last little bit
    while(ret < 0) ret += 2;
    while(ret > 2) ret -= 2;

    return ret;
    }
  }
  double FitDeltaInPiUnits::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return dcP_GetVal_Helper(osc);
  }
  stan::math::var FitDeltaInPiUnits::GetValue(const osc::IOscCalcAdjustableStan* osc) const
  {
    return dcP_GetVal_Helper(osc);
  }

  //----------------------------------------------------------------------
  void FitDeltaInPiUnits::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetdCP(M_PI*val);
  }
  void FitDeltaInPiUnits::SetValue(osc::IOscCalcAdjustableStan* osc, stan::math::var val) const
  {
    osc->SetdCP(M_PI*val);
  }

  //----------------------------------------------------------------------
  double FitSinSqTheta23::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return util::sqr(sin(osc->GetTh23()));
  }

  //----------------------------------------------------------------------
  stan::math::var FitSinSqTheta23::GetValue(const osc::IOscCalcAdjustableStan *osc) const
  {
    return util::sqr(sin(osc->GetTh23()));
  }

  //----------------------------------------------------------------------
  void FitSinSqTheta23::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetTh23(asin(sqrt(this->Clamp(val))));
  }

  //----------------------------------------------------------------------
  void FitSinSqTheta23::SetValue(osc::IOscCalcAdjustableStan *osc, stan::math::var val) const
  {
    osc->SetTh23(asin(sqrt(this->Clamp(val))));
  }

  //----------------------------------------------------------------------
  stan::math::var FitSinSqTheta23UpperOctant::GetValue(const osc::IOscCalcAdjustableStan *osc) const
  {
    return util::sqr(sin(osc->GetTh23()));
  }
  double FitSinSqTheta23UpperOctant::GetValue(const osc::IOscCalcAdjustable *osc) const
  {
    return util::sqr(sin(osc->GetTh23()));
  }
  //----------------------------------------------------------------------
  void FitSinSqTheta23UpperOctant::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetTh23(asin(sqrt(this->Clamp(val))));
  }

  //----------------------------------------------------------------------
  void FitSinSqTheta23UpperOctant::SetValue(osc::IOscCalcAdjustableStan *osc, stan::math::var val) const
  {
    osc->SetTh23(asin(sqrt(this->Clamp(val))));
  }

  //----------------------------------------------------------------------
  stan::math::var FitSinSqTheta23LowerOctant::GetValue(const osc::IOscCalcAdjustableStan *osc) const
  {
    return util::sqr(sin(osc->GetTh23()));
  }
  double FitSinSqTheta23LowerOctant::GetValue(const osc::IOscCalcAdjustable *osc) const
  {
    return util::sqr(sin(osc->GetTh23()));
  }

  //----------------------------------------------------------------------
  void FitSinSqTheta23LowerOctant::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetTh23(asin(sqrt(this->Clamp(val))));
  }

  //----------------------------------------------------------------------
  void FitSinSqTheta23LowerOctant::SetValue(osc::IOscCalcAdjustableStan *osc, stan::math::var val) const
  {
    osc->SetTh23(asin(sqrt(this->Clamp(val))));
  }

  //----------------------------------------------------------------------
  double FitSinSq2Theta23::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return util::sqr(sin(2*osc->GetTh23()));
  }

  //----------------------------------------------------------------------
  void FitSinSq2Theta23::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetTh23(asin(sqrt(Clamp(val)))/2);
  }

  //----------------------------------------------------------------------
  double FitDmSq32::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return osc->GetDmsq32();
  }

  //----------------------------------------------------------------------
  void FitDmSq32::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetDmsq32(this->Clamp(val));
  }

  //----------------------------------------------------------------------
  double FitDmSq32Scaled::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return osc->GetDmsq32()*1000.0;
  }

  //----------------------------------------------------------------------
  stan::math::var FitDmSq32Scaled::GetValue(const osc::IOscCalcAdjustableStan *osc) const
  {
    return osc->GetDmsq32()*1000.0;;
  }

  //----------------------------------------------------------------------
  void FitDmSq32Scaled::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetDmsq32(this->Clamp(val)/1000.0);
  }

  //----------------------------------------------------------------------
  void FitDmSq32Scaled::SetValue(osc::IOscCalcAdjustableStan *osc, stan::math::var val) const
  {
    osc->SetDmsq32(this->Clamp(val)/1000.0);
  }


//----------------------------------------------------------------------
  double FitDmSq32NormalHierarchy::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return osc->GetDmsq32();
  }
  
  //---------------------------------------------------------------------
  void FitDmSq32NormalHierarchy::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetDmsq32(Clamp(val));
  }
  
  //---------------------------------------------------------------------
  void FitDmSq32ScaledNH::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetDmsq32(Clamp(val)/1000.0);
  }
  
  //--------------------------------------------------------------------
  double FitDmSq32ScaledNH::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return osc->GetDmsq32()*1000.0;
  }

  //---------------------------------------------------------------------
  void FitDmSq32ScaledIH::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetDmsq32(Clamp(val)/1000.0);
  }
  
  //--------------------------------------------------------------------
  double FitDmSq32ScaledIH::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return osc->GetDmsq32()*1000.0;
  }

  //---------------------------------------------------------------------
double FitDmSq32InvertedHierarchy::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return osc->GetDmsq32();
  }
  
  //---------------------------------------------------------------------
  void FitDmSq32InvertedHierarchy::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetDmsq32(Clamp(val));
  }


  //----------------------------------------------------------------------
  double FitTanSqTheta12::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return util::sqr(tan(osc->GetTh12()));
  }

  //----------------------------------------------------------------------
  void FitTanSqTheta12::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetTh12(atan(sqrt(Clamp(val))));
  }

  //----------------------------------------------------------------------
  double FitSinSq2Theta12::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return util::sqr(sin(2*osc->GetTh12()));
  }

  //----------------------------------------------------------------------
  void FitSinSq2Theta12::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetTh12(asin(sqrt(Clamp(val)))/2);
  }

  //----------------------------------------------------------------------
  double FitDmSq21::GetValue(const osc::IOscCalcAdjustable* osc) const
  {
    return osc->GetDmsq21();
  }

  //----------------------------------------------------------------------
  void FitDmSq21::SetValue(osc::IOscCalcAdjustable* osc, double val) const
  {
    osc->SetDmsq21(Clamp(val));
  }
} // namespace