JRealExperiment.hh

Go to the documentation of this file.

#ifndef __JASTRONOMY__JREALEXPERIMENT__
#define __JASTRONOMY__JREALEXPERIMENT__
 
#include <vector>
 
#include "TObject.h"
#include "TH1.h"
#include "TH2.h"
#include "TH3.h"
 
#include "JAstronomy/JExperiment.hh"
#include "JAstronomy/JAspera.hh"
 
 
/**
 * \file
 *
 * Real experiment.
 * \author mdejong
 */
namespace JASTRONOMY {}
namespace JPP { using namespace JASTRONOMY; }
 
namespace JASTRONOMY {
 
  /**
   * Real experiment using PDF of signal and background.
   */
  struct JRealExperiment :
    public JExperiment,
    public JAspera
  {
    /**
     * Default constructor.
     */
    JRealExperiment()
    {}
 
 
    /**
     * Constructor.
     *
     * \param  hd         histogram with data
     * \param  hs         histogram with PDF of signal
     * \param  hb         histogram with PDF of background
     */
    template<class H_t>
    JRealExperiment(const H_t&   hd,
                    const H_t&   hs,
                    const H_t&   hb)
    {
      add(hd, hs, hb);
    }
 
 
    /**
     * Add objects with data and PDFs of signal and background.
     *
     * \param  pd         pointer to object with data
     * \param  ps         pointer to object with PDF of signal
     * \param  pb         pointer to object with PDF of background
     */
    void add(const TObject* pd,
             const TObject* ps,
             const TObject* pb)
    {
      if (add<TH3>(pd, ps, pb)) { return; }
      if (add<TH2>(pd, ps, pb)) { return; }
      if (add<TH1>(pd, ps, pb)) { return; }
    }
 
 
    /**
     * Add histograms with data and PDFs of signal and background.
     *
     * \param  hd         histogram with data
     * \param  hs         histogram with PDF of signal
     * \param  hb         histogram with PDF of background
     */
    void add(const TH1&   hd,
             const TH1&   hs,
             const TH1&   hb)
    {
      for (Int_t ix = 1; ix <= hs.GetXaxis()->GetNbins(); ++ix) {
        add(hd.GetBinContent(ix),
            hs.GetBinContent(ix),
            hb.GetBinContent(ix));
      }
    }
 
 
    /**
     * Add histograms with data and PDFs of signal and background.
     *
     * \param  hd         histogram with data
     * \param  hs         histogram with PDF of signal
     * \param  hb         histogram with PDF of background
     */
    void add(const TH2&   hd,
             const TH2&   hs,
             const TH2&   hb)
    {
      for (Int_t ix = 1; ix <= hs.GetXaxis()->GetNbins(); ++ix) {
        for (Int_t iy = 1; iy <= hs.GetYaxis()->GetNbins(); ++iy) {
          add(hd.GetBinContent(ix, iy),
              hs.GetBinContent(ix, iy),
              hb.GetBinContent(ix, iy));
        }
      }
    }
 
 
    /**
     * Add histograms with data and PDFs of signal and background.
     *
     * \param  hd         histogram with data
     * \param  hs         histogram with PDF of signal
     * \param  hb         histogram with PDF of background
     */
    void add(const TH3&   hd,
             const TH3&   hs,
             const TH3&   hb)
    {
      for (Int_t ix = 1; ix <= hs.GetXaxis()->GetNbins(); ++ix) {
        for (Int_t iy = 1; iy <= hs.GetYaxis()->GetNbins(); ++iy) {
          for (Int_t iz = 1; iz <= hs.GetZaxis()->GetNbins(); ++iz) {
            add(hd.GetBinContent(ix, iy, iz),
                hs.GetBinContent(ix, iy, iz),
                hb.GetBinContent(ix, iy, iz));
          }
        }
      }
    }
 
 
    /**
     * Add data, signal and background.
     *
     * \param  n          data
     * \param  s          signal
     * \param  b          background
     */
    void add(const size_t n,
             const double s,
             const double b)
    {
      if (check(s, b)) {
        put(n, s, b);
      } else {
        remnant.add(n, s, b);
      }
    }
 
 
    /**
     * Add remnant data, signal and background.
     */
    void add()
    {
      if (remnant.n != 0) {
 
        add(remnant.n, remnant.s, remnant.b);
 
        remnant.reset();
      }
    }
 
  protected:
    /**
     * Add objects with data and PDFs of signal and background.
     *
     * \param  pd         pointer to object with data
     * \param  ps         pointer to object with PDF of signal
     * \param  pb         pointer to object with PDF of background
     * \return            true if added; else false
     */
    template<class H_t>
    bool add(const TObject* pd,
             const TObject* ps,
             const TObject* pb)
    {
      if (dynamic_cast<const H_t*>(pd) != NULL &&
          dynamic_cast<const H_t*>(ps) != NULL &&
          dynamic_cast<const H_t*>(pb) != NULL) {
 
        const H_t& hd = dynamic_cast<const H_t&>(*pd);
        const H_t& hs = dynamic_cast<const H_t&>(*ps);
        const H_t& hb = dynamic_cast<const H_t&>(*pb);
 
        if (check(hd, hs) && check(hs, hb)) {
 
          add(hd, hs, hb);
 
          return true;
        }
      }
 
      return false;
    }
 
 
    struct remnant_type {
 
      void add(const size_t n, const double s, const double b)
      {
        this->n += n;
        this->s += s;
        this->b += b;
      }
 
      void reset()
      {
        this->n = 0;
        this->s = 0.0;
        this->b = 0.0;
      }
 
      size_t n = 0;
      double s = 0.0;
      double b = 0.0;
 
    } remnant;
  };
}
 
#endif