data_out function to dynamically allocate return value.
[pspp-builds.git] / src / ui / gui / psppire-data-store.c
index 71d7d5f53976c124fc779a7a56fa127b13eaa09e..13ad0cc6cd283edfb9656a67f4eb2b3f7ae7873b 100644 (file)
@@ -31,6 +31,7 @@
 #include <pango/pango-context.h>
 
 #include "psppire-data-store.h"
+#include <libpspp/i18n.h>
 #include "helper.h"
 
 #include <data/dictionary.h>
@@ -186,18 +187,18 @@ psppire_data_store_class_init (PsppireDataStoreClass *class)
 
 
 static gboolean
-psppire_data_store_insert_values (PsppireDataStore *ds,
-                                 gint n_values, gint where);
+psppire_data_store_insert_value (PsppireDataStore *ds,
+                                 gint width, gint where);
 
-static union value *
+static bool
 psppire_data_store_get_value (const PsppireDataStore *ds,
                              casenumber casenum, size_t idx,
-                             union value *value, int width);
+                             union value *value);
 
 
 static gboolean
 psppire_data_store_set_value (PsppireDataStore *ds, casenumber casenum,
-                             gint idx, union value *v, gint width);
+                             gint idx, union value *v);
 
 
 
@@ -213,7 +214,7 @@ psppire_data_store_get_var_count (const PsppireSheetModel *model)
 casenumber
 psppire_data_store_get_case_count (const PsppireDataStore *store)
 {
-  return datasheet_get_row_cnt (store->datasheet);
+  return datasheet_get_n_rows (store->datasheet);
 }
 
 size_t
@@ -222,6 +223,12 @@ psppire_data_store_get_value_count (const PsppireDataStore *store)
   return psppire_dict_get_value_cnt (store->dict);
 }
 
