ElectronAssociation.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
 
#include "egammaEvent/ElectronAssociation.h"
 
namespace Analysis
{
  ElectronAssociation::ElectronAssociation(const NameType& name)
  {
    this->setName(name);
  }  
  
  JetAssociationBase* ElectronAssociation::clone() const
  {
    return new ElectronAssociation( *this );
  }
  
  const Electron* ElectronAssociation::electron() const
  {
    //get electron from Navigable
    if (this->size() ==0) 
      return nullptr;
    if (this->size() > 1)
      return nullptr;
    // this shouldn't happen
    else 
      return getConstituentPtr((this->getConstituents()).begin());
  }
  
  void  ElectronAssociation::set_electron(const ElectronContainer* theContainer,
                      const Electron* the_electron,
                      double weight)
  {
    double newWeight = weight;
    // electron already in collection
    if ( this->contains(the_electron) )
      {
    // update weight
    newWeight *= this->getElectronWeight(the_electron);
    // remove the previous electron 
    this->remove(the_electron);
      }
    
    // from Navigable base
    this->putElement(theContainer,the_electron,newWeight);
  }
  
  void  ElectronAssociation::set_electron(const ElectronContainer* theContainer,
                      const index_type& theIndex,
                      double weight)
  {
    double newWeight = weight;
    // electron already in collection
    if ( this->contains(theContainer,theIndex) )
      {
    // update weight
    newWeight *= this->getElectronWeight(theContainer,theIndex);
    // remove the previous electron
    this->remove(theContainer,theIndex);
      }
    
    // from Navigable base
    this->putElement(theContainer,theIndex,newWeight);
  }
  
  double ElectronAssociation::getElectronWeight(const Electron* the_electron) const
  {
    // from Navigable
    return (this->contains(the_electron) )
      ? this->getParameter(the_electron)
      : 0.;
  }
  
  double ElectronAssociation::getElectronWeight(const ElectronContainer* theContainer,
                        const index_type& theIndex) const
  {
    // from Navigable
    return (this->contains(theContainer,theIndex))
      ? this->getParameter(theContainer,theIndex)
      : 0.;
  }
}