MbdVertexMapv1.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file MbdVertexMapv1.h

#ifndef G4MBD_MBDVERTEXMAPV1_H
#define G4MBD_MBDVERTEXMAPV1_H

#include "MbdVertexMap.h"

#include "MbdVertex.h"

#include <cstddef>  // for size_t
#include <iostream>
#include <map>

class MbdVertexMapv1 : public MbdVertexMap
{
 public:
  MbdVertexMapv1();
  ~MbdVertexMapv1() override;

  void identify(std::ostream& os = std::cout) const override;
  void Reset() override { clear(); }
  int isValid() const override { return 1; }

  bool empty() const override { return _map.empty(); }
  size_t size() const override { return _map.size(); }
  size_t count(unsigned int idkey) const override { return _map.count(idkey); }
  void clear() override;

  const MbdVertex* get(unsigned int idkey) const override;
  MbdVertex* get(unsigned int idkey) override;
  MbdVertex* insert(MbdVertex* vertex) override;
  size_t erase(unsigned int idkey) override
  {
    delete _map[idkey];
    return _map.erase(idkey);
  }

  ConstIter begin() const override { return _map.begin(); }
  ConstIter find(unsigned int idkey) const override { return _map.find(idkey); }
  ConstIter end() const override { return _map.end(); }

  Iter begin() override { return _map.begin(); }
  Iter find(unsigned int idkey) override { return _map.find(idkey); }
  Iter end() override { return _map.end(); }

 private:
  std::map<unsigned int, MbdVertex*> _map;

  ClassDefOverride(MbdVertexMapv1, 1);
};

#endif  // G4MBD_MBDVERTEXMAPV1_H