Absorbed Andreas' fakeData in the TPofTCalc-class
This commit is contained in:
Bastian M. Wojek
parent
999b82008b
commit
59e95d014f
@ -1071,9 +1071,9 @@ TLondon1D3LSub::TLondon1D3LSub() : fCalcNeeded(true), fFirstCall(true), fWeights
|
||||
}
|
||||
|
||||
//------------------
|
||||
// TLondon1D3L-Method that calls the procedures to create B(z), p(B) and P(t)
|
||||
// TLondon1D3LSub-Method that calls the procedures to create B(z), p(B) and P(t)
|
||||
// It finally returns P(t) for a given t.
|
||||
// Parameters: all the parameters for the function to be fitted through TLondon1D3L
|
||||
// Parameters: all the parameters for the function to be fitted through TLondon1D3LSub
|
||||
//------------------
|
||||
|
||||
double TLondon1D3LSub::operator()(double t, const vector<double> &par) const {
|
||||
|
||||
208
src/external/TFitPofB-lib/classes/TPofTCalc.cpp
vendored
208
src/external/TFitPofB-lib/classes/TPofTCalc.cpp
vendored
@ -5,20 +5,55 @@
|
||||
Author: Bastian M. Wojek
|
||||
e-mail: bastian.wojek@psi.ch
|
||||
|
||||
2008/09/02
|
||||
2008/11/16
|
||||
|
||||
***************************************************************************/
|
||||
|
||||
/***************************************************************************
|
||||
|
||||
TPofTCalc::FakeData Method based on Andreas Suter's fakeData
|
||||
|
||||
***************************************************************************/
|
||||
|
||||
/***************************************************************************
|
||||
* Copyright (C) 2008 by Andreas Suter *
|
||||
* andreas.suter@psi.ch *
|
||||
* *
|
||||
* This program is free software; you can redistribute it and/or modify *
|
||||
* it under the terms of the GNU General Public License as published by *
|
||||
* the Free Software Foundation; either version 2 of the License, or *
|
||||
* (at your option) any later version. *
|
||||
* *
|
||||
* This program is distributed in the hope that it will be useful, *
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
||||
* GNU General Public License for more details. *
|
||||
* *
|
||||
* You should have received a copy of the GNU General Public License *
|
||||
* along with this program; if not, write to the *
|
||||
* Free Software Foundation, Inc., *
|
||||
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
|
||||
***************************************************************************/
|
||||
|
||||
#include "TPofTCalc.h"
|
||||
#include "fftw3.h"
|
||||
#include <cmath>
|
||||
#include <iostream>
|
||||
#include <cstdio>
|
||||
|
||||
#include <TString.h>
|
||||
#include <TObjArray.h>
|
||||
#include <TObjString.h>
|
||||
#include <TFile.h>
|
||||
#include <TH1F.h>
|
||||
#include <TFolder.h>
|
||||
|
||||
#include "TLemRunHeader.h"
|
||||
|
||||
/* USED FOR DEBUGGING -----------------------
|
||||
#include <ctime>
|
||||
#include <fstream>
|
||||
/--------------------------------------------*/
|
||||
--------------------------------------------*/
|
||||
|
||||
//------------------
|
||||
// Constructor of the TPofTCalc class - it creates the FFT plan
|
||||
@ -120,16 +155,181 @@ void TPofTCalc::CalcPol(const vector<double> &par) {
|
||||
}
|
||||
}
|
||||
|
||||
//---------------------
|
||||
// Method for generating fake LEM decay histograms from p(B)
|
||||
// Parameters: par(dt, dB, timeres, channels, asyms, phases, t0s, N0s, bgs), output filename
|
||||
//---------------------
|
||||
|
||||
void TPofTCalc::FakeData(const vector<double> &par, const string &rootOutputFileName) {
|
||||
|
||||
//determine the number of histograms to be built
|
||||
unsigned int numHist(0);
|
||||
if(!((par.size()-4)%5))
|
||||
numHist=(par.size()-4)/5;
|
||||
|
||||
if(!numHist){
|
||||
cout << "The number of parameters for the histogram creation is not correct. Do nothing." << endl;
|
||||
return;
|
||||
}
|
||||
|
||||
cout << numHist << " histograms to be built" << endl;
|
||||
|
||||
vector<double> param;
|
||||
param.push_back(0.0);
|
||||
param.push_back(par[0]);
|
||||
param.push_back(par[1]);
|
||||
|
||||
vector< vector<double> > asy;
|
||||
vector<double> asydata;
|
||||
double ttime(0.0);
|
||||
double pol(0.0);
|
||||
|
||||
for(unsigned int i(0); i<numHist; i++) {
|
||||
param[0]=par[i+numHist*1+4];
|
||||
// calculate asymmetry
|
||||
CalcPol(param);
|
||||
for(unsigned int j(0); j<par[3]; j++){
|
||||
ttime=j*par[2];
|
||||
for(unsigned int k(0); k<fT.size()-1; k++){
|
||||
if (ttime < fT[k+1]) {
|
||||
pol=fPT[k]+(fPT[k+1]-fPT[k])/(fT[k+1]-fT[k])*(ttime-fT[k]);
|
||||
asydata.push_back(par[i+numHist*0+4]*pol);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
asy.push_back(asydata);
|
||||
asydata.clear();
|
||||
cout << "Asymmetry " << i+1 << "/" << numHist << " calculated!" << endl;
|
||||
}
|
||||
|
||||
// calculate the histograms
|
||||
vector< vector<double> > histo;
|
||||
vector<double> data;
|
||||
double value;
|
||||
|
||||
for (unsigned int i(0); i<numHist; i++) { // loop over all histos
|
||||
for (unsigned int j(0); j<par[3]; j++) { // loop over time
|
||||
if (j < par[i+numHist*2+4]) // j<t0
|
||||
value = par[i+numHist*4+4]; // background
|
||||
else
|
||||
value = par[i+numHist*3+4]*exp(-par[2]*(j-par[i+numHist*2+4])/tauMu)*(1.0+asy[i][j- int(par[i+numHist*2+4])])+par[i+numHist*4+4];
|
||||
data.push_back(value);
|
||||
}
|
||||
histo.push_back(data);
|
||||
data.clear();
|
||||
cout << endl << ">> histo " << i+1 << "/" << numHist << " done ...";
|
||||
}
|
||||
|
||||
// add Poisson noise to the histograms
|
||||
|
||||
cout << endl << ">> add Poisson noise ..." << endl;
|
||||
|
||||
TH1F* theoHisto;
|
||||
TH1F* fakeHisto;
|
||||
vector<TH1F*> histoData;
|
||||
|
||||
TString name;
|
||||
for (unsigned int i(0); i<numHist; i++) { // loop over all histos
|
||||
// create histos
|
||||
name = "theoHisto";
|
||||
name += i;
|
||||
theoHisto = new TH1F(name.Data(), name.Data(), int(par[3]), -par[2]/2.0, (par[3]+0.5)*par[2]);
|
||||
if (i < 10)
|
||||
name = "hDecay0";
|
||||
else
|
||||
name = "hDecay";
|
||||
name += i;
|
||||
fakeHisto = new TH1F(name.Data(), name.Data(), int(par[3]), -par[2]/2.0, (par[3]+0.5)*par[2]);
|
||||
// fill theoHisto
|
||||
for (unsigned int j(0); j<par[3]; j++)
|
||||
theoHisto->SetBinContent(j, histo[i][j]);
|
||||
// fill fakeHisto
|
||||
fakeHisto->FillRandom(theoHisto, (int)theoHisto->Integral());
|
||||
|
||||
// keep fake data
|
||||
histoData.push_back(fakeHisto);
|
||||
|
||||
// cleanup
|
||||
if (theoHisto) {
|
||||
delete theoHisto;
|
||||
theoHisto = 0;
|
||||
}
|
||||
}
|
||||
|
||||
// save the histograms as root files
|
||||
// create run info folder and content
|
||||
TFolder *runInfoFolder = new TFolder("RunInfo", "Run Info");
|
||||
TLemRunHeader *runHeader = new TLemRunHeader();
|
||||
//sprintf(str, "Fake Data generated from %s", pBFileName.Data());
|
||||
runHeader->SetRunTitle("Fake Data");
|
||||
float fval = par[2]*1000.; //us->ns
|
||||
runHeader->SetTimeResolution(fval);
|
||||
runHeader->SetNChannels(int(par[3]));
|
||||
runHeader->SetNHist(histoData.size());
|
||||
double *t0array = new double[histoData.size()];
|
||||
for (unsigned int i(0); i<histoData.size(); i++)
|
||||
t0array[i] = par[i+numHist*2+4];
|
||||
runHeader->SetTimeZero(t0array);
|
||||
if (t0array)
|
||||
delete t0array;
|
||||
runInfoFolder->Add(runHeader);
|
||||
|
||||
// create decay histo folder and content
|
||||
TFolder *histoFolder = new TFolder("histos", "histos");
|
||||
TFolder *decayAnaModule = new TFolder("DecayAnaModule", "DecayAnaModule");
|
||||
histoFolder->Add(decayAnaModule);
|
||||
// no post pileup corrected (NPP)
|
||||
for (unsigned int i(0); i<histoData.size(); i++)
|
||||
decayAnaModule->Add(histoData[i]);
|
||||
// post pileup corrected (PPC)
|
||||
vector<TH1F*> histoDataPPC;
|
||||
for (unsigned int i(0); i<histoData.size(); i++) {
|
||||
histoDataPPC.push_back(dynamic_cast<TH1F*>(histoData[i]->Clone()));
|
||||
if (i < 10)
|
||||
name = "hDecay2";
|
||||
else
|
||||
name = "hDecay";
|
||||
name += i;
|
||||
histoDataPPC[i]->SetNameTitle(name.Data(), name.Data());
|
||||
decayAnaModule->Add(histoDataPPC[i]);
|
||||
}
|
||||
|
||||
// write file
|
||||
TFile fdf(rootOutputFileName.c_str(), "recreate");
|
||||
runInfoFolder->Write("RunInfo", TObject::kSingleKey);
|
||||
histoFolder->Write();
|
||||
fdf.Close();
|
||||
|
||||
// clean up
|
||||
for (unsigned int i(0); i<histo.size(); i++) {
|
||||
asy[i].clear();
|
||||
histo[i].clear();
|
||||
}
|
||||
asy.clear();
|
||||
histo.clear();
|
||||
|
||||
for (unsigned int i(0); i<histoData.size(); i++) {
|
||||
delete histoData[i];
|
||||
delete histoDataPPC[i];
|
||||
}
|
||||
histoData.clear();
|
||||
histoDataPPC.clear();
|
||||
|
||||
cout << endl << ">> DONE." << endl;
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
//---------------------
|
||||
// Method for evaluating P(t) at a given t
|
||||
//---------------------
|
||||
|
||||
double TPofTCalc::Eval(double t) const {
|
||||
for (unsigned int i(0); i<fT.size()-1; i++) {
|
||||
if (t < fT[i+1]) {
|
||||
if (t < fT[i+1])
|
||||
return fPT[i]+(fPT[i+1]-fPT[i])/(fT[i+1]-fT[i])*(t-fT[i]);
|
||||
}
|
||||
}
|
||||
|
||||
cout << "No data for the time " << t << " us available! Returning -999.0 ..." << endl;
|
||||
return -999.0;
|
||||
|
||||
@ -5,7 +5,7 @@
|
||||
Author: Bastian M. Wojek
|
||||
e-mail: bastian.wojek@psi.ch
|
||||
|
||||
2008/06/03
|
||||
2008/11/16
|
||||
|
||||
***************************************************************************/
|
||||
|
||||
@ -17,6 +17,7 @@
|
||||
#include <string>
|
||||
|
||||
#define PI 3.14159265358979323846
|
||||
#define tauMu 2.197147
|
||||
|
||||
class TPofTCalc {
|
||||
|
||||
@ -29,6 +30,7 @@ public:
|
||||
vector<double> DataPT() const {return fPT;}
|
||||
void DoFFT(const TPofBCalc&);
|
||||
void CalcPol(const vector<double>&);
|
||||
void FakeData(const vector<double>&, const string&);
|
||||
double Eval(double) const;
|
||||
|
||||
private:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user