1 files changed, 170 insertions, 0 deletions
diff --git a/ACE/TAO/TAO_IDL/be/be_visitor_operation/argument_invoke.cpp b/ACE/TAO/TAO_IDL/be/be_visitor_operation/argument_invoke.cpp
new file mode 100644
index 00000000000..d222235b04d
--- /dev/null
+++ b/ACE/TAO/TAO_IDL/be/be_visitor_operation/argument_invoke.cpp
@@ -0,0 +1,170 @@
+//
+// $Id$
+//
+
+// ============================================================================
+//
+// = LIBRARY
+//    TAO IDL
+//
+// = FILENAME
+//    argument_invoke.cpp
+//
+// = DESCRIPTION
+//    Visitor to pass arguments to the CDR operators. This one helps in
+//    generating the && and the , at the right place. This one is for the
+//    client stub side.
+//
+// = AUTHOR
+//    Aniruddha Gokhale
+//
+// ============================================================================
+
+ACE_RCSID (be_visitor_operation, 
+           argument_invoke, 
+           "$Id$")
+
+// ************************************************************
+// operation visitor to handle the passing of arguments to the CDR operators
+// ************************************************************
+
+be_visitor_operation_argument_invoke::be_visitor_operation_argument_invoke (
+    be_visitor_context
+    *ctx
+  )
+  : be_visitor_operation_argument (ctx),
+    last_arg_printed_ (be_visitor_operation_argument_invoke::TAO_ARG_NONE)
+{
+}
+
+be_visitor_operation_argument_invoke::~be_visitor_operation_argument_invoke (
+    void
+  )
+{
+}
+
+int
+be_visitor_operation_argument_invoke::pre_process (be_decl *bd)
+{
+  TAO_OutStream *os = this->ctx_->stream ();
+
+  be_argument *arg = be_argument::narrow_from_decl (bd);
+
+  if (!arg)
+    {
+      ACE_ERROR_RETURN ((LM_ERROR,
+                         "(%N:%l) "
+                         "be_visitor_operation_argument_invoke"
+                         "::post_process - "
+                         "Bad argument node\n"),
+                        -1);
+    }
+
+  switch (arg->direction ())
+    {
+    case AST_Argument::dir_IN:
+      if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
+        {
+          if (this->last_arg_printed_ !=
+              be_visitor_operation_argument_invoke::TAO_ARG_NONE)
+            *os << " &&" << be_nl;
+        }
+      else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
+        {
+          // nothing
+        }
+      break;
+    case AST_Argument::dir_INOUT:
+      if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
+        {
+          if (this->last_arg_printed_ !=
+              be_visitor_operation_argument_invoke::TAO_ARG_NONE)
+            *os << " &&" << be_nl;
+        }
+      else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
+        {
+          if (this->last_arg_printed_ !=
+              be_visitor_operation_argument_invoke::TAO_ARG_NONE)
+            *os << " &&" << be_nl;
+        }
+      break;
+    case AST_Argument::dir_OUT:
+      if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
+        {
+          // nothing
+        }
+      else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
+        {
+          if (this->last_arg_printed_ !=
+              be_visitor_operation_argument_invoke::TAO_ARG_NONE)
+            *os << " &&" << be_nl;
+        }
+      break;
+    }
+
+  return 0;
+}
+
+int
+be_visitor_operation_argument_invoke::post_process (be_decl *bd)
+{
+  be_argument *arg = be_argument::narrow_from_decl (bd);
+
+  if (!arg)
+    {
+      ACE_ERROR_RETURN ((LM_ERROR,
+                         "(%N:%l) "
+                         "be_visitor_operation_argument_invoke"
+                         "::post_process - "
+                         "Bad argument node\n"),
+                        -1);
+    }
+  switch (this->ctx_->sub_state ())
+    {
+    case TAO_CodeGen::TAO_CDR_OUTPUT:
+      switch (arg->direction ())
+        {
+        case AST_Argument::dir_IN:
+          // only these arguments get printed
+          this->last_arg_printed_ =
+            be_visitor_operation_argument_invoke::TAO_ARG_IN;
+          break;
+        case AST_Argument::dir_INOUT:
+          // only these arguments get printed
+          this->last_arg_printed_ =
+            be_visitor_operation_argument_invoke::TAO_ARG_INOUT;
+          break;
+        case AST_Argument::dir_OUT:
+          // these arguments don't get printed for the << operator on the stub
+          break;
+        }
+      break;
+    case TAO_CodeGen::TAO_CDR_INPUT:
+      switch (arg->direction ())
+        {
+        case AST_Argument::dir_IN:
+          // these arguments don't get printed for the >> on the stub
+          break;
+        case AST_Argument::dir_INOUT:
+          // only these arguments get printed
+          this->last_arg_printed_ =
+            be_visitor_operation_argument_invoke::TAO_ARG_INOUT;
+          break;
+        case AST_Argument::dir_OUT:
+          // only these arguments get printed
+          this->last_arg_printed_ =
+            be_visitor_operation_argument_invoke::TAO_ARG_OUT;
+          break;
+        }
+      break;
+    default:
+      ACE_ERROR_RETURN ((LM_ERROR,
+                         "(%N:%l) "
+                         "be_visitor_operation_argument_invoke"
+                         "::post_process - "
+                         "Bad sub state\n"),
+                        -1);
+    }
+  return 0;
+}
+