//
// post_invoke_cs.cpp,v 1.8 2002/09/25 16:25:35 parsons Exp
//

// ============================================================================
//
// = LIBRARY
//    TAO IDL
//
// = FILENAME
//    post_invoke_cs.cpp
//
// = DESCRIPTION
//    Visitor generating code for post-processing of arguments following an
//    invocation.
//
// = AUTHOR
//    Jeff Parsons
//
// ============================================================================

ACE_RCSID (be_visitor_argument, 
           post_invoke_cs, 
           "post_invoke_cs.cpp,v 1.8 2002/09/25 16:25:35 parsons Exp")


// *************************************************************************
// visitor for argument to do any post invocation processing. Not all types need
// this. Only those that have an _out type need this.  In addition, interfaces
// need it because we need to convert from the interface type to the base Object
// type and vice versa.
// *************************************************************************

be_visitor_args_post_invoke_cs::be_visitor_args_post_invoke_cs (
    be_visitor_context *ctx
  )
  : be_visitor_args (ctx)
{
}

be_visitor_args_post_invoke_cs::~be_visitor_args_post_invoke_cs (void)
{
}

int
be_visitor_args_post_invoke_cs::visit_argument (be_argument *node)
{
  this->ctx_->node (node);

  // Retrieve the type of the argument.
  be_type *bt = be_type::narrow_from_decl (node->field_type ());

  if (!bt)
    {
      ACE_ERROR_RETURN ((LM_ERROR,
                         "be_visitor_args_pre_docall_cs::"
                         "visit_argument - "
                         "Bad argument type\n"),
                        -1);
    }

  // Different types have different mappings when used as in/out or
  // inout parameters. Let this visitor deal with the type

  if (bt->accept (this) == -1)
    {
      ACE_ERROR_RETURN ((LM_ERROR,
                         "be_visitor_args_post_docall_compiled_cs::"
                         "visit_argument - "
                         "cannot accept visitor\n"),
                        -1);
    }

  return 0;
}

int
be_visitor_args_post_invoke_cs::visit_interface (be_interface *node)
{
  // We must narrow the out object reference to the appropriate type.
  TAO_OutStream *os = this->ctx_->stream ();
  be_argument *arg = this->ctx_->be_node_as_argument ();

  switch (this->direction ())
    {
    case AST_Argument::dir_INOUT:
      {
        // Assign the narrowed obj reference.
        if (node->is_defined ())
          {
            *os << "CORBA::release (" << arg->local_name () << ");" << be_nl;
          }
        else
          {
            *os << "tao_" << node->flat_name () << "_release ("
                << arg->local_name () << ");" << be_nl;
          }
      }
      break;
    default:
      break;
    }

  return 0;
}

int
be_visitor_args_post_invoke_cs::visit_interface_fwd (be_interface_fwd *node)
{
  // We must narrow the out object reference to the appropriate type.
  TAO_OutStream *os = this->ctx_->stream ();
  be_argument *arg = this->ctx_->be_node_as_argument ();

  switch (this->direction ())
    {
    case AST_Argument::dir_INOUT:
      {
        // Assign the narrowed obj reference.
        if (node->full_definition ()->is_defined ())
          {
            *os << "CORBA::release (" << arg->local_name () << ");" << be_nl;
          }
        else
          {
            *os << "tao_" << node->flat_name () << "_release ("
                << arg->local_name () << ");" << be_nl;
          }
      }
      break;
    default:
      break;
    }

  return 0;
}

int
be_visitor_args_post_invoke_cs::visit_valuetype (be_valuetype *)
{
  // We must narrow the out object reference to the appropriate type.
  TAO_OutStream *os = this->ctx_->stream ();
  be_argument *arg = this->ctx_->be_node_as_argument ();

  switch (this->direction ())
    {
    case AST_Argument::dir_INOUT:
      {
        os->indent ();
        *os << "CORBA::remove_ref (" << arg->local_name ()
            << ");\n";
      }
      break;
    default:
      break;
    }

  return 0;
}

int
be_visitor_args_post_invoke_cs::visit_valuetype_fwd (be_valuetype_fwd *)
{
  // We must narrow the out object reference to the appropriate type.
  TAO_OutStream *os = this->ctx_->stream ();
  be_argument *arg = this->ctx_->be_node_as_argument ();

  switch (this->direction ())
    {
    case AST_Argument::dir_INOUT:
      {
        os->indent ();
        *os << "CORBA::remove_ref (" << arg->local_name ()
            << ");\n";
      }
      break;
    default:
      break;
    }

  return 0;
}

int
be_visitor_args_post_invoke_cs::visit_string (be_string *node)
{
  TAO_OutStream *os = this->ctx_->stream ();
  be_argument *arg = this->ctx_->be_node_as_argument ();

  switch (this->direction ())
    {
    case AST_Argument::dir_IN:
      break;
    case AST_Argument::dir_INOUT:
      os->indent ();

      if (node->width () == (long) sizeof (char))
        {
          *os << "CORBA::string_free (" << arg->local_name ()
              << ");" << be_nl;
        }
      else
        {
          *os << "CORBA::wstring_free (" << arg->local_name ()
              << ");" << be_nl;
        }

      break;
    case AST_Argument::dir_OUT:
      break;
    }

  return 0;
}

int
be_visitor_args_post_invoke_cs::visit_typedef (be_typedef *node)
{
  this->ctx_->alias (node);

  if (node->primitive_base_type ()->accept (this) == -1)
    {
      ACE_ERROR_RETURN ((LM_ERROR,
                         "be_visitor_args_post_docall_compiled_cs::"
                         "visit_typedef - "
                         "accept on primitive type failed\n"),
                        -1);
    }

  this->ctx_->alias (0);
  return 0;
}

int
be_visitor_args_post_invoke_cs::visit_component (
    be_component *node
  )
{
  return this->visit_interface (node);
}

int
be_visitor_args_post_invoke_cs::visit_component_fwd (
    be_component_fwd *node
  )
{
  return this->visit_interface_fwd (node);
}

int
be_visitor_args_post_invoke_cs::visit_eventtype (
    be_eventtype *node
  )
{
  return this->visit_valuetype (node);
}

int
be_visitor_args_post_invoke_cs::visit_eventtype_fwd (
    be_eventtype_fwd *node
  )
{
  return this->visit_valuetype_fwd (node);
}

int
be_visitor_args_post_invoke_cs::visit_home (
    be_home *node
  )
{
  return this->visit_interface (node);
}


