%%cpp -d
#include "RooStats/RooStatsUtils.h"
#include <iostream>

using namespace RooFit;
using namespace RooStats; // the utilities are in the RooStats namespace
using std::cout, std::endl;

double sExpected = 50;
double bExpected = 100;
double relativeBkgUncert = 0.1;

double pExp = NumberCountingUtils::BinomialExpP(sExpected, bExpected, relativeBkgUncert);
double zExp = NumberCountingUtils::BinomialExpZ(sExpected, bExpected, relativeBkgUncert);
cout << "expected p-value =" << pExp << "  Z value (Gaussian sigma) = " << zExp << endl;

double observed = 150;
double pObs = NumberCountingUtils::BinomialObsP(observed, bExpected, relativeBkgUncert);
double zObs = NumberCountingUtils::BinomialObsZ(observed, bExpected, relativeBkgUncert);
cout << "observed p-value =" << pObs << "  Z value (Gaussian sigma) = " << zObs << endl;

double tau = 1;

double pExpWithTau = NumberCountingUtils::BinomialWithTauExpP(sExpected, bExpected, tau);
double zExpWithTau = NumberCountingUtils::BinomialWithTauExpZ(sExpected, bExpected, tau);
cout << "expected p-value =" << pExpWithTau << "  Z value (Gaussian sigma) = " << zExpWithTau << endl;

double pObsWithTau = NumberCountingUtils::BinomialWithTauObsP(observed, bExpected, tau);
double zObsWithTau = NumberCountingUtils::BinomialWithTauObsZ(observed, bExpected, tau);
cout << "observed p-value =" << pObsWithTau << "  Z value (Gaussian sigma) = " << zObsWithTau << endl;