StatSample.hpp

Go to the documentation of this file.

#ifndef CAFE_STATSAMPLE_HPP__
#define CAFE_STATSAMPLE_HPP__

#include "cafe/Event.hpp" 
#include "cafe/StatWeight.hpp" 

#include <string>
#include <vector>

namespace cafe {
  
  /* \brief
     Class to collect number of events for the one sample of statistics.
     This class is used internally by the Stat class.
     
     \author  Viatcheslav Shary (shary@fnal.gov)
     
  */
  class StatSample {
    
    std::string _name ; //< sample name
    std::vector<std::string> _tags ; //< tags used to add events to the sample (ANY)
    std::vector<std::string> _tags_and ; //< tags requiered with AND 
    std::vector<StatSelection*> _events ; //< selected events or event weight
    StatWeight* _weight ; //< the global event weight 
    std::vector<std::pair<StatWeight*,StatWeight*> > _syst ; //< systematics +sigma, -sigma

  public: 
    
    StatSample(const std::string& sample="data") ;
    virtual ~StatSample() ;
    
    const std::string& name() const {return _name ;}
    const std::vector<std::string>& tags() const {return _tags ;}
    const std::vector<std::string>& tagsAnd() const {return _tags_and ;}
    
    void AddTags(const std::string& tag) {_tags.push_back(tag);}
    
    void AddTags(const std::vector<std::string>& tags) ;
    
    void AddAndTags(const std::vector<std::string>& tags) ;
    
    void AddAndTags(const std::string& tag) {_tags_and.push_back(tag);}
    
    //return false if events does not have tags used for this sample
    bool tagged(const cafe::Event* event) const  ;
    
    unsigned int size() const { return _events.size();} 
    
    unsigned long nevents(unsigned int n = 0) const ;
    
    unsigned long nevents(const std::string& name) const ;
    
    double eff(unsigned int n = 0) const ;
    
    double eff(const std::string& name) const ;
    
    double effGlob(unsigned int n = 0) const ;
    
    double effGlob(const std::string& name) const ;
    
    double effErr(unsigned int n = 0) const ;
    
    double effErr(const std::string& name) const ;
    
    double effErrGlob(unsigned int n = 0) const ;
    
    double effErrGlob(const std::string& name) const ;
    
    const StatSelection* eventSelection(unsigned int n = 0) const ;
    
    const StatSelection* eventSelection(const std::string& name) const ;
    
    const StatWeight* eventWeight(unsigned int n) const ;
    
    const StatWeight*  eventWeight(const std::string& name="global") const ;
    
    double correctedEff(const StatWeight* w=0) const ;
    
    double correctedEffErr(const StatWeight* w=0) const ;
    
    const std::pair<StatWeight*, StatWeight*>& systematics(unsigned int n = 0) const ;
    
    unsigned int systsize() const { return _syst.size();} 
    
    double totalsyst_pos() const ; 

    double totalsyst_neg() const ; 

    bool compareNames(const StatSample& sample) const ;

    unsigned long add(cafe::Event* event, const std::string& name="Initial") ;
    
    double applyWeight(cafe::Event* event, const std::string& name, 
                       double weight, double weight_pos, double weight_neg) ;

    double calculateGlobalWeight(cafe::Event* event) ;
    
    void Clear() ; 

    ostream& HtmlTable(ostream& os) const ;

    ostream& print_tex (ostream& os, const std::string title) const ;

    ostream& print_tex_syst (ostream& os) const ;

    static std::string tex(const std::string& init) ;

  public:
       ClassDef(StatSample, 0);     

   } ;
  
}

#endif // CAFE_STATSAMPLE_HPP__

---