#include "qv4debugging_p.h"
#include "qv4engine_p.h"
#include "qv4jsir_p.h"
#include "qv4isel_p.h"
#include "qv4isel_util_p.h"
#include "qv4functionobject_p.h"

#include <QString>

#include <cassert>

namespace {
QTextStream qout(stderr, QIODevice::WriteOnly);
} // anonymous namespace

using namespace QQmlJS;
using namespace QQmlJS::V4IR;

EvalInstructionSelection::EvalInstructionSelection(VM::ExecutionEngine *engine, Module *module)
    : _engine(engine)
    , useFastLookups(true)
{
    assert(engine);
    assert(module);

    createFunctionMapping(0, module->rootFunction);
    foreach (V4IR::Function *f, module->functions) {
        assert(_irToVM.contains(f));
    }
}

EvalInstructionSelection::~EvalInstructionSelection()
{}

EvalISelFactory::~EvalISelFactory()
{}

VM::Function *EvalInstructionSelection::createFunctionMapping(VM::Function *outer, Function *irFunction)
{
    VM::Function *vmFunction = _engine->newFunction(irFunction->name ? *irFunction->name : QString());
    _irToVM.insert(irFunction, vmFunction);

    vmFunction->hasDirectEval = irFunction->hasDirectEval;
    vmFunction->usesArgumentsObject = irFunction->usesArgumentsObject;
    vmFunction->hasNestedFunctions = !irFunction->nestedFunctions.isEmpty();
    vmFunction->isStrict = irFunction->isStrict;
    vmFunction->outer = outer;
    vmFunction->isNamedExpression = irFunction->isNamedExpression;

    if (outer)
        outer->nestedFunctions.append(vmFunction);

    foreach (const QString *formal, irFunction->formals)
        if (formal)
            vmFunction->formals.append(_engine->newString(*formal));
    foreach (const QString *local, irFunction->locals)
        if (local)
            vmFunction->locals.append(_engine->newString(*local));

    foreach (V4IR::Function *function, irFunction->nestedFunctions)
        createFunctionMapping(vmFunction, function);

    if (_engine->debugger)
        _engine->debugger->mapFunction(vmFunction, irFunction);

    return vmFunction;
}

VM::Function *EvalInstructionSelection::vmFunction(Function *f) {
    VM::Function *function = _irToVM[f];
    if (!function->code)
        run(function, f);
    return function;
}

void InstructionSelection::visitMove(V4IR::Move *s)
{
    if (s->op == V4IR::OpInvalid) {
        if (V4IR::Name *n = s->target->asName()) {
            if (s->source->asTemp()) {
                setActivationProperty(s->source->asTemp(), *n->id);
                return;
            }
        } else if (V4IR::Temp *t = s->target->asTemp()) {
            if (V4IR::Name *n = s->source->asName()) {
                if (*n->id == QStringLiteral("this")) // TODO: `this' should be a builtin.
                    loadThisObject(t);
                else
                    getActivationProperty(n, t);
                return;
            } else if (V4IR::Const *c = s->source->asConst()) {
                loadConst(c, t);
                return;
            } else if (V4IR::Temp *t2 = s->source->asTemp()) {
                copyValue(t2, t);
                return;
            } else if (V4IR::String *str = s->source->asString()) {
                loadString(*str->value, t);
                return;
            } else if (V4IR::RegExp *re = s->source->asRegExp()) {
                loadRegexp(re, t);
                return;
            } else if (V4IR::Closure *clos = s->source->asClosure()) {
                initClosure(clos, t);
                return;
            } else if (V4IR::New *ctor = s->source->asNew()) {
                if (Name *func = ctor->base->asName()) {
                    constructActivationProperty(func, ctor->args, t);
                    return;
                } else if (V4IR::Member *member = ctor->base->asMember()) {
                    constructProperty(member->base->asTemp(), *member->name, ctor->args, t);
                    return;
                } else if (V4IR::Temp *value = ctor->base->asTemp()) {
                    constructValue(value, ctor->args, t);
                    return;
                }
            } else if (V4IR::Member *m = s->source->asMember()) {
                if (V4IR::Temp *base = m->base->asTemp()) {
                    getProperty(base, *m->name, t);
                    return;
                }
            } else if (V4IR::Subscript *ss = s->source->asSubscript()) {
                getElement(ss->base->asTemp(), ss->index->asTemp(), t);
                return;
            } else if (V4IR::Unop *u = s->source->asUnop()) {
                if (V4IR::Temp *e = u->expr->asTemp()) {
                    unop(u->op, e, t);
                    return;
                }
            } else if (V4IR::Binop *b = s->source->asBinop()) {
                if (b->left->asTemp() && b->right->asTemp()) {
                    binop(b->op, b->left->asTemp(), b->right->asTemp(), t);
                    return;
                }
            } else if (V4IR::Call *c = s->source->asCall()) {
                if (c->base->asName()) {
                    callBuiltin(c, t);
                    return;
                } else if (Member *member = c->base->asMember()) {
                    callProperty(member->base, *member->name, c->args, t);
                    return;
                } else if (Subscript *s = c->base->asSubscript()) {
                    callSubscript(s->base, s->index, c->args, t);
                    return;
                } else if (V4IR::Temp *value = c->base->asTemp()) {
                    callValue(value, c->args, t);
                    return;
                }
            }
        } else if (V4IR::Member *m = s->target->asMember()) {
            if (V4IR::Temp *base = m->base->asTemp()) {
                if (s->source->asTemp()) {
                    setProperty(s->source->asTemp(), base, *m->name);
                    return;
                }
            }
        } else if (V4IR::Subscript *ss = s->target->asSubscript()) {
            if (s->source->asTemp()) {
                setElement(s->source->asTemp(), ss->base->asTemp(), ss->index->asTemp());
                return;
            }
        }
    } else {
        // inplace assignment, e.g. x += 1, ++x, ...
        if (V4IR::Temp *t = s->target->asTemp()) {
            if (s->source->asTemp()) {
                binop(s->op, t, s->source->asTemp(), t);
                return;
            }
        } else if (V4IR::Name *n = s->target->asName()) {
            if (s->source->asTemp()) {
                inplaceNameOp(s->op, s->source->asTemp(), *n->id);
                return;
            }
        } else if (V4IR::Subscript *ss = s->target->asSubscript()) {
            if (s->source->asTemp()) {
                inplaceElementOp(s->op, s->source->asTemp(), ss->base->asTemp(),
                                 ss->index->asTemp());
                return;
            }
        } else if (V4IR::Member *m = s->target->asMember()) {
            if (s->source->asTemp()) {
                inplaceMemberOp(s->op, s->source->asTemp(), m->base->asTemp(), *m->name);
                return;
            }
        }
    }

    // For anything else...:
    Q_UNIMPLEMENTED();
    s->dump(qout, V4IR::Stmt::MIR);
    qout << endl;
    assert(!"TODO");
}

