TAO/TAO_IDL/be/be_visitor_operation/argument_invoke.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    argument_invoke.cpp
   5  *
   6  *  Visitor to pass arguments to the CDR operators. This one helps in
   7  *  generating the && and the , at the right place. This one is for the
   8  *  client stub side.
   9  *
  10  *  @author Aniruddha Gokhale
  11  */
  12 //=============================================================================
  13
  14 #include "operation.h"
  15
  16 // ************************************************************
  17 // operation visitor to handle the passing of arguments to the CDR operators
  18 // ************************************************************
  19
  20 be_visitor_operation_argument_invoke::be_visitor_operation_argument_invoke (
  21     be_visitor_context
  22     *ctx)
  23   : be_visitor_operation_argument (ctx),
  24     last_arg_printed_ (be_visitor_operation_argument_invoke::TAO_ARG_NONE)
  25 {
  26 }
  27
  28 be_visitor_operation_argument_invoke::~be_visitor_operation_argument_invoke ()
  29 {
  30 }
  31
  32 int
  33 be_visitor_operation_argument_invoke::pre_process (be_decl *bd)
  34 {
  35   TAO_OutStream *os = this->ctx_->stream ();
  36
  37   be_argument *arg = dynamic_cast<be_argument*> (bd);
  38
  39   if (!arg)
  40     {
  41       ACE_ERROR_RETURN ((LM_ERROR,
  42                          "(%N:%l) "
  43                          "be_visitor_operation_argument_invoke"
  44                          "::pre_process - "
  45                          "Bad argument node\n"),
  46                         -1);
  47     }
  48
  49   switch (arg->direction ())
  50     {
  51     case AST_Argument::dir_IN:
  52       if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
  53         {
  54           if (this->last_arg_printed_ !=
  55               be_visitor_operation_argument_invoke::TAO_ARG_NONE)
  56             *os << " &&" << be_nl;
  57         }
  58       else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
  59         {
  60           // nothing
  61         }
  62       break;
  63     case AST_Argument::dir_INOUT:
  64       if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
  65         {
  66           if (this->last_arg_printed_ !=
  67               be_visitor_operation_argument_invoke::TAO_ARG_NONE)
  68             *os << " &&" << be_nl;
  69         }
  70       else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
  71         {
  72           if (this->last_arg_printed_ !=
  73               be_visitor_operation_argument_invoke::TAO_ARG_NONE)
  74             *os << " &&" << be_nl;
  75         }
  76       break;
  77     case AST_Argument::dir_OUT:
  78       if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
  79         {
  80           // nothing
  81         }
  82       else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
  83         {
  84           if (this->last_arg_printed_ !=
  85               be_visitor_operation_argument_invoke::TAO_ARG_NONE)
  86             *os << " &&" << be_nl;
  87         }
  88       break;
  89     }
  90
  91   return 0;
  92 }
  93
  94 int
  95 be_visitor_operation_argument_invoke::post_process (be_decl *bd)
  96 {
  97   be_argument *arg = dynamic_cast<be_argument*> (bd);
  98
  99   if (!arg)
 100     {
 101       ACE_ERROR_RETURN ((LM_ERROR,
 102                          "(%N:%l) "
 103                          "be_visitor_operation_argument_invoke"
 104                          "::post_process - "
 105                          "Bad argument node\n"),
 106                         -1);
 107     }
 108
 109   switch (this->ctx_->sub_state ())
 110     {
 111     case TAO_CodeGen::TAO_CDR_OUTPUT:
 112       switch (arg->direction ())
 113         {
 114         case AST_Argument::dir_IN:
 115           // only these arguments get printed
 116           this->last_arg_printed_ =
 117             be_visitor_operation_argument_invoke::TAO_ARG_IN;
 118           break;
 119         case AST_Argument::dir_INOUT:
 120           // only these arguments get printed
 121           this->last_arg_printed_ =
 122             be_visitor_operation_argument_invoke::TAO_ARG_INOUT;
 123           break;
 124         case AST_Argument::dir_OUT:
 125           // these arguments don't get printed for the << operator on the stub
 126           break;
 127         }
 128       break;
 129     case TAO_CodeGen::TAO_CDR_INPUT:
 130       switch (arg->direction ())
 131         {
 132         case AST_Argument::dir_IN:
 133           // these arguments don't get printed for the >> on the stub
 134           break;
 135         case AST_Argument::dir_INOUT:
 136           // only these arguments get printed
 137           this->last_arg_printed_ =
 138             be_visitor_operation_argument_invoke::TAO_ARG_INOUT;
 139           break;
 140         case AST_Argument::dir_OUT:
 141           // only these arguments get printed
 142           this->last_arg_printed_ =
 143             be_visitor_operation_argument_invoke::TAO_ARG_OUT;
 144           break;
 145         }
 146       break;
 147     default:
 148       ACE_ERROR_RETURN ((LM_ERROR,
 149                          "(%N:%l) "
 150                          "be_visitor_operation_argument_invoke"
 151                          "::post_process - "
 152                          "Bad sub state\n"),
 153                         -1);
 154     }
 155
 156   return 0;
 157 }
 158