Quelle  iio-test-rescale.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Kunit tests for IIO rescale conversions
 *
 * Copyright (c) 2021 Liam Beguin <liambeguin@gmail.com>
 */

#include <linux/gcd.h>
#include <linux/overflow.h>

#include <linux/iio/afe/rescale.h>
#include <linux/iio/iio.h>

#include <kunit/test.h>

struct rescale_tc_data {
 const char *name;

 const s32 numerator;
 const s32 denominator;
 const s32 offset;

 const int schan_val;
 const int schan_val2;
 const int schan_off;
 const int schan_scale_type;

 const char *expected;
 const char *expected_off;
};

static const struct rescale_tc_data scale_cases[] = {
 /*
 * Typical use cases
 */
 {
  .name = "typical IIO_VAL_INT, positive",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT,
  .schan_val = 42,
  .expected = "5210.918114143",
 },
 {
  .name = "typical IIO_VAL_INT, negative",
  .numerator = -1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT,
  .schan_val = 42,
  .expected = "-5210.918114143",
 },
 {
  .name = "typical IIO_VAL_FRACTIONAL, positive",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 42,
  .schan_val2 = 20,
  .expected = "260.545905707",
 },
 {
  .name = "typical IIO_VAL_FRACTIONAL, negative",
  .numerator = -1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 42,
  .schan_val2 = 20,
  .expected = "-260.545905707",
 },
 {
  .name = "typical IIO_VAL_FRACTIONAL_LOG2, positive",
  .numerator = 42,
  .denominator = 53,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 4096,
  .schan_val2 = 16,
  .expected = "0.049528301",
 },
 {
  .name = "typical IIO_VAL_FRACTIONAL_LOG2, negative",
  .numerator = -42,
  .denominator = 53,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 4096,
  .schan_val2 = 16,
  .expected = "-0.049528301",
 },
 {
  .name = "typical IIO_VAL_INT_PLUS_NANO, positive",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 10,
  .schan_val2 = 123456,
  .expected = "1240.710106203",
 },
 {
  .name = "typical IIO_VAL_INT_PLUS_NANO, negative",
  .numerator = -1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 10,
  .schan_val2 = 123456,
  .expected = "-1240.710106203",
 },
 {
  .name = "typical IIO_VAL_INT_PLUS_MICRO, positive",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 10,
  .schan_val2 = 1234,
  .expected = "1240.84789",
 },
 {
  .name = "typical IIO_VAL_INT_PLUS_MICRO, negative",
  .numerator = -1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 10,
  .schan_val2 = 1234,
  .expected = "-1240.84789",
 },
 /*
 * Use cases with small scales involving divisions
 */
 {
  .name = "small IIO_VAL_FRACTIONAL, 261/509 scaled by 90/1373754273",
  .numerator = 261,
  .denominator = 509,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 90,
  .schan_val2 = 1373754273,
  .expected = "0.000000033594",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL, 90/1373754273 scaled by 261/509",
  .numerator = 90,
  .denominator = 1373754273,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 261,
  .schan_val2 = 509,
  .expected = "0.000000033594",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL, 760/1373754273 scaled by 427/2727",
  .numerator = 760,
  .denominator = 1373754273,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 427,
  .schan_val2 = 2727,
  .expected = "0.000000086626",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL, 761/1373754273 scaled by 427/2727",
  .numerator = 761,
  .denominator = 1373754273,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 427,
  .schan_val2 = 2727,
  .expected = "0.000000086740",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL, 5/32768 scaled by 3/10000",
  .numerator = 5,
  .denominator = 32768,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 3,
  .schan_val2 = 10000,
  .expected = "0.0000000457763671875",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL, 0 < scale < 1",
  .numerator = 6,
  .denominator = 6,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 1,
  .schan_val2 = 3,
  .expected = "0.3333333333333333",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL, -1 < scale < 0",
  .numerator = -6,
  .denominator = 6,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 1,
  .schan_val2 = 3,
  .expected = "-0.3333333333333333",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL, 0 < scale < 2",
  .numerator = 8,
  .denominator = 2,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 1,
  .schan_val2 = 3,
  .expected = "1.3333333333333333",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL, -2 < scale < 0",
  .numerator = -8,
  .denominator = 2,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 1,
  .schan_val2 = 3,
  .expected = "-1.3333333333333333",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL_LOG2, 760/32768 scaled by 15/22",
  .numerator = 760,
  .denominator = 32768,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 15,
  .schan_val2 = 22,
  .expected = "0.000000082946",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL_LOG2, 761/32768 scaled by 15/22",
  .numerator = 761,
  .denominator = 32768,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 15,
  .schan_val2 = 22,
  .expected = "0.000000083055",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL_LOG2, 0 < scale < 1",
  .numerator = 16,
  .denominator = 3,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 1,
  .schan_val2 = 4,
  .expected = "0.3333333333333333",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL_LOG2, -1 < scale < 0",
  .numerator = -16,
  .denominator = 3,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 1,
  .schan_val2 = 4,
  .expected = "-0.3333333333333333",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL_LOG2, 0 < scale < 2",
  .numerator = 8,
  .denominator = 3,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 1,
  .schan_val2 = 1,
  .expected = "1.3333333333333333",
 },
 {
  .name = "small IIO_VAL_FRACTIONAL_LOG2, -2 < scale < 0",
  .numerator = -8,
  .denominator = 3,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 1,
  .schan_val2 = 1,
  .expected = "-1.3333333333333333",
 },
 {
  .name = "small IIO_VAL_INT_PLUS_MICRO, positive",
  .numerator = 1,
  .denominator = 2,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 5,
  .schan_val2 = 1234,
  .expected = "2.500617",
 },
 {
  .name = "small IIO_VAL_INT_PLUS_MICRO, negative",
  .numerator = -1,
  .denominator = 2,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 5,
  .schan_val2 = 1234,
  .expected = "-2.500617",
 },
 /*
 * INT_PLUS_{MICRO,NANO} positive/negative corner cases
 */
 {
  .name = "negative IIO_VAL_INT_PLUS_NANO, negative schan",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = -10,
  .schan_val2 = 123456,
  .expected = "-1240.710106203",
 },
 {
  .name = "negative IIO_VAL_INT_PLUS_NANO, both negative",
  .numerator = -1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = -10,
  .schan_val2 = 123456,
  .expected = "1240.710106203",
 },
 {
  .name = "negative IIO_VAL_INT_PLUS_NANO, 3 negative",
  .numerator = -1000000,
  .denominator = -8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = -10,
  .schan_val2 = 123456,
  .expected = "-1240.710106203",
 },
 {
  .name = "negative IIO_VAL_INT_PLUS_NANO, 4 negative",
  .numerator = -1000000,
  .denominator = -8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = -10,
  .schan_val2 = -123456,
  .expected = "-1240.710106203",
 },
 {
  .name = "negative IIO_VAL_INT_PLUS_NANO, negative, *val = 0",
  .numerator = 1,
  .denominator = -10,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 0,
  .schan_val2 = 123456789,
  .expected = "-0.012345678",
 },
 /*
 * INT_PLUS_{MICRO,NANO} decimal part overflow
 */
 {
  .name = "decimal overflow IIO_VAL_INT_PLUS_NANO, positive",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 10,
  .schan_val2 = 123456789,
  .expected = "1256.01200856",
 },
 {
  .name = "decimal overflow IIO_VAL_INT_PLUS_NANO, negative",
  .numerator = -1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 10,
  .schan_val2 = 123456789,
  .expected = "-1256.01200856",
 },
 {
  .name = "decimal overflow IIO_VAL_INT_PLUS_NANO, negative schan",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = -10,
  .schan_val2 = 123456789,
  .expected = "-1256.01200856",
 },
 {
  .name = "decimal overflow IIO_VAL_INT_PLUS_MICRO, positive",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 10,
  .schan_val2 = 123456789,
  .expected = "16557.914267",
 },
 {
  .name = "decimal overflow IIO_VAL_INT_PLUS_MICRO, negative",
  .numerator = -1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 10,
  .schan_val2 = 123456789,
  .expected = "-16557.914267",
 },
 {
  .name = "decimal overflow IIO_VAL_INT_PLUS_MICRO, negative schan",
  .numerator = 1000000,
  .denominator = 8060,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = -10,
  .schan_val2 = 123456789,
  .expected = "-16557.914267",
 },
 /*
 * 32-bit overflow conditions
 */
 {
  .name = "overflow IIO_VAL_FRACTIONAL, positive",
  .numerator = 2,
  .denominator = 20,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = S32_MAX,
  .schan_val2 = 1,
  .expected = "214748364.7",
 },
 {
  .name = "overflow IIO_VAL_FRACTIONAL, negative",
  .numerator = -2,
  .denominator = 20,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = S32_MAX,
  .schan_val2 = 1,
  .expected = "-214748364.7",
 },
 {
  .name = "overflow IIO_VAL_FRACTIONAL_LOG2, positive",
  .numerator = S32_MAX,
  .denominator = 4096,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 4096,
  .schan_val2 = 16,
  .expected = "32767.99998474121",
 },
 {
  .name = "overflow IIO_VAL_FRACTIONAL_LOG2, negative",
  .numerator = S32_MAX,
  .denominator = 4096,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = -4096,
  .schan_val2 = 16,
  .expected = "-32767.99998474121",
 },
 {
  .name = "overflow IIO_VAL_INT_PLUS_NANO, positive",
  .numerator = 2,
  .denominator = 20,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 10,
  .schan_val2 = S32_MAX,
  .expected = "1.214748364",
 },
 {
  .name = "overflow IIO_VAL_INT_PLUS_NANO, negative",
  .numerator = -2,
  .denominator = 20,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 10,
  .schan_val2 = S32_MAX,
  .expected = "-1.214748364",
 },
 {
  .name = "overflow IIO_VAL_INT_PLUS_NANO, negative schan",
  .numerator = 2,
  .denominator = 20,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = -10,
  .schan_val2 = S32_MAX,
  .expected = "-1.214748364",
 },
 {
  .name = "overflow IIO_VAL_INT_PLUS_MICRO, positive",
  .numerator = 2,
  .denominator = 20,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 10,
  .schan_val2 = S32_MAX,
  .expected = "215.748364",
 },
 {
  .name = "overflow IIO_VAL_INT_PLUS_MICRO, negative",
  .numerator = -2,
  .denominator = 20,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 10,
  .schan_val2 = S32_MAX,
  .expected = "-215.748364",
 },
 {
  .name = "overflow IIO_VAL_INT_PLUS_MICRO, negative schan",
  .numerator = 2,
  .denominator = 20,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = -10,
  .schan_val2 = S32_MAX,
  .expected = "-215.748364",
 },
};

