#include "pobject.h"
#include "pintobject.h"
#include "pboolobject.h"
#include "pstringobject.h"
#include "../pycodegen.h"


#if USE_RUNTIME_SWITCHES
DEFINEVAR fixed_switch_t psyfs_int;
DEFINEVAR fixed_switch_t psyfs_int_long;
DEFINEVAR fixed_switch_t psyfs_int_long_float;
DEFINEVAR fixed_switch_t psyfs_tuple_list;
DEFINEVAR fixed_switch_t psyfs_string_unicode;
DEFINEVAR fixed_switch_t psyfs_tuple;
DEFINEVAR fixed_switch_t psyfs_dict;
DEFINEVAR fixed_switch_t psyfs_none;
#endif


DEFINEFN
PyTypeObject* Psyco_NeedType(PsycoObject* po, vinfo_t* vi)
{
	if (is_compiletime(vi->source)) {
		/* optimization for the type of a known object */
		PyObject* o = (PyObject*) CompileTime_Get(vi->source)->value;
		return o->ob_type;
	}
	else {
		vinfo_t* vtp = psyco_get_const(po, vi, OB_type);
		if (vtp == NULL)
			return NULL;
		return (PyTypeObject*) psyco_pyobj_atcompiletime(po, vtp);
	}
}

DEFINEFN
PyTypeObject* Psyco_KnownType(vinfo_t* vi)
{
	if (is_compiletime(vi->source)) {
		PyObject* o = (PyObject*) CompileTime_Get(vi->source)->value;
		return o->ob_type;
	}
	else {
		vinfo_t* vtp = vinfo_getitem(vi, iOB_TYPE);
		if (vtp != NULL && is_compiletime(vtp->source))
			return (PyTypeObject*)
				CompileTime_Get(vtp->source)->value;
		else
			return NULL;
	}
}


DEFINEFN
vinfo_t* PsycoObject_IsTrue(PsycoObject* po, vinfo_t* vi)
{
	PyTypeObject* tp;
	tp = (PyTypeObject*) Psyco_NeedType(po, vi);
	if (tp == NULL)
		return NULL;

	if (tp == Py_None->ob_type)
		return psyco_vi_Zero();
	else if (tp->tp_as_number != NULL &&
		 tp->tp_as_number->nb_nonzero != NULL)
		return Psyco_META1(po, tp->tp_as_number->nb_nonzero,
				   CfReturnNormal|CfPyErrIfNeg, "v", vi);
	else if (tp->tp_as_mapping != NULL &&
		 tp->tp_as_mapping->mp_length != NULL)
		return Psyco_META1(po, tp->tp_as_mapping->mp_length,
				   CfReturnNormal|CfPyErrIfNeg, "v", vi);
	else if (tp->tp_as_sequence != NULL &&
		 tp->tp_as_sequence->sq_length != NULL)
		return Psyco_META1(po, tp->tp_as_sequence->sq_length,
				   CfReturnNormal|CfPyErrIfNeg, "v", vi);
	else
		return psyco_vi_One();
}

DEFINEFN
vinfo_t* PsycoObject_Repr(PsycoObject* po, vinfo_t* vi)
{
	/* XXX implement me */
	vinfo_t* vstr = psyco_generic_call(po, PyObject_Repr,
					   CfReturnRef|CfPyErrIfNull,
					   "v", vi);
	if (vstr == NULL)
		return NULL;
	
	/* the result is a string */
	Psyco_AssertType(po, vstr, &PyString_Type);
	return vstr;
}

