CollectionActions/FluxHistoAction/FluxHistoAction.cpp

/*
 * FluxHistoAction.cpp
 *
 *  Created on: 20-mag-2009
 *      Author: Nicola Mori
 */

/*! @file FluxHistoAction.cpp The FluxHistoAction class implementation file */

#include "FluxHistoAction.h"

FluxHistoAction::FluxHistoAction(const char *actionName, TString outFileBase, TString rigBinsFile, TString mode) :
  CollectionAction(actionName), _outFileBase(outFileBase), _bins(0), _rootHisto(), _textHisto(0), _mode(mode)
#ifdef DEBUGPAMCUT
, _outUp(0), _outDown(0)
#endif
{

  // Reading the bins from file
  ifstream rigBinListFile;
  rigBinListFile.open(rigBinsFile);

  TString auxString;
  while (!rigBinListFile.eof()) {
    rigBinListFile >> auxString;
    if (auxString != "") {
      _bins.push_back(auxString.Atof());
    }
  }
  rigBinListFile.close();

  // Initializing histograms
  _textHisto.assign(_bins.size() - 1, 0);
  _rootHisto.SetName("rigHisto");
  _rootHisto.SetTitle("Rigidity histogram");
  _rootHisto.GetXaxis()->SetTitle("R");
  _rootHisto.GetYaxis()->SetTitle("Events");

  Double_t *auxArray = new Double_t[_bins.size()];

  for (unsigned int i = 1; i < _bins.size(); i++) {
    auxArray[i] = _bins[i];
  }

  _rootHisto.SetBins(_bins.size() - 1, auxArray);

  delete[] auxArray;
}

void FluxHistoAction::OnGood(PamLevel2 *event) {

  float rig = 1. / event->GetTrack(0)->GetTrkTrack()->GetDeflection();
  if (rig >= _bins[0]) {
    int i = 1;
    int binningSize = _bins.size();
    bool found = false;
    while (!found && i < binningSize) {
      if (rig < _bins[i])
        found = true;
      else
        i++;
    }
    if (found) {
      _textHisto[i - 1]++;
      _rootHisto.Fill(rig);
    }
#ifdef DEBUGPAMCUT
    else
    _outUp++;
#endif
  }
#ifdef DEBUGPAMCUT
  else
  _outDown++;
#endif

}

void FluxHistoAction::Finalize() {

  // Write the text file
  // Currently, text output doesn't support append.
  ofstream outTextFile((_outFileBase + ".txt").Data(), ios_base::out);
  streamsize oldPrec = cout.precision();
  streamsize newPrec = 4;
  outTextFile.precision(newPrec);
  outTextFile.setf(ios::fixed, ios::floatfield);
  for (unsigned int i = 0; i < _textHisto.size(); i++)
    outTextFile << _bins[i] << "  " << _bins[i + 1] << "  " << _textHisto[i] << "\n";
  outTextFile.close();
  outTextFile.precision(oldPrec);
  outTextFile << resetiosflags(ios::floatfield);

  // Write the ROOT file
  TFile outRootFile((_outFileBase + ".root"), _mode);
  outRootFile.cd();
  _rootHisto.Write();
  //outRootFile.Close();

#ifdef DEBUGPAMCUT
  cout << "Debug informations from " << GetName() << ":\n";
  cout << "   Events below the lowest rigidity: " << _outDown << "\n";
  cout << "   Events above the highest rigidity: " << _outUp << endl;
#endif
}
1	pam-fi	1.1	/*
2			* FluxHistoAction.cpp
3			*
4			* Created on: 20-mag-2009
5			* Author: Nicola Mori
6			*/
7
8			/! @file FluxHistoAction.cpp The FluxHistoAction class implementation file /
9
10			#include "FluxHistoAction.h"
11
12	pam-fi	1.2	FluxHistoAction::FluxHistoAction(const char *actionName, TString outFileBase, TString rigBinsFile, TString mode) :
13			CollectionAction(actionName), _outFileBase(outFileBase), _bins(0), _rootHisto(), _textHisto(0), _mode(mode)
14	pam-fi	1.1	#ifdef DEBUGPAMCUT
15	pam-fi	1.2	, _outUp(0), _outDown(0)
16	pam-fi	1.1	#endif
17			{
18
19			// Reading the bins from file
20			ifstream rigBinListFile;
21			rigBinListFile.open(rigBinsFile);
22
23			TString auxString;
24			while (!rigBinListFile.eof()) {
25			rigBinListFile >> auxString;
26			if (auxString != "") {
27			_bins.push_back(auxString.Atof());
28			}
29			}
30			rigBinListFile.close();
31
32			// Initializing histograms
33			_textHisto.assign(_bins.size() - 1, 0);
34			_rootHisto.SetName("rigHisto");
35			_rootHisto.SetTitle("Rigidity histogram");
36			_rootHisto.GetXaxis()->SetTitle("R");
37			_rootHisto.GetYaxis()->SetTitle("Events");
38
39			Double_t *auxArray = new Double_t[_bins.size()];
40
41			for (unsigned int i = 1; i < _bins.size(); i++) {
42			auxArray[i] = _bins[i];
43			}
44
45			_rootHisto.SetBins(_bins.size() - 1, auxArray);
46
47			delete[] auxArray;
48			}
49
50			void FluxHistoAction::OnGood(PamLevel2 *event) {
51
52			float rig = 1. / event->GetTrack(0)->GetTrkTrack()->GetDeflection();
53			if (rig >= _bins[0]) {
54			int i = 1;
55			int binningSize = _bins.size();
56			bool found = false;
57			while (!found && i < binningSize) {
58			if (rig < _bins[i])
59			found = true;
60			else
61			i++;
62			}
63			if (found) {
64			_textHisto[i - 1]++;
65			_rootHisto.Fill(rig);
66			}
67			#ifdef DEBUGPAMCUT
68			else
69	pam-fi	1.2	_outUp++;
70	pam-fi	1.1	#endif
71			}
72			#ifdef DEBUGPAMCUT
73			else
74	pam-fi	1.2	_outDown++;
75	pam-fi	1.1	#endif
76
77			}
78
79			void FluxHistoAction::Finalize() {
80
81			// Write the text file
82	pam-fi	1.2	// Currently, text output doesn't support append.
83	pam-fi	1.1	ofstream outTextFile((_outFileBase + ".txt").Data(), ios_base::out);
84			streamsize oldPrec = cout.precision();
85			streamsize newPrec = 4;
86			outTextFile.precision(newPrec);
87			outTextFile.setf(ios::fixed, ios::floatfield);
88			for (unsigned int i = 0; i < _textHisto.size(); i++)
89			outTextFile << _bins[i] << " " << _bins[i + 1] << " " << _textHisto[i] << "\n";
90			outTextFile.close();
91			outTextFile.precision(oldPrec);
92			outTextFile << resetiosflags(ios::floatfield);
93
94			// Write the ROOT file
95	pam-fi	1.2	TFile outRootFile((_outFileBase + ".root"), _mode);
96	pam-fi	1.1	outRootFile.cd();
97			_rootHisto.Write();
98			//outRootFile.Close();
99
100			#ifdef DEBUGPAMCUT
101			cout << "Debug informations from " << GetName() << ":\n";
102			cout << " Events below the lowest rigidity: " << _outDown << "\n";
103			cout << " Events above the highest rigidity: " << _outUp << endl;
104			#endif
105			}