math_2d.cpp 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663
  1. /*************************************************************************/
  2. /* math_2d.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* http://www.godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2015 Juan Linietsky, Ariel Manzur. */
  9. /* */
  10. /* Permission is hereby granted, free of charge, to any person obtaining */
  11. /* a copy of this software and associated documentation files (the */
  12. /* "Software"), to deal in the Software without restriction, including */
  13. /* without limitation the rights to use, copy, modify, merge, publish, */
  14. /* distribute, sublicense, and/or sell copies of the Software, and to */
  15. /* permit persons to whom the Software is furnished to do so, subject to */
  16. /* the following conditions: */
  17. /* */
  18. /* The above copyright notice and this permission notice shall be */
  19. /* included in all copies or substantial portions of the Software. */
  20. /* */
  21. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  22. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  23. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  24. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  25. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  26. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  27. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  28. /*************************************************************************/
  29. #include "math_2d.h"
  30. real_t Vector2::angle() const {
  31. return Math::atan2(x,y);
  32. }
  33. float Vector2::length() const {
  34. return Math::sqrt( x*x + y*y );
  35. }
  36. float Vector2::length_squared() const {
  37. return x*x + y*y;
  38. }
  39. void Vector2::normalize() {
  40. float l = x*x + y*y;
  41. if (l!=0) {
  42. l=Math::sqrt(l);
  43. x/=l;
  44. y/=l;
  45. }
  46. }
  47. Vector2 Vector2::normalized() const {
  48. Vector2 v=*this;
  49. v.normalize();
  50. return v;
  51. }
  52. float Vector2::distance_to(const Vector2& p_vector2) const {
  53. return Math::sqrt( (x-p_vector2.x)*(x-p_vector2.x) + (y-p_vector2.y)*(y-p_vector2.y));
  54. }
  55. float Vector2::distance_squared_to(const Vector2& p_vector2) const {
  56. return (x-p_vector2.x)*(x-p_vector2.x) + (y-p_vector2.y)*(y-p_vector2.y);
  57. }
  58. float Vector2::angle_to(const Vector2& p_vector2) const {
  59. return Math::atan2( tangent().dot(p_vector2), dot(p_vector2) );
  60. }
  61. float Vector2::angle_to_point(const Vector2& p_vector2) const {
  62. return Math::atan2( x-p_vector2.x, y - p_vector2.y );
  63. }
  64. float Vector2::dot(const Vector2& p_other) const {
  65. return x*p_other.x + y*p_other.y;
  66. }
  67. float Vector2::cross(const Vector2& p_other) const {
  68. return x*p_other.y - y*p_other.x;
  69. }
  70. Vector2 Vector2::cross(real_t p_other) const {
  71. return Vector2(p_other*y,-p_other*x);
  72. }
  73. Vector2 Vector2::operator+(const Vector2& p_v) const {
  74. return Vector2(x+p_v.x,y+p_v.y);
  75. }
  76. void Vector2::operator+=(const Vector2& p_v) {
  77. x+=p_v.x; y+=p_v.y;
  78. }
  79. Vector2 Vector2::operator-(const Vector2& p_v) const {
  80. return Vector2(x-p_v.x,y-p_v.y);
  81. }
  82. void Vector2::operator-=(const Vector2& p_v) {
  83. x-=p_v.x; y-=p_v.y;
  84. }
  85. Vector2 Vector2::operator*(const Vector2 &p_v1) const {
  86. return Vector2(x * p_v1.x, y * p_v1.y);
  87. };
  88. Vector2 Vector2::operator*(const float &rvalue) const {
  89. return Vector2(x * rvalue, y * rvalue);
  90. };
  91. void Vector2::operator*=(const float &rvalue) {
  92. x *= rvalue; y *= rvalue;
  93. };
  94. Vector2 Vector2::operator/(const Vector2 &p_v1) const {
  95. return Vector2(x / p_v1.x, y / p_v1.y);
  96. };
  97. Vector2 Vector2::operator/(const float &rvalue) const {
  98. return Vector2(x / rvalue, y / rvalue);
  99. };
  100. void Vector2::operator/=(const float &rvalue) {
  101. x /= rvalue; y /= rvalue;
  102. };
  103. Vector2 Vector2::operator-() const {
  104. return Vector2(-x,-y);
  105. }
  106. bool Vector2::operator==(const Vector2& p_vec2) const {
  107. return x==p_vec2.x && y==p_vec2.y;
  108. }
  109. bool Vector2::operator!=(const Vector2& p_vec2) const {
  110. return x!=p_vec2.x || y!=p_vec2.y;
  111. }
  112. Vector2 Vector2::floor() const {
  113. return Vector2( Math::floor(x), Math::floor(y) );
  114. }
  115. Vector2 Vector2::rotated(float p_by) const {
  116. Vector2 v;
  117. v.set_rotation(angle()+p_by);
  118. v*=length();
  119. return v;
  120. }
  121. Vector2 Vector2::project(const Vector2& p_vec) const {
  122. Vector2 v1=p_vec;
  123. Vector2 v2=*this;
  124. return v2 * ( v1.dot(v2) / v2.dot(v2));
  125. }
  126. Vector2 Vector2::snapped(const Vector2& p_by) const {
  127. return Vector2(
  128. Math::stepify(x,p_by.x),
  129. Math::stepify(y,p_by.y)
  130. );
  131. }
  132. Vector2 Vector2::clamped(real_t p_len) const {
  133. return *this;
  134. real_t l = length();
  135. Vector2 v = *this;
  136. if (l>0 && p_len<l) {
  137. v/=l;
  138. v*=p_len;
  139. }
  140. return v;
  141. }
  142. Vector2 Vector2::cubic_interpolate_soft(const Vector2& p_b,const Vector2& p_pre_a, const Vector2& p_post_b,float p_t) const {
  143. #if 0
  144. k[0] = ((*this) (vi[0] + 1, vi[1], vi[2])) - ((*this) (vi[0],
  145. vi[1],vi[2])); //fk = a0
  146. k[1] = (((*this) (vi[0] + 1, vi[1], vi[2])) - ((*this) ((int) (v(0) -
  147. 1), vi[1],vi[2])))*0.5; //dk = a1
  148. k[2] = (((*this) ((int) (v(0) + 2), vi[1], vi[2])) - ((*this) (vi[0],
  149. vi[1],vi[2])))*0.5; //dk+1
  150. k[3] = k[0]*3 - k[1]*2 - k[2];//a2
  151. k[4] = k[1] + k[2] - k[0]*2;//a3
  152. //ip = a3(t-tk)³ + a2(t-tk)² + a1(t-tk) + a0
  153. //
  154. //a3 = dk + dk+1 - Dk
  155. //a2 = 3Dk - 2dk - dk+1
  156. //a1 = dk
  157. //a0 = fk
  158. //
  159. //dk = (fk+1 - fk-1)*0.5
  160. //Dk = (fk+1 - fk)
  161. float dk =
  162. #endif
  163. return Vector2();
  164. }
  165. Vector2 Vector2::cubic_interpolate(const Vector2& p_b,const Vector2& p_pre_a, const Vector2& p_post_b,float p_t) const {
  166. Vector2 p0=p_pre_a;
  167. Vector2 p1=*this;
  168. Vector2 p2=p_b;
  169. Vector2 p3=p_post_b;
  170. float t = p_t;
  171. float t2 = t * t;
  172. float t3 = t2 * t;
  173. Vector2 out;
  174. out = 0.5f * ( ( p1 * 2.0f) +
  175. ( -p0 + p2 ) * t +
  176. ( 2.0f * p0 - 5.0f * p1 + 4 * p2 - p3 ) * t2 +
  177. ( -p0 + 3.0f * p1 - 3.0f * p2 + p3 ) * t3 );
  178. return out;
  179. /*
  180. float mu = p_t;
  181. float mu2 = mu*mu;
  182. Vector2 a0 = p_post_b - p_b - p_pre_a + *this;
  183. Vector2 a1 = p_pre_a - *this - a0;
  184. Vector2 a2 = p_b - p_pre_a;
  185. Vector2 a3 = *this;
  186. return ( a0*mu*mu2 + a1*mu2 + a2*mu + a3 );
  187. */
  188. /*
  189. float t = p_t;
  190. real_t t2 = t*t;
  191. real_t t3 = t2*t;
  192. real_t a = 2.0*t3- 3.0*t2 + 1;
  193. real_t b = -2.0*t3+ 3.0*t2;
  194. real_t c = t3- 2.0*t2 + t;
  195. real_t d = t3- t2;
  196. Vector2 p_a=*this;
  197. return Vector2(
  198. (a * p_a.x) + (b *p_b.x) + (c * p_pre_a.x) + (d * p_post_b.x),
  199. (a * p_a.y) + (b *p_b.y) + (c * p_pre_a.y) + (d * p_post_b.y)
  200. );
  201. */
  202. }
  203. Vector2 Vector2::slide(const Vector2& p_vec) const {
  204. return p_vec - *this * this->dot(p_vec);
  205. }
  206. Vector2 Vector2::reflect(const Vector2& p_vec) const {
  207. return p_vec - *this * this->dot(p_vec) * 2.0;
  208. }
  209. bool Rect2::intersects_segment(const Point2& p_from, const Point2& p_to, Point2* r_pos,Point2* r_normal) const {
  210. real_t min=0,max=1;
  211. int axis=0;
  212. float sign=0;
  213. for(int i=0;i<2;i++) {
  214. real_t seg_from=p_from[i];
  215. real_t seg_to=p_to[i];
  216. real_t box_begin=pos[i];
  217. real_t box_end=box_begin+size[i];
  218. real_t cmin,cmax;
  219. float csign;
  220. if (seg_from < seg_to) {
  221. if (seg_from > box_end || seg_to < box_begin)
  222. return false;
  223. real_t length=seg_to-seg_from;
  224. cmin = (seg_from < box_begin)?((box_begin - seg_from)/length):0;
  225. cmax = (seg_to > box_end)?((box_end - seg_from)/length):1;
  226. csign=-1.0;
  227. } else {
  228. if (seg_to > box_end || seg_from < box_begin)
  229. return false;
  230. real_t length=seg_to-seg_from;
  231. cmin = (seg_from > box_end)?(box_end - seg_from)/length:0;
  232. cmax = (seg_to < box_begin)?(box_begin - seg_from)/length:1;
  233. csign=1.0;
  234. }
  235. if (cmin > min) {
  236. min = cmin;
  237. axis=i;
  238. sign=csign;
  239. }
  240. if (cmax < max)
  241. max = cmax;
  242. if (max < min)
  243. return false;
  244. }
  245. Vector2 rel=p_to-p_from;
  246. if (r_normal) {
  247. Vector2 normal;
  248. normal[axis]=sign;
  249. *r_normal=normal;
  250. }
  251. if (r_pos)
  252. *r_pos=p_from+rel*min;
  253. return true;
  254. }
  255. /* Point2i */
  256. Point2i Point2i::operator+(const Point2i& p_v) const {
  257. return Point2i(x+p_v.x,y+p_v.y);
  258. }
  259. void Point2i::operator+=(const Point2i& p_v) {
  260. x+=p_v.x; y+=p_v.y;
  261. }
  262. Point2i Point2i::operator-(const Point2i& p_v) const {
  263. return Point2i(x-p_v.x,y-p_v.y);
  264. }
  265. void Point2i::operator-=(const Point2i& p_v) {
  266. x-=p_v.x; y-=p_v.y;
  267. }
  268. Point2i Point2i::operator*(const Point2i &p_v1) const {
  269. return Point2i(x * p_v1.x, y * p_v1.y);
  270. };
  271. Point2i Point2i::operator*(const int &rvalue) const {
  272. return Point2i(x * rvalue, y * rvalue);
  273. };
  274. void Point2i::operator*=(const int &rvalue) {
  275. x *= rvalue; y *= rvalue;
  276. };
  277. Point2i Point2i::operator/(const Point2i &p_v1) const {
  278. return Point2i(x / p_v1.x, y / p_v1.y);
  279. };
  280. Point2i Point2i::operator/(const int &rvalue) const {
  281. return Point2i(x / rvalue, y / rvalue);
  282. };
  283. void Point2i::operator/=(const int &rvalue) {
  284. x /= rvalue; y /= rvalue;
  285. };
  286. Point2i Point2i::operator-() const {
  287. return Point2i(-x,-y);
  288. }
  289. bool Point2i::operator==(const Point2i& p_vec2) const {
  290. return x==p_vec2.x && y==p_vec2.y;
  291. }
  292. bool Point2i::operator!=(const Point2i& p_vec2) const {
  293. return x!=p_vec2.x || y!=p_vec2.y;
  294. }
  295. void Matrix32::invert() {
  296. SWAP(elements[0][1],elements[1][0]);
  297. elements[2] = basis_xform(-elements[2]);
  298. }
  299. Matrix32 Matrix32::inverse() const {
  300. Matrix32 inv=*this;
  301. inv.invert();
  302. return inv;
  303. }
  304. void Matrix32::affine_invert() {
  305. float det = elements[0][0]*elements[1][1] - elements[1][0]*elements[0][1];
  306. ERR_FAIL_COND(det==0);
  307. float idet = 1.0 / det;
  308. SWAP( elements[0][0],elements[1][1] );
  309. elements[0]*=Vector2(idet,-idet);
  310. elements[1]*=Vector2(-idet,idet);
  311. elements[2] = basis_xform(-elements[2]);
  312. }
  313. Matrix32 Matrix32::affine_inverse() const {
  314. Matrix32 inv=*this;
  315. inv.affine_invert();
  316. return inv;
  317. }
  318. void Matrix32::rotate(real_t p_phi) {
  319. Matrix32 rot(p_phi,Vector2());
  320. *this *= rot;
  321. }
  322. real_t Matrix32::get_rotation() const {
  323. return Math::atan2(elements[1].x,elements[1].y);
  324. }
  325. void Matrix32::set_rotation(real_t p_rot) {
  326. real_t cr = Math::cos(p_rot);
  327. real_t sr = Math::sin(p_rot);
  328. elements[0][0]=cr;
  329. elements[1][1]=cr;
  330. elements[0][1]=-sr;
  331. elements[1][0]=sr;
  332. }
  333. Matrix32::Matrix32(real_t p_rot, const Vector2& p_pos) {
  334. real_t cr = Math::cos(p_rot);
  335. real_t sr = Math::sin(p_rot);
  336. elements[0][0]=cr;
  337. elements[1][1]=cr;
  338. elements[0][1]=-sr;
  339. elements[1][0]=sr;
  340. elements[2]=p_pos;
  341. }
  342. Vector2 Matrix32::get_scale() const {
  343. return Vector2( elements[0].length(), elements[1].length() );
  344. }
  345. void Matrix32::scale(const Vector2& p_scale) {
  346. elements[0]*=p_scale;
  347. elements[1]*=p_scale;
  348. elements[2]*=p_scale;
  349. }
  350. void Matrix32::scale_basis(const Vector2& p_scale) {
  351. elements[0]*=p_scale;
  352. elements[1]*=p_scale;
  353. }
  354. void Matrix32::translate( real_t p_tx, real_t p_ty) {
  355. translate(Vector2(p_tx,p_ty));
  356. }
  357. void Matrix32::translate( const Vector2& p_translation ) {
  358. elements[2]+=basis_xform(p_translation);
  359. }
  360. void Matrix32::orthonormalize() {
  361. // Gram-Schmidt Process
  362. Vector2 x=elements[0];
  363. Vector2 y=elements[1];
  364. x.normalize();
  365. y = (y-x*(x.dot(y)));
  366. y.normalize();
  367. elements[0]=x;
  368. elements[1]=y;
  369. }
  370. Matrix32 Matrix32::orthonormalized() const {
  371. Matrix32 on=*this;
  372. on.orthonormalize();
  373. return on;
  374. }
  375. bool Matrix32::operator==(const Matrix32& p_transform) const {
  376. for(int i=0;i<3;i++) {
  377. if (elements[i]!=p_transform.elements[i])
  378. return false;
  379. }
  380. return true;
  381. }
  382. bool Matrix32::operator!=(const Matrix32& p_transform) const {
  383. for(int i=0;i<3;i++) {
  384. if (elements[i]!=p_transform.elements[i])
  385. return true;
  386. }
  387. return false;
  388. }
  389. void Matrix32::operator*=(const Matrix32& p_transform) {
  390. elements[2] = xform(p_transform.elements[2]);
  391. float x0,x1,y0,y1;
  392. /*
  393. x0 = p_transform.tdotx(elements[0]);
  394. x1 = p_transform.tdoty(elements[0]);
  395. y0 = p_transform.tdotx(elements[1]);
  396. y1 = p_transform.tdoty(elements[1]);*/
  397. x0 = tdotx(p_transform.elements[0]);
  398. x1 = tdoty(p_transform.elements[0]);
  399. y0 = tdotx(p_transform.elements[1]);
  400. y1 = tdoty(p_transform.elements[1]);
  401. elements[0][0]=x0;
  402. elements[0][1]=x1;
  403. elements[1][0]=y0;
  404. elements[1][1]=y1;
  405. }
  406. Matrix32 Matrix32::operator*(const Matrix32& p_transform) const {
  407. Matrix32 t = *this;
  408. t*=p_transform;
  409. return t;
  410. }
  411. Matrix32 Matrix32::scaled(const Vector2& p_scale) const {
  412. Matrix32 copy=*this;
  413. copy.scale(p_scale);
  414. return copy;
  415. }
  416. Matrix32 Matrix32::basis_scaled(const Vector2& p_scale) const {
  417. Matrix32 copy=*this;
  418. copy.scale_basis(p_scale);
  419. return copy;
  420. }
  421. Matrix32 Matrix32::untranslated() const {
  422. Matrix32 copy=*this;
  423. copy.elements[2]=Vector2();
  424. return copy;
  425. }
  426. Matrix32 Matrix32::translated(const Vector2& p_offset) const {
  427. Matrix32 copy=*this;
  428. copy.translate(p_offset);
  429. return copy;
  430. }
  431. Matrix32 Matrix32::rotated(float p_phi) const {
  432. Matrix32 copy=*this;
  433. copy.rotate(p_phi);
  434. return copy;
  435. }
  436. float Matrix32::basis_determinant() const {
  437. return elements[0].x * elements[1].y - elements[0].y * elements[1].x;
  438. }
  439. Matrix32 Matrix32::interpolate_with(const Matrix32& p_transform, float p_c) const {
  440. //extract parameters
  441. Vector2 p1 = get_origin();
  442. Vector2 p2 = p_transform.get_origin();
  443. real_t r1 = get_rotation();
  444. real_t r2 = p_transform.get_rotation();
  445. Vector2 s1 = get_scale();
  446. Vector2 s2 = p_transform.get_scale();
  447. //slerp rotation
  448. Vector2 v1(Math::cos(r1), Math::sin(r1));
  449. Vector2 v2(Math::cos(r2), Math::sin(r2));
  450. real_t dot = v1.dot(v2);
  451. dot = (dot < -1.0) ? -1.0 : ((dot > 1.0) ? 1.0 : dot); //clamp dot to [-1,1]
  452. Vector2 v;
  453. if (dot > 0.9995) {
  454. v = Vector2::linear_interpolate(v1, v2, p_c).normalized(); //linearly interpolate to avoid numerical precision issues
  455. } else {
  456. real_t angle = p_c*Math::acos(dot);
  457. Vector2 v3 = (v2 - v1*dot).normalized();
  458. v = v1*Math::cos(angle) + v3*Math::sin(angle);
  459. }
  460. //construct matrix
  461. Matrix32 res(Math::atan2(v.y, v.x), Vector2::linear_interpolate(p1, p2, p_c));
  462. res.scale_basis(Vector2::linear_interpolate(s1, s2, p_c));
  463. return res;
  464. }
  465. Matrix32::operator String() const {
  466. return String(String()+elements[0]+", "+elements[1]+", "+elements[2]);
  467. }