DEFINEFN
vinfo_t* PsycoObject_GetAttr(PsycoObject* po, vinfo_t* o, vinfo_t* attr_name)
{
	/* TypeSwitch */
	PyTypeObject* tp = Psyco_NeedType(po, attr_name);
	if (tp == NULL)
		return NULL;

	if (!PyType_TypeCheck(tp, &PyString_Type)) {

#ifdef Py_USING_UNICODE
		if (PyType_TypeCheck(tp, &PyUnicode_Type))
			goto generic;
#endif
		PycException_SetString(po, PyExc_TypeError,
				       "attribute name must be string");
		return NULL;
	}

	tp = (PyTypeObject*) Psyco_NeedType(po, o);
	if (tp == NULL)
		return NULL;
	if (tp->tp_getattro != NULL)
		return Psyco_META2(po, tp->tp_getattro,
				   CfReturnRef|CfPyErrIfNull,
				   "vv", o, attr_name);
	if (tp->tp_getattr != NULL)
		return Psyco_META2(po, tp->tp_getattr,
				   CfReturnRef|CfPyErrIfNull,
				   "vv", o,
                                   PsycoString_AS_STRING(po, attr_name));

   generic:
	/* when the above fails */
	return psyco_generic_call(po, PyObject_GetAttr,
				  CfReturnRef|CfPyErrIfNull,
				  "vv", o, attr_name);
}

DEFINEFN
bool PsycoObject_SetAttr(PsycoObject* po, vinfo_t* o,
                         vinfo_t* vattrname, vinfo_t* v)
{
	PyObject* name;
        PyTypeObject* tp;
	vinfo_t* vresult;
	/* as in PsycoObject_GenericGetAttr() we don't try to analyse
	   a non-constant vattrname */
	if (!is_compiletime(vattrname->source))
		goto generic;

	tp = (PyTypeObject*) Psyco_NeedType(po, o);
	if (tp == NULL)
		return false;

	name = (PyObject*) CompileTime_Get(vattrname->source)->value;
	if (!PyString_Check(name)){
#ifdef Py_USING_UNICODE
		/* The Unicode to string conversion is done here because the
		   existing tp_setattro slots expect a string object as name
		   and we wouldn't want to break those. */
		if (PyUnicode_Check(name)) {
# if PSYCO_CAN_CALL_UNICODE
			name = PyUnicode_AsEncodedString(name, NULL, NULL);
			if (name == NULL) {
				psyco_virtualize_exception(po);
				return NULL;
			}
# else
			goto generic;
# endif
		}
		else
#endif
		{
			PycException_SetString(po, PyExc_TypeError,
					"attribute name must be string");
			return false;
		}
	}
	else
		Py_INCREF(name);

	PyString_InternInPlace(&name);
	if (tp->tp_setattro != NULL) {
		vresult = Psyco_META3(po, tp->tp_setattro,
				      CfNoReturnValue|CfPyErrIfNonNull,
				      v ? "vlv" : "vll", o, name, v);
		Py_DECREF(name);
		return vresult != NULL;
	}
	if (tp->tp_setattr != NULL) {
		vresult = Psyco_META3(po, tp->tp_setattr,
				      CfNoReturnValue|CfPyErrIfNonNull,
				      v ? "vlv" : "vll", o,
				      (long)PyString_AS_STRING(name), v);
		Py_DECREF(name);
		return vresult != NULL;
	}
	Py_DECREF(name);

   generic:
	/* fall-back implementation */
	return psyco_generic_call(po, PyObject_SetAttr,
				  CfNoReturnValue|CfPyErrIfNonNull,
				  v ? "vvv" : "vvl", o, vattrname, v) != NULL;
}

/* Helper to get the offset an object's __dict__ slot, if any.
   Return a defield_t to the base offset, or NO_PREV_FIELD if there is
   no __dict__ slot.  A variable offset may be computed and stored in
   *varindex. */

#if NEW_STYLE_TYPES   /* Python >= 2.2b1 */

