X-Git-Url: http://g0dil.de/git?a=blobdiff_plain;f=senf%2FPackets%2FPacketRegistry.ct;h=d1ce4666f1143a837808cb44d70f0ad7ae97f7fa;hb=57daeae6f2e924ce3f16f9677c3474f531cba9e5;hp=b4babccc279bdbea2cd4639a0f7c2333d0889f59;hpb=af697a0c8591b38f7ee7dbc3d1d0293f4ff72d37;p=senf.git

diff --git a/senf/Packets/PacketRegistry.ct b/senf/Packets/PacketRegistry.ct
index b4babcc..d1ce466 100644
--- a/senf/Packets/PacketRegistry.ct
+++ b/senf/Packets/PacketRegistry.ct
@@ -34,98 +34,129 @@
 #include <senf/Utils/Format.hh>
 
 #define prefix_
-///////////////////////////////ct.p////////////////////////////////////////
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 
-///////////////////////////////////////////////////////////////////////////
-// senf::detail::PkReg_EntryImpl<PacketType>
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
+// senf::detail::PacketRegistryImpl<KeyType>::Entry
 
+template <class KeyType>
+prefix_ senf::detail::PacketRegistryImpl<KeyType>::Entry::Entry(KeyType const & key_,
+                                                                int priority_)
+    : key (key_), priority (priority_)
+{}
+
+template <class KeyType>
+prefix_ senf::detail::PacketRegistryImpl<KeyType>::Entry::~Entry()
+{}
+
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
+// senf::detail::PacketRegistryImpl<KeyType>::EntryImpl<PacketType>
+
+template <class KeyType>
 template <class PacketType>
-prefix_ senf::PacketInterpreterBase::factory_t senf::detail::PkReg_EntryImpl<PacketType>::factory()
+prefix_ senf::detail::PacketRegistryImpl<KeyType>::EntryImpl<PacketType>::
+EntryImpl(KeyType const & key, int priority)
+    : Entry (key, priority)
+{}
+
+template <class KeyType>
+template <class PacketType>
+prefix_ senf::Packet::factory_t
+senf::detail::PacketRegistryImpl<KeyType>::EntryImpl<PacketType>::factory()
     const
 {
     return PacketType::factory();
 }
 
+template <class KeyType>
 template <class PacketType>
-prefix_ std::string senf::detail::PkReg_EntryImpl<PacketType>::name()
+prefix_ std::string senf::detail::PacketRegistryImpl<KeyType>::EntryImpl<PacketType>::name()
     const
 {
     return prettyName(typeid(PacketType));
 }
 
-///////////////////////////////////////////////////////////////////////////
-// senf::PacketRegistry<Tag>
-
-template <class Tag>
-prefix_ typename senf::PacketRegistry<Tag>::Registry &
-senf::PacketRegistry<Tag>::registry()
+template <class KeyType>
+template <class PacketType>
+prefix_ std::type_info const &
+senf::detail::PacketRegistryImpl<KeyType>::EntryImpl<PacketType>::type()
+    const
 {
-    static Registry registry (prettyName(typeid(Tag)));
-    return registry;
+    return typeid(PacketType);
 }
 
-///////////////////////////////////////////////////////////////////////////
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::detail::PacketRegistryImpl<KeyType>:
 
 template <class KeyType>
 template <class PacketType>
