forked from davisking/dlib
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Preparing for correlation tracker - add drectangle
The correlation tracker deals with the drectangle (double rectangle) class which currently isn't wrapped. Therefore, I add the drectangle class and refactor rectangles into their own file. I also added a load of methods on rectangle that might be useful such as intersection/contains/area etc.
- Loading branch information
1 parent
8568c26
commit c4cf31a
Showing
4 changed files
with
154 additions
and
66 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,127 @@ | ||
| // Copyright (C) 2015 Davis E. King (davis@dlib.net) | ||
| // License: Boost Software License See LICENSE.txt for the full license. | ||
|
|
||
| #include <dlib/python.h> | ||
| #include <boost/python/args.hpp> | ||
| #include <dlib/geometry.h> | ||
| #include "indexing.h" | ||
|
|
||
| using namespace dlib; | ||
| using namespace std; | ||
| using namespace boost::python; | ||
|
|
||
| // ---------------------------------------------------------------------------------------- | ||
|
|
||
| long left(const rectangle& r) { return r.left(); } | ||
| long top(const rectangle& r) { return r.top(); } | ||
| long right(const rectangle& r) { return r.right(); } | ||
| long bottom(const rectangle& r) { return r.bottom(); } | ||
| long width(const rectangle& r) { return r.width(); } | ||
| long height(const rectangle& r) { return r.height(); } | ||
| unsigned long area(const rectangle& r) { return r.area(); } | ||
|
|
||
| double dleft(const drectangle& r) { return r.left(); } | ||
| double dtop(const drectangle& r) { return r.top(); } | ||
| double dright(const drectangle& r) { return r.right(); } | ||
| double dbottom(const drectangle& r) { return r.bottom(); } | ||
| double dwidth(const drectangle& r) { return r.width(); } | ||
| double dheight(const drectangle& r) { return r.height(); } | ||
| double darea(const drectangle& r) { return r.area(); } | ||
|
|
||
| template <typename rect_type> | ||
| bool is_empty(const rect_type& r) { return r.is_empty(); } | ||
|
|
||
| template <typename rect_type> | ||
| point center(const rect_type& r) { return center(r); } | ||
|
|
||
| template <typename rect_type> | ||
| point dcenter(const rect_type& r) { return dcenter(r); } | ||
|
|
||
| template <typename rect_type> | ||
| bool contains(const rect_type& r, const point& p) { return r.contains(p); } | ||
|
|
||
| template <typename rect_type> | ||
| bool contains_xy(const rect_type& r, const long x, const long y) { return r.contains(point(x, y)); } | ||
|
|
||
| template <typename rect_type> | ||
| bool contains_rec(const rect_type& r, const rect_type& r2) { return r.contains(r2); } | ||
|
|
||
| template <typename rect_type> | ||
| rect_type intersect(const rect_type& r, const rect_type& r2) { return r.intersect(r2); } | ||
|
|
||
| template <typename rect_type> | ||
| string print_rectangle_str(const rect_type& r) | ||
| { | ||
| std::ostringstream sout; | ||
| sout << r; | ||
| return sout.str(); | ||
| } | ||
|
|
||
| template <typename rect_type> | ||
| string print_rectangle_repr(const rect_type& r) | ||
| { | ||
| std::ostringstream sout; | ||
| sout << "rectangle(" << r.left() << "," << r.top() << "," << r.right() << "," << r.bottom() << ")"; | ||
| return sout.str(); | ||
| } | ||
|
|
||
| // ---------------------------------------------------------------------------------------- | ||
|
|
||
| void bind_rectangles() | ||
| { | ||
| using boost::python::arg; | ||
| { | ||
| typedef rectangle type; | ||
| class_<type>("rectangle", "This object represents a rectangular area of an image.") | ||
| .def(init<long,long,long,long>( (arg("left"),arg("top"),arg("right"),arg("bottom")) )) | ||
| .def("area", &::area) | ||
| .def("left", &::left) | ||
| .def("top", &::top) | ||
| .def("right", &::right) | ||
| .def("bottom", &::bottom) | ||
| .def("width", &::width) | ||
| .def("height", &::height) | ||
| .def("is_empty", &::is_empty<type>) | ||
| .def("center", &::center<type>) | ||
| .def("dcenter", &::dcenter<type>) | ||
| .def("contains", &::contains<type>, arg("point")) | ||
| .def("contains", &::contains_xy<type>, (arg("x"), arg("y"))) | ||
| .def("contains", &::contains_rec<type>, (arg("rectangle"))) | ||
| .def("intersect", &::intersect<type>, (arg("rectangle"))) | ||
| .def("__str__", &::print_rectangle_str<type>) | ||
| .def("__repr__", &::print_rectangle_repr<type>) | ||
| .def_pickle(serialize_pickle<type>()); | ||
| } | ||
| { | ||
| typedef drectangle type; | ||
| class_<type>("drectangle", "This object represents a rectangular area of an image with floating point coordinates.") | ||
| .def(init<double,double,double,double>( (arg("left"),arg("top"),arg("right"),arg("bottom")) )) | ||
| .def("area", &::darea) | ||
| .def("left", &::dleft) | ||
| .def("top", &::dtop) | ||
| .def("right", &::dright) | ||
| .def("bottom", &::dbottom) | ||
| .def("width", &::dwidth) | ||
| .def("height", &::dheight) | ||
| .def("is_empty", &::is_empty<type>) | ||
| .def("center", &::center<type>) | ||
| .def("dcenter", &::dcenter<type>) | ||
| .def("contains", &::contains<type>, arg("point")) | ||
| .def("contains", &::contains_xy<type>, (arg("x"), arg("y"))) | ||
| .def("contains", &::contains_rec<type>, (arg("rectangle"))) | ||
| .def("intersect", &::intersect<type>, (arg("rectangle"))) | ||
| .def("__str__", &::print_rectangle_str<type>) | ||
| .def("__repr__", &::print_rectangle_repr<type>) | ||
| .def_pickle(serialize_pickle<type>()); | ||
| } | ||
| { | ||
| typedef std::vector<rectangle> type; | ||
| class_<type>("rectangles", "An array of rectangle objects.") | ||
| .def(vector_indexing_suite<type>()) | ||
| .def("clear", &type::clear) | ||
| .def("resize", resize<type>) | ||
| .def_pickle(serialize_pickle<type>()); | ||
| } | ||
| } | ||
|
|
||
| // ---------------------------------------------------------------------------------------- |