static defield_t getdictoffset(PsycoObject* po, vinfo_t* obj, vinfo_t** varindex)
{
	long dictoffset;
	PyTypeObject *tp = Psyco_FastType(obj);

	/*if (!(tp->tp_flags & Py_TPFLAGS_HAVE_CLASS))
	  return NO_PREV_FIELD;*/
	dictoffset = tp->tp_dictoffset;
	if (dictoffset == 0)
		return NO_PREV_FIELD;
	if (dictoffset < 0) {
		vinfo_t* size1;
		vinfo_t* ob_size = psyco_get_field(po, obj, VAR_signed_size);
		if (ob_size == NULL)
			return NO_PREV_FIELD;
		size1 = integer_abs(po, ob_size, false);
		vinfo_decref(ob_size, po);
		if (size1 == NULL)
			return NO_PREV_FIELD;
		
		extra_assert(dictoffset % SIZEOF_VOID_P == 0);
		/* the following code emulates _PyObject_VAR_SIZE() */
		if ((tp->tp_itemsize & (SIZEOF_VOID_P-1)) == 0 &&
		    (tp->tp_basicsize & (SIZEOF_VOID_P-1)) == 0) {
			/* the result is automatically aligned */
			vinfo_t* a;
			a = integer_mul_i(po, size1,
					  tp->tp_itemsize / SIZEOF_VOID_P);
			vinfo_decref(size1, po);
			if (a == NULL)
				return NO_PREV_FIELD;
			*varindex = a;
			dictoffset = tp->tp_basicsize + dictoffset;
		}
		else {
			/* need to align the result to the next
			   SIZEOF_VOID_P boundary */
			vinfo_t* a;
			vinfo_t* b;
			vinfo_t* c;
			a = integer_mul_i(po, size1, tp->tp_itemsize);
			vinfo_decref(size1, po);
			if (a == NULL)
				return NO_PREV_FIELD;
			c = integer_add_i(po, a, tp->tp_basicsize + dictoffset +
						 SIZEOF_VOID_P - 1, false);
			vinfo_decref(a, po);
			if (c == NULL)
				return NO_PREV_FIELD;
			b = integer_urshift_i(po, c, SIZE_OF_LONG_BITS);
			vinfo_decref(c, po);
			if (b == NULL)
				return NO_PREV_FIELD;
			*varindex = b;
			dictoffset = 0;
		}
	}
	return FMUT(DEF_ARRAY(PyObject*, dictoffset));
}

