icestormelement.h

/*
 * Orca-Robotics Project: Components for robotics 
 *               http://orca-robotics.sf.net/
 * Copyright (c) 2004-2008 Alex Brooks, Alexei Makarenko, Tobias Kaupp
 *
 * This copy of Orca is licensed to you under the terms described in
 * the LICENSE file included in this distribution.
 *
 */

#ifndef ORCAGUI_ICESTORM_ELEMENT_H
#define ORCAGUI_ICESTORM_ELEMENT_H

#include <assert.h>
#include <Ice/Ice.h>
#include <IceStorm/IceStorm.h>
#include <orcaqguielementutil/icestormlistener.h>
#include <hydroqgui/hydroqgui.h>
#include <hydroqguielementutil/guielement2d.h>

namespace orcaqguielementutil {

template<class PainterType, class DataType, class ProxyType, class ConsumerType, class ConsumerPrxType>
class IceStormElement
{
  
public:

    IceStormElement( const orcaice::Context  &context,
                     const std::string       &proxyString,
                     PainterType             &painter,
                     const double            timeoutMs )
        : context_(context),
          listener_(context,proxyString),
          painter_(painter),
          timeoutMs_(timeoutMs),
          isConnected_(false)
    {};

    virtual ~IceStormElement() {}

    virtual void actionOnConnection()=0;

    void updateFromBuffer();

protected:

    orcaice::Context              context_;
    
    DataType                       data_;
    
    bool needToUpdate();
    
    orcaqguielementutil::IceStormListener<DataType,
                                          ProxyType,
                                          ConsumerType,
                                          ConsumerPrxType> listener_;

    PainterType                     &painter_;
    
    double                           timeoutMs_;
    bool                             isConnected_;
};

template<class PainterType, class DataType, class ProxyType, class ConsumerType, class ConsumerPrxType>
bool 
IceStormElement<PainterType,DataType,ProxyType,ConsumerType,ConsumerPrxType>::needToUpdate()
{
    if ( !isConnected_ )
    {
        if ( listener_.connect() == 0 )
        {
            isConnected_ = true;
            assert( listener_.proxy() != 0 );
            actionOnConnection();
        }
        return false;
    }

    if ( !listener_.buffer().isEmpty() )
    {
        // An object has arrived in our buffer.  We need to update.
        return true;
    }

    // The buffer is empty.  How long since we last received something?
    if ( timeoutMs_ != -1 && listener_.msSinceReceipt() >= timeoutMs_ )
    {
        std::cout<<"TRACE(icestormelement.h): Haven't received anything from " 
                 << listener_.interfaceName() << " for " << listener_.msSinceReceipt() << "ms" << std::endl;
        std::cout<<"TRACE(icestormelement.h): Timing out..." << std::endl;

        painter_.clear();
        listener_.resetTimer();
        if ( listener_.connect() == 0 )
            actionOnConnection();
        else
            isConnected_ = false;
    }
    return false;
}

template<class PainterType, class DataType, class ProxyType, class ConsumerType, class ConsumerPrxType>
void 
IceStormElement<PainterType,DataType,ProxyType,ConsumerType,ConsumerPrxType>::updateFromBuffer()
{
    if ( !needToUpdate() ) {
        return;
    }
    
    assert( !listener_.buffer().isEmpty() );

    // get data from the buffer
    listener_.buffer().getAndPop( data_ );

    // transfer data into painter
    painter_.setData( data_ );
}

} // namespace

#endif