moses-smt/mosesdecoder
1
1
Fork 0
mirror of https://github.com/moses-smt/mosesdecoder.git synced 2024-08-16 15:00:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Restore readability.

Browse Source
This commit is contained in:
Ulrich Germann 2015-08-02 21:22:52 +01:00
parent faaf0bdf87
commit b7f517bafa
1 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
moses/thread_safe_container.h
View File

@ -23,7 +23,7 @@ namespace Moses
template<typename KEY, typename VAL, class CONTAINER = std::map<KEY,VAL> > template<typename KEY, typename VAL, class CONTAINER = std::map<KEY,VAL> >
class class
ThreadSafeContainer ThreadSafeContainer
{ {
protected: protected:
mutable boost::shared_mutex m_lock; mutable boost::shared_mutex m_lock;
@ -49,28 +49,33 @@ public:
: m_lock(lock), m_container(container), m_iter(iter) : m_lock(lock), m_container(container), m_iter(iter)
{ } { }
entry_t const& operator->() { entry_t const& operator->()
{
UTIL_THROW_IF2(m_container == NULL, "This locking iterator is invalid " UTIL_THROW_IF2(m_container == NULL, "This locking iterator is invalid "
<< "or has not been assigned."); << "or has not been assigned.");
return m_iter.operator->(); return m_iter.operator->();
} }
// locking operators transfer the lock upon assignment and become invalid // locking operators transfer the lock upon assignment and become
// invalid
locking_iterator const& locking_iterator const&
operator=(locking_iterator& other) { operator=(locking_iterator& other)
{
m_lock.swap(other.m_lock); m_lock.swap(other.m_lock);
m_iter = other.m_iter; m_iter = other.m_iter;
other.m_iter = other.m_container.end(); other.m_iter = other.m_container.end();
} }
bool bool
operator==(const_iter_t const& other) { operator==(const_iter_t const& other)
{
return m_iter == other; return m_iter == other;
} }
locking_iterator const& locking_iterator const&
operator++() { operator++()
++m_iter; {
++m_iter;
return *this; return *this;
} }
@ -82,15 +87,21 @@ public:
operator++(int); operator++(int);
}; };
const_iter_t const& end() const { const_iter_t const&
end() const
{
return m_container.end(); return m_container.end();
} }
locking_iterator begin() const { locking_iterator
begin() const
{
return locking_iterator(m_lock, this, m_container.begin()); return locking_iterator(m_lock, this, m_container.begin());
} }
VAL const& set(KEY const& key, VAL const& val) { VAL const&
set(KEY const& key, VAL const& val)
{
boost::unique_lock< boost::shared_mutex > lock(m_lock); boost::unique_lock< boost::shared_mutex > lock(m_lock);
entry_t entry(key,val); entry_t entry(key,val);
iter_t foo = m_container.insert(entry).first; iter_t foo = m_container.insert(entry).first;
@ -98,21 +109,27 @@ public:
return foo->second; return foo->second;
} }
VAL const* get(KEY const& key, VAL const& default_val) { VAL const*
get(KEY const& key, VAL const& default_val)
{
boost::unique_lock< boost::shared_mutex > lock(m_lock); boost::unique_lock< boost::shared_mutex > lock(m_lock);
entry_t entry(key, default_val); entry_t entry(key, default_val);
iter_t foo = m_container.insert(entry).first; iter_t foo = m_container.insert(entry).first;
return &(foo->second); return &(foo->second);
} }
VAL const* get(KEY const& key) const { VAL const*
get(KEY const& key) const
{
boost::shared_lock< boost::shared_mutex > lock(m_lock); boost::shared_lock< boost::shared_mutex > lock(m_lock);
const_iter_t m = m_container.find(key); const_iter_t m = m_container.find(key);
if (m == m_container.end()) return NULL; if (m == m_container.end()) return NULL;
return &m->second; return &m->second;
} }
size_t erase(KEY const& key) { size_t
erase(KEY const& key)
{
boost::unique_lock< boost::shared_mutex > lock(m_lock); boost::unique_lock< boost::shared_mutex > lock(m_lock);
return m_container.erase(key); return m_container.erase(key);
} }