DEFINEFN
vinfo_t* PsycoObject_GenericGetAttr(PsycoObject* po, vinfo_t* obj,
				    vinfo_t* vname)
{
	PyTypeObject* tp;
	PyObject *descr = NULL;
	vinfo_t* res = NULL;
	descrgetfunc f;
	PyObject* name;
	defield_t dictofsbase;
	vinfo_t* dictofs;

	/* 'name' is generally known at compile-time.
	   We could promote it at compile-time with
	   
		name = psyco_pyobj_atcompiletime(po, vname);
		if (name == NULL)
			return NULL;

	   but we don't, because run-time names generally
	   mean that the Python code uses functions like getattr(),
	   probably inside of a loop, with a lot of various
	   attributes. Let's emit a call to PyObject_GenericGetAttr
	   instead. */
	if (!is_compiletime(vname->source)) {
	   fallback:
		return psyco_generic_call(po, PyObject_GenericGetAttr,
					  CfReturnRef|CfPyErrIfNull,
					  "vv", obj, vname);
        }
	name = (PyObject*) CompileTime_Get(vname->source)->value;

	if (!PyString_Check(name)){
#ifdef Py_USING_UNICODE
		/* The Unicode to string conversion is done here because the
		   existing tp_setattro slots expect a string object as name
		   and we wouldn't want to break those. */
		if (PyUnicode_Check(name)) {
# if PSYCO_CAN_CALL_UNICODE
			name = PyUnicode_AsEncodedString(name, NULL, NULL);
			if (name == NULL) {
				psyco_virtualize_exception(po);
				return NULL;
			}
# else
			goto fallback;
# endif
		}
		else
#endif
		{
			PycException_SetString(po, PyExc_TypeError,
					"attribute name must be string");
			return NULL;
		}
	}
	else
		Py_INCREF(name);

	/* we need the type of 'obj' at compile-time */
	tp = (PyTypeObject*) Psyco_NeedType(po, obj);
	if (tp == NULL)
		goto done;

	if (tp->tp_dict == NULL) {
		if (PyType_Ready(tp) < 0) {
			psyco_virtualize_exception(po);
			goto done;
		}
	}

	/* XXX this is broken because the type might have been modified
	   XXX since the last time we were here! */
	descr = _PyType_Lookup(tp, name);
	Py_XINCREF(descr);
	
	f = NULL;
	if (descr != NULL &&
	    PyType_HasFeature(descr->ob_type, Py_TPFLAGS_HAVE_CLASS)) {
		f = descr->ob_type->tp_descr_get;
		if (f != NULL && PyDescr_IsData(descr)) {
			res = Psyco_META3(po, f, CfReturnRef|CfPyErrIfNull,
					  "lvl", descr, obj, tp);
			goto done;
		}
	}

	dictofs = NULL;
	dictofsbase = getdictoffset(po, obj, &dictofs);
	if (dictofsbase == NO_PREV_FIELD) {
		if (PycException_Occurred(po))
			goto done;
		/* no __dict__ slot */
	}
	else {
		condition_code_t cc;
		int cond;
		vinfo_t* dict;
		if (dictofs == NULL)
			dict = psyco_get_field(po, obj, dictofsbase);
		else {
			dict = psyco_get_field_array(po, obj, dictofsbase,
						     dictofs);
			vinfo_decref(dictofs, po);
		}
		if (dict == NULL)
			goto done;
		cc = integer_non_null(po, dict);
		if (cc == CC_ERROR) {
			vinfo_decref(dict, po);
			goto done;
		}

		if (runtime_condition_t(po, cc)) {
			/* the __dict__ slot contains a non-NULL value */
			res = psyco_generic_call(po, PyDict_GetItem,
						 CfReturnNormal,
						 "vl", dict, (long) name);
			vinfo_decref(dict, po);
			if (res == NULL)
				goto done;
			Py_INCREF(name);  /* XXX ref hold by the code buf */

			/* if there is a method descriptor of the same name,
			   then it is likely that we don't find an attribute
			   with that name; otherwise, it is likely that we
			   do find it. */
			cc = integer_non_null(po, res);
			if (cc == CC_ERROR) {
				vinfo_decref(res, po);
				goto done;
			}
			
			if (f != NULL)
				cond = runtime_condition_f(po, cc);
			else
				cond = runtime_condition_t(po, cc);

			if (cond) {
				/* we have found the attribute */
				need_reference(po, res);
				goto done;
			}
			else {
				/* there is no such attribute */
				vinfo_decref(res, po);
				res = NULL;
			}
		}
		else {
			/* the __dict__ slot contains NULL */
			vinfo_decref(dict, po);
		}
	}
	
	if (f != NULL) {
		res = Psyco_META3(po, f, CfReturnRef|CfPyErrIfNull,
				  "lvl", descr, obj, tp);
		goto done;
	}

	if (descr != NULL) {
		source_known_t* sk = sk_new((long) descr, SkFlagPyObj);
		descr = NULL;
		res = vinfo_new(CompileTime_NewSk(sk));
		goto done;
	}

	PycException_SetFormat(po, PyExc_AttributeError,
			       "'%.50s' object has no attribute '%.400s'",
			       tp->tp_name, PyString_AS_STRING(name));
  done:
	Py_XDECREF(descr);
	Py_DECREF(name);
	return res;
}
#endif /* NEW_STYLE_TYPES */


/* Macro to get the tp_richcompare field of a type if defined */
#define RICHCOMPARE(t) (PyType_HasFeature((t), Py_TPFLAGS_HAVE_RICHCOMPARE) \
                         ? (t)->tp_richcompare : NULL)


/* Map rich comparison operators to their swapped version, e.g. LT --> GT */
static int swapped_op[] = {Py_GT, Py_GE, Py_EQ, Py_NE, Py_LT, Py_LE};