-prefix_ void senf::detail::PacketRegistryImpl<KeyType>::registerPacket(key_t key)
+prefix_ void senf::detail::PacketRegistryImpl<KeyType>::registerPacket(key_t key, int priority)
 {
-#ifndef SENF_DEBUG
-    registry_.insert(std::make_pair(key, Entry_ptr(new detail::PkReg_EntryImpl<PacketType>())));
-    reverseRegistry_.insert(std::make_pair(senf::typeIdValue<PacketType>(), key));
-#else
-    bool isUnique (
+    SENF_ASSERT_EXPRESSION(
         registry_.insert(
-            std::make_pair(key, Entry_ptr(new detail::PkReg_EntryImpl<PacketType>()))).second);
-    // If this assertion fails, a Packet was registered with an already known key
-    SENF_ASSERT( isUnique );
-    bool isNew (
-        reverseRegistry_.insert(
-            std::make_pair(senf::typeIdValue<PacketType>(), key)).second);
-    // If this assertion fails, the same Packet was registered with two different keys
-    SENF_ASSERT( isNew );
-#endif
+            typename Entry::ptr(new EntryImpl<PacketType>(key,priority))).second,
+        "Duplicate packet registration");
+}
+
+template <class KeyType>
+template <class PacketType>
+prefix_ void senf::detail::PacketRegistryImpl<KeyType>::unregisterPacket()
+{
+    registry_.template get<ByType>().erase(typeid(PacketType));
+}
+
+template <class KeyType>
+prefix_ void senf::detail::PacketRegistryImpl<KeyType>::unregisterPacket(key_t key, int priority)
+{
+    // Why doesn't this work:
+    // registry_.erase(boost::make_tuple(key,priority));
+    typename Registry::iterator i (registry_.find(boost::make_tuple(key,priority)));
+    if (i != registry_.end())
+        registry_.erase(i);
 }
 
 template <class KeyType>
 prefix_ typename senf::detail::PacketRegistryImpl<KeyType>::key_t
 senf::detail::PacketRegistryImpl<KeyType>::key(senf::TypeIdValue const & type)
 {
-    typename ReversePacketMap::iterator i (reverseRegistry_.find(type));
-    if (i==reverseRegistry_.end())
+    boost::optional<KeyType> k (key(type,true));
+    if (! k)
         throw PacketTypeNotRegisteredException();
-    return i->second;
+    return *k;
 }
 
 template <class KeyType>
 prefix_ boost::optional<typename senf::detail::PacketRegistryImpl<KeyType>::key_t>
 senf::detail::PacketRegistryImpl<KeyType>::key(senf::TypeIdValue const & type, bool)
 {
-    typename ReversePacketMap::iterator i (reverseRegistry_.find(type));
-    if (i==reverseRegistry_.end())
+    typedef typename Registry::template index<ByType>::type TypeIndex;
+    TypeIndex const & typeIndex (registry_.template get<ByType>());
+    typename TypeIndex::const_iterator i (typeIndex.find(type.id()));
+    if (i == typeIndex.end())
         return boost::optional<key_t>();
-    return i->second;
+    return (*i)->key;
 }
 
 template <class KeyType>
 prefix_ typename senf::detail::PacketRegistryImpl<KeyType>::Entry const &
 senf::detail::PacketRegistryImpl<KeyType>::lookup(key_t key)
 {
-    typename PacketMap::iterator i (registry_.find(key));
-    if (i==registry_.end())
+    Entry const * e (lookup(key, true));
+    if (!e)
         throw PacketTypeNotRegisteredException();
-    return *(i->second);
+    return *e;
 }
 
 template <class KeyType>
 prefix_ typename senf::detail::PacketRegistryImpl<KeyType>::Entry const *
 senf::detail::PacketRegistryImpl<KeyType>::lookup(key_t key, bool)
 {
-    typename PacketMap::iterator i (registry_.find(key));
-    if (i==registry_.end())
+    typedef typename Registry::template index<ByKey>::type KeyIndex;
+    KeyIndex const & keyIndex (registry_.template get<ByKey>());
+    typename KeyIndex::const_iterator i (keyIndex.lower_bound(key));
+    if (i == keyIndex.end() || (*i)->key != key)
         return 0;
-    return i->second.get();
+    return i->get();
 }
 
 template <class KeyType>
@@ -139,12 +170,13 @@ template <class KeyType>
 prefix_ void senf::detail::PacketRegistryImpl<KeyType>::v_dump(std::ostream & os)
     const
 {
-    typename PacketMap::const_iterator i (registry_.begin());
-    typename PacketMap::const_iterator const i_end (registry_.end());
-    for (; i != i_end; ++i) {
-        std::string n (i->second->name());
-        senf::detail::DumpKey<KeyType>::dump(i->first, os);
-        os << ' ' << n.substr(21,n.size()-22) << "\n";
+    typedef typename Registry::template index<ByKey>::type KeyIndex;
+    KeyIndex const & keyIndex (registry_.template get<ByKey>());
+    for (typename KeyIndex::iterator i (keyIndex.begin()), i_end (keyIndex.end());
+         i != i_end; ++i) {
+        std::string n ((*i)->name());
+        senf::detail::DumpKey<KeyType>::dump((*i)->key, os);
+        os << ' ' << std::setw(6) << (*i)->priority << ' ' << n.substr(21,n.size()-22) << '\n';
     }
 }
 
@@ -152,10 +184,9 @@ template <class KeyType>
 prefix_ void senf::detail::PacketRegistryImpl<KeyType>::v_clear()
 {
     registry_.clear();
-    reverseRegistry_.clear();
 }
 
-///////////////////////////////////////////////////////////////////////////
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 // senf::detail::DumpKey<KeyType,is_integral>
 
 template <class KeyType, bool is_integral>
@@ -173,7 +204,7 @@ prefix_ void senf::detail::DumpKey<KeyType, true>::dump(KeyType const & v, std::
     os << "  " << senf::format::dumpint(v);
 }
 
-///////////////////////////////ct.e////////////////////////////////////////
+//-/////////////////////////////////////////////////////////////////////////////////////////////////
 #undef prefix_