leverington
/
hit_analyse_v2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
data analysis scripts
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 Commits
1 Branch
0 Tags
15 MiB
 
 
 
 
Tree: 6f25f8675f
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '6f25f8675f'
${ noResults }
hit_analyse_v2/Scripts_20161126/analyse2.h

919 lines
29 KiB
Raw Blame History

//////////////////////////////////////////////////////////
// This class has been automatically generated on
// Tue Sep 12 15:40:57 2017 by ROOT version 5.34/32
// from TTree t/t
// found on file: ../../HIT_26-11-2016/with_timestamp/pin/PiN_run1.root
//////////////////////////////////////////////////////////
#ifndef analyse2_h
#define analyse2_h
#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include <TTree.h>
#include <stdio.h>
#include <TH2D.h>
#include <TH1D.h>
#include <TProfile.h>
#include <TCanvas.h>
#include <TF1.h>
#include <TMath.h>
#include <vector>
#include <utility>
#include <TString.h>
#include <TDirectory.h>
#include <TString.h>
#include <TVectorD.h>
#include <TGraph.h>
#include <TGraphErrors.h>
#include <cmath>
#include <TPaveStats.h>
#include <TText.h>
#include <TStyle.h>
#include <TObjArray.h>
#include <TLegend.h>
#include <TFitResult.h>
#include <gsl/gsl_statistics.h>
#include <math.h>
#include <gsl/gsl_errno.h>
#include <gsl/gsl_fft_complex.h>
using namespace std;
// Header file for the classes stored in the TTree if any.
// Fixed size dimensions of array or collections stored in the TTree if any.
class analyse2 {
public :
TTree *fChain; //!pointer to the analyzed TTree or TChain
Int_t fCurrent; //!current Tree number in a TChain
TTree *data;
TTree *newdata;
// Declaration of leaf types
Double_t time;
Double_t event_id;
Double_t sync_out;
Double_t sync_in;
Double_t data_ok;
Double_t led_voltage;
Double_t current;
Double_t ch00;
Double_t ch01;
Double_t ch02;
Double_t ch03;
Double_t ch04;
Double_t ch05;
Double_t ch06;
Double_t ch07;
Double_t ch08;
Double_t ch09;
Double_t ch10;
Double_t ch11;
Double_t ch12;
Double_t ch13;
Double_t ch14;
Double_t ch15;
Double_t ch16;
Double_t ch17;
Double_t ch18;
Double_t ch19;
Double_t ch20;
Double_t ch21;
Double_t ch22;
Double_t ch23;
Double_t ch24;
Double_t ch25;
Double_t ch26;
Double_t ch27;
Double_t ch28;
Double_t ch29;
Double_t ch30;
Double_t ch31;
Double_t ch32;
Double_t ch33;
Double_t ch34;
Double_t ch35;
Double_t ch36;
Double_t ch37;
Double_t ch38;
Double_t ch39;
Double_t ch40;
Double_t ch41;
Double_t ch42;
Double_t ch43;
Double_t ch44;
Double_t ch45;
Double_t ch46;
Double_t ch47;
Double_t ch48;
Double_t ch49;
Double_t ch50;
Double_t ch51;
Double_t ch52;
Double_t ch53;
Double_t ch54;
Double_t ch55;
Double_t ch56;
Double_t ch57;
Double_t ch58;
Double_t ch59;
Double_t ch60;
Double_t ch61;
Double_t ch62;
Double_t ch63;
Double_t ic1;
Double_t ic2;
Double_t mw1_focusx;
Double_t mw1_focusy;
Double_t mw2_focusx;
Double_t mw2_focusy;
Double_t mw1_posx;
Double_t mw1_posy;
Double_t mw2_posx;
Double_t mw2_posy;
// List of branches
TBranch *b_time; //!
TBranch *b_event_id; //!
TBranch *b_sync_out; //!
TBranch *b_sync_in; //!
TBranch *b_data_ok; //!
TBranch *b_led_voltage; //!
TBranch *b_current; //!
TBranch *b_ch00; //!
TBranch *b_ch01; //!
TBranch *b_ch02; //!
TBranch *b_ch03; //!
TBranch *b_ch04; //!
TBranch *b_ch05; //!
TBranch *b_ch06; //!
TBranch *b_ch07; //!
TBranch *b_ch08; //!
TBranch *b_ch09; //!
TBranch *b_ch10; //!
TBranch *b_ch11; //!
TBranch *b_ch12; //!
TBranch *b_ch13; //!
TBranch *b_ch14; //!
TBranch *b_ch15; //!
TBranch *b_ch16; //!
TBranch *b_ch17; //!
TBranch *b_ch18; //!
TBranch *b_ch19; //!
TBranch *b_ch20; //!
TBranch *b_ch21; //!
TBranch *b_ch22; //!
TBranch *b_ch23; //!
TBranch *b_ch24; //!
TBranch *b_ch25; //!
TBranch *b_ch26; //!
TBranch *b_ch27; //!
TBranch *b_ch28; //!
TBranch *b_ch29; //!
TBranch *b_ch30; //!
TBranch *b_ch31; //!
TBranch *b_ch32; //!
TBranch *b_ch33; //!
TBranch *b_ch34; //!
TBranch *b_ch35; //!
TBranch *b_ch36; //!
TBranch *b_ch37; //!
TBranch *b_ch38; //!
TBranch *b_ch39; //!
TBranch *b_ch40; //!
TBranch *b_ch41; //!
TBranch *b_ch42; //!
TBranch *b_ch43; //!
TBranch *b_ch44; //!
TBranch *b_ch45; //!
TBranch *b_ch46; //!
TBranch *b_ch47; //!
TBranch *b_ch48; //!
TBranch *b_ch49; //!
TBranch *b_ch50; //!
TBranch *b_ch51; //!
TBranch *b_ch52; //!
TBranch *b_ch53; //!
TBranch *b_ch54; //!
TBranch *b_ch55; //!
TBranch *b_ch56; //!
TBranch *b_ch57; //!
TBranch *b_ch58; //!
TBranch *b_ch59; //!
TBranch *b_ch60; //!
TBranch *b_ch61; //!
TBranch *b_ch62; //!
TBranch *b_ch63; //!
TBranch *b_ic1; //!
TBranch *b_ic2; //!
TBranch *b_mw1_focusx; //!
TBranch *b_mw1_focusy; //!
TBranch *b_mw2_focusx; //!
TBranch *b_mw2_focusy; //!
TBranch *b_mw1_posx; //!
TBranch *b_mw1_posy; //!
TBranch *b_mw2_posx; //!
TBranch *b_mw2_posy; //!
char *file;
TFile *fileIn;
TFile *fileOut;
TObjArray objectArray;
TBranch *b_channels[64];
TBranch *b_baseline[64];
double channels[64];
double channelamp_smooth[64];
double baseline[64];
double channellist[64];
double errorx[64];
double errory[64];
TBranch *b_beamNumX_1;
TBranch *b_beamPosX_1;
TBranch *b_beamFocusX_1;
TBranch *b_beamSignal_1;
TBranch *b_beamSidebandNoise_1;
TBranch *b_beamSkewX_1;
TBranch *b_beamKurtX_1;
TBranch *b_beamPosX_fit;
TBranch *b_beamFocusX_fit;
TBranch *b_beamChi2_fit;
TBranch *b_beamPeakX_fit;
TBranch *b_beamPosX_fit2;
TBranch *b_beamFocusX_fit2;
TBranch *b_beamFocus2X_fit2;
TBranch *b_beamChi2_fit2;
TBranch *b_beamPeakX_fit2;
TBranch *b_beamPeak2X_fit2;
TBranch *b_ic1_1; //!
TBranch *b_ic2_1; //!
TBranch *b_mw1_focusx_1; //!
TBranch *b_mw1_focusy_1; //!
TBranch *b_mw2_focusx_1; //!
TBranch *b_mw2_focusy_1; //!
TBranch *b_mw1_posx_1; //!
TBranch *b_mw1_posy_1; //!
TBranch *b_mw2_posx_1; //!
TBranch *b_mw2_posy_1; //!
TBranch *b_baseline_integral;
TBranch *b_time_1; //!
TBranch *b_eventid_1; //!
TBranch *b_rollingavg; //!
TBranch *b_beamon; //!
double time_1;
double ic1_1;
double ic2_1;
double mw1_focusx_1;
double mw2_focusx_1;
double mw2_posx_1;
double mw1_posx_1;
double mw1_focusy_1;
double mw2_focusy_1;
double mw2_posy_1;
double mw1_posy_1;
double beamNumX_1;
double beamPosX_1;
double beamFocusX_1;
double beamSignal_1;
double beamSidebandNoise_1;
double beamSkewX_1;
double beamKurtX_1;
double baseline_integral;
double chargeintegral;
double timeintegral;
double eventid_1;
double rollingavg;
int beamon;
double beamPosX_fit;
double beamFocusX_fit;
double beamChi2_fit;
double beamPeakX_fit;
double beamPosX_fit2;
double beamFocusX_fit2;
double beamFocus2X_fit2;
double beamChi2_fit2;
double beamPeakX_fit2;
double beamPeak2X_fit2;
analyse2(TTree *tree=0);
virtual ~analyse2();
virtual Int_t Cut(Long64_t entry);
virtual Int_t GetEntry(Long64_t entry);
virtual Long64_t LoadTree(Long64_t entry);
virtual void Init(TTree *tree);
virtual void Loop();
virtual Bool_t Notify();
virtual void Show(Long64_t entry = -1);
double channelWidth;
void Initialize(const char *dirname, const char *filename, int baselineEvents = 1000, int prelimEvents = 10000, double beamLevel = 0.1, bool firstFibermat = true, double readOutFrequency = 1000., int integrationTime = 100);
void Baseline(bool useFirstEvents, int nBaselineEvents, int StartEvent, bool run50hz);
void Save();
void Close();
double * FFTsmoothing(double arr[]);
// double RollingAverage(const int length, double newdata);
TH2D * th2d_mw1_beamPosX;
TH2D * th2d_ic1_beamSignal;
TH1D * th1d_beamSignal_ic1ratio;
TH2D * th2d_beamSignal_channel;
TH2D * th2d_bkg_channel;
TH1D * th1d_baseline;
TH1D * th1d_baselinesub;
TH1D * th1d_chbaseline;
TH1D * th1d_chbaselinesub;
TH2D * th2d_fitdiff_channel;
TH2D * th2d_fit2diff_channel;
TH2D * th2d_AmpFFT_channel;
TH2D * th2d_PhaseFFT_channel;
TFitResultPtr bkgfitResult;
Double_t bkgfitPar0[65];
Double_t bkgfitPar1[65];
Double_t bkgfitPar2[65];
Double_t bkgfitPar3[65];
Double_t bkgfitPar4[65];
};
#endif
#ifdef analyse2_cxx
analyse2::analyse2(TTree *tree) : fChain(0)
{
// if parameter tree is not specified (or zero), connect the file
// used to generate this class and read the Tree.
if (tree == 0) {
cerr << "no TTree found." << endl;
}
Init(tree);
}
analyse2::~analyse2()
{
if (!fChain) return;
delete fChain->GetCurrentFile();
}
Int_t analyse2::GetEntry(Long64_t entry)
{
// Read contents of entry.
if (!fChain) return 0;
return fChain->GetEntry(entry);
}
Long64_t analyse2::LoadTree(Long64_t entry)
{
// Set the environment to read one entry
if (!fChain) return -5;
Long64_t centry = fChain->LoadTree(entry);
if (centry < 0) return centry;
if (fChain->GetTreeNumber() != fCurrent) {
fCurrent = fChain->GetTreeNumber();
Notify();
}
return centry;
}
void analyse2::Init(TTree *tree)
{
// The Init() function is called when the selector needs to initialize
// a new tree or chain. Typically here the branch addresses and branch
// pointers of the tree will be set.
// It is normally not necessary to make changes to the generated
// code, but the routine can be extended by the user if needed.
// Init() will be called many times when running on PROOF
// (once per file to be processed).
// Set branch addresses and branch pointers
if (!tree) return;
fChain = tree;
fCurrent = -1;
fChain->SetMakeClass(1);
fChain->SetBranchAddress("time", &time, &b_time);
fChain->SetBranchAddress("event_id", &event_id, &b_event_id);
fChain->SetBranchAddress("sync_out", &sync_out, &b_sync_out);
fChain->SetBranchAddress("sync_in", &sync_in, &b_sync_in);
fChain->SetBranchAddress("data_ok", &data_ok, &b_data_ok);
fChain->SetBranchAddress("led_voltage", &led_voltage, &b_led_voltage);
fChain->SetBranchAddress("current", &current, &b_current);
fChain->SetBranchAddress("ch00", &ch00, &b_ch00);
fChain->SetBranchAddress("ch01", &ch01, &b_ch01);
fChain->SetBranchAddress("ch02", &ch02, &b_ch02);
fChain->SetBranchAddress("ch03", &ch03, &b_ch03);
fChain->SetBranchAddress("ch04", &ch04, &b_ch04);
fChain->SetBranchAddress("ch05", &ch05, &b_ch05);
fChain->SetBranchAddress("ch06", &ch06, &b_ch06);
fChain->SetBranchAddress("ch07", &ch07, &b_ch07);
fChain->SetBranchAddress("ch08", &ch08, &b_ch08);
fChain->SetBranchAddress("ch09", &ch09, &b_ch09);
fChain->SetBranchAddress("ch10", &ch10, &b_ch10);
fChain->SetBranchAddress("ch11", &ch11, &b_ch11);
fChain->SetBranchAddress("ch12", &ch12, &b_ch12);
fChain->SetBranchAddress("ch13", &ch13, &b_ch13);
fChain->SetBranchAddress("ch14", &ch14, &b_ch14);
fChain->SetBranchAddress("ch15", &ch15, &b_ch15);
fChain->SetBranchAddress("ch16", &ch16, &b_ch16);
fChain->SetBranchAddress("ch17", &ch17, &b_ch17);
fChain->SetBranchAddress("ch18", &ch18, &b_ch18);
fChain->SetBranchAddress("ch19", &ch19, &b_ch19);
fChain->SetBranchAddress("ch20", &ch20, &b_ch20);
fChain->SetBranchAddress("ch21", &ch21, &b_ch21);
fChain->SetBranchAddress("ch22", &ch22, &b_ch22);
fChain->SetBranchAddress("ch23", &ch23, &b_ch23);
fChain->SetBranchAddress("ch24", &ch24, &b_ch24);
fChain->SetBranchAddress("ch25", &ch25, &b_ch25);
fChain->SetBranchAddress("ch26", &ch26, &b_ch26);
fChain->SetBranchAddress("ch27", &ch27, &b_ch27);
fChain->SetBranchAddress("ch28", &ch28, &b_ch28);
fChain->SetBranchAddress("ch29", &ch29, &b_ch29);
fChain->SetBranchAddress("ch30", &ch30, &b_ch30);
fChain->SetBranchAddress("ch31", &ch31, &b_ch31);
fChain->SetBranchAddress("ch32", &ch32, &b_ch32);
fChain->SetBranchAddress("ch33", &ch33, &b_ch33);
fChain->SetBranchAddress("ch34", &ch34, &b_ch34);
fChain->SetBranchAddress("ch35", &ch35, &b_ch35);
fChain->SetBranchAddress("ch36", &ch36, &b_ch36);
fChain->SetBranchAddress("ch37", &ch37, &b_ch37);
fChain->SetBranchAddress("ch38", &ch38, &b_ch38);
fChain->SetBranchAddress("ch39", &ch39, &b_ch39);
fChain->SetBranchAddress("ch40", &ch40, &b_ch40);
fChain->SetBranchAddress("ch41", &ch41, &b_ch41);
fChain->SetBranchAddress("ch42", &ch42, &b_ch42);
fChain->SetBranchAddress("ch43", &ch43, &b_ch43);
fChain->SetBranchAddress("ch44", &ch44, &b_ch44);
fChain->SetBranchAddress("ch45", &ch45, &b_ch45);
fChain->SetBranchAddress("ch46", &ch46, &b_ch46);
fChain->SetBranchAddress("ch47", &ch47, &b_ch47);
fChain->SetBranchAddress("ch48", &ch48, &b_ch48);
fChain->SetBranchAddress("ch49", &ch49, &b_ch49);
fChain->SetBranchAddress("ch50", &ch50, &b_ch50);
fChain->SetBranchAddress("ch51", &ch51, &b_ch51);
fChain->SetBranchAddress("ch52", &ch52, &b_ch52);
fChain->SetBranchAddress("ch53", &ch53, &b_ch53);
fChain->SetBranchAddress("ch54", &ch54, &b_ch54);
fChain->SetBranchAddress("ch55", &ch55, &b_ch55);
fChain->SetBranchAddress("ch56", &ch56, &b_ch56);
fChain->SetBranchAddress("ch57", &ch57, &b_ch57);
fChain->SetBranchAddress("ch58", &ch58, &b_ch58);
fChain->SetBranchAddress("ch59", &ch59, &b_ch59);
fChain->SetBranchAddress("ch60", &ch60, &b_ch60);
fChain->SetBranchAddress("ch61", &ch61, &b_ch61);
fChain->SetBranchAddress("ch62", &ch62, &b_ch62);
fChain->SetBranchAddress("ch63", &ch63, &b_ch63);
fChain->SetBranchAddress("ic1", &ic1, &b_ic1);
fChain->SetBranchAddress("ic2", &ic2, &b_ic2);
fChain->SetBranchAddress("mw1_focusx", &mw1_focusx, &b_mw1_focusx);
fChain->SetBranchAddress("mw1_focusy", &mw1_focusy, &b_mw1_focusy);
fChain->SetBranchAddress("mw2_focusx", &mw2_focusx, &b_mw2_focusx);
fChain->SetBranchAddress("mw2_focusy", &mw2_focusy, &b_mw2_focusy);
fChain->SetBranchAddress("mw1_posx", &mw1_posx, &b_mw1_posx);
fChain->SetBranchAddress("mw1_posy", &mw1_posy, &b_mw1_posy);
fChain->SetBranchAddress("mw2_posx", &mw2_posx, &b_mw2_posx);
fChain->SetBranchAddress("mw2_posy", &mw2_posy, &b_mw2_posy);
// v->ResizeTo(10);
Notify();
}
Bool_t analyse2::Notify()
{
// The Notify() function is called when a new file is opened. This
// can be either for a new TTree in a TChain or when when a new TTree
// is started when using PROOF. It is normally not necessary to make changes
// to the generated code, but the routine can be extended by the
// user if needed. The return value is currently not used.
return kTRUE;
}
void analyse2::Show(Long64_t entry)
{
// Print contents of entry.
// If entry is not specified, print current entry
if (!fChain) return;
fChain->Show(entry);
}
Int_t analyse2::Cut(Long64_t entry)
{
// This function may be called from Loop.
// returns 1 if entry is accepted.
// returns -1 otherwise.
return 1;
}
void analyse2::Initialize(const char *dirname, const char *filename, int baselineEvents, int prelimEvents, double beamLevel, bool firstFibermat, double readOutFrequency, int integrationTime)
{
TString fn = TString(filename);
TString dn = TString(dirname);
channelWidth = 0.8;
TString inName = dn + "pin/" + fn;
fileIn = new TFile(inName, "READ");
fileIn->GetObject("t", fChain);
fChain->Print();
Init(fChain);
TString outname = filename;
outname.Insert(0, "SAVE_");
outname.Insert(0, dirname);
fileOut = new TFile(outname.Data(), "RECREATE");
newdata = new TTree("newdata", "newdata");
for (int n = 0; n < 64; n++)
{
baseline[n] = 0.;
channellist[n] = double(n);
errorx[n] = 0.0;
errory[n] = 7.0;
char bname[20];
sprintf(bname, "ch%02d", n);
b_channels[n] = fChain->GetBranch(bname);
b_channels[n]->SetAddress(&channels[n]);
}
b_eventid_1 = newdata->Branch("eventid_1", &eventid_1);
b_beamNumX_1 = newdata->Branch("beamNumX_1", &beamNumX_1);
b_beamPosX_1 = newdata->Branch("beamPosX_1", &beamPosX_1);
b_beamFocusX_1 = newdata->Branch("beamFocusX_1", &beamFocusX_1);
b_beamSignal_1 = newdata->Branch("beamSignal_1", &beamSignal_1);
b_beamSidebandNoise_1 = newdata->Branch("beamSidebandNoise_1", &beamSidebandNoise_1);
b_beamSkewX_1 = newdata->Branch("beamSkewX_1", &beamSkewX_1);
b_beamKurtX_1 = newdata->Branch("beamKurtX_1", &beamKurtX_1);
b_beamPosX_fit = newdata->Branch("beamPosX_fit", &beamPosX_fit);
b_beamFocusX_fit = newdata->Branch("beamFocusX_fit", &beamFocusX_fit);
b_beamChi2_fit = newdata->Branch("beamChi2_fit", &beamChi2_fit);
b_beamPeakX_fit = newdata->Branch("beamPeakX_fit", &beamPeakX_fit);
b_beamPosX_fit2 = newdata->Branch("beamPosX_fit2", &beamPosX_fit2);
b_beamFocusX_fit2 = newdata->Branch("beamFocusX_fit2", &beamFocusX_fit2);
b_beamFocus2X_fit2 = newdata->Branch("beamFocus2X_fit2", &beamFocus2X_fit2);
b_beamChi2_fit2 = newdata->Branch("beamChi2_fit2", &beamChi2_fit2);
b_beamPeakX_fit2 = newdata->Branch("beamPeakX_fit2", &beamPeakX_fit2);
b_beamPeak2X_fit2 = newdata->Branch("beamPeak2X_fit2", &beamPeak2X_fit2);
b_ic1_1 = newdata->Branch("ic1_1", &ic1_1);
b_ic2_1 = newdata->Branch("ic2_1", &ic2_1);
b_mw1_focusx_1 = newdata->Branch("mw1_focusx_1", &mw1_focusx_1);
b_mw1_posx_1 = newdata->Branch("mw1_posx_1", &mw1_posx_1);
b_mw2_focusx_1 = newdata->Branch("mw2_focusx_1", &mw2_focusx_1);
b_mw2_posx_1 = newdata->Branch("mw2_posx_1", &mw2_posx_1);
b_mw1_focusy_1 = newdata->Branch("mw1_focusy_1", &mw1_focusy_1);
b_mw1_posy_1 = newdata->Branch("mw1_posy_1", &mw1_posy_1);
b_mw2_focusy_1 = newdata->Branch("mw2_focusy_1", &mw2_focusy_1);
b_mw2_posy_1 = newdata->Branch("mw2_posy_1", &mw2_posy_1);
b_rollingavg = newdata->Branch("rollingavg", &rollingavg);
b_beamon = newdata->Branch("beamon", &beamon);
b_time_1 = newdata->Branch("time_1", &time_1);
//histograms;
th2d_mw1_beamPosX = new TH2D("th2d_mw1_beamPosX","th2d_mw1_beamPosX",1000,0,64,1000,0,60);
th2d_ic1_beamSignal = new TH2D("th2d_ic1_beamSignal","th2d_ic1_beamSignal",500,0,500,500,0,100000);
th1d_beamSignal_ic1ratio = new TH1D("th1d_beamSignal_ic1ratio","th1d_beamSignal_ic1ratio",500,0,500.);
th2d_beamSignal_channel = new TH2D("th2d_beamSignal_channel","th2d_beamSignal_channel",65,-0.5,64.5,5100,-1000,10000);
th2d_fitdiff_channel = new TH2D("th2d_fitdiff_channel","th2d_fitdiff_channel",65,-0.5,64.5,5100,-10000,1000);
th2d_fit2diff_channel = new TH2D("th2d_fit2diff_channel","th2d_fit2diff_channel",65,-0.5,64.5,5100,-10000,1000);
th2d_bkg_channel = new TH2D("th2d_bkg_channel","th2d_bkg_channel",65,-0.5,64.5,500,-500,500);
th1d_baseline = new TH1D("th1d_baseline","th1d_baseline",500,-1500,1500.);
th1d_baselinesub = new TH1D("th1d_baselinesub","th1d_baselinesub",500,-1500,1500.);
th1d_chbaseline = new TH1D("th1d_chbaseline","th1d_chbaseline",2100,-100,1000.);
th1d_chbaselinesub = new TH1D("th1d_chbaselinesub","th1d_chbaselinesub",200,-100,100.);
th2d_AmpFFT_channel = new TH2D("th2d_AmpFFT_channel","th2d_AmpFFT_channel",65,-0.5,64.5,500,0,1000);
th2d_PhaseFFT_channel = new TH2D("th2d_PhaseFFT_channel","th2d_PhaseFFT_channel",65,-0.5,64.5,100,-2,2);
}
void analyse2::Baseline(bool useFirstEvents, const int nBaselineEvents, int startevent, bool run50hz)
{
Long64_t nevents = fChain->GetEntriesFast();
baseline_integral = 0.;
// use the first/last nBaselineEvents events to calculate the average/baseline
cout << "Start background subtraction event:" << startevent << endl;
const int numbaseline = 3500;
Double_t channelamp[numbaseline];
Double_t basetime[numbaseline] = {0.};
TGraph * g1;
TF1 * fitfunc = new TF1("fitfunc","[0]+[1]*TMath::Cos(x*[2]*(2*3.14159)+[3])");
fitfunc->SetParameters( 0.0, 10.0, 0.05, 0.0 );
fitfunc->FixParameter(2,0.05);
fChain->GetEvent(1);
Double_t basetimeoffsetzero = time;
fChain->GetEvent(startevent+numbaseline);
Double_t basetimeoffset = time-basetimeoffsetzero;
fChain->GetEvent(startevent);
Double_t basetimeoffsetstart = time-basetimeoffsetzero;
Int_t counter = 0;
char bname[20];
for (int ch = 0; ch < 64; ch++)
{
baseline[ch] =0.;
}
if (!useFirstEvents)
{ // last
for (int i = nevents- numbaseline; i < nevents; i++)
{
fChain->GetEvent(i);
for (int ch = 0; ch < 64; ch++)
{
if (channels[ch]>-100) baseline[ch] += channels[ch] / double(numbaseline);
if (channels[ch]>-100) baseline_integral +=baseline[ch];
th1d_chbaseline->Fill(channels[ch]);
}
}
// cout << "baseline int: " << baseline_integral << endl;
}
else
{
for (int i = startevent ; i <= startevent+ numbaseline; i++)
{
fChain->GetEvent(i);
for (int ch = 0; ch < 64; ch++)
{
if (channels[ch]>-100) baseline[ch] += channels[ch] / double(numbaseline);
if (channels[ch]>-100) baseline_integral +=baseline[ch];
th1d_chbaseline->Fill(channels[ch]);
}
// cout << "baseline int: " << baseline_integral << endl;
}
}
if (run50hz){
///fit a 50Hz background noise to each channel
for (int ch = 0; ch < 64; ch++)
{
counter = 0;
for (int i = startevent ; i < startevent+numbaseline; i++)
{
fChain->GetEvent(i);
// for (int ch = 0; ch < 64; ch++)
//{
channelamp[counter]= channels[ch]- baseline[ch];
//}
basetime[counter] = time-basetimeoffsetzero;
counter++;
}
g1 = new TGraph(numbaseline,basetime,channelamp);
///////finished here. 1) need to fit the baseline to a Cosine function 2) return 3 parameters per channel. 50 Hz is fixed.
bkgfitResult = g1->Fit(fitfunc,"QSR","",basetimeoffsetstart,basetimeoffset );
bkgfitPar0[ch] = bkgfitResult->Parameter(0);
bkgfitPar1[ch] = bkgfitResult->Parameter(1);
bkgfitPar2[ch] = bkgfitResult->Parameter(2);
bkgfitPar3[ch] = bkgfitResult->Parameter(3);
sprintf(bname, "bkg_ch%02d_graph", ch);
g1->SetTitle(bname);
g1->SetName(bname);
//fitResult->Print("V");
// bkgfitResult->Write();
g1->Write();
} //end for (int ch = ...
for (int ch = 0; ch < 64; ch++)
{
counter = 0;
fitfunc->SetParameters( bkgfitPar0[ch], bkgfitPar1[ch], bkgfitPar2[ch], bkgfitPar3[ch]);
for (Long64_t i = startevent ; i <startevent+ numbaseline; i++)
{
fChain->GetEvent(i);
channelamp[counter] = channels[ch]- baseline[ch]- fitfunc->Eval(time-basetimeoffsetzero);
th1d_chbaselinesub->Fill(channelamp[counter]);
counter++;
th2d_bkg_channel->Fill(ch,channelamp[counter]);
}
}
Double_t channelampsum[numbaseline] = {0.};
counter = 0;
///fit a 50Hz background noise to the sum of all channels
for (Long64_t i = startevent ; i <startevent+ numbaseline; i++)
{
fChain->GetEvent(i);
channelampsum[counter]=0.;
for (int ch = 0; ch < 64; ch++){
if (channels[ch]>-100 && baseline[ch]>-100) channelampsum[counter] += channels[ch] - baseline[ch]; //integrate the signal
}
basetime[counter] = (time-basetimeoffsetzero);
counter++;
}
TGraph * g2 = new TGraph(counter,basetime,channelampsum);
TF1 * fitfunc2 = new TF1("fitfunc2","[0]+[1]*TMath::Cos(x*[2]*(2*3.14159)+[3])");
fitfunc2->SetParameters( 5120.0, 450.0, 0.05, 0.0 );
// fitfunc2->SetParLimits(1,0.,1000.);
fitfunc2->FixParameter(2,0.05);
TFitResultPtr sumbkgfitResult = g2->Fit(fitfunc2,"QRS","", basetimeoffsetstart, basetimeoffset);
g2->SetName("commonmodenoise"); g2->SetTitle("commonmodenoise");
g2->Write();
sumbkgfitResult->Write();
fitfunc2->Write();
bkgfitPar0[64] = fitfunc2->GetParameter(0);
bkgfitPar1[64] = fitfunc2->GetParameter(1);
bkgfitPar2[64] = fitfunc2->GetParameter(2);
bkgfitPar3[64] = fitfunc2->GetParameter(3);
counter=0;
for (Long64_t i = startevent ; i <startevent+ numbaseline; i++)
{
fChain->GetEvent(i);
channelampsum[counter] -= fitfunc2->Eval(time-basetimeoffsetzero);
th1d_baselinesub->Fill(channelampsum[counter]);
counter++;
}
th1d_baseline->Write();
th1d_baselinesub->Write();
th1d_chbaseline->Write();
th1d_chbaselinesub->Write();
TGraph * g3 = new TGraph(counter,basetime,channelampsum);
g3->SetName("commonmodenoisesub"); g2->SetTitle("commonmodenoisesub");
g3->Write();
}
}
void analyse2::Save()
{
//fileIn->Close();
//fileOut->cd();
newdata->AutoSave();
}
void analyse2::Close()
{
fileIn->Close();
fileOut->Close();
if(fChain) delete fChain->GetCurrentFile();
}
#define REAL(z,i) ((z)[2*(i)])
#define IMAG(z,i) ((z)[2*(i)+1])
double * analyse2::FFTsmoothing(double arr[])
{
int i; double data[2*64];
double channel[64];
double RealInput[64];
double ImagInput[64];
double RealFFT[64];
double ImagFFT[64];
double AmpFFT[64];
double PhaseFFT[64];
double RealOutput[64];
double ImagOutput[64];
double avgnoiseAmpFFT = 0.;
for (i = 0; i < 64; i++)
{
REAL(data,i) = arr[i]; IMAG(data,i) = 0.0;
}
// REAL(data,0) = 1.0;
// for (i = 1; i <= 10; i++)
// {
// REAL(data,i) 1.0;
// }
for (i = 0; i < 64; i++)
{
channel[i] = i;
RealInput[i] = REAL(data,i);
ImagInput[i] = IMAG(data,i);
// printf ("%d %e %e\n", i, REAL(data,i), IMAG(data,i));
}
// printf ("\n");
// TGraph * g1_input_real = new TGraph(64,channel,RealInput);
// TGraph * g1_input_imag = new TGraph(64,channel,ImagInput);
// g1_input_real->SetName("input_real"); g1_input_real->Write();
//g1_input_imag->SetName("input_imag"); g1_input_imag->Write();
//apply Fourier transform
gsl_fft_complex_radix2_forward (data, 1, 64);
for (i = 0; i < 64; i++)
{
// printf ("%d %e %e\n", i, REAL(data,i)/sqrt(64), IMAG(data,i)/sqrt(64));
RealFFT[i] = REAL(data,i)/sqrt(64);
ImagFFT[i] = IMAG(data,i)/sqrt(64);
AmpFFT[i] = sqrt(RealFFT[i]*RealFFT[i] + ImagFFT[i]*ImagFFT[i]);
PhaseFFT[i] = atan(ImagFFT[i]/RealFFT[i]);
if (beamon){
th2d_AmpFFT_channel->Fill(i,AmpFFT[i]);
th2d_PhaseFFT_channel->Fill(i,PhaseFFT[i]);
}
}
// printf ("\n");
//find the average noise frequency amplitude
for (i = 22; i < 32; i++){
avgnoiseAmpFFT +=AmpFFT[i]/10;
}
//find where the signal frequency amplitude falls below noise;
int cutchannel = 31;
for (i = 0; i < 32; i++)
{
if (AmpFFT[i]<avgnoiseAmpFFT) {
cutchannel = i;
break;
}
}
// TGraph * g1_FFT_real = new TGraph(64,channel,RealFFT);
// TGraph * g1_FFT_imag = new TGraph(64,channel,ImagFFT);
//g1_FFT_real->SetName("FFT_real"); g1_FFT_real->Write();
// g1_FFT_imag->SetName("FFT_imag");g1_FFT_imag->Write();
// TGraph * g1_FFT_amp = new TGraph(64,channel,AmpFFT);
// TGraph * g1_FFT_phase = new TGraph(64,channel,PhaseFFT);
// g1_FFT_amp->SetName("FFT_amp"); g1_FFT_amp->Write();
// g1_FFT_phase->SetName("FFT_phase");g1_FFT_phase->Write();
//suppress channels that are more likely noise; the signal has a lower frequency than noise;
for (i = cutchannel+2; i < 32; i++)
{
REAL(data,i) = 0.;
IMAG(data,i) = 0.;
REAL(data,64-i) = 0.;
IMAG(data,64-i) = 0.;
}
//apply inverse Fourier transform
gsl_fft_complex_radix2_inverse (data, 1, 64);
for (i = 0; i < 64; i++)
{
// printf ("%d %e %e\n", i, REAL(data,i), IMAG(data,i));
RealOutput[i] = REAL(data,i);
ImagOutput[i] = IMAG(data,i);
arr[i] = RealOutput[i];
}
//printf ("\n");
// TGraph * g1_output_real = new TGraph(64,channel,RealOutput);
// TGraph * g1_output_imag = new TGraph(64,channel,ImagOutput);
// g1_output_real->SetName("output_real");g1_output_real->Write();
// g1_output_imag->SetName("output_imag");g1_output_imag->Write();
return arr;
}
#endif // #ifdef analyse2_cxx