jashmenn-apriori 0.1.0

data/ext/math/src/chi2.c

@@ -0,0 +1,151 @@
+/*----------------------------------------------------------------------
+  File    : chi2.c
+  Contents: chi^2 distribution function
+  Author  : Christian Borgelt
+  History : 2003.05.19 file created as quantile.c
+            2008.03.14 main programs added
+----------------------------------------------------------------------*/
+#if defined(CHI2PDF_MAIN) \
+ || defined(CHI2CDF_MAIN) \
+ || defined(CHI2QTL_MAIN)
+#include <stdio.h>
+#include <stdlib.h>
+#endif
+#ifdef CHI2QTL_MAIN
+#ifndef CHI2QTL
+#define CHI2QTL
+#endif
+#ifndef GAMMAQTL
+#define GAMMAQTL
+#endif
+#endif
+#include <assert.h>
+#include <math.h>
+#include "gamma.h"
+#ifdef CHI2QTL
+#include "normal.h"
+#endif
+#include "chi2.h"
+
+/*----------------------------------------------------------------------
+  Functions
+----------------------------------------------------------------------*/
+
+double chi2pdf (double x, double df)
+{                               /* --- probability density function */
+  assert(df > 0);               /* check the function arguments */
+  if (x <=  0) return 0;        /* only non-zero for positive arg. */
+  if (df == 2) return 0.5 *exp(-0.5*x);
+  df *= 0.5;                    /* compute probability density */
+  return 0.5 *exp((df-1) *log(0.5*x) -0.5*x -logGamma(df));
+}  /* chi2pdf() */
+
+/*--------------------------------------------------------------------*/
+
+double chi2cdfP (double x, double df)
+{                               /* --- cumulative distribution fn. */
+  assert(df > 0);               /* check the function arguments */
+  return GammaP(0.5*df, 0.5*x); /* compute regularized Gamma function */
+}  /* chi2cdfP() */
+
+/*--------------------------------------------------------------------*/
+
+double chi2cdfQ (double x, double df)
+{                               /* --- cumulative distribution fn. */
+  assert(df > 0);               /* check the function arguments */
+  return GammaQ(0.5*df, 0.5*x); /* compute regularized Gamma function */
+}  /* chi2cdfQ() */
+
+/*--------------------------------------------------------------------*/
+#ifdef CHI2QTL
+
+double chi2qtlP (double prob, double df)
+{                               /* --- quantile of chi2 distribution */
+  return GammaqtlP(prob, 0.5*df, 2);
+}  /* chi2qtlP() */
+
+/*--------------------------------------------------------------------*/
+
+double chi2qtlQ (double prob, double df)
+{                               /* --- quantile of chi2 distribution */
+  return GammaqtlQ(prob, 0.5*df, 2);
+}  /* chi2qtlQ() */
+
+#endif
+/*----------------------------------------------------------------------
+  Main Functions
+----------------------------------------------------------------------*/
+#ifdef CHI2PDF_MAIN
+
+int main (int argc, char *argv[])
+{                               /* --- main function */
+  double df = 1;                /* degrees of freedom */
+  double x;                     /* argument value */
+
+  if ((argc < 2) || (argc > 3)){/* if wrong number of arguments */
+    printf("usage: %s arg [df]\n", argv[0]);
+    printf("compute probability density function of the\n"
+           "chi^2 distribution with df degrees of freedom\n");
+    return 0;                   /* print a usage message */
+  }                             /* and abort the program */
+  x = atof(argv[1]);            /* get the argument value */
+  if (argc > 2) df = atof(argv[2]);
+  if (df <= 0) {                /* get the degrees of freedom */
+    printf("%s: invalid degrees of freedom\n", argv[0]); return -1; }
+  printf("chi^2: f(%.16g; %.16g) = %.16g\n", x, df, chi2pdf(x, df));
+  return 0;                     /* compute and print density */
+}  /* main() */
+
+#endif
+/*--------------------------------------------------------------------*/
+#ifdef CHI2CDF_MAIN
+
+int main (int argc, char *argv[])
+{                               /* --- main function */
+  double df = 1;                /* degrees of freedom */
+  double x;                     /* argument value */
+
+  if ((argc < 2) || (argc > 3)){/* if wrong number of arguments */
+    printf("usage: %s arg [df]\n", argv[0]);
+    printf("compute cumulative distribution function of the\n"
+           "chi^2 distribution with df degrees of freedom\n");
+    return 0;                   /* print a usage message */
+  }                             /* and abort the program */
+  x = atof(argv[1]);            /* get the argument value */
+  if (argc > 2) df = atof(argv[2]);
+  if (df <= 0) {                /* get the degrees of freedom */
+    printf("%s: invalid degrees of freedom\n", argv[0]); return -1; }
+  printf("chi^2: F(%.16g; %.16g) = %.16g\n", x, df, chi2cdfP(x, df));
+  printf("   1 - F(%.16g; %.16g) = %.16g\n", x, df, chi2cdfQ(x, df));
+  return 0;                     /* compute and print probability */
+}  /* main() */
+
+#endif
+/*--------------------------------------------------------------------*/
+#ifdef CHI2QTL_MAIN
+
+int main (int argc, char *argv[])
+{                               /* --- main function */
+  double df = 1;                /* degrees of freedom */
+  double prob;                  /* probability */
+
+  if ((argc < 2) || (argc > 3)){/* if wrong number of arguments */
+    printf("usage: %s prob [df]\n", argv[0]);
+    printf("compute quantile of the chi^2 distribution "
+           "with df degrees of freedom\n");
+    return 0;                   /* print a usage message */
+  }                             /* and abort the program */
+  prob = atof(argv[1]);         /* get the probability */
+  if ((prob < 0) || (prob > 1)) {
+    printf("%s: invalid probability\n", argv[0]); return -1; }
+  if (argc > 2) df = atof(argv[2]);
+  if (df <= 0) {                /* get the degrees of freedom */
+    printf("%s: invalid degrees of freedom\n", argv[0]); return -1; }
+  printf("chi^2: F(%.16g; %.16g) = %.16g\n",
+         chi2qtlP(prob, df), df, prob);
+  printf("   1 - F(%.16g; %.16g) = %.16g\n",
+         chi2qtlQ(prob, df), df, prob);
+  return 0;                     /* compute and print quantile */
+}  /* main() */
+
+#endif