static const struct rescale_tc_data offset_cases[] = {
 /*
 * Typical use cases
 */
 {
  .name = "typical IIO_VAL_INT, positive",
  .offset = 1234,
  .schan_scale_type = IIO_VAL_INT,
  .schan_val = 123,
  .schan_val2 = 0,
  .schan_off = 14,
  .expected_off = "24", /* 23.872 */
 },
 {
  .name = "typical IIO_VAL_INT, negative",
  .offset = -1234,
  .schan_scale_type = IIO_VAL_INT,
  .schan_val = 12,
  .schan_val2 = 0,
  .schan_off = 14,
  .expected_off = "-88", /* -88.83333333333333 */
 },
 {
  .name = "typical IIO_VAL_FRACTIONAL, positive",
  .offset = 1234,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 12,
  .schan_val2 = 34,
  .schan_off = 14,
  .expected_off = "3510", /* 3510.333333333333 */
 },
 {
  .name = "typical IIO_VAL_FRACTIONAL, negative",
  .offset = -1234,
  .schan_scale_type = IIO_VAL_FRACTIONAL,
  .schan_val = 12,
  .schan_val2 = 34,
  .schan_off = 14,
  .expected_off = "-3482", /* -3482.333333333333 */
 },
 {
  .name = "typical IIO_VAL_FRACTIONAL_LOG2, positive",
  .offset = 1234,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 12,
  .schan_val2 = 16,
  .schan_off = 14,
  .expected_off = "6739299", /* 6739299.333333333 */
 },
 {
  .name = "typical IIO_VAL_FRACTIONAL_LOG2, negative",
  .offset = -1234,
  .schan_scale_type = IIO_VAL_FRACTIONAL_LOG2,
  .schan_val = 12,
  .schan_val2 = 16,
  .schan_off = 14,
  .expected_off = "-6739271", /* -6739271.333333333 */
 },
 {
  .name = "typical IIO_VAL_INT_PLUS_NANO, positive",
  .offset = 1234,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 10,
  .schan_val2 = 123456789,
  .schan_off = 14,
  .expected_off = "135", /* 135.8951219647469 */
 },
 {
  .name = "typical IIO_VAL_INT_PLUS_NANO, negative",
  .offset = -1234,
  .schan_scale_type = IIO_VAL_INT_PLUS_NANO,
  .schan_val = 10,
  .schan_val2 = 123456789,
  .schan_off = 14,
  .expected_off = "-107", /* -107.89512196474689 */
 },
 {
  .name = "typical IIO_VAL_INT_PLUS_MICRO, positive",
  .offset = 1234,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 10,
  .schan_val2 = 123456789,
  .schan_off = 14,
  .expected_off = "23", /* 23.246438560723952 */
 },
 {
  .name = "typical IIO_VAL_INT_PLUS_MICRO, negative",
  .offset = -12345,
  .schan_scale_type = IIO_VAL_INT_PLUS_MICRO,
  .schan_val = 10,
  .schan_val2 = 123456789,
  .schan_off = 14,
  .expected_off = "-78", /* -78.50185091745313 */
 },
};

