X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fdata%2Fdictionary.c;h=a5a36527c3d2803d48a686f02e82070c078de42c;hb=72147c12357446b304264b6944b8a1d92c45f53c;hp=161071ec159a038555da3467ef8586a07a143fa3;hpb=698e31dd616553a663d0b04f24d322ac47315534;p=pspp-builds.git diff --git a/src/data/dictionary.c b/src/data/dictionary.c index 161071ec..a5a36527 100644 --- a/src/data/dictionary.c +++ b/src/data/dictionary.c @@ -1,6 +1,5 @@ /* PSPP - computes sample statistics. Copyright (C) 1997-9, 2000, 2006 Free Software Foundation, Inc. - Written by Ben Pfaff . This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as @@ -29,15 +28,20 @@ #include "category.h" #include "settings.h" #include "value-labels.h" +#include "vardict.h" #include "variable.h" +#include "vector.h" #include #include #include #include #include #include +#include #include +#include "minmax.h" + #include "gettext.h" #define _(msgid) gettext (msgid) @@ -57,20 +61,37 @@ struct dictionary char *documents; /* Documents, as a string. */ struct vector **vector; /* Vectors of variables. */ size_t vector_cnt; /* Number of vectors. */ + const struct dict_callbacks *callbacks; /* Callbacks on dictionary + modification */ + void *cb_data ; /* Data passed to callbacks */ }; -/* Active file dictionary. */ -struct dictionary *default_dict; + +/* Associate CALLBACKS with DICT. Callbacks will be invoked whenever + the dictionary or any of the variables it contains are modified. + Each callback will get passed CALLBACK_DATA. + Any callback may be NULL, in which case it'll be ignored. +*/ +void +dict_set_callbacks (struct dictionary *dict, + const struct dict_callbacks *callbacks, + void *callback_data) +{ + dict->callbacks = callbacks; + dict->cb_data = callback_data; +} + /* Creates and returns a new dictionary. */ struct dictionary * -dict_create (void) +dict_create (void) { - struct dictionary *d = xmalloc (sizeof *d); - + struct dictionary *d = xzalloc (sizeof *d); + d->var = NULL; d->var_cnt = d->var_cap = 0; - d->name_tab = hsh_create (8, compare_var_names, hash_var_name, NULL, NULL); + d->name_tab = hsh_create (8, compare_vars_by_name, hash_var_by_name, + NULL, NULL); d->next_value_idx = 0; d->split = NULL; d->split_cnt = 0; @@ -88,7 +109,7 @@ dict_create (void) /* Creates and returns a (deep) copy of an existing dictionary. */ struct dictionary * -dict_clone (const struct dictionary *s) +dict_clone (const struct dictionary *s) { struct dictionary *d; size_t i; @@ -97,28 +118,28 @@ dict_clone (const struct dictionary *s) d = dict_create (); - for (i = 0; i < s->var_cnt; i++) + for (i = 0; i < s->var_cnt; i++) { struct variable *sv = s->var[i]; - struct variable *dv = dict_clone_var_assert (d, sv, sv->name); - var_set_short_name (dv, sv->short_name); + struct variable *dv = dict_clone_var_assert (d, sv, var_get_name (sv)); + var_set_short_name (dv, var_get_short_name (sv)); } d->next_value_idx = s->next_value_idx; d->split_cnt = s->split_cnt; - if (d->split_cnt > 0) + if (d->split_cnt > 0) { d->split = xnmalloc (d->split_cnt, sizeof *d->split); - for (i = 0; i < d->split_cnt; i++) - d->split[i] = dict_lookup_var_assert (d, s->split[i]->name); + for (i = 0; i < d->split_cnt; i++) + d->split[i] = dict_lookup_var_assert (d, var_get_name (s->split[i])); } - if (s->weight != NULL) - d->weight = dict_lookup_var_assert (d, s->weight->name); + if (s->weight != NULL) + d->weight = dict_lookup_var_assert (d, var_get_name (s->weight)); - if (s->filter != NULL) - d->filter = dict_lookup_var_assert (d, s->filter->name); + if (s->filter != NULL) + d->filter = dict_lookup_var_assert (d, var_get_name (s->filter)); d->case_limit = s->case_limit; dict_set_label (d, dict_get_label (s)); @@ -126,19 +147,8 @@ dict_clone (const struct dictionary *s) d->vector_cnt = s->vector_cnt; d->vector = xnmalloc (d->vector_cnt, sizeof *d->vector); - for (i = 0; i < s->vector_cnt; i++) - { - struct vector *sv = s->vector[i]; - struct vector *dv = d->vector[i] = xmalloc (sizeof *dv); - int j; - - dv->idx = i; - strcpy (dv->name, sv->name); - dv->cnt = sv->cnt; - dv->var = xnmalloc (dv->cnt, sizeof *dv->var); - for (j = 0; j < dv->cnt; j++) - dv->var[j] = d->var[sv->var[j]->index]; - } + for (i = 0; i < s->vector_cnt; i++) + d->vector[i] = vector_clone (s->vector[i], s, d); return d; } @@ -146,7 +156,7 @@ dict_clone (const struct dictionary *s) /* Clears the contents from a dictionary without destroying the dictionary itself. */ void -dict_clear (struct dictionary *d) +dict_clear (struct dictionary *d) { /* FIXME? Should we really clear case_limit, label, documents? Others are necessarily cleared by deleting all the variables.*/ @@ -154,13 +164,10 @@ dict_clear (struct dictionary *d) assert (d != NULL); - for (i = 0; i < d->var_cnt; i++) + for (i = 0; i < d->var_cnt; i++) { - struct variable *v = d->var[i]; - var_clear_aux (v); - val_labs_destroy (v->val_labs); - free (v->label); - free (v); + var_clear_vardict (d->var[i]); + var_destroy (d->var[i]); } free (d->var); d->var = NULL; @@ -183,12 +190,12 @@ dict_clear (struct dictionary *d) /* Destroys the aux data for every variable in D, by calling var_clear_aux() for each variable. */ void -dict_clear_aux (struct dictionary *d) +dict_clear_aux (struct dictionary *d) { int i; - + assert (d != NULL); - + for (i = 0; i < d->var_cnt; i++) var_clear_aux (d->var[i]); } @@ -197,7 +204,7 @@ dict_clear_aux (struct dictionary *d) void dict_destroy (struct dictionary *d) { - if (d != NULL) + if (d != NULL) { dict_clear (d); hsh_destroy (d->name_tab); @@ -207,18 +214,18 @@ dict_destroy (struct dictionary *d) /* Returns the number of variables in D. */ size_t -dict_get_var_cnt (const struct dictionary *d) +dict_get_var_cnt (const struct dictionary *d) { assert (d != NULL); return d->var_cnt; } -/* Returns the variable in D with index IDX, which must be - between 0 and the count returned by dict_get_var_cnt(), - exclusive. */ +/* Returns the variable in D with dictionary index IDX, which + must be between 0 and the count returned by + dict_get_var_cnt(), exclusive. */ struct variable * -dict_get_var (const struct dictionary *d, size_t idx) +dict_get_var (const struct dictionary *d, size_t idx) { assert (d != NULL); assert (idx < d->var_cnt); @@ -237,156 +244,107 @@ dict_get_vars (const struct dictionary *d, struct variable ***vars, { size_t count; size_t i; - + assert (d != NULL); assert (vars != NULL); assert (cnt != NULL); assert ((exclude_classes & ~((1u << DC_ORDINARY) | (1u << DC_SYSTEM) | (1u << DC_SCRATCH))) == 0); - + count = 0; for (i = 0; i < d->var_cnt; i++) - if (!(exclude_classes & (1u << dict_class_from_id (d->var[i]->name)))) - count++; + { + enum dict_class class = dict_class_from_id (var_get_name (d->var[i])); + if (!(exclude_classes & (1u << class))) + count++; + } *vars = xnmalloc (count, sizeof **vars); *cnt = 0; for (i = 0; i < d->var_cnt; i++) - if (!(exclude_classes & (1u << dict_class_from_id (d->var[i]->name)))) - (*vars)[(*cnt)++] = d->var[i]; + { + enum dict_class class = dict_class_from_id (var_get_name (d->var[i])); + if (!(exclude_classes & (1u << class))) + (*vars)[(*cnt)++] = d->var[i]; + } assert (*cnt == count); } - -/* Creates and returns a new variable in D with the given NAME - and WIDTH. Returns a null pointer if the given NAME would - duplicate that of an existing variable in the dictionary. */ -struct variable * -dict_create_var (struct dictionary *d, const char *name, int width) +static struct variable * +add_var (struct dictionary *d, struct variable *v) { - struct variable *v; - - assert (d != NULL); - assert (name != NULL); - - assert (width >= 0 && width < 256); - - assert (var_is_plausible_name(name,0)); - - /* Make sure there's not already a variable by that name. */ - if (dict_lookup_var (d, name) != NULL) - return NULL; - - /* Allocate and initialize variable. */ - v = xmalloc (sizeof *v); - str_copy_trunc (v->name, sizeof v->name, name); - v->type = width == 0 ? NUMERIC : ALPHA; - v->width = width; - v->fv = d->next_value_idx; - v->nv = width == 0 ? 1 : DIV_RND_UP (width, 8); - v->leave = dict_class_from_id (v->name) == DC_SCRATCH; - v->index = d->var_cnt; - mv_init (&v->miss, width); - if (v->type == NUMERIC) - { - v->print = f8_2; - v->alignment = ALIGN_RIGHT; - v->display_width = 8; - v->measure = MEASURE_SCALE; - } - else - { - v->print = make_output_format (FMT_A, v->width, 0); - v->alignment = ALIGN_LEFT; - v->display_width = 8; - v->measure = MEASURE_NOMINAL; - } - v->write = v->print; - v->val_labs = val_labs_create (v->width); - v->label = NULL; - var_clear_short_name (v); - v->aux = NULL; - v->aux_dtor = NULL; - v->obs_vals = NULL; + /* Add dictionary info to variable. */ + struct vardict_info vdi; + vdi.case_index = d->next_value_idx; + vdi.dict_index = d->var_cnt; + vdi.dict = d; + var_set_vardict (v, &vdi); /* Update dictionary. */ - if (d->var_cnt >= d->var_cap) + if (d->var_cnt >= d->var_cap) { - d->var_cap = 8 + 2 * d->var_cap; + d->var_cap = 8 + 2 * d->var_cap; d->var = xnrealloc (d->var, d->var_cap, sizeof *d->var); } - d->var[v->index] = v; - d->var_cnt++; + d->var[d->var_cnt++] = v; hsh_force_insert (d->name_tab, v); - d->next_value_idx += v->nv; + if ( d->callbacks && d->callbacks->var_added ) + d->callbacks->var_added (d, d->next_value_idx, d->cb_data); + + d->next_value_idx += var_get_value_cnt (v); return v; } +/* Creates and returns a new variable in D with the given NAME + and WIDTH. Returns a null pointer if the given NAME would + duplicate that of an existing variable in the dictionary. */ +struct variable * +dict_create_var (struct dictionary *d, const char *name, int width) +{ + return (dict_lookup_var (d, name) == NULL + ? dict_create_var_assert (d, name, width) + : NULL); +} + /* Creates and returns a new variable in D with the given NAME and WIDTH. Assert-fails if the given NAME would duplicate that of an existing variable in the dictionary. */ struct variable * dict_create_var_assert (struct dictionary *d, const char *name, int width) { - struct variable *v = dict_create_var (d, name, width); - assert (v != NULL); - return v; + assert (dict_lookup_var (d, name) == NULL); + return add_var (d, var_create (name, width)); } /* Creates and returns a new variable in D with name NAME, as a - copy of existing variable OV, which need not be in D or in any - dictionary. Returns a null pointer if the given NAME would - duplicate that of an existing variable in the dictionary. */ + copy of existing variable OLD_VAR, which need not be in D or + in any dictionary. Returns a null pointer if the given NAME + would duplicate that of an existing variable in the + dictionary. */ struct variable * -dict_clone_var (struct dictionary *d, const struct variable *ov, +dict_clone_var (struct dictionary *d, const struct variable *old_var, const char *name) { - struct variable *nv; - - assert (d != NULL); - assert (ov != NULL); - assert (name != NULL); - - assert (strlen (name) >= 1); - assert (strlen (name) <= LONG_NAME_LEN); - - nv = dict_create_var (d, name, ov->width); - if (nv == NULL) - return NULL; - - /* Copy most members not copied via dict_create_var(). - short_name[] is intentionally not copied, because there is - no reason to give a new variable with potentially a new name - the same short name. */ - nv->leave = ov->leave; - mv_copy (&nv->miss, &ov->miss); - nv->print = ov->print; - nv->write = ov->write; - val_labs_destroy (nv->val_labs); - nv->val_labs = val_labs_copy (ov->val_labs); - if (ov->label != NULL) - nv->label = xstrdup (ov->label); - nv->measure = ov->measure; - nv->display_width = ov->display_width; - nv->alignment = ov->alignment; - - return nv; + return (dict_lookup_var (d, name) == NULL + ? dict_clone_var_assert (d, old_var, name) + : NULL); } /* Creates and returns a new variable in D with name NAME, as a - copy of existing variable OV, which need not be in D or in any - dictionary. Assert-fails if the given NAME would duplicate - that of an existing variable in the dictionary. */ + copy of existing variable OLD_VAR, which need not be in D or + in any dictionary. Assert-fails if the given NAME would + duplicate that of an existing variable in the dictionary. */ struct variable * -dict_clone_var_assert (struct dictionary *d, const struct variable *ov, +dict_clone_var_assert (struct dictionary *d, const struct variable *old_var, const char *name) { - struct variable *v = dict_clone_var (d, ov, name); - assert (v != NULL); - return v; + struct variable *new_var = var_clone (old_var); + assert (dict_lookup_var (d, name) == NULL); + var_set_name (new_var, name); + return add_var (d, new_var); } /* Returns the variable named NAME in D, or a null pointer if no @@ -394,13 +352,10 @@ dict_clone_var_assert (struct dictionary *d, const struct variable *ov, struct variable * dict_lookup_var (const struct dictionary *d, const char *name) { - struct variable v; - - assert (d != NULL); - assert (name != NULL); - - str_copy_trunc (v.name, sizeof v.name, name); - return hsh_find (d->name_tab, &v); + struct variable *target = var_create (name, 0); + struct variable *result = hsh_find (d->name_tab, target); + var_destroy (target); + return result; } /* Returns the variable named NAME in D. Assert-fails if no @@ -418,16 +373,21 @@ dict_lookup_var_assert (const struct dictionary *d, const char *name) bool dict_contains_var (const struct dictionary *d, const struct variable *v) { - assert (d != NULL); - assert (v != NULL); - - return v->index >= 0 && v->index < d->var_cnt && d->var[v->index] == v; + if (var_has_vardict (v)) + { + const struct vardict_info *vdi = var_get_vardict (v); + return (vdi->dict_index >= 0 + && vdi->dict_index < d->var_cnt + && d->var[vdi->dict_index] == v); + } + else + return false; } /* Compares two double pointers to variables, which should point to elements of a struct dictionary's `var' member array. */ static int -compare_var_ptrs (const void *a_, const void *b_, void *aux UNUSED) +compare_var_ptrs (const void *a_, const void *b_, const void *aux UNUSED) { struct variable *const *a = a_; struct variable *const *b = b_; @@ -435,11 +395,40 @@ compare_var_ptrs (const void *a_, const void *b_, void *aux UNUSED) return *a < *b ? -1 : *a > *b; } +/* Sets the dict_index in V's vardict to DICT_INDEX. */ +static void +set_var_dict_index (struct variable *v, int dict_index) +{ + struct vardict_info vdi = *var_get_vardict (v); + vdi.dict_index = dict_index; + var_set_vardict (v, &vdi); +} + +/* Sets the case_index in V's vardict to DICT_INDEX. */ +static void +set_var_case_index (struct variable *v, int case_index) +{ + struct vardict_info vdi = *var_get_vardict (v); + vdi.case_index = case_index; + var_set_vardict (v, &vdi); +} + +/* Re-sets the dict_index in the dictionary variables with + indexes from FROM to TO (exclusive). */ +static void +reindex_vars (struct dictionary *d, size_t from, size_t to) +{ + size_t i; + + for (i = from; i < to; i++) + set_var_dict_index (d->var[i], i); +} + /* Deletes variable V from dictionary D and frees V. This is a very bad idea if there might be any pointers to V - from outside D. In general, no variable in default_dict - should be deleted when any transformations are active, because + from outside D. In general, no variable in should be deleted when + any transformations are active on the dictionary's dataset, because those transformations might reference the deleted variable. The safest time to delete a variable is just after a procedure has been executed, as done by MODIFY VARS. @@ -450,10 +439,8 @@ compare_var_ptrs (const void *a_, const void *b_, void *aux UNUSED) void dict_delete_var (struct dictionary *d, struct variable *v) { - size_t i; + int dict_index = var_get_dict_index (v); - assert (d != NULL); - assert (v != NULL); assert (dict_contains_var (d, v)); /* Delete aux data. */ @@ -469,21 +456,22 @@ dict_delete_var (struct dictionary *d, struct variable *v) dict_clear_vectors (d); /* Remove V from var array. */ - remove_element (d->var, d->var_cnt, sizeof *d->var, v->index); + remove_element (d->var, d->var_cnt, sizeof *d->var, dict_index); d->var_cnt--; - /* Update index. */ - for (i = v->index; i < d->var_cnt; i++) - d->var[i]->index = i; + /* Update dict_index for each affected variable. */ + reindex_vars (d, dict_index, d->var_cnt); /* Update name hash. */ hsh_force_delete (d->name_tab, v); + /* Free memory. */ - val_labs_destroy (v->val_labs); - cat_stored_values_destroy (v); - free (v->label); - free (v); + var_clear_vardict (v); + var_destroy (v); + + if (d->callbacks && d->callbacks->var_deleted ) + d->callbacks->var_deleted (d, dict_index, d->cb_data); } /* Deletes the COUNT variables listed in VARS from D. This is @@ -501,6 +489,20 @@ dict_delete_vars (struct dictionary *d, dict_delete_var (d, *vars++); } +/* Deletes the COUNT variables in D starting at index IDX. This + is unsafe; see the comment on dict_delete_var() for + details. */ +void +dict_delete_consecutive_vars (struct dictionary *d, size_t idx, size_t count) +{ + /* FIXME: this can be done in O(count) time, but this algorithm + is O(count**2). */ + assert (idx + count <= d->var_cnt); + + while (count-- > 0) + dict_delete_var (d, d->var[idx]); +} + /* Deletes scratch variables from dictionary D. */ void dict_delete_scratch_vars (struct dictionary *d) @@ -512,7 +514,7 @@ dict_delete_scratch_vars (struct dictionary *d) assert (d != NULL); for (i = 0; i < d->var_cnt; ) - if (dict_class_from_id (d->var[i]->name) == DC_SCRATCH) + if (dict_class_from_id (var_get_name (d->var[i])) == DC_SCRATCH) dict_delete_var (d, d->var[i]); else i++; @@ -522,23 +524,13 @@ dict_delete_scratch_vars (struct dictionary *d) if any, retain their relative positions. Runs in time linear in the distance moved. */ void -dict_reorder_var (struct dictionary *d, struct variable *v, - size_t new_index) +dict_reorder_var (struct dictionary *d, struct variable *v, size_t new_index) { - size_t min_idx, max_idx; - size_t i; - - assert (d != NULL); - assert (v != NULL); - assert (dict_contains_var (d, v)); - assert (new_index < d->var_cnt); - - move_element (d->var, d->var_cnt, sizeof *d->var, v->index, new_index); + size_t old_index = var_get_dict_index (v); - min_idx = min (v->index, new_index); - max_idx = max (v->index, new_index); - for (i = min_idx; i <= max_idx; i++) - d->var[i]->index = i; + assert (new_index < d->var_cnt); + move_element (d->var, d->var_cnt, sizeof *d->var, old_index, new_index); + reindex_vars (d, MIN (old_index, new_index), MAX (old_index, new_index) + 1); } /* Reorders the variables in D, placing the COUNT variables @@ -560,43 +552,56 @@ dict_reorder_vars (struct dictionary *d, memcpy (new_var, order, count * sizeof *new_var); for (i = 0; i < count; i++) { - assert (d->var[order[i]->index] != NULL); - d->var[order[i]->index] = NULL; - order[i]->index = i; + size_t index = var_get_dict_index (order[i]); + assert (d->var[index] == order[i]); + d->var[index] = NULL; + set_var_dict_index (order[i], i); } for (i = 0; i < d->var_cnt; i++) if (d->var[i] != NULL) { assert (count < d->var_cnt); new_var[count] = d->var[i]; - new_var[count]->index = count; + set_var_dict_index (new_var[count], count); count++; } free (d->var); d->var = new_var; } +/* Changes the name of variable V in dictionary D to NEW_NAME. */ +static void +rename_var (struct dictionary *d, struct variable *v, const char *new_name) +{ + struct vardict_info vdi; + + assert (dict_contains_var (d, v)); + + vdi = *var_get_vardict (v); + var_clear_vardict (v); + var_set_name (v, new_name); + var_set_vardict (v, &vdi); +} + /* Changes the name of V in D to name NEW_NAME. Assert-fails if a variable named NEW_NAME is already in D, except that NEW_NAME may be the same as V's existing name. */ -void +void dict_rename_var (struct dictionary *d, struct variable *v, - const char *new_name) + const char *new_name) { - assert (d != NULL); - assert (v != NULL); - assert (new_name != NULL); - assert (var_is_plausible_name (new_name, false)); - assert (dict_contains_var (d, v)); - assert (!compare_var_names (v->name, new_name, NULL) + assert (!strcasecmp (var_get_name (v), new_name) || dict_lookup_var (d, new_name) == NULL); hsh_force_delete (d->name_tab, v); - str_copy_trunc (v->name, sizeof v->name, new_name); + rename_var (d, v, new_name); hsh_force_insert (d->name_tab, v); if (get_algorithm () == ENHANCED) var_clear_short_name (v); + + if ( d->callbacks && d->callbacks->var_changed ) + d->callbacks->var_changed (d, var_get_dict_index (v), d->cb_data); } /* Renames COUNT variables specified in VARS to the names given @@ -607,73 +612,62 @@ dict_rename_var (struct dictionary *d, struct variable *v, is returned. */ bool dict_rename_vars (struct dictionary *d, - struct variable **vars, char **new_names, - size_t count, char **err_name) + struct variable **vars, char **new_names, size_t count, + char **err_name) { + struct pool *pool; char **old_names; size_t i; - bool success = true; - assert (d != NULL); assert (count == 0 || vars != NULL); assert (count == 0 || new_names != NULL); + /* Save the names of the variables to be renamed. */ + pool = pool_create (); + old_names = pool_nalloc (pool, count, sizeof *old_names); + for (i = 0; i < count; i++) + old_names[i] = pool_strdup (pool, var_get_name (vars[i])); + /* Remove the variables to be renamed from the name hash, - save their names, and rename them. */ - old_names = xnmalloc (count, sizeof *old_names); + and rename them. */ for (i = 0; i < count; i++) { - assert (d->var[vars[i]->index] == vars[i]); - assert (var_is_plausible_name (new_names[i], false)); hsh_force_delete (d->name_tab, vars[i]); - old_names[i] = xstrdup (vars[i]->name); - strcpy (vars[i]->name, new_names[i]); + rename_var (d, vars[i], new_names[i]); } /* Add the renamed variables back into the name hash, checking for conflicts. */ for (i = 0; i < count; i++) - { - assert (new_names[i] != NULL); - assert (*new_names[i] != '\0'); - assert (strlen (new_names[i]) >= 1); - assert (strlen (new_names[i]) <= LONG_NAME_LEN); - - if (hsh_insert (d->name_tab, vars[i]) != NULL) - { - /* There is a name conflict. - Back out all the name changes that have already - taken place, and indicate failure. */ - size_t fail_idx = i; - if (err_name != NULL) - *err_name = new_names[i]; - - for (i = 0; i < fail_idx; i++) - hsh_force_delete (d->name_tab, vars[i]); + if (hsh_insert (d->name_tab, vars[i]) != NULL) + { + /* There is a name conflict. + Back out all the name changes that have already + taken place, and indicate failure. */ + size_t fail_idx = i; + if (err_name != NULL) + *err_name = new_names[i]; + + for (i = 0; i < fail_idx; i++) + hsh_force_delete (d->name_tab, vars[i]); - for (i = 0; i < count; i++) - { - strcpy (vars[i]->name, old_names[i]); - hsh_force_insert (d->name_tab, vars[i]); - } - - success = false; - goto done; - } - } + for (i = 0; i < count; i++) + { + rename_var (d, vars[i], old_names[i]); + hsh_force_insert (d->name_tab, vars[i]); + } + + pool_destroy (pool); + return false; + } /* Clear short names. */ if (get_algorithm () == ENHANCED) for (i = 0; i < count; i++) var_clear_short_name (vars[i]); - done: - /* Free the old names we kept around. */ - for (i = 0; i < count; i++) - free (old_names[i]); - free (old_names); - - return success; + pool_destroy (pool); + return true; } /* Returns the weighting variable in dictionary D, or a null @@ -690,11 +684,11 @@ dict_get_weight (const struct dictionary *d) /* Returns the value of D's weighting variable in case C, except that a negative weight is returned as 0. Returns 1 if the dictionary is unweighted. Will warn about missing, negative, or zero values if - warn_on_invalid is nonzero. The function will set warn_on_invalid to zero + warn_on_invalid is true. The function will set warn_on_invalid to false if an invalid weight is found. */ double dict_get_case_weight (const struct dictionary *d, const struct ccase *c, - int *warn_on_invalid) + bool *warn_on_invalid) { assert (d != NULL); assert (c != NULL); @@ -703,11 +697,11 @@ dict_get_case_weight (const struct dictionary *d, const struct ccase *c, return 1.0; else { - double w = case_num (c, d->weight->fv); - if (w < 0.0 || mv_is_num_missing (&d->weight->miss, w)) + double w = case_num (c, d->weight); + if (w < 0.0 || var_is_num_missing (d->weight, w)) w = 0.0; if ( w == 0.0 && *warn_on_invalid ) { - *warn_on_invalid = 0; + *warn_on_invalid = false; msg (SW, _("At least one case in the data file had a weight value " "that was user-missing, system-missing, zero, or " "negative. These case(s) were ignored.")); @@ -723,7 +717,7 @@ dict_set_weight (struct dictionary *d, struct variable *v) { assert (d != NULL); assert (v == NULL || dict_contains_var (d, v)); - assert (v == NULL || v->type == NUMERIC); + assert (v == NULL || var_is_numeric (v)); d->weight = v; } @@ -770,8 +764,8 @@ dict_set_case_limit (struct dictionary *d, size_t case_limit) d->case_limit = case_limit; } -/* Returns the index of the next value to be added to D. This - value is the number of `union value's that need to be +/* Returns the case index of the next value to be added to D. + This value is the number of `union value's that need to be allocated to store a case for dictionary D. */ int dict_get_next_value_idx (const struct dictionary *d) @@ -803,10 +797,10 @@ dict_compact_values (struct dictionary *d) { struct variable *v = d->var[i]; - if (dict_class_from_id (v->name) != DC_SCRATCH) + if (dict_class_from_id (var_get_name (v)) != DC_SCRATCH) { - v->fv = d->next_value_idx; - d->next_value_idx += v->nv; + set_var_case_index (v, d->next_value_idx); + d->next_value_idx += var_get_value_cnt (v); i++; } else @@ -824,53 +818,79 @@ dict_get_compacted_value_cnt (const struct dictionary *d) cnt = 0; for (i = 0; i < d->var_cnt; i++) - if (dict_class_from_id (d->var[i]->name) != DC_SCRATCH) - cnt += d->var[i]->nv; + if (dict_class_from_id (var_get_name (d->var[i])) != DC_SCRATCH) + cnt += var_get_value_cnt (d->var[i]); return cnt; } /* Creates and returns an array mapping from a dictionary index - to the `fv' that the corresponding variable will have after - calling dict_compact_values(). Scratch variables receive -1 - for `fv' because dict_compact_values() will delete them. */ + to the case index that the corresponding variable will have + after calling dict_compact_values(). Scratch variables + receive -1 for case index because dict_compact_values() will + delete them. */ int * -dict_get_compacted_idx_to_fv (const struct dictionary *d) +dict_get_compacted_dict_index_to_case_index (const struct dictionary *d) { size_t i; size_t next_value_idx; - int *idx_to_fv; + int *map; - idx_to_fv = xnmalloc (d->var_cnt, sizeof *idx_to_fv); + map = xnmalloc (d->var_cnt, sizeof *map); next_value_idx = 0; for (i = 0; i < d->var_cnt; i++) { struct variable *v = d->var[i]; - if (dict_class_from_id (v->name) != DC_SCRATCH) + if (dict_class_from_id (var_get_name (v)) != DC_SCRATCH) { - idx_to_fv[i] = next_value_idx; - next_value_idx += v->nv; + map[i] = next_value_idx; + next_value_idx += var_get_value_cnt (v); } else - idx_to_fv[i] = -1; + map[i] = -1; } - return idx_to_fv; + return map; } /* Returns true if a case for dictionary D would be smaller after - compaction, false otherwise. Compacting a case eliminates + compacting, false otherwise. Compacting a case eliminates "holes" between values and after the last value. Holes are created by deleting variables (or by scratch variables). The return value may differ from whether compacting a case - from dictionary D would *change* the case: compaction could + from dictionary D would *change* the case: compacting could rearrange values even if it didn't reduce space requirements. */ bool -dict_needs_compaction (const struct dictionary *d) +dict_compacting_would_shrink (const struct dictionary *d) { return dict_get_compacted_value_cnt (d) < dict_get_next_value_idx (d); } + +/* Returns true if a case for dictionary D would change after + compacting, false otherwise. Compacting a case eliminates + "holes" between values and after the last value. Holes are + created by deleting variables (or by scratch variables). + + The return value may differ from whether compacting a case + from dictionary D would *shrink* the case: compacting could + rearrange values without reducing space requirements. */ +bool +dict_compacting_would_change (const struct dictionary *d) +{ + size_t case_idx; + size_t i; + + case_idx = 0; + for (i = 0; i < dict_get_var_cnt (d); i++) + { + struct variable *v = dict_get_var (d, i); + if (var_get_case_index (v) != case_idx) + return true; + case_idx += var_get_value_cnt (v); + } + return false; +} /* How to copy a contiguous range of values between cases. */ struct copy_map @@ -913,21 +933,21 @@ dict_make_compactor (const struct dictionary *d) { struct variable *v = d->var[i]; - if (dict_class_from_id (v->name) == DC_SCRATCH) + if (dict_class_from_id (var_get_name (v)) == DC_SCRATCH) continue; - if (map != NULL && map->src_idx + map->cnt == v->fv) - map->cnt += v->nv; + if (map != NULL && map->src_idx + map->cnt == var_get_case_index (v)) + map->cnt += var_get_value_cnt (v); else { if (compactor->map_cnt == map_allocated) compactor->maps = x2nrealloc (compactor->maps, &map_allocated, sizeof *compactor->maps); map = &compactor->maps[compactor->map_cnt++]; - map->src_idx = v->fv; + map->src_idx = var_get_case_index (v); map->dst_idx = value_idx; - map->cnt = v->nv; + map->cnt = var_get_value_cnt (v); } - value_idx += v->nv; + value_idx += var_get_value_cnt (v); } return compactor; @@ -1051,39 +1071,29 @@ dict_set_documents (struct dictionary *d, const char *documents) d->documents = xstrdup (documents); } -/* Creates in D a vector named NAME that contains CNT variables - VAR (see cmd_vector()). Returns true if successful, or - false if a vector named NAME already exists in D. */ +/* Creates in D a vector named NAME that contains the CNT + variables in VAR. Returns true if successful, or false if a + vector named NAME already exists in D. */ bool dict_create_vector (struct dictionary *d, const char *name, struct variable **var, size_t cnt) { - struct vector *vector; size_t i; - assert (d != NULL); - assert (name != NULL); - assert (var_is_plausible_name (name, false)); assert (var != NULL); assert (cnt > 0); - - if (dict_lookup_vector (d, name) != NULL) - return false; - - d->vector = xnrealloc (d->vector, d->vector_cnt + 1, sizeof *d->vector); - vector = d->vector[d->vector_cnt] = xmalloc (sizeof *vector); - vector->idx = d->vector_cnt++; - str_copy_trunc (vector->name, sizeof vector->name, name); - vector->var = xnmalloc (cnt, sizeof *var); for (i = 0; i < cnt; i++) + assert (dict_contains_var (d, var[i])); + + if (dict_lookup_vector (d, name) == NULL) { - assert (dict_contains_var (d, var[i])); - vector->var[i] = var[i]; + d->vector = xnrealloc (d->vector, d->vector_cnt + 1, sizeof *d->vector); + d->vector[d->vector_cnt++] = vector_create (name, var, cnt); + return true; } - vector->cnt = cnt; - - return true; + else + return false; } /* Returns the vector in D with index IDX, which must be less @@ -1112,12 +1122,8 @@ const struct vector * dict_lookup_vector (const struct dictionary *d, const char *name) { size_t i; - - assert (d != NULL); - assert (name != NULL); - for (i = 0; i < d->vector_cnt; i++) - if (!strcasecmp (d->vector[i]->name, name)) + if (!strcasecmp (vector_get_name (d->vector[i]), name)) return d->vector[i]; return NULL; } @@ -1128,32 +1134,73 @@ dict_clear_vectors (struct dictionary *d) { size_t i; - assert (d != NULL); - - for (i = 0; i < d->vector_cnt; i++) - { - free (d->vector[i]->var); - free (d->vector[i]); - } + for (i = 0; i < d->vector_cnt; i++) + vector_destroy (d->vector[i]); free (d->vector); + d->vector = NULL; d->vector_cnt = 0; } /* Compares two strings. */ static int -compare_strings (const void *a, const void *b, void *aux UNUSED) +compare_strings (const void *a, const void *b, const void *aux UNUSED) { return strcmp (a, b); } /* Hashes a string. */ static unsigned -hash_string (const void *s, void *aux UNUSED) +hash_string (const void *s, const void *aux UNUSED) { return hsh_hash_string (s); } + +/* Sets V's short name to BASE, followed by a suffix of the form + _A, _B, _C, ..., _AA, _AB, etc. according to the value of + SUFFIX_NUMBER. Truncates BASE as necessary to fit. */ +static void +set_var_short_name_suffix (struct variable *v, const char *base, + int suffix_number) +{ + char suffix[SHORT_NAME_LEN + 1]; + char short_name[SHORT_NAME_LEN + 1]; + char *start, *end; + int len, ofs; + + assert (v != NULL); + assert (suffix_number >= 0); + + /* Set base name. */ + var_set_short_name (v, base); + + /* Compose suffix. */ + start = end = suffix + sizeof suffix - 1; + *end = '\0'; + do + { + *--start = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"[suffix_number % 26]; + if (start <= suffix + 1) + msg (SE, _("Variable suffix too large.")); + suffix_number /= 26; + } + while (suffix_number > 0); + *--start = '_'; + + /* Append suffix to V's short name. */ + str_copy_trunc (short_name, sizeof short_name, base); + len = end - start; + if (len + strlen (short_name) > SHORT_NAME_LEN) + ofs = SHORT_NAME_LEN - len; + else + ofs = strlen (short_name); + strcpy (short_name + ofs, start); + + /* Set name. */ + var_set_short_name (v, short_name); +} + /* Assigns a valid, unique short_name[] to each variable in D. Each variable whose actual name is short has highest priority for that short name. Otherwise, variables with an existing @@ -1174,9 +1221,10 @@ dict_assign_short_names (struct dictionary *d) for (i = 0; i < d->var_cnt; i++) { struct variable *v = d->var[i]; - if (strlen (v->name) <= SHORT_NAME_LEN) - var_set_short_name (v, v->name); - else if (dict_lookup_var (d, v->short_name) != NULL) + const char *short_name = var_get_short_name (v); + if (strlen (var_get_name (v)) <= SHORT_NAME_LEN) + var_set_short_name (v, var_get_name (v)); + else if (short_name != NULL && dict_lookup_var (d, short_name) != NULL) var_clear_short_name (v); } @@ -1187,7 +1235,8 @@ dict_assign_short_names (struct dictionary *d) for (i = 0; i < d->var_cnt; i++) { struct variable *v = d->var[i]; - if (v->short_name[0] && hsh_insert (short_names, v->short_name) != NULL) + const char *name = var_get_short_name (v); + if (name != NULL && hsh_insert (short_names, (char *) name) != NULL) var_clear_short_name (v); } @@ -1195,19 +1244,44 @@ dict_assign_short_names (struct dictionary *d) for (i = 0; i < d->var_cnt; i++) { struct variable *v = d->var[i]; - if (v->short_name[0] == '\0') + const char *name = var_get_short_name (v); + if (name == NULL) { - int sfx; - - /* Form initial short_name. */ - var_set_short_name (v, v->name); + /* Form initial short_name from the variable name, then + try _A, _B, ... _AA, _AB, etc., if needed.*/ + int trial = 0; + do + { + if (trial == 0) + var_set_short_name (v, var_get_name (v)); + else + set_var_short_name_suffix (v, var_get_name (v), trial - 1); - /* Try _A, _B, ... _AA, _AB, etc., if needed. */ - for (sfx = 0; hsh_insert (short_names, v->short_name) != NULL; sfx++) - var_set_short_name_suffix (v, v->name, sfx); + trial++; + } + while (hsh_insert (short_names, (char *) var_get_short_name (v)) + != NULL); } } /* Get rid of hash table. */ hsh_destroy (short_names); } + + +/* Called from variable.c to notify the dictionary that some property of + the variable has changed */ +void +dict_var_changed (const struct variable *v) +{ + if ( var_has_vardict (v)) + { + const struct vardict_info *vdi = var_get_vardict (v); + struct dictionary *d; + + d = vdi->dict; + + if ( d->callbacks && d->callbacks->var_changed ) + d->callbacks->var_changed (d, var_get_dict_index (v), d->cb_data); + } +}