home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / pyth_os2.zip / python-1.0.2 / Python / bltinmodule.c next >
C/C++ Source or Header  |  1994-05-03  |  27KB  |  1,393 lines
  1. /***********************************************************
  2. Copyright 1991, 1992, 1993, 1994 by Stichting Mathematisch Centrum,
  3. Amsterdam, The Netherlands.
  4.  
  5.                         All Rights Reserved
  6.  
  7. Permission to use, copy, modify, and distribute this software and its 
  8. documentation for any purpose and without fee is hereby granted, 
  9. provided that the above copyright notice appear in all copies and that
  10. both that copyright notice and this permission notice appear in 
  11. supporting documentation, and that the names of Stichting Mathematisch
  12. Centrum or CWI not be used in advertising or publicity pertaining to
  13. distribution of the software without specific, written prior permission.
  14.  
  15. STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
  16. THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
  17. FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
  18. FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  19. WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  20. ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
  21. OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  22.  
  23. ******************************************************************/
  24.  
  25. /* Built-in functions */
  26.  
  27. #include "allobjects.h"
  28.  
  29. #include "node.h"
  30. #include "graminit.h"
  31. #include "sysmodule.h"
  32. #include "bltinmodule.h"
  33. #include "import.h"
  34. #include "pythonrun.h"
  35. #include "ceval.h"
  36. #include "modsupport.h"
  37. #include "compile.h"
  38. #include "eval.h"
  39.  
  40. /* Forward */
  41. static object *filterstring PROTO((object *, object *));
  42. static object *filtertuple  PROTO((object *, object *));
  43. static object *exec_eval PROTO((object *v, int start));
  44.  
  45. static object *
  46. builtin_abs(self, v)
  47.     object *self;
  48.     object *v;
  49. {
  50.     number_methods *nm;
  51.     if (v == NULL || (nm = v->ob_type->tp_as_number) == NULL) {
  52.         err_setstr(TypeError, "abs() requires numeric argument");
  53.         return NULL;
  54.     }
  55.     return (*nm->nb_absolute)(v);
  56. }
  57.  
  58. static object *
  59. builtin_apply(self, args)
  60.     object *self;
  61.     object *args;
  62. {
  63.     object *func, *arglist;
  64.     if (!getargs(args, "(OO)", &func, &arglist))
  65.         return NULL;
  66.     if (!is_tupleobject(arglist)) {
  67.         err_setstr(TypeError, "apply() 2nd argument must be tuple");
  68.         return NULL;
  69.     }
  70.     return call_object(func, arglist);
  71. }
  72.  
  73. static object *
  74. builtin_filter(self, args)
  75.     object *self;
  76.     object *args;
  77. {
  78.     object *func, *seq, *result;
  79.     sequence_methods *sqf;
  80.     int len;
  81.     register int i, j;
  82.  
  83.     if (!getargs(args, "(OO)", &func, &seq))
  84.         return NULL;
  85.  
  86.     if (is_stringobject(seq)) {
  87.         object *r = filterstring(func, seq);
  88.         return r;
  89.     }
  90.  
  91.     if (is_tupleobject(seq)) {
  92.         object *r = filtertuple(func, seq);
  93.         return r;
  94.     }
  95.  
  96.     if ((sqf = seq->ob_type->tp_as_sequence) == NULL) {
  97.         err_setstr(TypeError,
  98.                "argument 2 to filter() must be a sequence type");
  99.         goto Fail_2;
  100.     }
  101.  
  102.     if ((len = (*sqf->sq_length)(seq)) < 0)
  103.         goto Fail_2;
  104.  
  105.     if (is_listobject(seq) && seq->ob_refcnt == 1) {
  106.         INCREF(seq);
  107.         result = seq;
  108.     }
  109.     else {
  110.         if ((result = newlistobject(len)) == NULL)
  111.             goto Fail_2;
  112.     }
  113.  
  114.     for (i = j = 0; ; ++i) {
  115.         object *item, *good;
  116.         int ok;
  117.  
  118.         if ((item = (*sqf->sq_item)(seq, i)) == NULL) {
  119.             if (i < len)
  120.                 goto Fail_1;
  121.             if (err_occurred() == IndexError) {
  122.                 err_clear();
  123.                 break;
  124.             }
  125.             goto Fail_1;
  126.         }
  127.  
  128.         if (func == None) {
  129.             good = item;
  130.         }
  131.         else {
  132.             object *arg = mkvalue("(O)", item);
  133.             DECREF(item);
  134.             if (arg == NULL)
  135.                 goto Fail_1;
  136.             good = call_object(func, arg);
  137.             DECREF(arg);
  138.             if (good == NULL)
  139.                 goto Fail_1;
  140.         }
  141.         ok = testbool(good);
  142.         DECREF(good);
  143.         if (ok) {
  144.             INCREF(item);
  145.             if (j < len) {
  146.                 if (setlistitem(result, j++, item) < 0)
  147.                     goto Fail_1;
  148.             }
  149.             else {
  150.                 j++;
  151.                 if (addlistitem(result, item) < 0)
  152.                     goto Fail_1;
  153.             }
  154.         }
  155.     }
  156.  
  157.  
  158.     if (j < len && setlistslice(result, j, len, NULL) < 0)
  159.         goto Fail_1;
  160.  
  161.     return result;
  162.  
  163. Fail_1:
  164.     DECREF(result);
  165. Fail_2:
  166.     return NULL;
  167. }
  168.  
  169. static object *
  170. builtin_chr(self, args)
  171.     object *self;
  172.     object *args;
  173. {
  174.     long x;
  175.     char s[1];
  176.     if (!getargs(args, "l", &x))
  177.         return NULL;
  178.     if (x < 0 || x >= 256) {
  179.         err_setstr(ValueError, "chr() arg not in range(256)");
  180.         return NULL;
  181.     }
  182.     s[0] = x;
  183.     return newsizedstringobject(s, 1);
  184. }
  185.  
  186. static object *
  187. builtin_cmp(self, args)
  188.     object *self;
  189.     object *args;
  190. {
  191.     object *a, *b;
  192.     if (!getargs(args, "(OO)", &a, &b))
  193.         return NULL;
  194.     return newintobject((long)cmpobject(a, b));
  195. }
  196.  
  197. static object *
  198. builtin_coerce(self, args)
  199.     object *self;
  200.     object *args;
  201. {
  202.     object *v, *w;
  203.     object *res;
  204.  
  205.     if (!getargs(args, "(OO)", &v, &w))
  206.         return NULL;
  207.     if (coerce(&v, &w) < 0)
  208.         return NULL;
  209.     res = mkvalue("(OO)", v, w);
  210.     DECREF(v);
  211.     DECREF(w);
  212.     return res;
  213. }
  214.  
  215. static object *
  216. builtin_compile(self, args)
  217.     object *self;
  218.     object *args;
  219. {
  220.     char *str;
  221.     char *filename;
  222.     char *startstr;
  223.     int start;
  224.     if (!getargs(args, "(sss)", &str, &filename, &startstr))
  225.         return NULL;
  226.     if (strcmp(startstr, "exec") == 0)
  227.         start = file_input;
  228.     else if (strcmp(startstr, "eval") == 0)
  229.         start = eval_input;
  230.     else {
  231.         err_setstr(ValueError,
  232.                "compile() mode must be 'exec' or 'eval'");
  233.         return NULL;
  234.     }
  235.     return compile_string(str, filename, start);
  236. }
  237.  
  238. static object *
  239. builtin_dir(self, v)
  240.     object *self;
  241.     object *v;
  242. {
  243.     object *d;
  244.     if (v == NULL) {
  245.         d = getlocals();
  246.         INCREF(d);
  247.     }
  248.     else {
  249.         d = getattr(v, "__dict__");
  250.         if (d == NULL) {
  251.             err_setstr(TypeError,
  252.                 "dir() argument must have __dict__ attribute");
  253.             return NULL;
  254.         }
  255.     }
  256.     if (is_dictobject(d)) {
  257.         v = getdictkeys(d);
  258.         if (sortlist(v) != 0) {
  259.             DECREF(v);
  260.             v = NULL;
  261.         }
  262.     }
  263.     else {
  264.         v = newlistobject(0);
  265.     }
  266.     DECREF(d);
  267.     return v;
  268. }
  269.  
  270. static object *
  271. builtin_divmod(self, args)
  272.     object *self;
  273.     object *args;
  274. {
  275.     object *v, *w, *x;
  276.     if (!getargs(args, "(OO)", &v, &w))
  277.         return NULL;
  278.     if (v->ob_type->tp_as_number == NULL ||
  279.                 w->ob_type->tp_as_number == NULL) {
  280.         err_setstr(TypeError, "divmod() requires numeric arguments");
  281.         return NULL;
  282.     }
  283.     if (coerce(&v, &w) != 0)
  284.         return NULL;
  285.     x = (*v->ob_type->tp_as_number->nb_divmod)(v, w);
  286.     DECREF(v);
  287.     DECREF(w);
  288.     return x;
  289. }
  290.  
  291. static object *
  292. exec_eval(v, start)
  293.     object *v;
  294.     int start;
  295. {
  296.     object *str = NULL, *globals = NULL, *locals = NULL;
  297.     char *s;
  298.     int n;
  299.     /* XXX This is a bit of a mess.  Should make it varargs */
  300.     if (v != NULL) {
  301.         if (is_tupleobject(v) &&
  302.                 ((n = gettuplesize(v)) == 2 || n == 3)) {
  303.             str = gettupleitem(v, 0);
  304.             globals = gettupleitem(v, 1);
  305.             if (n == 3)
  306.                 locals = gettupleitem(v, 2);
  307.         }
  308.         else
  309.             str = v;
  310.     }
  311.     if (str == NULL || (!is_stringobject(str) && !is_codeobject(str)) ||
  312.             globals != NULL && !is_dictobject(globals) ||
  313.             locals != NULL && !is_dictobject(locals)) {
  314.         err_setstr(TypeError,
  315.           "eval arguments must be (string|code)[,dict[,dict]]");
  316.         return NULL;
  317.     }
  318.  
  319.     if (is_codeobject(str))
  320.         return eval_code((codeobject *) str, globals, locals,
  321.                  (object *)NULL, (object *)NULL);
  322.     s = getstringvalue(str);
  323.     if (strlen(s) != getstringsize(str)) {
  324.         err_setstr(ValueError, "embedded '\\0' in string arg");
  325.         return NULL;
  326.     }
  327.     if (start == eval_input) {
  328.         while (*s == ' ' || *s == '\t')
  329.             s++;
  330.     }
  331.     return run_string(s, start, globals, locals);
  332. }
  333.  
  334. static object *
  335. builtin_eval(self, v)
  336.     object *self;
  337.     object *v;
  338. {
  339.     return exec_eval(v, eval_input);
  340. }
  341.  
  342. static object *
  343. builtin_execfile(self, v)
  344.     object *self;
  345.     object *v;
  346. {
  347.     object *str = NULL, *globals = NULL, *locals = NULL, *w;
  348.     FILE* fp;
  349.     char *s;
  350.     int n;
  351.     if (v != NULL) {
  352.         if (is_stringobject(v))
  353.             str = v;
  354.         else if (is_tupleobject(v) &&
  355.                 ((n = gettuplesize(v)) == 2 || n == 3)) {
  356.             str = gettupleitem(v, 0);
  357.             globals = gettupleitem(v, 1);
  358.             if (n == 3)
  359.                 locals = gettupleitem(v, 2);
  360.         }
  361.     }
  362.     if (str == NULL || !is_stringobject(str) ||
  363.             globals != NULL && !is_dictobject(globals) ||
  364.             locals != NULL && !is_dictobject(locals)) {
  365.         err_setstr(TypeError,
  366.             "execfile arguments must be filename[,dict[,dict]]");
  367.         return NULL;
  368.     }
  369.     s = getstringvalue(str);
  370.     if (strlen(s) != getstringsize(str)) {
  371.         err_setstr(ValueError, "embedded '\\0' in string arg");
  372.         return NULL;
  373.     }
  374.     BGN_SAVE
  375.     fp = fopen(s, "r");
  376.     END_SAVE
  377.     if (fp == NULL) {
  378.         err_setstr(IOError, "execfile cannot open the file argument");
  379.         return NULL;
  380.     }
  381.     w = run_file(fp, getstringvalue(str), file_input, globals, locals);
  382.     BGN_SAVE
  383.     fclose(fp);
  384.     END_SAVE
  385.     return w;
  386. }
  387.  
  388. static object *
  389. builtin_float(self, v)
  390.     object *self;
  391.     object *v;
  392. {
  393.     number_methods *nb;
  394.     
  395.     if (v == NULL || (nb = v->ob_type->tp_as_number) == NULL ||
  396.         nb->nb_float == NULL) {
  397.         err_setstr(TypeError,
  398.                "float() argument can't be converted to float");
  399.         return NULL;
  400.     }
  401.     return (*nb->nb_float)(v);
  402. }
  403.  
  404. static object *
  405. builtin_getattr(self, args)
  406.     object *self;
  407.     object *args;
  408. {
  409.     object *v;
  410.     object *name;
  411.     if (!getargs(args, "(OS)", &v, &name))
  412.         return NULL;
  413.     return getattro(v, name);
  414. }
  415.  
  416. static object *
  417. builtin_hasattr(self, args)
  418.     object *self;
  419.     object *args;
  420. {
  421.     object *v;
  422.     object *name;
  423.     if (!getargs(args, "(OS)", &v, &name))
  424.         return NULL;
  425.     v = getattro(v, name);
  426.     if (v == NULL) {
  427.         err_clear();
  428.         return newintobject(0L);
  429.     }
  430.     DECREF(v);
  431.     return newintobject(1L);
  432. }
  433.  
  434. static object *
  435. builtin_id(self, args)
  436.     object *self;
  437.     object *args;
  438. {
  439.     object *v;
  440.     if (!getargs(args, "O", &v))
  441.         return NULL;
  442.     return newintobject((long)v);
  443. }
  444.  
  445. static object *
  446. builtin_map(self, args)
  447.     object *self;
  448.     object *args;
  449. {
  450.     typedef struct {
  451.         object *seq;
  452.         sequence_methods *sqf;
  453.         int len;
  454.     } sequence;
  455.  
  456.     object *func, *result;
  457.     sequence *seqs = NULL, *sqp;
  458.     int n, len, newfunc = 0;
  459.     register int i, j;
  460.  
  461.     if (args == NULL || !is_tupleobject(args)) {
  462.         err_setstr(TypeError, "map() requires at least two args");
  463.         return NULL;
  464.     }
  465.  
  466.     func = gettupleitem(args, 0);
  467.     n    = gettuplesize(args) - 1;
  468.  
  469.     if ((seqs = NEW(sequence, n)) == NULL) {
  470.         err_nomem();
  471.         goto Fail_2;
  472.     }
  473.  
  474.     for (len = 0, i = 0, sqp = seqs; i < n; ++i, ++sqp) {
  475.         int curlen;
  476.     
  477.         if ((sqp->seq = gettupleitem(args, i + 1)) == NULL)
  478.             goto Fail_2;
  479.  
  480.         if (! (sqp->sqf = sqp->seq->ob_type->tp_as_sequence)) {
  481.             static char errmsg[] =
  482.                 "argument %d to map() must be a sequence object";
  483.             char errbuf[sizeof(errmsg) + 3];
  484.  
  485.             sprintf(errbuf, errmsg, i+2);
  486.             err_setstr(TypeError, errbuf);
  487.             goto Fail_2;
  488.         }
  489.  
  490.         if ((curlen = sqp->len = (*sqp->sqf->sq_length)(sqp->seq)) < 0)
  491.             goto Fail_2;
  492.  
  493.         if (curlen > len)
  494.             len = curlen;
  495.     }
  496.  
  497.     if ((result = (object *) newlistobject(len)) == NULL)
  498.         goto Fail_2;
  499.  
  500.     /* XXX Special case map(None, single_list) could be more efficient */
  501.     for (i = 0; ; ++i) {
  502.         object *arglist, *item, *value;
  503.         int any = 0;
  504.  
  505.         if (func == None && n == 1)
  506.             arglist = NULL;
  507.         else {
  508.             if ((arglist = newtupleobject(n)) == NULL)
  509.                 goto Fail_1;
  510.         }
  511.  
  512.         for (j = 0, sqp = seqs; j < n; ++j, ++sqp) {
  513.             if (sqp->len < 0) {
  514.                 INCREF(None);
  515.                 item = None;
  516.             }
  517.             else {
  518.                 item = (*sqp->sqf->sq_item)(sqp->seq, i);
  519.                 if (item == NULL) {
  520.                     if (i < sqp->len)
  521.                         goto Fail_0;
  522.                     if (err_occurred() == IndexError) {
  523.                         err_clear();
  524.                         INCREF(None);
  525.                         item = None;
  526.                         sqp->len = -1;
  527.                     }
  528.                     else {
  529.                         goto Fail_0;
  530.                     }
  531.                 }
  532.                 else
  533.                     any = 1;
  534.  
  535.             }
  536.             if (!arglist)
  537.                 break;
  538.             if (settupleitem(arglist, j, item) < 0) {
  539.                 DECREF(item);
  540.                 goto Fail_0;
  541.             }
  542.             continue;
  543.  
  544.         Fail_0:
  545.             XDECREF(arglist);
  546.             goto Fail_1;
  547.         }
  548.  
  549.         if (!arglist)
  550.             arglist = item;
  551.  
  552.         if (!any) {
  553.             DECREF(arglist);
  554.             break;
  555.         }
  556.  
  557.         if (func == None)
  558.             value = arglist;
  559.         else {
  560.             value = call_object(func, arglist);
  561.             DECREF(arglist);
  562.             if (value == NULL)
  563.                 goto Fail_1;
  564.         }
  565.         if (i >= len) {
  566.             if (addlistitem(result, value) < 0)
  567.                 goto Fail_1;
  568.         }
  569.         else {
  570.             if (setlistitem(result, i, value) < 0)
  571.                 goto Fail_1;
  572.         }
  573.     }
  574.  
  575.     DEL(seqs);
  576.     return result;
  577.  
  578. Fail_1:
  579.     DECREF(result);
  580. Fail_2:
  581.     if (seqs) DEL(seqs);
  582.     return NULL;
  583. }
  584.  
  585. static object *
  586. builtin_setattr(self, args)
  587.     object *self;
  588.     object *args;
  589. {
  590.     object *v;
  591.     object *name;
  592.     object *value;
  593.     if (!getargs(args, "(OSO)", &v, &name, &value))
  594.         return NULL;
  595.     if (setattro(v, name, value) != 0)
  596.         return NULL;
  597.     INCREF(None);
  598.     return None;
  599. }
  600.  
  601. static object *
  602. builtin_hash(self, args)
  603.     object *self;
  604.     object *args;
  605. {
  606.     object *v;
  607.     long x;
  608.     if (!getargs(args, "O", &v))
  609.         return NULL;
  610.     x = hashobject(v);
  611.     if (x == -1)
  612.         return NULL;
  613.     return newintobject(x);
  614. }
  615.  
  616. static object *
  617. builtin_hex(self, v)
  618.     object *self;
  619.     object *v;
  620. {
  621.     number_methods *nb;
  622.     
  623.     if (v == NULL || (nb = v->ob_type->tp_as_number) == NULL ||
  624.         nb->nb_hex == NULL) {
  625.         err_setstr(TypeError,
  626.                "hex() argument can't be converted to hex");
  627.         return NULL;
  628.     }
  629.     return (*nb->nb_hex)(v);
  630. }
  631.  
  632. static object *builtin_raw_input PROTO((object *, object *));
  633.  
  634. static object *
  635. builtin_input(self, v)
  636.     object *self;
  637.     object *v;
  638. {
  639.     object *line = builtin_raw_input(self, v);
  640.     if (line == NULL)
  641.         return line;
  642.     v = exec_eval(line, eval_input);
  643.     DECREF(line);
  644.     return v;
  645. }
  646.  
  647. static object *
  648. builtin_int(self, v)
  649.     object *self;
  650.     object *v;
  651. {
  652.     number_methods *nb;
  653.     
  654.     if (v == NULL || (nb = v->ob_type->tp_as_number) == NULL ||
  655.         nb->nb_int == NULL) {
  656.         err_setstr(TypeError,
  657.                "int() argument can't be converted to int");
  658.         return NULL;
  659.     }
  660.     return (*nb->nb_int)(v);
  661. }
  662.  
  663. static object *
  664. builtin_len(self, v)
  665.     object *self;
  666.     object *v;
  667. {
  668.     long len;
  669.     typeobject *tp;
  670.     if (v == NULL) {
  671.         err_setstr(TypeError, "len() without argument");
  672.         return NULL;
  673.     }
  674.     tp = v->ob_type;
  675.     if (tp->tp_as_sequence != NULL) {
  676.         len = (*tp->tp_as_sequence->sq_length)(v);
  677.     }
  678.     else if (tp->tp_as_mapping != NULL) {
  679.         len = (*tp->tp_as_mapping->mp_length)(v);
  680.     }
  681.     else {
  682.         err_setstr(TypeError, "len() of unsized object");
  683.         return NULL;
  684.     }
  685.     if (len < 0)
  686.         return NULL;
  687.     else
  688.         return newintobject(len);
  689. }
  690.  
  691. static object *
  692. builtin_long(self, v)
  693.     object *self;
  694.     object *v;
  695. {
  696.     number_methods *nb;
  697.     
  698.     if (v == NULL || (nb = v->ob_type->tp_as_number) == NULL ||
  699.         nb->nb_long == NULL) {
  700.         err_setstr(TypeError,
  701.                "long() argument can't be converted to long");
  702.         return NULL;
  703.     }
  704.     return (*nb->nb_long)(v);
  705. }
  706.  
  707. static object *
  708. min_max(v, sign)
  709.     object *v;
  710.     int sign;
  711. {
  712.     int i;
  713.     object *w, *x;
  714.     sequence_methods *sq;
  715.     if (v == NULL) {
  716.         err_setstr(TypeError, "min() or max() without argument");
  717.         return NULL;
  718.     }
  719.     sq = v->ob_type->tp_as_sequence;
  720.     if (sq == NULL) {
  721.         err_setstr(TypeError, "min() or max() of non-sequence");
  722.         return NULL;
  723.     }
  724.     w = NULL;
  725.     for (i = 0; ; i++) {
  726.         x = (*sq->sq_item)(v, i); /* Implies INCREF */
  727.         if (x == NULL) {
  728.             if (err_occurred() == IndexError) {
  729.                 err_clear();
  730.                 break;
  731.             }
  732.             XDECREF(w);
  733.             return NULL;
  734.         }
  735.         if (w == NULL)
  736.             w = x;
  737.         else {
  738.             if (cmpobject(x, w) * sign > 0) {
  739.                 DECREF(w);
  740.                 w = x;
  741.             }
  742.             else
  743.                 DECREF(x);
  744.         }
  745.     }
  746.     if (w == NULL)
  747.         err_setstr(ValueError, "min() or max() of empty sequence");
  748.     return w;
  749. }
  750.  
  751. static object *
  752. builtin_min(self, v)
  753.     object *self;
  754.     object *v;
  755. {
  756.     return min_max(v, -1);
  757. }
  758.  
  759. static object *
  760. builtin_max(self, v)
  761.     object *self;
  762.     object *v;
  763. {
  764.     return min_max(v, 1);
  765. }
  766.  
  767. static object *
  768. builtin_oct(self, v)
  769.     object *self;
  770.     object *v;
  771. {
  772.     number_methods *nb;
  773.     
  774.     if (v == NULL || (nb = v->ob_type->tp_as_number) == NULL ||
  775.         nb->nb_oct == NULL) {
  776.         err_setstr(TypeError,
  777.                "oct() argument can't be converted to oct");
  778.         return NULL;
  779.     }
  780.     return (*nb->nb_oct)(v);
  781. }
  782.  
  783. static object *
  784. builtin_open(self, args)
  785.     object *self;
  786.     object *args;
  787. {
  788.     char *name;
  789.     char *mode = "r";
  790.     int bufsize = -1;
  791.     object *f;
  792.     if (!getargs(args, "s", &name) &&
  793.         (err_clear(), !getargs(args, "(ss)", &name, &mode)) &&
  794.         (err_clear(), !getargs(args, "(ssi)", &name, &mode, &bufsize)))
  795.         return NULL;
  796.     f = newfileobject(name, mode);
  797.     if (f != NULL)
  798.         setfilebufsize(f, bufsize);
  799.     return f;
  800. }
  801.  
  802. static object *
  803. builtin_ord(self, args)
  804.     object *self;
  805.     object *args;
  806. {
  807.     char *s;
  808.     int len;
  809.     if (!getargs(args, "s#", &s, &len))
  810.         return NULL;
  811.     if (len != 1) {
  812.         err_setstr(ValueError, "ord() arg must have length 1");
  813.         return NULL;
  814.     }
  815.     return newintobject((long)(s[0] & 0xff));
  816. }
  817.  
  818. static object *
  819. builtin_pow(self, args)
  820.     object *self;
  821.     object *args;
  822. {
  823.     object *v, *w, *x;
  824.     if (!getargs(args, "(OO)", &v, &w))
  825.         return NULL;
  826.     if (v->ob_type->tp_as_number == NULL ||
  827.                 w->ob_type->tp_as_number == NULL) {
  828.         err_setstr(TypeError, "pow() requires numeric arguments");
  829.         return NULL;
  830.     }
  831.     if (coerce(&v, &w) != 0)
  832.         return NULL;
  833.     x = (*v->ob_type->tp_as_number->nb_power)(v, w);
  834.     DECREF(v);
  835.     DECREF(w);
  836.     return x;
  837. }
  838.  
  839. static object *
  840. builtin_range(self, v)
  841.     object *self;
  842.     object *v;
  843. {
  844.     static char *errmsg = "range() requires 1-3 int arguments";
  845.     int i, n;
  846.     long ilow, ihigh, istep;
  847.     if (v != NULL && is_intobject(v)) {
  848.         ilow = 0; ihigh = getintvalue(v); istep = 1;
  849.     }
  850.     else if (v == NULL || !is_tupleobject(v)) {
  851.         err_setstr(TypeError, errmsg);
  852.         return NULL;
  853.     }
  854.     else {
  855.         n = gettuplesize(v);
  856.         if (n < 1 || n > 3) {
  857.             err_setstr(TypeError, errmsg);
  858.             return NULL;
  859.         }
  860.         for (i = 0; i < n; i++) {
  861.             if (!is_intobject(gettupleitem(v, i))) {
  862.                 err_setstr(TypeError, errmsg);
  863.                 return NULL;
  864.             }
  865.         }
  866.         if (n == 3) {
  867.             istep = getintvalue(gettupleitem(v, 2));
  868.             --n;
  869.         }
  870.         else
  871.             istep = 1;
  872.         ihigh = getintvalue(gettupleitem(v, --n));
  873.         if (n > 0)
  874.             ilow = getintvalue(gettupleitem(v, 0));
  875.         else
  876.             ilow = 0;
  877.     }
  878.     if (istep == 0) {
  879.         err_setstr(ValueError, "zero step for range()");
  880.         return NULL;
  881.     }
  882.     /* XXX ought to check overflow of subtraction */
  883.     if (istep > 0)
  884.         n = (ihigh - ilow + istep - 1) / istep;
  885.     else
  886.         n = (ihigh - ilow + istep + 1) / istep;
  887.     if (n < 0)
  888.         n = 0;
  889.     v = newlistobject(n);
  890.     if (v == NULL)
  891.         return NULL;
  892.     for (i = 0; i < n; i++) {
  893.         object *w = newintobject(ilow);
  894.         if (w == NULL) {
  895.             DECREF(v);
  896.             return NULL;
  897.         }
  898.         setlistitem(v, i, w);
  899.         ilow += istep;
  900.     }
  901.     return v;
  902. }
  903.  
  904. static object *
  905. builtin_xrange(self, v)
  906.     object *self;
  907.     object *v;
  908. {
  909.     static char *errmsg = "xrange() requires 1-3 int arguments";
  910.     int i, n;
  911.     long start, stop, step, len;
  912.     if (v != NULL && is_intobject(v))
  913.         start = 0, stop = getintvalue(v), step = 1;
  914.  
  915.     else if (v == NULL || !is_tupleobject(v)) {
  916.         err_setstr(TypeError, errmsg);
  917.         return NULL;
  918.     }
  919.     else {
  920.         n = gettuplesize(v);
  921.         if (n < 1 || n > 3) {
  922.             err_setstr(TypeError, errmsg);
  923.             return NULL;
  924.         }
  925.         for (i = 0; i < n; i++) {
  926.             if (!is_intobject(gettupleitem(v, i))) {
  927.                 err_setstr(TypeError, errmsg);
  928.                 return NULL;
  929.             }
  930.         }
  931.         if (n == 3) {
  932.             step = getintvalue(gettupleitem(v, 2));
  933.             --n;
  934.         }
  935.         else
  936.             step = 1;
  937.         stop = getintvalue(gettupleitem(v, --n));
  938.         if (n > 0)
  939.             start = getintvalue(gettupleitem(v, 0));
  940.         else
  941.             start = 0;
  942.     }
  943.  
  944.     if (step == 0) {
  945.         err_setstr(ValueError, "zero step for xrange()");
  946.         return NULL;
  947.     }
  948.  
  949.     len = (stop - start + step + ((step > 0) ? -1 : 1)) / step;
  950.     if (len < 0)
  951.         len = 0;
  952.  
  953.     return newrangeobject(start, len, step, 1);
  954. }
  955.  
  956. static object *
  957. builtin_raw_input(self, v)
  958.     object *self;
  959.     object *v;
  960. {
  961.     object *f = sysget("stdout");
  962.     if (f == NULL) {
  963.         err_setstr(RuntimeError, "lost sys.stdout");
  964.         return NULL;
  965.     }
  966.     flushline();
  967.     if (v != NULL) {
  968.         if (writeobject(v, f, PRINT_RAW) != 0)
  969.             return NULL;
  970.     }
  971.     return filegetline(sysget("stdin"), -1);
  972. }
  973.  
  974. static object *
  975. builtin_reduce(self, args)
  976.     object *self;
  977.     object *args;
  978. {
  979.     object *seq, *func, *result;
  980.     sequence_methods *sqf;
  981.     register int i;
  982.  
  983.     if (getargs(args, "(OO)", &func, &seq))
  984.         result = NULL;
  985.     else {
  986.         err_clear();
  987.         if (!getargs(args, "(OOO)", &func, &seq, &result))
  988.             return NULL;
  989.         INCREF(result);
  990.     }
  991.  
  992.     if ((sqf = seq->ob_type->tp_as_sequence) == NULL) {
  993.         err_setstr(TypeError,
  994.             "2nd argument to reduce() must be a sequence object");
  995.         return NULL;
  996.     }
  997.  
  998.     if ((args = newtupleobject(2)) == NULL)
  999.         goto Fail;
  1000.  
  1001.     for (i = 0; ; ++i) {
  1002.         object *op2;
  1003.  
  1004.         if (args->ob_refcnt > 1) {
  1005.             DECREF(args);
  1006.             if ((args = newtupleobject(2)) == NULL)
  1007.                 goto Fail;
  1008.         }
  1009.  
  1010.         if ((op2 = (*sqf->sq_item)(seq, i)) == NULL) {
  1011.             if (err_occurred() == IndexError) {
  1012.                 err_clear();
  1013.                 break;
  1014.             }
  1015.             goto Fail;
  1016.         }
  1017.  
  1018.         if (result == NULL)
  1019.             result = op2;
  1020.         else {
  1021.             settupleitem(args, 0, result);
  1022.             settupleitem(args, 1, op2);
  1023.             if ((result = call_object(func, args)) == NULL)
  1024.                 goto Fail;
  1025.         }
  1026.     }
  1027.  
  1028.     DECREF(args);
  1029.  
  1030.     if (result == NULL)
  1031.         err_setstr(TypeError,
  1032.                "reduce of empty sequence with no initial value");
  1033.  
  1034.     return result;
  1035.  
  1036. Fail:
  1037.     XDECREF(args);
  1038.     XDECREF(result);
  1039.     return NULL;
  1040. }
  1041.  
  1042. static object *
  1043. builtin_reload(self, v)
  1044.     object *self;
  1045.     object *v;
  1046. {
  1047.     return reload_module(v);
  1048. }
  1049.  
  1050. static object *
  1051. builtin_repr(self, v)
  1052.     object *self;
  1053.     object *v;
  1054. {
  1055.     if (v == NULL) {
  1056.         err_badarg();
  1057.         return NULL;
  1058.     }
  1059.     return reprobject(v);
  1060. }
  1061.  
  1062. static object *
  1063. builtin_round(self, args)
  1064.     object *self;
  1065.     object *args;
  1066. {
  1067.     extern double floor PROTO((double));
  1068.     extern double ceil PROTO((double));
  1069.     double x;
  1070.     double f;
  1071.     int ndigits = 0;
  1072.     int sign = 1;
  1073.     int i;
  1074.     if (!getargs(args, "d", &x)) {
  1075.         err_clear();
  1076.         if (!getargs(args, "(di)", &x, &ndigits))
  1077.             return NULL;
  1078.     }
  1079.     f = 1.0;
  1080.     for (i = ndigits; --i >= 0; )
  1081.         f = f*10.0;
  1082.     for (i = ndigits; ++i <= 0; )
  1083.         f = f*0.1;
  1084.     if (x >= 0.0)
  1085.         return newfloatobject(floor(x*f + 0.5) / f);
  1086.     else
  1087.         return newfloatobject(ceil(x*f - 0.5) / f);
  1088. }
  1089.  
  1090. static object *
  1091. builtin_str(self, v)
  1092.     object *self;
  1093.     object *v;
  1094. {
  1095.     if (v == NULL) {
  1096.         err_badarg();
  1097.         return NULL;
  1098.     }
  1099.     return strobject(v);
  1100. }
  1101.  
  1102. static object *
  1103. builtin_type(self, v)
  1104.     object *self;
  1105.     object *v;
  1106. {
  1107.     if (v == NULL) {
  1108.         err_setstr(TypeError, "type() requires an argument");
  1109.         return NULL;
  1110.     }
  1111.     v = (object *)v->ob_type;
  1112.     INCREF(v);
  1113.     return v;
  1114. }
  1115.  
  1116. static object *
  1117. builtin_vars(self, v)
  1118.     object *self;
  1119.     object *v;
  1120. {
  1121.     object *d;
  1122.     if (v == NULL) {
  1123.         d = getlocals();
  1124.         INCREF(d);
  1125.     }
  1126.     else {
  1127.         d = getattr(v, "__dict__");
  1128.         if (d == NULL) {
  1129.             err_setstr(TypeError,
  1130.                 "dir() argument must have __dict__ attribute");
  1131.             return NULL;
  1132.         }
  1133.     }
  1134.     return d;
  1135. }
  1136.  
  1137. static struct methodlist builtin_methods[] = {
  1138.     {"abs",        builtin_abs},
  1139.     {"apply",    builtin_apply},
  1140.     {"chr",        builtin_chr},
  1141.     {"cmp",        builtin_cmp},
  1142.     {"coerce",    builtin_coerce},
  1143.     {"compile",    builtin_compile},
  1144.     {"dir",        builtin_dir},
  1145.     {"divmod",    builtin_divmod},
  1146.     {"eval",    builtin_eval},
  1147.     {"execfile",    builtin_execfile},
  1148.     {"filter",    builtin_filter},
  1149.     {"float",    builtin_float},
  1150.     {"getattr",    builtin_getattr},
  1151.     {"hasattr",    builtin_hasattr},
  1152.     {"hash",    builtin_hash},
  1153.     {"hex",        builtin_hex},
  1154.     {"id",        builtin_id},
  1155.     {"input",    builtin_input},
  1156.     {"int",        builtin_int},
  1157.     {"len",        builtin_len},
  1158.     {"long",    builtin_long},
  1159.     {"map",        builtin_map},
  1160.     {"max",        builtin_max},
  1161.     {"min",        builtin_min},
  1162.     {"oct",        builtin_oct},
  1163.     {"open",    builtin_open},
  1164.     {"ord",        builtin_ord},
  1165.     {"pow",        builtin_pow},
  1166.     {"range",    builtin_range},
  1167.     {"raw_input",    builtin_raw_input},
  1168.     {"reduce",    builtin_reduce},
  1169.     {"reload",    builtin_reload},
  1170.     {"repr",    builtin_repr},
  1171.     {"round",    builtin_round},
  1172.     {"setattr",    builtin_setattr},
  1173.     {"str",        builtin_str},
  1174.     {"type",    builtin_type},
  1175.     {"vars",    builtin_vars},
  1176.     {"xrange",    builtin_xrange},
  1177.     {NULL,        NULL},
  1178. };
  1179.  
  1180. static object *builtin_dict;
  1181.  
  1182. object *
  1183. getbuiltin(name)
  1184.     object *name;
  1185. {
  1186.     return dict2lookup(builtin_dict, name);
  1187. }
  1188.  
  1189. /* Predefined exceptions */
  1190.  
  1191. object *AccessError;
  1192. object *AttributeError;
  1193. object *ConflictError;
  1194. object *EOFError;
  1195. object *IOError;
  1196. object *ImportError;
  1197. object *IndexError;
  1198. object *KeyError;
  1199. object *KeyboardInterrupt;
  1200. object *MemoryError;
  1201. object *NameError;
  1202. object *OverflowError;
  1203. object *RuntimeError;
  1204. object *SyntaxError;
  1205. object *SystemError;
  1206. object *SystemExit;
  1207. object *TypeError;
  1208. object *ValueError;
  1209. object *ZeroDivisionError;
  1210.  
  1211. static object *
  1212. newstdexception(name)
  1213.     char *name;
  1214. {
  1215.     object *v = newstringobject(name);
  1216.     if (v == NULL || dictinsert(builtin_dict, name, v) != 0)
  1217.         fatal("no mem for new standard exception");
  1218.     return v;
  1219. }
  1220.  
  1221. static void
  1222. initerrors()
  1223. {
  1224.     AccessError = newstdexception("AccessError");
  1225.     AttributeError = newstdexception("AttributeError");
  1226.     ConflictError = newstdexception("ConflictError");
  1227.     EOFError = newstdexception("EOFError");
  1228.     IOError = newstdexception("IOError");
  1229.     ImportError = newstdexception("ImportError");
  1230.     IndexError = newstdexception("IndexError");
  1231.     KeyError = newstdexception("KeyError");
  1232.     KeyboardInterrupt = newstdexception("KeyboardInterrupt");
  1233.     MemoryError = newstdexception("MemoryError");
  1234.     NameError = newstdexception("NameError");
  1235.     OverflowError = newstdexception("OverflowError");
  1236.     RuntimeError = newstdexception("RuntimeError");
  1237.     SyntaxError = newstdexception("SyntaxError");
  1238.     SystemError = newstdexception("SystemError");
  1239.     SystemExit = newstdexception("SystemExit");
  1240.     TypeError = newstdexception("TypeError");
  1241.     ValueError = newstdexception("ValueError");
  1242.     ZeroDivisionError = newstdexception("ZeroDivisionError");
  1243. }
  1244.  
  1245. void
  1246. initbuiltin()
  1247. {
  1248.     object *m;
  1249.     m = initmodule("__builtin__", builtin_methods);
  1250.     builtin_dict = getmoduledict(m);
  1251.     INCREF(builtin_dict);
  1252.     initerrors();
  1253.     (void) dictinsert(builtin_dict, "None", None);
  1254. }
  1255.  
  1256. /* Coerce two numeric types to the "larger" one.
  1257.    Increment the reference count on each argument.
  1258.    Return -1 and raise an exception if no coercion is possible
  1259.    (and then no reference count is incremented).
  1260. */
  1261.  
  1262. int
  1263. coerce(pv, pw)
  1264.     object **pv, **pw;
  1265. {
  1266.     register object *v = *pv;
  1267.     register object *w = *pw;
  1268.     int res;
  1269.  
  1270.     if (v->ob_type == w->ob_type && !is_instanceobject(v)) {
  1271.         INCREF(v);
  1272.         INCREF(w);
  1273.         return 0;
  1274.     }
  1275.     if (v->ob_type->tp_as_number && v->ob_type->tp_as_number->nb_coerce) {
  1276.         res = (*v->ob_type->tp_as_number->nb_coerce)(pv, pw);
  1277.         if (res <= 0)
  1278.             return res;
  1279.     }
  1280.     if (w->ob_type->tp_as_number && w->ob_type->tp_as_number->nb_coerce) {
  1281.         res = (*w->ob_type->tp_as_number->nb_coerce)(pw, pv);
  1282.         if (res <= 0)
  1283.             return res;
  1284.     }
  1285.     err_setstr(TypeError, "number coercion failed");
  1286.     return -1;
  1287. }
  1288.  
  1289.  
  1290. /* Helper for filter(): filter a tuple through a function */
  1291.  
  1292. static object *
  1293. filtertuple(func, tuple)
  1294.     object *func;
  1295.     object *tuple;
  1296. {
  1297.     object *result;
  1298.     register int i, j;
  1299.     int len = gettuplesize(tuple);
  1300.  
  1301.     if ((result = newtupleobject(len)) == NULL)
  1302.         return NULL;
  1303.  
  1304.     for (i = j = 0; i < len; ++i) {
  1305.         object *item, *good;
  1306.         int ok;
  1307.  
  1308.         if ((item = gettupleitem(tuple, i)) == NULL)
  1309.             goto Fail_1;
  1310.         if (func == None) {
  1311.             INCREF(item);
  1312.             good = item;
  1313.         }
  1314.         else {
  1315.             object *arg = mkvalue("(O)", item);
  1316.             if (arg == NULL)
  1317.                 goto Fail_1;
  1318.             good = call_object(func, arg);
  1319.             DECREF(arg);
  1320.             if (good == NULL)
  1321.                 goto Fail_1;
  1322.         }
  1323.         ok = testbool(good);
  1324.         DECREF(good);
  1325.         if (ok) {
  1326.             INCREF(item);
  1327.             if (settupleitem(result, j++, item) < 0)
  1328.                 goto Fail_1;
  1329.         }
  1330.     }
  1331.  
  1332.     if (resizetuple(&result, j, 0) < 0)
  1333.         return NULL;
  1334.  
  1335.     return result;
  1336.  
  1337. Fail_1:
  1338.     DECREF(result);
  1339.     return NULL;
  1340. }
  1341.  
  1342.  
  1343. /* Helper for filter(): filter a string through a function */
  1344.  
  1345. static object *
  1346. filterstring(func, strobj)
  1347.     object *func;
  1348.     object *strobj;
  1349. {
  1350.     object *result;
  1351.     register int i, j;
  1352.     int len = getstringsize(strobj);
  1353.  
  1354.     if (func == None) {
  1355.         /* No character is ever false -- share input string */
  1356.         INCREF(strobj);
  1357.         return strobj;
  1358.     }
  1359.     if ((result = newsizedstringobject(NULL, len)) == NULL)
  1360.         return NULL;
  1361.  
  1362.     for (i = j = 0; i < len; ++i) {
  1363.         object *item, *arg, *good;
  1364.         int ok;
  1365.  
  1366.         item = (*strobj->ob_type->tp_as_sequence->sq_item)(strobj, i);
  1367.         if (item == NULL)
  1368.             goto Fail_1;
  1369.         arg = mkvalue("(O)", item);
  1370.         DECREF(item);
  1371.         if (arg == NULL)
  1372.             goto Fail_1;
  1373.         good = call_object(func, arg);
  1374.         DECREF(arg);
  1375.         if (good == NULL)
  1376.             goto Fail_1;
  1377.         ok = testbool(good);
  1378.         DECREF(good);
  1379.         if (ok)
  1380.             GETSTRINGVALUE((stringobject *)result)[j++] =
  1381.                 GETSTRINGVALUE((stringobject *)item)[0];
  1382.     }
  1383.  
  1384.     if (resizestring(&result, j) < 0)
  1385.         return NULL;
  1386.  
  1387.     return result;
  1388.  
  1389. Fail_1:
  1390.     DECREF(result);
  1391.     return NULL;
  1392. }
  1393.