HarpoPedMgr.cxx

Go to the documentation of this file.

//
//
//
 
#include <iostream>
#include "TBranch.h"
#include "HarpoPedMgr.h"
#include "HarpoPedsRaw.h"
 
// Constructor
HarpoPedMgr::HarpoPedMgr()
{
  fn = NULL;
  f = 0; 
}
  
// Distructor
HarpoPedMgr::~HarpoPedMgr()
{
  // we will not distroy pointer to pedistal hire
  // will be distroed in Run Header
  //  if ( f != 0 ) f->Close();
}

// Open Femimos Pedistal File
void HarpoPedMgr::SetFile(TString * file)
{
  std::cout << "HarpoPedMgr::SetFile " << *file << std::endl;
  fn = file->Data(); 
}

// Serach pedestal and means for given plane
HarpoPedestal * HarpoPedMgr::Lookup(Long_t run, Long_t detno)
{
  p = new HarpoPedestal();  // Default values

  if (f == 0) { // Not Opened ?
    f = TFile::Open(fn);
  }

  if (f == 0) return p;  // Open errorr, no ped file

  HarpoPedsRaw *praw = new HarpoPedsRaw();
  std::cout << "Lookup " << run << " plane " << detno << " in file " << fn << " (" << f << ")" << std::endl; 
  //  f->ls();

  TTree * t = (TTree *) f->Get("PedsRaw");
  std::cout << "TTree at " << t << std::endl;
  t->ResetBranchAddresses();

  TBranch *brunno = t->GetBranch("nRun");
  TBranch *bpraw = t->GetBranch("PedsRaw");
  bpraw->SetAddress(&praw);
  
  Int_t nevent = t->GetEntries();
  std::cout << "N entries "<< nevent << std::endl; 

  for (Int_t i=0;i<nevent;i++) {
    brunno->GetEvent(i);

    if (praw->nRun != run) continue;
    //std::cout << "Run found " << std::endl; 
   
    t->GetEvent(i);
    if ( praw->Plane != detno ) continue;
    //std::cout << "plane found " << std::endl; 
    // Found
    p->SetPeds(praw->Mean,praw->Sigma);
    break;
  }
  delete f;
  f=0;
  delete praw;
  return p;
}


// Write on run pedestals to file
void HarpoPedMgr::WriteRunPedestals(Long_t run, HarpoPedestal * ptr,
                         Int_t plane, Int_t card)
{

  const char *pedfile = "pedsraw.root";
  HarpoPedsRaw s;
  s.Clear();
  s.nRun = run;
  
  s.Plane = plane;
  s.Pass = 1;
  //??  s.CardNo = 1;
  s.CardNo = card;
  s.Pass = 1;
  for (UInt_t i=0; i < ptr->GetPedLen(); i++) {
    s.Mean[i] = ptr->GetPed(i);
    s.Sigma[i] = ptr->GetSigma(i);
  }
  //Make a tree
  TFile f1(pedfile,"RECREATE");
  TTree *t = new TTree("PedsRaw","Harpo Raw Pedestal Tree");
  // Create branches in the tree
  t->Branch("PedsRaw",&s);

  t->Fill();
  //Close root file
  f1.Write();
  f1.Close();
}

// EOF HarpoPedMgr.cxx