TAO/TAO_IDL/be/be_visitor_argument/marshal_ss.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    marshal_ss.cpp
   5  *
   6  *  Visitor that generates code that passes the argument variable to the
   7  *  marshal operations.
   8  *
   9  *  @author Aniruddha Gokhale
  10  */
  11 //=============================================================================
  12
  13 #include "argument.h"
  14
  15 // ************************************************************************
  16 // Visitor to generate code for passing argument to the marshal/demarshal
  17 // routines
  18 // ************************************************************************
  19
  20 be_visitor_args_marshal_ss::
  21 be_visitor_args_marshal_ss (be_visitor_context *ctx)
  22   : be_visitor_args (ctx)
  23 {
  24 }
  25
  26 be_visitor_args_marshal_ss::
  27 ~be_visitor_args_marshal_ss ()
  28 {
  29 }
  30
  31 int be_visitor_args_marshal_ss::visit_argument (
  32   be_argument *node)
  33 {
  34   this->ctx_->node (node);
  35   be_type *bt = dynamic_cast<be_type*> (node->field_type ());
  36
  37   if (!bt)
  38     {
  39       ACE_ERROR_RETURN ((LM_ERROR,
  40                          ACE_TEXT ("be_visitor_args_marshal_ss::")
  41                          ACE_TEXT ("visit_argument - ")
  42                          ACE_TEXT ("Bad argument type\n")),
  43                         -1);
  44     }
  45
  46   // Different types have different mappings when used as in/out or
  47   // inout parameters. Let this visitor deal with the type
  48
  49   TAO_OutStream *os = this->ctx_->stream ();
  50
  51   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
  52     {
  53       switch (this->direction ())
  54         {
  55         case AST_Argument::dir_IN:
  56         case AST_Argument::dir_INOUT:
  57           *os << be_nl << "(_tao_in >> ";
  58           break;
  59         case AST_Argument::dir_OUT:
  60           break;
  61         }
  62     }
  63   else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
  64     {
  65       switch (this->direction ())
  66         {
  67         case AST_Argument::dir_IN:
  68           break;
  69         case AST_Argument::dir_INOUT:
  70         case AST_Argument::dir_OUT:
  71           *os << be_nl << "(_tao_out << ";
  72           break;
  73         }
  74     }
  75   else
  76     {
  77       ACE_ERROR_RETURN ((LM_ERROR,
  78                          ACE_TEXT ("be_visitor_args_marshal_ss::")
  79                          ACE_TEXT ("visit_argument - ")
  80                          ACE_TEXT ("Bad substate\n")),
  81                         -1);
  82     }
  83
  84   if (bt->accept (this) == -1)
  85     {
  86       ACE_ERROR_RETURN ((LM_ERROR,
  87                          ACE_TEXT ("be_visitor_args_marshal_ss::")
  88                          ACE_TEXT ("visit_argument - ")
  89                          ACE_TEXT ("cannot accept visitor\n")),
  90                         -1);
  91     }
  92
  93   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
  94     {
  95       switch (this->direction ())
  96         {
  97         case AST_Argument::dir_IN:
  98         case AST_Argument::dir_INOUT:
  99           *os << ")";
 100           break;
 101         case AST_Argument::dir_OUT:
 102           break;
 103         }
 104     }
 105   else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
 106     {
 107       switch (this->direction ())
 108         {
 109         case AST_Argument::dir_IN:
 110           break;
 111         case AST_Argument::dir_INOUT:
 112         case AST_Argument::dir_OUT:
 113           *os << ")";
 114           break;
 115         }
 116     }
 117   else
 118     {
 119       ACE_ERROR_RETURN ((LM_ERROR,
 120                          ACE_TEXT ("be_visitor_args_marshal_ss::")
 121                          ACE_TEXT ("visit_argument - ")
 122                          ACE_TEXT ("Bad substate\n")),
 123                         -1);
 124     }
 125
 126   return 0;
 127 }
 128
 129 int be_visitor_args_marshal_ss::visit_array (be_array *)
 130 {
 131   TAO_OutStream *os = this->ctx_->stream ();
 132   be_argument *arg =
 133     dynamic_cast<be_argument*> (this->ctx_->node ());
 134   const char *lname = arg->local_name ()->get_string ();
 135
 136   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
 137     {
 138       switch (this->direction ())
 139         {
 140         case AST_Argument::dir_IN:
 141         case AST_Argument::dir_INOUT:
 142           *os << "_tao_forany_" << lname;
 143           break;
 144         case AST_Argument::dir_OUT:
 145           break;
 146         }
 147     }
 148   else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
 149     {
 150       switch (this->direction ())
 151         {
 152         case AST_Argument::dir_IN:
 153           break;
 154         case AST_Argument::dir_INOUT:
 155         case AST_Argument::dir_OUT:
 156           *os << "_tao_forany_" << lname;
 157           break;
 158         }
 159     }
 160   else
 161     {
 162       ACE_ERROR_RETURN ((LM_ERROR,
 163                          ACE_TEXT ("be_visitor_args_marshal_ss::")
 164                          ACE_TEXT ("visit_array - ")
 165                          ACE_TEXT ("Bad substate\n")),
 166                         -1);
 167     }
 168
 169   return 0;
 170 }
 171
 172 int be_visitor_args_marshal_ss::visit_enum (be_enum *)
 173 {
 174   TAO_OutStream *os = this->ctx_->stream ();
 175   be_argument *arg =
 176     dynamic_cast<be_argument*> (this->ctx_->node ());
 177   const char *lname = arg->local_name ()->get_string ();
 178
 179   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
 180     {
 181       switch (this->direction ())
 182         {
 183         case AST_Argument::dir_IN:
 184         case AST_Argument::dir_INOUT:
 185           *os << lname;
 186           break;
 187         case AST_Argument::dir_OUT:
 188           break;
 189         }
 190     }
 191   else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
 192     {
 193       switch (this->direction ())
 194         {
 195         case AST_Argument::dir_IN:
 196           break;
 197         case AST_Argument::dir_INOUT:
 198         case AST_Argument::dir_OUT:
 199           *os << lname;
 200           break;
 201         }
 202     }
 203   else
 204     {
 205       ACE_ERROR_RETURN ((LM_ERROR,
 206                          ACE_TEXT ("be_visitor_args_marshal_ss::")
 207                          ACE_TEXT ("visit_enum - ")
 208                          ACE_TEXT ("Bad substate\n")),
 209                         -1);
 210     }
 211
 212   return 0;
 213 }
 214
 215 int be_visitor_args_marshal_ss::visit_interface (
 216   be_interface *)
 217 {
 218   return this->emit_common ();
 219 }
 220
 221 int be_visitor_args_marshal_ss::visit_interface_fwd (
 222   be_interface_fwd *)
 223 {
 224   return this->emit_common ();
 225 }
 226
 227 int be_visitor_args_marshal_ss::visit_valuebox (
 228   be_valuebox *)
 229 {
 230   return this->emit_common ();
 231 }
 232
 233 int be_visitor_args_marshal_ss::visit_valuetype (
 234   be_valuetype *)
 235 {
 236   return this->emit_common ();
 237 }
 238
 239 int be_visitor_args_marshal_ss::visit_valuetype_fwd (
 240   be_valuetype_fwd *)
 241 {
 242   return this->emit_common ();
 243 }
 244
 245 int be_visitor_args_marshal_ss::visit_predefined_type (
 246   be_predefined_type *node)
 247 {
 248   return this->gen_pd_arg (node, false);
 249 }
 250
 251 int be_visitor_args_marshal_ss::visit_sequence (be_sequence *)
 252 {
 253   TAO_OutStream *os = this->ctx_->stream ();
 254   be_argument *arg =
 255     dynamic_cast<be_argument*> (this->ctx_->node ());
 256   const char *lname = arg->local_name ()->get_string ();
 257
 258   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
 259     {
 260       switch (this->direction ())
 261         {
 262         case AST_Argument::dir_IN:
 263         case AST_Argument::dir_INOUT:
 264           *os << lname;
 265           break;
 266         case AST_Argument::dir_OUT:
 267           break;
 268         }
 269     }
 270   else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
 271     {
 272       switch (this->direction ())
 273         {
 274         case AST_Argument::dir_IN:
 275           break;
 276         case AST_Argument::dir_INOUT:
 277           *os << lname;
 278           break;
 279         case AST_Argument::dir_OUT:
 280           *os << lname << ".in ()";
 281           break;
 282         }
 283     }
 284   else
 285     {
 286       ACE_ERROR_RETURN ((LM_ERROR,
 287                          ACE_TEXT ("be_visitor_args_marshal_ss::")
 288                          ACE_TEXT ("visit_interface - ")
 289                          ACE_TEXT ("Bad substate\n")),
 290                         -1);
 291     }
 292
 293   return 0;
 294 }
 295
 296 int be_visitor_args_marshal_ss::visit_string (be_string *node)
 297 {
 298   TAO_OutStream *os = this->ctx_->stream ();
 299   be_argument *arg =
 300     dynamic_cast<be_argument*> (this->ctx_->node ());
 301   const char *lname = arg->local_name ()->get_string ();
 302   ACE_CDR::ULong bound = node->max_size ()->ev ()->u.ulval;
 303   bool wide =
 304     (node->width () != static_cast<long> (sizeof (char)));
 305
 306   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
 307     {
 308       switch (this->direction ())
 309         {
 310         case AST_Argument::dir_IN:
 311         case AST_Argument::dir_INOUT:
 312           if (bound == 0)
 313             {
 314               *os << lname << ".out ()";
 315             }
 316           else
 317             {
 318               *os << "::ACE_InputCDR::to_"
 319                   << (wide ? "w" : "" ) << "string ("
 320                   << lname << ".out (), " << bound << ")";
 321             }
 322
 323           break;
 324         case AST_Argument::dir_OUT:
 325           break;
 326         }
 327     }
 328   else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
 329     {
 330       // We need to make a distinction between
 331       // bounded and unbounded strings.
 332       if (bound == 0)
 333         {
 334           switch (this->direction ())
 335             {
 336             case AST_Argument::dir_IN:
 337               break;
 338             case AST_Argument::dir_INOUT:
 339             case AST_Argument::dir_OUT:
 340               *os << lname << ".in ()";
 341               break;
 342             }
 343         }
 344       else
 345         {
 346           switch (this->direction ())
 347             {
 348             case AST_Argument::dir_IN:
 349               break;
 350             case AST_Argument::dir_INOUT:
 351             case AST_Argument::dir_OUT:
 352               *os << "::ACE_OutputCDR::from_"
 353                   << (wide ? "w" : "" ) << "string (("
 354                   << (wide ? "CORBA::WChar" : "char") << " *)"
 355                   << lname << ".in (), " << bound << ")";
 356
 357               break;
 358             }
 359         }
 360     }
 361   else
 362     {
 363       ACE_ERROR_RETURN ((LM_ERROR,
 364                          ACE_TEXT ("be_visitor_args_marshal_ss::")
 365                          ACE_TEXT ("visit_string - ")
 366                          ACE_TEXT ("Bad substate\n")),
 367                         -1);
 368     }
 369
 370   return 0;
 371 }
 372
 373 int be_visitor_args_marshal_ss::visit_structure (be_structure *node)
 374 {
 375   return this->emit_common2 (node);
 376 }
 377
 378 int be_visitor_args_marshal_ss::visit_union (be_union *node)
 379 {
 380   return this->emit_common2 (node);
 381 }
 382
 383 int be_visitor_args_marshal_ss::visit_typedef (be_typedef *node)
 384 {
 385   this->ctx_->alias (node);
 386
 387   if (node->primitive_base_type ()->accept (this) == -1)
 388     {
 389       ACE_ERROR_RETURN ((LM_ERROR,
 390                          ACE_TEXT ("be_visitor_args_marshal_ss::")
 391                          ACE_TEXT ("visit_typedef - ")
 392                          ACE_TEXT ("accept on primitive ")
 393                          ACE_TEXT ("type failed\n")),
 394                         -1);
 395     }
 396
 397   this->ctx_->alias (nullptr);
 398   return 0;
 399 }
 400
 401 int
 402 be_visitor_args_marshal_ss::visit_component (
 403     be_component *node
 404   )
 405 {
 406   return this->visit_interface (node);
 407 }
 408
 409 int
 410 be_visitor_args_marshal_ss::visit_component_fwd (
 411     be_component_fwd *node
 412   )
 413 {
 414   return this->visit_interface_fwd (node);
 415 }
 416
 417 int
 418 be_visitor_args_marshal_ss::visit_eventtype (
 419     be_eventtype *node
 420   )
 421 {
 422   return this->visit_valuetype (node);
 423 }
 424
 425 int
 426 be_visitor_args_marshal_ss::visit_eventtype_fwd (
 427     be_eventtype_fwd *node
 428   )
 429 {
 430   return this->visit_valuetype_fwd (node);
 431 }
 432
 433 int
 434 be_visitor_args_marshal_ss::visit_home (
 435     be_home *node
 436   )
 437 {
 438   return this->visit_interface (node);
 439 }
 440
 441 int
 442 be_visitor_args_marshal_ss::emit_common ()
 443 {
 444   TAO_OutStream *os = this->ctx_->stream ();
 445   be_argument *arg =
 446     dynamic_cast<be_argument*> (this->ctx_->node ());
 447   const char *lname = arg->local_name ()->get_string ();
 448
 449   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
 450     {
 451       switch (this->direction ())
 452         {
 453         case AST_Argument::dir_IN:
 454         case AST_Argument::dir_INOUT:
 455           *os << lname << ".out ()";
 456           break;
 457         case AST_Argument::dir_OUT:
 458           break;
 459         }
 460     }
 461   else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
 462     {
 463       switch (this->direction ())
 464         {
 465         case AST_Argument::dir_IN:
 466           break;
 467         case AST_Argument::dir_INOUT:
 468         case AST_Argument::dir_OUT:
 469           *os << lname << ".in ()";
 470           break;
 471         }
 472     }
 473   else
 474     {
 475       ACE_ERROR_RETURN ((LM_ERROR,
 476                          ACE_TEXT ("be_visitor_args_marshal_ss")
 477                          ACE_TEXT ("::emit_common - ")
 478                          ACE_TEXT ("Bad substate\n")),
 479                         -1);
 480     }
 481
 482   return 0;
 483 }
 484
 485
 486 int
 487 be_visitor_args_marshal_ss::emit_common2 (be_type *node)
 488 {
 489   TAO_OutStream *os = this->ctx_->stream ();
 490   be_argument *arg =
 491     dynamic_cast<be_argument*> (this->ctx_->node ());
 492   const char *lname = arg->local_name ()->get_string ();
 493   AST_Type::SIZE_TYPE st = node->size_type ();
 494
 495   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_INPUT)
 496     {
 497       switch (this->direction ())
 498         {
 499         case AST_Argument::dir_IN:
 500         case AST_Argument::dir_INOUT:
 501           *os << lname;
 502           break;
 503         case AST_Argument::dir_OUT:
 504           break;
 505         }
 506     }
 507   else if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT)
 508     {
 509       switch (this->direction ())
 510         {
 511         case AST_Argument::dir_IN:
 512           break;
 513         case AST_Argument::dir_INOUT:
 514           *os << lname;
 515           break;
 516         case AST_Argument::dir_OUT:
 517           *os << lname
 518               << (st == AST_Type::VARIABLE ? ".in ()" : "");
 519
 520           break;
 521         }
 522     }
 523   else
 524     {
 525       ACE_ERROR_RETURN ((LM_ERROR,
 526                          ACE_TEXT ("be_visitor_args_marshal_ss")
 527                          ACE_TEXT ("::emit_common2 - ")
 528                          ACE_TEXT ("Bad substate\n")),
 529                         -1);
 530     }
 531
 532   return 0;
 533 }