NuEZRunsFitter.cxx

Go to the documentation of this file.

#include <string>
#include <vector>

#include "NtupleUtils/NuMMRunFC.h"
#include "NtupleUtils/NuMMParameters.h"
#include "NtupleUtils/NuSystFitter.h"
#include "NtupleUtils/NuEZRunsFitter.h"
#include "MessageService/MsgService.h"

CVSID("$Id: NuEZRunsFitter.cxx,v 1.3 2009/04/08 16:35:01 nickd Exp $");

using namespace std;

NuEZRunsFitter::NuEZRunsFitter() : fFitter(0)
{
  
}

NuEZRunsFitter::~NuEZRunsFitter()
{
  // Delete our owned fitters
  while (!fOwnedFitters.empty())
  {
    // delete and remove the last element
    if (fOwnedFitters.back()) delete fOwnedFitters.back(); 
    fOwnedFitters.pop_back();
  }
}

void NuEZRunsFitter::AddRun(std::string Helper,
    std::string NeardetData, std::string FardetData)
{
  NuEZFitter *f = new NuEZFitter(Helper, NeardetData, FardetData);
  fFitters.push_back(f);
  fOwnedFitters.push_back(f);
  
  BuildFitter();
}

void NuEZRunsFitter::AddRun(NuEZFitter *EZFitter)
{
  // Only accept if they passed a (valid) pointer
  if (EZFitter) fFitters.push_back(EZFitter); 
  
  BuildFitter();
}


NuMatrixSpectrum NuEZRunsFitter::PredictionNQ(Double_t dm2bar, Double_t sn2bar,
                                        Double_t dm2, Double_t sn2) const
{
  NuMMParameters mmPars;
  mmPars.Dm2(dm2);
  mmPars.Sn2(sn2);
  mmPars.Dm2Bar(dm2bar);
  mmPars.Sn2Bar(sn2bar);
  
  return PredictionNQ(mmPars);
}

NuMatrixSpectrum NuEZRunsFitter::PredictionPQ(Double_t dm2bar, Double_t sn2bar,
                                        Double_t dm2, Double_t sn2) const
{
  NuMMParameters mmPars;
  mmPars.Dm2(dm2);
  mmPars.Sn2(sn2);
  mmPars.Dm2Bar(dm2bar);
  mmPars.Sn2Bar(sn2bar);
  
  return PredictionPQ(mmPars);
}

NuMatrixSpectrum NuEZRunsFitter::PredictionNQ(const NuMMParameters &params) const
{
  NuMatrixSpectrum total;
  // Loop over all runs, and combine the spectra
  for (UInt_t i = 0; i < fFitters.size(); i++) {
    total.Add(fFitters[i]->PredictionNQ(params), true);
  }
  
  return total;
}
NuMatrixSpectrum NuEZRunsFitter::PredictionPQ(const NuMMParameters &params) const
{
  NuMatrixSpectrum total;
  // Loop over all runs, and combine the spectra
  for (UInt_t i = 0; i < fFitters.size(); i++) {
    total.Add(fFitters[i]->PredictionPQ(params), true);
  }
  
  return total;
}

NuMatrixSpectrum NuEZRunsFitter::FDDataNQ() const
{
  NuMatrixSpectrum total;
  // Loop over all runs, and combine the spectra
  for (UInt_t i = 0; i < fFitters.size(); i++) {
    // Add the data spectrum, accumulating POTs
    total.Add(fFitters[i]->FDDataNQ(), true);
  }
  
  return total;
}

NuMatrixSpectrum NuEZRunsFitter::FDDataPQ() const
{
  NuMatrixSpectrum total;
  // Loop over all runs, and combine the spectra
  for (UInt_t i = 0; i < fFitters.size(); i++) {
    total.Add(fFitters[i]->FDDataPQ(), true);
  }
  
  return total;
}

void NuEZRunsFitter::BuildFitter( void )
{
  // First delete the old fitter, if it exists
  if (fFitter) delete fFitter;
  fFitter = 0;

  // Only build if we have runs
  if (fFitters.size() == 0) return;
  
  // Create!
  fFitter = new NuSystFitter();
  
  // Add all of our runs to the new fitter
  for(UInt_t i = 0; i < fFitters.size(); i++) {
    fFitter->push_back(fFitters[i]->Run());
  }
}

Double_t NuEZRunsFitter::Chi2(Double_t dm2bar, Double_t sn2bar, Double_t dm2, Double_t sn2) const
{
  NuMMParameters mmPars;
  mmPars.Dm2(dm2);
  mmPars.Sn2(sn2);
  mmPars.Dm2Bar(dm2bar);
  mmPars.Sn2Bar(sn2bar);
  
  // return (*fFitter)(mmPars.VectorParameters());
  return Chi2(mmPars);
}

Double_t NuEZRunsFitter::Chi2(const NuMMParameters &params) const
{
  if (fFitter) {
    return (*fFitter)(params.VectorParameters());
  } else {
    MSG("NuEZRunsFitter",Msg::kError) << "No Fitter exists. Have you added FD data?" << endl;
    return 0;
  }
}

---