]> sigrok.org Git - libsigrok.git/blobdiff - src/analog.c
output/csv: use intermediate time_t var, silence compiler warning
[libsigrok.git] / src / analog.c
index 7c0dde1be5e9265c713371c78819b09f1189197b..76ea659caf99ec54a33813a023c2d6a6f51279c8 100644 (file)
@@ -22,6 +22,7 @@
 #include <stdint.h>
 #include <string.h>
 #include <ctype.h>
+#include <math.h>
 #include <libsigrok/libsigrok.h>
 #include "libsigrok-internal.h"
 
@@ -62,8 +63,8 @@ static struct unit_mq_string unit_strings[] = {
        { SR_UNIT_BOOLEAN, "" },
        { SR_UNIT_SECOND, "s" },
        { SR_UNIT_SIEMENS, "S" },
-       { SR_UNIT_DECIBEL_MW, "dBu" },
-       { SR_UNIT_DECIBEL_VOLT, "dBv" },
+       { SR_UNIT_DECIBEL_MW, "dBm" },
+       { SR_UNIT_DECIBEL_VOLT, "dBV" },
        { SR_UNIT_UNITLESS, "" },
        { SR_UNIT_DECIBEL_SPL, "dB" },
        { SR_UNIT_CONCENTRATION, "ppm" },
@@ -87,6 +88,9 @@ static struct unit_mq_string unit_strings[] = {
        { SR_UNIT_MOMME, "momme" },
        { SR_UNIT_TOLA, "tola" },
        { SR_UNIT_PIECE, "pcs" },
+       { SR_UNIT_JOULE, "J" },
+       { SR_UNIT_COULOMB, "C" },
+       { SR_UNIT_AMPERE_HOUR, "Ah" },
        ALL_ZERO
 };
 
@@ -118,6 +122,7 @@ static struct unit_mq_string mq_strings[] = {
        ALL_ZERO
 };
 
+/** @private */
 SR_PRIV int sr_analog_init(struct sr_datafeed_analog *analog,
                struct sr_analog_encoding *encoding,
                struct sr_analog_meaning *meaning,
@@ -155,14 +160,14 @@ SR_PRIV int sr_analog_init(struct sr_datafeed_analog *analog,
 /**
  * Convert an analog datafeed payload to an array of floats.
  *
+ * The caller must provide the #outbuf space for the conversion result,
+ * and is expected to free allocated space after use.
+ *
  * @param[in] analog The analog payload to convert. Must not be NULL.
  *                   analog->data, analog->meaning, and analog->encoding
  *                   must not be NULL.
  * @param[out] outbuf Memory where to store the result. Must not be NULL.
  *
- * Sufficient memory for outbuf must have been pre-allocated by the caller,
- * who is also responsible for freeing it when no longer needed.
- *
  * @retval SR_OK Success.
  * @retval SR_ERR Unsupported encoding.
  * @retval SR_ERR_ARG Invalid argument.
@@ -172,136 +177,293 @@ SR_PRIV int sr_analog_init(struct sr_datafeed_analog *analog,
 SR_API int sr_analog_to_float(const struct sr_datafeed_analog *analog,
                float *outbuf)
 {
-       float offset;
-       unsigned int b, i, count;
-       gboolean bigendian;
-
-       if (!analog || !(analog->data) || !(analog->meaning)
-                       || !(analog->encoding) || !outbuf)
+       size_t count;
+       gboolean host_bigendian;
+       gboolean input_float, input_signed, input_bigendian;
+       size_t input_unitsize;
+       double scale, offset, value;
+       const uint8_t *data8;
+       gboolean input_is_native;
+       char type_text[10];
+
+       if (!analog || !analog->data || !analog->meaning || !analog->encoding)
+               return SR_ERR_ARG;
+       if (!outbuf)
                return SR_ERR_ARG;
 
        count = analog->num_samples * g_slist_length(analog->meaning->channels);
 
+       /*
+        * Determine properties of the input data's and the host's
+        * native formats, to simplify test conditions below.
+        * Error messages for unsupported input property combinations
+        * will only be seen by developers and maintainers of input
+        * formats or acquisition device drivers. Terse output is
+        * acceptable there, users shall never see them.
+        */
 #ifdef WORDS_BIGENDIAN
-       bigendian = TRUE;
+       host_bigendian = TRUE;
 #else
-       bigendian = FALSE;
+       host_bigendian = FALSE;
 #endif
-       if (!analog->encoding->is_float) {
-               float offset = analog->encoding->offset.p / (float)analog->encoding->offset.q;
-               float scale = analog->encoding->scale.p / (float)analog->encoding->scale.q;
-               gboolean is_signed = analog->encoding->is_signed;
-               gboolean is_bigendian = analog->encoding->is_bigendian;
-               int8_t *data8 = (int8_t *)(analog->data);
-               int16_t *data16 = (int16_t *)(analog->data);
-               int32_t *data32 = (int32_t *)(analog->data);
-
-               switch (analog->encoding->unitsize) {
-               case 1:
-                       if (is_signed) {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * data8[i];
-                                       outbuf[i] += offset;
-                               }
-                       } else {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * R8(data8 + i);
-                                       outbuf[i] += offset;
-                               }
-                       }
-                       break;
-               case 2:
-                       if (is_signed && is_bigendian) {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * RB16S(&data16[i]);
-                                       outbuf[i] += offset;
-                               }
-                       } else if (is_bigendian) {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * RB16(&data16[i]);
-                                       outbuf[i] += offset;
-                               }
-                       } else if (is_signed) {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * RL16S(&data16[i]);
-                                       outbuf[i] += offset;
-                               }
-                       } else {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * RL16(&data16[i]);
-                                       outbuf[i] += offset;
-                               }
+       input_float = analog->encoding->is_float;
+       input_signed = analog->encoding->is_signed;
+       input_bigendian = analog->encoding->is_bigendian;
+       input_unitsize = analog->encoding->unitsize;
+
+       /*
+        * Prepare the iteration over the sample data: Get the common
+        * scale/offset factors which apply to all individual values.
+        * Position the read pointer on the first byte of input data.
+        */
+       offset = analog->encoding->offset.p;
+       offset /= analog->encoding->offset.q;
+       scale = analog->encoding->scale.p;
+       scale /= analog->encoding->scale.q;
+       data8 = analog->data;
+
+       /*
+        * Immediately handle the special case where input data needs
+        * no conversion because it already is in the application's
+        * native format. Do apply scale/offset though when applicable
+        * on our way out.
+        */
+       input_is_native = input_float &&
+               input_unitsize == sizeof(outbuf[0]) &&
+               input_bigendian == host_bigendian;
+       if (input_is_native) {
+               memcpy(outbuf, data8, count * sizeof(outbuf[0]));
+               if (scale != 1.0 || offset != 0.0) {
+                       while (count--) {
+                               *outbuf *= scale;
+                               *outbuf += offset;
+                               outbuf++;
                        }
-                       break;
-               case 4:
-                       if (is_signed && is_bigendian) {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * RB32S(&data32[i]);
-                                       outbuf[i] += offset;
-                               }
-                       } else if (is_bigendian) {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * RB32(&data32[i]);
-                                       outbuf[i] += offset;
-                               }
-                       } else if (is_signed) {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * RL32S(&data32[i]);
-                                       outbuf[i] += offset;
-                               }
-                       } else {
-                               for (unsigned int i = 0; i < count; i++) {
-                                       outbuf[i] = scale * RL32(&data32[i]);
-                                       outbuf[i] += offset;
-                               }
-                       }
-                       break;
-               default:
-                       sr_err("Unsupported unit size '%d' for analog-to-float conversion.",
-                               analog->encoding->unitsize);
-                       return SR_ERR;
                }
                return SR_OK;
        }
 
