Collections/EffCollection/EffCollection.cpp

/*
 * EffCollection.cpp
 *
 *  Created on: 10/ago/2009
 *      Author: Nicola Mori
 */

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

#include "EffCollection.h"
#include "TGraphAsymmErrors.h"

extern "C" {
bool efficiency_(Int_t*, Int_t*, Double_t*, Double_t*, Double_t*);
}

EffCollection::EffCollection(const char *collectionName, TString outFileBase, int errMethod) :
  VerboseCollection(collectionName), _selCollection("selCollection"), _detCollection("detCollection"), _outFileBase(
      outFileBase), _errMethod(errMethod), _det(0), _sel(0) {

}

void EffCollection::AddDetectorCut(PamCut &cut) {
  _detCollection.AddCut(cut);
}

void EffCollection::AddSelectionCut(PamCut &cut) {
  _selCollection.AddCut(cut);
}

void EffCollection::AddDetectorAction(CollectionAction &action) {
  _detCollection.AddAction(action);
}

void EffCollection::AddSelectionAction(CollectionAction &action) {
  _selCollection.AddAction(action);
}

int EffCollection::ApplyCut(PamLevel2 *event) {

  _nEv++;
  if (_selCollection.ApplyCut(event) == CUTOK) {
    _sel++;
    if (_detCollection.ApplyCut(event) == CUTOK) {
      _det++;
      _nGood++;
      return CUTOK;
    }
  }

  return 0;
}

