root/TMVA/PDEFoamDiscriminant.h

0001 // @(#)root/tmva $Id$
0002 // Author: Tancredi Carli, Dominik Dannheim, Alexander Voigt
0003
0004 /**********************************************************************************
0005  * Project: TMVA - a Root-integrated toolkit for multivariate data analysis       *
0006  * Package: TMVA                                                                  *
0007  * Classes: PDEFoamDiscriminant                                                   *
0008  *                                             *
0009  *                                                                                *
0010  * Description:                                                                   *
0011  *    Concrete PDEFoam sub-class.  This foam stores the discriminant D            *
0012  *    = N_sig / (N_bg + N_sig) with every cell, as well as the                    *
0013  *    statistical error on the discriminant.                                      *
0014  *                                                                                *
0015  * Authors (alphabetical):                                                        *
0016  *      S. Jadach        - Institute of Nuclear Physics, Cracow, Poland           *
0017  *      Tancredi Carli   - CERN, Switzerland                                      *
0018  *      Dominik Dannheim - CERN, Switzerland                                      *
0019  *      Alexander Voigt  - TU Dresden, Germany                                    *
0020  *                                                                                *
0021  * Copyright (c) 2008, 2010:                                                      *
0022  *      CERN, Switzerland                                                         *
0023  *      MPI-K Heidelberg, Germany                                                 *
0024  *                                                                                *
0025  * Redistribution and use in source and binary forms, with or without             *
0026  * modification, are permitted according to the terms listed in LICENSE           *
0027  * (see tmva/doc/LICENSE)                                          *
0028  **********************************************************************************/
0029
0030 #ifndef ROOT_TMVA_PDEFoamDiscriminant
0031 #define ROOT_TMVA_PDEFoamDiscriminant
0032
0033 #include "TMVA/PDEFoam.h"
0034
0035 namespace TMVA
0036 {
0037
0038    class PDEFoamDiscriminant : public PDEFoam
0039    {
0040
0041    protected:
0042       UInt_t fClass; // signal class
0043
0044       PDEFoamDiscriminant(const PDEFoamDiscriminant&); // Copy Constructor  NOT USED
0045
0046       // ---------- Public functions ----------------------------------
0047    public:
0048       PDEFoamDiscriminant();                  // Default constructor (used only by ROOT streamer)
0049       PDEFoamDiscriminant(const TString&, UInt_t); // Principal user-defined constructor
0050       virtual ~PDEFoamDiscriminant() {}       // Default destructor
0051
0052       // function to fill created cell with given value
0053       virtual void FillFoamCells(const Event* ev, Float_t wt);
0054
0055       // function to call after foam is grown
0056       virtual void Finalize();
0057
0058       // 2-dimensional projection
0059       virtual TH2D* Project2(Int_t, Int_t, ECellValue, PDEFoamKernelBase*, UInt_t);
0060
0061       // ---------- ROOT class definition
0062       ClassDef(PDEFoamDiscriminant, 1) // Tree of PDEFoamCells
0063          }; // end of PDEFoamDiscriminant
0064
0065 }  // namespace TMVA
0066
0067 // ---------- Inline functions
0068
0069 #endif