SubDetHitStatistics.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/
 
// Dear emacs, this is -*-c++-*-
 
#ifndef SUBDETHITSTATISTICS_H
#define SUBDETHITSTATISTICS_H
 
#include "AthenaKernel/CLASS_DEF.h"
 
class TrackTruthCollectionAccessor;
 
 
class SubDetHitStatistics {
public:
 
  class OverflowControlledRef {
    unsigned char *m_p;
  public:
    static const unsigned int OVERFLOW_VALUE = 255;
    
    OverflowControlledRef(unsigned char& c) : m_p(&c) {}
 
    operator unsigned int () const { return *m_p; }
    unsigned int value() const { return *m_p; }
 
    OverflowControlledRef operator=(unsigned int rhs) { *m_p = (rhs > OVERFLOW_VALUE) ? OVERFLOW_VALUE : rhs; return *this; }
 
    OverflowControlledRef operator++() { unsigned int tmp = 1 + *m_p; *m_p = (tmp > OVERFLOW_VALUE) ? OVERFLOW_VALUE : tmp; return *this; }
 
    OverflowControlledRef operator+=(unsigned int rhs) {
      unsigned int tmp = *m_p + rhs;
      *m_p = ((tmp > OVERFLOW_VALUE)
          ||(tmp < rhs)/*protection agains overflowing tmp */
          ) ? OVERFLOW_VALUE : tmp;
      return *this;
    }
  };  // end of class OverflowControlledRef
 
  //----------------------------------------------------------------
  // SubDetHitStatistics non-class members
 
  enum SubDetType { Pixel=0, SCT, TRT, MDT, RPC, TGC, CSC, MM, STGC, NUM_SUBDETECTORS };
  static const unsigned int OVERFLOW_VALUE = OverflowControlledRef::OVERFLOW_VALUE;
 
  SubDetHitStatistics();
 
  unsigned int operator[](SubDetType i) const { return m_numPRDs[i]; }
 
  OverflowControlledRef operator[](SubDetType i) { return m_numPRDs[i]; }
 
private:
  unsigned char m_numPRDs[NUM_SUBDETECTORS];
 
  friend class TrackTruthCollectionAccessor; // in TrackTruthTPCnv
 
public:
};
 
template<class Ostream> Ostream& operator<<(Ostream& os, const SubDetHitStatistics& m) {
  os<<"SubDetStat(";
  for(unsigned i=0; i<SubDetHitStatistics::NUM_SUBDETECTORS; i++) {
    os<<unsigned(m[SubDetHitStatistics::SubDetType(i)]);
    if(i!=SubDetHitStatistics::NUM_SUBDETECTORS) os<<",";
  }
  return os<<")";
}
 
// That method would be the *only* one in a .so file.
// So inline it and get rid of the library.  
// Also note that memset()-ing 7 bytes can be inlined by gcc.
#include <cstring>
inline SubDetHitStatistics::SubDetHitStatistics() {
  memset(m_numPRDs, 0, NUM_SUBDETECTORS);
}
 
CLASS_DEF(SubDetHitStatistics, 73400190, 1)
 
#endif/*SUBDETHITSTATISTICS_H*/