/* vim: set sw=8: -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */ /* * analysis-signed-rank-test.c: * * Author: * Andreas J. Guelzow * * (C) Copyright 2010 by Andreas J. Guelzow * * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ #include #include #include "gnumeric.h" #include "analysis-signed-rank-test.h" #include "analysis-tools.h" #include "value.h" #include "ranges.h" #include "expr.h" #include "func.h" #include "numbers.h" static inline GnmExpr const * make_int (int n) { return gnm_expr_new_constant (value_new_int (n)); } static inline GnmExpr const * make_float (gnm_float x) { return gnm_expr_new_constant (value_new_float (x)); } static gboolean analysis_tool_signed_rank_test_engine_run (data_analysis_output_t *dao, analysis_tools_data_sign_test_t *info) { guint col; GSList *data = info->base.input; gboolean first = TRUE; GnmExpr const *expr; GnmExpr const *expr_isnumber; GnmExpr const *expr_diff; GnmExpr const *expr_expect; GnmExpr const *expr_var; GnmExpr const *expr_abs; GnmExpr const *expr_big; GnmFunc *fd_median = analysis_tool_get_function ("MEDIAN", dao); GnmFunc *fd_if = analysis_tool_get_function ("IF", dao); GnmFunc *fd_sum = analysis_tool_get_function ("SUM", dao); GnmFunc *fd_min = analysis_tool_get_function ("MIN", dao); GnmFunc *fd_normdist = analysis_tool_get_function ("NORMDIST", dao); GnmFunc *fd_isnumber = analysis_tool_get_function ("ISNUMBER", dao); GnmFunc *fd_iferror = analysis_tool_get_function ("IFERROR", dao); GnmFunc *fd_rank = analysis_tool_get_function ("RANK.AVG", dao); GnmFunc *fd_abs = analysis_tool_get_function ("ABS", dao); GnmFunc *fd_sqrt = analysis_tool_get_function ("SQRT", dao); GnmFunc *fd_max = analysis_tool_get_function ("MAX", dao); dao_set_italic (dao, 0, 0, 0, 9); set_cell_text_col (dao, 0, 0, _("/Wilcoxon Signed Rank Test" "/Median" "/Predicted Median" "/N" "/S\xe2\x88\x92" "/S+" "/Test Statistic" "/\xce\xb1" "/P(T\xe2\x89\xa4t) one-tailed" "/P(T\xe2\x89\xa4t) two-tailed")); for (col = 0; data != NULL; data = data->next, col++) { GnmValue *val_org = value_dup (data->data); GnmExpr const *expr_org; /* Note that analysis_tools_write_label may modify val_org */ dao_set_italic (dao, col + 1, 0, col+1, 0); analysis_tools_write_label (val_org, dao, &info->base, col + 1, 0, col + 1); expr_org = gnm_expr_new_constant (val_org); if (first) { dao_set_cell_float (dao, col + 1, 2, info->median); dao_set_cell_float (dao, col + 1, 7, info->alpha); first = FALSE; } else { dao_set_cell_expr (dao, col + 1, 2, make_cellref (-1,0)); dao_set_cell_expr (dao, col + 1, 7, make_cellref (-1,0)); } expr_isnumber = gnm_expr_new_funcall3 (fd_if, gnm_expr_new_funcall1 (fd_isnumber, gnm_expr_copy (expr_org)), make_int (1), make_int (0)); expr = gnm_expr_new_funcall1 (fd_median, gnm_expr_copy (expr_org)); dao_set_cell_expr (dao, col + 1, 1, expr); expr_diff = gnm_expr_new_binary (gnm_expr_copy (expr_org), GNM_EXPR_OP_SUB, make_cellref (0,-2)); expr_abs = gnm_expr_new_funcall1 (fd_abs, gnm_expr_copy (expr_diff)); expr_big = gnm_expr_new_binary (gnm_expr_new_funcall1 (fd_max, gnm_expr_copy (expr_abs)), GNM_EXPR_OP_ADD, make_int (1)); expr = gnm_expr_new_funcall3 (fd_if, gnm_expr_new_funcall1 (fd_isnumber, gnm_expr_copy (expr_org)), gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (gnm_expr_copy (expr_org), GNM_EXPR_OP_EQUAL, make_cellref (0,-2)), gnm_expr_copy (expr_big), expr_abs), expr_big); expr = gnm_expr_new_funcall3 (fd_rank, gnm_expr_new_unary (GNM_EXPR_OP_UNARY_NEG, expr_diff), expr, make_int (1)); dao_set_cell_array_expr (dao, col + 1, 4, gnm_expr_new_funcall1 (fd_sum, gnm_expr_new_binary (gnm_expr_copy (expr_isnumber), GNM_EXPR_OP_MULT, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (gnm_expr_copy (expr_org), GNM_EXPR_OP_LT, make_cellref (0,-2)), expr, make_int (0))))); expr = gnm_expr_new_funcall1 (fd_sum, gnm_expr_new_binary (expr_isnumber, GNM_EXPR_OP_MULT, gnm_expr_new_funcall2 (fd_iferror, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (expr_org, GNM_EXPR_OP_NOT_EQUAL, make_cellref (0,-1)), make_int (1), make_int (0)), make_int (0)))); dao_set_cell_array_expr (dao, col + 1, 3, expr); dao_set_cell_expr (dao, col + 1, 5, gnm_expr_new_binary (gnm_expr_new_binary (gnm_expr_new_binary (make_cellref (0,-2), GNM_EXPR_OP_MULT, gnm_expr_new_binary (make_cellref (0,-2), GNM_EXPR_OP_ADD, make_int (1))), GNM_EXPR_OP_DIV, make_int (2)), GNM_EXPR_OP_SUB, make_cellref (0,-1))); dao_set_cell_expr (dao, col + 1, 6, gnm_expr_new_funcall2 (fd_min, make_cellref (0,-1), make_cellref (0,-2))); expr_expect = gnm_expr_new_binary (gnm_expr_new_binary (make_cellref (0,-5), GNM_EXPR_OP_MULT, gnm_expr_new_binary (make_cellref (0,-5), GNM_EXPR_OP_ADD, make_int (1))), GNM_EXPR_OP_DIV, make_int (4)); expr_var = gnm_expr_new_binary (gnm_expr_new_binary (gnm_expr_copy (expr_expect), GNM_EXPR_OP_MULT, gnm_expr_new_binary (gnm_expr_new_binary (make_int (2), GNM_EXPR_OP_MULT, make_cellref (0,-5)), GNM_EXPR_OP_ADD, make_int (1))), GNM_EXPR_OP_DIV, make_int (6)); expr = gnm_expr_new_funcall4 (fd_normdist, gnm_expr_new_binary (make_cellref (0,-2), GNM_EXPR_OP_ADD, make_float (0.5)), expr_expect, gnm_expr_new_funcall1 (fd_sqrt, expr_var), gnm_expr_new_constant (value_new_bool (TRUE))); dao_set_cell_expr (dao, col + 1, 8, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (make_cellref (0,-5), GNM_EXPR_OP_LT, make_int (12)), gnm_expr_new_constant (value_new_error_NA (NULL)), expr)); dao_set_cell_comment (dao, col + 1, 8, _("This p-value is calculated by a normal approximation.\n" "It is only valid if the sample size is at least 12.")); expr = gnm_expr_new_binary (make_int (2), GNM_EXPR_OP_MULT, make_cellref (0,-1)); dao_set_cell_expr (dao, col + 1, 9, expr); } gnm_func_unref (fd_median); gnm_func_unref (fd_if); gnm_func_unref (fd_min); gnm_func_unref (fd_sum); gnm_func_unref (fd_normdist); gnm_func_unref (fd_isnumber); gnm_func_unref (fd_iferror); gnm_func_unref (fd_rank); gnm_func_unref (fd_abs); gnm_func_unref (fd_sqrt); gnm_func_unref (fd_max); dao_redraw_respan (dao); return FALSE; } static gboolean analysis_tool_signed_rank_test_two_engine_run (data_analysis_output_t *dao, analysis_tools_data_sign_test_two_t *info) { GnmValue *val_1; GnmValue *val_2; GnmExpr const *expr_1; GnmExpr const *expr_2; GnmExpr const *expr; GnmExpr const *expr_diff; GnmExpr const *expr_diff_pred; GnmExpr const *expr_isnumber_1; GnmExpr const *expr_isnumber_2; GnmExpr const *expr_isnumber; GnmExpr const *expr_expect; GnmExpr const *expr_var; GnmExpr const *expr_abs; GnmExpr const *expr_big; GnmFunc *fd_median = analysis_tool_get_function ("MEDIAN", dao); GnmFunc *fd_if = analysis_tool_get_function ("IF", dao); GnmFunc *fd_sum = analysis_tool_get_function ("SUM", dao); GnmFunc *fd_min = analysis_tool_get_function ("MIN", dao); GnmFunc *fd_normdist = analysis_tool_get_function ("NORMDIST", dao); GnmFunc *fd_isnumber = analysis_tool_get_function ("ISNUMBER", dao); GnmFunc *fd_iferror = analysis_tool_get_function ("IFERROR", dao); GnmFunc *fd_rank = analysis_tool_get_function ("RANK.AVG", dao); GnmFunc *fd_abs = analysis_tool_get_function ("ABS", dao); GnmFunc *fd_sqrt = analysis_tool_get_function ("SQRT", dao); GnmFunc *fd_max = analysis_tool_get_function ("MAX", dao); dao_set_italic (dao, 0, 0, 0, 10); set_cell_text_col (dao, 0, 0, _("/Wilcoxon Signed Rank Test" "/Median" "/Observed Median Difference" "/Predicted Median Difference" "/N" "/S\xe2\x88\x92" "/S+" "/Test Statistic" "/\xce\xb1" "/P(T\xe2\x89\xa4t) one-tailed" "/P(T\xe2\x89\xa4t) two-tailed")); val_1 = value_dup (info->base.range_1); val_2 = value_dup (info->base.range_2); /* Labels */ dao_set_italic (dao, 1, 0, 2, 0); analysis_tools_write_label_ftest (val_1, dao, 1, 0, info->base.labels, 1); analysis_tools_write_label_ftest (val_2, dao, 2, 0, info->base.labels, 2); expr_1 = gnm_expr_new_constant (value_dup (val_1)); expr_2 = gnm_expr_new_constant (value_dup (val_2)); dao_set_cell_float (dao, 1, 3, info->median); dao_set_cell_float (dao, 1, 8, info->base.alpha); expr_isnumber_1 = gnm_expr_new_funcall3 (fd_if, gnm_expr_new_funcall1 (fd_isnumber, gnm_expr_copy (expr_1)), make_int (1), make_int (0)); expr_isnumber_2 = gnm_expr_new_funcall3 (fd_if, gnm_expr_new_funcall1 (fd_isnumber, gnm_expr_copy (expr_2)), make_int (1), make_int (0)); expr_isnumber = gnm_expr_new_binary (expr_isnumber_1, GNM_EXPR_OP_MULT, expr_isnumber_2); expr = gnm_expr_new_funcall1 (fd_median, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (gnm_expr_copy (expr_isnumber), GNM_EXPR_OP_EQUAL, make_int (1)), gnm_expr_copy (expr_1), gnm_expr_new_constant (value_new_string("")))); dao_set_cell_array_expr (dao, 1, 1, expr); expr = gnm_expr_new_funcall1 (fd_median, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (gnm_expr_copy (expr_isnumber), GNM_EXPR_OP_EQUAL, make_int (1)), gnm_expr_copy (expr_2), gnm_expr_new_constant (value_new_string("")))); dao_set_cell_array_expr (dao, 2, 1, expr); expr_diff = gnm_expr_new_binary (gnm_expr_copy (expr_1), GNM_EXPR_OP_SUB, gnm_expr_copy (expr_2)); dao_set_cell_array_expr (dao, 1, 2, gnm_expr_new_funcall1 (fd_median, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (gnm_expr_copy (expr_isnumber), GNM_EXPR_OP_EQUAL, make_int (1)), gnm_expr_copy (expr_diff), gnm_expr_new_constant (value_new_string(""))))); expr = gnm_expr_new_funcall1 (fd_sum, gnm_expr_new_binary (gnm_expr_copy (expr_isnumber), GNM_EXPR_OP_MULT, gnm_expr_new_funcall2 (fd_iferror, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (gnm_expr_copy (expr_diff), GNM_EXPR_OP_NOT_EQUAL, make_cellref (0,-2)), make_int (1), make_int (0)), make_int (0)))); dao_set_cell_array_expr (dao, 1, 4, expr); expr_diff_pred = gnm_expr_new_binary (gnm_expr_copy (expr_diff), GNM_EXPR_OP_SUB, make_cellref (0,-2)); expr_abs = gnm_expr_new_funcall1 (fd_abs, gnm_expr_copy (expr_diff_pred)); expr_big = gnm_expr_new_binary (gnm_expr_new_funcall1 (fd_max, gnm_expr_copy (expr_abs)), GNM_EXPR_OP_ADD, make_int (1)); expr = gnm_expr_new_funcall3 (fd_if, gnm_expr_new_funcall1 (fd_isnumber, expr_1), gnm_expr_new_funcall3 (fd_if, gnm_expr_new_funcall1 (fd_isnumber, expr_2), gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (gnm_expr_copy (expr_diff), GNM_EXPR_OP_EQUAL, make_cellref (0,-2)), gnm_expr_copy (expr_big), expr_abs), gnm_expr_copy (expr_big)), expr_big); expr = gnm_expr_new_funcall3 (fd_rank, gnm_expr_new_unary (GNM_EXPR_OP_UNARY_NEG, expr_diff_pred), expr, make_int (1)); expr = gnm_expr_new_funcall1 (fd_sum, gnm_expr_new_binary (expr_isnumber, GNM_EXPR_OP_MULT, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (expr_diff, GNM_EXPR_OP_LT, make_cellref (0,-2)), expr, make_int (0)))); dao_set_cell_array_expr (dao, 1, 5, expr); dao_set_cell_expr (dao, 1, 6, gnm_expr_new_binary (gnm_expr_new_binary (gnm_expr_new_binary (make_cellref (0,-2), GNM_EXPR_OP_MULT, gnm_expr_new_binary (make_cellref (0,-2), GNM_EXPR_OP_ADD, make_int (1))), GNM_EXPR_OP_DIV, make_int (2)), GNM_EXPR_OP_SUB, make_cellref (0,-1))); dao_set_cell_expr (dao, 1, 7, gnm_expr_new_funcall2 (fd_min, make_cellref (0,-1), make_cellref (0,-2))); expr_expect = gnm_expr_new_binary (gnm_expr_new_binary (make_cellref (0,-5), GNM_EXPR_OP_MULT, gnm_expr_new_binary (make_cellref (0,-5), GNM_EXPR_OP_ADD, make_int (1))), GNM_EXPR_OP_DIV, make_int (4)); expr_var = gnm_expr_new_binary (gnm_expr_new_binary (gnm_expr_copy (expr_expect), GNM_EXPR_OP_MULT, gnm_expr_new_binary (gnm_expr_new_binary (make_int (2), GNM_EXPR_OP_MULT, make_cellref (0,-5)), GNM_EXPR_OP_ADD, make_int (1))), GNM_EXPR_OP_DIV, make_int (6)); expr = gnm_expr_new_funcall4 (fd_normdist, gnm_expr_new_binary (make_cellref (0,-2), GNM_EXPR_OP_ADD, make_float (0.5)), expr_expect, gnm_expr_new_funcall1 (fd_sqrt, expr_var), gnm_expr_new_constant (value_new_bool (TRUE))); dao_set_cell_expr (dao, 1, 9, gnm_expr_new_funcall3 (fd_if, gnm_expr_new_binary (make_cellref (0,-5), GNM_EXPR_OP_LT, make_int (12)), gnm_expr_new_constant (value_new_error_NA (NULL)), expr)); dao_set_cell_comment (dao, 1, 9, _("This p-value is calculated by a normal approximation.\n" "It is only valid if the sample size is at least 12.")); expr = gnm_expr_new_binary (make_int (2), GNM_EXPR_OP_MULT, make_cellref (0,-1)); dao_set_cell_array_expr (dao, 1, 10, expr); gnm_func_unref (fd_median); gnm_func_unref (fd_if); gnm_func_unref (fd_min); gnm_func_unref (fd_sum); gnm_func_unref (fd_normdist); gnm_func_unref (fd_isnumber); gnm_func_unref (fd_iferror); gnm_func_unref (fd_rank); gnm_func_unref (fd_abs); gnm_func_unref (fd_sqrt); gnm_func_unref (fd_max); value_release (val_1); value_release (val_2); dao_redraw_respan (dao); return FALSE; } gboolean analysis_tool_signed_rank_test_engine (G_GNUC_UNUSED GOCmdContext *gcc, data_analysis_output_t *dao, gpointer specs, analysis_tool_engine_t selector, gpointer result) { analysis_tools_data_sign_test_t *info = specs; switch (selector) { case TOOL_ENGINE_UPDATE_DESCRIPTOR: return (dao_command_descriptor (dao, _("Wilcoxon Signed Rank Test (%s)"), result) == NULL); case TOOL_ENGINE_UPDATE_DAO: prepare_input_range (&info->base.input, info->base.group_by); dao_adjust (dao, 1 + g_slist_length (info->base.input), 10); return FALSE; case TOOL_ENGINE_CLEAN_UP: return analysis_tool_generic_clean (specs); case TOOL_ENGINE_LAST_VALIDITY_CHECK: return FALSE; case TOOL_ENGINE_PREPARE_OUTPUT_RANGE: dao_prepare_output (NULL, dao, _("Wilcoxon Signed Rank Test")); return FALSE; case TOOL_ENGINE_FORMAT_OUTPUT_RANGE: return dao_format_output (dao, _("Wilcoxon Signed Rank Test")); case TOOL_ENGINE_PERFORM_CALC: default: return analysis_tool_signed_rank_test_engine_run (dao, specs); } return TRUE; } gboolean analysis_tool_signed_rank_test_two_engine (G_GNUC_UNUSED GOCmdContext *gcc, data_analysis_output_t *dao, gpointer specs, analysis_tool_engine_t selector, gpointer result) { switch (selector) { case TOOL_ENGINE_UPDATE_DESCRIPTOR: return (dao_command_descriptor (dao, _("Wilcoxon Signed Rank Test (%s)"), result) == NULL); case TOOL_ENGINE_UPDATE_DAO: dao_adjust (dao, 3, 11); return FALSE; case TOOL_ENGINE_CLEAN_UP: return analysis_tool_generic_b_clean (specs); case TOOL_ENGINE_LAST_VALIDITY_CHECK: return FALSE; case TOOL_ENGINE_PREPARE_OUTPUT_RANGE: dao_prepare_output (NULL, dao, _("Wilcoxon Signed Rank Test")); return FALSE; case TOOL_ENGINE_FORMAT_OUTPUT_RANGE: return dao_format_output (dao, _("Wilcoxon Signed Rank Test")); case TOOL_ENGINE_PERFORM_CALC: default: return analysis_tool_signed_rank_test_two_engine_run (dao, specs); } return TRUE; /* We shouldn't get here */ }