syndilights/open-lighting-architecture/ola-0.8.4/ola/StreamingClient.cpp

157 lines
3.6 KiB
C++
Raw Normal View History

2010-10-31 17:32:25 +01:00
/*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* StreamingClient.cpp
* Implementation of Streaming Client
* Copyright (C) 2005-2008 Simon Newton
*/
#include <ola/BaseTypes.h>
#include <ola/Closure.h>
#include <ola/DmxBuffer.h>
#include <ola/Logging.h>
#include <ola/StreamingClient.h>
#include "common/protocol/Ola.pb.h"
#include "common/rpc/StreamRpcChannel.h"
namespace ola {
using ola::rpc::StreamRpcChannel;
using ola::proto::OlaServerService_Stub;
StreamingClient::StreamingClient()
: m_socket(NULL),
m_ss(NULL),
m_channel(NULL),
m_stub(NULL),
m_socket_closed(false) {
}
StreamingClient::~StreamingClient() {
Stop();
}
/*
* Setup the Streaming Client
* @returns true on success, false on failure
*/
bool StreamingClient::Setup() {
if (m_socket || m_channel || m_stub)
return false;
m_socket = TcpSocket::Connect("127.0.0.1", OLA_DEFAULT_PORT);
if (!m_socket)
return false;
m_ss = new SelectServer();
m_ss->AddSocket(m_socket);
m_channel = new StreamRpcChannel(NULL, m_socket);
if (!m_channel) {
delete m_socket;
m_socket = NULL;
return false;
}
m_stub = new OlaServerService_Stub(m_channel);
if (!m_stub) {
delete m_channel;
delete m_socket;
m_channel = NULL;
m_socket = NULL;
return false;
}
m_socket->SetOnClose(
NewSingleClosure(this, &StreamingClient::SocketClosed));
m_channel->SetOnClose(
NewSingleClosure(this, &StreamingClient::SocketClosed));
return true;
}
/*
* Close the ola connection.
*/
void StreamingClient::Stop() {
if (m_stub)
delete m_stub;
if (m_channel)
delete m_channel;
if (m_ss)
delete m_ss;
if (m_socket)
delete m_socket;
m_channel = NULL;
m_socket = NULL;
m_ss = NULL;
m_stub = NULL;
}
/*
* Send DMX to the remote OLA server
* @returns True is sent sucessfully, false if the connection to the server has
* been closed and Setup() needs to be run again.
*/
bool StreamingClient::SendDmx(unsigned int universe,
const DmxBuffer &data) {
if (!m_stub ||
m_socket->ReadDescriptor() == ola::network::Socket::INVALID_SOCKET)
return false;
// We select() on the fd here to see if the remove end has closed the
// connection. We could skip this and rely on the EPIPE delivered by the
// write() below, but that introduces a race condition in the unittests.
m_socket_closed = false;
m_ss->RunOnce(0, 0);
if (m_socket_closed) {
Stop();
return false;
}
ola::proto::DmxData request;
request.set_universe(universe);
request.set_data(data.Get());
m_stub->StreamDmxData(NULL, &request, NULL, NULL);
if (m_socket_closed) {
Stop();
return false;
}
return true;
}
/*
* Called when the socket is closed
*/
void StreamingClient::SocketClosed() {
m_socket_closed = true;
OLA_WARN << "The RPC socket has been closed, this is more than likely due"
<< " to a framing error, perhaps you're sending too fast?";
}
} // ola