1 files changed, 192 insertions, 0 deletions
diff --git a/trunk/TAO/TAO_IDL/be/be_visitor_operation/argument.cpp b/trunk/TAO/TAO_IDL/be/be_visitor_operation/argument.cpp
new file mode 100644
index 00000000000..413cb6a125e
--- /dev/null
+++ b/trunk/TAO/TAO_IDL/be/be_visitor_operation/argument.cpp
@@ -0,0 +1,192 @@
+//
+// $Id$
+//
+
+// ============================================================================
+//
+// = LIBRARY
+//    TAO IDL
+//
+// = FILENAME
+//    argument.cpp
+//
+// = DESCRIPTION
+//    Visitor that calls the visitor for arguments.
+//
+// = AUTHOR
+//    Aniruddha Gokhale
+//
+// ============================================================================
+
+ACE_RCSID (be_visitor_operation,
+           argument,
+           "$Id$")
+
+// ************************************************************
+// Generic operation visitor to handle the pre/post
+// do_static_call/upcall stuff with arguments.
+// ************************************************************
+
+be_visitor_operation_argument::be_visitor_operation_argument (
+    be_visitor_context *ctx
+  )
+  : be_visitor_operation (ctx)
+{
+}
+
+be_visitor_operation_argument::~be_visitor_operation_argument (void)
+{
+}
+
+int
+be_visitor_operation_argument::post_process (be_decl *bd)
+{
+  TAO_OutStream *os = this->ctx_->stream ();
+
+  // If we are not the last parameter, we insert a comma. This is only
+  // applicable for the upcalls or the call to (de)marshal that we use in the
+  // interpreted marshaling.
+  switch (this->ctx_->state ())
+    {
+    case TAO_CodeGen::TAO_OPERATION_ARG_UPCALL_SS:
+    case TAO_CodeGen::TAO_OPERATION_ARG_DEMARSHAL_SS:
+    case TAO_CodeGen::TAO_OPERATION_ARG_MARSHAL_SS:
+      if (!this->last_node (bd))
+        {
+          *os << "," << be_nl;
+        }
+
+      break;
+    case TAO_CodeGen::TAO_OPERATION_COLLOCATED_ARG_UPCALL_SS:
+      if (!this->last_node (bd))
+        {
+          *os << ",";
+        }
+
+      break;
+    default:
+      break;
+    }
+
+  return 0;
+}
+
+int
+be_visitor_operation_argument::visit_operation (be_operation *node)
+{
+  // All we do is hand over code generation to our scope.
+  if (this->visit_scope (node) == -1)
+    {
+      ACE_ERROR_RETURN ((LM_ERROR,
+                         "(%N:%l) be_visitor_operation_argument::"
+                         "visit_operation - "
+                         "codegen for scope failed\n"),
+                        -1);
+    }
+
+  return 0;
+}
+
+int
+be_visitor_operation_argument::visit_argument (be_argument *node)
+{
+  // Get the visitor that will dump the argument's mapping in the operation
+  // signature.
+  be_visitor_context ctx (*this->ctx_);
+
+  // first grab the interface definition inside which this operation is
+  // defined. We need this since argument types may very well be declared
+  // inside the scope of the interface node. In such cases, we would like to
+  // generate the appropriate relative scoped names.
+  be_operation *op = this->ctx_->be_scope_as_operation ();
+
+  if (!op)
+    {
+      ACE_ERROR_RETURN ((LM_ERROR,
+                         "(%N:%l) be_visitor_arglist::"
+                         "visit_argument - "
+                         "Bad operation\n"),
+                        -1);
+    }
+
+  // We need the interface node in which this operation was defined. However,
+  // if this operation node was an attribute node in disguise, we get this
+  // information from the context.
+  be_interface *intf;
+  intf = this->ctx_->attribute ()
+    ? be_interface::narrow_from_scope (this->ctx_->attribute ()->defined_in ())
+    : be_interface::narrow_from_scope (op->defined_in ());
+
+  if (!intf)
+    {
+      ACE_ERROR_RETURN ((LM_ERROR,
+                         "(%N:%l) be_visitor_arglist::"
+                         "visit_argument - "
+                         "Bad interface\n"),
+                        -1);
+    }
+
+  ctx.scope (intf);
+  int status = 0;
+
+  switch (this->ctx_->state ())
+    {
+    case TAO_CodeGen::TAO_OPERATION_ARG_INVOKE_CS:
+      {
+        be_visitor_args_invoke_cs visitor (&ctx);
+        status = node->accept (&visitor);
+        break;
+      }
+    case TAO_CodeGen::TAO_OPERATION_ARG_DECL_SS:
+      {
+        be_visitor_args_vardecl_ss visitor (&ctx);
+        status = node->accept (&visitor);
+        break;
+      }
+    case TAO_CodeGen::TAO_OPERATION_ARG_DEMARSHAL_SS:
+      {
+        be_visitor_args_marshal_ss visitor (&ctx);
+        status = node->accept (&visitor);
+        break;
+      }
+    case TAO_CodeGen::TAO_OPERATION_ARG_MARSHAL_SS:
+      {
+        be_visitor_args_marshal_ss visitor (&ctx);
+        status = node->accept (&visitor);
+        break;
+      }
+    case TAO_CodeGen::TAO_OPERATION_COLLOCATED_ARG_UPCALL_SS:
+      {
+        ctx.state (TAO_CodeGen::TAO_ARGUMENT_COLLOCATED_UPCALL_SS);
+        be_visitor_args_upcall_ss visitor (&ctx);
+        status = node->accept (&visitor);
+        break;
+      }
+    case TAO_CodeGen::TAO_OPERATION_ARG_UPCALL_SS:
+      {
+        be_visitor_args_upcall_ss visitor (&ctx);
+        status = node->accept (&visitor);
+        break;
+      }
+    default:
+      {
+        ACE_ERROR_RETURN ((LM_ERROR,
+                           "(%N:%l) be_visitor_argument::"
+                           "visit_argument - "
+                           "Bad context\n"),
+                          -1);
+      }
+    }
+
+  if (status == -1)
+    {
+      ACE_ERROR_RETURN ((LM_ERROR,
+                         "(%N:%l) be_visitor_argument::"
+
+                         "visit_argument - "
+                         "codegen for argument failed\n"),
+                        -1);
+    }
+
+  return 0;
+}