static vinfo_t* try_rich_compare(PsycoObject* po, vinfo_t* v, vinfo_t* w, int op)
{
	bool swap;
	PyTypeObject* vtp = Psyco_FastType(v);
	PyTypeObject* wtp = Psyco_FastType(w);
	richcmpfunc fv = RICHCOMPARE(vtp);
	richcmpfunc fw = RICHCOMPARE(wtp);
	vinfo_t* res;

#if NEW_STYLE_TYPES   /* Python >= 2.2b1 */
	swap = (vtp != wtp &&
		PyType_IsSubtype(wtp, vtp) &&
		fw != NULL);
#else
        swap = 0;
#endif
        
	if (swap) {
		res = Psyco_META3(po, fw, CfReturnRef|CfPyErrNotImplemented,
				  "vvl", w, v, swapped_op[op]);
		if (IS_IMPLEMENTED(res))
			return res;   /* 'res' might be NULL */
		vinfo_decref(res, po);
	}
	if (fv != NULL) {
		res = Psyco_META3(po, fv, CfReturnRef|CfPyErrNotImplemented,
				  "vvl", v, w, op);
		if (IS_IMPLEMENTED(res))
			return res;
		vinfo_decref(res, po);
	}
	if (!swap && fw != NULL) {
		return Psyco_META3(po, fw, CfReturnRef|CfPyErrNotImplemented,
				   "vvl", w, v, swapped_op[op]);
	}
	return psyco_vi_NotImplemented();
}

PSY_INLINE vinfo_t* convert_3way_to_object(PsycoObject* po, int op, vinfo_t* c)
{
	condition_code_t cc = integer_cmp_i(po, c, 0, op);
	if (cc == CC_ERROR)
		return NULL;
	return PsycoBool_FromCondition(po, cc);
}

PSY_INLINE vinfo_t* try_3way_to_rich_compare(PsycoObject* po, vinfo_t* v,
					 vinfo_t* w, int op)
{
	/* XXX implement me (some day) */
	return psyco_generic_call(po, PyObject_RichCompare,
				  CfReturnRef|CfPyErrIfNull,
				  "vvl", v, w, (long) op);
}

DEFINEFN vinfo_t* PsycoObject_RichCompare(PsycoObject* po, vinfo_t* v,
					  vinfo_t* w, int op)
{
	PyTypeObject* vtp;
	PyTypeObject* wtp;
	vinfo_t* res;
	cmpfunc f;
	extra_assert(Py_LT <= op && op <= Py_GE);
	
	/* XXX try to detect circular data structures */

	vtp = (PyTypeObject*) Psyco_NeedType(po, v);
	if (vtp == NULL)
		return NULL;
	wtp = (PyTypeObject*) Psyco_NeedType(po, w);
	if (wtp == NULL)
		return NULL;

	/* If the types are equal, don't bother with coercions etc. 
	   Instances are special-cased in try_3way_compare, since
	   a result of 2 does *not* mean one value being greater
	   than the other. */
	if (vtp == wtp
	    && (f = vtp->tp_compare) != NULL
	    && !PyType_TypeCheck(vtp, &PyInstance_Type)) {
		vinfo_t* c;
		richcmpfunc f1;
		if (vtp == &PyInt_Type) {
			/* Special-case integers because they don't use
			   rich comparison, and the plain 3-way comparisons
			   are too low-level for serious optimizations
			   (they would require building the value -1, 0 or
			   1 in a register and then testing it). */
			return PsycoIntInt_RichCompare(po, v, w, op);
		}
		if ((f1 = RICHCOMPARE(vtp)) != NULL) {
			/* If the type has richcmp, try it first.
			   try_rich_compare would try it two-sided,
			   which is not needed since we've a single
			   type only. */
			res = Psyco_META3(po, f1,
					  CfReturnRef|CfPyErrNotImplemented,
					  "vvl", v, w, (long) op);
			if (IS_IMPLEMENTED(res))
				return res;
			vinfo_decref(res, po);
		}
		c = Psyco_META2(po, f, CfReturnNormal|CfPyErrCheck,
				"vv", v, w);
		if (c == NULL)
			return NULL;
		res = convert_3way_to_object(po, op, c);
                vinfo_decref(c, po);
                return res;
	}

	res = try_rich_compare(po, v, w, op);
	if (IS_IMPLEMENTED(res))
		return res;
	vinfo_decref(res, po);

	return try_3way_to_rich_compare(po, v, w, op);
}

DEFINEFN
vinfo_t* PsycoObject_RichCompareBool(PsycoObject* po,
                                     vinfo_t* v, vinfo_t* w, int op)
{
	vinfo_t* diff;
	vinfo_t* result = PsycoObject_RichCompare(po, v, w, op);
	if (result == NULL)
		return NULL;
        diff = PsycoObject_IsTrue(po, result);
        vinfo_decref(result, po);
	return diff;
}

