IsolationCorrectionHelper.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
 
/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef XAODPRIMITIVES_ISOLATIONCORRECTIONHELPER_H
#define XAODPRIMITIVES_ISOLATIONCORRECTIONHELPER_H
 
#include <bitset>
#include <vector>
 
namespace xAOD {
 
  namespace Iso {
 
    class IsolationCaloCorrectionBitsetHelper {
    public:
      static void decode( const IsolationCaloCorrectionBitset& bitset, std::vector<IsolationCaloCorrection>& results ) {
        results.clear();
        for( int i=0;i<static_cast<int>(IsolationCaloCorrection::numIsolationCaloCorrections);++i ){
          if( bitset.test(i) ) results.push_back( static_cast<IsolationCaloCorrection>(i) );
        }
      }
      
      static void encode( const std::vector<IsolationCaloCorrection>& values, IsolationCaloCorrectionBitset& bitset ) {
        bitset.reset();
        for( auto value : values ){
          bitset.set(static_cast<int>(value));
        }
      }      
    };
    
    
  
  
    class IsolationTrackCorrectionBitsetHelper {
    public:
      static void decode( const IsolationTrackCorrectionBitset& bitset, std::vector<IsolationTrackCorrection>& results ) {
        results.clear();
        for( int i=0;i<static_cast<int>(IsolationTrackCorrection::numIsolationTrackCorrections);++i ){
          if( bitset.test(i) ) results.push_back( static_cast<IsolationTrackCorrection>(i) );
        }
      }
 
      static void encode( const std::vector<IsolationTrackCorrection>& values, IsolationTrackCorrectionBitset& bitset ) {
        bitset.reset();
        for( auto value : values ){
          bitset.set(static_cast<int>(value));
        }
      }      
    };
    
  }
}    
    
#endif