RooGenFitStudy.cxx

Go to the documentation of this file.

/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitCore                                                       *
 * @(#)root/roofitcore:$Id$
 * Authors:                                                                  *
 *   WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu       *
 *   DK, David Kirkby,    UC Irvine,         dkirkby@uci.edu                 *
 *                                                                           *
 * Copyright (c) 2000-2005, Regents of the University of California          *
 *                          and Stanford University. All rights reserved.    *
 *                                                                           *
 * 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)             *
 *****************************************************************************/
 
/**
\file RooGenFitStudy.cxx
\class RooGenFitStudy
\ingroup Roofitcore
 
Abstract base class for RooStudyManager modules
**/
 
#include "Riostream.h"
 
#include "RooGenFitStudy.h"
#include "RooWorkspace.h"
#include "RooMsgService.h"
#include "RooDataSet.h"
#include "RooAbsPdf.h"
#include "RooRealVar.h"
#include "RooGlobalFunc.h"
#include "RooFitResult.h"
 
 
ClassImp(RooGenFitStudy);
 
////////////////////////////////////////////////////////////////////////////////
/// Constructor
 
RooGenFitStudy::RooGenFitStudy(const char* name, const char* title) :
  RooAbsStudy(name?name:"RooGenFitStudy",title?title:"RooGenFitStudy")
{
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Copy constructor
 
RooGenFitStudy::RooGenFitStudy(const RooGenFitStudy& other) :
  RooAbsStudy(other),
  _genPdfName(other._genPdfName),
  _genObsName(other._genObsName),
  _fitPdfName(other._fitPdfName),
  _fitObsName(other._fitObsName)
{
  for(TObject * o : other._genOpts) _genOpts.Add(o->Clone());
  for(TObject * o : other._fitOpts) _fitOpts.Add(o->Clone());
}
 
////////////////////////////////////////////////////////////////////////////////
/// Function called after insertion into workspace
 
bool RooGenFitStudy::attach(RooWorkspace& w)
{
  bool ret = false ;
 
  RooAbsPdf* pdf = w.pdf(_genPdfName) ;
  if (pdf) {
    _genPdf = pdf ;
  } else {
    coutE(InputArguments) << "RooGenFitStudy(" << GetName() << ") ERROR: generator p.d.f named " << _genPdfName << " not found in workspace " << w.GetName() << std::endl ;
    ret = true ;
  }
 
  _genObs.add(w.argSet(_genObsName)) ;
  if (_genObs.empty()) {
    coutE(InputArguments) << "RooGenFitStudy(" << GetName() << ") ERROR: no generator observables defined" << std::endl ;
    ret = true ;
  }
 
  pdf = w.pdf(_fitPdfName) ;
  if (pdf) {
    _fitPdf = pdf ;
  } else {
    coutE(InputArguments) << "RooGenFitStudy(" << GetName() << ") ERROR: fitting p.d.f named " << _fitPdfName << " not found in workspace " << w.GetName() << std::endl ;
    ret = true ;
  }
 
  _fitObs.add(w.argSet(_fitObsName)) ;
  if (_fitObs.empty()) {
    coutE(InputArguments) << "RooGenFitStudy(" << GetName() << ") ERROR: no fitting observables defined" << std::endl ;
    ret = true ;
  }
 
  return ret ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
 
void RooGenFitStudy::setGenConfig(const char* pdfName, const char* obsName, const RooCmdArg& arg1,const RooCmdArg& arg2,const RooCmdArg& arg3)
{
  _genPdfName = pdfName ;
  _genObsName = obsName ;
  _genOpts.Add(arg1.Clone()) ;
  _genOpts.Add(arg2.Clone()) ;
  _genOpts.Add(arg3.Clone()) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
 
void RooGenFitStudy::setFitConfig(const char* pdfName, const char* obsName, const RooCmdArg& arg1,const RooCmdArg& arg2,const RooCmdArg& arg3)
{
  _fitPdfName = pdfName ;
  _fitObsName = obsName ;
  _fitOpts.Add(arg1.Clone()) ;
  _fitOpts.Add(arg2.Clone()) ;
  _fitOpts.Add(arg3.Clone()) ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// One-time initialization of study
 
bool RooGenFitStudy::initialize()
{
  _nllVar = new RooRealVar("NLL","-log(Likelihood)",0) ;
  _ngenVar = new RooRealVar("ngen","number of generated events",0) ;
 
  _params = std::unique_ptr<RooArgSet>{_fitPdf->getParameters(_genObs)};
  RooArgSet modelParams(*_params) ;
  _initParams = new RooArgSet;
  _params->snapshot(*_initParams);
  _params->add(*_nllVar) ;
  _params->add(*_ngenVar) ;
 
  _genSpec = _genPdf->prepareMultiGen(_genObs,static_cast<RooCmdArg&>(*_genOpts.At(0)),static_cast<RooCmdArg&>(*_genOpts.At(1)),static_cast<RooCmdArg&>(*_genOpts.At(2))) ;
 
  registerSummaryOutput(*_params,modelParams) ;
  return false ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Execute one study iteration
 
bool RooGenFitStudy::execute()
{
  _params->assign(*_initParams) ;
  std::unique_ptr<RooDataSet> data{_genPdf->generate(*_genSpec)};
  std::unique_ptr<RooFitResult> fr{_fitPdf->fitTo(*data,RooFit::Save(true),static_cast<RooCmdArg&>(*_fitOpts.At(0)),static_cast<RooCmdArg&>(*_fitOpts.At(1)),static_cast<RooCmdArg&>(*_fitOpts.At(2)))};
 
  if (fr->status()==0) {
    _ngenVar->setVal(data->sumEntries()) ;
    _nllVar->setVal(fr->minNll()) ;
    storeSummaryOutput(*_params) ;
    storeDetailedOutput(std::move(fr)) ;
  }
 
  return false ;
}
 
 
 
////////////////////////////////////////////////////////////////////////////////
/// Finalization of study
 
bool RooGenFitStudy::finalize()
{
  delete _nllVar ;
  delete _ngenVar ;
  delete _initParams ;
  delete _genSpec ;
  _params.reset();
  _nllVar = nullptr ;
  _ngenVar = nullptr ;
  _initParams = nullptr ;
  _genSpec = nullptr ;
 
 
  return false ;
}
 
 
////////////////////////////////////////////////////////////////////////////////
 
void RooGenFitStudy::Print(Option_t* /*options*/) const
{
}