TAO/orbsvcs/tests/AVStreams/Component_Switching/distributer.h

   1 /* -*- C++ -*- */
   2
   3 //=============================================================================
   4 /**
   5  *  @file    distributer.h
   6  *
   7  *  Process to receive data from the sender and send it to the
   8  *  receiver.
   9  *
  10  *  @author Yamuna Krishnamurthy <yamuna@cs.wustl.edu>
  11  */
  12 //=============================================================================
  13
  14
  15 #include "Connection_Manager.h"
  16 #include "orbsvcs/AV/AVStreams_i.h"
  17 #include "orbsvcs/AV/Endpoint_Strategy.h"
  18 #include "orbsvcs/AV/Policy.h"
  19
  20 /**
  21  * @class Signal_Handler
  22  TITLE
  23  * This class Handles the SIGINT signal through the Reactor.
  24  * Useful to gracefully release the process
  25  */
  26 class Signal_Handler : public ACE_Event_Handler
  27 {
  28
  29 public:
  30
  31   Signal_Handler (void);
  32
  33   /// Override this method to implement graceful shutdown.
  34   int handle_signal(int signum, siginfo_t*,ucontext_t*);
  35
  36 };
  37
  38
  39 /**
  40  * @class Distributer_Receiver_Callback
  41  *
  42  * @brief Application defined callback object.
  43  *
  44  * AVStreams calls this class when data shows up from a sender.
  45  */
  46 class Distributer_Receiver_Callback : public TAO_AV_Callback
  47 {
  48 public:
  49
  50   //// Constructor.
  51   Distributer_Receiver_Callback (void);
  52
  53   /// Method that is called when there is data to be received from a
  54   /// sender.
  55   int receive_frame (ACE_Message_Block *frame,
  56                      TAO_AV_frame_info *frame_info,
  57                      const ACE_Addr &peer_address);
  58
  59   /// Called when the sender is done sending data and wants to close
  60   /// down the connection.
  61   int handle_destroy (void);
  62
  63   /// Accessor methods for the flowname of the callback.
  64   ACE_CString &flowname (void);
  65   void flowname (const ACE_CString &flowname);
  66
  67 private:
  68   //// Count of the frames passing through us.
  69   int frame_count_;
  70
  71   //// Flowname of the flow.
  72   ACE_CString flowname_;
  73 };
  74
  75 /**
  76  * @class Distributer_Sender_Callback
  77  *
  78  * @brief Defines a class for the distributer application callback
  79  * for receiving data.
  80  *
  81  * This class overrides the methods of the TAO_AV_Callback so the
  82  * AVStreams can make upcalls to the application.
  83  */
  84 class Distributer_Sender_Callback : public TAO_AV_Callback
  85 {
  86
  87 public:
  88
  89   //// Called when the sender has finished reading the file and wants
  90   //// to close down the connection.
  91   int handle_destroy (void);
  92
  93   /// Accessor methods for the flowname of the callback.
  94   ACE_CString &flowname (void);
  95   void flowname (const ACE_CString &flowname);
  96
  97 private:
  98
  99   ACE_CString flowname_;
 100
 101 };
 102
 103 /**
 104  * @class Distributer_Receiver_StreamEndPoint
 105  *
 106  * @brief Application defined stream endpoint object.
 107  *
 108  * AVStreams calls this class during connection setup.
 109  */
 110 class Distributer_Receiver_StreamEndPoint : public TAO_Server_StreamEndPoint
 111 {
 112 public:
 113   /// Create a receiver application callback.
 114   int get_callback (const char *flowname,
 115                     TAO_AV_Callback *&callback);
 116
 117   //// Store the reference to the protocol object corresponding
 118   //// to the transport
 119   int set_protocol_object (const char *,
 120                            TAO_AV_Protocol_Object *object);
 121
 122   //// Called when a sender makes a connection request.
 123   virtual CORBA::Boolean handle_connection_requested (AVStreams::flowSpec &the_spec);
 124
 125 private:
 126   //// Receiver application callback.
 127   Distributer_Receiver_Callback callback_;
 128 };
 129
 130 /**
 131  * @class Distributer_Sender_StreamEndPoint
 132  *
 133  * @brief /    Defines a sender stream endpoint.
 134  */
 135 class Distributer_Sender_StreamEndPoint : public TAO_Client_StreamEndPoint
 136 {
 137 public:
 138   //// Create the application callback and return its handle to
 139   //// AVStreams for further application callbacks.
 140   int get_callback (const char *flowname,
 141                     TAO_AV_Callback *&callback);
 142
 143   //// Set protocol object corresponding to the transport protocol
 144   //// chosen.
 145   int set_protocol_object (const char *flowname,
 146                            TAO_AV_Protocol_Object *object);
 147
 148 protected:
 149   //// Application callback.
 150   Distributer_Sender_Callback callback_;
 151 };
 152
 153 typedef TAO_AV_Endpoint_Reactive_Strategy_A
 154           <Distributer_Sender_StreamEndPoint,
 155            TAO_VDev,
 156            AV_Null_MediaCtrl>
 157         SENDER_ENDPOINT_STRATEGY;
 158
 159 typedef TAO_AV_Endpoint_Reactive_Strategy_B
 160           <Distributer_Receiver_StreamEndPoint,
 161            TAO_VDev,
 162            AV_Null_MediaCtrl>
 163         RECEIVER_ENDPOINT_STRATEGY;
 164
 165 /**
 166  * @class Distributer
 167  *
 168  * @brief Distributer Application.
 169  *
 170  * The distributer is the intermediate receiver that receives
 171  * data from the sender and forwards to a receiver.
 172  */
 173 class Distributer
 174 {
 175 public:
 176   //// Constructor
 177   Distributer (void);
 178
 179   //// Destructor.
 180   ~Distributer (void);
 181
 182   //// Initialize data components.
 183   int init (int argc,
 184             ACE_TCHAR **argv);
 185
 186   //// Parse args.
 187   int parse_args (int argc,
 188                   ACE_TCHAR **argv);
 189
 190   /// Flag to know when we are done.
 191   bool done (void) const;
 192   void done (bool);
 193
 194   //// Accessor to connection manager.
 195   Connection_Manager &connection_manager (void);
 196
 197   //// Called when stream created
 198   void stream_created (void);
 199
 200   //// Called when stream destroyed
 201   void stream_destroyed (void);
 202
 203   ////Unbind the sender and receiver from the Naming Service
 204   void shut_down (void);
 205
 206 protected:
 207   //// Connection manager.
 208   Connection_Manager connection_manager_;
 209
 210   //// The sender endpoint strategy.
 211   SENDER_ENDPOINT_STRATEGY sender_endpoint_strategy_;
 212
 213   //// The receiver endpoint strategy.
 214   RECEIVER_ENDPOINT_STRATEGY receiver_endpoint_strategy_;
 215
 216   //// The distributer receiver multimedia device
 217   TAO_MMDevice* distributer_receiver_mmdevice_;
 218
 219   //// The distributer receiver multimedia device
 220   TAO_MMDevice* distributer_sender_mmdevice_;
 221
 222   //// The name of the sender to connect to.
 223   ACE_CString sender_name_;
 224
 225   //// Our name.
 226   ACE_CString distributer_name_;
 227
 228   //// Flag to know when we are done.
 229   bool done_;
 230
 231   //// Number of active streams. When a stream is disconnected this
 232   //// count is decremented.
 233   int stream_count_;
 234
 235   //// Reference to the signal handler.
 236   Signal_Handler signal_handler_;
 237 };