// @(#)root/tmva $Id$   
// Author: Andreas Hoecker, Joerg Stelzer, Helge Voss

/**********************************************************************************
 * Project: TMVA - a Root-integrated toolkit for multivariate data analysis       *
 * Package: TMVA                                                                  *
 * Class  : SdivSqrtSplusB                                                        *
 * Web    : http://tmva.sourceforge.net                                           *
 *                                                                                *
 * Description: Implementation of the SdivSqrtSplusB as separation criterion      *
 *              s / sqrt( s+b )                                                   * 
 *                                                                                *
 *                                                                                *
 * Authors (alphabetical):                                                        *
 *      Andreas Hoecker <Andreas.Hocker@cern.ch> - CERN, Switzerland              *
 *      Helge Voss      <Helge.Voss@cern.ch>     - MPI-K Heidelberg, Germany      *
 *      Kai Voss        <Kai.Voss@cern.ch>       - U. of Victoria, Canada         *
 *                                                                                *
 * Copyright (c) 2005:                                                            *
 *      CERN, Switzerland                                                         * 
 *      U. of Victoria, Canada                                                    * 
 *      Heidelberg U., Germany                                                    * 
 *                                                                                *
 * Redistribution and use in source and binary forms, with or without             *
 * modification, are permitted according to the terms listed in LICENSE           *
 * (http://tmva.sourceforge.net/LICENSE)                                          *
 **********************************************************************************/

#include "TMath.h"
#include "TMVA/SdivSqrtSplusB.h"

ClassImp(TMVA::SdivSqrtSplusB)

//_______________________________________________________________________
Double_t  TMVA::SdivSqrtSplusB::GetSeparationIndex( const Double_t &s, const Double_t &b )
{
   // Index = S/sqrt(S+B)  (statistical significance)                 
   if (s+b > 0) return s / TMath::Sqrt(s+b);
   else return 0;
}


 
//_______________________________________________________________________
Double_t TMVA::SdivSqrtSplusB::GetSeparationGain(const Double_t &nSelS, const Double_t& nSelB,
                                                 const Double_t& nTotS, const Double_t& nTotB)
{
   // Separation Gain:
   // the measure of how the quality of separation of the sample increases
   // by splitting the sample e.g. into a "left-node" and a "right-node"
   // (N * Index_parent) - (N_left * Index_left) - (N_right * Index_right)
   // this is then the quality crition which is optimized for when trying
   // to increase the information in the system (making the best selection

   if ( (nTotS-nSelS)==nSelS && (nTotB-nSelB)==nSelB) return 0.;

   Double_t parentIndex = (nTotS+nTotB) *this->GetSeparationIndex(nTotS,nTotB);

   Double_t leftIndex   = ( ((nTotS - nSelS) + (nTotB - nSelB))
                            * this->GetSeparationIndex(nTotS-nSelS,nTotB-nSelB) );
   Double_t rightIndex  = (nSelS+nSelB) * this->GetSeparationIndex(nSelS,nSelB);

   //Double_t diff = parentIndex - leftIndex - rightIndex;
   Double_t diff = (parentIndex - leftIndex - rightIndex)/(nTotS+nTotB);

   if(diff<fPrecisionCut ) {
      // std::cout << " Warning value in GetSeparation is below numerical presicion " 
      //           << diff/parentIndex 
      //           << std::endl;
      return 0;
   }

   return diff;
}

 SdivSqrtSplusB.cxx:1
 SdivSqrtSplusB.cxx:2
 SdivSqrtSplusB.cxx:3
 SdivSqrtSplusB.cxx:4
 SdivSqrtSplusB.cxx:5
 SdivSqrtSplusB.cxx:6
 SdivSqrtSplusB.cxx:7
 SdivSqrtSplusB.cxx:8
 SdivSqrtSplusB.cxx:9
 SdivSqrtSplusB.cxx:10
 SdivSqrtSplusB.cxx:11
 SdivSqrtSplusB.cxx:12
 SdivSqrtSplusB.cxx:13
 SdivSqrtSplusB.cxx:14
 SdivSqrtSplusB.cxx:15
 SdivSqrtSplusB.cxx:16
 SdivSqrtSplusB.cxx:17
 SdivSqrtSplusB.cxx:18
 SdivSqrtSplusB.cxx:19
 SdivSqrtSplusB.cxx:20
 SdivSqrtSplusB.cxx:21
 SdivSqrtSplusB.cxx:22
 SdivSqrtSplusB.cxx:23
 SdivSqrtSplusB.cxx:24
 SdivSqrtSplusB.cxx:25
 SdivSqrtSplusB.cxx:26
 SdivSqrtSplusB.cxx:27
 SdivSqrtSplusB.cxx:28
 SdivSqrtSplusB.cxx:29
 SdivSqrtSplusB.cxx:30
 SdivSqrtSplusB.cxx:31
 SdivSqrtSplusB.cxx:32
 SdivSqrtSplusB.cxx:33
 SdivSqrtSplusB.cxx:34
 SdivSqrtSplusB.cxx:35
 SdivSqrtSplusB.cxx:36
 SdivSqrtSplusB.cxx:37
 SdivSqrtSplusB.cxx:38
 SdivSqrtSplusB.cxx:39
 SdivSqrtSplusB.cxx:40
 SdivSqrtSplusB.cxx:41
 SdivSqrtSplusB.cxx:42
 SdivSqrtSplusB.cxx:43
 SdivSqrtSplusB.cxx:44
 SdivSqrtSplusB.cxx:45
 SdivSqrtSplusB.cxx:46
 SdivSqrtSplusB.cxx:47
 SdivSqrtSplusB.cxx:48
 SdivSqrtSplusB.cxx:49
 SdivSqrtSplusB.cxx:50
 SdivSqrtSplusB.cxx:51
 SdivSqrtSplusB.cxx:52
 SdivSqrtSplusB.cxx:53
 SdivSqrtSplusB.cxx:54
 SdivSqrtSplusB.cxx:55
 SdivSqrtSplusB.cxx:56
 SdivSqrtSplusB.cxx:57
 SdivSqrtSplusB.cxx:58
 SdivSqrtSplusB.cxx:59
 SdivSqrtSplusB.cxx:60
 SdivSqrtSplusB.cxx:61
 SdivSqrtSplusB.cxx:62
 SdivSqrtSplusB.cxx:63
 SdivSqrtSplusB.cxx:64
 SdivSqrtSplusB.cxx:65
 SdivSqrtSplusB.cxx:66
 SdivSqrtSplusB.cxx:67
 SdivSqrtSplusB.cxx:68
 SdivSqrtSplusB.cxx:69
 SdivSqrtSplusB.cxx:70
 SdivSqrtSplusB.cxx:71
 SdivSqrtSplusB.cxx:72
 SdivSqrtSplusB.cxx:73
 SdivSqrtSplusB.cxx:74
 SdivSqrtSplusB.cxx:75