dataStream.cpp

/* Copyright (c) 2007-2012, Stefan Eilemann <eile@equalizergraphics.com>
 *                    2012, Daniel Nachbaur <danielnachbaur@gmail.com>
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License version 2.1 as published
 * by the Free Software Foundation.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
 * details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <test.h>

#include <co/dataIStream.h>
#include <co/dataOStream.h>

#include <co/buffer.h>
#include <co/bufferCache.h>
#include <co/connectionDescription.h>
#include <co/command.h>
#include <co/commandQueue.h>
#include <co/connection.h>
#include <co/init.h>
#include <co/types.h>

#include <lunchbox/thread.h>

#include <co/objectDataOCommand.h> // private header
#include <co/objectDataCommand.h> // private header
#include <co/cpuCompressor.h> // private header

// Tests the functionality of the DataOStream and DataIStream

#define CONTAINER_SIZE LB_64KB

static std::string _message( "So long, and thanks for all the fish" );

class DataOStream : public co::DataOStream
{
public:
    DataOStream() {}

protected:
    virtual void sendData( const void* buffer, const uint64_t size,
                           const bool last )
        {
            co::ObjectDataOCommand( getConnections(), co::CMD_OBJECT_DELTA,
                                    co::COMMANDTYPE_CO_OBJECT, co::UUID(), 0,
                                    co::uint128_t(), 0, size, last, this );
        }
};

class DataIStream : public co::DataIStream
{
public:
    void addDataCommand( co::ConstBufferPtr buffer )
        {
            co::ObjectDataCommand command( buffer );
            TESTINFO( command.getCommand() == co::CMD_OBJECT_DELTA, command );
            _commands.push( buffer );
        }

    virtual size_t nRemainingBuffers() const { return _commands.getSize(); }
    virtual lunchbox::uint128_t getVersion() const { return co::VERSION_NONE;}
    virtual co::NodePtr getMaster() { return 0; }

protected:
    virtual bool getNextBuffer( uint32_t& compressor, uint32_t& nChunks,
                                const void** chunkData, uint64_t& size )
        {
            co::Command cmd = _commands.tryPop();
            if( !cmd.isValid( ))
                return false;

            co::ObjectDataCommand command( cmd );

            TEST( command.getCommand() == co::CMD_OBJECT_DELTA );

            size = command.getDataSize();
            compressor = command.getCompressor();
            nChunks = command.getChunks();
            *chunkData = command.getRemainingBuffer( size );
            return true;
        }

private:
    co::CommandQueue _commands;
};

namespace co
{
namespace DataStreamTest
{
class Sender : public lunchbox::Thread
{
public:
    Sender( lunchbox::RefPtr< co::Connection > connection )
            : Thread(),
              _connection( connection )
        {
            TEST( connection );
            TEST( connection->isConnected( ));
        }
    virtual ~Sender(){}

protected:
    virtual void run()
        {
            ::DataOStream stream;

            stream._setupConnection( _connection );
            stream._enable();

            int foo = 42;
            stream << foo;
            stream << 43.0f;
            stream << 44.0;

            std::vector< double > doubles;
            for( size_t i=0; i<CONTAINER_SIZE; ++i )
                doubles.push_back( static_cast< double >( i ));

            stream << doubles;
            stream << _message;

            stream.disable();
        }

private:
    lunchbox::RefPtr< co::Connection > _connection;
};
}
}

int main( int argc, char **argv )
{
    co::init( argc, argv );
    co::ConnectionDescriptionPtr desc = new co::ConnectionDescription;
    desc->type = co::CONNECTIONTYPE_PIPE;
    co::ConnectionPtr connection = co::Connection::create( desc );

    TEST( connection->connect( ));
    TEST( connection->isConnected( ));
    co::DataStreamTest::Sender sender( connection->acceptSync( ));
    TEST( sender.start( ));

    ::DataIStream stream;
    co::BufferCache bufferCache;
    bool receiving = true;

    while( receiving )
    {
        uint64_t size;
        connection->recvNB( &size, sizeof( size ));
        TEST( connection->recvSync( 0, 0 ));
        TEST( size );

        co::BufferPtr buffer = bufferCache.alloc( 0, 0, size );
        connection->recvNB( buffer->getData(), size );
        TEST( connection->recvSync( 0, 0 ) );
        TEST( buffer->isValid( ));

        co::ObjectDataCommand command( buffer );
        switch( command.getCommand( ))
        {
            case co::CMD_OBJECT_DELTA:
            {
                stream.addDataCommand( buffer );
                TEST( !buffer->isFree( ));
                receiving = !command.isLast();
                break;
            }
            default:
                TEST( false );
        }
    }

    int foo;
    stream >> foo;
    TESTINFO( foo == 42, foo );

    float fFoo;
    stream >> fFoo;
    TEST( fFoo == 43.f );

    double dFoo;
    stream >> dFoo;
    TEST( dFoo == 44.0 );

    std::vector< double > doubles;
    stream >> doubles;
    for( size_t i=0; i<CONTAINER_SIZE; ++i )
        TEST( doubles[i] == static_cast< double >( i ));

    std::string message;
    stream >> message;
    TEST( message.length() == _message.length() );
    TESTINFO( message == _message,
              '\'' <<  message << "' != '" << _message << '\'' );

    TEST( sender.join( ));
    connection->close();
    return EXIT_SUCCESS;
}