AtNuDataNtuple.cxx

Go to the documentation of this file.

#include "AtNuDataNtuple.h"
#include "AtNuRunTimes.h"

#include "AtNuEvent/AtmosEvent.h"

#include <iostream>

ClassImp(AtNuDataNtuple)

AtNuDataNtuple::AtNuDataNtuple() :
  fChain(0),
  fEvent(0)
{
  this->Reset();
}

AtNuDataNtuple::~AtNuDataNtuple()
{

}

void AtNuDataNtuple::Reset()
{
  if( !fChain ){
    fChain = new TChain("ntp","chain");
  }

  fChain->Reset();
  fChain->SetBranchAddress("evt",&fEvent);
 
  fAtNuDataFile = 0;
  fAtNuDataTree = 0;

  fFileName = "ntpcheck.root";

  return;
}

void AtNuDataNtuple::AddFile(const char* file)
{
  std::cout << " adding files from: " << file << std::endl;

  fChain->Add(file);

  std::cout << "   ... total entries=" << fChain->GetEntries() << std::endl;

  return;
}

void AtNuDataNtuple::SetFileName(const char* file)
{
  std::cout << " *** AtNuDataNtuple::SetFileName(...) *** " << std::endl;

  fFileName = file;

  std::cout << "   output file = " << fFileName.Data() << std::endl;
}

void AtNuDataNtuple::OpenFile()
{
  if( !fAtNuDataFile ){
    std::cout << " opening file: " << fFileName.Data() << std::endl;
    TDirectory* tmpd = gDirectory;
    fAtNuDataFile = new TFile(fFileName.Data(),"recreate");
    fAtNuDataTree = new TTree("ntuple","minos monitor ntuple");
    fAtNuDataTree->SetDirectory(fAtNuDataFile);

    fAtNuDataTree->Branch("run",&fRun,"run/I");
    fAtNuDataTree->Branch("subrun",&fSubRun,"subrun/I");
    fAtNuDataTree->Branch("unixtime",&fUnixTime,"unixtime/I");
    fAtNuDataTree->Branch("date",&fDate,"date/I");
    fAtNuDataTree->Branch("time",&fTime,"time/I");
    fAtNuDataTree->Branch("rate",&fRate,"rate/I");
    fAtNuDataTree->Branch("good",&fGood,"good/I");

    fAtNuDataTree->SetAutoSave(100);
    gDirectory = tmpd;
  }
}

void AtNuDataNtuple::CloseFile()
{
  if( fAtNuDataFile ){
    std::cout << " closing file: " << fAtNuDataFile->GetName() << std::endl;
    TDirectory* tmpd = gDirectory;
    fAtNuDataFile->cd();
    fAtNuDataTree->Write();
    fAtNuDataFile->Close();
    gDirectory = tmpd;
  }
}

void AtNuDataNtuple::WriteToFile()
{
  if( fAtNuDataFile ){
    TDirectory* tmpd = gDirectory;
    fAtNuDataFile->cd();
    fAtNuDataTree->Fill();
    gDirectory = tmpd;
  }
}

void AtNuDataNtuple::Run()
{
  std::cout << " *** AtNuDataNtuple::Run() *** " << std::endl;

  // open file
  this->OpenFile();

  // loop over events
  fUnixTime = -1;
  fRun = -1;
  fSubRun = -1;
  fDate = -1;
  fTime = -1;
  fRate = 0;
  fGood = 0;

  for( Int_t i=0; i<fChain->GetEntries(); i++ ){
    fChain->GetEntry(i);

    if( !(fEvent->Run==fRun 
       && fEvent->SubRun==fSubRun) ){
      std::cout << "  Run: " << fEvent->Run << std::endl;
    }

    if( fEvent->UnixTime!=fUnixTime ){

      if( fUnixTime>=0 ){
        this->WriteToFile();
      }

      fUnixTime = fEvent->UnixTime;
      fRun = fEvent->Run;
      fSubRun = fEvent->SubRun;
      fDate = fEvent->Date;
      fTime = fEvent->Time;
      fRate = 0;
      fGood = AtNuRunTimes::GoodData(fUnixTime);
    }

    fRate++;
  }

  // write last event
  if( fUnixTime>=0 ){
    this->WriteToFile();
  }

  // close file
  this->CloseFile();

  return;
}

---