Collections/EffRigCollection/EffRigCollection.cpp

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

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

#include "EffRigCollection.h"

EffRigCollection::EffRigCollection(const char *collectionName, TString outFileBase, TString rigBinsFile, bool absRig) :
  EffCollection(collectionName, outFileBase), _bins(0), _selVector(0), _detVector(0), _outUp(0), _outDown(0) {

  ifstream rigBinListFile;
  rigBinListFile.open(rigBinsFile);

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

  _selVector.resize(_bins.size() - 1, 0);
  _detVector.resize(_bins.size() - 1, 0);
}

int EffRigCollection::ApplyCut(PamLevel2 *event) {

  _nEv++;
  if (_selCollection.ApplyCut(event) == CUTOK) {
    // Check if the event is inside the rigidity range
    // NOTE: at this point a TrkPhSinCut should be already performed,
    //       since we are going to retrieve rigidity.
    float rig;
    if (_absRig) {
      rig = event->GetTrack(0)->GetTrkTrack()->GetRigidity();
    }
    else
      rig = 1. / event->GetTrack(0)->GetTrkTrack()->GetDeflection();
    if (rig >= _bins[0]) {
      int i = 1;
      while (rig >= _bins[i] && i < (int) _bins.size()) {
        i++;
      }
      i--;

      if (i < (int) (_selVector.size())) {
        _selVector[i]++;
        if (_detCollection.ApplyCut(event) == CUTOK) {
          _detVector[i]++;
          _nGood++;
          return CUTOK;
        }
      }
      else
        _outUp++;

    }
    else {
      _outDown++;
      return 0;
    }
  }

  return 0;
}

void EffRigCollection::Finalize() {

  // Correct the number of selected events by subtracting events outside the rigidity interval
  _sel -= _outUp + _outDown;
  // Print the report
  EffCollection::Finalize();
  cout << "    Events below the minimum rigidity: " << _outDown << "\n";
  cout << "    Events above the maximum rigidity: " << _outUp << "\n";

  // Write the output files
  if (_outFileBase != "") {
    ofstream outTextFile((_outFileBase + "-eff-rig.txt").Data(), ios_base::out);
    for (unsigned int i = 0; i < _selVector.size(); i++) {
      outTextFile << _detVector[i] << "    " << _selVector[i] << "    ";
      if (_selVector[i] != 0)
        outTextFile << (float) _detVector[i] / (float) _selVector[i] << "\n";
      else
        outTextFile << "0.\n";

    }
    outTextFile.close();
  }

}
1	/*
2	* EffRigCollection.cpp
3	*
4	* Created on: 10/ago/2009
5	* Author: Nicola Mori
6	*/
7
8	/! @file EffRigCollection.cpp The EffRigCollection class implementation file. /
9
10	#include "EffRigCollection.h"
11
12	EffRigCollection::EffRigCollection(const char *collectionName, TString outFileBase, TString rigBinsFile, bool absRig) :
13	EffCollection(collectionName, outFileBase), _bins(0), _selVector(0), _detVector(0), _outUp(0), _outDown(0) {
14
15	ifstream rigBinListFile;
16	rigBinListFile.open(rigBinsFile);
17
18	TString auxString;
19	while (!rigBinListFile.eof()) {
20	rigBinListFile >> auxString;
21	if (auxString != "") {
22	_bins.push_back(auxString.Atof());
23	}
24	}
25	rigBinListFile.close();
26
27	_selVector.resize(_bins.size() - 1, 0);
28	_detVector.resize(_bins.size() - 1, 0);
29	}
30
31	int EffRigCollection::ApplyCut(PamLevel2 *event) {
32
33	_nEv++;
34	if (_selCollection.ApplyCut(event) == CUTOK) {
35	// Check if the event is inside the rigidity range
36	// NOTE: at this point a TrkPhSinCut should be already performed,
37	// since we are going to retrieve rigidity.
38	float rig;
39	if (_absRig) {
40	rig = event->GetTrack(0)->GetTrkTrack()->GetRigidity();
41	}
42	else
43	rig = 1. / event->GetTrack(0)->GetTrkTrack()->GetDeflection();
44	if (rig >= _bins[0]) {
45	int i = 1;
46	while (rig >= _bins[i] && i < (int) _bins.size()) {
47	i++;
48	}
49	i--;
50
51	if (i < (int) (_selVector.size())) {
52	_selVector[i]++;
53	if (_detCollection.ApplyCut(event) == CUTOK) {
54	_detVector[i]++;
55	_nGood++;
56	return CUTOK;
57	}
58	}
59	else
60	_outUp++;
61
62	}
63	else {
64	_outDown++;
65	return 0;
66	}
67	}
68
69	return 0;
70	}
71
72	void EffRigCollection::Finalize() {
73
74	// Correct the number of selected events by subtracting events outside the rigidity interval
75	_sel -= _outUp + _outDown;
76	// Print the report
77	EffCollection::Finalize();
78	cout << " Events below the minimum rigidity: " << _outDown << "\n";
79	cout << " Events above the maximum rigidity: " << _outUp << "\n";
80
81	// Write the output files
82	if (_outFileBase != "") {
83	ofstream outTextFile((_outFileBase + "-eff-rig.txt").Data(), ios_base::out);
84	for (unsigned int i = 0; i < _selVector.size(); i++) {
85	outTextFile << _detVector[i] << " " << _selVector[i] << " ";
86	if (_selVector[i] != 0)
87	outTextFile << (float) _detVector[i] / (float) _selVector[i] << "\n";
88	else
89	outTextFile << "0.\n";
90
91	}
92	outTextFile.close();
93	}
94
95	}