+const struct caseproto *
+psppire_data_store_get_proto (const PsppireDataStore *store)
+{
+  return psppire_dict_get_proto (store->dict);
+}
+
 static casenumber
 psppire_data_store_get_case_count_wrapper (const PsppireSheetModel *model)
 {
@@ -263,6 +270,7 @@ static GtkJustification get_column_justification (const PsppireSheetModel *model
 
 static gchar * get_row_button_label (const PsppireSheetModel *model, gint row);
 static gboolean get_row_sensitivity (const PsppireSheetModel *model, gint row);
+static gboolean get_row_overstrike (const PsppireSheetModel *model, gint row);
 
 
 static void
@@ -285,6 +293,7 @@ psppire_data_store_sheet_model_init (PsppireSheetModelIface *iface)
 
   iface->get_row_title = get_row_button_label;
   iface->get_row_sensitivity = get_row_sensitivity;
+  iface->get_row_overstrike = get_row_overstrike;
 }
 
 
@@ -293,13 +302,15 @@ psppire_data_store_sheet_model_init (PsppireSheetModelIface *iface)
  */
 static void
 delete_variable_callback (GObject *obj, gint dict_index,
-                         gint case_index, gint val_cnt,
+                         gint case_index, gint width,
                          gpointer data)
 {
   PsppireDataStore *store  = PSPPIRE_DATA_STORE (data);
 
 
   psppire_sheet_model_columns_deleted (PSPPIRE_SHEET_MODEL (store), dict_index, 1);
+  datasheet_delete_columns (store->datasheet, case_index, 1);
+  datasheet_insert_column (store->datasheet, NULL, -1, case_index);
 #if AXIS_TRANSITION
 
 
@@ -308,18 +319,15 @@ delete_variable_callback (GObject *obj, gint dict_index,
 #endif
 }
 
-
-
 static void
 variable_changed_callback (GObject *obj, gint var_num, gpointer data)
 {
+#if AXIS_TRANSITION
   PsppireDataStore *store  = PSPPIRE_DATA_STORE (data);
 
-#if AXIS_TRANSITION
   psppire_sheet_column_columns_changed (PSPPIRE_SHEET_COLUMN (store),
                                  var_num, 1);
 
-
   psppire_sheet_model_range_changed (PSPPIRE_SHEET_MODEL (store),
                               -1, var_num,
                               -1, var_num);
@@ -329,6 +337,7 @@ variable_changed_callback (GObject *obj, gint var_num, gpointer data)
 static void
 insert_variable_callback (GObject *obj, gint var_num, gpointer data)
 {
+  struct variable *variable;
   PsppireDataStore *store;
   gint posn;
 
@@ -336,21 +345,9 @@ insert_variable_callback (GObject *obj, gint var_num, gpointer data)
 
   store  = PSPPIRE_DATA_STORE (data);
 
-  if ( var_num > 0 )
-    {
-      struct variable *variable =
-       psppire_dict_get_variable (store->dict, var_num);
-
-      g_assert (variable != NULL);
-
-      posn = var_get_case_index (variable);
-    }
-  else
-    {
-      posn = 0;
-    }
-
-  psppire_data_store_insert_values (store, 1, posn);
+  variable = psppire_dict_get_variable (store->dict, var_num);
+  posn = var_get_case_index (variable);
+  psppire_data_store_insert_value (store, var_get_width (variable), posn);
 
 #if AXIS_TRANSITION
 
@@ -361,21 +358,51 @@ insert_variable_callback (GObject *obj, gint var_num, gpointer data)
   psppire_sheet_model_columns_inserted (PSPPIRE_SHEET_MODEL (store), var_num, 1);
 }
 
+struct resize_datum_aux
+  {
+    int old_width;
+    int new_width;
+  };
+
+
+void
+resize_datum (const union value *old, union value *new, void *aux_)
+{
+  struct resize_datum_aux *aux = aux_;
+
+  if (aux->new_width == 0)
+    {
+      /* FIXME: try to parse string as number. */
+      new->f = SYSMIS;
+    }
+  else if (aux->old_width == 0)
+    {
+      /* FIXME: format number as string. */
+      value_set_missing (new, aux->new_width);
+    }
+  else
+    value_copy_rpad (new, aux->new_width, old, aux->old_width, ' ');
+}
 
 static void
 dict_size_change_callback (GObject *obj,
-                         gint posn, gint adjustment, gpointer data)
+                         gint var_num, gint old_width, gpointer data)
 {
   PsppireDataStore *store  = PSPPIRE_DATA_STORE (data);
+  struct variable *variable;
+  int posn;
 
-  const struct variable *v = psppire_dict_get_variable (store->dict, posn);
-
-  const gint new_val_width = value_cnt_from_width (var_get_width (v));
+  variable = psppire_dict_get_variable (store->dict, var_num);
+  posn = var_get_case_index (variable);
 
-  if ( adjustment > 0 )
-    psppire_data_store_insert_values (store, adjustment,
-                                    new_val_width - adjustment +
-                                    var_get_case_index(v));
+  if (old_width != var_get_width (variable))
+    {
+      struct resize_datum_aux aux;
+      aux.old_width = old_width;
+      aux.new_width = var_get_width (variable);
+      datasheet_resize_column (store->datasheet, posn, aux.new_width,
+                               resize_datum, &aux);
+    }
 }
 
 
@@ -530,28 +557,16 @@ gboolean
 psppire_data_store_insert_new_case (PsppireDataStore *ds, casenumber posn)
 {
   gboolean result;
-  gint val_cnt, v;
+  const struct caseproto *proto;
   struct ccase *cc;
   g_return_val_if_fail (ds, FALSE);
 
-  val_cnt = datasheet_get_column_cnt (ds->datasheet) ;
-
-  g_return_val_if_fail (val_cnt > 0, FALSE);
-
+  proto = datasheet_get_proto (ds->datasheet);
+  g_return_val_if_fail (caseproto_get_n_widths (proto) > 0, FALSE);
   g_return_val_if_fail (posn <= psppire_data_store_get_case_count (ds), FALSE);
 
-  cc = case_create (val_cnt);
-
-  memset ( case_data_rw_idx (cc, 0), 0, val_cnt * MAX_SHORT_STRING);
-
-  for (v = 0 ; v < psppire_dict_get_var_cnt (ds->dict) ; ++v)
-    {
-      const struct variable *pv = psppire_dict_get_variable (ds->dict, v);
-      if ( var_is_alpha (pv))
-       continue;
-
-      case_data_rw (cc, pv)->f = SYSMIS;
-    }
+  cc = case_create (proto);
+  case_set_missing (cc);
 
   result = psppire_data_store_insert_case (ds, cc, posn);
 
@@ -568,8 +583,8 @@ psppire_data_store_get_string (PsppireDataStore *store, glong row, glong column)
   char *text;
   const struct fmt_spec *fp ;
   const struct variable *pv ;
-  union value *v ;
-  GString *s;
+  union value v;
+  int width;
 
   g_return_val_if_fail (store->dict, NULL);
   g_return_val_if_fail (store->datasheet, NULL);
@@ -585,44 +600,35 @@ psppire_data_store_get_string (PsppireDataStore *store, glong row, glong column)
   g_assert (pv);
 
   idx = var_get_case_index (pv);
+  width = var_get_width (pv);
 
   g_assert (idx >= 0);
 
-  v = psppire_data_store_get_value (store, row, idx, NULL,
-                                   var_get_width (pv));
-
-  g_return_val_if_fail (v, NULL);
+  value_init (&v, width);
+  if (!psppire_data_store_get_value (store, row, idx, &v))
+    return NULL;
 
   if ( store->show_labels)
     {
-      const gchar *label = var_lookup_value_label (pv, v);
+      const gchar *label = var_lookup_value_label (pv, &v);
       if (label)
         {
-          free (v);
-         return pspp_locale_to_utf8 (label, -1, 0);
+          value_destroy (&v, width);
+         return recode_string (UTF8, psppire_dict_encoding (store->dict),
+                               label, -1);
         }
     }
 
   fp = var_get_write_format (pv);
 
-  s = g_string_sized_new (fp->w + 1);
-  g_string_set_size (s, fp->w);
-
-  memset (s->str, 0, fp->w);
-
-  g_assert (fp->w == s->len);
-
   /* Converts binary value V into printable form in the exactly
      FP->W character in buffer S according to format specification
      FP.  No null terminator is appended to the buffer.  */
-  data_out (v, fp, s->str);
-
-  text = pspp_locale_to_utf8 (s->str, fp->w, 0);
-  g_string_free (s, TRUE);
+  text = data_out (&v, fp);
 
   g_strchomp (text);
 
-  free (v);
+  value_destroy (&v, width);
   return text;
 }
 
@@ -635,19 +641,18 @@ psppire_data_store_clear_datum (PsppireSheetModel *model,
 
   union value v;
   const struct variable *pv = psppire_dict_get_variable (store->dict, col);
+  int width = var_get_width (pv);
 
   const gint index = var_get_case_index (pv) ;
 
-  if ( var_is_numeric (pv))
-    v.f = SYSMIS;
-  else
-    memcpy (v.s, "", MAX_SHORT_STRING);
-
-  psppire_data_store_set_value (store, row, index, &v,
-                               var_get_width (pv));
+  value_init (&v, width);
+  value_set_missing (&v, width);
+  psppire_data_store_set_value (store, row, index, &v);
+  value_destroy (&v, width);
 
   psppire_sheet_model_range_changed (model, row, col, row, col);
 
+
   return TRUE;
 }
 
@@ -660,6 +665,7 @@ gboolean
 psppire_data_store_set_string (PsppireDataStore *store,
                               const gchar *text, glong row, glong col)
 {
+  gchar *s;
   glong n_cases;
   const struct variable *pv = psppire_dict_get_variable (store->dict, col);
   if ( NULL == pv)
@@ -673,9 +679,12 @@ psppire_data_store_set_string (PsppireDataStore *store,
   if (row == n_cases)
     psppire_data_store_insert_new_case (store, row);
 
+  s = recode_string (psppire_dict_encoding (store->dict), UTF8, text, -1);
+
   psppire_data_store_data_in (store, row,
-                             var_get_case_index (pv), ss_cstr (text),
+                             var_get_case_index (pv), ss_cstr (s),
                              var_get_write_format (pv));
+  free (s);
 
   psppire_sheet_model_range_changed (PSPPIRE_SHEET_MODEL (store), row, col, row, col);
 
@@ -746,9 +755,11 @@ static const gchar null_var_name[]=N_("var");
 static gchar *
 get_row_button_label (const PsppireSheetModel *model, gint unit)
 {
+  PsppireDataStore *ds = PSPPIRE_DATA_STORE (model);
   gchar *s = g_strdup_printf (_("%d"), unit + FIRST_CASE_NUMBER);
 
-  gchar *text =  pspp_locale_to_utf8 (s, -1, 0);
+  gchar *text =  recode_string (UTF8, psppire_dict_encoding (ds->dict),
+                               s, -1);
 
   g_free (s);
 
@@ -772,7 +783,6 @@ get_row_sensitivity (const PsppireSheetModel *model, gint unit)
 static gchar *
 get_column_subtitle (const PsppireSheetModel *model, gint col)
 {
-  gchar *text;
   const struct variable *v ;
   PsppireDataStore *ds = PSPPIRE_DATA_STORE (model);
 
@@ -784,15 +794,12 @@ get_column_subtitle (const PsppireSheetModel *model, gint col)
   if ( ! var_has_label (v))
     return NULL;
 
-  text =  pspp_locale_to_utf8 (var_get_label (v), -1, 0);
-
-  return text;
+  return xstrdup (var_get_label (v));
 }
 
 static gchar *
 get_column_button_label (const PsppireSheetModel *model, gint col)
 {
-  gchar *text;
   struct variable *pv ;
   PsppireDataStore *ds = PSPPIRE_DATA_STORE (model);
 
@@ -801,9 +808,7 @@ get_column_button_label (const PsppireSheetModel *model, gint col)
 
   pv = psppire_dict_get_variable (ds->dict, col);
 
-  text =  pspp_locale_to_utf8 (var_get_name (pv), -1, 0);
-
-  return text;
+  return xstrdup (var_get_name (pv));
 }
 
 static gboolean
@@ -897,53 +902,38 @@ psppire_data_store_insert_case (PsppireDataStore *ds,
 }
 
 
-/* Copies the IDXth value from case CASENUM into VALUE.
-   If VALUE is null, then memory is allocated is allocated with
-   malloc.  Returns the value if successful, NULL on failure. */
-static union value *
+/* Copies the IDXth value from case CASENUM into VALUE, which
+   must be of the correct width for IDX.
+   Returns true if successful, false on failure. */
+static bool
 psppire_data_store_get_value (const PsppireDataStore *ds,
                              casenumber casenum, size_t idx,
-                             union value *value, int width)
+                             union value *value)
 {
-  bool allocated;
-
   g_return_val_if_fail (ds, false);
   g_return_val_if_fail (ds->datasheet, false);
+  g_return_val_if_fail (idx < datasheet_get_n_columns (ds->datasheet), false);
 
-  g_return_val_if_fail (idx < datasheet_get_column_cnt (ds->datasheet), false);
-
-  if (value == NULL)
-    {
-      value = xnmalloc (value_cnt_from_width (width), sizeof *value);
-      allocated = true;
-    }
-  else
-    allocated = false;
-  if (!datasheet_get_value (ds->datasheet, casenum, idx, value, width))
-    {
-      if (allocated)
-        free (value);
-      value = NULL;
-    }
-  return value;
+  return datasheet_get_value (ds->datasheet, casenum, idx, value);
 }
 
 
 
 /* Set the IDXth value of case C to V.
+   V must be the correct width for IDX.
    Returns true if successful, false on I/O error. */
 static gboolean
 psppire_data_store_set_value (PsppireDataStore *ds, casenumber casenum,
-                             gint idx, union value *v, gint width)
+                             gint idx, union value *v)
 {
   bool ok;
 
   g_return_val_if_fail (ds, FALSE);
   g_return_val_if_fail (ds->datasheet, FALSE);
 
-  g_return_val_if_fail (idx < datasheet_get_column_cnt (ds->datasheet), FALSE);
+  g_return_val_if_fail (idx < datasheet_get_n_columns (ds->datasheet), FALSE);
 
-  ok = datasheet_put_value (ds->datasheet, casenum, idx, v, width);
+  ok = datasheet_put_value (ds->datasheet, casenum, idx, v);
   if (ok)
     g_signal_emit (ds, signals [CASE_CHANGED], 0, casenum);
 
@@ -958,22 +948,24 @@ static gboolean
 psppire_data_store_data_in (PsppireDataStore *ds, casenumber casenum, gint idx,
                            struct substring input, const struct fmt_spec *fmt)
 {
-  union value *value = NULL;
+  union value value;
   int width;
   bool ok;
 
   g_return_val_if_fail (ds, FALSE);
   g_return_val_if_fail (ds->datasheet, FALSE);
 
-  g_return_val_if_fail (idx < datasheet_get_column_cnt (ds->datasheet), FALSE);
+  g_return_val_if_fail (idx < datasheet_get_n_columns (ds->datasheet), FALSE);
 
   width = fmt_var_width (fmt);
-  value = xmalloca (value_cnt_from_width (width) * sizeof *value);
-  ok = (datasheet_get_value (ds->datasheet, casenum, idx, value, width)
-        && data_in (input, LEGACY_NATIVE, fmt->type, 0, 0, 0, value, width)
-        && datasheet_put_value (ds->datasheet, casenum, idx, value, width));
-
-  freea (value);
+  g_return_val_if_fail (caseproto_get_width (
+                          datasheet_get_proto (ds->datasheet), idx) == width,
+                        FALSE);
+  value_init (&value, width);
+  ok = (datasheet_get_value (ds->datasheet, casenum, idx, &value)
+        && data_in (input, LEGACY_NATIVE, fmt->type, 0, 0, 0, &value, width)
+        && datasheet_put_value (ds->datasheet, casenum, idx, &value));
+  value_destroy (&value, width);
 
   if (ok)
     g_signal_emit (ds, signals [CASE_CHANGED], 0, casenum);
@@ -981,28 +973,58 @@ psppire_data_store_data_in (PsppireDataStore *ds, casenumber casenum, gint idx,
   return ok;
 }
 
-/* Resize the cases in the casefile, by inserting N_VALUES into every
-   one of them at the position immediately preceeding WHERE.
+/* Resize the cases in the casefile, by inserting a value of the
+   given WIDTH into every one of them at the position immediately
+   preceding WHERE.
 */
 static gboolean
-psppire_data_store_insert_values (PsppireDataStore *ds,
-                                 gint n_values, gint where)
+psppire_data_store_insert_value (PsppireDataStore *ds,
+                                 gint width, gint where)
 {
-  g_return_val_if_fail (ds, FALSE);
+  union value value;
 
-  if ( n_values == 0 )
-    return FALSE;
+  g_return_val_if_fail (ds, FALSE);
 
-  g_assert (n_values > 0);
+  g_assert (width >= 0);
 
   if ( ! ds->datasheet )
     ds->datasheet = datasheet_create (NULL);
 
-  {
-    union value *values = xcalloc (n_values, sizeof *values);
-    datasheet_insert_columns (ds->datasheet, values, n_values, where);
-    free (values);
-  }
+  value_init (&value, width);
+  if (width == 0)
+    value.f = 0;
+  else
+    value_set_missing (&value, width);
+
+  datasheet_insert_column (ds->datasheet, &value, width, where);
 
   return TRUE;
 }
+
+static gboolean
+get_row_overstrike (const PsppireSheetModel *model, gint row)
+{
+  union value val;
+  PsppireDataStore *ds = PSPPIRE_DATA_STORE (model);
+
+  const struct dictionary *dict = ds->dict->dict;
+
+  const struct variable *filter = dict_get_filter (dict);
+
+  if ( row < 0 || row >= datasheet_get_n_rows (ds->datasheet))
+    return FALSE;
+
+  if ( ! filter)
+    return FALSE;
+
+  g_assert (var_is_numeric (filter));
+
+  value_init (&val, 0);
+  if ( ! datasheet_get_value (ds->datasheet, row,
+                             var_get_case_index (filter),
+                             &val) )
+    return FALSE;
+
+
+  return (val.f == 0.0);
+}