SimpleVFATFrameCollection.h

Go to the documentation of this file.

/****************************************************************************
*
* This is a part of the TOTEM offline software.
* Authors: 
*   Mate Csanad (mate.csanad@cern.ch)
*   Jan Kašpar (jan.kaspar@gmail.com) 
*    
****************************************************************************/


#ifndef EventFilter_TotemRawToDigi_SimpleVFATFrameCollection
#define EventFilter_TotemRawToDigi_SimpleVFATFrameCollection

#include "EventFilter/TotemRawToDigi/interface/VFATFrameCollection.h"

#include <map>

class SimpleVFATFrameCollection : public VFATFrameCollection
{
  protected:
    typedef std::map<TotemFramePosition, VFATFrame> MapType;

    MapType data;

    virtual value_type BeginIterator() const;
    virtual value_type NextIterator(const value_type&) const;
    virtual bool IsEndIterator(const value_type&) const;

  public:
    SimpleVFATFrameCollection();
    ~SimpleVFATFrameCollection();

    const VFATFrame* GetFrameByID(unsigned int ID) const;
    const VFATFrame* GetFrameByIndex(TotemFramePosition index) const;

    virtual unsigned int Size() const
    {
      return data.size();
    }

    virtual bool Empty() const
    {
      return (data.size() == 0);
    }

    void Insert(const TotemFramePosition &index, const VFATFrame &frame)
    {
      data.insert({index, frame});
    }

    VFATFrame* InsertEmptyFrame(TotemFramePosition index)
    {
      return &data.insert({index, VFATFrame()}).first->second;
    }

    void Clear()
    {
      data.clear();
    }
};

#endif