LikeModule.h

Go to the documentation of this file.

#ifndef LIT_LIKEMODULE_H
#define LIT_LIKEMODULE_H

// C++
#include <iosfwd>
#include <map>

// Local
#include "Node.h"
#include "Result.h"

class TDirectory;

namespace Lit
{
   typedef std::vector<Lit::Event> EventVec;

   class LikeModule
   {
   public:

      typedef std::map<int, std::vector<double> > VarMap;
      typedef std::map<short, double> ProbMap;

   public:

      LikeModule();
      ~LikeModule();

      bool Read(const std::string &filepath, int nevent = 0, const std::string &treename = "events");
      bool Read(TDirectory *dir, int nevent = 0, const std::string &treename = "events");

      bool Write(const std::string &filepath, const std::string &treename = "events");
      void Write(TDirectory *dir, const std::string &treename = "events");

      void Add(const Event &event);

      bool Fill(unsigned short optimize_depth = 7, const std::string &option = "");

      bool Find(Event event, unsigned int nfind = 100) const;

      bool Find(unsigned int nfind, const std::string &option) const;
      
      const EventVec& GetEventVec() const;
      const Result& GetResult() const;

      const VarMap& GetVarMap() const;

      const std::map<int, double>& GetMetric() const;
      
      void Print() const;
      void Print(std::ostream &os) const;

   private:

      Node* Balance(unsigned int bdepth);

      void ComputeMetric(unsigned int ifrac = 10);

      const Event Scale(const Event &event) const;

   private:

      unsigned int fDimn;

      Node *fTree;

      std::map<int, double> fVarScale;

      mutable Result fResult;

      std::map<short, unsigned int> fCount; //count number of events of each type

      EventVec fEvent; // vector of all events used to build tree and analysis
      VarMap   fVar;   // sorted map of variables in each dimension for all event types
   };

   inline const Result& LikeModule::GetResult() const
   {
      return fResult;
   }
   inline const EventVec& LikeModule::GetEventVec() const
   {
      return fEvent;
   }
   inline const LikeModule::VarMap& LikeModule::GetVarMap() const
   {
      return fVar;
   }
   inline const std::map<int, double>& LikeModule::GetMetric() const
   {
      return fVarScale;
   }
}

#endif

---