InstructionSelection::~InstructionSelection()
{
}

void InstructionSelection::visitEnter(Enter *)
{
    Q_UNREACHABLE();
}

void InstructionSelection::visitLeave(Leave *)
{
    Q_UNREACHABLE();
}

void InstructionSelection::visitExp(V4IR::Exp *s)
{
    if (V4IR::Call *c = s->expr->asCall()) {
        // These are calls where the result is ignored.
        if (c->base->asName()) {
            callBuiltin(c, 0);
        } else if (Temp *value = c->base->asTemp()) {
            callValue(value, c->args, 0);
        } else if (Member *member = c->base->asMember()) {
            callProperty(member->base, *member->name, c->args, 0);
        } else if (Subscript *s = c->base->asSubscript()) {
            callSubscript(s->base, s->index, c->args, 0);
        } else {
            Q_UNIMPLEMENTED();
        }
    } else {
        Q_UNIMPLEMENTED();
    }
}

void InstructionSelection::callBuiltin(V4IR::Call *call, V4IR::Temp *result)
{
    V4IR::Name *baseName = call->base->asName();
    assert(baseName != 0);

    switch (baseName->builtin) {
    case V4IR::Name::builtin_invalid:
        callBuiltinInvalid(baseName, call->args, result);
        return;

    case V4IR::Name::builtin_typeof: {
        if (V4IR::Member *m = call->args->expr->asMember()) {
            callBuiltinTypeofMember(m->base->asTemp(), *m->name, result);
            return;
        } else if (V4IR::Subscript *ss = call->args->expr->asSubscript()) {
            callBuiltinTypeofSubscript(ss->base->asTemp(), ss->index->asTemp(), result);
            return;
        } else if (V4IR::Name *n = call->args->expr->asName()) {
            callBuiltinTypeofName(*n->id, result);
            return;
        } else if (V4IR::Temp *arg = call->args->expr->asTemp()){
            assert(arg != 0);
            callBuiltinTypeofValue(arg, result);
            return;
        }
    } break;

    case V4IR::Name::builtin_delete: {
        if (V4IR::Member *m = call->args->expr->asMember()) {
            callBuiltinDeleteMember(m->base->asTemp(), *m->name, result);
            return;
        } else if (V4IR::Subscript *ss = call->args->expr->asSubscript()) {
            callBuiltinDeleteSubscript(ss->base->asTemp(), ss->index->asTemp(), result);
            return;
        } else if (V4IR::Name *n = call->args->expr->asName()) {
            callBuiltinDeleteName(*n->id, result);
            return;
        } else if (call->args->expr->asTemp()){
            // TODO: should throw in strict mode
            callBuiltinDeleteValue(result);
            return;
        }
    } break;

    case V4IR::Name::builtin_postincrement: {
        if (V4IR::Member *m = call->args->expr->asMember()) {
            callBuiltinPostIncrementMember(m->base->asTemp(), *m->name, result);
            return;
        } else if (V4IR::Subscript *ss = call->args->expr->asSubscript()) {
            callBuiltinPostIncrementSubscript(ss->base->asTemp(), ss->index->asTemp(), result);
            return;
        } else if (V4IR::Name *n = call->args->expr->asName()) {
            callBuiltinPostIncrementName(*n->id, result);
            return;
        } else if (V4IR::Temp *arg = call->args->expr->asTemp()){
            assert(arg != 0);
            callBuiltinPostIncrementValue(arg, result);
            return;
        }
    } break;

    case V4IR::Name::builtin_postdecrement: {
        if (V4IR::Member *m = call->args->expr->asMember()) {
            callBuiltinPostDecrementMember(m->base->asTemp(), *m->name, result);
            return;
        } else if (V4IR::Subscript *ss = call->args->expr->asSubscript()) {
            callBuiltinPostDecrementSubscript(ss->base->asTemp(), ss->index->asTemp(), result);
            return;
        } else if (V4IR::Name *n = call->args->expr->asName()) {
            callBuiltinPostDecrementName(*n->id, result);
            return;
        } else if (V4IR::Temp *arg = call->args->expr->asTemp()){
            assert(arg != 0);
            callBuiltinPostDecrementValue(arg, result);
            return;
        }
    } break;

    case V4IR::Name::builtin_throw: {
        V4IR::Temp *arg = call->args->expr->asTemp();
        assert(arg != 0);
        callBuiltinThrow(arg);
    } return;

    case V4IR::Name::builtin_finish_try:
        callBuiltinFinishTry();
        return;

    case V4IR::Name::builtin_foreach_iterator_object: {
        V4IR::Temp *arg = call->args->expr->asTemp();
        assert(arg != 0);
        callBuiltinForeachIteratorObject(arg, result);
    } return;

    case V4IR::Name::builtin_foreach_next_property_name: {
        V4IR::Temp *arg = call->args->expr->asTemp();
        assert(arg != 0);
        callBuiltinForeachNextPropertyname(arg, result);
    } return;
    case V4IR::Name::builtin_push_with_scope: {
        V4IR::Temp *arg = call->args->expr->asTemp();
        assert(arg != 0);
        callBuiltinPushWithScope(arg);
    } return;

    case V4IR::Name::builtin_pop_scope:
        callBuiltinPopScope();
        return;

    case V4IR::Name::builtin_declare_vars: {
        if (!call->args)
            return;
        V4IR::Const *deletable = call->args->expr->asConst();
        assert(deletable->type == V4IR::BoolType);
        for (V4IR::ExprList *it = call->args->next; it; it = it->next) {
            V4IR::Name *arg = it->expr->asName();
            assert(arg != 0);
            callBuiltinDeclareVar(deletable->value != 0, *arg->id);
        }
    } return;

    case V4IR::Name::builtin_define_getter_setter: {
        if (!call->args)
            return;
        V4IR::ExprList *args = call->args;
        V4IR::Temp *object = args->expr->asTemp();
        assert(object);
        args = args->next;
        assert(args);
        V4IR::Name *name = args->expr->asName();
        args = args->next;
        assert(args);
        V4IR::Temp *getter = args->expr->asTemp();
        args = args->next;
        assert(args);
        V4IR::Temp *setter = args->expr->asTemp();

        callBuiltinDefineGetterSetter(object, *name->id, getter, setter);
    } return;

    case V4IR::Name::builtin_define_property: {
        if (!call->args)
            return;
        V4IR::ExprList *args = call->args;
        V4IR::Temp *object = args->expr->asTemp();
        assert(object);
        args = args->next;
        assert(args);
        V4IR::Name *name = args->expr->asName();
        args = args->next;
        assert(args);
        V4IR::Temp *value = args->expr->asTemp();

        callBuiltinDefineProperty(object, *name->id, value);
    } return;

    case V4IR::Name::builtin_define_array:
        callBuiltinDefineArray(result, call->args);
        return;

    default:
        break;
    }

    Q_UNIMPLEMENTED();
    call->dump(qout); qout << endl;
    assert(!"TODO!");
    Q_UNREACHABLE();
}