111 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
			
		
		
	
	
			111 lines
		
	
	
		
			3.3 KiB
		
	
	
	
		
			C++
		
	
	
| /*
 | |
|  * Rot2.cpp
 | |
|  *
 | |
|  *  Created on: Dec 9, 2009
 | |
|  *      Author: Frank Dellaert
 | |
|  */
 | |
| 
 | |
| #include "Rot2.h"
 | |
| #include "Lie-inl.h"
 | |
| 
 | |
| using namespace std;
 | |
| 
 | |
| namespace gtsam {
 | |
| 
 | |
|   /** Explicit instantiation of base class to export members */
 | |
|   INSTANTIATE_LIE(Rot2);
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	Rot2 Rot2::fromCosSin(double c, double s) {
 | |
| 		if (fabs(c * c + s * s - 1.0) > 1e-9) {
 | |
|     	double norm_cs = sqrt(c*c + s*s);
 | |
|     	c = c/norm_cs;
 | |
|     	s = s/norm_cs;
 | |
| 		}
 | |
| 		return Rot2(c, s);
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	Rot2 Rot2::atan2(double y, double x) {
 | |
| 		Rot2 R(x, y);
 | |
| 		return R.normalize();
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	void Rot2::print(const string& s) const {
 | |
| 		cout << s << ":" << theta() << endl;
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	bool Rot2::equals(const Rot2& R, double tol) const {
 | |
| 		return fabs(c_ - R.c_) <= tol && fabs(s_ - R.s_) <= tol;
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
|   Rot2& Rot2::normalize() {
 | |
|   	double scale = c_*c_ + s_*s_;
 | |
|   	if(fabs(scale-1.0)>1e-9) {
 | |
|   		scale = pow(scale, -0.5);
 | |
|   		c_ *= scale;
 | |
|   		s_ *= scale;
 | |
|   	}
 | |
|   	return *this;
 | |
|   }
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	Matrix Rot2::matrix() const {
 | |
| 		return Matrix_(2, 2, c_, -s_, s_, c_);
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	Matrix Rot2::transpose() const {
 | |
| 		return Matrix_(2, 2, c_, s_, -s_, c_);
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	Point2 Rot2::rotate(const Point2& p) const {
 | |
| 		return Point2(c_ * p.x() + -s_ * p.y(), s_ * p.x() + c_ * p.y());
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	Point2 Rot2::unrotate(const Point2& p) const {
 | |
| 		return Point2(c_ * p.x() + s_ * p.y(), -s_ * p.x() + c_ * p.y());
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
|   // see doc/math.lyx, SO(2) section
 | |
| 	Point2 rotate(const Rot2 & R, const Point2& p, boost::optional<Matrix&> H1,
 | |
| 			boost::optional<Matrix&> H2) {
 | |
| 		Point2 q = R * p;
 | |
| 		if (H1) *H1 = Matrix_(2, 1, -q.y(), q.x());
 | |
| 		if (H2) *H2 = R.matrix();
 | |
| 		return q;
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
|   // see doc/math.lyx, SO(2) section
 | |
| 	Point2 unrotate(const Rot2 & R, const Point2& p,
 | |
| 			boost::optional<Matrix&> H1, boost::optional<Matrix&> H2) {
 | |
| 		Point2 q = R.unrotate(p);
 | |
| 		if (H1) *H1 = Matrix_(2, 1, q.y(), -q.x());  // R_{pi/2}q
 | |
| 		if (H2) *H2 = R.transpose();
 | |
| 		return q;
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	Rot2 relativeBearing(const Point2& d) {
 | |
| 		double n = d.norm();
 | |
| 		return Rot2::fromCosSin(d.x() / n, d.y() / n);
 | |
| 	}
 | |
| 
 | |
| 	/* ************************************************************************* */
 | |
| 	Rot2 relativeBearing(const Point2& d, boost::optional<Matrix&> H) {
 | |
| 		double x = d.x(), y = d.y(), d2 = x * x + y * y, n = sqrt(d2);
 | |
| 		if (H) *H = Matrix_(1, 2, -y / d2, x / d2);
 | |
| 		return Rot2::fromCosSin(x / n, y / n);
 | |
| 	}
 | |
| 
 | |
| /* ************************************************************************* */
 | |
| 
 | |
| } // gtsam
 |