Rebinder.h

// -*- C++ -*-
//
// Rebinder.h is a part of ThePEG - Toolkit for HEP Event Generation
// Copyright (C) 1999-2019 Leif Lonnblad
//
// ThePEG is licenced under version 3 of the GPL, see COPYING for details.
// Please respect the MCnet academic guidelines, see GUIDELINES for details.
//
#ifndef ThePEG_Rebinder_H
#define ThePEG_Rebinder_H

#include "ThePEG/Config/ThePEG.h"
#include "Rebinder.fh"
#include <stdexcept>

namespace ThePEG {

template <typename T>
class Rebinder {

public:
  
  Rebinder() {}

public:

  ThePEG_DECLARE_TEMPLATE_POINTERS(T,TPtr);

  typedef std::map<cTPtr,TPtr> MapType;

  typedef typename MapType::const_iterator const_iterator;

public:

  TPtr & operator[](tcTPtr t) { return theMap[t]; }

  template <typename R>
  R translate(const R & r) const {
    const_iterator it = theMap.find(r);
    return it == theMap.end()? R(): dynamic_ptr_cast<R>(it->second);
  }

  template <typename OutputIterator, typename InputIterator>
  void translate(OutputIterator r,
                 InputIterator first, InputIterator last) const {
    while ( first != last ) *r++ = translate(*first++);
  }

  template <typename R>
  R alwaysTranslate(const R & r) const {
    R ret;
    if ( !r ) return ret;
    const_iterator it = theMap.find(r);
    ret = (it == theMap.end()? R(): dynamic_ptr_cast<R>(it->second));
    if ( !ret ) throw std::runtime_error("Rebinder exception");
    return ret;
  }

  template <typename OutputIterator, typename InputIterator>
  void alwaysTranslate(OutputIterator r, InputIterator first, InputIterator last)
    const {
    while ( first != last ) *r++ = alwaysTranslate(*first++);
  }

  const MapType & map() const { return theMap; }

private:


  MapType theMap;

private:

  Rebinder(const Rebinder &);

  Rebinder & operator=(const Rebinder &) = delete;

};


}

#endif /* ThePEG_Rebinder_H */