KVGaus2D.cpp

//Created by KVClassFactory on Tue Feb  7 16:49:15 2012
//Author: bonnet
 
#include "KVGaus2D.h"
#include "TH2.h"
#include "Riostream.h"
#include "TMath.h"
#include "TMarker.h"
 
ClassImp(KVGaus2D)
 
//________________________________________________________________
 
 
KVGaus2D::KVGaus2D(Bool_t batch) : KVAutoFit(batch)
{
   // Default constructor
   SetName("Gaus2D");
   is2D = kTRUE;
 
}
 
 
 
 
 
KVGaus2D::KVGaus2D(const KVGaus2D&)  : KVAutoFit()
{
   // Copy constructor
   // This ctor is used to make a copy of an existing object (for example
   // when a method returns an object), and it is always a good idea to
   // implement it.
   // If your class allocates memory in its constructor(s) then it is ESSENTIAL :-)
 
}
 
 
 
 
Bool_t KVGaus2D::NewFunction_2D()
{
 
   Int_t nf = lfunc->GetEntries();
 
   f2Dfit = new TF2(Form("func_%d", ++nf),
                    this,
                    &KVGaus2D::f2D,
                    Xmin,
                    Xmax,
                    Ymin,
                    Ymax,
                    6,
                    GetName(),
                    "f2D"
                   );
 
   lfunc->Add(f2Dfit);
 
   hclone->GetXaxis()->SetRangeUser(Xmin, Xmax);
   hclone->GetYaxis()->SetRangeUser(Ymin, Ymax);
 
   f2Dfit->SetParameters(
      hclone->GetMean(1),
      hclone->GetRMS(1),
      hclone->GetMean(2),
      hclone->GetRMS(2),
      ((TH2*)hclone)->GetCorrelationFactor(),
      hclone->Integral()
   );
 
   return kTRUE;
}
 
 
 
 
TF1* KVGaus2D::ReloadFunction_2D(const Char_t* nom, Int_t npar)
{
 
   TF2* f2 = new TF2(nom,
                     this,
                     &KVGaus2D::f2D,
                     0, 1, 0, 1,
                     npar,
                     GetName(),
                     "f2D"
                    );
 
   return f2;
}
 
 
 
 
Double_t KVGaus2D::f2D(Double_t* xx, Double_t* para)
{
 
//Function gaussienne 2D
//
// E1 vs E2
 
   userdefined = kTRUE;
 
   Double_t e2 = xx[0];
   Double_t e1 = xx[1];
 
   Double_t E2    =  para[0];
   Double_t sigE2 =  para[1];
   Double_t E1 =  para[2];
   Double_t sigE1 =  para[3];
   Double_t rho   =  para[4];
   Double_t norm  =  para[5];
 
   Double_t dE1   =  E1 - e1;
   Double_t dE2   =  E2 - e2;
 
   Double_t A = -0.5 / ((1. - TMath::Power(rho, 2.)) * TMath::Power(sigE1 * sigE2, 2.)) * (TMath::Power(dE1 * sigE2, 2.) + TMath::Power(dE2 * sigE1, 2.) - 2.*rho * dE1 * dE2 * sigE1 * sigE2);
   Double_t det = (1. - TMath::Power(rho, 2.)) * TMath::Power(sigE1 * sigE2, 2.);
   Double_t P = norm * 1. / det * exp(A);
 
   return P;
 
}
 
 
 
 
void KVGaus2D::ExtraDrawing()
{
   TMarker* m1 = new TMarker(f2Dfit->GetParameter(0), f2Dfit->GetParameter(2), 8);
   m1->SetMarkerSize(0.7);
 
   m1->Draw();
}