CSource.cpp

Go to the documentation of this file.

#include "stdafx.h"

namespace AxPipe {
    CSource::~CSource() {
        Plug();                             // Safety first Plug(), is ok to call many times.
    }
    CSource *
    CSource::Append(CSink *pSink) {
        CPipe::Append(pSink);
        return this;
    }

    CSource *
    CSource::Append(CSink& sink) {
        AppendSink(&sink, false);
        return this;
    }
 
    CSource *
    CSource::Open() {
        if (!m_fIsOpen) {
            OutPump((new CSeg)->SetType(eSegTypeOpen));
            m_fIsOpen = true;
        }
        return this;
    }
    
    CSource *
    CSource::Close() {
        if (m_fIsOpen) {
            OutPump((new CSeg)->SetType(eSegTypeClose));
            m_fIsOpen = false;
        }
        return this;
    }

    CSource *
    CSource::Drain() {
        ASSCHK(nGlobalInit != 0, _T("AxPipe::CGlobalInit object must exist"));

        if (!m_fIsOpen) {
            SetError(ERROR_CODE_NOTOPEN, ERROR_MSG_NOTOPEN);
            return this;
        }
        while (WorkStart(), !GetErrorCode()) {
            if (((m_pSeg = In()) != NULL) && (m_pSeg->Len() != 0)) {
                WorkSignal();
                // Always return with m_pSeg == NULL and WorkEnd() called
            } else {
                break;
            }
        }
        // If we have a segment, it has to be a zero-length segment we ignore.
        if (m_pSeg) {
            m_pSeg->Release();
            m_pSeg = NULL;
        }
        // At this point, WorkStart() is called, and WorkEnd() must be too.
        WorkEnd();
        return this;
    }

    CSource *
    CSource::Plug() {
        if (!m_fExit) {
            OutPump((new CSeg)->SetType(eSegTypePlug));
        }
        return this;
    }

    void
    CSource::Out(CSeg *pSeg) {
        if (m_pSink) {
            m_pSink->OutPump(pSeg);
        } else {
            pSeg->Release();
        }
    }
};

---