diff foosdk/sdk/foobar2000/helpers/duration_counter.h @ 1:20d02a178406 default tip

*: check in everything else yay
author Paper <paper@tflc.us>
date Mon, 05 Jan 2026 02:15:46 -0500
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/foosdk/sdk/foobar2000/helpers/duration_counter.h	Mon Jan 05 02:15:46 2026 -0500
@@ -0,0 +1,96 @@
+#pragma once
+
+#include <SDK/dsp.h>
+#include <pfc/map.h>
+
+//! Duration counter class - accumulates duration using sample values, without any kind of rounding error accumulation.
+class duration_counter {
+public:
+	duration_counter() {}
+	void set(double v) {
+		m_sampleCounts.remove_all();
+		m_offset = v;
+	}
+	void reset() {
+		set(0);
+	}
+
+	void add(double v) { m_offset += v; }
+	void subtract(double v) { m_offset -= v; }
+
+	double query() const {
+		double acc = m_offset;
+		for (auto& walk : m_sampleCounts) {
+			acc += audio_math::samples_to_time(walk.m_value, walk.m_key);
+		}
+		return acc;
+	}
+
+	uint64_t queryAsAnySampleCount() const {
+		if (m_sampleCounts.get_count() == 1) {
+			return m_sampleCounts.first()->m_value;
+		}
+		return 0;
+	}
+
+	uint64_t queryAsSampleCount(uint32_t rate) const {
+		uint64_t samples = 0;
+		double acc = m_offset;
+		for (auto& walk : m_sampleCounts) {
+			if (walk.m_key == rate) samples += walk.m_value;
+			else acc += audio_math::samples_to_time(walk.m_value, walk.m_key);
+		}
+		return samples + audio_math::time_to_samples(acc, rate);
+	}
+
+	void add(const audio_chunk & c) {
+		add(c.get_sample_count(), c.get_sample_rate());
+	}
+#ifdef FOOBAR2000_HAVE_DSP
+	void add(dsp_chunk_list const & c) {
+		const size_t num = c.get_count();
+		for (size_t walk = 0; walk < num; ++walk) {
+			add(*c.get_item(walk));
+		}
+	}
+#endif
+	void add(t_uint64 sampleCount, t_uint32 sampleRate) {
+		PFC_ASSERT(sampleRate > 0);
+		if (sampleRate > 0 && sampleCount > 0) {
+			m_sampleCounts.find_or_add(sampleRate) += sampleCount;
+		}
+	}
+	void add(const duration_counter & other) {
+		add(other.m_offset);
+		for (auto& walk : other.m_sampleCounts) add(walk.m_value, walk.m_key);
+	}
+	void subtract(const duration_counter & other) {
+		subtract(other.m_offset);
+		for (auto& walk : other.m_sampleCounts) subtract(walk.m_value, walk.m_key);
+	}
+	void subtract(t_uint64 sampleCount, t_uint32 sampleRate) {
+		PFC_ASSERT(sampleRate > 0);
+		if (sampleRate > 0 && sampleCount > 0) {
+			t_uint64 * val = m_sampleCounts.query_ptr(sampleRate);
+			if (val == NULL) throw pfc::exception_invalid_params();
+			if (*val < sampleCount) throw pfc::exception_invalid_params();
+			else if (*val == sampleCount) {
+				m_sampleCounts.remove(sampleRate);
+			} else {
+				*val -= sampleCount;
+			}
+
+		}
+	}
+	void subtract(const audio_chunk & c) {
+		subtract(c.get_sample_count(), c.get_sample_rate());
+	}
+	template<typename t_source> duration_counter & operator+=(const t_source & source) { add(source); return *this; }
+	template<typename t_source> duration_counter & operator-=(const t_source & source) { subtract(source); return *this; }
+	template<typename t_source> duration_counter & operator=(const t_source & source) { reset(); add(source); return *this; }
+private:
+	double m_offset = 0;
+	typedef pfc::map_t<t_uint32, t_uint64> t_map;
+	t_map m_sampleCounts;
+};
+