void EffCollection::Finalize() {
  // Let's add all the cuts to the vector of the collection before calling VerboseCollection::Finalize
  for (unsigned int i = 0; i < _selCollection.GetSize(); i++)
    _cuts.push_back(_selCollection.GetCut(i));
  for (unsigned int i = 0; i < _detCollection.GetSize(); i++)
    _cuts.push_back(_detCollection.GetCut(i));
  // Now all the cuts are in place, and VerboseCollection can print its report
  VerboseCollection::Finalize();

  // Compute the error
  Int_t sel = (Int_t) _sel;
  Int_t det = (Int_t) _det;
  Double_t eff, errLow, errHigh;
  if (_errMethod == EFFRIG_ROOT) {
    if (sel < 8) {
      eff = 1.1;
      errLow = errHigh = 0.;
    }
    else {
      TH1I pass("pass", "pass", 1, 0., 1.);
      TH1I total("total", "total", 1, 0., 1.);
      pass.Fill(0.5, det);
      total.Fill(0.5, sel);
      TGraphAsymmErrors errGraph;
      errGraph.BayesDivide(&pass, &total);
      Double_t dummy;
      errGraph.GetPoint(0, dummy, eff);
      errLow = errGraph.GetErrorYlow(0);
      errHigh = errGraph.GetErrorYhigh(0);
    }
  }
  if (_errMethod == EFFRIG_SERGIO) {
    efficiency_(&sel, &det, &eff, &errLow, &errHigh);
  }

  // Add some info about efficiency to the stdout
  cout << "    ****** Efficiency informations ******\n";
  cout << "    Detector cuts:\n";
  for (unsigned int i = 0; i < _detCollection.GetSize(); i++)
    cout << "      - " << _detCollection.GetCut(i)->GetName() << "\n";
  cout << "    Total detector efficiency: " << eff << " +" << errHigh << "-" << errLow << endl;

  // Write the output file
  if (_outFileBase != "") {

    ofstream outTextFile((_outFileBase + "-eff.txt").Data(), ios_base::out);
    streamsize newPrec = 4;
    outTextFile.precision(newPrec);
    outTextFile.setf(ios::fixed, ios::floatfield);
    outTextFile << setw(10) << det << setw(10) << sel << setw(10) << eff << setw(10) << errLow << setw(10) << errHigh
        << endl;
    outTextFile.close();
  }

}
1	pam-fi	1.1	/*
2			* EffCollection.cpp
3			*
4			* Created on: 10/ago/2009
5			* Author: Nicola Mori
6			*/
7
8			/! @file EffCollection.cpp The EffCollection class implementation file. /
9
10			#include "EffCollection.h"
11	pam-fi	1.6	#include "TGraphAsymmErrors.h"
12	pam-fi	1.1
13	pam-fi	1.5	extern "C" {
14			bool efficiency_(Int_t, Int_t, Double_t, Double_t, Double_t*);
15			}
16
17	pam-fi	1.6	EffCollection::EffCollection(const char *collectionName, TString outFileBase, int errMethod) :
18	pam-fi	1.1	VerboseCollection(collectionName), _selCollection("selCollection"), _detCollection("detCollection"), _outFileBase(
19	pam-fi	1.6	outFileBase), _errMethod(errMethod), _det(0), _sel(0) {
20	pam-fi	1.1
21			}
22
23			void EffCollection::AddDetectorCut(PamCut &cut) {
24			_detCollection.AddCut(cut);
25			}
26
27			void EffCollection::AddSelectionCut(PamCut &cut) {
28			_selCollection.AddCut(cut);
29			}
30
31			void EffCollection::AddDetectorAction(CollectionAction &action) {
32			_detCollection.AddAction(action);
33			}
34
35			void EffCollection::AddSelectionAction(CollectionAction &action) {
36			_selCollection.AddAction(action);
37			}
38
39			int EffCollection::ApplyCut(PamLevel2 *event) {
40
41	pam-fi	1.3	_nEv++;
42			if (_selCollection.ApplyCut(event) == CUTOK) {
43			_sel++;
44			if (_detCollection.ApplyCut(event) == CUTOK) {
45			_det++;
46			_nGood++;
47			return CUTOK;
48	pam-fi	1.1	}
49			}
50
51			return 0;
52			}
53
54			void EffCollection::Finalize() {
55			// Let's add all the cuts to the vector of the collection before calling VerboseCollection::Finalize
56			for (unsigned int i = 0; i < _selCollection.GetSize(); i++)
57			_cuts.push_back(_selCollection.GetCut(i));
58			for (unsigned int i = 0; i < _detCollection.GetSize(); i++)
59			_cuts.push_back(_detCollection.GetCut(i));
60			// Now all the cuts are in place, and VerboseCollection can print its report
61			VerboseCollection::Finalize();
62	pam-fi	1.3
63	pam-fi	1.5	// Compute the error
64			Int_t sel = (Int_t) _sel;
65			Int_t det = (Int_t) _det;
66			Double_t eff, errLow, errHigh;
67	pam-fi	1.6	if (_errMethod == EFFRIG_ROOT) {
68			if (sel < 8) {
69			eff = 1.1;
70			errLow = errHigh = 0.;
71			}
72			else {
73			TH1I pass("pass", "pass", 1, 0., 1.);
74			TH1I total("total", "total", 1, 0., 1.);
75			pass.Fill(0.5, det);
76			total.Fill(0.5, sel);
77			TGraphAsymmErrors errGraph;
78			errGraph.BayesDivide(&pass, &total);
79			Double_t dummy;
80			errGraph.GetPoint(0, dummy, eff);
81			errLow = errGraph.GetErrorYlow(0);
82			errHigh = errGraph.GetErrorYhigh(0);
83			}
84			}
85			if (_errMethod == EFFRIG_SERGIO) {
86			efficiency_(&sel, &det, &eff, &errLow, &errHigh);
87			}
88	pam-fi	1.5
89			// Add some info about efficiency to the stdout
90	pam-fi	1.2	cout << " **** Efficiency informations ****\n";
91			cout << " Detector cuts:\n";
92			for (unsigned int i = 0; i < _detCollection.GetSize(); i++)
93			cout << " - " << _detCollection.GetCut(i)->GetName() << "\n";
94	pam-fi	1.5	cout << " Total detector efficiency: " << eff << " +" << errHigh << "-" << errLow << endl;
95	pam-fi	1.1
96	pam-fi	1.3	// Write the output file
97	pam-fi	1.1	if (_outFileBase != "") {
98	pam-fi	1.5
99	pam-fi	1.3	ofstream outTextFile((_outFileBase + "-eff.txt").Data(), ios_base::out);
100	pam-fi	1.5	streamsize newPrec = 4;
101			outTextFile.precision(newPrec);
102			outTextFile.setf(ios::fixed, ios::floatfield);
103			outTextFile << setw(10) << det << setw(10) << sel << setw(10) << eff << setw(10) << errLow << setw(10) << errHigh
104			<< endl;
105	pam-fi	1.1	outTextFile.close();
106			}
107
108			}