FillEnergy.cxx

Go to the documentation of this file.

// $Id: FillEnergy.cxx,v 1.1 2008/01/11 19:09:26 rustem Exp $

// MINOS
#include "DataUtil/EnergyCorrections.h"
#include "RecoBase/CandShowerHandle.h"
#include "Registry/Registry.h"
#include "Validity/VldContext.h"

// Local
#include "PhysicsNtuple/Factory.h"

// Local
#include "FillEnergy.h"

REGISTER_ANP_OBJECT(AlgSnarl,FillEnergy)

namespace EnergyCorrections{}
using namespace EnergyCorrections;
using namespace std;

//-------------------------------------------------------------------
Anp::FillEnergy::FillEnergy()
{
}

//----------------------------------------------------------------------
Anp::FillEnergy::~FillEnergy()
{
}

//----------------------------------------------------------------------
bool Anp::FillEnergy::Run(Record& record)
{
   //
   // Make VldContext
   //
   const VldContext vldc = GetVldc(record.GetHeader());

   //
   // Correct shower energy using latest energy correction function
   //
   for(ShowerIterator sit = record.ShowerBegIterator(); sit != record.ShowerEndIterator(); ++sit)
   {
      sit -> SetEnergyCC(FullyCorrectShowerEnergy(sit -> EnergyCC(),
                                                  CandShowerHandle::kCC,
                                                  vldc,
                                                  record.GetHeader().Release()));
   }

   //
   // Apply latest energy corrections to event's hadronic energy
   //
   for(EventIterator ievent = record.EventBegIterator(); ievent != record.EventEndIterator(); ++ievent)
   {
      //
      // Event has no showers
      //
      if(ievent -> GetNShowers() < 1) continue;

      //
      // Find longest track
      //
      TrackIter itrack = LongestTrack(*ievent, record);

      double ehad = 0.0;
      for(ShowerIter ishower = record.ShowerBeg(); ishower != record.ShowerEnd(); ++ishower)
      {
         if(!(ievent -> MatchShower(ishower -> ShowerIndex())))
         {
            continue;
         }

         const float deltaz = ishower -> GetVtx().Z() - ievent -> GetVtx().Z();
            
         //
         // !!!hard coded constants!!!
         //
         if(itrack == record.TrackEnd())
         {
            ehad += ishower -> EnergyCC();
         }
         else if(std::fabs(deltaz) < 0.5 || ishower -> EnergyCC() > 2.0)
         {
            ehad += ishower -> EnergyCC();
         }
      }

      //
      // Set corrected hadronic energy
      //
      RecoNu rnu = ievent -> GetNu();
      rnu.SetEHad(ehad);
      ievent -> SetNu(rnu);
   }

   return true;
}

//----------------------------------------------------------------------
void Anp::FillEnergy::Config(const Registry &)
{
}

//------------------------------------------------------------------------------------------
const VldContext Anp::FillEnergy::GetVldc(const Header &header) const
{
   //
   // Convert header to VldContext, should make it a global utility function...
   //

   const VldTimeStamp time(header.Sec(), header.NSec());
   if(header.IsData())
   {
      if(header.IsNear())
      {
         return VldContext(Detector::kNear, SimFlag::kData, time);
      }
      else if(header.IsFar())
      {
         return VldContext(Detector::kFar, SimFlag::kData, time);
      }
   }
   else
   {
      if(header.IsNear())
      {
         return VldContext(Detector::kNear, SimFlag::kMC, time);
      }
      else if(header.IsFar())
      {
         return VldContext(Detector::kFar, SimFlag::kMC, time);
      }
   }

   return VldContext(Detector::kUnknown, SimFlag::kUnknown, time);
}

---