static void case_to_desc(const struct rescale_tc_data *t, char *desc)
{
 strcpy(desc, t->name);
}

KUNIT_ARRAY_PARAM(iio_rescale_scale, scale_cases, case_to_desc);
KUNIT_ARRAY_PARAM(iio_rescale_offset, offset_cases, case_to_desc);

/**
 * iio_str_to_nano() - Parse a fixed-point string to get an
 *                      IIO_VAL_INT_PLUS_NANO value
 * @str: The string to parse
 * @nano: The number as an integer
 *
 * Returns 0 on success, or a negative error code if the string cound not be
 * parsed.
 */
static int iio_str_to_nano(const char *str, s64 *nano)
{
 int tmp, tmp2;
 int ret = 0;

 /*
 * iio_str_to_fixpoint() uses 10^8 here instead of 10^9 as fract_mult is
 * the multiplier for the first decimal place.
 */
 ret = iio_str_to_fixpoint(str, 100000000, &tmp, &tmp2);
 if (ret < 0)
  return ret;

 if (tmp < 0)
  tmp2 *= -1;

 *nano = (s64)tmp * 1000000000UL + tmp2;

 return ret;
}

/**
 * iio_test_relative_error_ppm() - Compute relative error (in parts-per-million)
 *                                 between two fixed-point strings
 * @real_str: The real value as a string
 * @exp_str: The expected value as a string
 *
 * Returns a negative error code if the strings cound not be parsed, or the
 * relative error in parts-per-million.
 */
