RooEffResModel.h

Go to the documentation of this file.

/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitModels                                                     *
 *    File: $Id: RooGaussModel.h,v 1.21 2007/05/11 09:13:07 verkerke Exp $
 * Authors:                                                                  *
 *
 * Redistribution and use in source and binary forms,                        *
 * with or without modification, are permitted according to the terms        *
 * listed in LICENSE (http://roofit.sourceforge.net/license.txt)             *
 * 2012-08-24 imported from G. Raven's p2vv git repository - M. Schiller     *
 * 2012-09-08 re-import from upstream, with some substantial modifications   *
 *  to ensure that the code does not leak memory (Manuel Schiller)       *
 * 2012-10-04 re-import from upstream, with some experimental changes, which *
 *  which are disabled by default for now...                             *
 *****************************************************************************/
#ifndef ROO_EFF_RES_MODEL
#define ROO_EFF_RES_MODEL

#include <map>

#include "RooRealProxy.h"
#include "RooSetProxy.h"
#include "RooObjCacheManager.h"
#include "RooAbsEffResModel.h"
#include "RooResolutionModel.h"

class RooAbsAnaConvPdf;

class RooEffResModel : public RooResolutionModel, public RooAbsEffResModel {
    public:
    // Constructors, assignment etc
    inline RooEffResModel()  { }
    RooEffResModel(const char *name, const char *title,
        RooResolutionModel& model, RooAbsReal& eff);
    RooEffResModel(const RooEffResModel& other, const char* name = 0);
    virtual RooEffResModel* clone(const char* newname) const;
    virtual ~RooEffResModel();

    virtual Int_t basisCode(const char* name) const;
    virtual Int_t getAnalyticalIntegral(
        RooArgSet& allVars, RooArgSet& analVars,
        const char* rangeName=0) const;
    virtual Double_t analyticalIntegral(
        Int_t code, const char* rangeName) const;
    virtual Bool_t forceAnalyticalInt(const RooAbsArg& dep) const;

    virtual Int_t getGenerator(
        const RooArgSet& directVars, RooArgSet &generateVars,
        Bool_t staticInitOK = kTRUE) const;
    virtual void initGenerator(Int_t code);
    virtual void generateEvent(Int_t code);

    virtual RooAbsGenContext* modelGenContext(
        const RooAbsAnaConvPdf& convPdf, const RooArgSet &vars,
        const RooDataSet *prototype = 0, const RooArgSet* auxProto = 0,
        Bool_t verbose= kFALSE) const;

    virtual RooAbsReal* efficiency() const;

    virtual std::vector<RooAbsReal*> efficiencies() const;

    virtual RooResolutionModel& model() const;

    virtual RooArgSet* observables() const;

    const RooArgList& getIntegralRanges(
        const RooArgSet& iset, const char* rangeName = 0) const;

    protected:
    friend class RooMultiEffResModel;
    virtual Double_t evaluate() const;
    virtual RooEffResModel* convolution(
        RooFormulaVar* inBasis, RooAbsArg* owner) const;

    private:
    class CacheElem : public RooAbsCacheElement
        {
        public:
        CacheElem(const RooEffResModel& parent, const RooArgSet& iset,
            const TNamed *rangeName);
        virtual ~CacheElem();

        virtual RooArgList containedArgs(Action);
        Double_t getVal(const RooArgSet* nset = 0) const;

        private:
        // Payload
        std::vector<double> _bounds;
        RooRealVar* _x; // variable in which we parametrise
        RooAbsReal* _eff; // efficiency as function of _x
        RooRealVar* _xmin; // model integral low bound
        RooRealVar* _xmax; // model integral high bound
        RooAbsReal* _int; // integral of model
        mutable Double_t _val; 
    };

    CacheElem *getCache(
        const RooArgSet* iset, const TNamed* rangeName = 0) const;

    // Pointers to our underlying components
    RooSetProxy _observables;
    RooRealProxy _model;     // RooResolutionModel
    RooRealProxy _eff;       // RooAbsReal

    typedef std::map<std::string, RooArgList*> RangeMap;
    mutable RangeMap _ranges;

    mutable RooObjCacheManager _cacheMgr; 

    //ClassDef(RooEffResModel, 2) // EffResian Resolution Model
};

#endif