URL https://opencores.org/ocsvn/openrisc/openrisc/trunk
Subversion Repositories openrisc

[/] [openrisc/] [trunk/] [gnu-dev/] [or1k-gcc/] [gcc/] [go/] [gofrontend/] [statements.cc] - Blame information for rev 715

Go to most recent revision | Details | Compare with Previous | View Log

// statements.cc -- Go frontend statements.
 
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
 
#include "go-system.h"
 
#include <gmp.h>
 
#include "go-c.h"
#include "types.h"
#include "expressions.h"
#include "gogo.h"
#include "runtime.h"
#include "backend.h"
#include "statements.h"
#include "ast-dump.h"
 
// Class Statement.
 
Statement::Statement(Statement_classification classification,
                     Location location)
  : classification_(classification), location_(location)
{
}
 
Statement::~Statement()
{
}
 
// Traverse the tree.  The work of walking the components is handled
// by the subclasses.
 
int
Statement::traverse(Block* block, size_t* pindex, Traverse* traverse)
{
  if (this->classification_ == STATEMENT_ERROR)
    return TRAVERSE_CONTINUE;
 
  unsigned int traverse_mask = traverse->traverse_mask();
 
  if ((traverse_mask & Traverse::traverse_statements) != 0)
    {
      int t = traverse->statement(block, pindex, this);
      if (t == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
      else if (t == TRAVERSE_SKIP_COMPONENTS)
        return TRAVERSE_CONTINUE;
    }
 
  // No point in checking traverse_mask here--a statement may contain
  // other blocks or statements, and if we got here we always want to
  // walk them.
  return this->do_traverse(traverse);
}
 
// Traverse the contents of a statement.
 
int
Statement::traverse_contents(Traverse* traverse)
{
  return this->do_traverse(traverse);
}
 
// Traverse assignments.
 
bool
Statement::traverse_assignments(Traverse_assignments* tassign)
{
  if (this->classification_ == STATEMENT_ERROR)
    return false;
  return this->do_traverse_assignments(tassign);
}
 
// Traverse an expression in a statement.  This is a helper function
// for child classes.
 
int
Statement::traverse_expression(Traverse* traverse, Expression** expr)
{
  if ((traverse->traverse_mask()
       & (Traverse::traverse_types | Traverse::traverse_expressions)) == 0)
    return TRAVERSE_CONTINUE;
  return Expression::traverse(expr, traverse);
}
 
// Traverse an expression list in a statement.  This is a helper
// function for child classes.
 
int
Statement::traverse_expression_list(Traverse* traverse,
                                    Expression_list* expr_list)
{
  if (expr_list == NULL)
    return TRAVERSE_CONTINUE;
  if ((traverse->traverse_mask()
       & (Traverse::traverse_types | Traverse::traverse_expressions)) == 0)
    return TRAVERSE_CONTINUE;
  return expr_list->traverse(traverse);
}
 
// Traverse a type in a statement.  This is a helper function for
// child classes.
 
int
Statement::traverse_type(Traverse* traverse, Type* type)
{
  if ((traverse->traverse_mask()
       & (Traverse::traverse_types | Traverse::traverse_expressions)) == 0)
    return TRAVERSE_CONTINUE;
  return Type::traverse(type, traverse);
}
 
// Set type information for unnamed constants.  This is really done by
// the child class.
 
void
Statement::determine_types()
{
  this->do_determine_types();
}
 
// If this is a thunk statement, return it.
 
Thunk_statement*
Statement::thunk_statement()
{
  Thunk_statement* ret = this->convert<Thunk_statement, STATEMENT_GO>();
  if (ret == NULL)
    ret = this->convert<Thunk_statement, STATEMENT_DEFER>();
  return ret;
}
 
// Convert a Statement to the backend representation.  This is really
// done by the child class.
 
Bstatement*
Statement::get_backend(Translate_context* context)
{
  if (this->classification_ == STATEMENT_ERROR)
    return context->backend()->error_statement();
  return this->do_get_backend(context);
}
 
// Dump AST representation for a statement to a dump context.
 
void
Statement::dump_statement(Ast_dump_context* ast_dump_context) const
{
  this->do_dump_statement(ast_dump_context);
}
 
// Note that this statement is erroneous.  This is called by children
// when they discover an error.
 
void
Statement::set_is_error()
{
  this->classification_ = STATEMENT_ERROR;
}
 
// For children to call to report an error conveniently.
 
void
Statement::report_error(const char* msg)
{
  error_at(this->location_, "%s", msg);
  this->set_is_error();
}
 
// An error statement, used to avoid crashing after we report an
// error.
 
class Error_statement : public Statement
{
 public:
  Error_statement(Location location)
    : Statement(STATEMENT_ERROR, location)
  { }
 
 protected:
  int
  do_traverse(Traverse*)
  { return TRAVERSE_CONTINUE; }
 
  Bstatement*
  do_get_backend(Translate_context*)
  { go_unreachable(); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
};
 
// Dump the AST representation for an error statement.
 
void
Error_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "Error statement" << std::endl;
}
 
// Make an error statement.
 
Statement*
Statement::make_error_statement(Location location)
{
  return new Error_statement(location);
}
 
// Class Variable_declaration_statement.
 
Variable_declaration_statement::Variable_declaration_statement(
    Named_object* var)
  : Statement(STATEMENT_VARIABLE_DECLARATION, var->var_value()->location()),
    var_(var)
{
}
 
// We don't actually traverse the variable here; it was traversed
// while traversing the Block.
 
int
Variable_declaration_statement::do_traverse(Traverse*)
{
  return TRAVERSE_CONTINUE;
}
 
// Traverse the assignments in a variable declaration.  Note that this
// traversal is different from the usual traversal.
 
bool
Variable_declaration_statement::do_traverse_assignments(
    Traverse_assignments* tassign)
{
  tassign->initialize_variable(this->var_);
  return true;
}
 
// Lower the variable's initialization expression.
 
Statement*
Variable_declaration_statement::do_lower(Gogo* gogo, Named_object* function,
                                         Block*, Statement_inserter* inserter)
{
  this->var_->var_value()->lower_init_expression(gogo, function, inserter);
  return this;
}
 
// Convert a variable declaration to the backend representation.
 
Bstatement*
Variable_declaration_statement::do_get_backend(Translate_context* context)
{
  Variable* var = this->var_->var_value();
  Bvariable* bvar = this->var_->get_backend_variable(context->gogo(),
                                                     context->function());
  tree init = var->get_init_tree(context->gogo(), context->function());
  Bexpression* binit = init == NULL ? NULL : tree_to_expr(init);
 
  if (!var->is_in_heap())
    {
      go_assert(binit != NULL);
      return context->backend()->init_statement(bvar, binit);
    }
 
  // Something takes the address of this variable, so the value is
  // stored in the heap.  Initialize it to newly allocated memory
  // space, and assign the initial value to the new space.
  Location loc = this->location();
  Named_object* newfn = context->gogo()->lookup_global("new");
  go_assert(newfn != NULL && newfn->is_function_declaration());
  Expression* func = Expression::make_func_reference(newfn, NULL, loc);
  Expression_list* params = new Expression_list();
  params->push_back(Expression::make_type(var->type(), loc));
  Expression* call = Expression::make_call(func, params, false, loc);
  context->gogo()->lower_expression(context->function(), NULL, &call);
  Temporary_statement* temp = Statement::make_temporary(NULL, call, loc);
  Bstatement* btemp = temp->get_backend(context);
 
  Bstatement* set = NULL;
  if (binit != NULL)
    {
      Expression* e = Expression::make_temporary_reference(temp, loc);
      e = Expression::make_unary(OPERATOR_MULT, e, loc);
      Bexpression* be = tree_to_expr(e->get_tree(context));
      set = context->backend()->assignment_statement(be, binit, loc);
    }
 
  Expression* ref = Expression::make_temporary_reference(temp, loc);
  Bexpression* bref = tree_to_expr(ref->get_tree(context));
  Bstatement* sinit = context->backend()->init_statement(bvar, bref);
 
  std::vector<Bstatement*> stats;
  stats.reserve(3);
  stats.push_back(btemp);
  if (set != NULL)
    stats.push_back(set);
  stats.push_back(sinit);
  return context->backend()->statement_list(stats);
}
 
// Dump the AST representation for a variable declaration.
 
void
Variable_declaration_statement::do_dump_statement(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
 
  go_assert(var_->is_variable());
  ast_dump_context->ostream() << "var " << this->var_->name() <<  " ";
  Variable* var = this->var_->var_value();
  if (var->has_type())
    {
      ast_dump_context->dump_type(var->type());
      ast_dump_context->ostream() << " ";
    }
  if (var->init() != NULL)
    {
      ast_dump_context->ostream() <<  "= ";
      ast_dump_context->dump_expression(var->init());
    }
  ast_dump_context->ostream() << std::endl;
}
 
// Make a variable declaration.
 
Statement*
Statement::make_variable_declaration(Named_object* var)
{
  return new Variable_declaration_statement(var);
}
 
// Class Temporary_statement.
 
// Return the type of the temporary variable.
 
Type*
Temporary_statement::type() const
{
  return this->type_ != NULL ? this->type_ : this->init_->type();
}
 
// Traversal.
 
int
Temporary_statement::do_traverse(Traverse* traverse)
{
  if (this->type_ != NULL
      && this->traverse_type(traverse, this->type_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  if (this->init_ == NULL)
    return TRAVERSE_CONTINUE;
  else
    return this->traverse_expression(traverse, &this->init_);
}
 
// Traverse assignments.
 
bool
Temporary_statement::do_traverse_assignments(Traverse_assignments* tassign)
{
  if (this->init_ == NULL)
    return false;
  tassign->value(&this->init_, true, true);
  return true;
}
 
// Determine types.
 
void
Temporary_statement::do_determine_types()
{
  if (this->type_ != NULL && this->type_->is_abstract())
    this->type_ = this->type_->make_non_abstract_type();
 
  if (this->init_ != NULL)
    {
      if (this->type_ == NULL)
        this->init_->determine_type_no_context();
      else
        {
          Type_context context(this->type_, false);
          this->init_->determine_type(&context);
        }
    }
 
  if (this->type_ == NULL)
    {
      this->type_ = this->init_->type();
      go_assert(!this->type_->is_abstract());
    }
}
 
// Check types.
 
void
Temporary_statement::do_check_types(Gogo*)
{
  if (this->type_ != NULL && this->init_ != NULL)
    {
      std::string reason;
      bool ok;
      if (this->are_hidden_fields_ok_)
        ok = Type::are_assignable_hidden_ok(this->type_, this->init_->type(),
                                            &reason);
      else
        ok = Type::are_assignable(this->type_, this->init_->type(), &reason);
      if (!ok)
        {
          if (reason.empty())
            error_at(this->location(), "incompatible types in assignment");
          else
            error_at(this->location(), "incompatible types in assignment (%s)",
                     reason.c_str());
          this->set_is_error();
        }
    }
}
 
// Convert to backend representation.
 
Bstatement*
Temporary_statement::do_get_backend(Translate_context* context)
{
  go_assert(this->bvariable_ == NULL);
 
  // FIXME: Permitting FUNCTION to be NULL here is a temporary measure
  // until we have a better representation of the init function.
  Named_object* function = context->function();
  Bfunction* bfunction;
  if (function == NULL)
    bfunction = NULL;
  else
    bfunction = tree_to_function(function->func_value()->get_decl());
 
  Btype* btype = this->type()->get_backend(context->gogo());
 
  Bexpression* binit;
  if (this->init_ == NULL)
    binit = NULL;
  else if (this->type_ == NULL)
    binit = tree_to_expr(this->init_->get_tree(context));
  else
    {
      Expression* init = Expression::make_cast(this->type_, this->init_,
                                               this->location());
      context->gogo()->lower_expression(context->function(), NULL, &init);
      binit = tree_to_expr(init->get_tree(context));
    }
 
  Bstatement* statement;
  this->bvariable_ =
    context->backend()->temporary_variable(bfunction, context->bblock(),
                                           btype, binit,
                                           this->is_address_taken_,
                                           this->location(), &statement);
  return statement;
}
 
// Return the backend variable.
 
Bvariable*
Temporary_statement::get_backend_variable(Translate_context* context) const
{
  if (this->bvariable_ == NULL)
    {
      go_assert(saw_errors());
      return context->backend()->error_variable();
    }
  return this->bvariable_;
}
 
// Dump the AST represemtation for a temporary statement
 
void
Temporary_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_temp_variable_name(this);
  if (this->type_ != NULL)
    {
      ast_dump_context->ostream() << " ";
      ast_dump_context->dump_type(this->type_);
    }
  if (this->init_ != NULL)
    {
      ast_dump_context->ostream() << " = ";
      ast_dump_context->dump_expression(this->init_);
    }
  ast_dump_context->ostream() << std::endl;
}
 
// Make and initialize a temporary variable in BLOCK.
 
Temporary_statement*
Statement::make_temporary(Type* type, Expression* init,
                          Location location)
{
  return new Temporary_statement(type, init, location);
}
 
// An assignment statement.
 
class Assignment_statement : public Statement
{
 public:
  Assignment_statement(Expression* lhs, Expression* rhs,
                       Location location)
    : Statement(STATEMENT_ASSIGNMENT, location),
      lhs_(lhs), rhs_(rhs), are_hidden_fields_ok_(false)
  { }
 
  // Note that it is OK for this assignment statement to set hidden
  // fields.
  void
  set_hidden_fields_are_ok()
  { this->are_hidden_fields_ok_ = true; }
 
 protected:
  int
  do_traverse(Traverse* traverse);
 
  bool
  do_traverse_assignments(Traverse_assignments*);
 
  void
  do_determine_types();
 
  void
  do_check_types(Gogo*);
 
  Bstatement*
  do_get_backend(Translate_context*);
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // Left hand side--the lvalue.
  Expression* lhs_;
  // Right hand side--the rvalue.
  Expression* rhs_;
  // True if this statement may set hidden fields in the assignment
  // statement.  This is used for generated method stubs.
  bool are_hidden_fields_ok_;
};
 
// Traversal.
 
int
Assignment_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->lhs_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->traverse_expression(traverse, &this->rhs_);
}
 
bool
Assignment_statement::do_traverse_assignments(Traverse_assignments* tassign)
{
  tassign->assignment(&this->lhs_, &this->rhs_);
  return true;
}
 
// Set types for the assignment.
 
void
Assignment_statement::do_determine_types()
{
  this->lhs_->determine_type_no_context();
  Type_context context(this->lhs_->type(), false);
  this->rhs_->determine_type(&context);
}
 
// Check types for an assignment.
 
void
Assignment_statement::do_check_types(Gogo*)
{
  // The left hand side must be either addressable, a map index
  // expression, or the blank identifier.
  if (!this->lhs_->is_addressable()
      && this->lhs_->map_index_expression() == NULL
      && !this->lhs_->is_sink_expression())
    {
      if (!this->lhs_->type()->is_error())
        this->report_error(_("invalid left hand side of assignment"));
      return;
    }
 
  Type* lhs_type = this->lhs_->type();
  Type* rhs_type = this->rhs_->type();
  std::string reason;
  bool ok;
  if (this->are_hidden_fields_ok_)
    ok = Type::are_assignable_hidden_ok(lhs_type, rhs_type, &reason);
  else
    ok = Type::are_assignable(lhs_type, rhs_type, &reason);
  if (!ok)
    {
      if (reason.empty())
        error_at(this->location(), "incompatible types in assignment");
      else
        error_at(this->location(), "incompatible types in assignment (%s)",
                 reason.c_str());
      this->set_is_error();
    }
 
  if (lhs_type->is_error() || rhs_type->is_error())
    this->set_is_error();
}
 
// Convert an assignment statement to the backend representation.
 
Bstatement*
Assignment_statement::do_get_backend(Translate_context* context)
{
  tree rhs_tree = this->rhs_->get_tree(context);
  if (this->lhs_->is_sink_expression())
    return context->backend()->expression_statement(tree_to_expr(rhs_tree));
  tree lhs_tree = this->lhs_->get_tree(context);
  rhs_tree = Expression::convert_for_assignment(context, this->lhs_->type(),
                                                this->rhs_->type(), rhs_tree,
                                                this->location());
  return context->backend()->assignment_statement(tree_to_expr(lhs_tree),
                                                  tree_to_expr(rhs_tree),
                                                  this->location());
}
 
// Dump the AST representation for an assignment statement.
 
void
Assignment_statement::do_dump_statement(Ast_dump_context* ast_dump_context)
    const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(this->lhs_);
  ast_dump_context->ostream() << " = " ;
  ast_dump_context->dump_expression(this->rhs_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make an assignment statement.
 
Statement*
Statement::make_assignment(Expression* lhs, Expression* rhs,
                           Location location)
{
  return new Assignment_statement(lhs, rhs, location);
}
 
// The Move_subexpressions class is used to move all top-level
// subexpressions of an expression.  This is used for things like
// index expressions in which we must evaluate the index value before
// it can be changed by a multiple assignment.
 
class Move_subexpressions : public Traverse
{
 public:
  Move_subexpressions(int skip, Block* block)
    : Traverse(traverse_expressions),
      skip_(skip), block_(block)
  { }
 
 protected:
  int
  expression(Expression**);
 
 private:
  // The number of subexpressions to skip moving.  This is used to
  // avoid moving the array itself, as we only need to move the index.
  int skip_;
  // The block where new temporary variables should be added.
  Block* block_;
};
 
int
Move_subexpressions::expression(Expression** pexpr)
{
  if (this->skip_ > 0)
    --this->skip_;
  else if ((*pexpr)->temporary_reference_expression() == NULL)
    {
      Location loc = (*pexpr)->location();
      Temporary_statement* temp = Statement::make_temporary(NULL, *pexpr, loc);
      this->block_->add_statement(temp);
      *pexpr = Expression::make_temporary_reference(temp, loc);
    }
  // We only need to move top-level subexpressions.
  return TRAVERSE_SKIP_COMPONENTS;
}
 
// The Move_ordered_evals class is used to find any subexpressions of
// an expression that have an evaluation order dependency.  It creates
// temporary variables to hold them.
 
class Move_ordered_evals : public Traverse
{
 public:
  Move_ordered_evals(Block* block)
    : Traverse(traverse_expressions),
      block_(block)
  { }
 
 protected:
  int
  expression(Expression**);
 
 private:
  // The block where new temporary variables should be added.
  Block* block_;
};
 
int
Move_ordered_evals::expression(Expression** pexpr)
{
  // We have to look at subexpressions first.
  if ((*pexpr)->traverse_subexpressions(this) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
 
  int i;
  if ((*pexpr)->must_eval_subexpressions_in_order(&i))
    {
      Move_subexpressions ms(i, this->block_);
      if ((*pexpr)->traverse_subexpressions(&ms) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
 
  if ((*pexpr)->must_eval_in_order())
    {
      Location loc = (*pexpr)->location();
      Temporary_statement* temp = Statement::make_temporary(NULL, *pexpr, loc);
      this->block_->add_statement(temp);
      *pexpr = Expression::make_temporary_reference(temp, loc);
    }
  return TRAVERSE_SKIP_COMPONENTS;
}
 
// An assignment operation statement.
 
class Assignment_operation_statement : public Statement
{
 public:
  Assignment_operation_statement(Operator op, Expression* lhs, Expression* rhs,
                                 Location location)
    : Statement(STATEMENT_ASSIGNMENT_OPERATION, location),
      op_(op), lhs_(lhs), rhs_(rhs)
  { }
 
 protected:
  int
  do_traverse(Traverse*);
 
  bool
  do_traverse_assignments(Traverse_assignments*)
  { go_unreachable(); }
 
  Statement*
  do_lower(Gogo*, Named_object*, Block*, Statement_inserter*);
 
  Bstatement*
  do_get_backend(Translate_context*)
  { go_unreachable(); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // The operator (OPERATOR_PLUSEQ, etc.).
  Operator op_;
  // Left hand side.
  Expression* lhs_;
  // Right hand side.
  Expression* rhs_;
};
 
// Traversal.
 
int
Assignment_operation_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->lhs_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->traverse_expression(traverse, &this->rhs_);
}
 
// Lower an assignment operation statement to a regular assignment
// statement.
 
Statement*
Assignment_operation_statement::do_lower(Gogo*, Named_object*,
                                         Block* enclosing, Statement_inserter*)
{
  Location loc = this->location();
 
  // We have to evaluate the left hand side expression only once.  We
  // do this by moving out any expression with side effects.
  Block* b = new Block(enclosing, loc);
  Move_ordered_evals moe(b);
  this->lhs_->traverse_subexpressions(&moe);
 
  Expression* lval = this->lhs_->copy();
 
  Operator op;
  switch (this->op_)
    {
    case OPERATOR_PLUSEQ:
      op = OPERATOR_PLUS;
      break;
    case OPERATOR_MINUSEQ:
      op = OPERATOR_MINUS;
      break;
    case OPERATOR_OREQ:
      op = OPERATOR_OR;
      break;
    case OPERATOR_XOREQ:
      op = OPERATOR_XOR;
      break;
    case OPERATOR_MULTEQ:
      op = OPERATOR_MULT;
      break;
    case OPERATOR_DIVEQ:
      op = OPERATOR_DIV;
      break;
    case OPERATOR_MODEQ:
      op = OPERATOR_MOD;
      break;
    case OPERATOR_LSHIFTEQ:
      op = OPERATOR_LSHIFT;
      break;
    case OPERATOR_RSHIFTEQ:
      op = OPERATOR_RSHIFT;
      break;
    case OPERATOR_ANDEQ:
      op = OPERATOR_AND;
      break;
    case OPERATOR_BITCLEAREQ:
      op = OPERATOR_BITCLEAR;
      break;
    default:
      go_unreachable();
    }
 
  Expression* binop = Expression::make_binary(op, lval, this->rhs_, loc);
  Statement* s = Statement::make_assignment(this->lhs_, binop, loc);
  if (b->statements()->empty())
    {
      delete b;
      return s;
    }
  else
    {
      b->add_statement(s);
      return Statement::make_block_statement(b, loc);
    }
}
 
// Dump the AST representation for an assignment operation statement
 
void
Assignment_operation_statement::do_dump_statement(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(this->lhs_);
  ast_dump_context->dump_operator(this->op_);
  ast_dump_context->dump_expression(this->rhs_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make an assignment operation statement.
 
Statement*
Statement::make_assignment_operation(Operator op, Expression* lhs,
                                     Expression* rhs, Location location)
{
  return new Assignment_operation_statement(op, lhs, rhs, location);
}
 
// A tuple assignment statement.  This differs from an assignment
// statement in that the right-hand-side expressions are evaluated in
// parallel.
 
class Tuple_assignment_statement : public Statement
{
 public:
  Tuple_assignment_statement(Expression_list* lhs, Expression_list* rhs,
                             Location location)
    : Statement(STATEMENT_TUPLE_ASSIGNMENT, location),
      lhs_(lhs), rhs_(rhs), are_hidden_fields_ok_(false)
  { }
 
  // Note that it is OK for this assignment statement to set hidden
  // fields.
  void
  set_hidden_fields_are_ok()
  { this->are_hidden_fields_ok_ = true; }
 
 protected:
  int
  do_traverse(Traverse* traverse);
 
  bool
  do_traverse_assignments(Traverse_assignments*)
  { go_unreachable(); }
 
  Statement*
  do_lower(Gogo*, Named_object*, Block*, Statement_inserter*);
 
  Bstatement*
  do_get_backend(Translate_context*)
  { go_unreachable(); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // Left hand side--a list of lvalues.
  Expression_list* lhs_;
  // Right hand side--a list of rvalues.
  Expression_list* rhs_;
  // True if this statement may set hidden fields in the assignment
  // statement.  This is used for generated method stubs.
  bool are_hidden_fields_ok_;
};
 
// Traversal.
 
int
Tuple_assignment_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression_list(traverse, this->lhs_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->traverse_expression_list(traverse, this->rhs_);
}
 
// Lower a tuple assignment.  We use temporary variables to split it
// up into a set of single assignments.
 
Statement*
Tuple_assignment_statement::do_lower(Gogo*, Named_object*, Block* enclosing,
                                     Statement_inserter*)
{
  Location loc = this->location();
 
  Block* b = new Block(enclosing, loc);
 
  // First move out any subexpressions on the left hand side.  The
  // right hand side will be evaluated in the required order anyhow.
  Move_ordered_evals moe(b);
  for (Expression_list::iterator plhs = this->lhs_->begin();
       plhs != this->lhs_->end();
       ++plhs)
    Expression::traverse(&*plhs, &moe);
 
  std::vector<Temporary_statement*> temps;
  temps.reserve(this->lhs_->size());
 
  Expression_list::const_iterator prhs = this->rhs_->begin();
  for (Expression_list::const_iterator plhs = this->lhs_->begin();
       plhs != this->lhs_->end();
       ++plhs, ++prhs)
    {
      go_assert(prhs != this->rhs_->end());
 
      if ((*plhs)->is_error_expression()
          || (*plhs)->type()->is_error()
          || (*prhs)->is_error_expression()
          || (*prhs)->type()->is_error())
        continue;
 
      if ((*plhs)->is_sink_expression())
        {
          b->add_statement(Statement::make_statement(*prhs, true));
          continue;
        }
 
      Temporary_statement* temp = Statement::make_temporary((*plhs)->type(),
                                                            *prhs, loc);
      if (this->are_hidden_fields_ok_)
        temp->set_hidden_fields_are_ok();
      b->add_statement(temp);
      temps.push_back(temp);
 
    }
  go_assert(prhs == this->rhs_->end());
 
  prhs = this->rhs_->begin();
  std::vector<Temporary_statement*>::const_iterator ptemp = temps.begin();
  for (Expression_list::const_iterator plhs = this->lhs_->begin();
       plhs != this->lhs_->end();
       ++plhs, ++prhs)
    {
      if ((*plhs)->is_error_expression()
          || (*plhs)->type()->is_error()
          || (*prhs)->is_error_expression()
          || (*prhs)->type()->is_error())
        continue;
 
      if ((*plhs)->is_sink_expression())
        continue;
 
      Expression* ref = Expression::make_temporary_reference(*ptemp, loc);
      Statement* s = Statement::make_assignment(*plhs, ref, loc);
      if (this->are_hidden_fields_ok_)
        {
          Assignment_statement* as = static_cast<Assignment_statement*>(s);
          as->set_hidden_fields_are_ok();
        }
      b->add_statement(s);
      ++ptemp;
    }
  go_assert(ptemp == temps.end());
 
  return Statement::make_block_statement(b, loc);
}
 
// Dump the AST representation for a tuple assignment statement.
 
void
Tuple_assignment_statement::do_dump_statement(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression_list(this->lhs_);
  ast_dump_context->ostream() << " = ";
  ast_dump_context->dump_expression_list(this->rhs_);
  ast_dump_context->ostream()  << std::endl;
}
 
// Make a tuple assignment statement.
 
Statement*
Statement::make_tuple_assignment(Expression_list* lhs, Expression_list* rhs,
                                 Location location)
{
  return new Tuple_assignment_statement(lhs, rhs, location);
}
 
// A tuple assignment from a map index expression.
//   v, ok = m[k]
 
class Tuple_map_assignment_statement : public Statement
{
public:
  Tuple_map_assignment_statement(Expression* val, Expression* present,
                                 Expression* map_index,
                                 Location location)
    : Statement(STATEMENT_TUPLE_MAP_ASSIGNMENT, location),
      val_(val), present_(present), map_index_(map_index)
  { }
 
 protected:
  int
  do_traverse(Traverse* traverse);
 
  bool
  do_traverse_assignments(Traverse_assignments*)
  { go_unreachable(); }
 
  Statement*
  do_lower(Gogo*, Named_object*, Block*, Statement_inserter*);
 
  Bstatement*
  do_get_backend(Translate_context*)
  { go_unreachable(); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // Lvalue which receives the value from the map.
  Expression* val_;
  // Lvalue which receives whether the key value was present.
  Expression* present_;
  // The map index expression.
  Expression* map_index_;
};
 
// Traversal.
 
int
Tuple_map_assignment_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT
      || this->traverse_expression(traverse, &this->present_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->traverse_expression(traverse, &this->map_index_);
}
 
// Lower a tuple map assignment.
 
Statement*
Tuple_map_assignment_statement::do_lower(Gogo*, Named_object*,
                                         Block* enclosing, Statement_inserter*)
{
  Location loc = this->location();
 
  Map_index_expression* map_index = this->map_index_->map_index_expression();
  if (map_index == NULL)
    {
      this->report_error(_("expected map index on right hand side"));
      return Statement::make_error_statement(loc);
    }
  Map_type* map_type = map_index->get_map_type();
  if (map_type == NULL)
    return Statement::make_error_statement(loc);
 
  Block* b = new Block(enclosing, loc);
 
  // Move out any subexpressions to make sure that functions are
  // called in the required order.
  Move_ordered_evals moe(b);
  this->val_->traverse_subexpressions(&moe);
  this->present_->traverse_subexpressions(&moe);
 
  // Copy the key value into a temporary so that we can take its
  // address without pushing the value onto the heap.
 
  // var key_temp KEY_TYPE = MAP_INDEX
  Temporary_statement* key_temp =
    Statement::make_temporary(map_type->key_type(), map_index->index(), loc);
  b->add_statement(key_temp);
 
  // var val_temp VAL_TYPE
  Temporary_statement* val_temp =
    Statement::make_temporary(map_type->val_type(), NULL, loc);
  b->add_statement(val_temp);
 
  // var present_temp bool
  Temporary_statement* present_temp =
    Statement::make_temporary(Type::lookup_bool_type(), NULL, loc);
  b->add_statement(present_temp);
 
  // present_temp = mapaccess2(DESCRIPTOR, MAP, &key_temp, &val_temp)
  Expression* a1 = Expression::make_type_descriptor(map_type, loc);
  Expression* a2 = map_index->map();
  Temporary_reference_expression* ref =
    Expression::make_temporary_reference(key_temp, loc);
  Expression* a3 = Expression::make_unary(OPERATOR_AND, ref, loc);
  ref = Expression::make_temporary_reference(val_temp, loc);
  Expression* a4 = Expression::make_unary(OPERATOR_AND, ref, loc);
  Expression* call = Runtime::make_call(Runtime::MAPACCESS2, loc, 4,
                                        a1, a2, a3, a4);
 
  ref = Expression::make_temporary_reference(present_temp, loc);
  ref->set_is_lvalue();
  Statement* s = Statement::make_assignment(ref, call, loc);
  b->add_statement(s);
 
  // val = val_temp
  ref = Expression::make_temporary_reference(val_temp, loc);
  s = Statement::make_assignment(this->val_, ref, loc);
  b->add_statement(s);
 
  // present = present_temp
  ref = Expression::make_temporary_reference(present_temp, loc);
  s = Statement::make_assignment(this->present_, ref, loc);
  b->add_statement(s);
 
  return Statement::make_block_statement(b, loc);
}
 
// Dump the AST representation for a tuple map assignment statement.
 
void
Tuple_map_assignment_statement::do_dump_statement(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(this->val_);
  ast_dump_context->ostream() << ", ";
  ast_dump_context->dump_expression(this->present_);
  ast_dump_context->ostream() << " = ";
  ast_dump_context->dump_expression(this->map_index_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make a map assignment statement which returns a pair of values.
 
Statement*
Statement::make_tuple_map_assignment(Expression* val, Expression* present,
                                     Expression* map_index,
                                     Location location)
{
  return new Tuple_map_assignment_statement(val, present, map_index, location);
}
 
// Assign a pair of entries to a map.
//   m[k] = v, p
 
class Map_assignment_statement : public Statement
{
 public:
  Map_assignment_statement(Expression* map_index,
                           Expression* val, Expression* should_set,
                           Location location)
    : Statement(STATEMENT_MAP_ASSIGNMENT, location),
      map_index_(map_index), val_(val), should_set_(should_set)
  { }
 
 protected:
  int
  do_traverse(Traverse* traverse);
 
  bool
  do_traverse_assignments(Traverse_assignments*)
  { go_unreachable(); }
 
  Statement*
  do_lower(Gogo*, Named_object*, Block*, Statement_inserter*);
 
  Bstatement*
  do_get_backend(Translate_context*)
  { go_unreachable(); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // A reference to the map index which should be set or deleted.
  Expression* map_index_;
  // The value to add to the map.
  Expression* val_;
  // Whether or not to add the value.
  Expression* should_set_;
};
 
// Traverse a map assignment.
 
int
Map_assignment_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->map_index_) == TRAVERSE_EXIT
      || this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->traverse_expression(traverse, &this->should_set_);
}
 
// Lower a map assignment to a function call.
 
Statement*
Map_assignment_statement::do_lower(Gogo*, Named_object*, Block* enclosing,
                                   Statement_inserter*)
{
  Location loc = this->location();
 
  Map_index_expression* map_index = this->map_index_->map_index_expression();
  if (map_index == NULL)
    {
      this->report_error(_("expected map index on left hand side"));
      return Statement::make_error_statement(loc);
    }
  Map_type* map_type = map_index->get_map_type();
  if (map_type == NULL)
    return Statement::make_error_statement(loc);
 
  Block* b = new Block(enclosing, loc);
 
  // Evaluate the map first to get order of evaluation right.
  // map_temp := m // we are evaluating m[k] = v, p
  Temporary_statement* map_temp = Statement::make_temporary(map_type,
                                                            map_index->map(),
                                                            loc);
  b->add_statement(map_temp);
 
  // var key_temp MAP_KEY_TYPE = k
  Temporary_statement* key_temp =
    Statement::make_temporary(map_type->key_type(), map_index->index(), loc);
  b->add_statement(key_temp);
 
  // var val_temp MAP_VAL_TYPE = v
  Temporary_statement* val_temp =
    Statement::make_temporary(map_type->val_type(), this->val_, loc);
  b->add_statement(val_temp);
 
  // var insert_temp bool = p
  Temporary_statement* insert_temp =
    Statement::make_temporary(Type::lookup_bool_type(), this->should_set_,
                              loc);
  b->add_statement(insert_temp);
 
  // mapassign2(map_temp, &key_temp, &val_temp, p)
  Expression* p1 = Expression::make_temporary_reference(map_temp, loc);
  Expression* ref = Expression::make_temporary_reference(key_temp, loc);
  Expression* p2 = Expression::make_unary(OPERATOR_AND, ref, loc);
  ref = Expression::make_temporary_reference(val_temp, loc);
  Expression* p3 = Expression::make_unary(OPERATOR_AND, ref, loc);
  Expression* p4 = Expression::make_temporary_reference(insert_temp, loc);
  Expression* call = Runtime::make_call(Runtime::MAPASSIGN2, loc, 4,
                                        p1, p2, p3, p4);
  Statement* s = Statement::make_statement(call, true);
  b->add_statement(s);
 
  return Statement::make_block_statement(b, loc);
}
 
// Dump the AST representation for a map assignment statement.
 
void
Map_assignment_statement::do_dump_statement(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(this->map_index_);
  ast_dump_context->ostream() << " = ";
  ast_dump_context->dump_expression(this->val_);
  ast_dump_context->ostream() << ", ";
  ast_dump_context->dump_expression(this->should_set_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make a statement which assigns a pair of entries to a map.
 
Statement*
Statement::make_map_assignment(Expression* map_index,
                               Expression* val, Expression* should_set,
                               Location location)
{
  return new Map_assignment_statement(map_index, val, should_set, location);
}
 
// A tuple assignment from a receive statement.
 
class Tuple_receive_assignment_statement : public Statement
{
 public:
  Tuple_receive_assignment_statement(Expression* val, Expression* closed,
                                     Expression* channel, Location location)
    : Statement(STATEMENT_TUPLE_RECEIVE_ASSIGNMENT, location),
      val_(val), closed_(closed), channel_(channel)
  { }
 
 protected:
  int
  do_traverse(Traverse* traverse);
 
  bool
  do_traverse_assignments(Traverse_assignments*)
  { go_unreachable(); }
 
  Statement*
  do_lower(Gogo*, Named_object*, Block*, Statement_inserter*);
 
  Bstatement*
  do_get_backend(Translate_context*)
  { go_unreachable(); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // Lvalue which receives the value from the channel.
  Expression* val_;
  // Lvalue which receives whether the channel is closed.
  Expression* closed_;
  // The channel on which we receive the value.
  Expression* channel_;
};
 
// Traversal.
 
int
Tuple_receive_assignment_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT
      || this->traverse_expression(traverse, &this->closed_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->traverse_expression(traverse, &this->channel_);
}
 
// Lower to a function call.
 
Statement*
Tuple_receive_assignment_statement::do_lower(Gogo*, Named_object*,
                                             Block* enclosing,
                                             Statement_inserter*)
{
  Location loc = this->location();
 
  Channel_type* channel_type = this->channel_->type()->channel_type();
  if (channel_type == NULL)
    {
      this->report_error(_("expected channel"));
      return Statement::make_error_statement(loc);
    }
  if (!channel_type->may_receive())
    {
      this->report_error(_("invalid receive on send-only channel"));
      return Statement::make_error_statement(loc);
    }
 
  Block* b = new Block(enclosing, loc);
 
  // Make sure that any subexpressions on the left hand side are
  // evaluated in the right order.
  Move_ordered_evals moe(b);
  this->val_->traverse_subexpressions(&moe);
  this->closed_->traverse_subexpressions(&moe);
 
  // var val_temp ELEMENT_TYPE
  Temporary_statement* val_temp =
    Statement::make_temporary(channel_type->element_type(), NULL, loc);
  b->add_statement(val_temp);
 
  // var closed_temp bool
  Temporary_statement* closed_temp =
    Statement::make_temporary(Type::lookup_bool_type(), NULL, loc);
  b->add_statement(closed_temp);
 
  // closed_temp = chanrecv2(type, channel, &val_temp)
  Expression* td = Expression::make_type_descriptor(this->channel_->type(),
                                                    loc);
  Temporary_reference_expression* ref =
    Expression::make_temporary_reference(val_temp, loc);
  Expression* p2 = Expression::make_unary(OPERATOR_AND, ref, loc);
  Expression* call = Runtime::make_call(Runtime::CHANRECV2,
                                        loc, 3, td, this->channel_, p2);
  ref = Expression::make_temporary_reference(closed_temp, loc);
  ref->set_is_lvalue();
  Statement* s = Statement::make_assignment(ref, call, loc);
  b->add_statement(s);
 
  // val = val_temp
  ref = Expression::make_temporary_reference(val_temp, loc);
  s = Statement::make_assignment(this->val_, ref, loc);
  b->add_statement(s);
 
  // closed = closed_temp
  ref = Expression::make_temporary_reference(closed_temp, loc);
  s = Statement::make_assignment(this->closed_, ref, loc);
  b->add_statement(s);
 
  return Statement::make_block_statement(b, loc);
}
 
// Dump the AST representation for a tuple receive statement.
 
void
Tuple_receive_assignment_statement::do_dump_statement(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(this->val_);
  ast_dump_context->ostream() << ", ";
  ast_dump_context->dump_expression(this->closed_);
  ast_dump_context->ostream() << " <- ";
  ast_dump_context->dump_expression(this->channel_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make a nonblocking receive statement.
 
Statement*
Statement::make_tuple_receive_assignment(Expression* val, Expression* closed,
                                         Expression* channel,
                                         Location location)
{
  return new Tuple_receive_assignment_statement(val, closed, channel,
                                                location);
}
 
// An assignment to a pair of values from a type guard.  This is a
// conditional type guard.  v, ok = i.(type).
 
class Tuple_type_guard_assignment_statement : public Statement
{
 public:
  Tuple_type_guard_assignment_statement(Expression* val, Expression* ok,
                                        Expression* expr, Type* type,
                                        Location location)
    : Statement(STATEMENT_TUPLE_TYPE_GUARD_ASSIGNMENT, location),
      val_(val), ok_(ok), expr_(expr), type_(type)
  { }
 
 protected:
  int
  do_traverse(Traverse*);
 
  bool
  do_traverse_assignments(Traverse_assignments*)
  { go_unreachable(); }
 
  Statement*
  do_lower(Gogo*, Named_object*, Block*, Statement_inserter*);
 
  Bstatement*
  do_get_backend(Translate_context*)
  { go_unreachable(); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  Call_expression*
  lower_to_type(Runtime::Function);
 
  void
  lower_to_object_type(Block*, Runtime::Function);
 
  // The variable which recieves the converted value.
  Expression* val_;
  // The variable which receives the indication of success.
  Expression* ok_;
  // The expression being converted.
  Expression* expr_;
  // The type to which the expression is being converted.
  Type* type_;
};
 
// Traverse a type guard tuple assignment.
 
int
Tuple_type_guard_assignment_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT
      || this->traverse_expression(traverse, &this->ok_) == TRAVERSE_EXIT
      || this->traverse_type(traverse, this->type_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->traverse_expression(traverse, &this->expr_);
}
 
// Lower to a function call.
 
Statement*
Tuple_type_guard_assignment_statement::do_lower(Gogo*, Named_object*,
                                                Block* enclosing,
                                                Statement_inserter*)
{
  Location loc = this->location();
 
  Type* expr_type = this->expr_->type();
  if (expr_type->interface_type() == NULL)
    {
      if (!expr_type->is_error() && !this->type_->is_error())
        this->report_error(_("type assertion only valid for interface types"));
      return Statement::make_error_statement(loc);
    }
 
  Block* b = new Block(enclosing, loc);
 
  // Make sure that any subexpressions on the left hand side are
  // evaluated in the right order.
  Move_ordered_evals moe(b);
  this->val_->traverse_subexpressions(&moe);
  this->ok_->traverse_subexpressions(&moe);
 
  bool expr_is_empty = expr_type->interface_type()->is_empty();
  Call_expression* call;
  if (this->type_->interface_type() != NULL)
    {
      if (this->type_->interface_type()->is_empty())
        call = Runtime::make_call((expr_is_empty
                                   ? Runtime::IFACEE2E2
                                   : Runtime::IFACEI2E2),
                                  loc, 1, this->expr_);
      else
        call = this->lower_to_type(expr_is_empty
                                   ? Runtime::IFACEE2I2
                                   : Runtime::IFACEI2I2);
    }
  else if (this->type_->points_to() != NULL)
    call = this->lower_to_type(expr_is_empty
                               ? Runtime::IFACEE2T2P
                               : Runtime::IFACEI2T2P);
  else
    {
      this->lower_to_object_type(b,
                                 (expr_is_empty
                                  ? Runtime::IFACEE2T2
                                  : Runtime::IFACEI2T2));
      call = NULL;
    }
 
  if (call != NULL)
    {
      Expression* res = Expression::make_call_result(call, 0);
      res = Expression::make_unsafe_cast(this->type_, res, loc);
      Statement* s = Statement::make_assignment(this->val_, res, loc);
      b->add_statement(s);
 
      res = Expression::make_call_result(call, 1);
      s = Statement::make_assignment(this->ok_, res, loc);
      b->add_statement(s);
    }
 
  return Statement::make_block_statement(b, loc);
}
 
// Lower a conversion to a non-empty interface type or a pointer type.
 
Call_expression*
Tuple_type_guard_assignment_statement::lower_to_type(Runtime::Function code)
{
  Location loc = this->location();
  return Runtime::make_call(code, loc, 2,
                            Expression::make_type_descriptor(this->type_, loc),
                            this->expr_);
}
 
// Lower a conversion to a non-interface non-pointer type.
 
void
Tuple_type_guard_assignment_statement::lower_to_object_type(
    Block* b,
    Runtime::Function code)
{
  Location loc = this->location();
 
  // var val_temp TYPE
  Temporary_statement* val_temp = Statement::make_temporary(this->type_,
                                                            NULL, loc);
  b->add_statement(val_temp);
 
  // ok = CODE(type_descriptor, expr, &val_temp)
  Expression* p1 = Expression::make_type_descriptor(this->type_, loc);
  Expression* ref = Expression::make_temporary_reference(val_temp, loc);
  Expression* p3 = Expression::make_unary(OPERATOR_AND, ref, loc);
  Expression* call = Runtime::make_call(code, loc, 3, p1, this->expr_, p3);
  Statement* s = Statement::make_assignment(this->ok_, call, loc);
  b->add_statement(s);
 
  // val = val_temp
  ref = Expression::make_temporary_reference(val_temp, loc);
  s = Statement::make_assignment(this->val_, ref, loc);
  b->add_statement(s);
}
 
// Dump the AST representation for a tuple type guard statement.
 
void
Tuple_type_guard_assignment_statement::do_dump_statement(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(this->val_);
  ast_dump_context->ostream() << ", ";
  ast_dump_context->dump_expression(this->ok_);
  ast_dump_context->ostream() << " = ";
  ast_dump_context->dump_expression(this->expr_);
  ast_dump_context->ostream() << " . ";
  ast_dump_context->dump_type(this->type_);
  ast_dump_context->ostream()  << std::endl;
}
 
// Make an assignment from a type guard to a pair of variables.
 
Statement*
Statement::make_tuple_type_guard_assignment(Expression* val, Expression* ok,
                                            Expression* expr, Type* type,
                                            Location location)
{
  return new Tuple_type_guard_assignment_statement(val, ok, expr, type,
                                                   location);
}
 
// An expression statement.
 
class Expression_statement : public Statement
{
 public:
  Expression_statement(Expression* expr, bool is_ignored)
    : Statement(STATEMENT_EXPRESSION, expr->location()),
      expr_(expr), is_ignored_(is_ignored)
  { }
 
  Expression*
  expr()
  { return this->expr_; }
 
 protected:
  int
  do_traverse(Traverse* traverse)
  { return this->traverse_expression(traverse, &this->expr_); }
 
  void
  do_determine_types()
  { this->expr_->determine_type_no_context(); }
 
  void
  do_check_types(Gogo*);
 
  bool
  do_may_fall_through() const;
 
  Bstatement*
  do_get_backend(Translate_context* context);
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  Expression* expr_;
  // Whether the value of this expression is being explicitly ignored.
  bool is_ignored_;
};
 
// Check the types of an expression statement.  The only check we do
// is to possibly give an error about discarding the value of the
// expression.
 
void
Expression_statement::do_check_types(Gogo*)
{
  if (!this->is_ignored_)
    this->expr_->discarding_value();
}
 
// An expression statement may fall through unless it is a call to a
// function which does not return.
 
bool
Expression_statement::do_may_fall_through() const
{
  const Call_expression* call = this->expr_->call_expression();
  if (call != NULL)
    {
      const Expression* fn = call->fn();
      const Func_expression* fe = fn->func_expression();
      if (fe != NULL)
        {
          const Named_object* no = fe->named_object();
 
          Function_type* fntype;
          if (no->is_function())
            fntype = no->func_value()->type();
          else if (no->is_function_declaration())
            fntype = no->func_declaration_value()->type();
          else
            fntype = NULL;
 
          // The builtin function panic does not return.
          if (fntype != NULL && fntype->is_builtin() && no->name() == "panic")
            return false;
        }
    }
  return true;
}
 
// Convert to backend representation.
 
Bstatement*
Expression_statement::do_get_backend(Translate_context* context)
{
  tree expr_tree = this->expr_->get_tree(context);
  return context->backend()->expression_statement(tree_to_expr(expr_tree));
}
 
// Dump the AST representation for an expression statement
 
void
Expression_statement::do_dump_statement(Ast_dump_context* ast_dump_context)
    const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(expr_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make an expression statement from an Expression.
 
Statement*
Statement::make_statement(Expression* expr, bool is_ignored)
{
  return new Expression_statement(expr, is_ignored);
}
 
// A block statement--a list of statements which may include variable
// definitions.
 
class Block_statement : public Statement
{
 public:
  Block_statement(Block* block, Location location)
    : Statement(STATEMENT_BLOCK, location),
      block_(block)
  { }
 
 protected:
  int
  do_traverse(Traverse* traverse)
  { return this->block_->traverse(traverse); }
 
  void
  do_determine_types()
  { this->block_->determine_types(); }
 
  bool
  do_may_fall_through() const
  { return this->block_->may_fall_through(); }
 
  Bstatement*
  do_get_backend(Translate_context* context);
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  Block* block_;
};
 
// Convert a block to the backend representation of a statement.
 
Bstatement*
Block_statement::do_get_backend(Translate_context* context)
{
  Bblock* bblock = this->block_->get_backend(context);
  return context->backend()->block_statement(bblock);
}
 
// Dump the AST for a block statement
 
void
Block_statement::do_dump_statement(Ast_dump_context*) const
{
  // block statement braces are dumped when traversing.
}
 
// Make a block statement.
 
Statement*
Statement::make_block_statement(Block* block, Location location)
{
  return new Block_statement(block, location);
}
 
// An increment or decrement statement.
 
class Inc_dec_statement : public Statement
{
 public:
  Inc_dec_statement(bool is_inc, Expression* expr)
    : Statement(STATEMENT_INCDEC, expr->location()),
      expr_(expr), is_inc_(is_inc)
  { }
 
 protected:
  int
  do_traverse(Traverse* traverse)
  { return this->traverse_expression(traverse, &this->expr_); }
 
  bool
  do_traverse_assignments(Traverse_assignments*)
  { go_unreachable(); }
 
  Statement*
  do_lower(Gogo*, Named_object*, Block*, Statement_inserter*);
 
  Bstatement*
  do_get_backend(Translate_context*)
  { go_unreachable(); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // The l-value to increment or decrement.
  Expression* expr_;
  // Whether to increment or decrement.
  bool is_inc_;
};
 
// Lower to += or -=.
 
Statement*
Inc_dec_statement::do_lower(Gogo*, Named_object*, Block*, Statement_inserter*)
{
  Location loc = this->location();
 
  mpz_t oval;
  mpz_init_set_ui(oval, 1UL);
  Expression* oexpr = Expression::make_integer(&oval, NULL, loc);
  mpz_clear(oval);
 
  Operator op = this->is_inc_ ? OPERATOR_PLUSEQ : OPERATOR_MINUSEQ;
  return Statement::make_assignment_operation(op, this->expr_, oexpr, loc);
}
 
// Dump the AST representation for a inc/dec statement.
 
void
Inc_dec_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(expr_);
  ast_dump_context->ostream() << (is_inc_? "++": "--") << std::endl;
}
 
// Make an increment statement.
 
Statement*
Statement::make_inc_statement(Expression* expr)
{
  return new Inc_dec_statement(true, expr);
}
 
// Make a decrement statement.
 
Statement*
Statement::make_dec_statement(Expression* expr)
{
  return new Inc_dec_statement(false, expr);
}
 
// Class Thunk_statement.  This is the base class for go and defer
// statements.
 
// Constructor.
 
Thunk_statement::Thunk_statement(Statement_classification classification,
                                 Call_expression* call,
                                 Location location)
    : Statement(classification, location),
      call_(call), struct_type_(NULL)
{
}
 
// Return whether this is a simple statement which does not require a
// thunk.
 
bool
Thunk_statement::is_simple(Function_type* fntype) const
{
  // We need a thunk to call a method, or to pass a variable number of
  // arguments.
  if (fntype->is_method() || fntype->is_varargs())
    return false;
 
  // A defer statement requires a thunk to set up for whether the
  // function can call recover.
  if (this->classification() == STATEMENT_DEFER)
    return false;
 
  // We can only permit a single parameter of pointer type.
  const Typed_identifier_list* parameters = fntype->parameters();
  if (parameters != NULL
      && (parameters->size() > 1
          || (parameters->size() == 1
              && parameters->begin()->type()->points_to() == NULL)))
    return false;
 
  // If the function returns multiple values, or returns a type other
  // than integer, floating point, or pointer, then it may get a
  // hidden first parameter, in which case we need the more
  // complicated approach.  This is true even though we are going to
  // ignore the return value.
  const Typed_identifier_list* results = fntype->results();
  if (results != NULL
      && (results->size() > 1
          || (results->size() == 1
              && !results->begin()->type()->is_basic_type()
              && results->begin()->type()->points_to() == NULL)))
    return false;
 
  // If this calls something which is not a simple function, then we
  // need a thunk.
  Expression* fn = this->call_->call_expression()->fn();
  if (fn->interface_field_reference_expression() != NULL)
    return false;
 
  return true;
}
 
// Traverse a thunk statement.
 
int
Thunk_statement::do_traverse(Traverse* traverse)
{
  return this->traverse_expression(traverse, &this->call_);
}
 
// We implement traverse_assignment for a thunk statement because it
// effectively copies the function call.
 
bool
Thunk_statement::do_traverse_assignments(Traverse_assignments* tassign)
{
  Expression* fn = this->call_->call_expression()->fn();
  Expression* fn2 = fn;
  tassign->value(&fn2, true, false);
  return true;
}
 
// Determine types in a thunk statement.
 
void
Thunk_statement::do_determine_types()
{
  this->call_->determine_type_no_context();
 
  // Now that we know the types of the call, build the struct used to
  // pass parameters.
  Call_expression* ce = this->call_->call_expression();
  if (ce == NULL)
    return;
  Function_type* fntype = ce->get_function_type();
  if (fntype != NULL && !this->is_simple(fntype))
    this->struct_type_ = this->build_struct(fntype);
}
 
// Check types in a thunk statement.
 
void
Thunk_statement::do_check_types(Gogo*)
{
  Call_expression* ce = this->call_->call_expression();
  if (ce == NULL)
    {
      if (!this->call_->is_error_expression())
        this->report_error("expected call expression");
      return;
    }
}
 
// The Traverse class used to find and simplify thunk statements.
 
class Simplify_thunk_traverse : public Traverse
{
 public:
  Simplify_thunk_traverse(Gogo* gogo)
    : Traverse(traverse_functions | traverse_blocks),
      gogo_(gogo), function_(NULL)
  { }
 
  int
  function(Named_object*);
 
  int
  block(Block*);
 
 private:
  // General IR.
  Gogo* gogo_;
  // The function we are traversing.
  Named_object* function_;
};
 
// Keep track of the current function while looking for thunks.
 
int
Simplify_thunk_traverse::function(Named_object* no)
{
  go_assert(this->function_ == NULL);
  this->function_ = no;
  int t = no->func_value()->traverse(this);
  this->function_ = NULL;
  if (t == TRAVERSE_EXIT)
    return t;
  return TRAVERSE_SKIP_COMPONENTS;
}
 
// Look for thunks in a block.
 
int
Simplify_thunk_traverse::block(Block* b)
{
  // The parser ensures that thunk statements always appear at the end
  // of a block.
  if (b->statements()->size() < 1)
    return TRAVERSE_CONTINUE;
  Thunk_statement* stat = b->statements()->back()->thunk_statement();
  if (stat == NULL)
    return TRAVERSE_CONTINUE;
  if (stat->simplify_statement(this->gogo_, this->function_, b))
    return TRAVERSE_SKIP_COMPONENTS;
  return TRAVERSE_CONTINUE;
}
 
// Simplify all thunk statements.
 
void
Gogo::simplify_thunk_statements()
{
  Simplify_thunk_traverse thunk_traverse(this);
  this->traverse(&thunk_traverse);
}
 
// Return true if the thunk function is a constant, which means that
// it does not need to be passed to the thunk routine.
 
bool
Thunk_statement::is_constant_function() const
{
  Call_expression* ce = this->call_->call_expression();
  Function_type* fntype = ce->get_function_type();
  if (fntype == NULL)
    {
      go_assert(saw_errors());
      return false;
    }
  if (fntype->is_builtin())
    return true;
  Expression* fn = ce->fn();
  if (fn->func_expression() != NULL)
    return fn->func_expression()->closure() == NULL;
  if (fn->interface_field_reference_expression() != NULL)
    return true;
  return false;
}
 
// Simplify complex thunk statements into simple ones.  A complicated
// thunk statement is one which takes anything other than zero
// parameters or a single pointer parameter.  We rewrite it into code
// which allocates a struct, stores the parameter values into the
// struct, and does a simple go or defer statement which passes the
// struct to a thunk.  The thunk does the real call.
 
bool
Thunk_statement::simplify_statement(Gogo* gogo, Named_object* function,
                                    Block* block)
{
  if (this->classification() == STATEMENT_ERROR)
    return false;
  if (this->call_->is_error_expression())
    return false;
 
  if (this->classification() == STATEMENT_DEFER)
    {
      // Make sure that the defer stack exists for the function.  We
      // will use when converting this statement to the backend
      // representation, but we want it to exist when we start
      // converting the function.
      function->func_value()->defer_stack(this->location());
    }
 
  Call_expression* ce = this->call_->call_expression();
  Function_type* fntype = ce->get_function_type();
  if (fntype == NULL)
    {
      go_assert(saw_errors());
      this->set_is_error();
      return false;
    }
  if (this->is_simple(fntype))
    return false;
 
  Expression* fn = ce->fn();
  Interface_field_reference_expression* interface_method =
    fn->interface_field_reference_expression();
 
  Location location = this->location();
 
  std::string thunk_name = Gogo::thunk_name();
 
  // Build the thunk.
  this->build_thunk(gogo, thunk_name);
 
  // Generate code to call the thunk.
 
  // Get the values to store into the struct which is the single
  // argument to the thunk.
 
  Expression_list* vals = new Expression_list();
  if (!this->is_constant_function())
    vals->push_back(fn);
 
  if (interface_method != NULL)
    vals->push_back(interface_method->expr());
 
  if (ce->args() != NULL)
    {
      for (Expression_list::const_iterator p = ce->args()->begin();
           p != ce->args()->end();
           ++p)
        vals->push_back(*p);
    }
 
  // Build the struct.
  Expression* constructor =
    Expression::make_struct_composite_literal(this->struct_type_, vals,
                                              location);
 
  // Allocate the initialized struct on the heap.
  constructor = Expression::make_heap_composite(constructor, location);
 
  // Look up the thunk.
  Named_object* named_thunk = gogo->lookup(thunk_name, NULL);
  go_assert(named_thunk != NULL && named_thunk->is_function());
 
  // Build the call.
  Expression* func = Expression::make_func_reference(named_thunk, NULL,
                                                     location);
  Expression_list* params = new Expression_list();
  params->push_back(constructor);
  Call_expression* call = Expression::make_call(func, params, false, location);
 
  // Build the simple go or defer statement.
  Statement* s;
  if (this->classification() == STATEMENT_GO)
    s = Statement::make_go_statement(call, location);
  else if (this->classification() == STATEMENT_DEFER)
    s = Statement::make_defer_statement(call, location);
  else
    go_unreachable();
 
  // The current block should end with the go statement.
  go_assert(block->statements()->size() >= 1);
  go_assert(block->statements()->back() == this);
  block->replace_statement(block->statements()->size() - 1, s);
 
  // We already ran the determine_types pass, so we need to run it now
  // for the new statement.
  s->determine_types();
 
  // Sanity check.
  gogo->check_types_in_block(block);
 
  // Return true to tell the block not to keep looking at statements.
  return true;
}
 
// Set the name to use for thunk parameter N.
 
void
Thunk_statement::thunk_field_param(int n, char* buf, size_t buflen)
{
  snprintf(buf, buflen, "a%d", n);
}
 
// Build a new struct type to hold the parameters for a complicated
// thunk statement.  FNTYPE is the type of the function call.
 
Struct_type*
Thunk_statement::build_struct(Function_type* fntype)
{
  Location location = this->location();
 
  Struct_field_list* fields = new Struct_field_list();
 
  Call_expression* ce = this->call_->call_expression();
  Expression* fn = ce->fn();
 
  if (!this->is_constant_function())
    {
      // The function to call.
      fields->push_back(Struct_field(Typed_identifier("fn", fntype,
                                                      location)));
    }
 
  // If this thunk statement calls a method on an interface, we pass
  // the interface object to the thunk.
  Interface_field_reference_expression* interface_method =
    fn->interface_field_reference_expression();
  if (interface_method != NULL)
    {
      Typed_identifier tid("object", interface_method->expr()->type(),
                           location);
      fields->push_back(Struct_field(tid));
    }
 
  // The predeclared recover function has no argument.  However, we
  // add an argument when building recover thunks.  Handle that here.
  if (ce->is_recover_call())
    {
      fields->push_back(Struct_field(Typed_identifier("can_recover",
                                                      Type::lookup_bool_type(),
                                                      location)));
    }
 
  const Expression_list* args = ce->args();
  if (args != NULL)
    {
      int i = 0;
      for (Expression_list::const_iterator p = args->begin();
           p != args->end();
           ++p, ++i)
        {
          char buf[50];
          this->thunk_field_param(i, buf, sizeof buf);
          fields->push_back(Struct_field(Typed_identifier(buf, (*p)->type(),
                                                          location)));
        }
    }
 
  return Type::make_struct_type(fields, location);
}
 
// Build the thunk we are going to call.  This is a brand new, albeit
// artificial, function.
 
void
Thunk_statement::build_thunk(Gogo* gogo, const std::string& thunk_name)
{
  Location location = this->location();
 
  Call_expression* ce = this->call_->call_expression();
 
  bool may_call_recover = false;
  if (this->classification() == STATEMENT_DEFER)
    {
      Func_expression* fn = ce->fn()->func_expression();
      if (fn == NULL)
        may_call_recover = true;
      else
        {
          const Named_object* no = fn->named_object();
          if (!no->is_function())
            may_call_recover = true;
          else
            may_call_recover = no->func_value()->calls_recover();
        }
    }
 
  // Build the type of the thunk.  The thunk takes a single parameter,
  // which is a pointer to the special structure we build.
  const char* const parameter_name = "__go_thunk_parameter";
  Typed_identifier_list* thunk_parameters = new Typed_identifier_list();
  Type* pointer_to_struct_type = Type::make_pointer_type(this->struct_type_);
  thunk_parameters->push_back(Typed_identifier(parameter_name,
                                               pointer_to_struct_type,
                                               location));
 
  Typed_identifier_list* thunk_results = NULL;
  if (may_call_recover)
    {
      // When deferring a function which may call recover, add a
      // return value, to disable tail call optimizations which will
      // break the way we check whether recover is permitted.
      thunk_results = new Typed_identifier_list();
      thunk_results->push_back(Typed_identifier("", Type::lookup_bool_type(),
                                                location));
    }
 
  Function_type* thunk_type = Type::make_function_type(NULL, thunk_parameters,
                                                       thunk_results,
                                                       location);
 
  // Start building the thunk.
  Named_object* function = gogo->start_function(thunk_name, thunk_type, true,
                                                location);
 
  gogo->start_block(location);
 
  // For a defer statement, start with a call to
  // __go_set_defer_retaddr.  */
  Label* retaddr_label = NULL;
  if (may_call_recover)
    {
      retaddr_label = gogo->add_label_reference("retaddr", location, false);
      Expression* arg = Expression::make_label_addr(retaddr_label, location);
      Expression* call = Runtime::make_call(Runtime::SET_DEFER_RETADDR,
                                            location, 1, arg);
 
      // This is a hack to prevent the middle-end from deleting the
      // label.
      gogo->start_block(location);
      gogo->add_statement(Statement::make_goto_statement(retaddr_label,
                                                         location));
      Block* then_block = gogo->finish_block(location);
      then_block->determine_types();
 
      Statement* s = Statement::make_if_statement(call, then_block, NULL,
                                                  location);
      s->determine_types();
      gogo->add_statement(s);
    }
 
  // Get a reference to the parameter.
  Named_object* named_parameter = gogo->lookup(parameter_name, NULL);
  go_assert(named_parameter != NULL && named_parameter->is_variable());
 
  // Build the call.  Note that the field names are the same as the
  // ones used in build_struct.
  Expression* thunk_parameter = Expression::make_var_reference(named_parameter,
                                                               location);
  thunk_parameter = Expression::make_unary(OPERATOR_MULT, thunk_parameter,
                                           location);
 
  Interface_field_reference_expression* interface_method =
    ce->fn()->interface_field_reference_expression();
 
  Expression* func_to_call;
  unsigned int next_index;
  if (this->is_constant_function())
    {
      func_to_call = ce->fn();
      next_index = 0;
    }
  else
    {
      func_to_call = Expression::make_field_reference(thunk_parameter,
                                                      0, location);
      next_index = 1;
    }
 
  if (interface_method != NULL)
    {
      // The main program passes the interface object.
      go_assert(next_index == 0);
      Expression* r = Expression::make_field_reference(thunk_parameter, 0,
                                                       location);
      const std::string& name(interface_method->name());
      func_to_call = Expression::make_interface_field_reference(r, name,
                                                                location);
      next_index = 1;
    }
 
  Expression_list* call_params = new Expression_list();
  const Struct_field_list* fields = this->struct_type_->fields();
  Struct_field_list::const_iterator p = fields->begin();
  for (unsigned int i = 0; i < next_index; ++i)
    ++p;
  bool is_recover_call = ce->is_recover_call();
  Expression* recover_arg = NULL;
  for (; p != fields->end(); ++p, ++next_index)
    {
      Expression* thunk_param = Expression::make_var_reference(named_parameter,
                                                               location);
      thunk_param = Expression::make_unary(OPERATOR_MULT, thunk_param,
                                           location);
      Expression* param = Expression::make_field_reference(thunk_param,
                                                           next_index,
                                                           location);
      if (!is_recover_call)
        call_params->push_back(param);
      else
        {
          go_assert(call_params->empty());
          recover_arg = param;
        }
    }
 
  if (call_params->empty())
    {
      delete call_params;
      call_params = NULL;
    }
 
  Call_expression* call = Expression::make_call(func_to_call, call_params,
                                                false, location);
 
  // This call expression was already lowered before entering the
  // thunk statement.  Don't try to lower varargs again, as that will
  // cause confusion for, e.g., method calls which already have a
  // receiver parameter.
  call->set_varargs_are_lowered();
 
  Statement* call_statement = Statement::make_statement(call, true);
 
  gogo->add_statement(call_statement);
 
  // If this is a defer statement, the label comes immediately after
  // the call.
  if (may_call_recover)
    {
      gogo->add_label_definition("retaddr", location);
 
      Expression_list* vals = new Expression_list();
      vals->push_back(Expression::make_boolean(false, location));
      gogo->add_statement(Statement::make_return_statement(vals, location));
    }
 
  Block* b = gogo->finish_block(location);
 
  gogo->add_block(b, location);
 
  gogo->lower_block(function, b);
 
  // We already ran the determine_types pass, so we need to run it
  // just for the call statement now.  The other types are known.
  call_statement->determine_types();
 
  if (may_call_recover || recover_arg != NULL)
    {
      // Dig up the call expression, which may have been changed
      // during lowering.
      go_assert(call_statement->classification() == STATEMENT_EXPRESSION);
      Expression_statement* es =
        static_cast<Expression_statement*>(call_statement);
      Call_expression* ce = es->expr()->call_expression();
      go_assert(ce != NULL);
      if (may_call_recover)
        ce->set_is_deferred();
      if (recover_arg != NULL)
        ce->set_recover_arg(recover_arg);
    }
 
  // That is all the thunk has to do.
  gogo->finish_function(location);
}
 
// Get the function and argument expressions.
 
bool
Thunk_statement::get_fn_and_arg(Expression** pfn, Expression** parg)
{
  if (this->call_->is_error_expression())
    return false;
 
  Call_expression* ce = this->call_->call_expression();
 
  *pfn = ce->fn();
 
  const Expression_list* args = ce->args();
  if (args == NULL || args->empty())
    *parg = Expression::make_nil(this->location());
  else
    {
      go_assert(args->size() == 1);
      *parg = args->front();
    }
 
  return true;
}
 
// Class Go_statement.
 
Bstatement*
Go_statement::do_get_backend(Translate_context* context)
{
  Expression* fn;
  Expression* arg;
  if (!this->get_fn_and_arg(&fn, &arg))
    return context->backend()->error_statement();
 
  Expression* call = Runtime::make_call(Runtime::GO, this->location(), 2,
                                        fn, arg);
  tree call_tree = call->get_tree(context);
  Bexpression* call_bexpr = tree_to_expr(call_tree);
  return context->backend()->expression_statement(call_bexpr);
}
 
// Dump the AST representation for go statement.
 
void
Go_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "go ";
  ast_dump_context->dump_expression(this->call());
  ast_dump_context->ostream() << std::endl;
}
 
// Make a go statement.
 
Statement*
Statement::make_go_statement(Call_expression* call, Location location)
{
  return new Go_statement(call, location);
}
 
// Class Defer_statement.
 
Bstatement*
Defer_statement::do_get_backend(Translate_context* context)
{
  Expression* fn;
  Expression* arg;
  if (!this->get_fn_and_arg(&fn, &arg))
    return context->backend()->error_statement();
 
  Location loc = this->location();
  Expression* ds = context->function()->func_value()->defer_stack(loc);
 
  Expression* call = Runtime::make_call(Runtime::DEFER, loc, 3,
                                        ds, fn, arg);
  tree call_tree = call->get_tree(context);
  Bexpression* call_bexpr = tree_to_expr(call_tree);
  return context->backend()->expression_statement(call_bexpr);
}
 
// Dump the AST representation for defer statement.
 
void
Defer_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "defer ";
  ast_dump_context->dump_expression(this->call());
  ast_dump_context->ostream() << std::endl;
}
 
// Make a defer statement.
 
Statement*
Statement::make_defer_statement(Call_expression* call,
                                Location location)
{
  return new Defer_statement(call, location);
}
 
// Class Return_statement.
 
// Traverse assignments.  We treat each return value as a top level
// RHS in an expression.
 
bool
Return_statement::do_traverse_assignments(Traverse_assignments* tassign)
{
  Expression_list* vals = this->vals_;
  if (vals != NULL)
    {
      for (Expression_list::iterator p = vals->begin();
           p != vals->end();
           ++p)
        tassign->value(&*p, true, true);
    }
  return true;
}
 
// Lower a return statement.  If we are returning a function call
// which returns multiple values which match the current function,
// split up the call's results.  If the return statement lists
// explicit values, implement this statement by assigning the values
// to the result variables and change this statement to a naked
// return.  This lets panic/recover work correctly.
 
Statement*
Return_statement::do_lower(Gogo*, Named_object* function, Block* enclosing,
                           Statement_inserter*)
{
  if (this->is_lowered_)
    return this;
 
  Expression_list* vals = this->vals_;
  this->vals_ = NULL;
  this->is_lowered_ = true;
 
  Location loc = this->location();
 
  size_t vals_count = vals == NULL ? 0 : vals->size();
  Function::Results* results = function->func_value()->result_variables();
  size_t results_count = results == NULL ? 0 : results->size();
 
  if (vals_count == 0)
    {
      if (results_count > 0 && !function->func_value()->results_are_named())
        {
          this->report_error(_("not enough arguments to return"));
          return this;
        }
      return this;
    }
 
  if (results_count == 0)
    {
      this->report_error(_("return with value in function "
                           "with no return type"));
      return this;
    }
 
  // If the current function has multiple return values, and we are
  // returning a single call expression, split up the call expression.
  if (results_count > 1
      && vals->size() == 1
      && vals->front()->call_expression() != NULL)
    {
      Call_expression* call = vals->front()->call_expression();
      delete vals;
      vals = new Expression_list;
      for (size_t i = 0; i < results_count; ++i)
        vals->push_back(Expression::make_call_result(call, i));
      vals_count = results_count;
    }
 
  if (vals_count < results_count)
    {
      this->report_error(_("not enough arguments to return"));
      return this;
    }
 
  if (vals_count > results_count)
    {
      this->report_error(_("too many values in return statement"));
      return this;
    }
 
  Block* b = new Block(enclosing, loc);
 
  Expression_list* lhs = new Expression_list();
  Expression_list* rhs = new Expression_list();
 
  Expression_list::const_iterator pe = vals->begin();
  int i = 1;
  for (Function::Results::const_iterator pr = results->begin();
       pr != results->end();
       ++pr, ++pe, ++i)
    {
      Named_object* rv = *pr;
      Expression* e = *pe;
 
      // Check types now so that we give a good error message.  The
      // result type is known.  We determine the expression type
      // early.
 
      Type *rvtype = rv->result_var_value()->type();
      Type_context type_context(rvtype, false);
      e->determine_type(&type_context);
 
      std::string reason;
      bool ok;
      if (this->are_hidden_fields_ok_)
        ok = Type::are_assignable_hidden_ok(rvtype, e->type(), &reason);
      else
        ok = Type::are_assignable(rvtype, e->type(), &reason);
      if (ok)
        {
          Expression* ve = Expression::make_var_reference(rv, e->location());
          lhs->push_back(ve);
          rhs->push_back(e);
        }
      else
        {
          if (reason.empty())
            error_at(e->location(), "incompatible type for return value %d", i);
          else
            error_at(e->location(),
                     "incompatible type for return value %d (%s)",
                     i, reason.c_str());
        }
    }
  go_assert(lhs->size() == rhs->size());
 
  if (lhs->empty())
    ;
  else if (lhs->size() == 1)
    {
      Statement* s = Statement::make_assignment(lhs->front(), rhs->front(),
                                                loc);
      if (this->are_hidden_fields_ok_)
        {
          Assignment_statement* as = static_cast<Assignment_statement*>(s);
          as->set_hidden_fields_are_ok();
        }
      b->add_statement(s);
      delete lhs;
      delete rhs;
    }
  else
    {
      Statement* s = Statement::make_tuple_assignment(lhs, rhs, loc);
      if (this->are_hidden_fields_ok_)
        {
          Tuple_assignment_statement* tas =
            static_cast<Tuple_assignment_statement*>(s);
          tas->set_hidden_fields_are_ok();
        }
      b->add_statement(s);
    }
 
  b->add_statement(this);
 
  delete vals;
 
  return Statement::make_block_statement(b, loc);
}
 
// Convert a return statement to the backend representation.
 
Bstatement*
Return_statement::do_get_backend(Translate_context* context)
{
  Location loc = this->location();
 
  Function* function = context->function()->func_value();
  tree fndecl = function->get_decl();
 
  Function::Results* results = function->result_variables();
  std::vector<Bexpression*> retvals;
  if (results != NULL && !results->empty())
    {
      retvals.reserve(results->size());
      for (Function::Results::const_iterator p = results->begin();
           p != results->end();
           p++)
        {
          Expression* vr = Expression::make_var_reference(*p, loc);
          retvals.push_back(tree_to_expr(vr->get_tree(context)));
        }
    }
 
  return context->backend()->return_statement(tree_to_function(fndecl),
                                              retvals, loc);
}
 
// Dump the AST representation for a return statement.
 
void
Return_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "return " ;
  ast_dump_context->dump_expression_list(this->vals_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make a return statement.
 
Return_statement*
Statement::make_return_statement(Expression_list* vals,
                                 Location location)
{
  return new Return_statement(vals, location);
}
 
// A break or continue statement.
 
class Bc_statement : public Statement
{
 public:
  Bc_statement(bool is_break, Unnamed_label* label, Location location)
    : Statement(STATEMENT_BREAK_OR_CONTINUE, location),
      label_(label), is_break_(is_break)
  { }
 
  bool
  is_break() const
  { return this->is_break_; }
 
 protected:
  int
  do_traverse(Traverse*)
  { return TRAVERSE_CONTINUE; }
 
  bool
  do_may_fall_through() const
  { return false; }
 
  Bstatement*
  do_get_backend(Translate_context* context)
  { return this->label_->get_goto(context, this->location()); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // The label that this branches to.
  Unnamed_label* label_;
  // True if this is "break", false if it is "continue".
  bool is_break_;
};
 
// Dump the AST representation for a break/continue statement
 
void
Bc_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << (this->is_break_ ? "break" : "continue");
  if (this->label_ != NULL)
    {
      ast_dump_context->ostream() << " ";
      ast_dump_context->dump_label_name(this->label_);
    }
  ast_dump_context->ostream() << std::endl;
}
 
// Make a break statement.
 
Statement*
Statement::make_break_statement(Unnamed_label* label, Location location)
{
  return new Bc_statement(true, label, location);
}
 
// Make a continue statement.
 
Statement*
Statement::make_continue_statement(Unnamed_label* label,
                                   Location location)
{
  return new Bc_statement(false, label, location);
}
 
// A goto statement.
 
class Goto_statement : public Statement
{
 public:
  Goto_statement(Label* label, Location location)
    : Statement(STATEMENT_GOTO, location),
      label_(label)
  { }
 
 protected:
  int
  do_traverse(Traverse*)
  { return TRAVERSE_CONTINUE; }
 
  void
  do_check_types(Gogo*);
 
  bool
  do_may_fall_through() const
  { return false; }
 
  Bstatement*
  do_get_backend(Translate_context*);
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  Label* label_;
};
 
// Check types for a label.  There aren't any types per se, but we use
// this to give an error if the label was never defined.
 
void
Goto_statement::do_check_types(Gogo*)
{
  if (!this->label_->is_defined())
    {
      error_at(this->location(), "reference to undefined label %qs",
               Gogo::message_name(this->label_->name()).c_str());
      this->set_is_error();
    }
}
 
// Convert the goto statement to the backend representation.
 
Bstatement*
Goto_statement::do_get_backend(Translate_context* context)
{
  Blabel* blabel = this->label_->get_backend_label(context);
  return context->backend()->goto_statement(blabel, this->location());
}
 
// Dump the AST representation for a goto statement.
 
void
Goto_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "goto " << this->label_->name() << std::endl;
}
 
// Make a goto statement.
 
Statement*
Statement::make_goto_statement(Label* label, Location location)
{
  return new Goto_statement(label, location);
}
 
// A goto statement to an unnamed label.
 
class Goto_unnamed_statement : public Statement
{
 public:
  Goto_unnamed_statement(Unnamed_label* label, Location location)
    : Statement(STATEMENT_GOTO_UNNAMED, location),
      label_(label)
  { }
 
 protected:
  int
  do_traverse(Traverse*)
  { return TRAVERSE_CONTINUE; }
 
  bool
  do_may_fall_through() const
  { return false; }
 
  Bstatement*
  do_get_backend(Translate_context* context)
  { return this->label_->get_goto(context, this->location()); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  Unnamed_label* label_;
};
 
// Dump the AST representation for an unnamed goto statement
 
void
Goto_unnamed_statement::do_dump_statement(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "goto ";
  ast_dump_context->dump_label_name(this->label_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make a goto statement to an unnamed label.
 
Statement*
Statement::make_goto_unnamed_statement(Unnamed_label* label,
                                       Location location)
{
  return new Goto_unnamed_statement(label, location);
}
 
// Class Label_statement.
 
// Traversal.
 
int
Label_statement::do_traverse(Traverse*)
{
  return TRAVERSE_CONTINUE;
}
 
// Return the backend representation of the statement defining this
// label.
 
Bstatement*
Label_statement::do_get_backend(Translate_context* context)
{
  Blabel* blabel = this->label_->get_backend_label(context);
  return context->backend()->label_definition_statement(blabel);
}
 
// Dump the AST for a label definition statement.
 
void
Label_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << this->label_->name() << ":" << std::endl;
}
 
// Make a label statement.
 
Statement*
Statement::make_label_statement(Label* label, Location location)
{
  return new Label_statement(label, location);
}
 
// An unnamed label statement.
 
class Unnamed_label_statement : public Statement
{
 public:
  Unnamed_label_statement(Unnamed_label* label)
    : Statement(STATEMENT_UNNAMED_LABEL, label->location()),
      label_(label)
  { }
 
 protected:
  int
  do_traverse(Traverse*)
  { return TRAVERSE_CONTINUE; }
 
  Bstatement*
  do_get_backend(Translate_context* context)
  { return this->label_->get_definition(context); }
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // The label.
  Unnamed_label* label_;
};
 
// Dump the AST representation for an unnamed label definition statement.
 
void
Unnamed_label_statement::do_dump_statement(Ast_dump_context* ast_dump_context)
    const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_label_name(this->label_);
  ast_dump_context->ostream() << ":" << std::endl;
}
 
// Make an unnamed label statement.
 
Statement*
Statement::make_unnamed_label_statement(Unnamed_label* label)
{
  return new Unnamed_label_statement(label);
}
 
// An if statement.
 
class If_statement : public Statement
{
 public:
  If_statement(Expression* cond, Block* then_block, Block* else_block,
               Location location)
    : Statement(STATEMENT_IF, location),
      cond_(cond), then_block_(then_block), else_block_(else_block)
  { }
 
 protected:
  int
  do_traverse(Traverse*);
 
  void
  do_determine_types();
 
  void
  do_check_types(Gogo*);
 
  bool
  do_may_fall_through() const;
 
  Bstatement*
  do_get_backend(Translate_context*);
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  Expression* cond_;
  Block* then_block_;
  Block* else_block_;
};
 
// Traversal.
 
int
If_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->cond_) == TRAVERSE_EXIT
      || this->then_block_->traverse(traverse) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  if (this->else_block_ != NULL)
    {
      if (this->else_block_->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  return TRAVERSE_CONTINUE;
}
 
void
If_statement::do_determine_types()
{
  Type_context context(Type::lookup_bool_type(), false);
  this->cond_->determine_type(&context);
  this->then_block_->determine_types();
  if (this->else_block_ != NULL)
    this->else_block_->determine_types();
}
 
// Check types.
 
void
If_statement::do_check_types(Gogo*)
{
  Type* type = this->cond_->type();
  if (type->is_error())
    this->set_is_error();
  else if (!type->is_boolean_type())
    this->report_error(_("expected boolean expression"));
}
 
// Whether the overall statement may fall through.
 
bool
If_statement::do_may_fall_through() const
{
  return (this->else_block_ == NULL
          || this->then_block_->may_fall_through()
          || this->else_block_->may_fall_through());
}
 
// Get the backend representation.
 
Bstatement*
If_statement::do_get_backend(Translate_context* context)
{
  go_assert(this->cond_->type()->is_boolean_type()
             || this->cond_->type()->is_error());
  tree cond_tree = this->cond_->get_tree(context);
  Bexpression* cond_expr = tree_to_expr(cond_tree);
  Bblock* then_block = this->then_block_->get_backend(context);
  Bblock* else_block = (this->else_block_ == NULL
                        ? NULL
                        : this->else_block_->get_backend(context));
  return context->backend()->if_statement(cond_expr, then_block,
                                          else_block, this->location());
}
 
// Dump the AST representation for an if statement
 
void
If_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "if ";
  ast_dump_context->dump_expression(this->cond_);
  ast_dump_context->ostream() << std::endl;
  if (ast_dump_context->dump_subblocks())
    {
      ast_dump_context->dump_block(this->then_block_);
      if (this->else_block_ != NULL)
        {
          ast_dump_context->print_indent();
          ast_dump_context->ostream() << "else" << std::endl;
          ast_dump_context->dump_block(this->else_block_);
        }
    }
}
 
// Make an if statement.
 
Statement*
Statement::make_if_statement(Expression* cond, Block* then_block,
                             Block* else_block, Location location)
{
  return new If_statement(cond, then_block, else_block, location);
}
 
// Class Case_clauses::Hash_integer_value.
 
class Case_clauses::Hash_integer_value
{
 public:
  size_t
  operator()(Expression*) const;
};
 
size_t
Case_clauses::Hash_integer_value::operator()(Expression* pe) const
{
  Type* itype;
  mpz_t ival;
  mpz_init(ival);
  if (!pe->integer_constant_value(true, ival, &itype))
    go_unreachable();
  size_t ret = mpz_get_ui(ival);
  mpz_clear(ival);
  return ret;
}
 
// Class Case_clauses::Eq_integer_value.
 
class Case_clauses::Eq_integer_value
{
 public:
  bool
  operator()(Expression*, Expression*) const;
};
 
bool
Case_clauses::Eq_integer_value::operator()(Expression* a, Expression* b) const
{
  Type* atype;
  Type* btype;
  mpz_t aval;
  mpz_t bval;
  mpz_init(aval);
  mpz_init(bval);
  if (!a->integer_constant_value(true, aval, &atype)
      || !b->integer_constant_value(true, bval, &btype))
    go_unreachable();
  bool ret = mpz_cmp(aval, bval) == 0;
  mpz_clear(aval);
  mpz_clear(bval);
  return ret;
}
 
// Class Case_clauses::Case_clause.
 
// Traversal.
 
int
Case_clauses::Case_clause::traverse(Traverse* traverse)
{
  if (this->cases_ != NULL
      && (traverse->traverse_mask()
          & (Traverse::traverse_types | Traverse::traverse_expressions)) != 0)
    {
      if (this->cases_->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  if (this->statements_ != NULL)
    {
      if (this->statements_->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  return TRAVERSE_CONTINUE;
}
 
// Check whether all the case expressions are integer constants.
 
bool
Case_clauses::Case_clause::is_constant() const
{
  if (this->cases_ != NULL)
    {
      for (Expression_list::const_iterator p = this->cases_->begin();
           p != this->cases_->end();
           ++p)
        if (!(*p)->is_constant() || (*p)->type()->integer_type() == NULL)
          return false;
    }
  return true;
}
 
// Lower a case clause for a nonconstant switch.  VAL_TEMP is the
// value we are switching on; it may be NULL.  If START_LABEL is not
// NULL, it goes at the start of the statements, after the condition
// test.  We branch to FINISH_LABEL at the end of the statements.
 
void
Case_clauses::Case_clause::lower(Block* b, Temporary_statement* val_temp,
                                 Unnamed_label* start_label,
                                 Unnamed_label* finish_label) const
{
  Location loc = this->location_;
  Unnamed_label* next_case_label;
  if (this->cases_ == NULL || this->cases_->empty())
    {
      go_assert(this->is_default_);
      next_case_label = NULL;
    }
  else
    {
      Expression* cond = NULL;
 
      for (Expression_list::const_iterator p = this->cases_->begin();
           p != this->cases_->end();
           ++p)
        {
          Expression* this_cond;
          if (val_temp == NULL)
            this_cond = *p;
          else
            {
              Expression* ref = Expression::make_temporary_reference(val_temp,
                                                                     loc);
              this_cond = Expression::make_binary(OPERATOR_EQEQ, ref, *p, loc);
            }
 
          if (cond == NULL)
            cond = this_cond;
          else
            cond = Expression::make_binary(OPERATOR_OROR, cond, this_cond, loc);
        }
 
      Block* then_block = new Block(b, loc);
      next_case_label = new Unnamed_label(Linemap::unknown_location());
      Statement* s = Statement::make_goto_unnamed_statement(next_case_label,
                                                            loc);
      then_block->add_statement(s);
 
      // if !COND { goto NEXT_CASE_LABEL }
      cond = Expression::make_unary(OPERATOR_NOT, cond, loc);
      s = Statement::make_if_statement(cond, then_block, NULL, loc);
      b->add_statement(s);
    }
 
  if (start_label != NULL)
    b->add_statement(Statement::make_unnamed_label_statement(start_label));
 
  if (this->statements_ != NULL)
    b->add_statement(Statement::make_block_statement(this->statements_, loc));
 
  Statement* s = Statement::make_goto_unnamed_statement(finish_label, loc);
  b->add_statement(s);
 
  if (next_case_label != NULL)
    b->add_statement(Statement::make_unnamed_label_statement(next_case_label));
}
 
// Determine types.
 
void
Case_clauses::Case_clause::determine_types(Type* type)
{
  if (this->cases_ != NULL)
    {
      Type_context case_context(type, false);
      for (Expression_list::iterator p = this->cases_->begin();
           p != this->cases_->end();
           ++p)
        (*p)->determine_type(&case_context);
    }
  if (this->statements_ != NULL)
    this->statements_->determine_types();
}
 
// Check types.  Returns false if there was an error.
 
bool
Case_clauses::Case_clause::check_types(Type* type)
{
  if (this->cases_ != NULL)
    {
      for (Expression_list::iterator p = this->cases_->begin();
           p != this->cases_->end();
           ++p)
        {
          if (!Type::are_assignable(type, (*p)->type(), NULL)
              && !Type::are_assignable((*p)->type(), type, NULL))
            {
              error_at((*p)->location(),
                       "type mismatch between switch value and case clause");
              return false;
            }
        }
    }
  return true;
}
 
// Return true if this clause may fall through to the following
// statements.  Note that this is not the same as whether the case
// uses the "fallthrough" keyword.
 
bool
Case_clauses::Case_clause::may_fall_through() const
{
  if (this->statements_ == NULL)
    return true;
  return this->statements_->may_fall_through();
}
 
// Convert the case values and statements to the backend
// representation.  BREAK_LABEL is the label which break statements
// should branch to.  CASE_CONSTANTS is used to detect duplicate
// constants.  *CASES should be passed as an empty vector; the values
// for this case will be added to it.  If this is the default case,
// *CASES will remain empty.  This returns the statement to execute if
// one of these cases is selected.
 
Bstatement*
Case_clauses::Case_clause::get_backend(Translate_context* context,
                                       Unnamed_label* break_label,
                                       Case_constants* case_constants,
                                       std::vector<Bexpression*>* cases) const
{
  if (this->cases_ != NULL)
    {
      go_assert(!this->is_default_);
      for (Expression_list::const_iterator p = this->cases_->begin();
           p != this->cases_->end();
           ++p)
        {
          Expression* e = *p;
          if (e->classification() != Expression::EXPRESSION_INTEGER)
            {
              Type* itype;
              mpz_t ival;
              mpz_init(ival);
              if (!(*p)->integer_constant_value(true, ival, &itype))
                {
                  // Something went wrong.  This can happen with a
                  // negative constant and an unsigned switch value.
                  go_assert(saw_errors());
                  continue;
                }
              go_assert(itype != NULL);
              e = Expression::make_integer(&ival, itype, e->location());
              mpz_clear(ival);
            }
 
          std::pair<Case_constants::iterator, bool> ins =
            case_constants->insert(e);
          if (!ins.second)
            {
              // Value was already present.
              error_at(this->location_, "duplicate case in switch");
              continue;
            }
 
          tree case_tree = e->get_tree(context);
          Bexpression* case_expr = tree_to_expr(case_tree);
          cases->push_back(case_expr);
        }
    }
 
  Bstatement* statements;
  if (this->statements_ == NULL)
    statements = NULL;
  else
    {
      Bblock* bblock = this->statements_->get_backend(context);
      statements = context->backend()->block_statement(bblock);
    }
 
  Bstatement* break_stat;
  if (this->is_fallthrough_)
    break_stat = NULL;
  else
    break_stat = break_label->get_goto(context, this->location_);
 
  if (statements == NULL)
    return break_stat;
  else if (break_stat == NULL)
    return statements;
  else
    return context->backend()->compound_statement(statements, break_stat);
}
 
// Dump the AST representation for a case clause
 
void
Case_clauses::Case_clause::dump_clause(Ast_dump_context* ast_dump_context)
    const
{
  ast_dump_context->print_indent();
  if (this->is_default_)
    {
      ast_dump_context->ostream() << "default:";
    }
  else
    {
      ast_dump_context->ostream() << "case ";
      ast_dump_context->dump_expression_list(this->cases_);
      ast_dump_context->ostream() << ":" ;
    }
  ast_dump_context->dump_block(this->statements_);
  if (this->is_fallthrough_)
    {
      ast_dump_context->print_indent();
      ast_dump_context->ostream() <<  " (fallthrough)" << std::endl;
    }
}
 
// Class Case_clauses.
 
// Traversal.
 
int
Case_clauses::traverse(Traverse* traverse)
{
  for (Clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    {
      if (p->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  return TRAVERSE_CONTINUE;
}
 
// Check whether all the case expressions are constant.
 
bool
Case_clauses::is_constant() const
{
  for (Clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    if (!p->is_constant())
      return false;
  return true;
}
 
// Lower case clauses for a nonconstant switch.
 
void
Case_clauses::lower(Block* b, Temporary_statement* val_temp,
                    Unnamed_label* break_label) const
{
  // The default case.
  const Case_clause* default_case = NULL;
 
  // The label for the fallthrough of the previous case.
  Unnamed_label* last_fallthrough_label = NULL;
 
  // The label for the start of the default case.  This is used if the
  // case before the default case falls through.
  Unnamed_label* default_start_label = NULL;
 
  // The label for the end of the default case.  This normally winds
  // up as BREAK_LABEL, but it will be different if the default case
  // falls through.
  Unnamed_label* default_finish_label = NULL;
 
  for (Clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    {
      // The label to use for the start of the statements for this
      // case.  This is NULL unless the previous case falls through.
      Unnamed_label* start_label = last_fallthrough_label;
 
      // The label to jump to after the end of the statements for this
      // case.
      Unnamed_label* finish_label = break_label;
 
      last_fallthrough_label = NULL;
      if (p->is_fallthrough() && p + 1 != this->clauses_.end())
        {
          finish_label = new Unnamed_label(p->location());
          last_fallthrough_label = finish_label;
        }
 
      if (!p->is_default())
        p->lower(b, val_temp, start_label, finish_label);
      else
        {
          // We have to move the default case to the end, so that we
          // only use it if all the other tests fail.
          default_case = &*p;
          default_start_label = start_label;
          default_finish_label = finish_label;
        }
    }
 
  if (default_case != NULL)
    default_case->lower(b, val_temp, default_start_label,
                        default_finish_label);
}
 
// Determine types.
 
void
Case_clauses::determine_types(Type* type)
{
  for (Clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    p->determine_types(type);
}
 
// Check types.  Returns false if there was an error.
 
bool
Case_clauses::check_types(Type* type)
{
  bool ret = true;
  for (Clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    {
      if (!p->check_types(type))
        ret = false;
    }
  return ret;
}
 
// Return true if these clauses may fall through to the statements
// following the switch statement.
 
bool
Case_clauses::may_fall_through() const
{
  bool found_default = false;
  for (Clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    {
      if (p->may_fall_through() && !p->is_fallthrough())
        return true;
      if (p->is_default())
        found_default = true;
    }
  return !found_default;
}
 
// Convert the cases to the backend representation.  This sets
// *ALL_CASES and *ALL_STATEMENTS.
 
void
Case_clauses::get_backend(Translate_context* context,
                          Unnamed_label* break_label,
                          std::vector<std::vector<Bexpression*> >* all_cases,
                          std::vector<Bstatement*>* all_statements) const
{
  Case_constants case_constants;
 
  size_t c = this->clauses_.size();
  all_cases->resize(c);
  all_statements->resize(c);
 
  size_t i = 0;
  for (Clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p, ++i)
    {
      std::vector<Bexpression*> cases;
      Bstatement* stat = p->get_backend(context, break_label, &case_constants,
                                        &cases);
      (*all_cases)[i].swap(cases);
      (*all_statements)[i] = stat;
    }
}
 
// Dump the AST representation for case clauses (from a switch statement)
 
void
Case_clauses::dump_clauses(Ast_dump_context* ast_dump_context) const
{
  for (Clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    p->dump_clause(ast_dump_context);
}
 
// A constant switch statement.  A Switch_statement is lowered to this
// when all the cases are constants.
 
class Constant_switch_statement : public Statement
{
 public:
  Constant_switch_statement(Expression* val, Case_clauses* clauses,
                            Unnamed_label* break_label,
                            Location location)
    : Statement(STATEMENT_CONSTANT_SWITCH, location),
      val_(val), clauses_(clauses), break_label_(break_label)
  { }
 
 protected:
  int
  do_traverse(Traverse*);
 
  void
  do_determine_types();
 
  void
  do_check_types(Gogo*);
 
  bool
  do_may_fall_through() const;
 
  Bstatement*
  do_get_backend(Translate_context*);
 
  void
  do_dump_statement(Ast_dump_context*) const;
 
 private:
  // The value to switch on.
  Expression* val_;
  // The case clauses.
  Case_clauses* clauses_;
  // The break label, if needed.
  Unnamed_label* break_label_;
};
 
// Traversal.
 
int
Constant_switch_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->clauses_->traverse(traverse);
}
 
// Determine types.
 
void
Constant_switch_statement::do_determine_types()
{
  this->val_->determine_type_no_context();
  this->clauses_->determine_types(this->val_->type());
}
 
// Check types.
 
void
Constant_switch_statement::do_check_types(Gogo*)
{
  if (!this->clauses_->check_types(this->val_->type()))
    this->set_is_error();
}
 
// Return whether this switch may fall through.
 
bool
Constant_switch_statement::do_may_fall_through() const
{
  if (this->clauses_ == NULL)
    return true;
 
  // If we have a break label, then some case needed it.  That implies
  // that the switch statement as a whole can fall through.
  if (this->break_label_ != NULL)
    return true;
 
  return this->clauses_->may_fall_through();
}
 
// Convert to GENERIC.
 
Bstatement*
Constant_switch_statement::do_get_backend(Translate_context* context)
{
  tree switch_val_tree = this->val_->get_tree(context);
  Bexpression* switch_val_expr = tree_to_expr(switch_val_tree);
 
  Unnamed_label* break_label = this->break_label_;
  if (break_label == NULL)
    break_label = new Unnamed_label(this->location());
 
  std::vector<std::vector<Bexpression*> > all_cases;
  std::vector<Bstatement*> all_statements;
  this->clauses_->get_backend(context, break_label, &all_cases,
                              &all_statements);
 
  Bstatement* switch_statement;
  switch_statement = context->backend()->switch_statement(switch_val_expr,
                                                          all_cases,
                                                          all_statements,
                                                          this->location());
  Bstatement* ldef = break_label->get_definition(context);
  return context->backend()->compound_statement(switch_statement, ldef);
}
 
// Dump the AST representation for a constant switch statement.
 
void
Constant_switch_statement::do_dump_statement(Ast_dump_context* ast_dump_context)
    const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "switch ";
  ast_dump_context->dump_expression(this->val_);
 
  if (ast_dump_context->dump_subblocks())
    {
      ast_dump_context->ostream() << " {" << std::endl;
      this->clauses_->dump_clauses(ast_dump_context);
      ast_dump_context->ostream() << "}";
    }
 
   ast_dump_context->ostream() << std::endl;
}
 
// Class Switch_statement.
 
// Traversal.
 
int
Switch_statement::do_traverse(Traverse* traverse)
{
  if (this->val_ != NULL)
    {
      if (this->traverse_expression(traverse, &this->val_) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  return this->clauses_->traverse(traverse);
}
 
// Lower a Switch_statement to a Constant_switch_statement or a series
// of if statements.
 
Statement*
Switch_statement::do_lower(Gogo*, Named_object*, Block* enclosing,
                           Statement_inserter*)
{
  Location loc = this->location();
 
  if (this->val_ != NULL
      && (this->val_->is_error_expression()
          || this->val_->type()->is_error()))
    return Statement::make_error_statement(loc);
 
  if (this->val_ != NULL
      && this->val_->type()->integer_type() != NULL
      && !this->clauses_->empty()
      && this->clauses_->is_constant())
    return new Constant_switch_statement(this->val_, this->clauses_,
                                         this->break_label_, loc);
 
  Block* b = new Block(enclosing, loc);
 
  if (this->clauses_->empty())
    {
      Expression* val = this->val_;
      if (val == NULL)
        val = Expression::make_boolean(true, loc);
      return Statement::make_statement(val, true);
    }
 
  Temporary_statement* val_temp;
  if (this->val_ == NULL)
    val_temp = NULL;
  else
    {
      // var val_temp VAL_TYPE = VAL
      val_temp = Statement::make_temporary(NULL, this->val_, loc);
      b->add_statement(val_temp);
    }
 
  this->clauses_->lower(b, val_temp, this->break_label());
 
  Statement* s = Statement::make_unnamed_label_statement(this->break_label_);
  b->add_statement(s);
 
  return Statement::make_block_statement(b, loc);
}
 
// Return the break label for this switch statement, creating it if
// necessary.
 
Unnamed_label*
Switch_statement::break_label()
{
  if (this->break_label_ == NULL)
    this->break_label_ = new Unnamed_label(this->location());
  return this->break_label_;
}
 
// Dump the AST representation for a switch statement.
 
void
Switch_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "switch ";
  if (this->val_ != NULL)
    {
      ast_dump_context->dump_expression(this->val_);
    }
  if (ast_dump_context->dump_subblocks())
    {
      ast_dump_context->ostream() << " {" << std::endl;
      this->clauses_->dump_clauses(ast_dump_context);
      ast_dump_context->print_indent();
      ast_dump_context->ostream() << "}";
    }
  ast_dump_context->ostream() << std::endl;
}
 
// Make a switch statement.
 
Switch_statement*
Statement::make_switch_statement(Expression* val, Location location)
{
  return new Switch_statement(val, location);
}
 
// Class Type_case_clauses::Type_case_clause.
 
// Traversal.
 
int
Type_case_clauses::Type_case_clause::traverse(Traverse* traverse)
{
  if (!this->is_default_
      && ((traverse->traverse_mask()
           & (Traverse::traverse_types | Traverse::traverse_expressions)) != 0)
      && Type::traverse(this->type_, traverse) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  if (this->statements_ != NULL)
    return this->statements_->traverse(traverse);
  return TRAVERSE_CONTINUE;
}
 
// Lower one clause in a type switch.  Add statements to the block B.
// The type descriptor we are switching on is in DESCRIPTOR_TEMP.
// BREAK_LABEL is the label at the end of the type switch.
// *STMTS_LABEL, if not NULL, is a label to put at the start of the
// statements.
 
void
Type_case_clauses::Type_case_clause::lower(Type* switch_val_type,
                                           Block* b,
                                           Temporary_statement* descriptor_temp,
                                           Unnamed_label* break_label,
                                           Unnamed_label** stmts_label) const
{
  Location loc = this->location_;
 
  Unnamed_label* next_case_label = NULL;
  if (!this->is_default_)
    {
      Type* type = this->type_;
 
      std::string reason;
      if (switch_val_type->interface_type() != NULL
          && !type->is_nil_constant_as_type()
          && type->interface_type() == NULL
          && !switch_val_type->interface_type()->implements_interface(type,
                                                                      &reason))
        {
          if (reason.empty())
            error_at(this->location_, "impossible type switch case");
          else
            error_at(this->location_, "impossible type switch case (%s)",
                     reason.c_str());
        }
 
      Expression* ref = Expression::make_temporary_reference(descriptor_temp,
                                                             loc);
 
      Expression* cond;
      // The language permits case nil, which is of course a constant
      // rather than a type.  It will appear here as an invalid
      // forwarding type.
      if (type->is_nil_constant_as_type())
        cond = Expression::make_binary(OPERATOR_EQEQ, ref,
                                       Expression::make_nil(loc),
                                       loc);
      else
        cond = Runtime::make_call((type->interface_type() == NULL
                                   ? Runtime::IFACETYPEEQ
                                   : Runtime::IFACEI2TP),
                                  loc, 2,
                                  Expression::make_type_descriptor(type, loc),
                                  ref);
 
      Unnamed_label* dest;
      if (!this->is_fallthrough_)
        {
          // if !COND { goto NEXT_CASE_LABEL }
          next_case_label = new Unnamed_label(Linemap::unknown_location());
          dest = next_case_label;
          cond = Expression::make_unary(OPERATOR_NOT, cond, loc);
        }
      else
        {
          // if COND { goto STMTS_LABEL }
          go_assert(stmts_label != NULL);
          if (*stmts_label == NULL)
            *stmts_label = new Unnamed_label(Linemap::unknown_location());
          dest = *stmts_label;
        }
      Block* then_block = new Block(b, loc);
      Statement* s = Statement::make_goto_unnamed_statement(dest, loc);
      then_block->add_statement(s);
      s = Statement::make_if_statement(cond, then_block, NULL, loc);
      b->add_statement(s);
    }
 
  if (this->statements_ != NULL
      || (!this->is_fallthrough_
          && stmts_label != NULL
          && *stmts_label != NULL))
    {
      go_assert(!this->is_fallthrough_);
      if (stmts_label != NULL && *stmts_label != NULL)
        {
          go_assert(!this->is_default_);
          if (this->statements_ != NULL)
            (*stmts_label)->set_location(this->statements_->start_location());
          Statement* s = Statement::make_unnamed_label_statement(*stmts_label);
          b->add_statement(s);
          *stmts_label = NULL;
        }
      if (this->statements_ != NULL)
        b->add_statement(Statement::make_block_statement(this->statements_,
                                                         loc));
    }
 
  if (this->is_fallthrough_)
    go_assert(next_case_label == NULL);
  else
    {
      Location gloc = (this->statements_ == NULL
                              ? loc
                              : this->statements_->end_location());
      b->add_statement(Statement::make_goto_unnamed_statement(break_label,
                                                              gloc));
      if (next_case_label != NULL)
        {
          Statement* s =
            Statement::make_unnamed_label_statement(next_case_label);
          b->add_statement(s);
        }
    }
}
 
// Dump the AST representation for a type case clause
 
void
Type_case_clauses::Type_case_clause::dump_clause(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  if (this->is_default_)
    {
      ast_dump_context->ostream() << "default:";
    }
  else
    {
      ast_dump_context->ostream() << "case ";
      ast_dump_context->dump_type(this->type_);
      ast_dump_context->ostream() << ":" ;
    }
  ast_dump_context->dump_block(this->statements_);
  if (this->is_fallthrough_)
    {
      ast_dump_context->print_indent();
      ast_dump_context->ostream() <<  " (fallthrough)" << std::endl;
    }
}
 
// Class Type_case_clauses.
 
// Traversal.
 
int
Type_case_clauses::traverse(Traverse* traverse)
{
  for (Type_clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    {
      if (p->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  return TRAVERSE_CONTINUE;
}
 
// Check for duplicate types.
 
void
Type_case_clauses::check_duplicates() const
{
  typedef Unordered_set_hash(const Type*, Type_hash_identical,
                             Type_identical) Types_seen;
  Types_seen types_seen;
  for (Type_clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    {
      Type* t = p->type();
      if (t == NULL)
        continue;
      if (t->is_nil_constant_as_type())
        t = Type::make_nil_type();
      std::pair<Types_seen::iterator, bool> ins = types_seen.insert(t);
      if (!ins.second)
        error_at(p->location(), "duplicate type in switch");
    }
}
 
// Lower the clauses in a type switch.  Add statements to the block B.
// The type descriptor we are switching on is in DESCRIPTOR_TEMP.
// BREAK_LABEL is the label at the end of the type switch.
 
void
Type_case_clauses::lower(Type* switch_val_type, Block* b,
                         Temporary_statement* descriptor_temp,
                         Unnamed_label* break_label) const
{
  const Type_case_clause* default_case = NULL;
 
  Unnamed_label* stmts_label = NULL;
  for (Type_clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    {
      if (!p->is_default())
        p->lower(switch_val_type, b, descriptor_temp, break_label,
                 &stmts_label);
      else
        {
          // We are generating a series of tests, which means that we
          // need to move the default case to the end.
          default_case = &*p;
        }
    }
  go_assert(stmts_label == NULL);
 
  if (default_case != NULL)
    default_case->lower(switch_val_type, b, descriptor_temp, break_label,
                        NULL);
}
 
// Dump the AST representation for case clauses (from a switch statement)
 
void
Type_case_clauses::dump_clauses(Ast_dump_context* ast_dump_context) const
{
  for (Type_clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    p->dump_clause(ast_dump_context);
}
 
// Class Type_switch_statement.
 
// Traversal.
 
int
Type_switch_statement::do_traverse(Traverse* traverse)
{
  if (this->var_ == NULL)
    {
      if (this->traverse_expression(traverse, &this->expr_) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  if (this->clauses_ != NULL)
    return this->clauses_->traverse(traverse);
  return TRAVERSE_CONTINUE;
}
 
// Lower a type switch statement to a series of if statements.  The gc
// compiler is able to generate a table in some cases.  However, that
// does not work for us because we may have type descriptors in
// different shared libraries, so we can't compare them with simple
// equality testing.
 
Statement*
Type_switch_statement::do_lower(Gogo*, Named_object*, Block* enclosing,
                                Statement_inserter*)
{
  const Location loc = this->location();
 
  if (this->clauses_ != NULL)
    this->clauses_->check_duplicates();
 
  Block* b = new Block(enclosing, loc);
 
  Type* val_type = (this->var_ != NULL
                    ? this->var_->var_value()->type()
                    : this->expr_->type());
 
  // var descriptor_temp DESCRIPTOR_TYPE
  Type* descriptor_type = Type::make_type_descriptor_ptr_type();
  Temporary_statement* descriptor_temp =
    Statement::make_temporary(descriptor_type, NULL, loc);
  b->add_statement(descriptor_temp);
 
  if (val_type->interface_type() == NULL)
    {
      // Doing a type switch on a non-interface type.  Should we issue
      // a warning for this case?
      Expression* lhs = Expression::make_temporary_reference(descriptor_temp,
                                                             loc);
      Expression* rhs;
      if (val_type->is_nil_type())
        rhs = Expression::make_nil(loc);
      else
        {
          if (val_type->is_abstract())
            val_type = val_type->make_non_abstract_type();
          rhs = Expression::make_type_descriptor(val_type, loc);
        }
      Statement* s = Statement::make_assignment(lhs, rhs, loc);
      b->add_statement(s);
    }
  else
    {
      // descriptor_temp = ifacetype(val_temp)
      // FIXME: This should be inlined.
      bool is_empty = val_type->interface_type()->is_empty();
      Expression* ref;
      if (this->var_ == NULL)
        ref = this->expr_;
      else
        ref = Expression::make_var_reference(this->var_, loc);
      Expression* call = Runtime::make_call((is_empty
                                             ? Runtime::EFACETYPE
                                             : Runtime::IFACETYPE),
                                            loc, 1, ref);
      Temporary_reference_expression* lhs =
        Expression::make_temporary_reference(descriptor_temp, loc);
      lhs->set_is_lvalue();
      Statement* s = Statement::make_assignment(lhs, call, loc);
      b->add_statement(s);
    }
 
  if (this->clauses_ != NULL)
    this->clauses_->lower(val_type, b, descriptor_temp, this->break_label());
 
  Statement* s = Statement::make_unnamed_label_statement(this->break_label_);
  b->add_statement(s);
 
  return Statement::make_block_statement(b, loc);
}
 
// Return the break label for this type switch statement, creating it
// if necessary.
 
Unnamed_label*
Type_switch_statement::break_label()
{
  if (this->break_label_ == NULL)
    this->break_label_ = new Unnamed_label(this->location());
  return this->break_label_;
}
 
// Dump the AST representation for a type switch statement
 
void
Type_switch_statement::do_dump_statement(Ast_dump_context* ast_dump_context)
    const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "switch " << this->var_->name() << " = ";
  ast_dump_context->dump_expression(this->expr_);
  ast_dump_context->ostream() << " .(type)";
  if (ast_dump_context->dump_subblocks())
    {
      ast_dump_context->ostream() << " {" << std::endl;
      this->clauses_->dump_clauses(ast_dump_context);
      ast_dump_context->ostream() << "}";
    }
  ast_dump_context->ostream() << std::endl;
}
 
// Make a type switch statement.
 
Type_switch_statement*
Statement::make_type_switch_statement(Named_object* var, Expression* expr,
                                      Location location)
{
  return new Type_switch_statement(var, expr, location);
}
 
// Class Send_statement.
 
// Traversal.
 
int
Send_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->channel_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->traverse_expression(traverse, &this->val_);
}
 
// Determine types.
 
void
Send_statement::do_determine_types()
{
  this->channel_->determine_type_no_context();
  Type* type = this->channel_->type();
  Type_context context;
  if (type->channel_type() != NULL)
    context.type = type->channel_type()->element_type();
  this->val_->determine_type(&context);
}
 
// Check types.
 
void
Send_statement::do_check_types(Gogo*)
{
  Type* type = this->channel_->type();
  if (type->is_error())
    {
      this->set_is_error();
      return;
    }
  Channel_type* channel_type = type->channel_type();
  if (channel_type == NULL)
    {
      error_at(this->location(), "left operand of %<<-%> must be channel");
      this->set_is_error();
      return;
    }
  Type* element_type = channel_type->element_type();
  if (!Type::are_assignable(element_type, this->val_->type(), NULL))
    {
      this->report_error(_("incompatible types in send"));
      return;
    }
  if (!channel_type->may_send())
    {
      this->report_error(_("invalid send on receive-only channel"));
      return;
    }
}
 
// Convert a send statement to the backend representation.
 
Bstatement*
Send_statement::do_get_backend(Translate_context* context)
{
  Location loc = this->location();
 
  Channel_type* channel_type = this->channel_->type()->channel_type();
  Type* element_type = channel_type->element_type();
  Expression* val = Expression::make_cast(element_type, this->val_, loc);
 
  bool is_small;
  bool can_take_address;
  switch (element_type->base()->classification())
    {
    case Type::TYPE_BOOLEAN:
    case Type::TYPE_INTEGER:
    case Type::TYPE_FUNCTION:
    case Type::TYPE_POINTER:
    case Type::TYPE_MAP:
    case Type::TYPE_CHANNEL:
      is_small = true;
      can_take_address = false;
      break;
 
    case Type::TYPE_FLOAT:
    case Type::TYPE_COMPLEX:
    case Type::TYPE_STRING:
    case Type::TYPE_INTERFACE:
      is_small = false;
      can_take_address = false;
      break;
 
    case Type::TYPE_STRUCT:
      is_small = false;
      can_take_address = true;
      break;
 
    case Type::TYPE_ARRAY:
      is_small = false;
      can_take_address = !element_type->is_slice_type();
      break;
 
    default:
    case Type::TYPE_ERROR:
    case Type::TYPE_VOID:
    case Type::TYPE_SINK:
    case Type::TYPE_NIL:
    case Type::TYPE_NAMED:
    case Type::TYPE_FORWARD:
      go_assert(saw_errors());
      return context->backend()->error_statement();
    }
 
  // Only try to take the address of a variable.  We have already
  // moved variables to the heap, so this should not cause that to
  // happen unnecessarily.
  if (can_take_address
      && val->var_expression() == NULL
      && val->temporary_reference_expression() == NULL)
    can_take_address = false;
 
  Expression* td = Expression::make_type_descriptor(this->channel_->type(),
                                                    loc);
 
  Runtime::Function code;
  Bstatement* btemp = NULL;
  if (is_small)
      {
        // Type is small enough to handle as uint64.
        code = Runtime::SEND_SMALL;
        val = Expression::make_unsafe_cast(Type::lookup_integer_type("uint64"),
                                           val, loc);
      }
  else if (can_take_address)
    {
      // Must pass address of value.  The function doesn't change the
      // value, so just take its address directly.
      code = Runtime::SEND_BIG;
      val = Expression::make_unary(OPERATOR_AND, val, loc);
    }
  else
    {
      // Must pass address of value, but the value is small enough
      // that it might be in registers.  Copy value into temporary
      // variable to take address.
      code = Runtime::SEND_BIG;
      Temporary_statement* temp = Statement::make_temporary(element_type,
                                                            val, loc);
      Expression* ref = Expression::make_temporary_reference(temp, loc);
      val = Expression::make_unary(OPERATOR_AND, ref, loc);
      btemp = temp->get_backend(context);
    }
 
  Expression* call = Runtime::make_call(code, loc, 3, td, this->channel_, val);
 
  context->gogo()->lower_expression(context->function(), NULL, &call);
  Bexpression* bcall = tree_to_expr(call->get_tree(context));
  Bstatement* s = context->backend()->expression_statement(bcall);
 
  if (btemp == NULL)
    return s;
  else
    return context->backend()->compound_statement(btemp, s);
}
 
// Dump the AST representation for a send statement
 
void
Send_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->dump_expression(this->channel_);
  ast_dump_context->ostream() << " <- ";
  ast_dump_context->dump_expression(this->val_);
  ast_dump_context->ostream() << std::endl;
}
 
// Make a send statement.
 
Send_statement*
Statement::make_send_statement(Expression* channel, Expression* val,
                               Location location)
{
  return new Send_statement(channel, val, location);
}
 
// Class Select_clauses::Select_clause.
 
// Traversal.
 
int
Select_clauses::Select_clause::traverse(Traverse* traverse)
{
  if (!this->is_lowered_
      && (traverse->traverse_mask()
          & (Traverse::traverse_types | Traverse::traverse_expressions)) != 0)
    {
      if (this->channel_ != NULL)
        {
          if (Expression::traverse(&this->channel_, traverse) == TRAVERSE_EXIT)
            return TRAVERSE_EXIT;
        }
      if (this->val_ != NULL)
        {
          if (Expression::traverse(&this->val_, traverse) == TRAVERSE_EXIT)
            return TRAVERSE_EXIT;
        }
      if (this->closed_ != NULL)
        {
          if (Expression::traverse(&this->closed_, traverse) == TRAVERSE_EXIT)
            return TRAVERSE_EXIT;
        }
    }
  if (this->statements_ != NULL)
    {
      if (this->statements_->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  return TRAVERSE_CONTINUE;
}
 
// Lowering.  We call a function to register this clause, and arrange
// to set any variables in any receive clause.
 
void
Select_clauses::Select_clause::lower(Gogo* gogo, Named_object* function,
                                     Block* b, Temporary_statement* sel)
{
  Location loc = this->location_;
 
  Expression* selref = Expression::make_temporary_reference(sel, loc);
 
  mpz_t ival;
  mpz_init_set_ui(ival, this->index_);
  Expression* index_expr = Expression::make_integer(&ival, NULL, loc);
  mpz_clear(ival);
 
  if (this->is_default_)
    {
      go_assert(this->channel_ == NULL && this->val_ == NULL);
      this->lower_default(b, selref, index_expr);
      this->is_lowered_ = true;
      return;
    }
 
  // Evaluate the channel before the select statement.
  Temporary_statement* channel_temp = Statement::make_temporary(NULL,
                                                                this->channel_,
                                                                loc);
  b->add_statement(channel_temp);
  Expression* chanref = Expression::make_temporary_reference(channel_temp,
                                                             loc);
 
  if (this->is_send_)
    this->lower_send(b, selref, chanref, index_expr);
  else
    this->lower_recv(gogo, function, b, selref, chanref, index_expr);
 
  // Now all references should be handled through the statements, not
  // through here.
  this->is_lowered_ = true;
  this->val_ = NULL;
  this->var_ = NULL;
}
 
// Lower a default clause in a select statement.
 
void
Select_clauses::Select_clause::lower_default(Block* b, Expression* selref,
                                             Expression* index_expr)
{
  Location loc = this->location_;
  Expression* call = Runtime::make_call(Runtime::SELECTDEFAULT, loc, 2, selref,
                                        index_expr);
  b->add_statement(Statement::make_statement(call, true));
}
 
// Lower a send clause in a select statement.
 
void
Select_clauses::Select_clause::lower_send(Block* b, Expression* selref,
                                          Expression* chanref,
                                          Expression* index_expr)
{
  Location loc = this->location_;
 
  Channel_type* ct = this->channel_->type()->channel_type();
  if (ct == NULL)
    return;
 
  Type* valtype = ct->element_type();
 
  // Note that copying the value to a temporary here means that we
  // evaluate the send values in the required order.
  Temporary_statement* val = Statement::make_temporary(valtype, this->val_,
                                                       loc);
  b->add_statement(val);
 
  Expression* valref = Expression::make_temporary_reference(val, loc);
  Expression* valaddr = Expression::make_unary(OPERATOR_AND, valref, loc);
 
  Expression* call = Runtime::make_call(Runtime::SELECTSEND, loc, 4, selref,
                                        chanref, valaddr, index_expr);
  b->add_statement(Statement::make_statement(call, true));
}
 
// Lower a receive clause in a select statement.
 
void
Select_clauses::Select_clause::lower_recv(Gogo* gogo, Named_object* function,
                                          Block* b, Expression* selref,
                                          Expression* chanref,
                                          Expression* index_expr)
{
  Location loc = this->location_;
 
  Channel_type* ct = this->channel_->type()->channel_type();
  if (ct == NULL)
    return;
 
  Type* valtype = ct->element_type();
  Temporary_statement* val = Statement::make_temporary(valtype, NULL, loc);
  b->add_statement(val);
 
  Expression* valref = Expression::make_temporary_reference(val, loc);
  Expression* valaddr = Expression::make_unary(OPERATOR_AND, valref, loc);
 
  Temporary_statement* closed_temp = NULL;
 
  Expression* call;
  if (this->closed_ == NULL && this->closedvar_ == NULL)
    call = Runtime::make_call(Runtime::SELECTRECV, loc, 4, selref, chanref,
                              valaddr, index_expr);
  else
    {
      closed_temp = Statement::make_temporary(Type::lookup_bool_type(), NULL,
                                              loc);
      b->add_statement(closed_temp);
      Expression* cref = Expression::make_temporary_reference(closed_temp,
                                                              loc);
      Expression* caddr = Expression::make_unary(OPERATOR_AND, cref, loc);
      call = Runtime::make_call(Runtime::SELECTRECV2, loc, 5, selref, chanref,
                                valaddr, caddr, index_expr);
    }
 
  b->add_statement(Statement::make_statement(call, true));
 
  // If the block of statements is executed, arrange for the received
  // value to move from VAL to the place where the statements expect
  // it.
 
  Block* init = NULL;
 
  if (this->var_ != NULL)
    {
      go_assert(this->val_ == NULL);
      valref = Expression::make_temporary_reference(val, loc);
      this->var_->var_value()->set_init(valref);
      this->var_->var_value()->clear_type_from_chan_element();
    }
  else if (this->val_ != NULL && !this->val_->is_sink_expression())
    {
      init = new Block(b, loc);
      valref = Expression::make_temporary_reference(val, loc);
      init->add_statement(Statement::make_assignment(this->val_, valref, loc));
    }
 
  if (this->closedvar_ != NULL)
    {
      go_assert(this->closed_ == NULL);
      Expression* cref = Expression::make_temporary_reference(closed_temp,
                                                              loc);
      this->closedvar_->var_value()->set_init(cref);
    }
  else if (this->closed_ != NULL && !this->closed_->is_sink_expression())
    {
      if (init == NULL)
        init = new Block(b, loc);
      Expression* cref = Expression::make_temporary_reference(closed_temp,
                                                              loc);
      init->add_statement(Statement::make_assignment(this->closed_, cref,
                                                     loc));
    }
 
  if (init != NULL)
    {
      gogo->lower_block(function, init);
 
      if (this->statements_ != NULL)
        init->add_statement(Statement::make_block_statement(this->statements_,
                                                            loc));
      this->statements_ = init;
    }
}
 
// Determine types.
 
void
Select_clauses::Select_clause::determine_types()
{
  go_assert(this->is_lowered_);
  if (this->statements_ != NULL)
    this->statements_->determine_types();
}
 
// Check types.
 
void
Select_clauses::Select_clause::check_types()
{
  if (this->is_default_)
    return;
 
  Channel_type* ct = this->channel_->type()->channel_type();
  if (ct == NULL)
    {
      error_at(this->channel_->location(), "expected channel");
      return;
    }
 
  if (this->is_send_ && !ct->may_send())
    error_at(this->location(), "invalid send on receive-only channel");
  else if (!this->is_send_ && !ct->may_receive())
    error_at(this->location(), "invalid receive on send-only channel");
}
 
// Whether this clause may fall through to the statement which follows
// the overall select statement.
 
bool
Select_clauses::Select_clause::may_fall_through() const
{
  if (this->statements_ == NULL)
    return true;
  return this->statements_->may_fall_through();
}
 
// Return the backend representation for the statements to execute.
 
Bstatement*
Select_clauses::Select_clause::get_statements_backend(
    Translate_context* context)
{
  if (this->statements_ == NULL)
    return NULL;
  Bblock* bblock = this->statements_->get_backend(context);
  return context->backend()->block_statement(bblock);
}
 
// Dump the AST representation for a select case clause
 
void
Select_clauses::Select_clause::dump_clause(
    Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  if (this->is_default_)
    {
      ast_dump_context->ostream() << "default:";
    }
  else
    {
      ast_dump_context->ostream() << "case "  ;
      if (this->is_send_)
        {
          ast_dump_context->dump_expression(this->channel_);
          ast_dump_context->ostream() << " <- " ;
          if (this->val_ != NULL)
            ast_dump_context->dump_expression(this->val_);
        }
      else
        {
          if (this->val_ != NULL)
            ast_dump_context->dump_expression(this->val_);
          if (this->closed_ != NULL)
            {
              // FIXME: can val_ == NULL and closed_ ! = NULL?
              ast_dump_context->ostream() << " , " ;
              ast_dump_context->dump_expression(this->closed_);
            }
          if (this->closedvar_ != NULL || this->var_ != NULL)
            ast_dump_context->ostream() << " := " ;
 
          ast_dump_context->ostream() << " <- " ;
          ast_dump_context->dump_expression(this->channel_);
        }
      ast_dump_context->ostream() << ":" ;
    }
  ast_dump_context->dump_block(this->statements_);
}
 
// Class Select_clauses.
 
// Traversal.
 
int
Select_clauses::traverse(Traverse* traverse)
{
  for (Clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    {
      if (p->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  return TRAVERSE_CONTINUE;
}
 
// Lowering.  Here we pull out the channel and the send values, to
// enforce the order of evaluation.  We also add explicit send and
// receive statements to the clauses.
 
void
Select_clauses::lower(Gogo* gogo, Named_object* function, Block* b,
                      Temporary_statement* sel)
{
  for (Clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    p->lower(gogo, function, b, sel);
}
 
// Determine types.
 
void
Select_clauses::determine_types()
{
  for (Clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    p->determine_types();
}
 
// Check types.
 
void
Select_clauses::check_types()
{
  for (Clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    p->check_types();
}
 
// Return whether these select clauses fall through to the statement
// following the overall select statement.
 
bool
Select_clauses::may_fall_through() const
{
  for (Clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    if (p->may_fall_through())
      return true;
  return false;
}
 
// Convert to the backend representation.  We have already accumulated
// all the select information.  Now we call selectgo, which will
// return the index of the clause to execute.
 
Bstatement*
Select_clauses::get_backend(Translate_context* context,
                            Temporary_statement* sel,
                            Unnamed_label *break_label,
                            Location location)
{
  size_t count = this->clauses_.size();
  std::vector<std::vector<Bexpression*> > cases(count);
  std::vector<Bstatement*> clauses(count);
 
  int i = 0;
  for (Clauses::iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p, ++i)
    {
      int index = p->index();
      mpz_t ival;
      mpz_init_set_ui(ival, index);
      Expression* index_expr = Expression::make_integer(&ival, NULL, location);
      mpz_clear(ival);
      cases[i].push_back(tree_to_expr(index_expr->get_tree(context)));
 
      Bstatement* s = p->get_statements_backend(context);
      Location gloc = (p->statements() == NULL
                       ? p->location()
                       : p->statements()->end_location());
      Bstatement* g = break_label->get_goto(context, gloc);
 
      if (s == NULL)
        clauses[i] = g;
      else
        clauses[i] = context->backend()->compound_statement(s, g);
    }
 
  Expression* selref = Expression::make_temporary_reference(sel, location);
  Expression* call = Runtime::make_call(Runtime::SELECTGO, location, 1,
                                        selref);
  context->gogo()->lower_expression(context->function(), NULL, &call);
  Bexpression* bcall = tree_to_expr(call->get_tree(context));
 
  if (count == 0)
    return context->backend()->expression_statement(bcall);
 
  std::vector<Bstatement*> statements;
  statements.reserve(2);
 
  Bstatement* switch_stmt = context->backend()->switch_statement(bcall,
                                                                 cases,
                                                                 clauses,
                                                                 location);
  statements.push_back(switch_stmt);
 
  Bstatement* ldef = break_label->get_definition(context);
  statements.push_back(ldef);
 
  return context->backend()->statement_list(statements);
}
// Dump the AST representation for select clauses.
 
void
Select_clauses::dump_clauses(Ast_dump_context* ast_dump_context) const
{
  for (Clauses::const_iterator p = this->clauses_.begin();
       p != this->clauses_.end();
       ++p)
    p->dump_clause(ast_dump_context);
}
 
// Class Select_statement.
 
// Return the break label for this switch statement, creating it if
// necessary.
 
Unnamed_label*
Select_statement::break_label()
{
  if (this->break_label_ == NULL)
    this->break_label_ = new Unnamed_label(this->location());
  return this->break_label_;
}
 
// Lower a select statement.  This will still return a select
// statement, but it will be modified to implement the order of
// evaluation rules, and to include the send and receive statements as
// explicit statements in the clauses.
 
Statement*
Select_statement::do_lower(Gogo* gogo, Named_object* function,
                           Block* enclosing, Statement_inserter*)
{
  if (this->is_lowered_)
    return this;
 
  Location loc = this->location();
 
  Block* b = new Block(enclosing, loc);
 
  go_assert(this->sel_ == NULL);
 
  mpz_t ival;
  mpz_init_set_ui(ival, this->clauses_->size());
  Expression* size_expr = Expression::make_integer(&ival, NULL, loc);
  mpz_clear(ival);
 
  Expression* call = Runtime::make_call(Runtime::NEWSELECT, loc, 1, size_expr);
 
  this->sel_ = Statement::make_temporary(NULL, call, loc);
  b->add_statement(this->sel_);
 
  this->clauses_->lower(gogo, function, b, this->sel_);
  this->is_lowered_ = true;
  b->add_statement(this);
 
  return Statement::make_block_statement(b, loc);
}
 
// Return the backend representation for a select statement.
 
Bstatement*
Select_statement::do_get_backend(Translate_context* context)
{
  return this->clauses_->get_backend(context, this->sel_, this->break_label(),
                                     this->location());
}
 
// Dump the AST representation for a select statement.
 
void
Select_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "select";
  if (ast_dump_context->dump_subblocks())
    {
      ast_dump_context->ostream() << " {" << std::endl;
      this->clauses_->dump_clauses(ast_dump_context);
      ast_dump_context->ostream() << "}";
    }
  ast_dump_context->ostream() << std::endl;
}
 
// Make a select statement.
 
Select_statement*
Statement::make_select_statement(Location location)
{
  return new Select_statement(location);
}
 
// Class For_statement.
 
// Traversal.
 
int
For_statement::do_traverse(Traverse* traverse)
{
  if (this->init_ != NULL)
    {
      if (this->init_->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  if (this->cond_ != NULL)
    {
      if (this->traverse_expression(traverse, &this->cond_) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  if (this->post_ != NULL)
    {
      if (this->post_->traverse(traverse) == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  return this->statements_->traverse(traverse);
}
 
// Lower a For_statement into if statements and gotos.  Getting rid of
// complex statements make it easier to handle garbage collection.
 
Statement*
For_statement::do_lower(Gogo*, Named_object*, Block* enclosing,
                        Statement_inserter*)
{
  Statement* s;
  Location loc = this->location();
 
  Block* b = new Block(enclosing, this->location());
  if (this->init_ != NULL)
    {
      s = Statement::make_block_statement(this->init_,
                                          this->init_->start_location());
      b->add_statement(s);
    }
 
  Unnamed_label* entry = NULL;
  if (this->cond_ != NULL)
    {
      entry = new Unnamed_label(this->location());
      b->add_statement(Statement::make_goto_unnamed_statement(entry, loc));
    }
 
  Unnamed_label* top = new Unnamed_label(this->location());
  b->add_statement(Statement::make_unnamed_label_statement(top));
 
  s = Statement::make_block_statement(this->statements_,
                                      this->statements_->start_location());
  b->add_statement(s);
 
  Location end_loc = this->statements_->end_location();
 
  Unnamed_label* cont = this->continue_label_;
  if (cont != NULL)
    b->add_statement(Statement::make_unnamed_label_statement(cont));
 
  if (this->post_ != NULL)
    {
      s = Statement::make_block_statement(this->post_,
                                          this->post_->start_location());
      b->add_statement(s);
      end_loc = this->post_->end_location();
    }
 
  if (this->cond_ == NULL)
    b->add_statement(Statement::make_goto_unnamed_statement(top, end_loc));
  else
    {
      b->add_statement(Statement::make_unnamed_label_statement(entry));
 
      Location cond_loc = this->cond_->location();
      Block* then_block = new Block(b, cond_loc);
      s = Statement::make_goto_unnamed_statement(top, cond_loc);
      then_block->add_statement(s);
 
      s = Statement::make_if_statement(this->cond_, then_block, NULL, cond_loc);
      b->add_statement(s);
    }
 
  Unnamed_label* brk = this->break_label_;
  if (brk != NULL)
    b->add_statement(Statement::make_unnamed_label_statement(brk));
 
  b->set_end_location(end_loc);
 
  return Statement::make_block_statement(b, loc);
}
 
// Return the break label, creating it if necessary.
 
Unnamed_label*
For_statement::break_label()
{
  if (this->break_label_ == NULL)
    this->break_label_ = new Unnamed_label(this->location());
  return this->break_label_;
}
 
// Return the continue LABEL_EXPR.
 
Unnamed_label*
For_statement::continue_label()
{
  if (this->continue_label_ == NULL)
    this->continue_label_ = new Unnamed_label(this->location());
  return this->continue_label_;
}
 
// Set the break and continue labels a for statement.  This is used
// when lowering a for range statement.
 
void
For_statement::set_break_continue_labels(Unnamed_label* break_label,
                                         Unnamed_label* continue_label)
{
  go_assert(this->break_label_ == NULL && this->continue_label_ == NULL);
  this->break_label_ = break_label;
  this->continue_label_ = continue_label;
}
 
// Dump the AST representation for a for statement.
 
void
For_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
  if (this->init_ != NULL && ast_dump_context->dump_subblocks())
    {
      ast_dump_context->print_indent();
      ast_dump_context->indent();
      ast_dump_context->ostream() << "// INIT  " << std::endl;
      ast_dump_context->dump_block(this->init_);
      ast_dump_context->unindent();
    }
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "for ";
  if (this->cond_ != NULL)
    ast_dump_context->dump_expression(this->cond_);
 
  if (ast_dump_context->dump_subblocks())
    {
      ast_dump_context->ostream() << " {" << std::endl;
      ast_dump_context->dump_block(this->statements_);
      if (this->init_ != NULL)
        {
          ast_dump_context->print_indent();
          ast_dump_context->ostream() << "// POST " << std::endl;
          ast_dump_context->dump_block(this->post_);
        }
      ast_dump_context->unindent();
 
      ast_dump_context->print_indent();
      ast_dump_context->ostream() << "}";
    }
 
  ast_dump_context->ostream() << std::endl;
}
 
// Make a for statement.
 
For_statement*
Statement::make_for_statement(Block* init, Expression* cond, Block* post,
                              Location location)
{
  return new For_statement(init, cond, post, location);
}
 
// Class For_range_statement.
 
// Traversal.
 
int
For_range_statement::do_traverse(Traverse* traverse)
{
  if (this->traverse_expression(traverse, &this->index_var_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  if (this->value_var_ != NULL)
    {
      if (this->traverse_expression(traverse, &this->value_var_)
          == TRAVERSE_EXIT)
        return TRAVERSE_EXIT;
    }
  if (this->traverse_expression(traverse, &this->range_) == TRAVERSE_EXIT)
    return TRAVERSE_EXIT;
  return this->statements_->traverse(traverse);
}
 
// Lower a for range statement.  For simplicity we lower this into a
// for statement, which will then be lowered in turn to goto
// statements.
 
Statement*
For_range_statement::do_lower(Gogo* gogo, Named_object*, Block* enclosing,
                              Statement_inserter*)
{
  Type* range_type = this->range_->type();
  if (range_type->points_to() != NULL
      && range_type->points_to()->array_type() != NULL
      && !range_type->points_to()->is_slice_type())
    range_type = range_type->points_to();
 
  Type* index_type;
  Type* value_type = NULL;
  if (range_type->array_type() != NULL)
    {
      index_type = Type::lookup_integer_type("int");
      value_type = range_type->array_type()->element_type();
    }
  else if (range_type->is_string_type())
    {
      index_type = Type::lookup_integer_type("int");
      value_type = Type::lookup_integer_type("int32");
    }
  else if (range_type->map_type() != NULL)
    {
      index_type = range_type->map_type()->key_type();
      value_type = range_type->map_type()->val_type();
    }
  else if (range_type->channel_type() != NULL)
    {
      index_type = range_type->channel_type()->element_type();
      if (this->value_var_ != NULL)
        {
          if (!this->value_var_->type()->is_error())
            this->report_error(_("too many variables for range clause "
                                 "with channel"));
          return Statement::make_error_statement(this->location());
        }
    }
  else
    {
      this->report_error(_("range clause must have "
                           "array, slice, string, map, or channel type"));
      return Statement::make_error_statement(this->location());
    }
 
  Location loc = this->location();
  Block* temp_block = new Block(enclosing, loc);
 
  Named_object* range_object = NULL;
  Temporary_statement* range_temp = NULL;
  Var_expression* ve = this->range_->var_expression();
  if (ve != NULL)
    range_object = ve->named_object();
  else
    {
      range_temp = Statement::make_temporary(NULL, this->range_, loc);
      temp_block->add_statement(range_temp);
      this->range_ = NULL;
    }
 
  Temporary_statement* index_temp = Statement::make_temporary(index_type,
                                                              NULL, loc);
  temp_block->add_statement(index_temp);
 
  Temporary_statement* value_temp = NULL;
  if (this->value_var_ != NULL)
    {
      value_temp = Statement::make_temporary(value_type, NULL, loc);
      temp_block->add_statement(value_temp);
    }
 
  Block* body = new Block(temp_block, loc);
 
  Block* init;
  Expression* cond;
  Block* iter_init;
  Block* post;
 
  // Arrange to do a loop appropriate for the type.  We will produce
  //   for INIT ; COND ; POST {
  //           ITER_INIT
  //           INDEX = INDEX_TEMP
  //           VALUE = VALUE_TEMP // If there is a value
  //           original statements
  //   }
 
  if (range_type->is_slice_type())
    this->lower_range_slice(gogo, temp_block, body, range_object, range_temp,
                            index_temp, value_temp, &init, &cond, &iter_init,
                            &post);
  else if (range_type->array_type() != NULL)
    this->lower_range_array(gogo, temp_block, body, range_object, range_temp,
                            index_temp, value_temp, &init, &cond, &iter_init,
                            &post);
  else if (range_type->is_string_type())
    this->lower_range_string(gogo, temp_block, body, range_object, range_temp,
                             index_temp, value_temp, &init, &cond, &iter_init,
                             &post);
  else if (range_type->map_type() != NULL)
    this->lower_range_map(gogo, temp_block, body, range_object, range_temp,
                          index_temp, value_temp, &init, &cond, &iter_init,
                          &post);
  else if (range_type->channel_type() != NULL)
    this->lower_range_channel(gogo, temp_block, body, range_object, range_temp,
                              index_temp, value_temp, &init, &cond, &iter_init,
                              &post);
  else
    go_unreachable();
 
  if (iter_init != NULL)
    body->add_statement(Statement::make_block_statement(iter_init, loc));
 
  Statement* assign;
  Expression* index_ref = Expression::make_temporary_reference(index_temp, loc);
  if (this->value_var_ == NULL)
    {
      assign = Statement::make_assignment(this->index_var_, index_ref, loc);
    }
  else
    {
      Expression_list* lhs = new Expression_list();
      lhs->push_back(this->index_var_);
      lhs->push_back(this->value_var_);
 
      Expression_list* rhs = new Expression_list();
      rhs->push_back(index_ref);
      rhs->push_back(Expression::make_temporary_reference(value_temp, loc));
 
      assign = Statement::make_tuple_assignment(lhs, rhs, loc);
    }
  body->add_statement(assign);
 
  body->add_statement(Statement::make_block_statement(this->statements_, loc));
 
  body->set_end_location(this->statements_->end_location());
 
  For_statement* loop = Statement::make_for_statement(init, cond, post,
                                                      this->location());
  loop->add_statements(body);
  loop->set_break_continue_labels(this->break_label_, this->continue_label_);
 
  temp_block->add_statement(loop);
 
  return Statement::make_block_statement(temp_block, loc);
}
 
// Return a reference to the range, which may be in RANGE_OBJECT or in
// RANGE_TEMP.
 
Expression*
For_range_statement::make_range_ref(Named_object* range_object,
                                    Temporary_statement* range_temp,
                                    Location loc)
{
  if (range_object != NULL)
    return Expression::make_var_reference(range_object, loc);
  else
    return Expression::make_temporary_reference(range_temp, loc);
}
 
// Return a call to the predeclared function FUNCNAME passing a
// reference to the temporary variable ARG.
 
Expression*
For_range_statement::call_builtin(Gogo* gogo, const char* funcname,
                                  Expression* arg,
                                  Location loc)
{
  Named_object* no = gogo->lookup_global(funcname);
  go_assert(no != NULL && no->is_function_declaration());
  Expression* func = Expression::make_func_reference(no, NULL, loc);
  Expression_list* params = new Expression_list();
  params->push_back(arg);
  return Expression::make_call(func, params, false, loc);
}
 
// Lower a for range over an array.
 
void
For_range_statement::lower_range_array(Gogo* gogo,
                                       Block* enclosing,
                                       Block* body_block,
                                       Named_object* range_object,
                                       Temporary_statement* range_temp,
                                       Temporary_statement* index_temp,
                                       Temporary_statement* value_temp,
                                       Block** pinit,
                                       Expression** pcond,
                                       Block** piter_init,
                                       Block** ppost)
{
  Location loc = this->location();
 
  // The loop we generate:
  //   len_temp := len(range)
  //   for index_temp = 0; index_temp < len_temp; index_temp++ {
  //           value_temp = range[index_temp]
  //           index = index_temp
  //           value = value_temp
  //           original body
  //   }
 
  // Set *PINIT to
  //   var len_temp int
  //   len_temp = len(range)
  //   index_temp = 0
 
  Block* init = new Block(enclosing, loc);
 
  Expression* ref = this->make_range_ref(range_object, range_temp, loc);
  Expression* len_call = this->call_builtin(gogo, "len", ref, loc);
  Temporary_statement* len_temp = Statement::make_temporary(index_temp->type(),
                                                            len_call, loc);
  init->add_statement(len_temp);
 
  mpz_t zval;
  mpz_init_set_ui(zval, 0UL);
  Expression* zexpr = Expression::make_integer(&zval, NULL, loc);
  mpz_clear(zval);
 
  Temporary_reference_expression* tref =
    Expression::make_temporary_reference(index_temp, loc);
  tref->set_is_lvalue();
  Statement* s = Statement::make_assignment(tref, zexpr, loc);
  init->add_statement(s);
 
  *pinit = init;
 
  // Set *PCOND to
  //   index_temp < len_temp
 
  ref = Expression::make_temporary_reference(index_temp, loc);
  Expression* ref2 = Expression::make_temporary_reference(len_temp, loc);
  Expression* lt = Expression::make_binary(OPERATOR_LT, ref, ref2, loc);
 
  *pcond = lt;
 
  // Set *PITER_INIT to
  //   value_temp = range[index_temp]
 
  Block* iter_init = NULL;
  if (value_temp != NULL)
    {
      iter_init = new Block(body_block, loc);
 
      ref = this->make_range_ref(range_object, range_temp, loc);
      Expression* ref2 = Expression::make_temporary_reference(index_temp, loc);
      Expression* index = Expression::make_index(ref, ref2, NULL, loc);
 
      tref = Expression::make_temporary_reference(value_temp, loc);
      tref->set_is_lvalue();
      s = Statement::make_assignment(tref, index, loc);
 
      iter_init->add_statement(s);
    }
  *piter_init = iter_init;
 
  // Set *PPOST to
  //   index_temp++
 
  Block* post = new Block(enclosing, loc);
  tref = Expression::make_temporary_reference(index_temp, loc);
  tref->set_is_lvalue();
  s = Statement::make_inc_statement(tref);
  post->add_statement(s);
  *ppost = post;
}
 
// Lower a for range over a slice.
 
void
For_range_statement::lower_range_slice(Gogo* gogo,
                                       Block* enclosing,
                                       Block* body_block,
                                       Named_object* range_object,
                                       Temporary_statement* range_temp,
                                       Temporary_statement* index_temp,
                                       Temporary_statement* value_temp,
                                       Block** pinit,
                                       Expression** pcond,
                                       Block** piter_init,
                                       Block** ppost)
{
  Location loc = this->location();
 
  // The loop we generate:
  //   for_temp := range
  //   len_temp := len(for_temp)
  //   for index_temp = 0; index_temp < len_temp; index_temp++ {
  //           value_temp = for_temp[index_temp]
  //           index = index_temp
  //           value = value_temp
  //           original body
  //   }
  //
  // Using for_temp means that we don't need to check bounds when
  // fetching range_temp[index_temp].
 
  // Set *PINIT to
  //   range_temp := range
  //   var len_temp int
  //   len_temp = len(range_temp)
  //   index_temp = 0
 
  Block* init = new Block(enclosing, loc);
 
  Expression* ref = this->make_range_ref(range_object, range_temp, loc);
  Temporary_statement* for_temp = Statement::make_temporary(NULL, ref, loc);
  init->add_statement(for_temp);
 
  ref = Expression::make_temporary_reference(for_temp, loc);
  Expression* len_call = this->call_builtin(gogo, "len", ref, loc);
  Temporary_statement* len_temp = Statement::make_temporary(index_temp->type(),
                                                            len_call, loc);
  init->add_statement(len_temp);
 
  mpz_t zval;
  mpz_init_set_ui(zval, 0UL);
  Expression* zexpr = Expression::make_integer(&zval, NULL, loc);
  mpz_clear(zval);
 
  Temporary_reference_expression* tref =
    Expression::make_temporary_reference(index_temp, loc);
  tref->set_is_lvalue();
  Statement* s = Statement::make_assignment(tref, zexpr, loc);
  init->add_statement(s);
 
  *pinit = init;
 
  // Set *PCOND to
  //   index_temp < len_temp
 
  ref = Expression::make_temporary_reference(index_temp, loc);
  Expression* ref2 = Expression::make_temporary_reference(len_temp, loc);
  Expression* lt = Expression::make_binary(OPERATOR_LT, ref, ref2, loc);
 
  *pcond = lt;
 
  // Set *PITER_INIT to
  //   value_temp = range[index_temp]
 
  Block* iter_init = NULL;
  if (value_temp != NULL)
    {
      iter_init = new Block(body_block, loc);
 
      ref = Expression::make_temporary_reference(for_temp, loc);
      Expression* ref2 = Expression::make_temporary_reference(index_temp, loc);
      Expression* index = Expression::make_index(ref, ref2, NULL, loc);
 
      tref = Expression::make_temporary_reference(value_temp, loc);
      tref->set_is_lvalue();
      s = Statement::make_assignment(tref, index, loc);
 
      iter_init->add_statement(s);
    }
  *piter_init = iter_init;
 
  // Set *PPOST to
  //   index_temp++
 
  Block* post = new Block(enclosing, loc);
  tref = Expression::make_temporary_reference(index_temp, loc);
  tref->set_is_lvalue();
  s = Statement::make_inc_statement(tref);
  post->add_statement(s);
  *ppost = post;
}
 
// Lower a for range over a string.
 
void
For_range_statement::lower_range_string(Gogo*,
                                        Block* enclosing,
                                        Block* body_block,
                                        Named_object* range_object,
                                        Temporary_statement* range_temp,
                                        Temporary_statement* index_temp,
                                        Temporary_statement* value_temp,
                                        Block** pinit,
                                        Expression** pcond,
                                        Block** piter_init,
                                        Block** ppost)
{
  Location loc = this->location();
 
  // The loop we generate:
  //   var next_index_temp int
  //   for index_temp = 0; ; index_temp = next_index_temp {
  //           next_index_temp, value_temp = stringiter2(range, index_temp)
  //           if next_index_temp == 0 {
  //                   break
  //           }
  //           index = index_temp
  //           value = value_temp
  //           original body
  //   }
 
  // Set *PINIT to
  //   var next_index_temp int
  //   index_temp = 0
 
  Block* init = new Block(enclosing, loc);
 
  Temporary_statement* next_index_temp =
    Statement::make_temporary(index_temp->type(), NULL, loc);
  init->add_statement(next_index_temp);
 
  mpz_t zval;
  mpz_init_set_ui(zval, 0UL);
  Expression* zexpr = Expression::make_integer(&zval, NULL, loc);
 
  Temporary_reference_expression* ref =
    Expression::make_temporary_reference(index_temp, loc);
  ref->set_is_lvalue();
  Statement* s = Statement::make_assignment(ref, zexpr, loc);
 
  init->add_statement(s);
  *pinit = init;
 
  // The loop has no condition.
 
  *pcond = NULL;
 
  // Set *PITER_INIT to
  //   next_index_temp = runtime.stringiter(range, index_temp)
  // or
  //   next_index_temp, value_temp = runtime.stringiter2(range, index_temp)
  // followed by
  //   if next_index_temp == 0 {
  //           break
  //   }
 
  Block* iter_init = new Block(body_block, loc);
 
  Expression* p1 = this->make_range_ref(range_object, range_temp, loc);
  Expression* p2 = Expression::make_temporary_reference(index_temp, loc);
  Call_expression* call = Runtime::make_call((value_temp == NULL
                                              ? Runtime::STRINGITER
                                              : Runtime::STRINGITER2),
                                             loc, 2, p1, p2);
 
  if (value_temp == NULL)
    {
      ref = Expression::make_temporary_reference(next_index_temp, loc);
      ref->set_is_lvalue();
      s = Statement::make_assignment(ref, call, loc);
    }
  else
    {
      Expression_list* lhs = new Expression_list();
 
      ref = Expression::make_temporary_reference(next_index_temp, loc);
      ref->set_is_lvalue();
      lhs->push_back(ref);
 
      ref = Expression::make_temporary_reference(value_temp, loc);
      ref->set_is_lvalue();
      lhs->push_back(ref);
 
      Expression_list* rhs = new Expression_list();
      rhs->push_back(Expression::make_call_result(call, 0));
      rhs->push_back(Expression::make_call_result(call, 1));
 
      s = Statement::make_tuple_assignment(lhs, rhs, loc);
    }
  iter_init->add_statement(s);
 
  ref = Expression::make_temporary_reference(next_index_temp, loc);
  zexpr = Expression::make_integer(&zval, NULL, loc);
  mpz_clear(zval);
  Expression* equals = Expression::make_binary(OPERATOR_EQEQ, ref, zexpr, loc);
 
  Block* then_block = new Block(iter_init, loc);
  s = Statement::make_break_statement(this->break_label(), loc);
  then_block->add_statement(s);
 
  s = Statement::make_if_statement(equals, then_block, NULL, loc);
  iter_init->add_statement(s);
 
  *piter_init = iter_init;
 
  // Set *PPOST to
  //   index_temp = next_index_temp
 
  Block* post = new Block(enclosing, loc);
 
  Temporary_reference_expression* lhs =
    Expression::make_temporary_reference(index_temp, loc);
  lhs->set_is_lvalue();
  Expression* rhs = Expression::make_temporary_reference(next_index_temp, loc);
  s = Statement::make_assignment(lhs, rhs, loc);
 
  post->add_statement(s);
  *ppost = post;
}
 
// Lower a for range over a map.
 
void
For_range_statement::lower_range_map(Gogo*,
                                     Block* enclosing,
                                     Block* body_block,
                                     Named_object* range_object,
                                     Temporary_statement* range_temp,
                                     Temporary_statement* index_temp,
                                     Temporary_statement* value_temp,
                                     Block** pinit,
                                     Expression** pcond,
                                     Block** piter_init,
                                     Block** ppost)
{
  Location loc = this->location();
 
  // The runtime uses a struct to handle ranges over a map.  The
  // struct is four pointers long.  The first pointer is NULL when we
  // have completed the iteration.
 
  // The loop we generate:
  //   var hiter map_iteration_struct
  //   for mapiterinit(range, &hiter); hiter[0] != nil; mapiternext(&hiter) {
  //           mapiter2(hiter, &index_temp, &value_temp)
  //           index = index_temp
  //           value = value_temp
  //           original body
  //   }
 
  // Set *PINIT to
  //   var hiter map_iteration_struct
  //   runtime.mapiterinit(range, &hiter)
 
  Block* init = new Block(enclosing, loc);
 
  Type* map_iteration_type = Runtime::map_iteration_type();
  Temporary_statement* hiter = Statement::make_temporary(map_iteration_type,
                                                         NULL, loc);
  init->add_statement(hiter);
 
  Expression* p1 = this->make_range_ref(range_object, range_temp, loc);
  Expression* ref = Expression::make_temporary_reference(hiter, loc);
  Expression* p2 = Expression::make_unary(OPERATOR_AND, ref, loc);
  Expression* call = Runtime::make_call(Runtime::MAPITERINIT, loc, 2, p1, p2);
  init->add_statement(Statement::make_statement(call, true));
 
  *pinit = init;
 
  // Set *PCOND to
  //   hiter[0] != nil
 
  ref = Expression::make_temporary_reference(hiter, loc);
 
  mpz_t zval;
  mpz_init_set_ui(zval, 0UL);
  Expression* zexpr = Expression::make_integer(&zval, NULL, loc);
  mpz_clear(zval);
 
  Expression* index = Expression::make_index(ref, zexpr, NULL, loc);
 
  Expression* ne = Expression::make_binary(OPERATOR_NOTEQ, index,
                                           Expression::make_nil(loc),
                                           loc);
 
  *pcond = ne;
 
  // Set *PITER_INIT to
  //   mapiter1(hiter, &index_temp)
  // or
  //   mapiter2(hiter, &index_temp, &value_temp)
 
  Block* iter_init = new Block(body_block, loc);
 
  ref = Expression::make_temporary_reference(hiter, loc);
  p1 = Expression::make_unary(OPERATOR_AND, ref, loc);
  ref = Expression::make_temporary_reference(index_temp, loc);
  p2 = Expression::make_unary(OPERATOR_AND, ref, loc);
  if (value_temp == NULL)
    call = Runtime::make_call(Runtime::MAPITER1, loc, 2, p1, p2);
  else
    {
      ref = Expression::make_temporary_reference(value_temp, loc);
      Expression* p3 = Expression::make_unary(OPERATOR_AND, ref, loc);
      call = Runtime::make_call(Runtime::MAPITER2, loc, 3, p1, p2, p3);
    }
  iter_init->add_statement(Statement::make_statement(call, true));
 
  *piter_init = iter_init;
 
  // Set *PPOST to
  //   mapiternext(&hiter)
 
  Block* post = new Block(enclosing, loc);
 
  ref = Expression::make_temporary_reference(hiter, loc);
  p1 = Expression::make_unary(OPERATOR_AND, ref, loc);
  call = Runtime::make_call(Runtime::MAPITERNEXT, loc, 1, p1);
  post->add_statement(Statement::make_statement(call, true));
 
  *ppost = post;
}
 
// Lower a for range over a channel.
 
void
For_range_statement::lower_range_channel(Gogo*,
                                         Block*,
                                         Block* body_block,
                                         Named_object* range_object,
                                         Temporary_statement* range_temp,
                                         Temporary_statement* index_temp,
                                         Temporary_statement* value_temp,
                                         Block** pinit,
                                         Expression** pcond,
                                         Block** piter_init,
                                         Block** ppost)
{
  go_assert(value_temp == NULL);
 
  Location loc = this->location();
 
  // The loop we generate:
  //   for {
  //           index_temp, ok_temp = <-range
  //           if !ok_temp {
  //                   break
  //           }
  //           index = index_temp
  //           original body
  //   }
 
  // We have no initialization code, no condition, and no post code.
 
  *pinit = NULL;
  *pcond = NULL;
  *ppost = NULL;
 
  // Set *PITER_INIT to
  //   index_temp, ok_temp = <-range
  //   if !ok_temp {
  //           break
  //   }
 
  Block* iter_init = new Block(body_block, loc);
 
  Temporary_statement* ok_temp =
    Statement::make_temporary(Type::lookup_bool_type(), NULL, loc);
  iter_init->add_statement(ok_temp);
 
  Expression* cref = this->make_range_ref(range_object, range_temp, loc);
  Temporary_reference_expression* iref =
    Expression::make_temporary_reference(index_temp, loc);
  iref->set_is_lvalue();
  Temporary_reference_expression* oref =
    Expression::make_temporary_reference(ok_temp, loc);
  oref->set_is_lvalue();
  Statement* s = Statement::make_tuple_receive_assignment(iref, oref, cref,
                                                          loc);
  iter_init->add_statement(s);
 
  Block* then_block = new Block(iter_init, loc);
  s = Statement::make_break_statement(this->break_label(), loc);
  then_block->add_statement(s);
 
  oref = Expression::make_temporary_reference(ok_temp, loc);
  Expression* cond = Expression::make_unary(OPERATOR_NOT, oref, loc);
  s = Statement::make_if_statement(cond, then_block, NULL, loc);
  iter_init->add_statement(s);
 
  *piter_init = iter_init;
}
 
// Return the break LABEL_EXPR.
 
Unnamed_label*
For_range_statement::break_label()
{
  if (this->break_label_ == NULL)
    this->break_label_ = new Unnamed_label(this->location());
  return this->break_label_;
}
 
// Return the continue LABEL_EXPR.
 
Unnamed_label*
For_range_statement::continue_label()
{
  if (this->continue_label_ == NULL)
    this->continue_label_ = new Unnamed_label(this->location());
  return this->continue_label_;
}
 
// Dump the AST representation for a for range statement.
 
void
For_range_statement::do_dump_statement(Ast_dump_context* ast_dump_context) const
{
 
  ast_dump_context->print_indent();
  ast_dump_context->ostream() << "for ";
  ast_dump_context->dump_expression(this->index_var_);
  if (this->value_var_ != NULL)
    {
      ast_dump_context->ostream() << ", ";
      ast_dump_context->dump_expression(this->value_var_);
    }
 
  ast_dump_context->ostream() << " = range ";
  ast_dump_context->dump_expression(this->range_);
  if (ast_dump_context->dump_subblocks())
    {
      ast_dump_context->ostream() << " {" << std::endl;
 
      ast_dump_context->indent();
 
      ast_dump_context->dump_block(this->statements_);
 
      ast_dump_context->unindent();
      ast_dump_context->print_indent();
      ast_dump_context->ostream() << "}";
    }
  ast_dump_context->ostream() << std::endl;
}
 
// Make a for statement with a range clause.
 
For_range_statement*
Statement::make_for_range_statement(Expression* index_var,
                                    Expression* value_var,
                                    Expression* range,
                                    Location location)
{
  return new For_range_statement(index_var, value_var, range, location);
}
Browse

Tools

Subversion Repositories openrisc

[/] [openrisc/] [trunk/] [gnu-dev/] [or1k-gcc/] [gcc/] [go/] [gofrontend/] [statements.cc] - Blame information for rev 715

Line No.	Rev	Author	Line
1	714	jeremybenn	`// statements.cc -- Go frontend statements.`
2
3			`// Copyright 2009 The Go Authors. All rights reserved.`
4			`// Use of this source code is governed by a BSD-style`
5			`// license that can be found in the LICENSE file.`
6
7			`#include "go-system.h"`
8
9			`#include <gmp.h>`
10
11			`#include "go-c.h"`
12			`#include "types.h"`
13			`#include "expressions.h"`
14			`#include "gogo.h"`
15			`#include "runtime.h"`
16			`#include "backend.h"`
17			`#include "statements.h"`
18			`#include "ast-dump.h"`
19
20			`// Class Statement.`
21
22			`Statement::Statement(Statement_classification classification,`
23			`Location location)`
24			`: classification_(classification), location_(location)`
25			`{`
26			`}`
27
28			`Statement::~Statement()`
29			`{`
30			`}`
31
32			`// Traverse the tree. The work of walking the components is handled`
33			`// by the subclasses.`
34
35			`int`
36			`Statement::traverse(Block* block, size_t* pindex, Traverse* traverse)`
37			`{`
38			`if (this->classification_ == STATEMENT_ERROR)`
39			`return TRAVERSE_CONTINUE;`
40