diff options
| author | a1xd <[email protected]> | 2021-09-05 21:42:04 -0400 |
|---|---|---|
| committer | a1xd <[email protected]> | 2021-09-23 22:28:44 -0400 |
| commit | 83545569d958ecd157e20cb683b1d47a81996266 (patch) | |
| tree | f25cefa50d5bdc63283401ab4c5d9e8db073066e | |
| parent | fix kernel bugcheck (diff) | |
| download | rawaccel-83545569d958ecd157e20cb683b1d47a81996266.tar.xz rawaccel-83545569d958ecd157e20cb683b1d47a81996266.zip | |
change lookup impl to use binary search
less data, less code, and latency is good at 8, 128, and 1024 points
| -rw-r--r-- | common/accel-lookup.hpp | 184 |
1 files changed, 42 insertions, 142 deletions
diff --git a/common/accel-lookup.hpp b/common/accel-lookup.hpp index b8c689a..493a7a4 100644 --- a/common/accel-lookup.hpp +++ b/common/accel-lookup.hpp @@ -32,160 +32,60 @@ namespace rawaccel { } }; - struct si_pair { - float slope = 0; - float intercept = 0; - }; - - struct arbitrary_lut_point { - float applicable_speed = 0; - si_pair slope_intercept = {}; - }; - struct lookup { enum { capacity = LUT_POINTS_CAPACITY }; - fp_rep_range range; - bool velocity_points; - arbitrary_lut_point data[capacity] = {}; - int log_lookup[capacity] = {}; - double first_point_speed; - double last_point_speed; - int last_arbitrary_index; - int last_log_lookup_index; - double last_log_lookup_speed; - double first_log_lookup_speed; - - double operator()(double speed, const accel_args&) const - { - int index = 0; - int last_arb_index = last_arbitrary_index; - int last_log_index = last_log_lookup_index; - - if (speed <= 0) return 1; - - if (unsigned(last_arb_index) < capacity && - unsigned(last_log_index) < capacity && - speed > first_point_speed) - { - if (speed > last_point_speed) - { - index = last_arb_index; - } - else if (speed > last_log_lookup_speed) - { - int last_log = log_lookup[last_log_index]; - if (unsigned(last_log) >= capacity) return 1; - index = search_from(last_log, last_arb_index, speed); - } - else if (speed < first_log_lookup_speed) - { - index = search_from(0, last_arb_index, speed); - } - else - { - int log_index = get_log_index(speed); - if (unsigned(log_index) >= capacity) return 1; - int arbitrary_index = log_lookup[log_index]; - if (arbitrary_index < 0) return 1; - index = search_from(arbitrary_index, last_arb_index, speed); - } - - } - - return apply(index, speed); - } - - int inline get_log_index(double speed) const - { - double speed_log = log(speed) - range.start; - int index = (int)floor(speed_log * range.num); - return index; - } + int size; + bool velocity; - int inline search_from(int index, int last, double speed) const - { - do - { - index++; - } - while (index <= last && data[index].applicable_speed < speed); + lookup(const accel_args& args) : + size(args.length / 2), + velocity(args.gain) {} - return index - 1; - } - - double inline apply(int index, double speed) const + double operator()(double x, const accel_args& args) const { - auto [slope, intercept] = data[index].slope_intercept; - - if (velocity_points) - { - return slope + intercept / speed; - } - else - { - return slope * speed + intercept; - } - } + auto* points = reinterpret_cast<const vec2<float>*>(args.data); + + int lo = 0; + int hi = size - 2; + + if (x <= 0) return 0; + + if (hi < capacity - 1) { + + while (lo <= hi) { + int mid = (lo + hi) / 2; + auto p = points[mid]; + + if (x < p.x) { + hi = mid - 1; + } + else if (x > p.x) { + lo = mid + 1; + } + else { + double y = p.y; + if (velocity) y /= x; + return y; + } + } - void fill(const float* raw_data, int raw_length) - { - auto* points = reinterpret_cast<const vec2<float>*>(raw_data); - int length = raw_length / 2; - - first_point_speed = points[0].x; - last_arbitrary_index = length - 1; - // -2 because the last index in the arbitrary array is used for slope-intercept only - last_point_speed = points[length-2].x; - - int start = static_cast<int>(floor(log(first_point_speed))); - first_log_lookup_speed = exp(start*1.0); - int end = static_cast<int>(floor(log(last_point_speed))); - last_log_lookup_speed = exp(end*1.0); - int num = end > start ? static_cast<int>(capacity / (end - start)) : 1; - range = fp_rep_range{ start, end, num }; - last_log_lookup_index = end > start ? num * (end - start) - 1 : 0; - - vec2<float> current = {0, velocity_points ? 0.0f : 1.0f }; - vec2<float> next; - int log_index = 0; - double log_inner_iterator = range.start; - double log_inner_slice = 1.0 / (range.num * 1.0); - double log_value = exp(log_inner_iterator); - - for (int i = 0; i < length; i++) - { - next = points[i]; - double slope = (next.y - current.y) / (next.x - current.x); - double intercept = next.y - slope * next.x; - si_pair current_si = { - static_cast<float>(slope), - static_cast<float>(intercept) - }; - arbitrary_lut_point current_lut_point = { - static_cast<float>(current.x), - current_si - }; - - this->data[i] = current_lut_point; - - while (log_value < next.x && log_inner_iterator < end) - { - this->log_lookup[log_index] = i; - log_index++; - log_inner_iterator += log_inner_slice; - log_value = exp(log_inner_iterator); + if (lo > 0) { + auto& a = points[lo - 1]; + auto& b = points[lo]; + double t = (x - a.x) / (b.x - a.x); + double y = lerp(a.y, b.y, t); + if (velocity) y /= x; + return y; } - current = next; } - } - lookup(const accel_args& args) - { - velocity_points = args.gain; - fill(args.data, args.length); + double y = points[0].y; + if (velocity) y /= points[0].x; + return y; } }; + } |