Stat.hpp

Go to the documentation of this file.

#ifndef CAFE_STAT_HPP__
#define CAFE_STAT_HPP__

#include "cafe/StatSample.hpp"
#include "cafe/Processor.hpp"

#include <string>
#include <vector>
#include <map>

namespace cafe {

  struct Syst {
    std::string name ;
    const StatSample* reference;
    StatSample pos, neg ;
    Syst(const std::string& syst_name, const StatSample* reference_sample) :
      name(syst_name), reference(reference_sample), 
      pos(reference_sample->name()+"_"+name+"_pos"), 
      neg(reference_sample->name()+"_"+name+"_neg") {

      pos.AddTags(reference->tags()) ;
      pos.AddAndTags(reference->tagsAnd()) ;
      pos.AddAndTags(syst_name+"_pos") ;

      neg.AddTags(reference->tags()) ;
      neg.AddAndTags(reference->tagsAnd()) ;
      neg.AddAndTags(syst_name+"_neg") ;

    }
  };
    
  class Stat : public cafe::Processor{

    std::vector<StatSample> _samples ; //< number of selected events for the different samples
    std::vector<Syst> _syst ; //< vector with systematics samples
    cafe::Event* _event ; //< pointer to the event object
    std::vector<std::string> _tags ; //< initial tags used to tag events
    bool _ignoreauto ;  //< ignore records made automatically

    double syst_pos(std::vector<Syst>::const_iterator jt) const ;

    double syst_neg(std::vector<Syst>::const_iterator jt) const ;

    double systerr_pos(std::vector<Syst>::const_iterator jt) const ;

    double systerr_neg(std::vector<Syst>::const_iterator jt) const ;

        Stat *_parent_stat;

  public:

    Stat(const char *name);
        Stat(const char *name, bool duplicate_ok);
    ~Stat();

    double eventWeight(const std::string& name="global", 
                       const std::string sample_name="not_specified") const ;

    double nevents(const std::string& name="Initial", const std::string& sample_name="not_specified") const ;

    double efficiencyCorrected(const std::string& sample_name="not_specified") const;

    double efficiency(const std::string& sample_name="not_specified") const;

    double syst_pos(const std::string& name, const std::string& sample_name="not_specified") const ; 

    double syst_neg(const std::string& name, const std::string& sample_name="not_specified") const ; 

    double systerr_pos(const std::string& name, const std::string& sample_name="not_specified") const ; 

    double systerr_neg(const std::string& name, const std::string& sample_name="not_specified") const ; 

    std::vector<const cafe::StatSample*> get_samples() const ;

    std::vector<const cafe::Syst*> get_syst() const ;

    void begin();

    void finish() ;

    bool processEvent(cafe::Event& event);

    void EventSelected(const std::string& selection_name) ;

    void applyWeight(const std::string& weight_name, 
                     double weight, double weight_pos=-1.0, double weight_neg=-1.0) ;

    double calculateGlobalWeight() ;

    void eventEnd() ;

    StatSample&  add_sample(const std::string& name) {_samples.push_back(name); return _samples.back();} 

    /* \brief Add samples for the systematics (with name "name") study. If sample == 0, 
              systematics samples will be added to all existing samples .
    */
    void add_syst(const std::string& name, StatSample* sample=0) ;

    void tag(const std::vector<std::string>& tags) ;

    ostream& print_html(ostream& os) const ; 
    ostream& print_tex(ostream& os, const std::string title="Efficiency") const ; 

        // Use when you need to use a sub-stat processor.
        void chain (void);
        void unchain (void);
        
        // Use to inherrit current weights from parent
        void inheritWeights();
    
  public:
    ClassDef(Stat, 0) ;
  };

  class StatPointer {
    Stat* _stat ;
    
  public :
    StatPointer(Stat* stat=0) ;
    virtual ~StatPointer() ;
    Stat* pointer() {return _stat;}
 
    void EventSelected(const std::string& selection_name) ;

    void applyWeight(const std::string& weight_name, 
                     double weight, double weight_pos=-1.0, double weight_neg=-1.0) ;

    double calculateGlobalWeight() ;
    
    ClassDef(StatPointer, 0) ;

  } ;
}

#endif // CAFE_STAT_HPP__

---