// Copyright (C) 2001 Dominic Letourneau

#include "BufferedNode.h"
#include "operators.h"
#include "Stream.h"
#include "SocketStream.h"

using namespace std;

namespace FD {

class Socket;

DECLARE_NODE(Socket)
/*Node
 *
 * @name Socket
 * @category Network
 * @description Create a network socket of any type
 *
 * @parameter_name TYPE
 * @parameter_type string
 * @parameter_value TCP_STREAM
 * @parameter_description Type of socket : BROADCAST, TCP_STREAM, etc.
 *
 * @parameter_name PORT
 * @parameter_type int
 * @parameter_description Communication port
 *
 * @output_name OUTPUT
 * @output_description The socket to be used for input/output operations
 * @output_type socket
 *
END*/


class Socket : public BufferedNode {
   
  int outputID;

  ObjectRef m_socketRef;

public:

   Socket(string nodeName, ParameterSet params)
   : BufferedNode(nodeName, params) {

     const String &socketType = object_cast<String> (parameters.get("TYPE"));
     int port = dereference_cast<int> (parameters.get("PORT"));
     
     m_socketRef = nilObject;

     if (socketType == "BROADCAST") {

       socket_iostream *my_socket = new socket_iostream(network_socket::BROADCAST_TYPE, port);

       m_socketRef = ObjectRef(new IOStream(my_socket,true));

     }

     if (socketType == "TCP_STREAM") {

       socket_iostream *my_socket = new socket_iostream(network_socket::TCP_STREAM_TYPE, port);

       m_socketRef = ObjectRef(new IOStream(my_socket,true));

     }

     outputID = addOutput("OUTPUT");
   }

   void calculate(int output_id, int count, Buffer &out) {
     out[count] = m_socketRef;
   }

};

}//namespace FD