//////////////////////////////////////////////////////////////////////////
//                                                                      //
// RhoSimpleElectronSelector                                            //
//                                                                      //
// Selector classes for particle selection                              //
//                                                                      //
// Author List:                                                         //
// Marcel Kunze,  RUB, Feb. 99                                          //
// Copyright (C) 1999-2001, Ruhr-University Bochum.                     //
// Ralf Kliemt, HIM/GSI Feb.2013 (Cleanup & Restructuring)              //
//                                                                      //
//////////////////////////////////////////////////////////////////////////

#include <math.h>
#include "TDatabasePDG.h"

#include "RhoSelector/RhoSimpleElectronSelector.h"
#include "RhoBase/RhoCandidate.h"
#include "FairRecoCandidate.h"





ClassImp ( RhoSimpleElectronSelector )

TBuffer& operator>> ( TBuffer& buf, RhoSimpleElectronSelector *&obj )
{
  obj = ( RhoSimpleElectronSelector* ) buf.ReadObject ( RhoSimpleElectronSelector::Class() );
  return buf;
}

RhoSimpleElectronSelector::RhoSimpleElectronSelector ( const char* name, const char* type ) :
  RhoParticleSelectorBase ( name,type )
{}

Bool_t RhoSimpleElectronSelector::Accept ( RhoCandidate* b )
{
  if ( b == 0 ) { return kFALSE; }

  SetTypeAndMass ( b );

  double Le  = b->GetPidInfo ( 0 );
  double Lmu = b->GetPidInfo ( 1 );
  double Lpi = b->GetPidInfo ( 2 );
  double Lk  = b->GetPidInfo ( 3 );
  double Lp  = b->GetPidInfo ( 4 );

  if ( fCriterion == loose ) {
    if ( Le<0.2 ) { return kFALSE; }
  } else if ( fCriterion == veryLoose || fCriterion == variable || fCriterion == all ) {
    return kTRUE;
  } else if ( fCriterion == best ) {
    if ( Le<Lmu || Le<Lpi || Le<Lk || Le<Lp ) { return kFALSE; }
  } else if ( fCriterion == tight ) {
    if ( Le<0.5 ) { return kFALSE; }
  } else {
    if ( Le<0.9 ) { return kFALSE; }
  }

  return kTRUE;
}

Bool_t RhoSimpleElectronSelector::Accept ( FairRecoCandidate* b )
{
  return kFALSE;
}