include/pwvstream.h

/* -*- Mode: C++ -*-
 * Worldvisions Weaver Software:
 *   Copyright (C) 1997-2002 Net Integration Technologies, Inc.
 */ 
#ifndef __PWVSTREAM_H
#define __PWVSTREAM_H
 
#include "iwvstream.h"
#include "wvstream.h"      // FIXME: don't include this!
#include "wvstreamclone.h" // FIXME: don't include this!
#include "wvmoniker.h"
#include "wvtr1.h"

class PWvStream : public wv::shared_ptr<WvStream>
{
    static WvStream *clean_stream(IWvStream *s)
    {
        WvStream *ss = dynamic_cast<WvStream *>(s);
        if (ss)
            return ss;
        else
            return new WvStreamClone(s);
    }
    
    static WvStream *make_stream(WvStringParm moniker, IObject *obj)
    {
        IWvStream *s = IWvStream::create(moniker, obj);
        if (!s)
            s = wvcreate<IWvStream>("null:");
        assert(s != NULL);
        return clean_stream(s);
    }
    
public:
    PWvStream()
    {
        // Pointer points to NULL right now, but it could be reassigned
        // later using operator=().
    }
    
    PWvStream(IWvStream *s)
        : wv::shared_ptr<WvStream>(clean_stream(s),
                                  wv::bind(&IWvStream::release, _1))
    {
    }
    
    PWvStream(WvStringParm moniker, IObject *obj = 0)
        : wv::shared_ptr<WvStream>(make_stream(moniker, obj),
                                  wv::bind(&IWvStream::release, _1))
    {
        // Note: pointer is definitely not NULL here, because make_stream is
        // careful.
    }
    
    WvStream *addRef() const
    {
        if (get())
            get()->addRef();
        return get();
    }
};
 
#endif // __PWVSTREAM_H