blob: 4dc0b9034d2129d274c0474ab7c8c014d191179a [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001/***********************************************************
Guido van Rossum9bfef441993-03-29 10:43:31 +00002Copyright 1991, 1992, 1993 by Stichting Mathematisch Centrum,
3Amsterdam, The Netherlands.
Guido van Rossumf70e43a1991-02-19 12:39:46 +00004
5 All Rights Reserved
6
7Permission to use, copy, modify, and distribute this software and its
8documentation for any purpose and without fee is hereby granted,
9provided that the above copyright notice appear in all copies and that
10both that copyright notice and this permission notice appear in
11supporting documentation, and that the names of Stichting Mathematisch
12Centrum or CWI not be used in advertising or publicity pertaining to
13distribution of the software without specific, written prior permission.
14
15STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
16THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
17FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
18FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
19WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
20ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
21OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
22
23******************************************************************/
24
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000025/* Function object implementation */
26
Guido van Rossum3f5da241990-12-20 15:06:42 +000027#include "allobjects.h"
Guido van Rossum9bfef441993-03-29 10:43:31 +000028#include "compile.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +000029#include "structmember.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000030
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000031object *
Guido van Rossum846e4311990-11-18 17:44:06 +000032newfuncobject(code, globals)
33 object *code;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000034 object *globals;
35{
36 funcobject *op = NEWOBJ(funcobject, &Functype);
37 if (op != NULL) {
Guido van Rossum846e4311990-11-18 17:44:06 +000038 INCREF(code);
39 op->func_code = code;
40 INCREF(globals);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000041 op->func_globals = globals;
Guido van Rossum25831651993-05-19 14:50:45 +000042 op->func_name = ((codeobject*)(op->func_code))->co_name;
43 INCREF(op->func_name);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000044 }
45 return (object *)op;
46}
47
Guido van Rossum846e4311990-11-18 17:44:06 +000048object *
49getfunccode(op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000050 object *op;
51{
52 if (!is_funcobject(op)) {
Guido van Rossum2a9096b1990-10-21 22:15:08 +000053 err_badcall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000054 return NULL;
55 }
Guido van Rossum846e4311990-11-18 17:44:06 +000056 return ((funcobject *) op) -> func_code;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000057}
58
59object *
60getfuncglobals(op)
61 object *op;
62{
63 if (!is_funcobject(op)) {
Guido van Rossum2a9096b1990-10-21 22:15:08 +000064 err_badcall();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000065 return NULL;
66 }
67 return ((funcobject *) op) -> func_globals;
68}
69
70/* Methods */
71
Guido van Rossum3f5da241990-12-20 15:06:42 +000072#define OFF(x) offsetof(funcobject, x)
73
74static struct memberlist func_memberlist[] = {
Guido van Rossuma38c0ff1992-01-14 18:32:20 +000075 {"func_code", T_OBJECT, OFF(func_code), READONLY},
76 {"func_globals",T_OBJECT, OFF(func_globals), READONLY},
Guido van Rossum25831651993-05-19 14:50:45 +000077 {"func_name", T_OBJECT, OFF(func_name), READONLY},
Guido van Rossum3f5da241990-12-20 15:06:42 +000078 {NULL} /* Sentinel */
79};
80
81static object *
82func_getattr(op, name)
83 funcobject *op;
84 char *name;
85{
86 return getmember((char *)op, func_memberlist, name);
87}
88
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000089static void
Guido van Rossum3f5da241990-12-20 15:06:42 +000090func_dealloc(op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000091 funcobject *op;
92{
Guido van Rossum846e4311990-11-18 17:44:06 +000093 DECREF(op->func_code);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000094 DECREF(op->func_globals);
Guido van Rossum846e4311990-11-18 17:44:06 +000095 DEL(op);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000096}
97
Guido van Rossum9bfef441993-03-29 10:43:31 +000098static object*
99func_repr(op)
100 funcobject *op;
101{
102 char buf[140];
103 sprintf(buf, "<function %.100s at %lx>",
Guido van Rossum25831651993-05-19 14:50:45 +0000104 getstringvalue(op->func_name),
Guido van Rossum9bfef441993-03-29 10:43:31 +0000105 (long)op);
106 return newstringobject(buf);
107}
108
Guido van Rossum2e8f8a31993-11-05 10:20:10 +0000109static int
110func_compare(f, g)
111 funcobject *f, *g;
112{
113 if (f->func_globals != g->func_globals)
114 return (f->func_globals < g->func_globals) ? -1 : 1;
115 return cmpobject(f->func_code, g->func_code);
116}
117
118static long
119func_hash(f)
120 funcobject *f;
121{
122 long h;
123 h = hashobject(f->func_code);
124 if (h == -1) return h;
125 h = h ^ (long)f->func_globals;
126 if (h == -1) h = -2;
127 return h;
128}
129
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000130typeobject Functype = {
131 OB_HEAD_INIT(&Typetype)
132 0,
133 "function",
134 sizeof(funcobject),
135 0,
Guido van Rossum3f5da241990-12-20 15:06:42 +0000136 func_dealloc, /*tp_dealloc*/
Guido van Rossum846e4311990-11-18 17:44:06 +0000137 0, /*tp_print*/
Guido van Rossum3f5da241990-12-20 15:06:42 +0000138 func_getattr, /*tp_getattr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000139 0, /*tp_setattr*/
Guido van Rossum2e8f8a31993-11-05 10:20:10 +0000140 func_compare, /*tp_compare*/
Guido van Rossum9bfef441993-03-29 10:43:31 +0000141 func_repr, /*tp_repr*/
Guido van Rossum2e8f8a31993-11-05 10:20:10 +0000142 0, /*tp_as_number*/
143 0, /*tp_as_sequence*/
144 0, /*tp_as_mapping*/
145 func_hash, /*tp_hash*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000146};