treewide: Replace <name>_cnt by n_<name>s and <name>_cap by allocated_<name>.
[pspp] / src / language / stats / rank.c
index cd3833dbba7a78583ad0aa5016754c8990ad863c..ecf9a65914bd6403ceb3defd673463b3dd647cff 100644 (file)
@@ -91,14 +91,14 @@ enum rank_func
   };
 
 static const struct fmt_spec dest_format[n_RANK_FUNCS] = {
-  {FMT_F, 9, 3}, /* rank */
-  {FMT_F, 6, 4}, /* normal */
-  {FMT_F, 6, 2}, /* percent */
-  {FMT_F, 6, 4}, /* rfraction */
-  {FMT_F, 6, 4}, /* proportion */
-  {FMT_F, 6, 0}, /* n */
-  {FMT_F, 3, 0}, /* ntiles */
-  {FMT_F, 8, 4}  /* savage */
+  [RANK]       = { .type = FMT_F, .w = 9, .d = 3 },
+  [NORMAL]     = { .type = FMT_F, .w = 6, .d = 4 },
+  [PERCENT]    = { .type = FMT_F, .w = 6, .d = 2 },
+  [RFRACTION]  = { .type = FMT_F, .w = 6, .d = 4 },
+  [PROPORTION] = { .type = FMT_F, .w = 6, .d = 4 },
+  [N]          = { .type = FMT_F, .w = 6, .d = 0 },
+  [NTILES]     = { .type = FMT_F, .w = 3, .d = 0 },
+  [SAVAGE]     = { .type = FMT_F, .w = 8, .d = 4 }
 };
 
 static const char * const function_name[n_RANK_FUNCS] = {
@@ -287,16 +287,16 @@ parse_into (struct lexer *lexer, struct rank *cmd,
     }
   else if (lex_match_id (lexer, "NTILES"))
     {
-      if ( !lex_force_match (lexer, T_LPAREN))
+      if (!lex_force_match (lexer, T_LPAREN))
        return false;
 
-      if (! lex_force_int (lexer) )
+      if (! lex_force_int_range (lexer, "NTILES", 1, INT_MAX))
        return false;
 
       cmd->k_ntiles = lex_integer (lexer);
       lex_get (lexer);
 
-      if ( !lex_force_match (lexer, T_RPAREN))
+      if (!lex_force_match (lexer, T_RPAREN))
        return false;
 
       rs->rfunc = NTILES;
@@ -313,13 +313,13 @@ parse_into (struct lexer *lexer, struct rank *cmd,
 
   if (lex_match_id (lexer, "INTO"))
     {
-      while( lex_token (lexer) == T_ID )
+      while(lex_token (lexer) == T_ID)
        {
          const char *name = lex_tokcstr (lexer);
 
-         if ( var_count >= subcase_get_n_fields (&cmd->sc) )
+         if (var_count >= subcase_get_n_fields (&cmd->sc))
             msg (SE, _("Too many variables in %s clause."), "INTO");
-         else if ( dict_lookup_var (cmd->dict, name) != NULL )
+         else if (dict_lookup_var (cmd->dict, name) != NULL)
             msg (SE, _("Variable %s already exists."), name);
           else if (string_set_contains (new_names, name))
             msg (SE, _("Duplicate variable name %s."), name);
@@ -354,7 +354,7 @@ rank_rank (const struct rank *cmd, double c, double cc, double cc_1,
 {
   double rank;
 
-  if ( c >= 1.0 )
+  if (c >= 1.0)
     {
       switch (cmd->ties)
        {
@@ -423,7 +423,7 @@ rank_proportion (const struct rank *cmd, double c, double cc, double cc_1,
 
   double f;
 
-  switch ( cmd->fraction )
+  switch (cmd->fraction)
     {
     case FRAC_BLOM:
       f =  (r - 3.0/8.0) / (w + 0.25);
@@ -435,7 +435,7 @@ rank_proportion (const struct rank *cmd, double c, double cc, double cc_1,
       f = (r - 1.0/3.0) / (w + 1.0/3.0);
       break;
     case FRAC_VW:
-      f = r / ( w + 1.0);
+      f = r / (w + 1.0);
       break;
     default:
       NOT_REACHED ();
@@ -461,7 +461,7 @@ rank_ntiles (const struct rank *cmd, double c, double cc, double cc_1,
   double r = rank_rank (cmd, c, cc, cc_1, i, w);
 
 
-  return ( floor (( r * cmd->k_ntiles) / ( w + 1) ) + 1);
+  return (floor ((r * cmd->k_ntiles) / (w + 1)) + 1);
 }
 
 /* Expected value of the order statistics from an exponential distribution */
@@ -472,7 +472,7 @@ ee (int j, double w_star)
   double sum = 0.0;
 
   for (k = 1 ; k <= j; k++)
-    sum += 1.0 / ( w_star + 1 - k );
+    sum += 1.0 / (w_star + 1 - k);
 
   return sum;
 }
@@ -486,29 +486,29 @@ rank_savage (const struct rank *cmd UNUSED, double c, double cc, double cc_1,
   const int i_1 = floor (cc_1);
   const int i_2 = floor (cc);
 
-  const double w_star = (modf (w, &int_part) == 0 ) ? w : floor (w) + 1;
+  const double w_star = (modf (w, &int_part) == 0) ? w : floor (w) + 1;
 
   const double g_1 = cc_1 - i_1;
   const double g_2 = cc - i_2;
 
   /* The second factor is infinite, when the first is zero.
      Therefore, evaluate the second, only when the first is non-zero */
-  const double expr1 =  (1 - g_1) ? (1 - g_1) * ee(i_1+1, w_star) : ( 1 - g_1);
+  const double expr1 =  (1 - g_1) ? (1 - g_1) * ee(i_1+1, w_star) : (1 - g_1);
   const double expr2 =  g_2 ? g_2 * ee (i_2+1, w_star) : g_2 ;
 
-  if ( i_1 == i_2 )
+  if (i_1 == i_2)
     return ee (i_1 + 1, w_star) - 1;
 
-  if ( i_1 + 1 == i_2 )
-    return ( ( expr1 + expr2 )/c ) - 1;
+  if (i_1 + 1 == i_2)
+    return ((expr1 + expr2)/c) - 1;
 
-  if ( i_1 + 2 <= i_2 )
+  if (i_1 + 2 <= i_2)
     {
       int j;
       double sigma = 0.0;
-      for (j = i_1 + 2 ; j <= i_2; ++j )
+      for (j = i_1 + 2 ; j <= i_2; ++j)
        sigma += ee (j, w_star);
-      return ( (expr1 + expr2 + sigma) / c) -1;
+      return ((expr1 + expr2 + sigma) / c) -1;
     }
 
   NOT_REACHED();
@@ -573,12 +573,12 @@ rank_sorted_file (struct casereader *input,
           size_t i;
 
           out_case = case_create (casewriter_get_proto (output));
-          case_data_rw_idx (out_case, 0)->f = case_num_idx (c, 1);
+          *case_num_rw_idx (out_case, 0) = case_num_idx (c, 1);
           for (i = 0; i < cmd->n_rs; ++i)
             {
               rank_function_t func = rank_func[cmd->rs[i].rfunc];
               double rank = func (cmd, tw, cc, cc_1, tie_group, w);
-              case_data_rw_idx (out_case, i + 1)->f = rank;
+              *case_num_rw_idx (out_case, i + 1) = rank;
             }
 
           casewriter_write (output, out_case);
@@ -595,7 +595,7 @@ rank_cmd (struct dataset *ds,  const struct rank *cmd);
 static const char *
 fraction_name (const struct rank *cmd)
 {
-  switch (cmd->fraction )
+  switch (cmd->fraction)
     {
     case FRAC_BLOM:   return "BLOM";
     case FRAC_RANKIT: return "RANKIT";
@@ -615,16 +615,16 @@ create_var_label (struct rank *cmd, const struct variable *src_var,
 
   ds_init_empty (&label);
 
-  if ( cmd->n_group_vars > 0 )
+  if (cmd->n_group_vars > 0)
     {
       struct string group_var_str;
       int g;
 
       ds_init_empty (&group_var_str);
 
-      for (g = 0 ; g < cmd->n_group_vars ; ++g )
+      for (g = 0 ; g < cmd->n_group_vars ; ++g)
        {
-         if ( g > 0 ) ds_put_cstr (&group_var_str, " ");
+         if (g > 0) ds_put_cstr (&group_var_str, " ");
          ds_put_cstr (&group_var_str, var_get_name (cmd->group_vars[g]));
        }
 
@@ -661,6 +661,7 @@ cmd_rank (struct lexer *lexer, struct dataset *ds)
   rank.ties = TIES_MEAN;
   rank.fraction = FRAC_BLOM;
   rank.print = true;
+  rank.vars = NULL;
   rank.pool = pool_create ();
 
   string_set_init (&new_names);
@@ -676,16 +677,16 @@ cmd_rank (struct lexer *lexer, struct dataset *ds)
 
   rank.n_vars = rank.sc.n_fields;
 
-  if (lex_match (lexer, T_BY) )
+  if (lex_match (lexer, T_BY))
     {
-      if ( ! parse_variables_const (lexer, rank.dict,
+      if (! parse_variables_const (lexer, rank.dict,
                                    &rank.group_vars, &rank.n_group_vars,
                                    PV_NO_DUPLICATE | PV_NO_SCRATCH))
        goto error;
     }
 
 
-  while (lex_token (lexer) != T_ENDCMD )
+  while (lex_token (lexer) != T_ENDCMD)
     {
       if (! lex_force_match (lexer, T_SLASH))
        goto error;
@@ -783,7 +784,7 @@ cmd_rank (struct lexer *lexer, struct dataset *ds)
 
 
   /* If no rank specs are given, then apply a default */
-  if ( rank.n_rs == 0)
+  if (rank.n_rs == 0)
     {
       struct rank_spec *rs;
 
@@ -802,10 +803,10 @@ cmd_rank (struct lexer *lexer, struct dataset *ds)
     {
       rs->dest_labels = pool_calloc (rank.pool, rank.n_vars,
                                      sizeof *rs->dest_labels);
-      for (int v = 0 ; v < rank.n_vars ;  v ++ )
+      for (int v = 0 ; v < rank.n_vars ;  v ++)
         {
           const char **dst_name = &rs->dest_names[v];
-          if ( *dst_name == NULL )
+          if (*dst_name == NULL)
             {
               *dst_name = rank_choose_dest_name (rank.dict, &new_names,
                                                  rs->rfunc,
@@ -819,11 +820,10 @@ cmd_rank (struct lexer *lexer, struct dataset *ds)
         }
     }
 
-  if ( rank.print )
+  if (rank.print)
     {
       struct pivot_table *table = pivot_table_create (
         N_("Variables Created by RANK"));
-      table->omit_empty = true;
 
       pivot_dimension_create (table, PIVOT_AXIS_COLUMN, N_("New Variable"),
                               N_("New Variable"), N_("Function"),
@@ -834,15 +834,15 @@ cmd_rank (struct lexer *lexer, struct dataset *ds)
         N_("Existing Variable"));
       variables->root->show_label = true;
 
-      for (size_t i = 0 ; i <  rank.n_rs ; ++i )
+      for (size_t i = 0 ; i <  rank.n_rs ; ++i)
        {
-         for (size_t v = 0 ; v < rank.n_vars ;  v ++ )
+         for (size_t v = 0 ; v < rank.n_vars ;  v ++)
            {
               int row_idx = pivot_category_create_leaf (
                 variables->root, pivot_value_new_variable (rank.vars[v]));
 
               struct string group_vars = DS_EMPTY_INITIALIZER;
-              for (int g = 0 ; g < rank.n_group_vars ; ++g )
+              for (int g = 0 ; g < rank.n_group_vars ; ++g)
                 {
                   if (g)
                     ds_put_byte (&group_vars, ' ');
@@ -865,6 +865,7 @@ cmd_rank (struct lexer *lexer, struct dataset *ds)
                     pivot_table_put2 (table, j, row_idx,
                                       pivot_value_new_user_text (entry, -1));
                 }
+              ds_destroy (&group_vars);
            }
        }
 
@@ -928,7 +929,7 @@ rank_trns_proc (void *trns_, struct ccase **c, casenumber case_idx UNUSED)
             size_t i;
 
             for (i = 0; i < trns->n_funcs; i++)
-              case_data_rw (*c, iv->output_vars[i])->f
+              *case_num_rw (*c, iv->output_vars[i])
                 = case_num_idx (iv->current, i + 1);
             advance_ranking (iv);
             break;
@@ -996,7 +997,7 @@ rank_cmd (struct dataset *ds, const struct rank *cmd)
   /* Open the active file and make one pass per input variable. */
   input = proc_open (ds);
   input = casereader_create_filter_weight (input, d, NULL, NULL);
-  for (i = 0 ; i < cmd->n_vars ; ++i )
+  for (i = 0 ; i < cmd->n_vars ; ++i)
     {
       const struct variable *input_var = cmd->vars[i];
       struct casereader *input_pass;
@@ -1031,11 +1032,11 @@ rank_cmd (struct dataset *ds, const struct rank *cmd)
       subcase_add_vars_always (&projection,
                                cmd->group_vars, cmd->n_group_vars);
       subcase_add_vars_always (&projection, dict_get_split_vars (d),
-                               dict_get_split_cnt (d));
+                               dict_get_n_splits (d));
       if (weight_var != NULL)
         {
           subcase_add_var_always (&projection, weight_var, SC_ASCEND);
-          weight_idx = 2 + cmd->n_group_vars + dict_get_split_cnt (d);
+          weight_idx = 2 + cmd->n_group_vars + dict_get_n_splits (d);
         }
       else
         weight_idx = -1;
@@ -1056,7 +1057,7 @@ rank_cmd (struct dataset *ds, const struct rank *cmd)
 
       /* Group by split variables */
       subcase_init_empty (&split_vars);
-      for (j = 0; j < dict_get_split_cnt (d); j++)
+      for (j = 0; j < dict_get_n_splits (d); j++)
         subcase_add_always (&split_vars, 2 + j + cmd->n_group_vars,
                             var_get_width (dict_get_split_vars (d)[j]),
                             SC_ASCEND);