CollectionActions/SaveEventsAction/SaveEventsAction.cpp

/*
 * SaveEventsAction.cpp
 *
 *  Created on: 16-mag-2009
 *      Author: Nicola Mori
 */

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

#include "SaveEventsAction.h"

SaveEventsAction::SaveEventsAction(const char *actionName, TString outFileName, TString outOptions,
    Long64_t maxFileSize) :
  CollectionAction(actionName), _outTreeFile(NULL), _events(NULL), _outOptions(outOptions), _outFileName(outFileName),
      _maxFileSize(maxFileSize) {

}

void SaveEventsAction::Setup(PamLevel2 *events) {

  // Open output file
  /* Due to the combined weirdness of PamLevel2 and ROOT, this pointer must NOT
   * be explicitly deleted, nor the file closed. This would generate a double
   * deletion in the destructors chain. This mechanism is not completely clear
   * but it indeed happens...
   */

  _outTreeFile = new TFile(_outFileName, "RECREATE");

  if (_outTreeFile->IsZombie()) {
    cout << "Output file could not be created\n";
    _outTreeFile->Delete();
    //TODO Gestire con un'eccezione
    return;
  }

  _events = events;
  if (_outOptions.Length() > 0)
    _events->SetWhichTrees(_outOptions);
  _events->CreateCloneTrees(_outTreeFile);

  // Set the output file size
  TTree * currTree = NULL;
  if ((currTree = _events->GetCloneTree("Run")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("SelectionList")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("Tracker")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("Calorimeter")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("ToF")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("Trigger")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("S4")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("NeutronD")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("Anticounter")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("OrbitalInfo")))
    currTree->SetMaxTreeSize(_maxFileSize);
  currTree = NULL;
  if ((currTree = _events->GetCloneTree("h20")))
    currTree->SetMaxTreeSize(_maxFileSize);

}

void SaveEventsAction::OnGood(PamLevel2 *event) {
  // Fill the tree
  event->FillCloneTrees();
}

void SaveEventsAction::Finalize() {
  _outTreeFile->cd();
  _events->WriteCloneTrees();

}
1	/*
2	* SaveEventsAction.cpp
3	*
4	* Created on: 16-mag-2009
5	* Author: Nicola Mori
6	*/
7
8	/! @file SaveEventsAction.cpp The SaveEventsAction class implementation file /
9
10	#include "SaveEventsAction.h"
11
12	SaveEventsAction::SaveEventsAction(const char *actionName, TString outFileName, TString outOptions,
13	Long64_t maxFileSize) :
14	CollectionAction(actionName), _outTreeFile(NULL), _events(NULL), _outOptions(outOptions), _outFileName(outFileName),
15	_maxFileSize(maxFileSize) {
16
17	}
18
19	void SaveEventsAction::Setup(PamLevel2 *events) {
20
21	// Open output file
22	/* Due to the combined weirdness of PamLevel2 and ROOT, this pointer must NOT
23	* be explicitly deleted, nor the file closed. This would generate a double
24	* deletion in the destructors chain. This mechanism is not completely clear
25	* but it indeed happens...
26	*/
27
28	_outTreeFile = new TFile(_outFileName, "RECREATE");
29
30	if (_outTreeFile->IsZombie()) {
31	cout << "Output file could not be created\n";
32	_outTreeFile->Delete();
33	//TODO Gestire con un'eccezione
34	return;
35	}
36
37	_events = events;
38	if (_outOptions.Length() > 0)
39	_events->SetWhichTrees(_outOptions);
40	_events->CreateCloneTrees(_outTreeFile);
41
42	// Set the output file size
43	TTree * currTree = NULL;
44	if ((currTree = _events->GetCloneTree("Run")))
45	currTree->SetMaxTreeSize(_maxFileSize);
46	currTree = NULL;
47	if ((currTree = _events->GetCloneTree("SelectionList")))
48	currTree->SetMaxTreeSize(_maxFileSize);
49	currTree = NULL;
50	if ((currTree = _events->GetCloneTree("Tracker")))
51	currTree->SetMaxTreeSize(_maxFileSize);
52	currTree = NULL;
53	if ((currTree = _events->GetCloneTree("Calorimeter")))
54	currTree->SetMaxTreeSize(_maxFileSize);
55	currTree = NULL;
56	if ((currTree = _events->GetCloneTree("ToF")))
57	currTree->SetMaxTreeSize(_maxFileSize);
58	currTree = NULL;
59	if ((currTree = _events->GetCloneTree("Trigger")))
60	currTree->SetMaxTreeSize(_maxFileSize);
61	currTree = NULL;
62	if ((currTree = _events->GetCloneTree("S4")))
63	currTree->SetMaxTreeSize(_maxFileSize);
64	currTree = NULL;
65	if ((currTree = _events->GetCloneTree("NeutronD")))
66	currTree->SetMaxTreeSize(_maxFileSize);
67	currTree = NULL;
68	if ((currTree = _events->GetCloneTree("Anticounter")))
69	currTree->SetMaxTreeSize(_maxFileSize);
70	currTree = NULL;
71	if ((currTree = _events->GetCloneTree("OrbitalInfo")))
72	currTree->SetMaxTreeSize(_maxFileSize);
73	currTree = NULL;
74	if ((currTree = _events->GetCloneTree("h20")))
75	currTree->SetMaxTreeSize(_maxFileSize);
76
77	}
78
79	void SaveEventsAction::OnGood(PamLevel2 *event) {
80	// Fill the tree
81	event->FillCloneTrees();
82	}
83
84	void SaveEventsAction::Finalize() {
85	_outTreeFile->cd();
86	_events->WriteCloneTrees();
87
88	}