static vinfo_t* collect_undeletable_vars(vinfo_t* vi, vinfo_t* link)
{
	int i;
	PyTypeObject* tp;
	switch (gettime(vi->source)) {

	case RunTime:
		if (vi->tmp != NULL || (vi->source & RunTime_NoRef) != 0)
			break;  /* already seen or not holding a ref */
		tp = Psyco_KnownType(vi);
		if (tp && (tp == &PyInt_Type || tp == &PyString_Type ||
#if BOOLEAN_TYPE
			   tp == &PyBool_Type ||
#endif
			   tp == &PyFloat_Type || tp == &PyLong_Type ||
			   tp == Py_None->ob_type || tp == &PyRange_Type))
			break;  /* known safe type */
		/* make a linked list of results */
		vi->tmp = link;
		link = vi;
		break;

	case VirtualTime:
		for (i=vi->array->count; i--; )
			if (vi->array->items[i] != NULL)
				link = collect_undeletable_vars(
					vi->array->items[i], link);
		break;

	default:
		break;
	}
	return link;
}

DEFINEFN
vinfo_t* Psyco_SafelyDeleteVar(PsycoObject* po, vinfo_t* vi)
{
	vinfo_t* result;
	vinfo_t* head;
	vinfo_t* queue = (vinfo_t*) 1;
	vinfo_t* p;
	int count;
	vi->tmp = NULL;
	clear_tmp_marks(vi->array);

	head = collect_undeletable_vars(vi, queue);
	count = 0;
	for (p = head; p != queue; p = p->tmp) {
		vinfo_array_t* a = p->array;
		p->array = NullArray;
		array_delete(a, po);
		count++;
	}
	if (count == 0) {
		result = psyco_vi_Zero();
	}
	else if (count == 1) {
		result = head;
		vinfo_incref(result);
	}
	else {
		result = vinfo_new(
			VirtualTime_New(&psyco_vsource_not_important));
		count += iOB_TYPE + 1;
		result->array = array_new(count);
		for (p = head; p != queue; p = p->tmp) {
			vinfo_incref(p);
			result->array->items[--count] = p;
		}
	}
	return result;
}


INITIALIZATIONFN
void psy_object_init(void)
{
#if USE_RUNTIME_SWITCHES
	long values[3];
        int cnt;

	values[0] = (long)(&PyInt_Type);
	psyco_build_run_time_switch(&psyfs_int, SkFlagFixed, values, 1);

	values[0] = (long)(&PyInt_Type);
	values[1] = (long)(&PyLong_Type);
	values[2] = (long)(&PyFloat_Type);
        psyco_build_run_time_switch(&psyfs_int_long, SkFlagFixed, values, 2);
	psyco_build_run_time_switch(&psyfs_int_long_float, SkFlagFixed, values, 3);

	values[0] = (long)(&PyTuple_Type);
	values[1] = (long)(&PyList_Type);
        psyco_build_run_time_switch(&psyfs_tuple_list, SkFlagFixed, values, 2);

	values[0] = (long)(&PyString_Type);
#ifdef Py_USING_UNICODE
	values[1] = (long)(&PyUnicode_Type);
        cnt = 2;
#else
        cnt = 1;
#endif
        psyco_build_run_time_switch(&psyfs_string_unicode, SkFlagFixed,
                                    values, cnt);

	values[0] = (long)(Py_None->ob_type);
	psyco_build_run_time_switch(&psyfs_none, SkFlagFixed, values, 1);

        values[0] = (long)(&PyTuple_Type);
	psyco_build_run_time_switch(&psyfs_tuple, SkFlagFixed, values, 1);

        values[0] = (long)(&PyDict_Type);
	psyco_build_run_time_switch(&psyfs_dict, SkFlagFixed, values, 1);
#endif   /* USE_RUNTIME_SWITCHES */

        /* associate the Python implementation of some functions with
           the one from Psyco */
#if NEW_STYLE_TYPES   /* Python >= 2.2b1 */
        Psyco_DefineMeta(PyObject_GenericGetAttr, PsycoObject_GenericGetAttr);
#endif
}