LauWeightedSumEffModel.cc

Go to the documentation of this file.

// Copyright University of Warwick 2004 - 2013.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

// Authors:
// Thomas Latham
// John Back
// Paul Harrison

#include <cstdlib>
#include <iostream>

#include "TSystem.h"
#include "Lau2DHistDP.hh"
#include "Lau2DSplineDP.hh"
#include "LauDaughters.hh"
#include "LauWeightedSumEffModel.hh"
#include "LauKinematics.hh"
#include "LauVetoes.hh"

ClassImp(LauWeightedSumEffModel)


LauWeightedSumEffModel::LauWeightedSumEffModel(const LauDaughters* daughters) :
        daughters_( daughters ),
        effModel_( 0 ),
        squareDP_( kFALSE ),
        fluctuateEffHisto_( kFALSE ),
        lowBinWarningIssued_( kFALSE ),
        highBinWarningIssued_( kFALSE )
{
        if ( daughters_ == 0 ) {
                std::cerr << "ERROR in LauWeightedSumEffModel Constructor : invalid pointer to daughters object supplied." << std::endl;
                gSystem->Exit(EXIT_FAILURE);
        }
}

void LauWeightedSumEffModel::addEffModel(const LauAbsEffModel* effModel, Double_t coeff)
{
        const LauDaughters* otherDaughters = effModel->getDaughters();
        if( otherDaughters->getTypeDaug1()!=daughters_->getTypeDaug1() ||
            otherDaughters->getTypeDaug2()!=daughters_->getTypeDaug2() ||
            otherDaughters->getTypeDaug3()!=daughters_->getTypeDaug3() ||
            otherDaughters->getTypeParent()!=daughters_->getTypeParent() ) {
                std::cerr << "ERROR in LauWeightedSumEffModel::addEffModel : daughters of provided efficiency model do not match those expected." << std::endl;
                gSystem->Exit(EXIT_FAILURE);
        }

        if( effModel_.empty() ) {
                squareDP_=effModel->usingSquareDP();
        } else if( effModel->usingSquareDP() != usingSquareDP() ) {
                std::cerr << "ERROR in LauWeightedSumEffModel::addEffModel : either all efficiency models must use the normal DP or all efficiency models must use the square DP." << std::endl;
                gSystem->Exit(EXIT_FAILURE);
        }

        effModel_.push_back(effModel);
        coeff_.push_back(coeff);
        if(effModel->fluctuateEffHisto()) fluctuateEffHisto_=kTRUE;
}

Double_t LauWeightedSumEffModel::calcEfficiency( const LauKinematics* kinematics ) const
{
        // Routine to calculate the efficiency for the given event/point in
        // the Dalitz plot. This routine uses the models set by the
        // addEffModel() function.
        Double_t eff(0.0);

        std::vector<const LauAbsEffModel*>::const_iterator it = effModel_.begin();
        std::vector<const LauAbsEffModel*>::const_iterator end = effModel_.end();

        std::vector<Double_t>::const_iterator coeffIt = coeff_.begin();

        for( ; it!=end; ++it) {
                eff += (*coeffIt)*(*it)->calcEfficiency( kinematics );
                ++coeffIt;
        }

        // Check that the efficiency is in the allowed range (0-1)
        // Out of range efficiencies could be caused by incorrect coefficients.
        if ( eff < 0.0 ) {
                if(!lowBinWarningIssued_) {
                        std::cerr << "WARNING in LauWeightedSumEffModel::calcEfficiency : Efficiency " << eff << " is less than 0 - setting to 0.  You may want to check your coefficients!" << std::endl
                                  << "                                       : Further warnings will be suppressed." << std::endl;
                        lowBinWarningIssued_=kTRUE;
                }
                eff = 0.0;
        } else if ( eff > 1.0 ) {
                if(!highBinWarningIssued_) {
                        std::cerr << "WARNING in LauWeightedSumEffModel::calcEfficiency : Efficiency " << eff << " is greater than 1 - setting to 1.  You may want to check your coefficients!" << std::endl
                                  << "                                       : Further warnings will be suppressed." << std::endl;
                        highBinWarningIssued_=kTRUE;
                }
                eff = 1.0;
        }

        return eff;
}

Bool_t LauWeightedSumEffModel::passVeto( const LauKinematics* kinematics ) const
{
        std::vector<const LauAbsEffModel*>::const_iterator it = effModel_.begin();
        std::vector<const LauAbsEffModel*>::const_iterator end = effModel_.end();

        for( ; it!=end; ++it) {
                if(!(*it)->passVeto( kinematics )) return kFALSE;
        }
        return kTRUE;
}