PhysicsNtuple/kNNAlg/Event.cxx

Go to the documentation of this file.

// $Id: Event.cxx,v 1.5 2009/02/17 19:15:40 rustem Exp $

// C++
#include <iomanip>

// Local
#include "Event.h"

//-------------------------------------------------------------------------------------------
Lit::Event::Event() 
   :fVar(),
    fTgt(),
    fWeight(-1.0),
    fType(-1)
{
}

//-------------------------------------------------------------------------------------------
Lit::Event::Event(const VarVec &vvec,
                  const double weight,
                  const short type)
   :fVar(vvec),
    fTgt(),
    fWeight(weight),
    fType(type)
{
}

//-------------------------------------------------------------------------------------------
Lit::Event::Event(const VarVec &vvec,
                  const double weight,
                  const short type, 
                  const VarVec &tvec)
   :fVar(vvec),
    fTgt(tvec),
    fWeight(weight),
    fType(type)
{
}

//-------------------------------------------------------------------------------------------
Lit::Event::~Event()
{
}

//-------------------------------------------------------------------------------------------
Lit::VarType Lit::Event::GetVar(const unsigned int i) const
{
   assert(i < fVar.size() && "index is out of range");
   return fVar[i];
}

//-------------------------------------------------------------------------------------------
Lit::VarType Lit::Event::GetTgt(const unsigned int i) const
{
   assert(i < fTgt.size() && "index is out of range");
   return fTgt[i];
}

//-------------------------------------------------------------------------------------------
Lit::VarType Lit::Distance(const Event &event1, const Event &event2)
{
   const unsigned int size = event1.GetNVar();

   if(size != event2.GetNVar())
   {
      std::cerr << "Distance: two events have different dimensions" << std::endl;
      return -1.0;
   }

   VarType sum = 0.0;
   for(unsigned int ivar = 0; ivar != size; ++ivar)
   {
      sum += Distance(event1.GetVar(ivar), event2.GetVar(ivar));
   }
   
   return sum;
}

//-------------------------------------------------------------------------------------------
void Lit::Event::Print(std::ostream& os, const unsigned int w, const unsigned int p) const
{
   os << "Event  (weight, type) = ("
      << std::setw(w) << std::setprecision(p) << fWeight << ", "
      << fType << ") ";

   for(unsigned int ivar = 0; ivar != GetNVar(); ++ivar)
   {
      if(ivar == 0)
      {
         os << "var = (";
      }
      else
      {
         os << ", ";
      }

      os << std::setfill(' ') << std::setw(5) << std::setprecision(3) << GetVar(ivar);
   }

   if(GetNVar() > 0)
   {
      os << ")";
   }
   else
   {
      os << " no variables";
   }

   for(unsigned int itgt = 0; itgt != GetNTgt(); ++itgt)
   {
      if(itgt == 0)
      {
         os << "tgt = (";
      }
      else
      {
         os << ", ";
      }

      os << std::setfill(' ') << std::setw(5) << std::setprecision(3) << GetTgt(itgt);
   }

   if(GetNTgt() > 0)
   {
      os << ")";
   }
   else
   {
      os << " no targets";
   }
}

//-------------------------------------------------------------------------------------------
std::ostream& Lit::operator<<(std::ostream& os, const Lit::Event& event)
{ 
   event.Print(os);
   return os;
}

---