-       if (analog->encoding->unitsize == sizeof(float)
-                       && analog->encoding->is_bigendian == bigendian
-                       && analog->encoding->scale.p == 1
-                       && analog->encoding->scale.q == 1
-                       && analog->encoding->offset.p / (float)analog->encoding->offset.q == 0) {
-               /* The data is already in the right format. */
-               memcpy(outbuf, analog->data, count * sizeof(float));
-       } else {
-               for (i = 0; i < count; i += analog->encoding->unitsize) {
-                       for (b = 0; b < analog->encoding->unitsize; b++) {
-                               if (analog->encoding->is_bigendian == bigendian)
-                                       ((uint8_t *)outbuf)[i + b] =
-                                               ((uint8_t *)analog->data)[i * analog->encoding->unitsize + b];
-                               else
-                                       ((uint8_t *)outbuf)[i + (analog->encoding->unitsize - b)] =
-                                               ((uint8_t *)analog->data)[i * analog->encoding->unitsize + b];
-                       }
-                       if (analog->encoding->scale.p != 1
-                                       || analog->encoding->scale.q != 1)
-                               outbuf[i] = (outbuf[i] * analog->encoding->scale.p) / analog->encoding->scale.q;
-                       offset = ((float)analog->encoding->offset.p / (float)analog->encoding->offset.q);
-                       outbuf[i] += offset;
+       /*
+        * Accept sample values in different widths and data types and
+        * endianess formats (floating point or signed or unsigned
+        * integer, in either endianess, for a set of supported widths).
+        * Common scale/offset factors apply to all sample values.
+        *
+        * Do most internal calculations on double precision values.
+        * Only trim the result data to single precision, since that's
+        * the routine's result data type in its public API which needs
+        * to be kept for compatibility. It remains an option for later
+        * to add another public routine which returns double precision
+        * result data, call sites could migrate at their own pace.
+        */
+       if (input_float && input_unitsize == sizeof(float)) {
+               float (*reader)(const uint8_t **p);
+               if (input_bigendian)
+                       reader = read_fltbe_inc;
+               else
+                       reader = read_fltle_inc;
+               while (count--) {
+                       value = reader(&data8);
+                       value *= scale;
+                       value += offset;
+                       *outbuf++ = value;
                }
+               return SR_OK;
+       }
+       if (input_float && input_unitsize == sizeof(double)) {
+               double (*reader)(const uint8_t **p);
+               if (input_bigendian)
+                       reader = read_dblbe_inc;
+               else
+                       reader = read_dblle_inc;
+               while (count--) {
+                       value = reader(&data8);
+                       value *= scale;
+                       value += offset;
+                       *outbuf++ = value;
+               }
+               return SR_OK;
+       }
+       if (input_float) {
+               snprintf(type_text, sizeof(type_text), "%c%zu%s",
+                       'f', input_unitsize * 8, input_bigendian ? "be" : "le");
+               sr_err("Unsupported type for analog-to-float conversion: %s.",
+                       type_text);
+               return SR_ERR;
        }
 
-       return SR_OK;
+       if (input_unitsize == sizeof(uint8_t) && input_signed) {
+               int8_t (*reader)(const uint8_t **p);
+               reader = read_i8_inc;
+               while (count--) {
+                       value = reader(&data8);
+                       value *= scale;
+                       value += offset;
+                       *outbuf++ = value;
+               }
+               return SR_OK;
+       }
+       if (input_unitsize == sizeof(uint8_t)) {
+               uint8_t (*reader)(const uint8_t **p);
+               reader = read_u8_inc;
+               while (count--) {
+                       value = reader(&data8);
+                       value *= scale;
+                       value += offset;
+                       *outbuf++ = value;
+               }
+               return SR_OK;
+       }
+       if (input_unitsize == sizeof(uint16_t) && input_signed) {
+               int16_t (*reader)(const uint8_t **p);
+               if (input_bigendian)
+                       reader = read_i16be_inc;
+               else
+                       reader = read_i16le_inc;
+               while (count--) {
+                       value = reader(&data8);
+                       value *= scale;
+                       value += offset;
+                       *outbuf++ = value;
+               }
+               return SR_OK;
+       }
+       if (input_unitsize == sizeof(uint16_t)) {
+               uint16_t (*reader)(const uint8_t **p);
+               if (input_bigendian)
+                       reader = read_u16be_inc;
+               else
+                       reader = read_u16le_inc;
+               while (count--) {
+                       value = reader(&data8);
+                       value *= scale;
+                       value += offset;
+                       *outbuf++ = value;
+               }
+               return SR_OK;
+       }
+       if (input_unitsize == sizeof(uint32_t) && input_signed) {
+               int32_t (*reader)(const uint8_t **p);
+               if (input_bigendian)
+                       reader = read_i32be_inc;
+               else
+                       reader = read_i32le_inc;
+               while (count--) {
+                       value = reader(&data8);
+                       value *= scale;
+                       value += offset;
+                       *outbuf++ = value;
+               }
+               return SR_OK;
+       }
+       if (input_unitsize == sizeof(uint32_t)) {
+               uint32_t (*reader)(const uint8_t **p);
+               if (input_bigendian)
+                       reader = read_u32be_inc;
+               else
+                       reader = read_u32le_inc;
+               while (count--) {
+                       value = reader(&data8);
+                       value *= scale;
+                       value += offset;
+                       *outbuf++ = value;
+               }
+               return SR_OK;
+       }
+       snprintf(type_text, sizeof(type_text), "%c%zu%s",
+               input_float ? 'f' : input_signed ? 'i' : 'u',
+               input_unitsize * 8, input_bigendian ? "be" : "le");
+       sr_err("Unsupported type for analog-to-float conversion: %s.",
+               type_text);
+       return SR_ERR;
+}
+
+/**
+ * Scale a float value to the appropriate SI prefix.
+ *
+ * @param[in,out] value The float value to convert to appropriate SI prefix.
+ * @param[in,out] digits The number of significant decimal digits in value.
+ *
+ * @return The SI prefix to which value was scaled, as a printable string.
+ *
+ * @since 0.5.0
+ */
+SR_API const char *sr_analog_si_prefix(float *value, int *digits)
+{
+/** @cond PRIVATE */
+#define NEG_PREFIX_COUNT 5 /* number of prefixes below unity */
+#define POS_PREFIX_COUNT (int)(ARRAY_SIZE(prefixes) - NEG_PREFIX_COUNT - 1)
+/** @endcond */
+       static const char *prefixes[] = { "f", "p", "n", "ยต", "m", "", "k", "M", "G", "T" };
+
+       if (!value || !digits || isnan(*value))
+               return prefixes[NEG_PREFIX_COUNT];
+
+       float logval = log10f(fabsf(*value));
+       int prefix = (logval / 3) - (logval < 1);
+
+       if (prefix < -NEG_PREFIX_COUNT)
+               prefix = -NEG_PREFIX_COUNT;
+       if (3 * prefix < -*digits)
+               prefix = (-*digits + 2 * (*digits < 0)) / 3;
+       if (prefix > POS_PREFIX_COUNT)
+               prefix = POS_PREFIX_COUNT;
+
+       *value *= powf(10, -3 * prefix);
+       *digits += 3 * prefix;
+
+       return prefixes[prefix + NEG_PREFIX_COUNT];
+}
+
+/**
+ * Check if a unit "accepts" an SI prefix.
+ *
+ * E.g. SR_UNIT_VOLT is SI prefix friendly while SR_UNIT_DECIBEL_MW or
+ * SR_UNIT_PERCENTAGE are not.
+ *
+ * @param[in] unit The unit to check for SI prefix "friendliness".
+ *
+ * @return TRUE if the unit "accept" an SI prefix.
+ *
+ * @since 0.5.0
+ */
+SR_API gboolean sr_analog_si_prefix_friendly(enum sr_unit unit)
+{
+       static const enum sr_unit prefix_friendly_units[] = {
+               SR_UNIT_VOLT,
+               SR_UNIT_AMPERE,
+               SR_UNIT_OHM,
+               SR_UNIT_FARAD,
+               SR_UNIT_KELVIN,
+               SR_UNIT_HERTZ,
+               SR_UNIT_SECOND,
+               SR_UNIT_SIEMENS,
+               SR_UNIT_VOLT_AMPERE,
+               SR_UNIT_WATT,
+               SR_UNIT_WATT_HOUR,
+               SR_UNIT_METER_SECOND,
+               SR_UNIT_HENRY,
+               SR_UNIT_GRAM
+       };
+       unsigned int i;
+
+       for (i = 0; i < ARRAY_SIZE(prefix_friendly_units); i++)
+               if (unit == prefix_friendly_units[i])
+                       return TRUE;
+
+       return FALSE;
 }
 
 /**
  * Convert the unit/MQ/MQ flags in the analog struct to a string.
  *
+ * The string is allocated by the function and must be freed by the caller
+ * after use by calling g_free().
+ *
  * @param[in] analog Struct containing the unit, MQ and MQ flags.
  *                   Must not be NULL. analog->meaning must not be NULL.
  * @param[out] result Pointer to store result. Must not be NULL.
  *
- * The string is allocated by the function and must be freed by the caller
- * after use by calling g_free().
- *
  * @retval SR_OK Success.
  * @retval SR_ERR_ARG Invalid argument.
  *
@@ -330,8 +492,7 @@ SR_API int sr_analog_unit_to_string(const struct sr_datafeed_analog *analog,
                if (analog->meaning->mqflags & mq_strings[i].value)
                        g_string_append(buf, mq_strings[i].str);
 
-       *result = buf->str;
-       g_string_free(buf, FALSE);
+       *result = g_string_free(buf, FALSE);
 
        return SR_OK;
 }
@@ -400,15 +561,15 @@ static void mult_uint64(struct sr_uint128_t *res, const uint64_t a,
 #endif
 
 /**
- * Compare two sr_rational for equality
+ * Compare two sr_rational for equality.
  *
- * @param[in] a First value
- * @param[in] b Second value
+ * The values are compared for numerical equality, i.e. 2/10 == 1/5.
  *
- * The values are compared for numerical equality, i.e. 2/10 == 1/5
+ * @param[in] a First value.
+ * @param[in] b Second value.
  *
- * @retval 1 if both values are equal
- * @retval 0 otherwise
+ * @retval 1 if both values are equal.
+ * @retval 0 Otherwise.
  *
  * @since 0.5.0
  */
@@ -434,18 +595,20 @@ SR_API int sr_rational_eq(const struct sr_rational *a, const struct sr_rational
 }
 
 /**
- * Multiply two sr_rational
- *
- * @param[in] a First value
- * @param[in] b Second value
- * @param[out] res Result
+ * Multiply two sr_rational.
  *
  * The resulting nominator/denominator are reduced if the result would not fit
  * otherwise. If the resulting nominator/denominator are relatively prime,
  * this may not be possible.
  *
+ * It is safe to use the same variable for result and input values.
+ *
+ * @param[in] a First value.
+ * @param[in] b Second value.
+ * @param[out] res Result.
+ *
  * @retval SR_OK Success.
- * @retval SR_ERR_ARG Resulting value to large
+ * @retval SR_ERR_ARG Resulting value too large.
  *
  * @since 0.5.0
  */
@@ -471,8 +634,8 @@ SR_API int sr_rational_mult(struct sr_rational *res, const struct sr_rational *a
                return SR_ERR_ARG;
        }
 
-       res->p = (int64_t)(p);
-       res->q = (uint64_t)(q);
+       res->p = (int64_t)p;
+       res->q = (uint64_t)q;
 
        return SR_OK;
 
@@ -485,10 +648,12 @@ SR_API int sr_rational_mult(struct sr_rational *res, const struct sr_rational *a
 
        while (!(p.low & 1) && !(q.low & 1)) {
                p.low /= 2;
-               if (p.high & 1) p.low |= (1ll << 63);
+               if (p.high & 1)
+                       p.low |= (1ll << 63);
                p.high >>= 1;
                q.low /= 2;
-               if (q.high & 1) q.low |= (1ll << 63);
+               if (q.high & 1)
+                       q.low |= (1ll << 63);
                q.high >>= 1;
        }
 
@@ -506,4 +671,44 @@ SR_API int sr_rational_mult(struct sr_rational *res, const struct sr_rational *a
 #endif
 }
 
+/**
+ * Divide rational a by rational b.
+ *
+ * The resulting nominator/denominator are reduced if the result would not fit
+ * otherwise. If the resulting nominator/denominator are relatively prime,
+ * this may not be possible.
+ *
+ * It is safe to use the same variable for result and input values.
+ *
+ * @param[in] num Numerator.
+ * @param[in] div Divisor.
+ * @param[out] res Result.
+ *
+ * @retval SR_OK Success.
+ * @retval SR_ERR_ARG Division by zero, denominator of divisor too large,
+ *                    or resulting value too large.
+ *
+ * @since 0.5.0
+ */
+SR_API int sr_rational_div(struct sr_rational *res, const struct sr_rational *num,
+       const struct sr_rational *div)
+{
+       struct sr_rational t;
+
+       if (div->q > INT64_MAX)
+               return SR_ERR_ARG;
+       if (div->p == 0)
+               return SR_ERR_ARG;
+
+       if (div->p > 0) {
+               t.p = div->q;
+               t.q = div->p;
+       } else {
+               t.p = -div->q;
+               t.q = -div->p;
+       }
+
+       return sr_rational_mult(res, num, &t);
+}
+
 /** @} */