Bu::StreamStack works, it's tested, reasonably, it will be used first in the

gats project in Gats::ProtocolGats.

1 files changed, 88 insertions, 1 deletions

diff --git a/src/streamstack.h b/src/streamstack.h
index 03d8d8f..89e683d 100644
--- a/src/streamstack.h
+++ b/src/streamstack.h
@@ -3,14 +3,99 @@
 
 #include "bu/stream.h"
 
+#include <typeinfo>
+
 namespace Bu
 {
         class StreamStack : public Bu::Stream
         {
+        private:
+                typedef Bu::List<Bu::Stream *> FilterList;
+
         public:
                 StreamStack();
+                StreamStack( Bu::Stream *pStream );
                 virtual ~StreamStack();
 
+                bool isEmpty();
+                bool hasStream();
+                void setStream( Bu::Stream *pStream );
+
+                void clearStack();
+                void popFilter();
+                Bu::Stream *getTop();
+
+                Bu::Stream *getStream();
+
+                template<typename filter>
+                Bu::Stream *findFilter()
+                {
+                        for( FilterList::iterator i = lFilts.begin(); i; i++ )
+                        {
+                                if( typeid(**i) == typeid( filter ) )
+                                {
+                                        return *i;
+                                }
+                        }
+
+                        throw Bu::ExceptionBase("Filter not found.");
+                }
+
+                template<typename filter>
+                void pushFilter()
+                {
+                        checkStack();
+
+                        filter *pFlt = new filter( *lFilts.first() );
+                        lFilts.prepend( pFlt );
+                }
+
+                template<typename filter, typename p1t>
+                void pushFilter( p1t p1 )
+                {
+                        checkStack();
+
+                        filter *pFlt = new filter( *lFilts.first(), p1 );
+                        lFilts.prepend( pFlt );
+                }
+
+                template<typename filter, typename p1t, typename p2t>
+                void pushFilter( p1t p1, p2t p2 )
+                {
+                        checkStack();
+
+                        filter *pFlt = new filter( *lFilts.first(), p1, p2 );
+                        lFilts.prepend( pFlt );
+                }
+
+                template<typename filter, typename p1t, typename p2t, typename p3t>
+                void pushFilter( p1t p1, p2t p2, p3t p3 )
+                {
+                        checkStack();
+
+                        filter *pFlt = new filter( *lFilts.first(), p1, p2, p3 );
+                        lFilts.prepend( pFlt );
+                }
+
+                template<typename filter, typename p1t, typename p2t, typename p3t,
+                        typename p4t>
+                void pushFilter( p1t p1, p2t p2, p3t p3, p4t p4 )
+                {
+                        checkStack();
+
+                        filter *pFlt = new filter( *lFilts.first(), p1, p2, p3, p4 );
+                        lFilts.prepend( pFlt );
+                }
+
+                template<typename filter, typename p1t, typename p2t, typename p3t,
+                        typename p4t, typename p5t>
+                void pushFilter( p1t p1, p2t p2, p3t p3, p4t p4, p5t p5 )
+                {
+                        checkStack();
+
+                        filter *pFlt = new filter( *lFilts.first(), p1, p2, p3, p4, p5 );
+                        lFilts.prepend( pFlt );
+                }
 
                 //
                 // Everything below here merely passes on the call to the top of the
@@ -39,7 +124,9 @@ namespace Bu
                 virtual void setSize( long iSize );
 
         private:
-                typedef Bu::List<Bu::Stream *> FilterList;
+                void checkStack();
+
+        private:
                 FilterList lFilts;
         };
 };