Added custom psppire-selector widget.
[pspp-builds.git] / src / ui / gui / psppire-dict.c
1 /*
2   PSPPIRE --- A Graphical User Interface for PSPP
3   Copyright (C) 2004, 2006, 2007  Free Software Foundation
4
5   This program is free software; you can redistribute it and/or modify
6   it under the terms of the GNU General Public License as published by
7   the Free Software Foundation; either version 2 of the License, or
8   (at your option) any later version.
9
10   This program is distributed in the hope that it will be useful,
11   but WITHOUT ANY WARRANTY; without even the implied warranty of
12   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13   GNU General Public License for more details.
14
15   You should have received a copy of the GNU General Public License
16   along with this program; if not, write to the Free Software
17   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
18   02110-1301, USA. */
19
20 #include <config.h>
21 #include <string.h>
22 #include <stdlib.h>
23
24 #include <gtk/gtk.h>
25 #include <gtksheet/gtkextra-marshal.h>
26
27 #include "psppire-dict.h"
28 #include <data/format.h>
29 #include <data/dictionary.h>
30 #include <data/missing-values.h>
31 #include <data/value-labels.h>
32 #include <data/variable.h>
33
34 #include "helper.h"
35 #include "message-dialog.h"
36
37 /* --- prototypes --- */
38 static void psppire_dict_class_init     (PsppireDictClass       *class);
39 static void psppire_dict_init   (PsppireDict            *dict);
40 static void psppire_dict_finalize       (GObject                *object);
41
42 static void dictionary_tree_model_init (GtkTreeModelIface *iface);
43
44
45 /* --- variables --- */
46 static GObjectClass     *parent_class = NULL;
47
48 enum  {VARIABLE_CHANGED,
49        VARIABLE_RESIZED,
50        VARIABLE_INSERTED,
51        VARIABLES_DELETED,
52        WEIGHT_CHANGED,
53        FILTER_CHANGED,
54        SPLIT_CHANGED,
55        n_SIGNALS};
56
57 static guint signals [n_SIGNALS];
58
59 /* --- functions --- */
60 /**
61  * psppire_dict_get_type:
62  * @returns: the type ID for accelerator groups.
63  */
64 GType
65 psppire_dict_get_type (void)
66 {
67   static GType object_type = 0;
68
69   if (!object_type)
70     {
71       static const GTypeInfo object_info = {
72         sizeof (PsppireDictClass),
73         (GBaseInitFunc) NULL,
74         (GBaseFinalizeFunc) NULL,
75         (GClassInitFunc) psppire_dict_class_init,
76         NULL,   /* class_finalize */
77         NULL,   /* class_data */
78         sizeof (PsppireDict),
79         0,      /* n_preallocs */
80         (GInstanceInitFunc) psppire_dict_init,
81       };
82
83       static const GInterfaceInfo tree_model_info = {
84         (GInterfaceInitFunc) dictionary_tree_model_init,
85         NULL,
86         NULL
87       };
88
89       object_type = g_type_register_static (G_TYPE_OBJECT,
90                                             "PsppireDict",
91                                             &object_info, 0);
92
93       g_type_add_interface_static (object_type, GTK_TYPE_TREE_MODEL,
94                                    &tree_model_info);
95
96
97     }
98
99   return object_type;
100 }
101
102
103 static void
104 psppire_dict_class_init (PsppireDictClass *class)
105 {
106   GObjectClass *object_class = G_OBJECT_CLASS (class);
107
108   parent_class = g_type_class_peek_parent (class);
109
110   object_class->finalize = psppire_dict_finalize;
111
112   signals [VARIABLE_CHANGED] =
113     g_signal_new ("variable_changed",
114                   G_TYPE_FROM_CLASS (class),
115                   G_SIGNAL_RUN_FIRST,
116                   0,
117                   NULL, NULL,
118                   g_cclosure_marshal_VOID__INT,
119                   G_TYPE_NONE,
120                   1,
121                   G_TYPE_INT);
122
123
124
125   signals [VARIABLE_INSERTED] =
126     g_signal_new ("variable_inserted",
127                   G_TYPE_FROM_CLASS (class),
128                   G_SIGNAL_RUN_FIRST,
129                   0,
130                   NULL, NULL,
131                   g_cclosure_marshal_VOID__INT,
132                   G_TYPE_NONE,
133                   1,
134                   G_TYPE_INT);
135
136
137   signals [VARIABLES_DELETED] =
138     g_signal_new ("variables_deleted",
139                   G_TYPE_FROM_CLASS (class),
140                   G_SIGNAL_RUN_FIRST,
141                   0,
142                   NULL, NULL,
143                   gtkextra_VOID__INT_INT,
144                   G_TYPE_NONE,
145                   2,
146                   G_TYPE_INT,
147                   G_TYPE_INT);
148
149
150   signals [VARIABLE_RESIZED] =
151     g_signal_new ("dict-size-changed",
152                   G_TYPE_FROM_CLASS (class),
153                   G_SIGNAL_RUN_FIRST,
154                   0,
155                   NULL, NULL,
156                   gtkextra_VOID__INT_INT,
157                   G_TYPE_NONE,
158                   2,
159                   G_TYPE_INT,
160                   G_TYPE_INT);
161
162
163   signals [WEIGHT_CHANGED] =
164     g_signal_new ("weight-changed",
165                   G_TYPE_FROM_CLASS (class),
166                   G_SIGNAL_RUN_FIRST,
167                   0,
168                   NULL, NULL,
169                   g_cclosure_marshal_VOID__INT,
170                   G_TYPE_NONE,
171                   1,
172                   G_TYPE_INT);
173
174
175   signals [FILTER_CHANGED] =
176     g_signal_new ("filter-changed",
177                   G_TYPE_FROM_CLASS (class),
178                   G_SIGNAL_RUN_FIRST,
179                   0,
180                   NULL, NULL,
181                   g_cclosure_marshal_VOID__INT,
182                   G_TYPE_NONE,
183                   1,
184                   G_TYPE_INT);
185
186
187   signals [SPLIT_CHANGED] =
188     g_signal_new ("split-changed",
189                   G_TYPE_FROM_CLASS (class),
190                   G_SIGNAL_RUN_FIRST,
191                   0,
192                   NULL, NULL,
193                   g_cclosure_marshal_VOID__VOID,
194                   G_TYPE_NONE,
195                   0);
196 }
197
198 static void
199 psppire_dict_finalize (GObject *object)
200 {
201   PsppireDict *d = PSPPIRE_DICT (object);
202
203   dict_destroy (d->dict);
204
205   G_OBJECT_CLASS (parent_class)->finalize (object);
206 }
207
208 /* Pass on callbacks from src/data/dictionary, as
209    signals in the Gtk library */
210 static void
211 addcb (struct dictionary *d, int idx, void *pd)
212 {
213   g_signal_emit (pd, signals [VARIABLE_INSERTED], 0, idx);
214 }
215
216 static void
217 delcb (struct dictionary *d, int idx, void *pd)
218 {
219   g_signal_emit (pd, signals [VARIABLES_DELETED], 0, idx, 1);
220 }
221
222 static void
223 mutcb (struct dictionary *d, int idx, void *pd)
224 {
225   g_signal_emit (pd, signals [VARIABLE_CHANGED], 0, idx);
226 }
227
228 static void
229 weight_changed_callback (struct dictionary *d, int idx, void *pd)
230 {
231   g_signal_emit (pd, signals [WEIGHT_CHANGED], 0, idx);
232 }
233
234 static void
235 filter_changed_callback (struct dictionary *d, int idx, void *pd)
236 {
237   g_signal_emit (pd, signals [FILTER_CHANGED], 0, idx);
238 }
239
240 static void
241 split_changed_callback (struct dictionary *d, void *pd)
242 {
243   g_signal_emit (pd, signals [SPLIT_CHANGED], 0);
244 }
245
246
247 static const struct dict_callbacks gui_callbacks =
248   {
249     addcb,
250     delcb,
251     mutcb,
252     weight_changed_callback,
253     filter_changed_callback,
254     split_changed_callback
255   };
256
257 static void
258 psppire_dict_init (PsppireDict *psppire_dict)
259 {
260   psppire_dict->stamp = g_random_int ();
261 }
262
263 /**
264  * psppire_dict_new_from_dict:
265  * @returns: a new #PsppireDict object
266  *
267  * Creates a new #PsppireDict.
268  */
269 PsppireDict*
270 psppire_dict_new_from_dict (struct dictionary *d)
271 {
272   PsppireDict *new_dict = g_object_new (G_TYPE_PSPPIRE_DICT, NULL);
273   new_dict->dict = d;
274
275   dict_set_callbacks (new_dict->dict, &gui_callbacks, new_dict);
276
277   return new_dict;
278 }
279
280
281 void
282 psppire_dict_replace_dictionary (PsppireDict *dict, struct dictionary *d)
283 {
284   dict->dict = d;
285 }
286
287
288 /* Returns a valid name for a new variable in DICT.
289    The return value is statically allocated */
290 static gchar *
291 auto_generate_var_name (PsppireDict *dict)
292 {
293   gint d = 0;
294   static gchar name[10];
295
296   while (g_snprintf (name, 10, "VAR%05d",d++),
297          psppire_dict_lookup_var (dict, name))
298     ;
299
300   return name;
301 }
302
303 /* Insert a new variable at posn IDX, with the name NAME.
304    If NAME is null, then a name will be automatically assigned.
305 */
306 void
307 psppire_dict_insert_variable (PsppireDict *d, gint idx, const gchar *name)
308 {
309   struct variable *var ;
310   g_return_if_fail (idx >= 0);
311   g_return_if_fail (d);
312   g_return_if_fail (PSPPIRE_IS_DICT (d));
313
314   if ( ! name )
315     name = auto_generate_var_name (d);
316
317   var = dict_create_var (d->dict, name, 0);
318
319   dict_reorder_var (d->dict, var, idx);
320 }
321
322 /* Delete N variables beginning at FIRST */
323 void
324 psppire_dict_delete_variables (PsppireDict *d, gint first, gint n)
325 {
326   gint idx;
327   g_return_if_fail (d);
328   g_return_if_fail (d->dict);
329   g_return_if_fail (PSPPIRE_IS_DICT (d));
330
331   for (idx = 0 ; idx < n ; ++idx )
332     {
333       struct variable *var;
334
335       /* Do nothing if it's out of bounds */
336       if ( first >= dict_get_var_cnt (d->dict))
337         break;
338
339       var = dict_get_var (d->dict, first);
340       dict_delete_var (d->dict, var);
341     }
342   dict_compact_values (d->dict);
343 }
344
345
346 void
347 psppire_dict_set_name (PsppireDict* d, gint idx, const gchar *name)
348 {
349   struct variable *var;
350   g_assert (d);
351   g_assert (PSPPIRE_IS_DICT (d));
352
353
354   if ( idx < dict_get_var_cnt (d->dict))
355     {
356       /* This is an existing variable? */
357       var = dict_get_var (d->dict, idx);
358       dict_rename_var (d->dict, var, name);
359     }
360   else
361     {
362       /* new variable */
363       dict_create_var (d->dict, name, 0);
364     }
365 }
366
367
368
369 /* Return the IDXth variable */
370 struct variable *
371 psppire_dict_get_variable (const PsppireDict *d, gint idx)
372 {
373   g_return_val_if_fail (d, NULL);
374   g_return_val_if_fail (d->dict, NULL);
375
376   if ( dict_get_var_cnt (d->dict) <= idx )
377     return NULL;
378
379   return dict_get_var (d->dict, idx);
380 }
381
382
383 /* Return the number of variables in the dictionary */
384 gint
385 psppire_dict_get_var_cnt (const PsppireDict *d)
386 {
387   g_return_val_if_fail (d, -1);
388   g_return_val_if_fail (d->dict, -1);
389
390   return dict_get_var_cnt (d->dict);
391 }
392
393
394 /* Return a variable by name.
395    Return NULL if it doesn't exist
396 */
397 struct variable *
398 psppire_dict_lookup_var (const PsppireDict *d, const gchar *name)
399 {
400   g_return_val_if_fail (d, NULL);
401   g_return_val_if_fail (d->dict, NULL);
402
403   return dict_lookup_var (d->dict, name);
404 }
405
406 /* Clears the contents of D */
407 void
408 psppire_dict_clear (PsppireDict *d)
409 {
410   g_return_if_fail (d);
411   g_return_if_fail (d->dict);
412
413   {
414     dict_clear (d->dict);
415   }
416 }
417
418
419 /* Return true is NAME would be a valid name of a variable to add to the
420    dictionary.  False otherwise.
421    If REPORT is true, then invalid names will be reported as such as errors
422 */
423 gboolean
424 psppire_dict_check_name (const PsppireDict *dict,
425                          const gchar *name, gboolean report)
426 {
427   if ( ! var_is_valid_name (name, report ) )
428     return FALSE;
429
430   if (psppire_dict_lookup_var (dict, name))
431     {
432       if ( report )
433         msg (ME,"Duplicate variable name.");
434       return FALSE;
435     }
436
437   return TRUE;
438 }
439
440
441 inline gint
442 psppire_dict_get_next_value_idx (const PsppireDict *dict)
443 {
444   return dict_get_next_value_idx (dict->dict);
445 }
446
447
448 void
449 psppire_dict_resize_variable (PsppireDict *d, const struct variable *pv,
450                               gint old_size, gint new_size)
451 {
452   gint fv;
453   g_return_if_fail (d);
454   g_return_if_fail (d->dict);
455
456   if ( old_size == new_size )
457     return ;
458
459   dict_compact_values (d->dict);
460
461   fv = var_get_case_index (pv);
462
463   g_signal_emit (d, signals [VARIABLE_RESIZED], 0,
464                  fv + old_size,
465                  new_size - old_size );
466 }
467
468
469 /* Tree Model Stuff */
470
471 static GtkTreeModelFlags tree_model_get_flags (GtkTreeModel *model);
472
473 static gint tree_model_n_columns (GtkTreeModel *model);
474
475 static GType tree_model_column_type (GtkTreeModel *model, gint index);
476
477 static gboolean tree_model_get_iter (GtkTreeModel *model, GtkTreeIter *iter,
478                                      GtkTreePath *path);
479
480 static gboolean tree_model_iter_next (GtkTreeModel *model, GtkTreeIter *iter);
481
482 static GtkTreePath * tree_model_get_path (GtkTreeModel *model,
483                                           GtkTreeIter *iter);
484
485 static void tree_model_get_value (GtkTreeModel *model, GtkTreeIter *iter,
486                                   gint column, GValue *value);
487
488 static gboolean tree_model_nth_child (GtkTreeModel *model, GtkTreeIter *iter,
489                                       GtkTreeIter *parent, gint n);
490
491 static gint tree_model_n_children (GtkTreeModel *tree_model,
492                                    GtkTreeIter  *iter);
493
494 static gboolean tree_model_iter_children (GtkTreeModel *,
495                                           GtkTreeIter *,
496                                           GtkTreeIter *);
497
498 static gboolean tree_model_iter_parent (GtkTreeModel *tree_model,
499                                         GtkTreeIter *iter,
500                                         GtkTreeIter *child);
501
502 static gboolean tree_model_iter_has_child  (GtkTreeModel *tree_model,
503                                             GtkTreeIter  *iter);
504
505 static void
506 dictionary_tree_model_init (GtkTreeModelIface *iface)
507 {
508   iface->get_flags = tree_model_get_flags;
509   iface->get_n_columns = tree_model_n_columns;
510   iface->get_column_type = tree_model_column_type;
511   iface->get_iter = tree_model_get_iter;
512   iface->iter_next = tree_model_iter_next;
513   iface->get_path = tree_model_get_path;
514   iface->get_value = tree_model_get_value;
515
516   iface->iter_children = tree_model_iter_children ;
517   iface->iter_has_child = tree_model_iter_has_child ;
518   iface->iter_n_children = tree_model_n_children ;
519   iface->iter_nth_child = tree_model_nth_child ;
520   iface->iter_parent = tree_model_iter_parent ;
521 }
522
523 static gboolean
524 tree_model_iter_has_child  (GtkTreeModel *tree_model,
525                             GtkTreeIter  *iter)
526 {
527   return FALSE;
528 }
529
530 static gboolean
531 tree_model_iter_parent (GtkTreeModel *tree_model,
532                         GtkTreeIter *iter,
533                         GtkTreeIter *child)
534 {
535   return TRUE;
536 }
537
538 static GtkTreeModelFlags
539 tree_model_get_flags (GtkTreeModel *model)
540 {
541   g_return_val_if_fail (PSPPIRE_IS_DICT (model), (GtkTreeModelFlags) 0);
542
543   return GTK_TREE_MODEL_LIST_ONLY;
544 }
545
546
547 static gint
548 tree_model_n_columns (GtkTreeModel *model)
549 {
550   return n_DICT_COLS;
551 }
552
553 static GType
554 tree_model_column_type (GtkTreeModel *model, gint index)
555 {
556   g_return_val_if_fail (PSPPIRE_IS_DICT (model), (GType) 0);
557
558   switch (index)
559     {
560     case DICT_TVM_COL_NAME:
561       return G_TYPE_STRING;
562       break;
563     case DICT_TVM_COL_VAR:
564       return G_TYPE_POINTER;
565       break;
566     default:
567       g_return_val_if_reached ((GType)0);
568       break;
569     }
570
571   g_assert_not_reached ();
572   return ((GType)0);
573 }
574
575 static gboolean
576 tree_model_get_iter (GtkTreeModel *model, GtkTreeIter *iter, GtkTreePath *path)
577 {
578   gint *indices, depth;
579   gint n;
580   struct variable *var;
581
582   PsppireDict *dict = PSPPIRE_DICT (model);
583
584   g_return_val_if_fail (path, FALSE);
585
586   indices = gtk_tree_path_get_indices (path);
587   depth = gtk_tree_path_get_depth (path);
588
589   g_return_val_if_fail (depth == 1, FALSE);
590
591   n = indices [0];
592
593   if ( n < 0 || n >= psppire_dict_get_var_cnt (dict))
594     {
595       iter->stamp = 0;
596       iter->user_data = NULL;
597       return FALSE;
598     }
599
600   var = psppire_dict_get_variable (dict, n);
601
602   g_assert (var_get_dict_index (var) == n);
603
604   iter->stamp = dict->stamp;
605   iter->user_data = var;
606
607   return TRUE;
608 }
609
610
611 static gboolean
612 tree_model_iter_next (GtkTreeModel *model, GtkTreeIter *iter)
613 {
614   PsppireDict *dict = PSPPIRE_DICT (model);
615   struct variable *var;
616   gint idx;
617
618   g_return_val_if_fail (iter->stamp == dict->stamp, FALSE);
619
620   if ( iter == NULL || iter->user_data == NULL)
621     return FALSE;
622
623   var = iter->user_data;
624
625   idx = var_get_dict_index (var);
626
627   if ( idx + 1 >= psppire_dict_get_var_cnt (dict))
628     {
629       iter->user_data = NULL;
630       iter->stamp = 0;
631       return FALSE;
632     }
633
634   var = psppire_dict_get_variable (dict, idx + 1);
635
636   g_assert (var_get_dict_index (var) == idx + 1);
637
638   iter->user_data = var;
639
640   return TRUE;
641 }
642
643 static GtkTreePath *
644 tree_model_get_path (GtkTreeModel *model, GtkTreeIter *iter)
645 {
646   GtkTreePath *path;
647   struct variable *var;
648   PsppireDict *dict = PSPPIRE_DICT (model);
649
650   g_return_val_if_fail (iter->stamp == dict->stamp, FALSE);
651
652   var = iter->user_data;
653
654   path = gtk_tree_path_new ();
655   gtk_tree_path_append_index (path, var_get_dict_index (var));
656
657   return path;
658 }
659
660
661 static void
662 tree_model_get_value (GtkTreeModel *model, GtkTreeIter *iter,
663                       gint column, GValue *value)
664 {
665   struct variable *var;
666   PsppireDict *dict = PSPPIRE_DICT (model);
667
668   g_return_if_fail (iter->stamp == dict->stamp);
669
670   var =  iter->user_data;
671
672   switch (column)
673     {
674     case DICT_TVM_COL_NAME:
675       {
676       gchar *name = pspp_locale_to_utf8(var_get_name (var), -1, NULL);
677       g_value_init (value, G_TYPE_STRING);
678       g_value_set_string (value, name);
679       g_free (name);
680       }
681       break;
682     case DICT_TVM_COL_VAR:
683       g_value_init (value, G_TYPE_POINTER);
684       g_value_set_pointer (value, var);
685       break;
686     default:
687       g_return_if_reached ();
688       break;
689     }
690 }
691
692 static gboolean
693 tree_model_iter_children (GtkTreeModel *tree_model,
694                           GtkTreeIter *iter,
695                           GtkTreeIter *parent)
696 {
697   return FALSE;
698 }
699
700 static gint
701 tree_model_n_children (GtkTreeModel *model,
702                        GtkTreeIter  *iter)
703 {
704   PsppireDict *dict = PSPPIRE_DICT (model);
705
706   if ( iter == NULL )
707     return psppire_dict_get_var_cnt (dict);
708
709   return 0;
710 }
711
712 static gboolean
713 tree_model_nth_child (GtkTreeModel *model, GtkTreeIter *iter,
714                       GtkTreeIter *parent, gint n)
715 {
716   PsppireDict *dict;
717
718   g_return_val_if_fail (PSPPIRE_IS_DICT (model), FALSE);
719
720   dict = PSPPIRE_DICT (model);
721
722   if ( parent )
723     return FALSE;
724
725   if ( n >= psppire_dict_get_var_cnt (dict) )
726     return FALSE;
727
728   iter->stamp = dict->stamp;
729   iter->user_data = psppire_dict_get_variable (dict, n);
730
731   if ( !iter->user_data)
732     return FALSE;
733
734   return TRUE;
735 }
736
737
738 void
739 psppire_dict_rename_var (PsppireDict *dict, struct variable *v,
740                          const gchar *text)
741 {
742   dict_rename_var (dict->dict, v, text);
743 }
744
745
746 struct variable *
747 psppire_dict_get_weight_variable (const PsppireDict *dict)
748 {
749   return dict_get_weight (dict->dict);
750 }