mirror of https://github.com/python/cpython
1028 lines
22 KiB
C
1028 lines
22 KiB
C
/***********************************************************
|
|
Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam,
|
|
The Netherlands.
|
|
|
|
All Rights Reserved
|
|
|
|
Permission to use, copy, modify, and distribute this software and its
|
|
documentation for any purpose and without fee is hereby granted,
|
|
provided that the above copyright notice appear in all copies and that
|
|
both that copyright notice and this permission notice appear in
|
|
supporting documentation, and that the names of Stichting Mathematisch
|
|
Centrum or CWI or Corporation for National Research Initiatives or
|
|
CNRI not be used in advertising or publicity pertaining to
|
|
distribution of the software without specific, written prior
|
|
permission.
|
|
|
|
While CWI is the initial source for this software, a modified version
|
|
is made available by the Corporation for National Research Initiatives
|
|
(CNRI) at the Internet address ftp://ftp.python.org.
|
|
|
|
STICHTING MATHEMATISCH CENTRUM AND CNRI DISCLAIM ALL WARRANTIES WITH
|
|
REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF
|
|
MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH
|
|
CENTRUM OR CNRI BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL
|
|
DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
|
|
PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
|
|
TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
******************************************************************/
|
|
|
|
/* Mapping object implementation; using a hash table */
|
|
|
|
/* This file should really be called "dictobject.c", since "mapping"
|
|
is the generic name for objects with an unorderred arbitrary key
|
|
set (just like lists are sequences), but since it improves (and was
|
|
originally derived from) a file by that name I had to change its
|
|
name. For the user these objects are still called "dictionaries". */
|
|
|
|
#include "allobjects.h"
|
|
#include "modsupport.h"
|
|
|
|
|
|
/*
|
|
* MINSIZE is the minimum size of a mapping.
|
|
*/
|
|
|
|
#define MINSIZE 4
|
|
|
|
/*
|
|
Table of irreducible polynomials to efficiently cycle through
|
|
GF(2^n)-{0}, 2<=n<=30.
|
|
*/
|
|
static long polys[] = {
|
|
4 + 3,
|
|
8 + 3,
|
|
16 + 3,
|
|
32 + 5,
|
|
64 + 3,
|
|
128 + 3,
|
|
256 + 29,
|
|
512 + 17,
|
|
1024 + 9,
|
|
2048 + 5,
|
|
4096 + 83,
|
|
8192 + 27,
|
|
16384 + 43,
|
|
32768 + 3,
|
|
65536 + 45,
|
|
131072 + 9,
|
|
262144 + 39,
|
|
524288 + 39,
|
|
1048576 + 9,
|
|
2097152 + 5,
|
|
4194304 + 3,
|
|
8388608 + 33,
|
|
16777216 + 27,
|
|
33554432 + 9,
|
|
67108864 + 71,
|
|
134217728 + 39,
|
|
268435456 + 9,
|
|
536870912 + 5,
|
|
1073741824 + 83,
|
|
0
|
|
};
|
|
|
|
/* Object used as dummy key to fill deleted entries */
|
|
static object *dummy; /* Initialized by first call to newmappingobject() */
|
|
|
|
/*
|
|
Invariant for entries: when in use, de_value is not NULL and de_key is
|
|
not NULL and not dummy; when not in use, de_value is NULL and de_key
|
|
is either NULL or dummy. A dummy key value cannot be replaced by
|
|
NULL, since otherwise other keys may be lost.
|
|
*/
|
|
typedef struct {
|
|
long me_hash;
|
|
object *me_key;
|
|
object *me_value;
|
|
} mappingentry;
|
|
|
|
/*
|
|
To ensure the lookup algorithm terminates, the table size must be a
|
|
prime number and there must be at least one NULL key in the table.
|
|
The value ma_fill is the number of non-NULL keys; ma_used is the number
|
|
of non-NULL, non-dummy keys.
|
|
To avoid slowing down lookups on a near-full table, we resize the table
|
|
when it is more than half filled.
|
|
*/
|
|
typedef struct {
|
|
OB_HEAD
|
|
int ma_fill;
|
|
int ma_used;
|
|
int ma_size;
|
|
int ma_poly;
|
|
mappingentry *ma_table;
|
|
} mappingobject;
|
|
|
|
object *
|
|
newmappingobject()
|
|
{
|
|
register mappingobject *mp;
|
|
if (dummy == NULL) { /* Auto-initialize dummy */
|
|
dummy = newstringobject("<dummy key>");
|
|
if (dummy == NULL)
|
|
return NULL;
|
|
}
|
|
mp = NEWOBJ(mappingobject, &Mappingtype);
|
|
if (mp == NULL)
|
|
return NULL;
|
|
mp->ma_size = 0;
|
|
mp->ma_poly = 0;
|
|
mp->ma_table = NULL;
|
|
mp->ma_fill = 0;
|
|
mp->ma_used = 0;
|
|
return (object *)mp;
|
|
}
|
|
|
|
/*
|
|
The basic lookup function used by all operations.
|
|
This is based on Algorithm D from Knuth Vol. 3, Sec. 6.4.
|
|
Open addressing is preferred over chaining since the link overhead for
|
|
chaining would be substantial (100% with typical malloc overhead).
|
|
However, instead of going through the table at constant steps, we cycle
|
|
through the values of GF(2^n)-{0}. This avoids modulo computations, being
|
|
much cheaper on RISC machines, without leading to clustering.
|
|
|
|
First a 32-bit hash value, 'sum', is computed from the key string.
|
|
The first character is added an extra time shifted by 8 to avoid hashing
|
|
single-character keys (often heavily used variables) too close together.
|
|
All arithmetic on sum should ignore overflow.
|
|
|
|
The initial probe index is then computed as sum mod the table size.
|
|
Subsequent probe indices use the values of x^i in GF(2^n) as an offset,
|
|
where x is a root. The initial value is derived from sum, too.
|
|
|
|
(This version is due to Reimer Behrends, some ideas are also due to
|
|
Jyrki Alakuijala.)
|
|
*/
|
|
static mappingentry *lookmapping PROTO((mappingobject *, object *, long));
|
|
static mappingentry *
|
|
lookmapping(mp, key, hash)
|
|
mappingobject *mp;
|
|
object *key;
|
|
long hash;
|
|
{
|
|
register int i;
|
|
register unsigned incr;
|
|
register unsigned long sum = (unsigned long) hash;
|
|
register mappingentry *freeslot = NULL;
|
|
register int mask = mp->ma_size-1;
|
|
mappingentry *ep0 = mp->ma_table;
|
|
register mappingentry *ep;
|
|
/* We must come up with (i, incr) such that 0 <= i < ma_size
|
|
and 0 < incr < ma_size and both are a function of hash */
|
|
i = (~sum) & mask;
|
|
/* We use ~sum instead if sum, as degenerate hash functions, such
|
|
as for ints <sigh>, can have lots of leading zeros. It's not
|
|
really a performance risk, but better safe than sorry. */
|
|
ep = &ep0[i];
|
|
if (ep->me_key == NULL)
|
|
return ep;
|
|
if (ep->me_key == dummy)
|
|
freeslot = ep;
|
|
else if (ep->me_key == key ||
|
|
(ep->me_hash == hash && cmpobject(ep->me_key, key) == 0)) {
|
|
return ep;
|
|
}
|
|
/* Derive incr from sum, just to make it more arbitrary. Note that
|
|
incr must not be 0, or we will get into an infinite loop.*/
|
|
incr = (sum ^ (sum >> 3)) & mask;
|
|
if (!incr)
|
|
incr = mask;
|
|
if (incr > mask) /* Cycle through GF(2^n)-{0} */
|
|
incr ^= mp->ma_poly; /* This will implicitly clear the
|
|
highest bit */
|
|
for (;;) {
|
|
ep = &ep0[(i+incr)&mask];
|
|
if (ep->me_key == NULL) {
|
|
if (freeslot != NULL)
|
|
return freeslot;
|
|
else
|
|
return ep;
|
|
}
|
|
if (ep->me_key == dummy) {
|
|
if (freeslot == NULL)
|
|
freeslot = ep;
|
|
}
|
|
else if (ep->me_key == key ||
|
|
(ep->me_hash == hash &&
|
|
cmpobject(ep->me_key, key) == 0)) {
|
|
return ep;
|
|
}
|
|
/* Cycle through GF(2^n)-{0} */
|
|
incr = incr << 1;
|
|
if (incr > mask)
|
|
incr ^= mp->ma_poly;
|
|
}
|
|
}
|
|
|
|
/*
|
|
Internal routine to insert a new item into the table.
|
|
Used both by the internal resize routine and by the public insert routine.
|
|
Eats a reference to key and one to value.
|
|
*/
|
|
static void insertmapping PROTO((mappingobject *, object *, long, object *));
|
|
static void
|
|
insertmapping(mp, key, hash, value)
|
|
register mappingobject *mp;
|
|
object *key;
|
|
long hash;
|
|
object *value;
|
|
{
|
|
object *old_value;
|
|
register mappingentry *ep;
|
|
ep = lookmapping(mp, key, hash);
|
|
if (ep->me_value != NULL) {
|
|
old_value = ep->me_value;
|
|
ep->me_value = value;
|
|
DECREF(old_value); /* which **CAN** re-enter */
|
|
DECREF(key);
|
|
}
|
|
else {
|
|
if (ep->me_key == NULL)
|
|
mp->ma_fill++;
|
|
else
|
|
DECREF(ep->me_key);
|
|
ep->me_key = key;
|
|
ep->me_hash = hash;
|
|
ep->me_value = value;
|
|
mp->ma_used++;
|
|
}
|
|
}
|
|
|
|
/*
|
|
Restructure the table by allocating a new table and reinserting all
|
|
items again. When entries have been deleted, the new table may
|
|
actually be smaller than the old one.
|
|
*/
|
|
static int mappingresize PROTO((mappingobject *));
|
|
static int
|
|
mappingresize(mp)
|
|
mappingobject *mp;
|
|
{
|
|
register int oldsize = mp->ma_size;
|
|
register int newsize, newpoly;
|
|
register mappingentry *oldtable = mp->ma_table;
|
|
register mappingentry *newtable;
|
|
register mappingentry *ep;
|
|
register int i;
|
|
newsize = mp->ma_size;
|
|
for (i = 0, newsize = MINSIZE; ; i++, newsize <<= 1) {
|
|
if (i > sizeof(polys)/sizeof(polys[0])) {
|
|
/* Ran out of polynomials */
|
|
err_nomem();
|
|
return -1;
|
|
}
|
|
if (newsize > mp->ma_used*2) {
|
|
newpoly = polys[i];
|
|
break;
|
|
}
|
|
}
|
|
newtable = (mappingentry *) calloc(sizeof(mappingentry), newsize);
|
|
if (newtable == NULL) {
|
|
err_nomem();
|
|
return -1;
|
|
}
|
|
mp->ma_size = newsize;
|
|
mp->ma_poly = newpoly;
|
|
mp->ma_table = newtable;
|
|
mp->ma_fill = 0;
|
|
mp->ma_used = 0;
|
|
|
|
/* Make two passes, so we can avoid decrefs
|
|
(and possible side effects) till the table is copied */
|
|
for (i = 0, ep = oldtable; i < oldsize; i++, ep++) {
|
|
if (ep->me_value != NULL)
|
|
insertmapping(mp,ep->me_key,ep->me_hash,ep->me_value);
|
|
}
|
|
for (i = 0, ep = oldtable; i < oldsize; i++, ep++) {
|
|
if (ep->me_value == NULL)
|
|
XDECREF(ep->me_key);
|
|
}
|
|
|
|
XDEL(oldtable);
|
|
return 0;
|
|
}
|
|
|
|
object *
|
|
mappinglookup(op, key)
|
|
object *op;
|
|
object *key;
|
|
{
|
|
long hash;
|
|
if (!is_mappingobject(op)) {
|
|
err_badcall();
|
|
return NULL;
|
|
}
|
|
if (((mappingobject *)op)->ma_table == NULL)
|
|
return NULL;
|
|
#ifdef CACHE_HASH
|
|
if (!is_stringobject(key) ||
|
|
(hash = ((stringobject *) key)->ob_shash) == -1)
|
|
#endif
|
|
{
|
|
hash = hashobject(key);
|
|
if (hash == -1)
|
|
return NULL;
|
|
}
|
|
return lookmapping((mappingobject *)op, key, hash) -> me_value;
|
|
}
|
|
|
|
int
|
|
mappinginsert(op, key, value)
|
|
register object *op;
|
|
object *key;
|
|
object *value;
|
|
{
|
|
register mappingobject *mp;
|
|
register long hash;
|
|
if (!is_mappingobject(op)) {
|
|
err_badcall();
|
|
return -1;
|
|
}
|
|
mp = (mappingobject *)op;
|
|
#ifdef CACHE_HASH
|
|
if (is_stringobject(key)) {
|
|
#ifdef INTERN_STRINGS
|
|
if (((stringobject *)key)->ob_sinterned != NULL) {
|
|
key = ((stringobject *)key)->ob_sinterned;
|
|
hash = ((stringobject *)key)->ob_shash;
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
hash = ((stringobject *)key)->ob_shash;
|
|
if (hash == -1)
|
|
hash = hashobject(key);
|
|
}
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
hash = hashobject(key);
|
|
if (hash == -1)
|
|
return -1;
|
|
}
|
|
/* if fill >= 2/3 size, resize */
|
|
if (mp->ma_fill*3 >= mp->ma_size*2) {
|
|
if (mappingresize(mp) != 0) {
|
|
if (mp->ma_fill+1 > mp->ma_size)
|
|
return -1;
|
|
}
|
|
}
|
|
INCREF(value);
|
|
INCREF(key);
|
|
insertmapping(mp, key, hash, value);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
mappingremove(op, key)
|
|
object *op;
|
|
object *key;
|
|
{
|
|
register mappingobject *mp;
|
|
register long hash;
|
|
register mappingentry *ep;
|
|
object *old_value, *old_key;
|
|
|
|
if (!is_mappingobject(op)) {
|
|
err_badcall();
|
|
return -1;
|
|
}
|
|
#ifdef CACHE_HASH
|
|
if (!is_stringobject(key) ||
|
|
(hash = ((stringobject *) key)->ob_shash) == -1)
|
|
#endif
|
|
{
|
|
hash = hashobject(key);
|
|
if (hash == -1)
|
|
return -1;
|
|
}
|
|
mp = (mappingobject *)op;
|
|
if (((mappingobject *)op)->ma_table == NULL)
|
|
goto empty;
|
|
ep = lookmapping(mp, key, hash);
|
|
if (ep->me_value == NULL) {
|
|
empty:
|
|
err_setval(KeyError, key);
|
|
return -1;
|
|
}
|
|
old_key = ep->me_key;
|
|
INCREF(dummy);
|
|
ep->me_key = dummy;
|
|
old_value = ep->me_value;
|
|
ep->me_value = NULL;
|
|
mp->ma_used--;
|
|
DECREF(old_value);
|
|
DECREF(old_key);
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
mappingclear(op)
|
|
object *op;
|
|
{
|
|
int i, n;
|
|
register mappingentry *table;
|
|
mappingobject *mp;
|
|
if (!is_mappingobject(op))
|
|
return;
|
|
mp = (mappingobject *)op;
|
|
table = mp->ma_table;
|
|
if (table == NULL)
|
|
return;
|
|
n = mp->ma_size;
|
|
mp->ma_size = mp->ma_used = mp->ma_fill = 0;
|
|
mp->ma_table = NULL;
|
|
for (i = 0; i < n; i++) {
|
|
XDECREF(table[i].me_key);
|
|
XDECREF(table[i].me_value);
|
|
}
|
|
DEL(table);
|
|
}
|
|
|
|
int
|
|
mappinggetnext(op, ppos, pkey, pvalue)
|
|
object *op;
|
|
int *ppos;
|
|
object **pkey;
|
|
object **pvalue;
|
|
{
|
|
int i;
|
|
register mappingobject *mp;
|
|
if (!is_dictobject(op))
|
|
return 0;
|
|
mp = (mappingobject *)op;
|
|
i = *ppos;
|
|
if (i < 0)
|
|
return 0;
|
|
while (i < mp->ma_size && mp->ma_table[i].me_value == NULL)
|
|
i++;
|
|
*ppos = i+1;
|
|
if (i >= mp->ma_size)
|
|
return 0;
|
|
if (pkey)
|
|
*pkey = mp->ma_table[i].me_key;
|
|
if (pvalue)
|
|
*pvalue = mp->ma_table[i].me_value;
|
|
return 1;
|
|
}
|
|
|
|
/* Methods */
|
|
|
|
static void
|
|
mapping_dealloc(mp)
|
|
register mappingobject *mp;
|
|
{
|
|
register int i;
|
|
register mappingentry *ep;
|
|
for (i = 0, ep = mp->ma_table; i < mp->ma_size; i++, ep++) {
|
|
if (ep->me_key != NULL)
|
|
DECREF(ep->me_key);
|
|
if (ep->me_value != NULL)
|
|
DECREF(ep->me_value);
|
|
}
|
|
XDEL(mp->ma_table);
|
|
DEL(mp);
|
|
}
|
|
|
|
static int
|
|
mapping_print(mp, fp, flags)
|
|
register mappingobject *mp;
|
|
register FILE *fp;
|
|
register int flags;
|
|
{
|
|
register int i;
|
|
register int any;
|
|
register mappingentry *ep;
|
|
fprintf(fp, "{");
|
|
any = 0;
|
|
for (i = 0, ep = mp->ma_table; i < mp->ma_size; i++, ep++) {
|
|
if (ep->me_value != NULL) {
|
|
if (any++ > 0)
|
|
fprintf(fp, ", ");
|
|
if (printobject((object *)ep->me_key, fp, 0) != 0)
|
|
return -1;
|
|
fprintf(fp, ": ");
|
|
if (printobject(ep->me_value, fp, 0) != 0)
|
|
return -1;
|
|
}
|
|
}
|
|
fprintf(fp, "}");
|
|
return 0;
|
|
}
|
|
|
|
static object *
|
|
mapping_repr(mp)
|
|
mappingobject *mp;
|
|
{
|
|
auto object *v;
|
|
object *sepa, *colon;
|
|
register int i;
|
|
register int any;
|
|
register mappingentry *ep;
|
|
v = newstringobject("{");
|
|
sepa = newstringobject(", ");
|
|
colon = newstringobject(": ");
|
|
any = 0;
|
|
for (i = 0, ep = mp->ma_table; i < mp->ma_size && v; i++, ep++) {
|
|
if (ep->me_value != NULL) {
|
|
if (any++)
|
|
joinstring(&v, sepa);
|
|
joinstring_decref(&v, reprobject(ep->me_key));
|
|
joinstring(&v, colon);
|
|
joinstring_decref(&v, reprobject(ep->me_value));
|
|
}
|
|
}
|
|
joinstring_decref(&v, newstringobject("}"));
|
|
XDECREF(sepa);
|
|
XDECREF(colon);
|
|
return v;
|
|
}
|
|
|
|
static int
|
|
mapping_length(mp)
|
|
mappingobject *mp;
|
|
{
|
|
return mp->ma_used;
|
|
}
|
|
|
|
static object *
|
|
mapping_subscript(mp, key)
|
|
mappingobject *mp;
|
|
register object *key;
|
|
{
|
|
object *v;
|
|
long hash;
|
|
if (mp->ma_table == NULL) {
|
|
err_setval(KeyError, key);
|
|
return NULL;
|
|
}
|
|
#ifdef CACHE_HASH
|
|
if (!is_stringobject(key) ||
|
|
(hash = ((stringobject *) key)->ob_shash) == -1)
|
|
#endif
|
|
{
|
|
hash = hashobject(key);
|
|
if (hash == -1)
|
|
return NULL;
|
|
}
|
|
v = lookmapping(mp, key, hash) -> me_value;
|
|
if (v == NULL)
|
|
err_setval(KeyError, key);
|
|
else
|
|
INCREF(v);
|
|
return v;
|
|
}
|
|
|
|
static int
|
|
mapping_ass_sub(mp, v, w)
|
|
mappingobject *mp;
|
|
object *v, *w;
|
|
{
|
|
if (w == NULL)
|
|
return mappingremove((object *)mp, v);
|
|
else
|
|
return mappinginsert((object *)mp, v, w);
|
|
}
|
|
|
|
static mapping_methods mapping_as_mapping = {
|
|
(inquiry)mapping_length, /*mp_length*/
|
|
(binaryfunc)mapping_subscript, /*mp_subscript*/
|
|
(objobjargproc)mapping_ass_sub, /*mp_ass_subscript*/
|
|
};
|
|
|
|
static object *
|
|
mapping_keys(mp, args)
|
|
register mappingobject *mp;
|
|
object *args;
|
|
{
|
|
register object *v;
|
|
register int i, j;
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
v = newlistobject(mp->ma_used);
|
|
if (v == NULL)
|
|
return NULL;
|
|
for (i = 0, j = 0; i < mp->ma_size; i++) {
|
|
if (mp->ma_table[i].me_value != NULL) {
|
|
object *key = mp->ma_table[i].me_key;
|
|
INCREF(key);
|
|
setlistitem(v, j, key);
|
|
j++;
|
|
}
|
|
}
|
|
return v;
|
|
}
|
|
|
|
static object *
|
|
mapping_values(mp, args)
|
|
register mappingobject *mp;
|
|
object *args;
|
|
{
|
|
register object *v;
|
|
register int i, j;
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
v = newlistobject(mp->ma_used);
|
|
if (v == NULL)
|
|
return NULL;
|
|
for (i = 0, j = 0; i < mp->ma_size; i++) {
|
|
if (mp->ma_table[i].me_value != NULL) {
|
|
object *value = mp->ma_table[i].me_value;
|
|
INCREF(value);
|
|
setlistitem(v, j, value);
|
|
j++;
|
|
}
|
|
}
|
|
return v;
|
|
}
|
|
|
|
static object *
|
|
mapping_items(mp, args)
|
|
register mappingobject *mp;
|
|
object *args;
|
|
{
|
|
register object *v;
|
|
register int i, j;
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
v = newlistobject(mp->ma_used);
|
|
if (v == NULL)
|
|
return NULL;
|
|
for (i = 0, j = 0; i < mp->ma_size; i++) {
|
|
if (mp->ma_table[i].me_value != NULL) {
|
|
object *key = mp->ma_table[i].me_key;
|
|
object *value = mp->ma_table[i].me_value;
|
|
object *item = newtupleobject(2);
|
|
if (item == NULL) {
|
|
DECREF(v);
|
|
return NULL;
|
|
}
|
|
INCREF(key);
|
|
settupleitem(item, 0, key);
|
|
INCREF(value);
|
|
settupleitem(item, 1, value);
|
|
setlistitem(v, j, item);
|
|
j++;
|
|
}
|
|
}
|
|
return v;
|
|
}
|
|
|
|
int
|
|
getmappingsize(mp)
|
|
object *mp;
|
|
{
|
|
if (mp == NULL || !is_mappingobject(mp)) {
|
|
err_badcall();
|
|
return 0;
|
|
}
|
|
return ((mappingobject *)mp)->ma_used;
|
|
}
|
|
|
|
object *
|
|
getmappingkeys(mp)
|
|
object *mp;
|
|
{
|
|
if (mp == NULL || !is_mappingobject(mp)) {
|
|
err_badcall();
|
|
return NULL;
|
|
}
|
|
return mapping_keys((mappingobject *)mp, (object *)NULL);
|
|
}
|
|
|
|
object *
|
|
getmappingvalues(mp)
|
|
object *mp;
|
|
{
|
|
if (mp == NULL || !is_mappingobject(mp)) {
|
|
err_badcall();
|
|
return NULL;
|
|
}
|
|
return mapping_values((mappingobject *)mp, (object *)NULL);
|
|
}
|
|
|
|
object *
|
|
getmappingitems(mp)
|
|
object *mp;
|
|
{
|
|
if (mp == NULL || !is_mappingobject(mp)) {
|
|
err_badcall();
|
|
return NULL;
|
|
}
|
|
return mapping_items((mappingobject *)mp, (object *)NULL);
|
|
}
|
|
|
|
#define NEWCMP
|
|
|
|
#ifdef NEWCMP
|
|
|
|
/* Subroutine which returns the smallest key in a for which b's value
|
|
is different or absent. The value is returned too, through the
|
|
pval argument. No reference counts are incremented. */
|
|
|
|
static object *
|
|
characterize(a, b, pval)
|
|
mappingobject *a;
|
|
mappingobject *b;
|
|
object **pval;
|
|
{
|
|
object *diff = NULL;
|
|
int i;
|
|
|
|
*pval = NULL;
|
|
for (i = 0; i < a->ma_size; i++) {
|
|
if (a->ma_table[i].me_value != NULL) {
|
|
object *key = a->ma_table[i].me_key;
|
|
object *aval, *bval;
|
|
if (diff != NULL && cmpobject(key, diff) > 0)
|
|
continue;
|
|
aval = a->ma_table[i].me_value;
|
|
bval = mappinglookup((object *)b, key);
|
|
if (bval == NULL || cmpobject(aval, bval) != 0) {
|
|
diff = key;
|
|
*pval = aval;
|
|
}
|
|
}
|
|
}
|
|
return diff;
|
|
}
|
|
|
|
static int
|
|
mapping_compare(a, b)
|
|
mappingobject *a, *b;
|
|
{
|
|
object *adiff, *bdiff, *aval, *bval;
|
|
int res;
|
|
|
|
/* Compare lengths first */
|
|
if (a->ma_used < b->ma_used)
|
|
return -1; /* a is shorter */
|
|
else if (a->ma_used > b->ma_used)
|
|
return 1; /* b is shorter */
|
|
/* Same length -- check all keys */
|
|
adiff = characterize(a, b, &aval);
|
|
if (adiff == NULL)
|
|
return 0; /* a is a subset with the same length */
|
|
bdiff = characterize(b, a, &bval);
|
|
/* bdiff == NULL would be impossible now */
|
|
res = cmpobject(adiff, bdiff);
|
|
if (res == 0)
|
|
res = cmpobject(aval, bval);
|
|
return res;
|
|
}
|
|
|
|
#else /* !NEWCMP */
|
|
|
|
static int
|
|
mapping_compare(a, b)
|
|
mappingobject *a, *b;
|
|
{
|
|
object *akeys, *bkeys;
|
|
int i, n, res;
|
|
if (a == b)
|
|
return 0;
|
|
if (a->ma_used == 0) {
|
|
if (b->ma_used != 0)
|
|
return -1;
|
|
else
|
|
return 0;
|
|
}
|
|
else {
|
|
if (b->ma_used == 0)
|
|
return 1;
|
|
}
|
|
akeys = mapping_keys(a, (object *)NULL);
|
|
bkeys = mapping_keys(b, (object *)NULL);
|
|
if (akeys == NULL || bkeys == NULL) {
|
|
/* Oops, out of memory -- what to do? */
|
|
/* For now, sort on address! */
|
|
XDECREF(akeys);
|
|
XDECREF(bkeys);
|
|
if (a < b)
|
|
return -1;
|
|
else
|
|
return 1;
|
|
}
|
|
sortlist(akeys);
|
|
sortlist(bkeys);
|
|
n = a->ma_used < b->ma_used ? a->ma_used : b->ma_used; /* smallest */
|
|
res = 0;
|
|
for (i = 0; i < n; i++) {
|
|
object *akey, *bkey, *aval, *bval;
|
|
long ahash, bhash;
|
|
akey = getlistitem(akeys, i);
|
|
bkey = getlistitem(bkeys, i);
|
|
res = cmpobject(akey, bkey);
|
|
if (res != 0)
|
|
break;
|
|
#ifdef CACHE_HASH
|
|
if (!is_stringobject(akey) ||
|
|
(ahash = ((stringobject *) akey)->ob_shash) == -1)
|
|
#endif
|
|
{
|
|
ahash = hashobject(akey);
|
|
if (ahash == -1)
|
|
err_clear(); /* Don't want errors here */
|
|
}
|
|
#ifdef CACHE_HASH
|
|
if (!is_stringobject(bkey) ||
|
|
(bhash = ((stringobject *) bkey)->ob_shash) == -1)
|
|
#endif
|
|
{
|
|
bhash = hashobject(bkey);
|
|
if (bhash == -1)
|
|
err_clear(); /* Don't want errors here */
|
|
}
|
|
aval = lookmapping(a, akey, ahash) -> me_value;
|
|
bval = lookmapping(b, bkey, bhash) -> me_value;
|
|
res = cmpobject(aval, bval);
|
|
if (res != 0)
|
|
break;
|
|
}
|
|
if (res == 0) {
|
|
if (a->ma_used < b->ma_used)
|
|
res = -1;
|
|
else if (a->ma_used > b->ma_used)
|
|
res = 1;
|
|
}
|
|
DECREF(akeys);
|
|
DECREF(bkeys);
|
|
return res;
|
|
}
|
|
|
|
#endif /* !NEWCMP */
|
|
|
|
static object *
|
|
mapping_has_key(mp, args)
|
|
register mappingobject *mp;
|
|
object *args;
|
|
{
|
|
object *key;
|
|
long hash;
|
|
register long ok;
|
|
if (!getargs(args, "O", &key))
|
|
return NULL;
|
|
#ifdef CACHE_HASH
|
|
if (!is_stringobject(key) ||
|
|
(hash = ((stringobject *) key)->ob_shash) == -1)
|
|
#endif
|
|
{
|
|
hash = hashobject(key);
|
|
if (hash == -1)
|
|
return NULL;
|
|
}
|
|
ok = mp->ma_size != 0 && lookmapping(mp, key, hash)->me_value != NULL;
|
|
return newintobject(ok);
|
|
}
|
|
|
|
static object *
|
|
mapping_clear(mp, args)
|
|
register mappingobject *mp;
|
|
object *args;
|
|
{
|
|
if (!getnoarg(args))
|
|
return NULL;
|
|
mappingclear((object *)mp);
|
|
INCREF(None);
|
|
return None;
|
|
}
|
|
|
|
static struct methodlist mapp_methods[] = {
|
|
{"clear", (method)mapping_clear},
|
|
{"has_key", (method)mapping_has_key},
|
|
{"items", (method)mapping_items},
|
|
{"keys", (method)mapping_keys},
|
|
{"values", (method)mapping_values},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
static object *
|
|
mapping_getattr(mp, name)
|
|
mappingobject *mp;
|
|
char *name;
|
|
{
|
|
return findmethod(mapp_methods, (object *)mp, name);
|
|
}
|
|
|
|
typeobject Mappingtype = {
|
|
OB_HEAD_INIT(&Typetype)
|
|
0,
|
|
"dictionary",
|
|
sizeof(mappingobject),
|
|
0,
|
|
(destructor)mapping_dealloc, /*tp_dealloc*/
|
|
(printfunc)mapping_print, /*tp_print*/
|
|
(getattrfunc)mapping_getattr, /*tp_getattr*/
|
|
0, /*tp_setattr*/
|
|
(cmpfunc)mapping_compare, /*tp_compare*/
|
|
(reprfunc)mapping_repr, /*tp_repr*/
|
|
0, /*tp_as_number*/
|
|
0, /*tp_as_sequence*/
|
|
&mapping_as_mapping, /*tp_as_mapping*/
|
|
};
|
|
|
|
/* For backward compatibility with old dictionary interface */
|
|
|
|
static object *last_name_object;
|
|
static char *last_name_char; /* NULL or == getstringvalue(last_name_object) */
|
|
|
|
object *
|
|
getattro(v, name)
|
|
object *v;
|
|
object *name;
|
|
{
|
|
if (v->ob_type->tp_getattro != NULL)
|
|
return (*v->ob_type->tp_getattro)(v, name);
|
|
|
|
if (name != last_name_object) {
|
|
XDECREF(last_name_object);
|
|
INCREF(name);
|
|
last_name_object = name;
|
|
last_name_char = getstringvalue(name);
|
|
}
|
|
return getattr(v, last_name_char);
|
|
}
|
|
|
|
int
|
|
setattro(v, name, value)
|
|
object *v;
|
|
object *name;
|
|
object *value;
|
|
{
|
|
int err;
|
|
INCREF(name);
|
|
PyString_InternInPlace(&name);
|
|
if (v->ob_type->tp_setattro != NULL)
|
|
err = (*v->ob_type->tp_setattro)(v, name, value);
|
|
else {
|
|
if (name != last_name_object) {
|
|
XDECREF(last_name_object);
|
|
INCREF(name);
|
|
last_name_object = name;
|
|
last_name_char = getstringvalue(name);
|
|
}
|
|
err = setattr(v, last_name_char, value);
|
|
}
|
|
DECREF(name);
|
|
return err;
|
|
}
|
|
|
|
object *
|
|
dictlookup(v, key)
|
|
object *v;
|
|
char *key;
|
|
{
|
|
if (key != last_name_char) {
|
|
XDECREF(last_name_object);
|
|
last_name_object = newstringobject(key);
|
|
if (last_name_object == NULL) {
|
|
last_name_char = NULL;
|
|
return NULL;
|
|
}
|
|
PyString_InternInPlace(&last_name_object);
|
|
last_name_char = getstringvalue(last_name_object);
|
|
}
|
|
return mappinglookup(v, last_name_object);
|
|
}
|
|
|
|
int
|
|
dictinsert(v, key, item)
|
|
object *v;
|
|
char *key;
|
|
object *item;
|
|
{
|
|
if (key != last_name_char) {
|
|
XDECREF(last_name_object);
|
|
last_name_object = newstringobject(key);
|
|
if (last_name_object == NULL) {
|
|
last_name_char = NULL;
|
|
return -1;
|
|
}
|
|
PyString_InternInPlace(&last_name_object);
|
|
last_name_char = getstringvalue(last_name_object);
|
|
}
|
|
return mappinginsert(v, last_name_object, item);
|
|
}
|
|
|
|
int
|
|
dictremove(v, key)
|
|
object *v;
|
|
char *key;
|
|
{
|
|
if (key != last_name_char) {
|
|
XDECREF(last_name_object);
|
|
last_name_object = newstringobject(key);
|
|
if (last_name_object == NULL) {
|
|
last_name_char = NULL;
|
|
return -1;
|
|
}
|
|
last_name_char = getstringvalue(last_name_object);
|
|
}
|
|
return mappingremove(v, last_name_object);
|
|
}
|