TAO/orbsvcs/tests/AVStreams/Bidirectional_Flows/sender.h

   1 /* -*- C++ -*- */
   2
   3 //=============================================================================
   4 /**
   5  *  @file    sender.h
   6  *
   7  *  This application reads data from a file and sends it to s
   8  *  receiver.
   9  *
  10  *  @author Yamuna Krishnamurthy <yamuna@cs.wustl.edu>
  11  */
  12 //=============================================================================
  13
  14
  15 #include "orbsvcs/Naming/Naming_Client.h"
  16 #include "orbsvcs/AV/AVStreams_i.h"
  17 #include "orbsvcs/AV/Endpoint_Strategy.h"
  18 #include "orbsvcs/AV/Protocol_Factory.h"
  19
  20 /**
  21  * @class Sender_Callback
  22  *
  23  * @brief Application defined callback object.
  24  *
  25  * AVStreams calls this class when data shows up from a sender.
  26  */
  27 class Sender_Callback : public TAO_AV_Callback
  28 {
  29 public:
  30   /// Constructor.
  31   Sender_Callback ();
  32
  33   // Method that is called when there is data to be received from a
  34   // sender.
  35   int receive_frame (ACE_Message_Block *frame,
  36                      TAO_AV_frame_info *frame_info,
  37                      const ACE_Addr &peer_address);
  38
  39   // Called when the sender is done sending data and wants to close
  40   // down the connection.
  41   //  int handle_destroy ();
  42
  43 private:
  44   /// Keeping a count of the incoming frames.
  45   int frame_count_;
  46 };
  47
  48 /**
  49  * @class Sender_StreamEndPoint
  50  *
  51  * @brief Defines a sender stream endpoint.
  52  */
  53 class Sender_StreamEndPoint : public TAO_Client_StreamEndPoint
  54 {
  55 public:
  56   /// Create the application callback and return its handle to
  57   /// AVStreams for further application callbacks.
  58   int get_callback (const char *flowname,
  59                     TAO_AV_Callback *&callback);
  60
  61   /// Set protocol object corresponding to the transport protocol
  62   /// chosen.
  63   int set_protocol_object (const char *flowname,
  64                            TAO_AV_Protocol_Object *object);
  65
  66 protected:
  67   /// Application callback.
  68   Sender_Callback callback_;
  69 };
  70
  71 typedef TAO_AV_Endpoint_Reactive_Strategy_A
  72           <Sender_StreamEndPoint,
  73            TAO_VDev,
  74            AV_Null_MediaCtrl>
  75         SENDER_ENDPOINT_STRATEGY;
  76
  77 /**
  78  * @class Sender
  79  *
  80  * @brief Sender Application.
  81  *
  82  * Class is responsible for streaming (and pacing) data to a
  83  * receiver.
  84  */
  85 class Sender
  86 {
  87 public:
  88   /// Constructor
  89   Sender ();
  90
  91   /// Method to initialize the various data components.
  92   int init (int argc,
  93             ACE_TCHAR *argv[]);
  94
  95   /// Method to pace and send data from a file.
  96   int pace_data ();
  97
  98   /// Set the protocol object corresponding to the transport protocol chosen.
  99   void protocol_object (TAO_AV_Protocol_Object *protocol_object);
 100
 101   int eof ();
 102
 103   void shutdown ();
 104
 105 private:
 106   /// Method to parse the command line arguments.
 107   int parse_args (int argc, ACE_TCHAR *argv[]);
 108
 109   /// Method that binds the sender to the receiver.
 110   int bind_to_receiver ();
 111
 112   /// The endpoint strategy used by the sender.
 113   SENDER_ENDPOINT_STRATEGY endpoint_strategy_;
 114
 115   /// The receiver MMDevice that the sender connects to.
 116   AVStreams::MMDevice_var receiver_mmdevice_;
 117
 118   /// The sender MMDevice.
 119   TAO_MMDevice *sender_mmdevice_;
 120
 121   /// Stream controller
 122   TAO_StreamCtrl *streamctrl_;
 123
 124   /// Number of frames sent.
 125   int frame_count_;
 126
 127   /// File from which data is read.
 128   ACE_CString filename_;
 129
 130   /// The Naming Service client.
 131   TAO_Naming_Client naming_client_;
 132
 133   /// File handle of the file read from.
 134   FILE *input_file_;
 135
 136   /// Selected protocol - default is UDP
 137   ACE_CString protocol_;
 138
 139   /// Rate at which the data will be sent.
 140   int frame_rate_;
 141
 142   /// Message block into which data is read from a file and then sent.
 143   ACE_Message_Block mb_;
 144
 145   /// Protocol object corresponding to the transport protocol selected.
 146   TAO_AV_Protocol_Object *protocol_object_;
 147
 148   int eof_;
 149 };