TAO/TAO_IDL/be/be_visitor_union/union_ch.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    union_ch.cpp
   5  *
   6  *  Visitor generating code for Unions in the client header
   7  *
   8  *  @author Aniruddha Gokhale
   9  */
  10 //=============================================================================
  11
  12 #include "union.h"
  13
  14 be_visitor_union_ch::be_visitor_union_ch (be_visitor_context *ctx)
  15   : be_visitor_union (ctx)
  16 {
  17 }
  18
  19 be_visitor_union_ch::~be_visitor_union_ch ()
  20 {
  21 }
  22
  23 int be_visitor_union_ch::visit_union (be_union *node)
  24 {
  25   if (node->cli_hdr_gen () || node->imported ())
  26     {
  27       return 0;
  28     }
  29
  30   // Evaluate the member in time for the decision to generate
  31   // the recursive typecode include in the stub source file.
  32   ACE_Unbounded_Queue<AST_Type *> list;
  33   (void) node->in_recursion (list);
  34
  35   // Instantiate a visitor context with a copy of our context. This info
  36   // will be modified based on what type of node we are visiting.
  37   be_visitor_context ctx (*this->ctx_);
  38   ctx.node (node);
  39
  40   TAO_OutStream *os = this->ctx_->stream ();
  41
  42   // Generate _var and _out class typedefs.
  43   node->gen_common_varout (os);
  44
  45   *os << be_nl_2
  46       << "class " << be_global->stub_export_macro () << " "
  47       << node->local_name () << be_nl
  48       << "{" << be_nl
  49       << "public:" << be_idt_nl
  50
  51     // Generate default and copy constructors.
  52       << node->local_name () << " (void);" << be_nl
  53       << node->local_name () << " (const " << node->local_name ()
  54       << " &);" << be_nl
  55     // Generate destructor.
  56       << "~" << node->local_name () << " (void);";
  57
  58     // Generate assignment operator.
  59   *os << be_nl_2
  60       << node->local_name () << " &operator= (const "
  61       << node->local_name () << " &);";
  62
  63   // Retrieve the disriminant type.
  64   be_type *bt = dynamic_cast<be_type*> (node->disc_type ());
  65
  66   if (!bt)
  67     {
  68       ACE_ERROR_RETURN ((LM_ERROR,
  69                          "(%N:%l) be_visitor_union_ch::"
  70                          "visit_union - "
  71                          "bad disciminant type\n"),
  72                         -1);
  73     }
  74
  75   // The discriminant type may have to be defined here if it was an enum
  76   // declaration inside of the union statement.
  77
  78   be_visitor_union_discriminant_ch ud_visitor (&ctx);
  79
  80   if (bt->accept (&ud_visitor) == -1)
  81     {
  82       ACE_ERROR_RETURN ((LM_ERROR,
  83                          ACE_TEXT ("be_visitor_union_ch::")
  84                          ACE_TEXT (" visit_union - ")
  85                          ACE_TEXT ("codegen for discriminant failed\n")),
  86                         -1);
  87     }
  88
  89   node->gen_stub_decls (os);
  90
  91   // Now generate the public defn for the union branch members. For this,
  92   // set our state to reflect what we are aiming to do.
  93   this->ctx_->state (TAO_CodeGen::TAO_UNION_PUBLIC_CH);
  94
  95   if (this->visit_scope (node) == -1)
  96     {
  97       ACE_ERROR_RETURN ((LM_ERROR,
  98                          ACE_TEXT ("be_visitor_union_ch::")
  99                          ACE_TEXT ("visit_union - ")
 100                          ACE_TEXT ("codegen for public ")
 101                          ACE_TEXT ("defn of union members\n")),
 102                         -1);
 103     }
 104
 105   // Now check if we need to generate the _default () method.
 106   be_union::DefaultValue dv;
 107
 108   if (node->default_value (dv) == -1)
 109     {
 110       ACE_ERROR_RETURN ((LM_ERROR,
 111                          ACE_TEXT ("be_visitor_union_ch::")
 112                          ACE_TEXT ("visit_union - ")
 113                          ACE_TEXT ("computing default ")
 114                          ACE_TEXT ("value failed\n")),
 115                         -1);
 116     }
 117
 118   if ((dv.computed_ != 0) && (node->default_index () == -1))
 119     {
 120       TAO_INSERT_COMMENT (os);
 121
 122       // Only if all cases are not covered AND there is no explicit
 123       // default, we get the _default () method.
 124       *os << be_nl_2
 125           << "void _default (void);";
 126     }
 127
 128   *os << be_uidt_nl;
 129
 130   // Now generate the private data members of the union.
 131   *os << "private:" << be_idt_nl;
 132   *os << bt->nested_type_name (node) << " disc_;" << be_nl_2;
 133
 134   // The members are inside of a union.
 135   *os << "union" << be_nl;
 136   *os << "{" << be_idt;
 137
 138   this->ctx_->state (TAO_CodeGen::TAO_UNION_PRIVATE_CH);
 139
 140   if (this->visit_scope (node) == -1)
 141     {
 142       ACE_ERROR_RETURN ((LM_ERROR,
 143                          ACE_TEXT ("be_visitor_union_ch::")
 144                          ACE_TEXT ("visit_union - ")
 145                          ACE_TEXT ("codegen for private")
 146                          ACE_TEXT (" members of union\n")),
 147                         -1);
 148     }
 149
 150   *os << be_uidt_nl;
 151   *os << "} u_;";
 152
 153   // The reset method (TAO extension).
 154   *os << be_nl_2
 155       << "/// TAO extension - frees any allocated storage." << be_nl;
 156   *os << "void _reset (void);";
 157
 158   *os << be_uidt_nl << "};";
 159
 160   if (be_global->tc_support ())
 161     {
 162       ctx = *this->ctx_;
 163       be_visitor_typecode_decl tc_visitor (&ctx);
 164
 165       if (node->accept (&tc_visitor) == -1)
 166         {
 167           ACE_ERROR_RETURN ((LM_ERROR,
 168                              ACE_TEXT ("be_visitor_union_ch::")
 169                              ACE_TEXT ("visit_union - ")
 170                              ACE_TEXT ("TypeCode declaration failed\n")),
 171                             -1);
 172         }
 173     }
 174
 175   node->cli_hdr_gen (true);
 176   return 0;
 177 }