Changeset 7139

Timestamp:

07/19/07 16:38:46

Author:

robert

Message:

Added BlockCount?

Files:

• OpenThreads/trunk/include/OpenThreads/Block (modified) (2 diffs)

OpenThreads/trunk/include/OpenThreads/Block

@@ -22 +22 @@
-namespace OpenThreads {
-
+/** Block is a block that can be used to halt a thread that is waiting another thread to release it.*/
-class Block 
-{
@@ -92 +93 @@
-};
-
+/** BlockCount is a block that can be used to halt a thread that is waiting for a specified number of operations to be completed.*/
+class BlockCount 
+{
+    public:
+    
+        BlockCount(unsigned int blockCount):
+            _blockCount(blockCount),
+            _currentCount(0) {}
+
+        ~BlockCount()
+        {
+            _blockCount = 0;
+            release();
+        }
+
+        inline void completed()
+        {
+            OpenThreads::ScopedLock<OpenThreads::Mutex> mutlock(_mut);
+            if (_currentCount>0)
+            {
+                --_currentCount;
+
+                if (_currentCount==0)
+                {
+                    // osg::notify(osg::NOTICE)<<"Released"<<std::endl;
+                    _cond.broadcast();
+                }
+            }
+        }
+
+        inline void block()
+        {
+            OpenThreads::ScopedLock<OpenThreads::Mutex> mutlock(_mut);
+            if (_currentCount)
+                _cond.wait(&_mut);
+        }
+        
+        inline void reset()
+        {
+            OpenThreads::ScopedLock<OpenThreads::Mutex> mutlock(_mut);
+            if (_currentCount!=_blockCount)
+            {
+                if (_blockCount==0) _cond.broadcast();
+                _currentCount = _blockCount;
+            }
+        }
+
+        inline void release()
+        {
+            OpenThreads::ScopedLock<OpenThreads::Mutex> mutlock(_mut);
+            if (_currentCount)
+            {
+                _currentCount = 0;
+                _cond.broadcast();
+            }
+        }
+
+        inline void setBlockCount(unsigned int blockCount) { _blockCount = blockCount; }
+        
+        inline unsigned int getBlockCount() const { return _blockCount; }
+        
+        inline unsigned int getCurrentCount() const { return _currentCount; }
+
+    protected:
+
+
+        OpenThreads::Mutex _mut;
+        OpenThreads::Condition _cond;
+        unsigned int _blockCount;
+        unsigned int _currentCount;
+};
+
-}
-