stddevw_pop.c

Go to the documentation of this file.

/* stddevw_pop.c (weighted standard deviation of population) */

/***********************************************************************
*  This code is part of Statistics for MySQL.
*
*  Copyright (C) 2015 Heinrich Schuchardt (xypron.glpk@gmx.de)
*
*  Licensed under the Apache License, Version 2.0 (the "License");
*  you may not use this file except in compliance with the License.
*  You may obtain a copy of the License at
*
*      http://www.apache.org/licenses/LICENSE-2.0
*
*  Unless required by applicable law or agreed to in writing, software
*  distributed under the License is distributed on an "AS IS" BASIS,
*  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*  See the License for the specific language governing permissions and
*  limitations under the License.
***********************************************************************/

#include "sqlstat.h"

struct stddevw_pop_storage {
  int     argc;  
  double  count;  
  double  sumX;   
  double  sumXX;  
};

my_bool stddevw_pop_init(UDF_INIT *initid, UDF_ARGS *args, char *message) {
  struct stddevw_pop_storage * data;

  if (args->arg_count < 1 || args->arg_count > 2) {
    strcpy(message,"stddevw_pop() requires one or two arguments");
    return 1;
  }
  args->arg_type[0] = REAL_RESULT;
  if (args->arg_count > 1) {
    args->arg_type[1] = REAL_RESULT;
  }

  data = (struct stddevw_pop_storage *) malloc( sizeof(struct stddevw_pop_storage));
  if (data == NULL) {
    strcpy(message,"Couldn't allocate memory");
    return 1;
  }
  data->argc = args->arg_count;

  initid->maybe_null = 1;
  initid->decimals   = NOT_FIXED_DEC;
  initid->max_length = 13 + initid->decimals;
  initid->ptr        = (char *) data;
  initid->const_item = 0;

  return 0;
}

void stddevw_pop_reset(UDF_INIT *initid, UDF_ARGS *args, char *is_null, char *error) {
  stddevw_pop_clear(initid, is_null, error);
  stddevw_pop_add(initid, args, is_null, error);
}

void stddevw_pop_clear(UDF_INIT *initid, char *is_null, char *error) {
  struct stddevw_pop_storage *data;
  data = (struct stddevw_pop_storage *) initid->ptr;
  data->count = 0;
  data->sumX  = 0;
  data->sumXX = 0;
}

void stddevw_pop_add(UDF_INIT *initid, UDF_ARGS *args, char *is_null, char *error) {
  struct stddevw_pop_storage * data;
  double x;
  double w;

  if (!args->args[0]) {
    return;
  }
  data = (struct stddevw_pop_storage *) initid->ptr;
  if (data->argc > 1) {
    if (!args->args[1]) {
      return;
    }
    w = *((double*) args->args[1]);
  } else {
    w = 1.;
  }
  x = *((double*) args->args[0]);

  data->count  += w;
  data->sumX   += w * x;
  data->sumXX  += w * x * x;
}

double stddevw_pop(UDF_INIT *initid, UDF_ARGS *args, char *is_null, char *error) {
  struct stddevw_pop_storage * data;
  double m2;
  double mean;

  double ret;

  data = (struct stddevw_pop_storage *) initid->ptr;

  if (data->count <= 0) {
    *is_null = 1;
    return 0;
  }

  mean = data->sumX / data->count;

  m2 = (data->sumXX - mean * data->sumX) / data->count;

  ret = sqrt(m2);

  return ret;
}

void stddevw_pop_deinit(UDF_INIT *initid) {
  if (initid->ptr) {
    free(initid->ptr);
  }
}