#include <CombDataSetFactory.h>

Inheritance diagram for CombDataSetFactory:

Public Member Functions
void	addSetAny (const char label, RooDataSet set)

void	addSetAny (const char label, RooDataHist set)

void	addSetBin (const char label, RooDataHist set)

	CombDataSetFactory ()

	CombDataSetFactory (const RooArgSet &vars, RooCategory &cat)

RooDataHist *	done (const char name, const char title)

RooDataSet *	doneUnbinned (const char name, const char title)

	~CombDataSetFactory ()

Public Attributes
ClassDef(CombDataSetFactory, 1) private RooCategory *	cat_

std::map< std::string, RooDataHist * >	map_

std::map< std::string, RooDataSet * >	mapUB_

RooRealVar *	weight_

Detailed Description

Definition at line 20 of file CombDataSetFactory.h.

Constructor & Destructor Documentation

CombDataSetFactory::CombDataSetFactory ( )

inline

Definition at line 23 of file CombDataSetFactory.h.

23 {}

CombDataSetFactory::CombDataSetFactory	(	const RooArgSet &	vars,
		RooCategory &	cat
	)

Definition at line 6 of file CombDataSetFactory.cc.

                                                                               :
     vars_(vars), cat_(&cat), weight_(0)
 {
     if (vars_.contains(cat)) vars_.remove(cat);
 }

CombDataSetFactory::~CombDataSetFactory ( )

Definition at line 12 of file CombDataSetFactory.cc.

12 {}

Member Function Documentation

void CombDataSetFactory::addSetAny	(	const char *	label,
		RooDataSet *	set
	)

Definition at line 29 of file CombDataSetFactory.cc.

References diffTwoXMLs::label, mapUB_, and runtimedef::set().

                                                                      {
     mapUB_[label] = set;
 }

void CombDataSetFactory::addSetAny	(	const char *	label,
		RooDataHist *	set
	)

Definition at line 18 of file CombDataSetFactory.cc.

References data, i, diffTwoXMLs::label, mapUB_, n, and weight_.

                                                                       {
     if (weight_ == 0) weight_ = new RooRealVar("_weight_","",1);
     RooDataSet *data = new RooDataSet(TString(set->GetName())+"_unbin", "", RooArgSet(*set->get(), *weight_), "_weight_");
     for (int i = 0, n = set->numEntries(); i < n; ++i) {
         const RooArgSet *entry = set->get(i);
         data->add(*entry, set->weight());
     }
     mapUB_[label] = data;
 }

void CombDataSetFactory::addSetBin	(	const char *	label,
		RooDataHist *	set
	)

Definition at line 14 of file CombDataSetFactory.cc.

References diffTwoXMLs::label, map_, and runtimedef::set().

                                                                       {
     map_[label] = set;
 }

RooDataHist * CombDataSetFactory::done	(	const char *	name,
		const char *	title
	)

Definition at line 34 of file CombDataSetFactory.cc.

References cat_, and map_.

                                                                          {
     return new RooDataHist(name,title,vars_,*cat_,map_);
     map_.clear();
 }

RooDataSet * CombDataSetFactory::doneUnbinned	(	const char *	name,
		const char *	title
	)

Definition at line 39 of file CombDataSetFactory.cc.

References cat_, data, i, mapUB_, n, run_regression::ret, and weight_.

                                                                                 {
     using namespace RooFit; 
     RooDataSet *ret = 0;
     if (weight_) {
 #if 0   // This works most of the time, but sometimes weights end up wrong.
         // I don't knopw why the @$^#&^# this happens, but it nevertheless 
         // means that I can't use the constructor with the map directly
         //
         for (std::map<std::string, RooDataSet *>::iterator it = mapUB_.begin(), ed = mapUB_.end(); it != ed; ++it) {
             RooDataSet *data = it->second;
             if (!data->isWeighted()) {
                 weight_->setVal(1.0);
                 data->addColumn(*weight_);
                 it->second = new RooDataSet(data->GetName(), data->GetTitle(), data, *data->get(), /*cut=*/(char*)0, weight_->GetName());
             }
             //std::cout << "\n\n\n========== DATASET " << it->first << " =============" << std::endl;
             //utils::printRDH(it->second);
         }
         RooArgSet varsPlusWeight(vars_); varsPlusWeight.add(*weight_);
         ret = new RooDataSet(name,title,varsPlusWeight,Index(*cat_),Import(mapUB_),WeightVar(*weight_));
         //std::cout << "\n\n\n========== COMBINED DATASET =============" << std::endl;
         //utils::printRDH(ret);
 #else
         RooArgSet varsPlusCat(vars_); varsPlusCat.add(*cat_);
         RooArgSet varsPlusWeight(varsPlusCat); varsPlusWeight.add(*weight_);
         ret = new RooDataSet(name,title,varsPlusWeight,WeightVar(*weight_));
         for (std::map<std::string, RooDataSet *>::iterator it = mapUB_.begin(), ed = mapUB_.end(); it != ed; ++it) {
             //std::cout << "\n\n\n========== DATASET " << it->first << " =============" << std::endl;
             //utils::printRDH(it->second);
             cat_->setLabel(it->first.c_str());
             RooDataSet *data = it->second;
             for (unsigned int i = 0, n = data->numEntries(); i < n; ++i) {
                 varsPlusCat = *data->get(i);
                 ret->add(varsPlusCat, data->weight());
             }
         }
         //std::cout << "\n\n\n========== COMBINED DATASET =============" << std::endl;
         //utils::printRDH(ret);
 #endif
     } else {
         ret = new RooDataSet(name,title,vars_,Index(*cat_),Import(mapUB_));
     }
     mapUB_.clear();
     return ret;
 }