TreeOut.cxx

Go to the documentation of this file.

// $Id: TreeOut.cxx,v 1.2 2006/07/19 22:41:29 gmieg Exp $
//
// Author: B. Speakman 2005.05.23
//
#include <fstream>
#include <iostream>
#include <sstream>
#include <string>
#include <cassert>

#include "TCanvas.h"
#include "TDirectory.h"

#include "MessageService/MsgService.h"

#include "AtNuEvent/AtmosEvent.h"//Whole TObject

#include "AtNuUtils/TreeOut.h"

ClassImp(TreeOut)
CVSID("$Id: TreeOut.cxx,v 1.2 2006/07/19 22:41:29 gmieg Exp $");

TreeOut::TreeOut(std::string fname) {
  MSG("TreeOut",Msg::kVerbose) << "TreeOut::TreeOut" << endl;

  TDirectory *tmpd = gDirectory;
  fFile = new TFile(fname.c_str(), "RECREATE");
  fTree = new TTree("ntp", "FarDet Atmos Events");
  fTree->SetDirectory(fFile);
  fEvent = 0;
  fTree->Branch("evt", "AtmosEvent", &fEvent);
  tmpd->cd();
}

TreeOut::~TreeOut() {
  MSG("TreeOut",Msg::kVerbose) << "TreeOut::~TreeOut" << endl;
}

void TreeOut::MakeHistos() {
  MSG("TreeOut",Msg::kVerbose) << "TreeOut::MakeHistos" << endl;
}

void TreeOut::MakeCanvases() {
  MSG("TreeOut",Msg::kVerbose) << "TreeOut::MakeCanvases" << endl;
}

bool TreeOut::FillEvent(const AtmosEvent *event) {
  MSG("TreeOut",Msg::kVerbose) << "TreeOut::FillEvent" << endl;

  if (!event) {
    MSG("TreeOut",Msg::kWarning)<< "No Event in TreeOut::FillEvent" << endl;
    return false;
  }

  //fEvent = new AtmosEvent(*event);
  fEvent = event;
  if(!fEvent) return false;//wtf

  TDirectory* tmpd = gDirectory;
  fFile->cd();
  fTree->Fill();
  gDirectory = tmpd;

  return(true);
}

void TreeOut::Reset() {
  MSG("TreeOut",Msg::kVerbose) << "TreeOut::Reset" << endl;
}

void TreeOut::Finish() {
  MSG("TreeOut",Msg::kVerbose) << "TreeOut::Finish" << endl;
  TDirectory* tmpd = gDirectory;
  fFile->cd();
  fTree->SetDirectory(fFile);
  fTree->Write();
  fFile->Write();
  fFile->Close();
  gDirectory = tmpd;
}

---