1 files changed, 108 insertions, 0 deletions
diff --git a/TAO/TAO_IDL/be_include/be_visitor_amh_pre_proc.h b/TAO/TAO_IDL/be_include/be_visitor_amh_pre_proc.h
new file mode 100644
index 00000000000..e362eb94d23
--- /dev/null
+++ b/TAO/TAO_IDL/be_include/be_visitor_amh_pre_proc.h
@@ -0,0 +1,108 @@
+//=============================================================================
+/**
+*  @file   be_visitor_amh_pre_proc.h
+*
+*  $Id$
+*
+*  This visitor creates for AMH implied IDL constructs for the ResponseHandler,
+*  the appropriate AST
+*  (Abstract Syntax Tree) node, sets the corresponding interface or operation
+*  strategy on it and enters the node into the AST.
+*
+*  @author Darrell Brunsch <brunsch@cs.wustl.edu>
+*  @author Mayur Deshpande <mayur@ics.uci.edu>
+*  @author Carlos O'Ryan   <coryan@uci.edu>
+*/
+//=============================================================================
+
+#ifndef TAO_BE_VISITOR_AMH_PRE_PROC_H
+#define TAO_BE_VISITOR_AMH_PRE_PROC_H
+
+#include "be_visitor_scope.h"
+#include "ace/SString.h"
+
+/**
+ * @class be_visitor_amh_pre_proc
+ *
+ * @brief Adds AMH implied IDL code into the AST
+ */
+class be_visitor_amh_pre_proc : public be_visitor_scope
+{
+private:
+  typedef enum 
+  {
+    NORMAL,
+    GET_OPERATION,
+    SET_OPERATION
+  } Operation_Kind;
+
+public:
+  be_visitor_amh_pre_proc (be_visitor_context *ctx);
+
+  virtual ~be_visitor_amh_pre_proc (void);
+
+  virtual int visit_root (be_root *node);
+
+  virtual int visit_module (be_module *node);
+
+  virtual int visit_interface (be_interface *node);
+
+  virtual int visit_operation (be_operation *node);
+
+private:
+  /// Create the reponse handler interface
+  be_interface *create_response_handler (be_interface *node,
+                                         be_valuetype *exception_holder);
+
+  /// method to add memebers to the response handler interfaces
+  int add_rh_node_members (be_interface *node,
+                           be_interface *response_handler,
+                           be_valuetype *exception_holder);
+
+  /// Create an operation with return, OUT and INOUT arguments
+  int create_response_handler_operation (be_operation *node,
+                                         be_interface *response_handler,
+                                         be_valuetype *exception_holder);
+
+  /// Create the get_* operation associated with an attribute.
+  int create_response_handler_attribute (be_attribute *node,
+                                         be_interface *response_handler,
+                                         be_valuetype *exception_holder);
+
+  /// Add the normal reply handler for the @c node operation
+  int add_normal_reply (be_operation *node,
+                        be_interface *response_handler);
+
+  /// Add the exception version of @c node
+  int add_exception_reply (be_operation *node,
+                           be_interface *response_handler,
+                           be_valuetype *exception_holder);
+
+  /// Create the exception holder valuetype
+  be_valuetype *create_exception_holder (be_interface *node);
+
+  /// And the concrete implemenation of the exception holder valuetype
+  //  be_valuetype *create_exception_holder_concrete (be_valuetype *parent);
+
+  /// Creates a raise operation from node and inserts it in
+  /// excep_holder, while obeying if it is a normal operation
+  /// or a set or get attribute.
+  int create_raise_operation (be_decl *node,
+                              be_valuetype *excep_holder,
+                              Operation_Kind operation_kind);
+
+  /// Concatenate 3 strings into one.
+  int generate_name (ACE_CString &destination,
+                     const char *prefix,
+                     const char *middle_name,
+                     const char *suffix);
+
+  /// Generate a get operation out of the attribute.
+  be_operation *generate_get_operation (be_attribute *node);
+
+  /// Generate a set operation out of the attribute.
+  be_operation *generate_set_operation (be_attribute *node);
+};
+
+
+#endif // TAO_BE_VISITOR_AMH_PRE_PROC_H