test_fitvarsproduct.C

Go to the documentation of this file.

#include "CAFAna/Vars/FitVarsProduct.h"
using namespace ana;

#include "OscLib/OscCalcPMNSOpt.h"

#include "TCanvas.h"
#include "TGraph.h"
#include "TH2.h"
#include "TRandom3.h"

#include <iostream>

void test_fitvarsproduct()
{
  auto alpha = FitVarSinSqTheta23SinSq2Theta13();
  auto beta  = FitVarSinSqTheta23SinSq2Theta13Marg();

  const IFitVar& x = kFitSinSq2Theta13;
  const IFitVar& y = kFitSinSqTheta23;

  osc::OscCalcPMNSOpt osc;

  // Check that you can set the variables in any order and get the same values
  // back.
  for(int i = 0; i < 1000; ++i){
    const double vx = gRandom->Uniform(0, 1);
    x.SetValue(&osc, vx);
    assert(fabs(vx - x.GetValue(&osc) < 1e-6));
    const double vy = gRandom->Uniform(0, 1);
    y.SetValue(&osc, vy);
    assert(fabs(vy - y.GetValue(&osc) < 1e-6));
    assert(fabs(vx - x.GetValue(&osc) < 1e-6));
  }

  for(int i = 0; i < 1000; ++i){
    const double va = gRandom->Uniform(0, 1);
    alpha.SetValue(&osc, va);
    assert(fabs(va - alpha.GetValue(&osc) < 1e-6));
    const double vb = gRandom->Uniform(0, 1);
    beta.SetValue(&osc, vb);
    assert(fabs(vb - beta.GetValue(&osc) < 1e-6));
    assert(fabs(va - alpha.GetValue(&osc) < 1e-6));
  }

  // Show what scanning x and y looks like on alpha, beta axes
  TH2* axes = new TH2F("", "", 120, -.1, 1.1, 120, -.1, 1.1);
  axes->GetXaxis()->SetTitle(alpha.LatexName().c_str());
  axes->GetYaxis()->SetTitle(beta.LatexName().c_str());
  axes->Draw();
  for(int ix = 0; ix <= 20; ++ix){
    TGraph* g = new TGraph;
    for(int iy = 0; iy <= 100; ++iy){
      x.SetValue(&osc, ix*.05);
      y.SetValue(&osc, iy*.01);
      g->SetPoint(iy, alpha.GetValue(&osc), beta.GetValue(&osc));
    }
    g->SetLineWidth(2);
    g->Draw("l same");
  }

  for(int iy = 0; iy <= 20; ++iy){
    TGraph* g = new TGraph;
    for(int ix = 0; ix <= 100; ++ix){
      x.SetValue(&osc, ix*.01);
      y.SetValue(&osc, iy*.05);
      g->SetPoint(ix, alpha.GetValue(&osc), beta.GetValue(&osc));
    }
    g->SetLineWidth(2);
    g->Draw("l same");
  }


  // Show what scanning alpha and beta looks like on x, y axes
  new TCanvas;
  axes = new TH2F("", "", 120, -.1, 1.1, 120, -.1, 1.1);
  axes->GetXaxis()->SetTitle(x.LatexName().c_str());
  axes->GetYaxis()->SetTitle(y.LatexName().c_str());
  axes->Draw();

  for(int ia = 0; ia <= 20; ++ia){
    TGraph* g = new TGraph;
    for(int ib = 0; ib <= 100; ++ib){
      alpha.SetValue(&osc, ia*.05);
      beta.SetValue(&osc, ib*.01);
      g->SetPoint(ib, x.GetValue(&osc), y.GetValue(&osc));
    }
    g->SetLineWidth(2);
    g->Draw("l same");
  }

  for(int ib = 0; ib <= 20; ++ib){
    TGraph* g = new TGraph;
    for(int ia = 0; ia <= 100; ++ia){
      alpha.SetValue(&osc, ia*.01);
      beta.SetValue(&osc, ib*.05);
      g->SetPoint(ia, x.GetValue(&osc), y.GetValue(&osc));
    }
    g->SetLineWidth(2);
    g->Draw("l same");
  }
}