//
// Created by Kosho on 2020年8月16日.
//

#include "object/hiObject.hpp"
#include "object/hiString.hpp"
#include "object/hiObject.hpp"
#include "object/hiDict.hpp"
#include "runtime/frameObject.hpp"

// this constructor is used for module only.
FrameObject::FrameObject(CodeObject *codes) {
 _codes = codes;
 _consts = codes->_consts;
 _names = codes->_names;

 _locals = new HiDict();
 _globals = _locals;
 _fast_locals = NULL;

 _stack = new HiList();
 _loop_stack = new ArrayList<Block *>();

 _pc = 0;
 _sender = NULL;
 _entry_frame = false;
}

FrameObject::FrameObject(FunctionObject *func, ObjList args, int op_arg) {
 assert((args && op_arg != 0) || (args == NULL && op_arg == 0));

 _codes = func->_func_code;
 _consts = _codes->_consts;
 _names = _codes->_names;

 _locals = new HiDict();
 _globals = func->_globals;
 _fast_locals = new HiList();

 // 形参数量
 const int argcnt = _codes->_argcount;
 const int na = op_arg & 0xff;
 const int nk = op_arg >> 8;
 int kw_pos = argcnt;

 if (func->_defaults) {
 int dft_num = func->_defaults->length();
 int arg_num = _codes->_argcount;
 while (dft_num--) {
 _fast_locals->set(--arg_num, func->_defaults->get(dft_num));
 }
 }

 // 扩展位置参数
 HiList *alist = NULL;
 HiDict *adict = NULL;

 if (argcnt < na) {
 int i = 0;
 for (; i < argcnt; i++) {
 _fast_locals->set(i, args->get(i));
 }
 alist = new HiList();
 for (; i < na; i++) {
 alist->append(args->get(i));
 }
 } else {
 for (int i = 0; i < na; i++) {
 _fast_locals->set(i, args->get(i));
 }
 }

 // 处理键参数
 for (int i = 0; i < nk; i++) {
 HiObject *key = args->get(na + i * 2);
 HiObject *val = args->get(na + i * 2 + 1);

 int index = _codes->_var_names->index(key);
 if (index >= 0) {
 _fast_locals->set(index, val);
 } else {
 if (adict == NULL) {
 adict = new HiDict();
 }

 adict->put(key, val);
 }
 }

 // 存在扩展位置参数
 if (_codes->_flag & FunctionObject::CO_VARARGS) {
 if (alist == NULL)
 alist = new HiList();
 _fast_locals->set(argcnt, alist);
 kw_pos += 1;
 } else {
 // give more parameters than need.
 if (alist != NULL) {
 printf("takes more extend parameters.\n");
 assert(false);
 }
 }

 // 存在键参数
 if (_codes->_flag & FunctionObject::CO_VARKEYWORDS) {
 if (adict == NULL)
 adict = new HiDict();

 _fast_locals->set(kw_pos, adict);
 } else {
 if (adict != NULL) {
 printf("takes more extend kw parameters.\n");
 assert(false);
 }
 }

 _closure = NULL;
 ArrayList<HiObject *> *cells = _codes->_cell_vars;
 if (cells && cells->size() > 0) {
 _closure = new HiList();

 for (int i = 0; i < cells->size(); i++) {
 _closure->append(NULL);
 }
 }

 if (func->closure() && func->closure()->size() > 0) {
 if (_closure == NULL)
 _closure = func->closure();
 else {
 _closure = (HiList *) _closure->add(func->closure());
 }
 }

 _stack = new HiList();
 _loop_stack = new ArrayList<Block *>();

 _pc = 0;
 _sender = NULL;
 _entry_frame = false;
}

int FrameObject::get_op_arg() {
 int byte1 = _codes->_bytecodes->value()[_pc++] & 0xff;
 int byte2 = _codes->_bytecodes->value()[_pc++] & 0xff;
 return byte2 << 8 | byte1;
}

unsigned char FrameObject::get_op_code() {
 return _codes->_bytecodes->value()[_pc++];
}

bool FrameObject::has_more_codes() {
 return _pc < _codes->_bytecodes->length();
}

HiObject *FrameObject::get_cell_from_parameter(int i) {
 HiObject *cell_name = _codes->_cell_vars->get(i);
 i = _codes->_var_names->index(cell_name);
 return _fast_locals->get(i);
}

int FrameObject::lineno() {
 return _codes->_lineno;
}

HiString *FrameObject::file_name() {
 return _codes->_file_name;
}

HiString *FrameObject::func_name() {
 return _codes->_co_name;
}

void FrameObject::oops_do(OopClosure *f) {
 f->do_array_list(&_consts);
 f->do_array_list(&_names);

 f->do_oop((HiObject **) &_globals);
 f->do_oop((HiObject **) &_locals);
 f->do_oop((HiObject **) &_fast_locals);
 f->do_oop((HiObject **) &_closure);
 f->do_oop((HiObject **) &_stack);

 f->do_oop((HiObject **) &_codes);

 if (_sender) {
 // sender is not in heap
 _sender->oops_do(f);
 }
}