PhysicsNtuple/kNNAlg/Event.h

Go to the documentation of this file.

#ifndef LIT_EVENT_H
#define LIT_EVENT_H

// $Id: Event.h,v 1.5 2009/02/17 19:15:40 rustem Exp $
//
// Internal knn class that stores classification variables
//

// C/C++
#include <cassert>
#include <iostream>
#include <vector>

namespace Lit
{
   typedef float VarType;
   typedef std::vector<VarType> VarVec;

   class Event
   {
   public:

      Event();
      Event(const VarVec &vvec, double weight, short type);
      Event(const VarVec &vvec, double weight, short type, const VarVec &tvec);
      ~Event();
      
      double GetWeight() const;

      VarType GetVar(unsigned int i) const;
      VarType GetTgt(unsigned int i) const;

      unsigned int GetNVar() const;
      unsigned int GetNTgt() const;

      const VarVec& GetVarVec() const;
      const VarVec& GetTgtVec() const;

      short GetType() const;
      
      void Print(std::ostream& os = std::cout, unsigned int w = 5, unsigned int p = 3) const;

   private:

      VarVec fVar;     // classificaion variables vector
      VarVec fTgt;     // target variables vector
      double fWeight;  // event weight
      short  fType;    // event type: signal, background, etc
   };

   //
   // Global functions used by k-nearest neighbor algorithm
   //
   inline VarType Distance(const VarType var1, const VarType var2)
   {
      return (var1 - var2) * (var1 - var2);
   }

   VarType Distance(const Event &event1, const Event &event2);

   std::ostream& operator<<(std::ostream& os, const Event& event);
   

   //
   // Inlined member functions
   //
   inline double Event::GetWeight() const
   {
      return fWeight;
   }

   inline unsigned int Event::GetNVar() const { return fVar.size(); }
   inline unsigned int Event::GetNTgt() const { return fTgt.size(); }
      
   inline const VarVec& Event::GetVarVec() const { return fVar; }
   inline const VarVec&Event:: GetTgtVec() const { return fTgt; }

   inline short Event::GetType() const
   {
      return fType;
   }
}

#endif

---