static int iio_test_relative_error_ppm(const char *real_str, const char *exp_str)
{
 s64 real, exp, err;
 int ret;

 ret = iio_str_to_nano(real_str, &real);
 if (ret < 0)
  return ret;

 ret = iio_str_to_nano(exp_str, &exp);
 if (ret < 0)
  return ret;

 if (!exp) {
  pr_err("Expected value is null, relative error is undefined\n");
  return -EINVAL;
 }

 err = 1000000UL * abs(exp - real);

 return (int)div64_u64(err, abs(exp));
}

static void iio_rescale_test_scale(struct kunit *test)
{
 struct rescale_tc_data *t = (struct rescale_tc_data *)test->param_value;
 char *buff = kunit_kmalloc(test, PAGE_SIZE, GFP_KERNEL);
 struct rescale rescale;
 int values[2];
 int rel_ppm;
 int ret;

 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, buff);

 rescale.numerator = t->numerator;
 rescale.denominator = t->denominator;
 rescale.offset = t->offset;
 values[0] = t->schan_val;
 values[1] = t->schan_val2;

 ret = rescale_process_scale(&rescale, t->schan_scale_type,
        &values[0], &values[1]);

 ret = iio_format_value(buff, ret, 2, values);
 KUNIT_EXPECT_EQ(test, (int)strlen(buff), ret);

 rel_ppm = iio_test_relative_error_ppm(buff, t->expected);
 KUNIT_EXPECT_GE_MSG(test, rel_ppm, 0, "failed to compute ppm\n");

 KUNIT_EXPECT_EQ_MSG(test, rel_ppm, 0,
       "\t real=%s"
       "\texpected=%s\n",
       buff, t->expected);
}

static void iio_rescale_test_offset(struct kunit *test)
{
 struct rescale_tc_data *t = (struct rescale_tc_data *)test->param_value;
 char *buff_off = kunit_kmalloc(test, PAGE_SIZE, GFP_KERNEL);
 struct rescale rescale;
 int values[2];
 int ret;

 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, buff_off);

 rescale.numerator = t->numerator;
 rescale.denominator = t->denominator;
 rescale.offset = t->offset;
 values[0] = t->schan_val;
 values[1] = t->schan_val2;

 ret = rescale_process_offset(&rescale, t->schan_scale_type,
         t->schan_val, t->schan_val2, t->schan_off,
         &values[0], &values[1]);

 ret = iio_format_value(buff_off, ret, 2, values);
 KUNIT_EXPECT_EQ(test, (int)strlen(buff_off), ret);

 KUNIT_EXPECT_STREQ(test, strim(buff_off), t->expected_off);
}

static struct kunit_case iio_rescale_test_cases[] = {
 KUNIT_CASE_PARAM(iio_rescale_test_scale, iio_rescale_scale_gen_params),
 KUNIT_CASE_PARAM(iio_rescale_test_offset, iio_rescale_offset_gen_params),
 { }
};

static struct kunit_suite iio_rescale_test_suite = {
 .name = "iio-rescale",
 .test_cases = iio_rescale_test_cases,
};
kunit_test_suite(iio_rescale_test_suite);

MODULE_AUTHOR("Liam Beguin ");
MODULE_DESCRIPTION("Test IIO rescale conversion functions");
MODULE_LICENSE("GPL v2");
MODULE_IMPORT_NS("IIO_RESCALE");