Collections/SmartCollection/SmartCollection.cpp

/*
 * SmartCollection.cpp
 *
 *  Created on: 14-mag-2009
 *      Author: Nicola Mori
 */

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

#include "SmartCollection.h"

SmartCollection::~SmartCollection(){
  if (_owns){
     for (unsigned int i = 0; i < _actions.size(); i++)
       if (_actions[i] != NULL){
         delete _actions[i];
         _actions[i] = NULL;
       }
   }
}

void SmartCollection::AddAction(CollectionAction *action) {
  _actions.push_back(action);
  _actionsPositions.push_back(GetSize() - 1);
}

CollectionAction *SmartCollection::GetAction(unsigned int iAction) {
  if (_actions.size() == 0)
    return NULL;
  if (iAction < 0 || iAction > _actions.size() - 1)
    return NULL;
  else
    return _actions[iAction];
}

CollectionAction *SmartCollection::GetAction(const char *actionName) {
  if (_actions.size() == 0)
    return NULL;
  for (unsigned int i = 0; i < _actions.size(); i++) {
    if (strcmp(_actions[i]->GetName(), actionName) == 0)
      return _actions[i];
  }

  return NULL;
}

void SmartCollection::Setup(PamLevel2 *events) {
  PamCutCollection::Setup(events);
  //cout << GetName() << endl;

  for (unsigned int i = 0; i < _actions.size(); i++) {
    _actions[i]->Setup(events);
  }
}

void SmartCollection::Finalize() {
  PamCutCollection::Finalize();

  for (unsigned int i = 0; i < _actions.size(); i++) {
    _actions[i]->Finalize();
  }
}

