HarpoAnalyse.cxx

Go to the documentation of this file.

// 
// File HarpoAnalyse.cxx
//
#include "HarpoAnalyse.h"
#include "TSystem.h"
#include "TKey.h"
#include "TFile.h"
#include <cstdlib>
#include <cstring>
#include <iostream>

ClassImp(HarpoAnalyse)

HarpoHistManager* HarpoHistManager::fManager = 0;


HarpoHistManager::HarpoHistManager()
{
  fHistFile = NULL;
}

HarpoHistManager* HarpoHistManager::GetPointer()
{
  if(!fManager) {
    static HarpoHistManager manager;
    fManager = &manager;
  }
  return fManager;
}



TFile* HarpoHistManager::OpenHistFile(const char* ananame, HarpoRunHeader* rh)
{
  Info("OpenHistFile","%s, %p",ananame,(void*)rh);
  //  TFile *fHistFile = NULL;
  if(fHistFile==NULL){
    Bool_t isZombie = kTRUE;
    TString * hstFile = gHConfig->GetHistFile();
    if ( hstFile != NULL ){
      fHistFile = new TFile(hstFile->Data(),"UPDATE");
      Info("OpenHistFile","Opened %s",fHistFile->GetName());
    }
    isZombie = fHistFile->IsZombie();
  
    std::cout << "fHistFile = " << fHistFile << std::endl;
    if(fHistFile == NULL || isZombie){
      std::cout << gHConfig->GetProgramName() << " " <<
        "No Hist File name given, use default" << std::endl;
      const char* dir = Form("$HARPO_ANA_DIR/%s",ananame);
      
      if(gSystem->AccessPathName(dir)) // If directory not available, write on local dir
        hstFile = new TString(Form("%s%lli.root",ananame,gHConfig->GetRunNo()));
      else
        hstFile = new TString(Form("%s/%s%lli.root",dir,ananame,gHConfig->GetRunNo()));
      
      fHistFile = new TFile(hstFile->Data(),"RECREATE");
      Info("OpenHistFile","Opened %s",fHistFile->GetName());
      if(rh != NULL)
        rh->Write("fRunHeader");
    }
  
    if(rh != NULL){
      Info("OpenHistFile","RunHeader: %p",(void*)rh);
      fHistFile->Delete("fRunHeader;*");
      fHistFile->Delete("fDetHeader*;*");
      Info("OpenHistFile","RunHeader: %p",(void*)rh);
      for (UInt_t ndet = 0; ndet < gkNDetectors; ndet++){
        if (gHDetSet->isExist(ndet)){
          HarpoDetHeader* hdr = rh->GetHeader(ndet);
          std::cout << "Save Header " << ndet << " " << hdr << std::endl;
          if(hdr) hdr->Write(Form("fDetHeader%d",ndet));
        }
      }
      std::cout << "Save RunHeader " << rh->GetRun() << std::endl;
      Info("OpenHistFile","RunHeader->GetRun() = %d",rh->GetRun());
      rh->Write("fRunHeader");
      if(gHConfig->GetRunNo()==0)
        gHConfig->SetRunNo(rh->GetRun());
    }
  }


  const char* dirname1 = Form("run%lli",gHConfig->GetRunNo());
  const char* dirname2 = Form("%s/%s",dirname1,ananame);

  if(fHistFile->GetDirectory(dirname2) == NULL){
    Info("OpenHistFile","make dir %s",dirname2);
    fHistFile->mkdir(dirname2);
  }else{
    TDirectory* dir = fHistFile->GetDirectory(dirname2);
    dir->Delete("*;*");
  }
  fHistFile->cd(dirname2);

  return fHistFile;
}


void HarpoHistManager::CloseHistFile()
{ 
  if(fHistFile){
    Info("CloseHistFile","Closing %s",fHistFile->GetName());
    fHistFile->Write();
    fHistFile->Close();
  }
}








void   HarpoAnalyse::Set(HarpoEvent *event)
{
  fEvt = event;
}

void   HarpoAnalyse::print()
{
}

 void   HarpoAnalyse::process()
 {
   nEvents++;  
 }

void   HarpoAnalyse::ProcessEvent(HarpoEvent *event)
{
  if ( event->GetHeader()->GetEvAnaStatus() < hAnaOK )
    return;

  Set(event);
  process();
  //  delete fEvt;
  //  fEvt->Delete();
}

TFile* HarpoAnalyse::OpenHistFile(const char* ananame)
{
  Info("HarpoAnalyse::OpenHistFile","RunHeader->GetrRun() = %d",fRunHeader->GetRun());
  return (HarpoHistManager::GetPointer())->OpenHistFile(ananame,fRunHeader);
}