/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitCore                                                       *
 * @(#)root/roofitcore:$Id$
 * Authors:                                                                  *
 *   GR, Gerhard Raven, NIKHEF/VU, Gerhard.Raven@nikhf.nl                    *
 *                                                                           *
 * Copyright (c) 2005, NIKHEF.  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)             *
 *****************************************************************************/


//////////////////////////////////////////////////////////////////////////////
// 
// BEGIN_HTML
// RooEffGenContext is a specialized generator context for p.d.fs represented
// by class RooEffProd, which are p.d.fs multiplied with an efficiency function.
// This generator context generates events from such products by first
// generating events from a dedicated generator context of the input p.d.f.
// and applying an extra rejection step based on the efficiency function.
// END_HTML
//

#include <memory>

#include "RooFit.h"
#include "RooEffGenContext.h"
#include "RooAbsPdf.h"
#include "RooRandom.h"

using namespace std;

ClassImp(RooEffGenContext);

//_____________________________________________________________________________
RooEffGenContext::RooEffGenContext(const RooAbsPdf &model, 
                                   const RooAbsPdf& pdf, const RooAbsReal& eff,
                                   const RooArgSet &vars,
                                   const RooDataSet *prototype, const RooArgSet* auxProto,
                                   Bool_t verbose, const RooArgSet* /*forceDirect*/) :
   RooAbsGenContext(model, vars, prototype, auxProto, verbose), _maxEff(0.)
{
   // Constructor of generator context for RooEffProd products

   RooArgSet x(eff,eff.GetName());
   _cloneSet = static_cast<RooArgSet*>(x.snapshot(kTRUE));
   _eff = dynamic_cast<RooAbsReal*>(_cloneSet->find(eff.GetName()));
   _generator = pdf.genContext(vars, prototype, auxProto, verbose);
   _vars = static_cast<RooArgSet*>(vars.snapshot(kTRUE));
}

//_____________________________________________________________________________
RooEffGenContext::~RooEffGenContext()
{
   // Destructor
   delete _generator;
   delete _cloneSet;
   delete _vars;
}

//_____________________________________________________________________________
void RooEffGenContext::initGenerator(const RooArgSet &theEvent)
{
   // One-time initialization of generator.

   _eff->recursiveRedirectServers(theEvent);
   _generator->initGenerator(theEvent);

   // Check if PDF supports maximum finding
   Int_t code = _eff->getMaxVal(*_vars);
   if (!code) {
     _maxEff = 1.;
   } else {
     _maxEff = _eff->maxVal(code);
   }
}

//_____________________________________________________________________________
void RooEffGenContext::generateEvent(RooArgSet &theEvent, Int_t remaining)
{
   // Generate one event. Generate an event from the p.d.f and
   // then perform an accept/reject sampling based on the efficiency
   // function

   while (true) {
      _generator->generateEvent(theEvent, remaining);
      double val = _eff->getVal();
      if (val > _maxEff && !_eff->getMaxVal(*_vars)) {
         coutE(Generation) << ClassName() << "::" << GetName() 
              << ":generateEvent: value of efficiency is larger than assumed maximum of 1."  << std::endl;
         continue;
      }
      if (val > RooRandom::uniform() * _maxEff) {
         break;
      }
   }
}