/* vim: set ts=4 sw=4 tw=99 et:

  *

  * Copyright (C) 2012 David Anderson

  *

  * This file is part of JITCraft.

  *

  * JITCraft is free software: you can redistribute it and/or modify it under

  * the terms of the GNU General Public License as published by the Free

  * Software Foundation, either version 3 of the License, or (at your option)

  * any later version.

  * 

  * Foobar is distributed in the hope that it will be useful, but WITHOUT ANY

  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

  * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

  *

  * You should have received a copy of the GNU General Public License along with

  * JITCraft. If not, see http://www.gnu.org/licenses/.

  */

 #include "AST.h"

 #include "HIR.h"

 #include "BytecodeEmitter.h"

 #include "../Strings.h"

 #include "../Heap-inl.h"

 #include "../Handles-inl.h"

 using namespace ke;

 class HIRTranslator : public HIRVisitor

 {

   public:

     HIRTranslator(Zone *zone, BytecodeEmitter &emitter)

       : zone_(zone),

         emitter_(emitter)

     {

     }

     void translate(HIR *hir) {

         hir->accept(this);

     }

     void visitBoolean(HBoolean *hir);

     void visitFloat(HFloat *hir);

     void visitInteger(HInteger *hir);

     void visitArray(HArray *hir);

     void visitCall(HCall *call);

     void visitStore(HStore *store);

     void visitUnary(HUnary *unary);

     void visitBinary(HBinary *binary);

     void visitLocal(HLocal *local);

     void visitGlobal(HGlobal *global);

     void visitConvert(HConvert *convert);

     void visitIndex(HIndex *index);

     void visitField(HField *field);

     void visitPostIncDec(HPostIncDec *hir);

     void visitToRef(HToRef *hir);

     void visitNewDependentArray(HNewDependentArray *hir);

     void visitImport(HImport *import);

   private:

     Zone *zone_;

     BytecodeEmitter &emitter_;

 };

 #define __ emitter_.

 void

 HIRTranslator::visitBoolean(HBoolean *hir)

 {

     if (hir->value())

         __ true_();

     else

         __ false_();

 }

 void

 HIRTranslator::visitFloat(HFloat *hir)

 {

     __ float_(hir->value());

 }

 void

 HIRTranslator::visitInteger(HInteger *hir)

 {

     __ int_(hir->value());

 }

 void

 HIRTranslator::visitArray(HArray *hir)

 {

     __ object(hir->array());

     if (hir->duplicate())

         __ duparray(); 

 }

 void

 HIRTranslator::visitCall(HCall *call)

 {

     Local<FunctionType> fun(zone_, FunctionType::cast(call->callee()->type()));

     assert(call->args()->length() == fun->parameters()->length());

     __ note_position(call->node()->pos());

     unsigned argc = call->args()->length();

     for (size_t i = call->args()->length() - 1; i < call->args()->length(); i--) {

         HIR *hir = call->args()->at(i);

         hir->accept(this);

     }

     call->callee()->accept(this);

     __ call(fun, argc);

 }

 void

 HGlobal::bind(HIRVisitor *visitor, BytecodeEmitter &emitter_)

 {

 }

 void

 HGlobal::hold(BytecodeEmitter &emitter_)

 {

 }

 void

 HGlobal::swapAndPick(BytecodeEmitter &emitter_)

 {

     __ dup();

 }

 void

 HGlobal::store(BytecodeEmitter &emitter_)

 {

     __ setglobal(sym_);

 }

 void

 HGlobal::load(BytecodeEmitter &emitter_)

 {

     __ getglobal(sym_);

 }

 bool

 HGlobal::ref(BytecodeEmitter &emitter_)

 {

     return false;

 }

 static inline Type *

 MaybeDereference(Type *type)

 {

     if (type->isReference())

         return ReferenceType::cast(type)->contained();

     return type;

 }

 HLocal::HLocal(AstNode *node, VariableSymbol *sym)

   : HLValue(node, MaybeDereference(sym->type())),

     sym_(sym)

 {

 }

 void

 HLocal::bind(HIRVisitor *visitor, BytecodeEmitter &emitter_)

 {

     if (sym_->type()->isReference()) {

         if (sym_->storage() == VariableSymbol::Local)

             __ getlocal(sym_);

         else if (sym_->storage() == VariableSymbol::Arg)

             __ getarg(sym_);

         else

             assert(false);

     }

 }

 void

 HLocal::hold(BytecodeEmitter &emitter_)

 {

     if (sym_->type()->isReference())

         __ dup();

 }

 void

 HLocal::swapAndPick(BytecodeEmitter &emitter_)

 {

     if (sym_->type()->isReference()) {

         __ swap();

         __ pick2();

     } else {

         __ dup();

     }

 }

 void

 HLocal::store(BytecodeEmitter &emitter_)

 {

     if (sym_->type()->isReference()) {

         __ setref(ReferenceType::cast(sym_->type())->contained());

     } else {

         if (sym_->storage() == VariableSymbol::Local)

             __ setlocal(sym_);

         else if (sym_->storage() == VariableSymbol::Arg)

             __ setarg(sym_);

         else

             assert(false);

     }

 }

 void

 HLocal::load(BytecodeEmitter &emitter_)

 {

     if (sym_->type()->isReference())

         __ getref(ReferenceType::cast(sym_->type())->contained());

     else if (sym_->storage() == VariableSymbol::Local)

         __ getlocal(sym_);

     else if (sym_->storage() == VariableSymbol::Arg)

         __ getarg(sym_);

     else

         assert(false);

 }

 bool

 HLocal::ref(BytecodeEmitter &emitter_)

 {

     if (sym_->storage() == VariableSymbol::Heap) {

         // Closing over a reference should never pass SemA.

         assert(!sym_->type()->isReference());

         // We don't support direct references to heap-allocated locals. This

         // should probably be banned in SemA? But for now, just assert. :TODO:

         assert(false);

         return false;

     }

     if (sym_->type()->isReference()) {

         if (sym_->storage() == VariableSymbol::Local)

             __ getlocal(sym_);

         else if (sym_->storage() == VariableSymbol::Arg)

             __ getarg(sym_);

         else

             assert(false);

     } else {

         __ newref(sym_);

     }

     return true;

 }

 void

 HIndex::bind(HIRVisitor *visitor, BytecodeEmitter &emitter_)

 {

     __ note_position(node()->pos());

     left_->accept(visitor);

     right_->accept(visitor);

 }

 void

 HIndex::hold(BytecodeEmitter &emitter_)

 {

     __ dup2();

 }

 void

 HIndex::swapAndPick(BytecodeEmitter &emitter_)

 {

     // BASE INDEX VALUE

     __ roll3();         // INDEX VALUE BASE

     __ roll3();         // VALUE BASE INDEX

     __ pick3();         // VALUE BASE INDEX VALUE

 }

 void

 HIndex::store(BytecodeEmitter &emitter_)

 {

     __ setelem(type());

 }

 void

 HIndex::load(BytecodeEmitter &emitter_)

 {

     __ getelem(type());

 }

 bool

 HIndex::ref(BytecodeEmitter &emitter_)

 {

     return false;

 }

 void

 HField::bind(HIRVisitor *visitor, BytecodeEmitter &emitter_)

 {

     __ note_position(node()->pos());

     base_->accept(visitor);

 }

 void

 HField::hold(BytecodeEmitter &emitter_)

 {

     __ dup();

 }

 void

 HField::swapAndPick(BytecodeEmitter &emitter_)

 {

     // BASE VALUE

     __ swap();      // VALUE BASE

     __ pick2();     // VALUE BASE VALUE

 }

 void

 HField::store(BytecodeEmitter &emitter_)

 {

     __ setfield(index_, type());

 }

 void

 HField::load(BytecodeEmitter &emitter_)

 {

     __ getfield(index_, type());

 }

 bool

 HField::ref(BytecodeEmitter &emitter_)

 {

     return false;

 }

 void

 HIRTranslator::visitLocal(HLocal *local)

 {

     local->bind(this, emitter_);

     local->load(emitter_);

 }

 void

 HIRTranslator::visitIndex(HIndex *index)

 {

     index->bind(this, emitter_);

     index->load(emitter_);

 }

 void

 HIRTranslator::visitField(HField *field)

 {

     field->bind(this, emitter_);

     field->load(emitter_);

 }

 void

 HIRTranslator::visitGlobal(HGlobal *global)

 {

     global->bind(this, emitter_);

     global->load(emitter_);

 }

 static inline Opcode

 TokenToOpcode(TokenKind kind)

 {

     switch (kind) {

       case TOK_PLUS:

       case TOK_ASSIGN_ADD:

       case TOK_INCREMENT:

         return OP_ADD;

       case TOK_MINUS:

       case TOK_ASSIGN_SUB:

       case TOK_DECREMENT:

         return OP_SUB;

       case TOK_STAR:

       case TOK_ASSIGN_MUL:

         return OP_MUL;

       case TOK_SLASH:

       case TOK_ASSIGN_DIV:

         return OP_DIV;

       case TOK_PERCENT:

       case TOK_ASSIGN_MOD:

         return OP_MOD;

       case TOK_SHR:

       case TOK_ASSIGN_SHR:

         return OP_SHR;

       case TOK_USHR:

       case TOK_ASSIGN_USHR:

         return OP_USHR;

       case TOK_SHL:

       case TOK_ASSIGN_SHL:

         return OP_SHL;

       case TOK_BITAND:

       case TOK_ASSIGN_BITAND:

         return OP_BITAND;

       case TOK_BITOR:

       case TOK_ASSIGN_BITOR:

         return OP_BITOR;

       case TOK_GT:

           return OP_GT;

       case TOK_GE:

           return OP_GE;

       case TOK_LT:

           return OP_LT;

       case TOK_LE:

           return OP_LE;

       case TOK_EQUALS:

           return OP_EQ;

       case TOK_NOTEQUALS:

           return OP_NE;

       default:

         assert(kind == TOK_BITXOR);

         return OP_BITXOR;

     }

 }

 void

 HIRTranslator::visitStore(HStore *store)

 {

     store->lval()->bind(this, emitter_);

     if (store->kind() == TOK_ASSIGN) {

         // Fixed-length arrays and structs have copy semantics.

         Local<Type> type(zone_, store->lval()->type());

         if (type->isArray() && ArrayType::cast(type)->isFixedLength()) {

             store->lval()->load(emitter_);

             store->rval()->accept(this);

             __ copyarray();

         } else if (type->isStruct()) {

             store->lval()->load(emitter_);

             store->rval()->accept(this);

             __ copystruct();

         } else {

             store->rval()->accept(this);

             store->lval()->store(emitter_);

         }

     } else {

         // Advanced case: dup the ref, load, op, store.

         store->lval()->hold(emitter_);

         store->lval()->load(emitter_);

         store->rval()->accept(this);

         __ binary(TokenToOpcode(store->kind()));

         store->lval()->store(emitter_);

     }

 }

 void

 HIRTranslator::visitPostIncDec(HPostIncDec *hir)

 {

     hir->lval()->bind(this, emitter_);

     hir->lval()->hold(emitter_);

     hir->lval()->load(emitter_);

     hir->lval()->swapAndPick(emitter_);

     hir->rval()->accept(this);

     __ binary(TokenToOpcode(hir->kind()));

     hir->lval()->store(emitter_);

     __ pop();

 }

 void

 HIRTranslator::visitBinary(HBinary *binary)

 {

     binary->left()->accept(this);

     binary->right()->accept(this);

     // We define three sets of math operators:

     //

     //  (1) ADD, SUB, MUL, DIV, MOD, which take two equal types and produce

     //      the same type.

     //  (2) LT, LE, GE, GT, EQ, NE, which take two equal types and produce

     //      a boolean.

     //  (3) SHL, SR, USHR, BITAND, BITOR, BITXOR, BITNOT, which take two

     //      equal integer types and return that integer type.

     //

     Opcode op = TokenToOpcode(binary->token());

     if (op == OP_DIV)

         __ note_position(binary->node()->pos());

     if (op >= OP_ADD && op <= OP_NE) {

         if (binary->type()->isFloat())

             op = Opcode(int(op) + int(OP_ADD_F - OP_ADD));

     } else {

         assert(binary->type()->isInt32());

     }

     __ binary(op);

 }

 void

 HIRTranslator::visitUnary(HUnary *unary)

 {

     unary->in()->accept(this);

     switch (unary->token()) {

       case TOK_TILDE:

         __ bitnot();

         break;

       case TOK_MINUS:

         if (unary->type()->isFloat())

             __ neg_f();

         else

             __ neg();

         break;

       default:

         assert(unary->token() == TOK_LABEL);

         __ bitcast(unary->type());

         break;

     }

 }

 void

 HIRTranslator::visitConvert(HConvert *convert)

 {

     convert->in()->accept(this);

     __ unary(convert->opcode());

 }

 void

 HIRTranslator::visitToRef(HToRef *hir)

 {

     HIR *in = hir->in();

     // See if we can bind a direct reference.

     if (in->isLValue() && in->toLValue()->ref(emitter_))

         return;

     // No, we can't. So instead we create an on-stack reference.

     in->accept(this);

     __ slotref_lifo();

 }

 void

 HIRTranslator::visitNewDependentArray(HNewDependentArray *hir)

 {

     hir->base()->accept(this);

     hir->index()->accept(this);

     Local<ArrayType> type(zone_, ArrayType::cast(hir->type()));

     // The dependent map is used for the opcode, but is not attached.

     Local<ArrayMap> map(zone_, ArrayMap::NewDependent(zone_, type));

     if (!map)

         return;

     __ deparray(map);

 }

 void

 HIRTranslator::visitImport(HImport *import)

 {

     __ import(import->index());

 }

 void

 ke::EmitHIR(Zone *zone, BytecodeEmitter &emitter, HIR *hir)

 {

     HIRTranslator translator(zone, emitter);

     translator.translate(hir);

 }