//------------------------------------------------------------------------------
// InvertAxis
//------------------------------------------------------------------------------
TH2F* InvertAxis(TH2* h2, TString hname)
{
TArrayD* xarray = (TArrayD*)h2->GetXaxis()->GetXbins();
TArrayD* yarray = (TArrayD*)h2->GetYaxis()->GetXbins();
const UInt_t xsize = xarray->GetSize();
const UInt_t ysize = yarray->GetSize();
Double_t* xbins = new Double_t[xsize];
Double_t* ybins = new Double_t[ysize];
for (UInt_t i=0; i<xsize; i++) xbins[i] = xarray->GetAt(i);
for (UInt_t i=0; i<ysize; i++) ybins[i] = yarray->GetAt(i);
TH2F* h2inverted = new TH2F(hname, hname,
h2->GetNbinsY(), ybins,
h2->GetNbinsX(), xbins);
for (UInt_t i=1; i<=h2inverted->GetNbinsX(); i++) {
for (UInt_t j=1; j<=h2inverted->GetNbinsY(); j++) {
h2inverted->SetBinContent(i, j, h2->GetBinContent(j, i));
h2inverted->SetBinError (i, j, h2->GetBinError (j, i));
}
}
delete [] xbins;
delete [] ybins;
return h2inverted;
}
void showFunction2d(TF1* function,
const TArrayD& testValuesX, const TArrayD& testValuesY, double metResolution, double metErr,
const TString& xAxisTitle, const TString& yAxisTitle,
const std::string& outputFileName)
{
int numTestValuesX = testValuesX.GetSize();
for ( int iTestValueX = 0; iTestValueX < numTestValuesX; ++iTestValueX ) {
double testValueX = testValuesX[iTestValueX];
std::vector<TF1*> functions;
std::vector<std::string> legendEntries;
int numTestValuesY = testValuesY.GetSize();
for ( int iTestValueY = 0; iTestValueY < numTestValuesY; ++iTestValueY ) {
double testValueY = testValuesY[iTestValueY];
std::string functionName_nll = Form("%s_nll_%i", function->GetName(), iTestValueY);
TF1* function_nll = 0;
if ( metResolution > 0. ) {
function_nll =
new TF1(functionName_nll.data(),
Form("-TMath::Log((%s)*TMath::Gaus(([1] - [0])*x, ([1] - [0]) + [2], [3]))",
function->GetTitle()),
function->GetXmin(), function->GetXmax());
} else {
function_nll = new TF1(functionName_nll.data(), function->GetTitle(), function->GetXmin(), function->GetXmax());
}
function_nll->SetParameter(0, testValueX);
function_nll->SetParameter(1, testValueY);
function_nll->SetParameter(2, metErr);
function_nll->SetParameter(3, metResolution);
functions.push_back(function_nll);
legendEntries.push_back(std::string(Form("m_{2} = %1.1f", testValueY)));
}
double yMax = 0.5*(0.1 + testValueX);
size_t idx = outputFileName.find_last_of('.');
TString outputFileName_plot = std::string(outputFileName, 0, idx).data();
outputFileName_plot.Append(Form("_m1Eq%1.1f", testValueX));
outputFileName_plot.ReplaceAll(".", "_");
if ( idx != std::string::npos ) outputFileName_plot.Append(std::string(outputFileName, idx).data());
showFunctions1d(functions, legendEntries, xAxisTitle, 0., yMax, yAxisTitle, outputFileName_plot.Data());
for ( std::vector<TF1*>::iterator it = functions.begin();
it != functions.end(); ++it ) {
delete (*it);
}
}
}
double* Divide(const TArrayD* array , double val){
TArrayD* temp = new TArrayD();
array->Copy(*temp);
for(int i = 0 ; i < temp->GetSize() ; i++)
temp->SetAt(array->At(i) / val , i);
return temp->GetArray();
}