int SmartCollection::ApplyCut(PamLevel2 *event) {

  _nEv++;

  // Execute the actions placed before the cuts
  unsigned int iBeforeCuts = 0;
  //cout << GetName() << endl;
  if (_actions.size() > 0) {
    while (_actionsPositions[iBeforeCuts] == -1) {
      _actions[iBeforeCuts]->OnGood(event);
      iBeforeCuts++;
      if (iBeforeCuts == _actions.size())
        break;
    }
  }

  // Apply the cuts
  if (_cuts.size() == 0) {
    _nGood++;
    OnGood(event);
    return CUTOK;
  }

  unsigned int firstFailed = _cuts.size();
  for (unsigned int iCut = 0; iCut < _cuts.size(); iCut++) {
    if (_cuts[iCut]->ApplyCut(event) != CUTOK && firstFailed == _cuts.size()) {
      firstFailed = iCut;
      break;
    }
  }

  //Do actions
  if (_actions.size() > 0) {
    unsigned int lastPosition = _cuts.size();
    for (unsigned int i = iBeforeCuts; i < _actions.size(); i++) {
      if (_actionsPositions[i] < (int) firstFailed)
        _actions[i]->OnGood(event);
      else { // A cut has failed
        if (lastPosition == _cuts.size()) // Record the position of the end of the bunch
          lastPosition = _actionsPositions[i];
        if (_actionsPositions[i] > (int) lastPosition) // Don't do actions at the end of successive bunches
          break;
        _actions[i]->OnBad(event, (int) firstFailed);
      }
    }
  }

  if (firstFailed == _cuts.size()) {
    _nGood++;
    OnGood(event);
    return CUTOK;
  }
  else {
    OnBad(event, firstFailed);
    return firstFailed;
  }
}
1	pam-fi	1.1	/*
2			* SmartCollection.cpp
3			*
4			* Created on: 14-mag-2009
5			* Author: Nicola Mori
6			*/
7
8			/! @file SmartCollection.cpp The SmartCollection class implementation file /
9
10			#include "SmartCollection.h"
11
12	pam-fi	1.6	SmartCollection::~SmartCollection(){
13			if (_owns){
14			for (unsigned int i = 0; i < _actions.size(); i++)
15			if (_actions[i] != NULL){
16			delete _actions[i];
17			_actions[i] = NULL;
18			}
19			}
20			}
21
22			void SmartCollection::AddAction(CollectionAction *action) {
23			_actions.push_back(action);
24	pam-fi	1.3	_actionsPositions.push_back(GetSize() - 1);
25	pam-fi	1.1	}
26
27			CollectionAction *SmartCollection::GetAction(unsigned int iAction) {
28			if (_actions.size() == 0)
29			return NULL;
30			if (iAction < 0 \|\| iAction > _actions.size() - 1)
31			return NULL;
32			else
33			return _actions[iAction];
34			}
35
36	pam-fi	1.5	CollectionAction SmartCollection::GetAction(const char actionName) {
37			if (_actions.size() == 0)
38			return NULL;
39	pam-fi	1.6	for (unsigned int i = 0; i < _actions.size(); i++) {
40	pam-fi	1.5	if (strcmp(_actions[i]->GetName(), actionName) == 0)
41			return _actions[i];
42			}
43
44			return NULL;
45			}
46
47	pam-fi	1.2	void SmartCollection::Setup(PamLevel2 *events) {
48	pam-fi	1.1	PamCutCollection::Setup(events);
49	pam-fi	1.6	//cout << GetName() << endl;
50	pam-fi	1.1
51	pam-fi	1.2	for (unsigned int i = 0; i < _actions.size(); i++) {
52	pam-fi	1.1	_actions[i]->Setup(events);
53			}
54			}
55
56	pam-fi	1.2	void SmartCollection::Finalize() {
57	pam-fi	1.1	PamCutCollection::Finalize();
58
59	pam-fi	1.2	for (unsigned int i = 0; i < _actions.size(); i++) {
60	pam-fi	1.1	_actions[i]->Finalize();
61			}
62			}
63
64	pam-fi	1.3	int SmartCollection::ApplyCut(PamLevel2 *event) {
65	pam-fi	1.1
66	pam-fi	1.3	_nEv++;
67	pam-fi	1.6
68			// Execute the actions placed before the cuts
69			unsigned int iBeforeCuts = 0;
70			//cout << GetName() << endl;
71			if (_actions.size() > 0) {
72			while (_actionsPositions[iBeforeCuts] == -1) {
73			_actions[iBeforeCuts]->OnGood(event);
74			iBeforeCuts++;
75			if (iBeforeCuts == _actions.size())
76			break;
77			}
78			}
79
80	pam-fi	1.3	// Apply the cuts
81			if (_cuts.size() == 0) {
82			_nGood++;
83			OnGood(event);
84			return CUTOK;
85	pam-fi	1.1	}
86
87	pam-fi	1.4	unsigned int firstFailed = _cuts.size();
88	pam-fi	1.6	for (unsigned int iCut = 0; iCut < _cuts.size(); iCut++) {
89			if (_cuts[iCut]->ApplyCut(event) != CUTOK && firstFailed == _cuts.size()) {
90			firstFailed = iCut;
91	pam-fi	1.4	break;
92			}
93	pam-fi	1.3	}
94	pam-fi	1.1
95	pam-fi	1.4	//Do actions
96			if (_actions.size() > 0) {
97			unsigned int lastPosition = _cuts.size();
98	pam-fi	1.6	for (unsigned int i = iBeforeCuts; i < _actions.size(); i++) {
99			if (_actionsPositions[i] < (int) firstFailed)
100	pam-fi	1.4	_actions[i]->OnGood(event);
101			else { // A cut has failed
102			if (lastPosition == _cuts.size()) // Record the position of the end of the bunch
103			lastPosition = _actionsPositions[i];
104	pam-fi	1.6	if (_actionsPositions[i] > (int) lastPosition) // Don't do actions at the end of successive bunches
105	pam-fi	1.4	break;
106	pam-fi	1.6	_actions[i]->OnBad(event, (int) firstFailed);
107	pam-fi	1.3	}
108			}
109	pam-fi	1.1	}
110	pam-fi	1.3
111	pam-fi	1.4	if (firstFailed == _cuts.size()) {
112			_nGood++;
113			OnGood(event);
114			return CUTOK;
115			}
116			else {
117			OnBad(event, firstFailed);
118			return firstFailed;
119			}
120	pam-fi	1.1	}