splimes 0.1.0

Spline interpolation over irregularly sampled time series, on SIMD/parallel CPU or GPU (wgpu), with BigDecimal values.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

use anyhow::{Context, Result};
use bigdecimal::{BigDecimal, FromPrimitive, ToPrimitive};
use chrono::{DateTime, Utc};

use crate::{Point, Resolution};

// NOTE: get_max_buffer_size removed - use GpuInterpolator::get_max_buffer_size_static() instead
// The old function was creating a new wgpu Instance per call, causing massive overhead.

pub fn convert_points_to_gpu_format_f64(points: &[Point], _resolution: Resolution) -> Result<(Vec<f64>, Vec<f64>)> {
	if points.is_empty() {
		return Ok((Vec::new(), Vec::new()));
	}

	// Sort points by timestamp to match CPU/parallel implementations
	let mut sorted_points = points.to_vec();
	sorted_points.sort_by_key(|p| p.timestamp);

	let base_time = sorted_points[0].timestamp;
	let mut times = Vec::with_capacity(sorted_points.len());
	let mut values = Vec::with_capacity(sorted_points.len());

	for point in &sorted_points {
		let duration = point.timestamp.signed_duration_since(base_time);
		// Use nanoseconds for internal time calculations to avoid integer division issues
		// (e.g., minute data with Years resolution = 0). Nanoseconds provide sufficient
		// precision for interpolation while staying within f64 range.
		let time_offset = duration.num_nanoseconds().context("Invalid duration")?;
		let time_offset = f64::from_i64(time_offset).context("Invalid time offset")?;
		times.push(time_offset);
		let value = point.value.to_f64().context("Invalid value")?;
		values.push(value); // Don't clamp - preserve full precision
	}

	Ok((times, values))
}

pub fn convert_points_to_gpu_format_f32(points: &[Point], _resolution: Resolution) -> Result<(Vec<f32>, Vec<f32>)> {
	if points.is_empty() {
		return Ok((Vec::new(), Vec::new()));
	}

	// Sort points by timestamp to match CPU/parallel implementations
	let mut sorted_points = points.to_vec();
	sorted_points.sort_by_key(|p| p.timestamp);

	let base_time = sorted_points[0].timestamp;
	let mut times = Vec::with_capacity(sorted_points.len());
	let mut values = Vec::with_capacity(sorted_points.len());

	for point in &sorted_points {
		let duration = point.timestamp.signed_duration_since(base_time);
		// Use nanoseconds for internal time calculations to avoid integer division issues
		// (e.g., minute data with Years resolution = 0). Nanoseconds provide sufficient
		// precision for interpolation while staying within f64 range, then convert to f32.
		let time_offset = duration.num_nanoseconds().context("Invalid duration")?;
		let time_offset = f32::from_i64(time_offset).context("Invalid time offset")?;
		times.push(time_offset);
		let value = point.value.to_f32().context("Invalid value")?;
		values.push(value); // Don't clamp - preserve full precision
	}

	Ok((times, values))
}

pub fn convert_datetimes_to_gpu_format_f64(target_times: &[DateTime<Utc>], _resolution: Resolution, base_time: DateTime<Utc>) -> Result<Vec<f64>> {
	let mut times = Vec::with_capacity(target_times.len());
	for &target_time in target_times {
		let duration = target_time.signed_duration_since(base_time);
		// Use nanoseconds for internal time calculations to avoid integer division issues
		// (e.g., minute data with Years resolution = 0). Nanoseconds provide sufficient
		// precision for interpolation while staying within f64 range.
		let time_offset = duration.num_nanoseconds().context("Invalid duration")?;
		let time_offset = f64::from_i64(time_offset).context("Invalid time offset")?;
		times.push(time_offset);
	}
	Ok(times)
}

pub fn convert_datetimes_to_gpu_format_f32(target_times: &[DateTime<Utc>], _resolution: Resolution, base_time: DateTime<Utc>) -> Result<Vec<f32>> {
	let mut times = Vec::with_capacity(target_times.len());
	for &target_time in target_times {
		let duration = target_time.signed_duration_since(base_time);
		// Use nanoseconds for internal time calculations to avoid integer division issues
		// (e.g., minute data with Years resolution = 0). Nanoseconds provide sufficient
		// precision for interpolation while staying within f64 range, then convert to f32.
		let time_offset = duration.num_nanoseconds().context("Invalid duration")?;
		let time_offset = f32::from_i64(time_offset).context("Invalid time offset")?;
		times.push(time_offset);
	}
	Ok(times)
}

pub fn convert_gpu_results_to_points_f64(results: Vec<f64>, target_times: &[DateTime<Utc>]) -> Vec<Point> {
	results.into_iter()
		.zip(target_times)
		.map(|(value, timestamp)| {
			let value = BigDecimal::from_f64(value).unwrap_or_default();
			Point { timestamp: *timestamp, value }
		})
		.collect()
}

pub fn convert_gpu_results_to_points_f32(results: Vec<f32>, target_times: &[DateTime<Utc>]) -> Vec<Point> {
	results.into_iter()
		.zip(target_times)
		.map(|(value, timestamp)| {
			let value = BigDecimal::from_f32(value).unwrap_or_default();
			Point { timestamp: *timestamp, value }
		})
		.collect()
}