Skip to content
Snippets Groups Projects
Commit 81652825 authored by Wouter Verkerke's avatar Wouter Verkerke
Browse files

back out new version of RooKeysPdf that was accidentally committed 

git-svn-id: http://root.cern.ch/svn/root/trunk@20164 27541ba8-7e3a-0410-8455-c3a389f83636
parent 391bd3ff
No related branches found
No related tags found
No related merge requests found
Expand all Hide whitespace changes
Inline Side-by-side
roofit/inc/RooKeysPdf.h
+ 28
− 174
View file @ 81652825
/*****************************************************************************
* Project: RooFit *
* Package: RooFitModels *
* File: $Id: RooKeysPdf.rdl,v 1.1 2007/03/01 15:23:18 hellas Exp $
* File: $Id: RooKeysPdf.h,v 1.10 2007/05/11 09:13:07 verkerke Exp $
* Authors: *
* GR, Gerhard Raven, UC San Diego, raven@slac.stanford.edu *
* DK, David Kirkby, UC Irvine, dkirkby@uci.edu *
......@@ -14,201 +14,55 @@
* with or without modification, are permitted according to the terms *
* listed in LICENSE (http://roofit.sourceforge.net/license.txt) *
*****************************************************************************/
#ifndef ROO_KEYS_PDF
#define ROO_KEYS_PDF
#ifndef ROO_KEYS
#define ROO_KEYS
#include "RooAbsPdf.h"
#include "RooRealProxy.h"
#include "RooSetProxy.h"
#include "RooRealConstant.h"
#include "TVectorD.h"
#include "TMatrixD.h"
#include "TMatrixDSym.h"
#include <map>
#include <vector>
#include <string>
class RooRealVar;
class RooArgList;
class RooArgSet;
using namespace std;
typedef std::pair<Int_t, vector<vector<Double_t> >::iterator > iiPair;
typedef std::vector< iiPair > iiVec;
typedef std::pair<Int_t, vector<TVectorD>::iterator > itPair;
typedef std::vector< itPair > itVec;
class RooKeysPdf : public RooAbsPdf {
public:
enum Mirror {NoMirror, MirrorLeft, MirrorRight, MirrorBoth,
MirrorAsymLeft, MirrorAsymLeftRight,
MirrorAsymRight, MirrorLeftAsymRight,
MirrorAsymBoth };
RooKeysPdf(const char *name, const char *title,
const RooArgList& varList, RooDataSet& data,
TString options="a", Double_t rho=1, Double_t nSigma=3,
RooAbsReal& weight=RooRealConstant::value(1));
enum Mirror { NoMirror, MirrorLeft, MirrorRight, MirrorBoth,
MirrorAsymLeft, MirrorAsymLeftRight,
MirrorAsymRight, MirrorLeftAsymRight,
MirrorAsymBoth };
RooKeysPdf(const char *name, const char *title,
RooAbsReal& x, RooDataSet& data,
Mirror mirror= NoMirror, Double_t rho=1, Double_t nSigma=3,
RooAbsReal& weight=RooRealConstant::value(1));
RooKeysPdf(const char *name, const char *title,
RooAbsReal& x, RooAbsReal &y, RooDataSet& data,
TString options="a", Double_t rho = 1.0, Double_t nSigma=3,
RooAbsReal& weight=RooRealConstant::value(1));
RooAbsReal& x, RooDataSet& data, Mirror mirror= NoMirror,
Double_t rho=1);
RooKeysPdf(const RooKeysPdf& other, const char* name=0);
virtual TObject* clone(const char* newname) const {return new RooKeysPdf(*this,newname); }
virtual ~RooKeysPdf();
virtual TObject* clone(const char* newname) const { return new RooKeysPdf(*this,newname); }
Int_t getAnalyticalIntegral(RooArgSet& allVars, RooArgSet& analVars, const char* rangeName=0) const ;
Double_t analyticalIntegral(Int_t code, const char* rangeName=0) const ;
inline void fixShape(Bool_t fix) {
createPdf(kFALSE);
_fixedShape=fix;
}
void LoadDataSet( RooDataSet& data);
protected:
RooListProxy _varList ;
TIterator* _varItr ; //! do not persist
TIterator* _weightItr; //!
RooRealProxy _x ;
Double_t evaluate() const;
private:
struct BoxInfo {
Bool_t filled;
Bool_t netFluxZ;
Double_t nEventsBW;
Double_t nEventsBMSW;
vector<Double_t> xVarLo, xVarHi;
vector<Double_t> xVarLoM3s, xVarLoP3s, xVarHiM3s, xVarHiP3s;
map<Int_t,Bool_t> bpsIdcs;
vector<Int_t> sIdcs;
vector<Int_t> bIdcs;
vector<Int_t> bmsIdcs;
} ;
void createPdf(Bool_t firstCall=kTRUE) const;
void setOptions() const;
void initialize() const;
void loadDataSet(Bool_t firstCall) const;
void mirrorDataSet() const;
void loadWeightSet() const;
void calculateShell(BoxInfo* bi) const;
void calculatePreNorm(BoxInfo* bi) const;
void sortDataIndices(BoxInfo* bi=0) const;
void calculateBandWidth() const;
Double_t gauss(vector<Double_t>& x, vector<vector<Double_t> >& weights) const;
void loopRange(vector<Double_t>& x, map<Int_t,Bool_t>& ibMap) const;
void boxInfoInit(BoxInfo* bi, const char* rangeName, Int_t code) const;
virtual Bool_t redirectServersHook(const RooAbsCollection& /*newServerList*/, Bool_t /*mustReplaceAll*/,
Bool_t /*nameChange*/, Bool_t /*isRecursive*/) ;
mutable RooDataSet& _data;
mutable TString _options;
mutable Double_t _widthFactor;
mutable Double_t _nSigma;
mutable RooRealProxy _weight ;
mutable RooSetProxy _weightParams ;
mutable RooAbsReal* _weightDep ;
mutable Bool_t _fixedShape;
mutable Bool_t _mirror;
mutable Bool_t _debug;
mutable Bool_t _verbose;
mutable Double_t _sqrt2pi;
mutable Int_t _nDim;
mutable Int_t _nEvents;
mutable Int_t _nEventsM;
mutable Double_t _nEventsW;
mutable Double_t _d;
mutable Double_t _n;
// cached info on variable
mutable vector<vector<Double_t> > _dataPts;
mutable vector<TVectorD> _dataPtsR;
mutable vector<vector<Double_t> > _weights0;
mutable vector<vector<Double_t> > _weights1;
mutable vector<vector<Double_t> >* _weights; //!
mutable vector<iiVec> _sortIdcs;
mutable vector<itVec> _sortTVIdcs;
mutable vector<string> _varName;
mutable vector<Double_t> _rho;
mutable RooArgSet _dataVars;
mutable vector<Double_t> _x;
mutable vector<Double_t> _x0, _x1, _x2;
mutable vector<Double_t> _mean, _sigma;
mutable vector<Double_t> _xDatLo, _xDatHi;
mutable vector<Double_t> _xDatLo3s, _xDatHi3s;
Double_t evaluateFull(Double_t x) const;
mutable Bool_t _netFluxZ;
mutable Double_t _nEventsBW;
mutable Double_t _nEventsBMSW;
mutable vector<Double_t> _xVarLo, _xVarHi;
mutable vector<Double_t> _xVarLoM3s, _xVarLoP3s, _xVarHiM3s, _xVarHiP3s;
mutable map<Int_t,Bool_t> _bpsIdcs;
mutable vector<Int_t> _sIdcs;
mutable vector<Int_t> _bIdcs;
mutable vector<Int_t> _bmsIdcs;
mutable map<pair<string,int>,BoxInfo*> _rangeBoxInfo ;
mutable BoxInfo _fullBoxInfo ;
mutable vector<Int_t> _idx;
mutable Double_t _minWeight;
mutable Double_t _maxWeight;
mutable map<Int_t,Double_t> _wMap;
mutable TMatrixDSym* _covMat;
mutable TMatrixDSym* _corrMat;
mutable TMatrixD* _rotMat;
mutable TVectorD* _sigmaR;
mutable TVectorD* _dx;
mutable Double_t _sigmaAvgR;
/*
/// sorter function
struct SorterL2H {
Int_t idx;
SorterL2H (Int_t index) : idx(index) {}
bool operator() (const iiPair& a, const iiPair& b) {
const vector<Double_t>& aVec = *(a.second);
const vector<Double_t>& bVec = *(b.second);
return (aVec[idx]<bVec[idx]);
}
};
*/
Int_t _nEvents;
Double_t *_dataPts; //!
Double_t *_weights; //!
enum { _nPoints = 1000 };
Double_t _lookupTable[_nPoints+1];
Double_t g(Double_t x,Double_t sigma) const;
/// sorter function
struct SorterTV_L2H {
Int_t idx;
SorterTV_L2H (Int_t index) : idx(index) {}
bool operator() (const itPair& a, const itPair& b) {
const TVectorD& aVec = *(a.second);
const TVectorD& bVec = *(b.second);
return (aVec[idx]<bVec[idx]);
}
};
Bool_t _mirrorLeft, _mirrorRight;
Bool_t _asymLeft, _asymRight;
// cached info on variable
Char_t _varName[128];
Double_t _lo, _hi, _binWidth;
Double_t _rho;
ClassDef(RooKeysPdf,1) // Non-Parametric KEYS PDF
};
......
roofit/src/RooKeysPdf.cxx
+ 156
− 1092
View file @ 81652825
This diff is collapsed.
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment