TAO/TAO_IDL/be/be_visitor_union/union.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    union.cpp
   5  *
   6  *  Generic visitor generating code for Unions
   7  *
   8  *  @author Aniruddha Gokhale
   9  */
  10 //=============================================================================
  11
  12 #include "union.h"
  13
  14 be_visitor_union::be_visitor_union (be_visitor_context *ctx)
  15   : be_visitor_scope (ctx)
  16 {
  17 }
  18
  19 be_visitor_union::~be_visitor_union ()
  20 {
  21 }
  22
  23 int
  24 be_visitor_union::visit_union (be_union *)
  25 {
  26   return -1; // Must be overriden.
  27 }
  28
  29 int
  30 be_visitor_union::visit_union_branch (be_union_branch *node)
  31 {
  32   // Instantiate a visitor context with a copy of our context. This info
  33   // will be modified based on what type of node we are visiting.
  34   be_visitor_context ctx (*this->ctx_);
  35   ctx.node (node);
  36   int status = 0;
  37
  38   switch (this->ctx_->state ())
  39     {
  40     case TAO_CodeGen::TAO_UNION_PUBLIC_CH:
  41       {
  42         be_visitor_union_branch_public_ch visitor (&ctx);
  43         status = node->accept (&visitor);
  44         break;
  45       }
  46     case TAO_CodeGen::TAO_UNION_PRIVATE_CH:
  47       {
  48         be_visitor_union_branch_private_ch visitor (&ctx);
  49         status = node->accept (&visitor);
  50         break;
  51       }
  52     case TAO_CodeGen::TAO_ROOT_CI:
  53       {
  54         be_visitor_union_branch_public_ci visitor (&ctx);
  55         status = node->accept (&visitor);
  56         break;
  57       }
  58     case TAO_CodeGen::TAO_UNION_PUBLIC_CS:
  59       {
  60         be_visitor_union_branch_public_cs visitor (&ctx);
  61         status = node->accept (&visitor);
  62         break;
  63       }
  64     case TAO_CodeGen::TAO_UNION_PUBLIC_ASSIGN_CS:
  65       {
  66         be_visitor_union_branch_public_assign_cs visitor (&ctx);
  67         status = node->accept (&visitor);
  68         break;
  69       }
  70     case TAO_CodeGen::TAO_UNION_PUBLIC_RESET_CS:
  71       {
  72         be_visitor_union_branch_public_reset_cs visitor (&ctx);
  73         status = node->accept (&visitor);
  74         break;
  75       }
  76     case TAO_CodeGen::TAO_ROOT_CDR_OP_CH:
  77       {
  78         be_visitor_union_branch_cdr_op_ch visitor (&ctx);
  79         status = node->accept (&visitor);
  80         break;
  81       }
  82     case TAO_CodeGen::TAO_ROOT_CDR_OP_CS:
  83       {
  84         be_visitor_union_branch_cdr_op_cs visitor (&ctx);
  85         status = node->accept (&visitor);
  86         break;
  87       }
  88     default:
  89       {
  90         ACE_ERROR_RETURN ((LM_ERROR,
  91                            "(%N:%l) be_visitor_union::"
  92                            "visit_union_branch - "
  93                            "Bad context state\n"),
  94                           -1);
  95       }
  96     }
  97
  98   if (status == -1)
  99     {
 100       ACE_ERROR_RETURN ((LM_ERROR,
 101                          "(%N:%l) be_visitor_union::"
 102                          "visit_union_branch - "
 103                          "failed to accept visitor\n"),
 104                         -1);
 105     }
 106
 107   return 0;
 108 }
 109
 110 be_visitor_union::BoolUnionBranch
 111 be_visitor_union::boolean_branch (be_union_branch *b)
 112 {
 113   be_union *u = dynamic_cast<be_union*> (b->defined_in ());
 114   if (!u || u->udisc_type () != AST_Expression::EV_bool)
 115     return BUB_NONE;
 116
 117   bool br_d = false, br_t = false, br_f = false;
 118   for (unsigned long i = 0; i < b->label_list_length (); ++i)
 119     if (b->label (i)->label_kind () == AST_UnionLabel::UL_default)
 120       br_d = true;
 121     else
 122       (b->label (i)->label_val ()->ev ()->u.bval ? br_t : br_f) = true;
 123
 124   if ((br_t && br_f) || (u->nfields () == 1 && br_d))
 125     return BUB_UNCONDITIONAL;
 126
 127   bool has_other = false, other_val = false;
 128   for (unsigned int i = 0; br_d && i < u->nfields (); ++i)
 129     {
 130       AST_Field **f;
 131       u->field (f, i);
 132       if (*f != b)
 133         {
 134           AST_UnionBranch *other = dynamic_cast<AST_UnionBranch*> (*f);
 135           for (unsigned long j = 0; j < other->label_list_length (); ++j)
 136             {
 137               has_other = true;
 138               other_val = other->label (j)->label_val ()->ev ()->u.bval;
 139             }
 140         }
 141     }
 142
 143   return (br_t || (has_other && !other_val)) ? BUB_TRUE : BUB_FALSE;
 144 }
 145
 146 int
 147 be_visitor_union_cdr_op_cs::pre_process (be_decl *bd)
 148 {
 149   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_SCOPE)
 150     {
 151       return 0;
 152     }
 153
 154   // Enum val nodes are added just to help check reference
 155   // clashes, since an enum declared in our scope is not itself
 156   // a scope.
 157   if (bd->node_type () == AST_Decl::NT_enum_val)
 158     {
 159       return 0;
 160     }
 161
 162   TAO_OutStream *os = this->ctx_->stream ();
 163
 164   be_union_branch* b =
 165     dynamic_cast<be_union_branch*> (bd);
 166
 167   // Could be a type decl.
 168   if (b == 0)
 169     {
 170       return 0;
 171     }
 172
 173   this->latest_branch_ = boolean_branch (b);
 174   if (this->latest_branch_ != BUB_NONE) {
 175     if (this->latest_branch_ != BUB_UNCONDITIONAL) {
 176       *os << "if (" << (this->latest_branch_ == BUB_TRUE ? "" : "!")
 177           << (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_OUTPUT ?
 178               "_tao_union._d ()" : "_tao_discriminant") << ")" << be_idt_nl
 179           << "{" << be_idt_nl;
 180     }
 181     return 0;
 182   }
 183
 184   *os << be_nl;
 185
 186   for (unsigned long i = 0; i < b->label_list_length (); ++i)
 187     {
 188       // check if we are printing the default case
 189       if (b->label (i)->label_kind () == AST_UnionLabel::UL_default)
 190         {
 191           *os << "default:";
 192         }
 193       else
 194         {
 195           *os << "case ";
 196
 197           b->gen_label_value (os, i);
 198
 199           *os << ":";
 200         }
 201
 202       if (i == (b->label_list_length () - 1))
 203         {
 204           *os << be_idt_nl;
 205         }
 206       else
 207         {
 208           *os << be_nl;
 209         }
 210     }
 211
 212   *os << "{" << be_idt_nl;
 213
 214   return 0;
 215 }
 216
 217 int
 218 be_visitor_union_cdr_op_cs::post_process (be_decl *bd)
 219 {
 220   if (this->ctx_->sub_state () == TAO_CodeGen::TAO_CDR_SCOPE)
 221     {
 222       return 0;
 223     }
 224
 225   if (bd->node_type () == AST_Decl::NT_enum_val)
 226     {
 227       return 0;
 228     }
 229
 230   TAO_OutStream *os = this->ctx_->stream ();
 231
 232   switch (this->latest_branch_)
 233     {
 234     case BUB_NONE:
 235       *os << be_uidt_nl << "}" << be_nl << "break;" << be_uidt;
 236       break;
 237     case BUB_TRUE:
 238     case BUB_FALSE:
 239       *os << be_uidt_nl << "}" << be_uidt_nl << be_nl;
 240       break;
 241     case BUB_UNCONDITIONAL:
 242       *os << be_nl;
 243     }
 244
 245   return 0;
 246 }