DataBlock.cxx

Go to the documentation of this file.

// $Id: DataBlock.cxx,v 1.17 2009/10/15 18:30:13 jyuko Exp $

// C/C++
#include <cmath>
#include <cassert>
#include <iomanip>
#include <iostream>
#include <sstream>

// ROOT
#include "TFile.h"

// Local
#include "PhysicsNtuple/Record.h"
#include "PhysicsNtuple/DataBlock.h"

ClassImp(DataBlock)

using namespace std;

//-------------------------------------------------------------------
DataBlock::DataBlock()
   :nSnarlIn(0),
    nEventIn(0),
    nSnarlOut(0),
    nEventOut(0),
    ProtonsIn(0.0),
    ProtonsOut(0.0)
{
}

//-------------------------------------------------------------------
DataBlock::~DataBlock()
{
}

//-------------------------------------------------------------------
void DataBlock::Add(const DataBlock &block)
{
   nSnarlIn   += block.nSnarlIn;
   nEventIn   += block.nEventIn;
   nSnarlOut  += block.nSnarlOut;
   nEventOut  += block.nEventOut;
   ProtonsIn  += block.ProtonsIn;
   ProtonsOut += block.ProtonsOut;
}

//-------------------------------------------------------------------
void DataBlock::SetInput(const DataBlock &block)
{
   nSnarlIn  = block.nSnarlOut;
   nEventIn  = block.nEventOut;
   ProtonsIn = block.ProtonsOut;
}

//-------------------------------------------------------------------
void DataBlock::SetProtons(double protons)
{
   ProtonsOut = protons;
}

//-------------------------------------------------------------------
void DataBlock::ScaleOut(const double factor)
{
   if(factor > 0.0)
   {
      nSnarlOut = static_cast<UInt_t>(factor*nSnarlOut);
      nEventOut = static_cast<UInt_t>(factor*nEventOut);
      ProtonsOut *= factor; 
   }
   else
   {
      cerr << "DataBlock::ScaleOut - scale factor "<< factor << " is not positive" << endl;
   }
}
 
//-------------------------------------------------------------------
void DataBlock::Add(const Anp::Record &record)
{
   ++nSnarlOut;

   nEventOut += record.GetNEvents();

   if     (record.GetHeader().TorTgt() > 0.01) ProtonsOut += record.GetHeader().TorTgt();
   else if(record.GetHeader().TrTgtD() > 0.01) ProtonsOut += record.GetHeader().TrTgtD();
   else if(record.GetHeader().Tr101D() > 0.01) ProtonsOut += record.GetHeader().Tr101D();
   else if(record.GetHeader().Tor101() > 0.01) ProtonsOut += record.GetHeader().Tor101();
}

//-------------------------------------------------------------------
Double_t DataBlock::GetProtonsPerSnarl() const
{
   if(nSnarlOut > 0 && ProtonsOut > 0.0)
   {
      return ProtonsOut/double(nSnarlOut);
   }
   
   return 0.0;
}

//-------------------------------------------------------------------
Float_t DataBlock::Ratio(const Float_t nom, const Float_t den) const
{
   if(den > 0.0) return nom/den;

   return 0.0;
}

//-------------------------------------------------------------------
Double_t DataBlock::Ratio(const Double_t nom, const Double_t den) const
{
   if(den > 0.0) return nom/den;

   return 0.0;
}

//-------------------------------------------------------------------
Float_t DataBlock::Ratio(const UInt_t nom, const UInt_t den) const
{
   if(den > 0) return float(nom)/float(den);

   return 0.0;
}

//-------------------------------------------------------------------
void DataBlock::print(ostream& o) const
{   
   stringstream os;

   os << "Printing DataBlock" << endl
      << "   nSnarl: " << nSnarlOut << " / " << nSnarlIn << " = " << Ratio(nSnarlOut, nSnarlIn) << endl
      << "   nEvent: " << nEventOut << " / " << nEventIn << " = " << Ratio(nEventOut, nEventIn) << endl
      << "   Protons: " << scientific << setprecision(5) << ProtonsOut << " / " 
      << scientific << setprecision(5) << ProtonsIn << " = " << Ratio(ProtonsOut, ProtonsIn) << endl
      << "   Protons per record = " << fixed << setprecision(5) << GetProtonsPerSnarl() << endl;
   
   o << os.str();
}

//-------------------------------------------------------------------
ostream& operator<<(ostream& o, const DataBlock& self)
{
   self.print(o);
   return o;
}

//-------------------------------------------------------------------
const DataBlock FindDataBlock(const std::string &path, const std::string &name)
{
   TFile *file_ptr = new TFile(path.c_str(), "READ");
   if(!file_ptr || !file_ptr -> IsOpen())
   {
      cerr << "Failed to open root file " << path << endl;
      return DataBlock();
   }
   
   DataBlock *block_ptr = dynamic_cast<DataBlock *>(file_ptr -> Get(name.c_str()));
   if(!block_ptr)
   {
      cout << "Failed to find DataBlock" << endl;
      return DataBlock();
   }

   const DataBlock block = *block_ptr;

   file_ptr -> Close();
   
   return block;
}

---