curvedraw

annotate src/curve.cc @ 0:8e524989c904

find changesets by author, revision, files, or words in the commit message
getting there
author John Tsiombikas <nuclear@member.fsf.org>
date Tue, 15 Dec 2015 07:15:53 +0200
parents
children ce7aa9a0594c
rev   line source
nuclear@0 1 #include <float.h>
nuclear@0 2 #include "curve.h"
nuclear@0 3
nuclear@0 4 Curve::Curve(CurveType type)
nuclear@0 5 {
nuclear@0 6 this->type = type;
nuclear@0 7 }
nuclear@0 8
nuclear@0 9 void Curve::set_type(CurveType type)
nuclear@0 10 {
nuclear@0 11 this->type = type;
nuclear@0 12 }
nuclear@0 13
nuclear@0 14 CurveType Curve::get_type() const
nuclear@0 15 {
nuclear@0 16 return type;
nuclear@0 17 }
nuclear@0 18
nuclear@0 19 void Curve::add_point(const Vector2 &p, float weight)
nuclear@0 20 {
nuclear@0 21 cp.push_back(Vector3(p.x, p.y, weight));
nuclear@0 22 }
nuclear@0 23
nuclear@0 24 bool Curve::remove_point(int idx)
nuclear@0 25 {
nuclear@0 26 if(idx < 0 || idx >= (int)cp.size()) {
nuclear@0 27 return false;
nuclear@0 28 }
nuclear@0 29 cp.erase(cp.begin() + idx);
nuclear@0 30 return true;
nuclear@0 31 }
nuclear@0 32
nuclear@0 33 int Curve::nearest_point(const Vector2 &p)
nuclear@0 34 {
nuclear@0 35 int res = -1;
nuclear@0 36 float bestsq = FLT_MAX;
nuclear@0 37
nuclear@0 38 for(size_t i=0; i<cp.size(); i++) {
nuclear@0 39 float d = (get_point(i) - p).length_sq();
nuclear@0 40 if(d < bestsq) {
nuclear@0 41 bestsq = d;
nuclear@0 42 res = i;
nuclear@0 43 }
nuclear@0 44 }
nuclear@0 45
nuclear@0 46 return res;
nuclear@0 47 }
nuclear@0 48
nuclear@0 49 int Curve::get_point_count() const
nuclear@0 50 {
nuclear@0 51 return (int)cp.size();
nuclear@0 52 }
nuclear@0 53
nuclear@0 54 const Vector3 &Curve::get_homo_point(int idx) const
nuclear@0 55 {
nuclear@0 56 return cp[idx];
nuclear@0 57 }
nuclear@0 58
nuclear@0 59 Vector2 Curve::get_point(int idx) const
nuclear@0 60 {
nuclear@0 61 return Vector2(cp[idx].x, cp[idx].y);
nuclear@0 62 }
nuclear@0 63
nuclear@0 64 float Curve::get_weight(int idx) const
nuclear@0 65 {
nuclear@0 66 return cp[idx].z;
nuclear@0 67 }
nuclear@0 68
nuclear@0 69 bool Curve::set_point(int idx, const Vector2 &p, float weight)
nuclear@0 70 {
nuclear@0 71 if(idx < 0 || idx >= (int)cp.size()) {
nuclear@0 72 return false;
nuclear@0 73 }
nuclear@0 74 cp[idx] = Vector3(p.x, p.y, weight);
nuclear@0 75 return true;
nuclear@0 76 }
nuclear@0 77
nuclear@0 78 bool Curve::set_weight(int idx, float weight)
nuclear@0 79 {
nuclear@0 80 if(idx < 0 || idx >= (int)cp.size()) {
nuclear@0 81 return false;
nuclear@0 82 }
nuclear@0 83 cp[idx].z = weight;
nuclear@0 84 return true;
nuclear@0 85 }
nuclear@0 86
nuclear@0 87 Vector2 Curve::interpolate(float t, CurveType type) const
nuclear@0 88 {
nuclear@0 89 int num_cp = (int)cp.size();
nuclear@0 90 if(!num_cp) {
nuclear@0 91 return Vector2(0, 0);
nuclear@0 92 }
nuclear@0 93 if(num_cp == 1) {
nuclear@0 94 return Vector2(cp[0].x, cp[0].y);
nuclear@0 95 }
nuclear@0 96
nuclear@0 97 Vector3 res;
nuclear@0 98 int idx0 = (int)floor(t * (num_cp - 1));
nuclear@0 99 int idx1 = idx0 + 1;
nuclear@0 100
nuclear@0 101 float dt = 1.0 / (float)(num_cp - 1);
nuclear@0 102 float t0 = (float)idx0 * dt;
nuclear@0 103 float t1 = (float)idx1 * dt;
nuclear@0 104
nuclear@0 105 t = (t - t0) / (t1 - t0);
nuclear@0 106 if(t < 0.0) t = 0.0;
nuclear@0 107 if(t > 1.0) t = 1.0;
nuclear@0 108
nuclear@0 109 if(type == CURVE_LINEAR || num_cp <= 2) {
nuclear@0 110 res = lerp(cp[idx0], cp[idx1], t);
nuclear@0 111 } else {
nuclear@0 112 int idx_prev = idx0 <= 0 ? idx0 : idx0 - 1;
nuclear@0 113 int idx_next = idx1 >= num_cp - 1 ? idx1 : idx1 + 1;
nuclear@0 114
nuclear@0 115 if(type == CURVE_HERMITE) {
nuclear@0 116 res = catmull_rom_spline(cp[idx_prev], cp[idx0], cp[idx1], cp[idx_next], t);
nuclear@0 117 } else {
nuclear@0 118 res = bspline(cp[idx_prev], cp[idx0], cp[idx1], cp[idx_next], t);
nuclear@0 119 if(res.z != 0.0f) {
nuclear@0 120 res.x /= res.z;
nuclear@0 121 res.y /= res.z;
nuclear@0 122 }
nuclear@0 123 }
nuclear@0 124 }
nuclear@0 125
nuclear@0 126 return Vector2(res.x, res.y);
nuclear@0 127 }
nuclear@0 128
nuclear@0 129 Vector2 Curve::interpolate(float t) const
nuclear@0 130 {
nuclear@0 131 return interpolate(t, type);
nuclear@0 132 }
nuclear@0 133
nuclear@0 134 Vector2 Curve::operator ()(float t) const
nuclear@0 135 {
nuclear@0 136 return interpolate(t);
nuclear@0 137 }
nuclear@0 138
nuclear@0 139 float Curve::map_param(float x) const
nuclear@0 140 {
nuclear@0 141 if(x < 0.0f) x = 0.0f;
nuclear@0 142 if(x >= 1.0f) x = 1.0f;
nuclear@0 143 return x * ((float)cp.size() - 1);
nuclear@0 144 }