/*
 * 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) :
  CollectionAction(actionName), _outTreeFile(NULL), _events(NULL), _outOptions(outOptions) {

  // 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;
  }
}

void SaveEventsAction::Setup(PamLevel2 *events){
  _events = events;
  _events->CreateCloneTrees(_outTreeFile);
  if (_outOptions.Length() > 0)
    _events->SetWhichTrees(_outOptions);
}

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

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

}