/*
   +----------------------------------------------------------------------+
   | HipHop for PHP                                                       |
   +----------------------------------------------------------------------+
   | Copyright (c) 2010-2013 Facebook, Inc. (http://www.facebook.com)     |
   +----------------------------------------------------------------------+
   | This source file is subject to version 3.01 of the PHP license,      |
   | that is bundled with this package in the file LICENSE, and is        |
   | available through the world-wide-web at the following url:           |
   | http://www.php.net/license/3_01.txt                                  |
   | If you did not receive a copy of the PHP license and are unable to   |
   | obtain it through the world-wide-web, please send a note to          |
   | license@php.net so we can mail you a copy immediately.               |
   +----------------------------------------------------------------------+
*/

#include "hphp/compiler/expression/yield_expression.h"
#include "hphp/compiler/analysis/function_scope.h"

using namespace HPHP;

///////////////////////////////////////////////////////////////////////////////
// constructors/destructors

YieldExpression::YieldExpression
(EXPRESSION_CONSTRUCTOR_PARAMETERS,
 ExpressionPtr keyExp, ExpressionPtr valExp)
  : Expression(EXPRESSION_CONSTRUCTOR_PARAMETER_VALUES(YieldExpression)),
    m_keyExp(keyExp), m_valExp(valExp), m_label(-1) {
}

ExpressionPtr YieldExpression::clone() {
  YieldExpressionPtr exp(new YieldExpression(*this));
  Expression::deepCopy(exp);
  exp->m_keyExp = Clone(m_keyExp);
  exp->m_valExp = Clone(m_valExp);
  exp->m_label = m_label;
  return exp;
}

///////////////////////////////////////////////////////////////////////////////
// parser functions


///////////////////////////////////////////////////////////////////////////////
// static analysis functions

void YieldExpression::analyzeProgram(AnalysisResultPtr ar) {
  assert(getFunctionScope() && getFunctionScope()->isGenerator());
  if (m_keyExp) {
    m_keyExp->analyzeProgram(ar);
  }
  m_valExp->setChildOfYield();
  m_valExp->analyzeProgram(ar);
  if (m_label == -1) {
    setLabel(getFunctionScope()->allocYieldLabel());
  }
}

ConstructPtr YieldExpression::getNthKid(int n) const {
  switch (n) {
    case 0:
      return m_keyExp;
    case 1:
      return m_valExp;
    default:
      assert(false);
      break;
  }
  return ConstructPtr();
}

int YieldExpression::getKidCount() const {
  return 2;
}

void YieldExpression::setNthKid(int n, ConstructPtr cp) {
  switch (n) {
    case 0:
      m_keyExp = boost::dynamic_pointer_cast<Expression>(cp);
      break;
    case 1:
      m_valExp = boost::dynamic_pointer_cast<Expression>(cp);
      break;
    default:
      assert(false);
      break;
  }
}

TypePtr YieldExpression::inferTypes(AnalysisResultPtr ar, TypePtr type,
                                    bool coerce) {
  if (m_keyExp) {
    m_keyExp->inferAndCheck(ar, Type::Some, false);
  }
  m_valExp->inferAndCheck(ar, Type::Some, false);
  return Type::Variant;
}


///////////////////////////////////////////////////////////////////////////////
// code generation functions

void YieldExpression::outputPHP(CodeGenerator &cg, AnalysisResultPtr ar) {
  cg_printf("yield ");
  if (m_keyExp) {
    m_keyExp->outputPHP(cg, ar);
    cg_printf(" => ");
  }
  m_valExp->outputPHP(cg, ar);
}