data/ext/math/src/chi2.h

@@ -0,0 +1,27 @@
+/*----------------------------------------------------------------------
+  File    : chi2.h
+  Contents: chi^2 distribution functions
+  Author  : Christian Borgelt
+  History : 2003.05.19 file created as quantile.h
+----------------------------------------------------------------------*/
+#ifndef __CHI2__
+#define __CHI2__
+
+/*----------------------------------------------------------------------
+  Functions
+----------------------------------------------------------------------*/
+extern double chi2pdf  (double x,    double df);
+extern double chi2cdf  (double x,    double df);
+extern double chi2cdfP (double x,    double df);
+extern double chi2cdfQ (double x,    double df);
+extern double chi2qtl  (double prob, double df);
+extern double chi2qtlP (double prob, double df);
+extern double chi2qtlQ (double prob, double df);
+
+/*----------------------------------------------------------------------
+  Preprocessor Definitions
+----------------------------------------------------------------------*/
+#define chi2cdf(x,d)   chi2cdfP(x,d)
+#define chi2qtl(p,d)   chi2qtlP(p,d)
+
+#endif

data/ext/math/src/choose.c

@@ -0,0 +1,71 @@
+/*----------------------------------------------------------------------
+  File    : choose.c
+  Contents: compute n choose k
+  Author  : Christian Borgelt
+  History : 1996.04.29 file created
+            2003.05.17 (n-i+1) replaced by n--
+            2008.03.14 main function added
+----------------------------------------------------------------------*/
+#ifdef CHOOSE_MAIN
+#include <stdio.h>
+#include <stdlib.h>
+#endif
+#include <limits.h>
+#include "choose.h"
+
+/*----------------------------------------------------------------------
+  Functions
+----------------------------------------------------------------------*/
+
+unsigned int choose (unsigned int n, unsigned int k)
+{                               /* --- compute n choose k */
+  unsigned int i, t;            /* loop variable, buffer */
+  unsigned int r = 1;           /* result */
+
+  if (k > n) return 0;          /* check range of k */
+  for (i = 1; i <= k; i++) {    /* calculation loop */
+    t = n--;                    /* calculate next factor in numerator */
+    if (UINT_MAX /t < r)        /* if result of multiplication */
+      return 0;                 /* is out of range, abort */
+    r = (r *t) /i;              /* calculate \prod_{i=1}^k (n-i+1)/i */
+  }
+  return r;                     /* return result */
+}  /* choose() */
+
+/*--------------------------------------------------------------------*/
+
+double dchoose (unsigned int n, unsigned int k)
+{                               /* --- compute n choose k */
+  unsigned int i;               /* loop variable, buffer */
+  double r = 1.0;               /* result */
+
+  if (k > n) return 0;          /* check range of k */
+  for (i = 1; i <= k; i++)      /* calculation loop */
+    r = (r *n--) /i;            /* calculate \prod_{i=1}^k (n-i+1)/i */
+  return r;                     /* return result */
+}  /* dchoose() */
+
+/*--------------------------------------------------------------------*/
+#ifdef CHOOSE_MAIN
+
+int main (int argc, char *argv[])
+{                               /* --- main function */
+  int          n, k;            /* arguments */
+  unsigned int r;               /* result */
+
+  if (argc != 3) {              /* if number of arguments is wrong */
+    printf("usage: %s <n> <k>\n", argv[0]);
+    printf("calculate n choose k, i.e. n!/(k!(n-k)!)\n");
+    return 0;                   /* print a usage message */
+  }                             /* and abort the program */
+  n = atoi(argv[1]);            /* get argument 1, i.e. n */
+  k = atoi(argv[2]);            /* get argument 2, i.e. k */
+  if ((n < 0) || (k < 0)) {     /* check range of n and k */
+    printf("%s: value out of range\n", argv[0]); return 1; }
+  r = choose(n, k);             /* calculate n choose k */
+  if (r > 0) printf("%d choose %d: %d\n",  n, k, r);
+  else       printf("%d choose %d: %.0f\n", n, k, dchoose(n, k));
+  return 0;                     /* print result and terminate program */
+}  /* main() */
+
+#endif

data/ext/math/src/choose.h

@@ -0,0 +1,16 @@
+/*----------------------------------------------------------------------
+  File    : choose.h
+  Contents: compute n choose k
+  Author  : Christian Borgelt
+  History : 29.04.1996 file created
+----------------------------------------------------------------------*/
+#ifndef __CHOOSE__
+#define __CHOOSE__
+
+/*----------------------------------------------------------------------
+  Functions
+----------------------------------------------------------------------*/
+extern unsigned int choose  (unsigned int n, unsigned int k);
+extern double       dchoose (unsigned int n, unsigned int k);
+
+#endif