/[PAMELA software]/PamCut/Collections/SmartCollection/SmartCollection.cpp
ViewVC logotype

Diff of /PamCut/Collections/SmartCollection/SmartCollection.cpp

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.1 by pam-fi, Wed May 27 13:30:09 2009 UTC revision 1.3 by pam-fi, Fri Jun 5 13:14:35 2009 UTC
# Line 11  Line 11 
11    
12  void SmartCollection::AddAction(CollectionAction& action) {  void SmartCollection::AddAction(CollectionAction& action) {
13    _actions.push_back(&action);    _actions.push_back(&action);
14      _actionsPositions.push_back(GetSize() - 1);
15  }  }
16    
17  CollectionAction *SmartCollection::GetAction(unsigned int iAction) {  CollectionAction *SmartCollection::GetAction(unsigned int iAction) {
# Line 22  CollectionAction *SmartCollection::GetAc Line 23  CollectionAction *SmartCollection::GetAc
23      return _actions[iAction];      return _actions[iAction];
24  }  }
25    
26  void SmartCollection::Setup(PamLevel2 *events){  void SmartCollection::Setup(PamLevel2 *events) {
27    PamCutCollection::Setup(events);    PamCutCollection::Setup(events);
28    
29    for (unsigned int i = 0; i < GetSize(); i++){    for (unsigned int i = 0; i < _actions.size(); i++) {
30      _actions[i]->Setup(events);      _actions[i]->Setup(events);
31    }    }
32  }  }
33    
34  void SmartCollection::Finalize(){  void SmartCollection::Finalize() {
35    PamCutCollection::Finalize();    PamCutCollection::Finalize();
36    
37    for (unsigned int i = 0; i < GetSize(); i++){    for (unsigned int i = 0; i < _actions.size(); i++) {
38      _actions[i]->Finalize();      _actions[i]->Finalize();
39    }    }
40  }  }
41    
42  void SmartCollection::OnGood(PamLevel2 *event){  int SmartCollection::ApplyCut(PamLevel2 *event) {
   PamCutCollection::OnGood(event);  
43    
44    for (unsigned int i = 0; i < GetSize(); i++){    _nEv++;
45      _actions[i]->OnGood(event);    // Apply the cuts
46      if (_cuts.size() == 0) {
47        _nGood++;
48        OnGood(event);
49        return CUTOK;
50    }    }
 }  
51    
52  void SmartCollection::OnBad(PamLevel2 *event, int selectionResult){    unsigned int bunchEnd = -1;
53    PamCutCollection::OnBad(event, selectionResult);    unsigned int nActions = 0;
54      unsigned int firstActionOfBunch = -1;
55      if (_actions.size() > 0) {
56        // Set the bunch end to the position of the first action
57        bunchEnd = _actionsPositions[0];
58        // Set the position of the first action at the end of the bunch
59        firstActionOfBunch = 0;
60        // Set the number of actions at the end of this bunch of cuts
61        nActions = 1;
62        while (_actionsPositions[nActions] == bunchEnd)
63          nActions++;
64      }
65    
66    for (unsigned int i = 0; i < GetSize(); i++){    for (unsigned int icut = 0; icut < _cuts.size(); icut++) {
67      _actions[i]->OnBad(event, selectionResult);      if (_cuts[icut]->ApplyCut(event) != CUTOK) {
68          // Check if there are actions at the end of the current bunch
69          // (last bunch may have no actions at its end)
70          if (bunchEnd >= icut) {
71            for (unsigned int i = 0; i < nActions; i++) {
72              _actions[firstActionOfBunch + i]->OnBad(event, icut);
73            }
74          }
75          OnBad(event, icut);
76          return icut;
77        }
78        else if (bunchEnd == icut) {
79          // We are at the end of the bunch of cuts, so let's execute the actions
80          for (unsigned int i = 0; i < nActions; i++) {
81            _actions[firstActionOfBunch + i]->OnGood(event);
82          }
83          if (firstActionOfBunch + nActions < _actions.size()) {
84            // Current bunch is done, but actions are not finished yet. Set up the next bunch.
85            // Set the end of the new bunch and the number of actions at this end
86            firstActionOfBunch += nActions;
87            bunchEnd = _actionsPositions[firstActionOfBunch];
88            nActions = 1;
89            while (_actionsPositions[firstActionOfBunch + nActions] == bunchEnd)
90              nActions++;
91          }
92        }
93    }    }
94    
95      _nGood++;
96      OnGood(event);
97      return CUTOK;
98  }  }
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.3
  ViewVC Help
Powered by ViewVC 1.1.23