PortWriterBuffer.h

Go to the documentation of this file.

/*
 * Copyright (C) 2006-2020 Istituto Italiano di Tecnologia (IIT)
 * Copyright (C) 2006-2010 RobotCub Consortium
 * All rights reserved.
 *
 * This software may be modified and distributed under the terms of the
 * BSD-3-Clause license. See the accompanying LICENSE file for details.
 */
 
#ifndef YARP_OS_PORTWRITERBUFFER_H
#define YARP_OS_PORTWRITERBUFFER_H
 
#include <yarp/os/Portable.h>
#include <yarp/os/PortWriterBufferBase.h>
 
namespace yarp {
namespace os {
 
class Port;
 
#ifndef DOXYGEN_SHOULD_SKIP_THIS
 
template <class T>
class PortWriterBufferAdaptor : public PortWriterWrapper
{
public:
    PortWriterBufferManager& creator;
    T writer;
    void* tracker;
 
    PortWriterBufferAdaptor(PortWriterBufferManager& creator,
                            void* tracker) :
            creator(creator),
            tracker(tracker)
    {
    }
 
    bool write(ConnectionWriter& connection) const override
    {
        return writer.write(connection);
    }
 
    void onCompletion() const override
    {
        writer.onCompletion();
        creator.onCompletion(tracker);
    }
 
    void onCommencement() const override
    {
        writer.onCommencement();
    }
 
    PortWriter* getInternal() override
    {
        return &writer;
    }
};
 
#endif // DOXYGEN_SHOULD_SKIP_THIS
 
 
template <class T>
class PortWriterBuffer : public PortWriterBufferBase
{
public:
    T& prepare()
    {
        return get();
    }
 
    bool unprepare()
    {
        return releaseContent();
    }
 
    T& get()
    {
        PortWriterBufferAdaptor<T>* content = (PortWriterBufferAdaptor<T>*)getContent(); // guaranteed to be non-NULL
        return content->writer;
    }
 
    int getCount()
    {
        return PortWriterBufferBase::getCount();
    }
 
    void attach(Port& port)
    {
        PortWriterBufferBase::attach(port);
    }
 
    void write(bool forceStrict = false)
    {
        PortWriterBufferBase::write(forceStrict);
    }
 
    void waitForWrite()
    {
        PortWriterBufferBase::waitForWrite();
    }
 
    virtual PortWriterWrapper* create(PortWriterBufferManager& man,
                                      void* tracker) override
    {
        return new PortWriterBufferAdaptor<T>(man, tracker);
    }
};
 
} // namespace os
} // namespace yarp
 
#endif // YARP_OS_PORTWRITERBUFFER_H