/* Function: tp_set * Attribute modification. * * This is the counterpart of tp_get, it does the same as self[k] = v would do * in actual tinypy code. */ void tp_set(TP,tp_obj self, tp_obj k, tp_obj v) { int type = self.type; if (type == TP_DICT) { TP_META_BEGIN(self,"__set__"); tp_call(tp,meta,tp_params_v(tp,2,k,v)); return; TP_META_END; _tp_dict_set(tp,self.dict.val,k,v); return; } else if (type == TP_LIST) { if (k.type == TP_NUMBER) { _tp_list_set(tp,self.list.val,k.number.val,v,"tp_set"); return; } else if (k.type == TP_NONE) { _tp_list_append(tp,self.list.val,v); return; } else if (k.type == TP_STRING) { if (tp_cmp(tp,tp_string("*"),k) == 0) { tp_params_v(tp,2,self,v); tp_extend(tp); return; } } } tp_raise(,tp_string("(tp_set) TypeError: object does not support item assignment")); }
tp_obj tp_max(TP) { tp_obj r = TP_OBJ(); tp_obj e; TP_LOOP(e) if (tp_cmp(tp,r,e) < 0) { r = e; } TP_END; return r; }
tp_obj tp_istype(TP) { tp_obj v = TP_OBJ(); tp_obj t = TP_STR(); if (tp_cmp(tp,t,tp_string("string")) == 0) { return tp_number(v.type == TP_STRING); } if (tp_cmp(tp,t,tp_string("list")) == 0) { return tp_number(v.type == TP_LIST); } if (tp_cmp(tp,t,tp_string("dict")) == 0) { return tp_number(v.type == TP_DICT); } if (tp_cmp(tp,t,tp_string("number")) == 0) { return tp_number(v.type == TP_NUMBER); } if (tp_cmp(tp,t,tp_string("fnc")) == 0) { return tp_number(v.type == TP_FNC && (v.fnc.ftype&2) == 0); } if (tp_cmp(tp,t,tp_string("method")) == 0) { return tp_number(v.type == TP_FNC && (v.fnc.ftype&2) != 0); } tp_raise(tp_None,tp_string("(is_type) TypeError: ?")); }
int tp_cmp(TP,tp_obj a, tp_obj b) { if (a.type != b.type) { return a.type-b.type; } switch(a.type) { case TP_NONE: return 0; case TP_NUMBER: return _tp_sign(a.number.val-b.number.val); case TP_STRING: { int v = memcmp(a.string.val,b.string.val,_tp_min(a.string.len,b.string.len)); if (v == 0) { v = a.string.len-b.string.len; } return v; } case TP_LIST: { int n,v; for(n=0; n<_tp_min(a.list.val->len,b.list.val->len); n++) { tp_obj aa = a.list.val->items[n]; tp_obj bb = b.list.val->items[n]; if (aa.type == TP_LIST && bb.type == TP_LIST) { v = aa.list.val-bb.list.val; } else { v = tp_cmp(tp,aa,bb); } if (v) { return v; } } return a.list.val->len-b.list.val->len; } case TP_DICT: return a.dict.val - b.dict.val; case TP_FNC: return a.fnc.info - b.fnc.info; case TP_DATA: return (char*)a.data.val - (char*)b.data.val; } tp_raise(0,"tp_cmp(%s,%s)",TP_CSTR(a),TP_CSTR(b)); }
int tp_cmp(tp_vm *tp, tp_obj a, tp_obj b) { if (obj_type(a) != obj_type(b)) { return obj_type(a) - obj_type(b); } switch (obj_type(a)) { case TP_NONE: return 0; case TP_NUMBER: return _tp_sign(tp_number_val(a) - tp_number_val(b)); case TP_STRING: { int minlen = _tp_min(tp_str_len(a), tp_str_len(b)); int v = memcmp(tp_str_val(a), tp_str_val(b), minlen); if (v == 0) { v = tp_str_len(a) - tp_str_len(b); } return v; } case TP_LIST: { int n,v; int minlen = _tp_min(tp_list_val(a)->len,tp_list_val(b)->len); for (n=0; n < minlen; n++) { tp_obj aa = tp_list_val(a)->items[n]; tp_obj bb = tp_list_val(b)->items[n]; if (obj_type(aa) == TP_LIST && obj_type(bb) == TP_LIST) { v = tp_list_val(aa) - tp_list_val(bb); } else { v = tp_cmp(tp,aa,bb); } if (v) { return v; } } return tp_list_val(a)->len - tp_list_val(b)->len; } case TP_DICT: return tp_dict_val(a) - tp_dict_val(b); case TP_FNC: return tp_fnc_val(a) - tp_fnc_val(b); case TP_DATA: return (char*)tp_data_val(a) - (char*)tp_data_val(b); } tp_raise(0,"tp_cmp(%s,%s)",STR(a),STR(b)); }
/* Function: tp_get * Attribute lookup. * * This returns the result of using self[k] in actual code. It works for * dictionaries (including classes and instantiated objects), lists and strings. * * As a special case, if self is a list, self[None] will return the first * element in the list and subsequently remove it from the list. */ tp_obj tp_get(TP,tp_obj self, tp_obj k) { int type = self.type; tp_obj r; if (type == TP_DICT) { TP_META_BEGIN(self,"__get__"); return tp_call(tp,meta,tp_params_v(tp,1,k)); TP_META_END; if (self.dict.dtype && _tp_lookup(tp,self,k,&r)) { return r; } return _tp_dict_get(tp,self.dict.val,k,"tp_get"); } else if (type == TP_LIST) { if (k.type == TP_NUMBER) { int l = tp_len(tp,self).number.val; int n = k.number.val; n = (n<0?l+n:n); return _tp_list_get(tp,self.list.val,n,"tp_get"); } else if (k.type == TP_STRING) { if (tp_cmp(tp,tp_string("append"),k) == 0) { return tp_method(tp,self,tp_append); } else if (tp_cmp(tp,tp_string("pop"),k) == 0) { return tp_method(tp,self,tp_pop); } else if (tp_cmp(tp,tp_string("index"),k) == 0) { return tp_method(tp,self,tp_index); } else if (tp_cmp(tp,tp_string("sort"),k) == 0) { return tp_method(tp,self,tp_sort); } else if (tp_cmp(tp,tp_string("extend"),k) == 0) { return tp_method(tp,self,tp_extend); } else if (tp_cmp(tp,tp_string("*"),k) == 0) { tp_params_v(tp,1,self); r = tp_copy(tp); self.list.val->len=0; return r; } } else if (k.type == TP_NONE) { return _tp_list_pop(tp,self.list.val,0,"tp_get"); } } else if (type == TP_STRING) { if (k.type == TP_NUMBER) { int l = self.string.len; int n = k.number.val; n = (n<0?l+n:n); if (n >= 0 && n < l) { return tp_string_n(tp->chars[(unsigned char)self.string.val[n]],1); } } else if (k.type == TP_STRING) { if (tp_cmp(tp,tp_string("join"),k) == 0) { return tp_method(tp,self,tp_join); } else if (tp_cmp(tp,tp_string("split"),k) == 0) { return tp_method(tp,self,tp_split); } else if (tp_cmp(tp,tp_string("index"),k) == 0) { return tp_method(tp,self,tp_str_index); } else if (tp_cmp(tp,tp_string("strip"),k) == 0) { return tp_method(tp,self,tp_strip); } else if (tp_cmp(tp,tp_string("replace"),k) == 0) { return tp_method(tp,self,tp_replace); } } } if (k.type == TP_LIST) { int a,b,l; tp_obj tmp; l = tp_len(tp,self).number.val; tmp = tp_get(tp,k,tp_number(0)); if (tmp.type == TP_NUMBER) { a = tmp.number.val; } else if(tmp.type == TP_NONE) { a = 0; } else { tp_raise(tp_None,tp_string("(tp_get) TypeError: indices must be numbers")); } tmp = tp_get(tp,k,tp_number(1)); if (tmp.type == TP_NUMBER) { b = tmp.number.val; } else if(tmp.type == TP_NONE) { b = l; } else { tp_raise(tp_None,tp_string("(tp_get) TypeError: indices must be numbers")); } a = _tp_max(0,(a<0?l+a:a)); b = _tp_min(l,(b<0?l+b:b)); if (type == TP_LIST) { return tp_list_n(tp,b-a,&self.list.val->items[a]); } else if (type == TP_STRING) { /* tp_obj r = tp_string_t(tp,b-a); char *ptr = r.string.info->s; memcpy(ptr,self.string.val+a,b-a); ptr[b-a]=0; return tp_track(tp,r);*/ return tp_string_slice(tp,self,a,b); } } tp_raise(tp_None,tp_string("(tp_get) TypeError: ?")); }