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

   1 /* -*- C++ -*- */
   2
   3 //=============================================================================
   4 /**
   5  *  @file    receiver.h
   6  *
   7  *  This application receives data from a AV sender and writes it to
   8  *  a file.
   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/Policy.h"
  19 #include "orbsvcs/AV/Protocol_Factory.h"
  20
  21 /**
  22  * @class Receiver_Callback
  23  *
  24  * @brief Application defined callback object.
  25  *
  26  * AVStreams calls this class when data shows up from a sender.
  27  */
  28 class Receiver_Callback : public TAO_AV_Callback
  29 {
  30 public:
  31   /// Constructor.
  32   Receiver_Callback ();
  33
  34   // Method that is called when there is data to be received from a
  35   // sender.
  36   int receive_frame (ACE_Message_Block *frame,
  37                      TAO_AV_frame_info *frame_info,
  38                      const ACE_Addr &peer_address);
  39
  40   // Called when the sender is done sending data and wants to close
  41   // down the connection.
  42   int handle_destroy ();
  43
  44 private:
  45   /// Keeping a count of the incoming frames.
  46   int frame_count_;
  47
  48   /// Message block into which data is read from a file and then sent.
  49   ACE_Message_Block mb_;
  50 };
  51
  52 /**
  53  * @class Receiver_StreamEndPoint
  54  *
  55  * @brief Application defined stream endpoint object.
  56  *
  57  * AVStreams calls this class during connection setup.
  58  */
  59 class Receiver_StreamEndPoint : public TAO_Server_StreamEndPoint
  60 {
  61 public:
  62   // Create a receiver application callback.
  63   int get_callback (const char *flowname,
  64                     TAO_AV_Callback *&callback);
  65
  66   /// Set protocol object corresponding to the transport protocol
  67   /// chosen.
  68   int set_protocol_object (const char *flowname,
  69                            TAO_AV_Protocol_Object *object);
  70
  71
  72 private:
  73   /// Receiver application callback.
  74   Receiver_Callback callback_;
  75 };
  76
  77 /**
  78  * @class Receiver
  79  *
  80  * @brief Receiver application class.
  81  *
  82  * This class receives data from a AV sender and writes it to
  83  * a file.
  84  */
  85 class Receiver
  86 {
  87 public:
  88   /// Constructor
  89   Receiver ();
  90
  91   /// Destructor.
  92   ~Receiver ();
  93
  94   /// Initialize data components.
  95   int init (int argc,
  96             ACE_TCHAR *argv[]);
  97
  98   /// Set the protocol object corresponding to the transport protocol chosen.
  99   void protocol_object (TAO_AV_Protocol_Object *protocol_object);
 100
 101   /// Set the protocol object corresponding to the transport protocol chosen.
 102   TAO_AV_Protocol_Object * protocol_object ();
 103
 104   FILE * input_file ();
 105
 106 protected:
 107   /// The Naming Service Client.
 108   TAO_Naming_Client naming_client_;
 109
 110   /// The endpoint reactive strategy.
 111   TAO_AV_Endpoint_Reactive_Strategy_B
 112   <Receiver_StreamEndPoint,TAO_VDev,AV_Null_MediaCtrl> reactive_strategy_;
 113
 114   /// Receiver MMDevice.
 115   TAO_MMDevice *mmdevice_;
 116
 117   /// Rate at which the data will be sent.
 118   int frame_rate_;
 119
 120   /// Protocol object corresponding to the transport protocol selected.
 121   TAO_AV_Protocol_Object *protocol_object_;
 122
 123   /// File handle of the file read from.
 124   FILE *input_file_;
 125
 126   /// Number of frames sent.
 127   int frame_count_;
 128
 129   /// File from which data is read.
 130   ACE_CString filename_;
 131
 132   /// Method to parse the command line arguments.
 133   int parse_args (int argc, ACE_TCHAR *argv[]);
 134
 135   /// Message block into which data is read from a file and then sent.
 136   ACE_Message_Block mb_;
 137 };