Ticket #1330: repeated-PS-HM-16.1.patch

source/Lib/TLibCommon/TComSlice.h

@@ -1522 +1522 @@
 
   ~ParameterSetMap()
   {
-    for (typename std::map<Int,T *>::iterator i = m_paramsetMap.begin(); i!= m_paramsetMap.end(); i++)
+    for (typename std::map<Int, std::vector<T *> >::iterator i = m_paramsetMap.begin(); i!= m_paramsetMap.end(); i++)
     {
-      delete (*i).second;
+      for (typename std::vector<T *>::iterator i2 = (*i).second.begin(); i2!= (*i).second.end(); ++i2) {
+        delete *i2;
+      }
     }
   }
 
   Void storePS(Int psId, T *ps)
   {
     assert ( psId < m_maxId );
-    if ( m_paramsetMap.find(psId) != m_paramsetMap.end() )
-    {
-      delete m_paramsetMap[psId];
-    }
-    m_paramsetMap[psId] = ps;
+    m_paramsetMap[psId].push_back(ps);
   }
 
   Void mergePSList(ParameterSetMap<T> &rPsList)
   {
-    for (typename std::map<Int,T *>::iterator i = rPsList.m_paramsetMap.begin(); i!= rPsList.m_paramsetMap.end(); i++)
+    for (typename std::map<Int, std::vector<T *> >::iterator i = rPsList.m_paramsetMap.begin(); i!= rPsList.m_paramsetMap.end(); i++)
     {
-      storePS(i->first, i->second);
+      size_t sz1=m_paramsetMap[i->first].size();
+      size_t sz2=i->second.size();
+      m_paramsetMap[i->first].resize(sz1+sz2);
+      memcpy(&(m_paramsetMap[i->first][sz1]), &(i->second[0]), sz2*sizeof(T*));
     }
     rPsList.m_paramsetMap.clear();
   }
@@ -1550 +1551 @@
 
   T* getPS(Int psId)
   {
-    return ( m_paramsetMap.find(psId) == m_paramsetMap.end() ) ? NULL : m_paramsetMap[psId];
+    return ( m_paramsetMap.find(psId) == m_paramsetMap.end() ) ? NULL : m_paramsetMap[psId].back();
   }
 
   T* getFirstPS()
   {
-    return (m_paramsetMap.begin() == m_paramsetMap.end() ) ? NULL : m_paramsetMap.begin()->second;
+    return (m_paramsetMap.begin() == m_paramsetMap.end() ) ? NULL : m_paramsetMap.begin()->second.back();
   }
 
 private:
-  std::map<Int,T *> m_paramsetMap;
+  std::map<Int, std::vector<T *> > m_paramsetMap;
   Int               m_maxId;
 };