// Copyright (C) 2008 Davis E. King (davis@dlib.net) // License: Boost Software License See LICENSE.txt for the full license. #ifndef DLIB_DRAW_IMAGe_ #define DLIB_DRAW_IMAGe_ #include "draw_abstract.h" #include "../algs.h" #include "../pixel.h" #include <cmath> namespace dlib { // ---------------------------------------------------------------------------------------- template < typename image_type, typename pixel_type > void draw_line ( long x1, long y1, long x2, long y2, image_type& c, const pixel_type& val ) { if (x1 == x2) { // make sure y1 comes before y2 if (y1 > y2) swap(y1,y2); if (x1 < 0 || x1 >= c.nc()) return; // this is a vertical line for (long y = y1; y <= y2; ++y) { if (y < 0 || y >= c.nr()) continue; assign_pixel(c[y][x1], val); } } else if (y1 == y2) { // make sure x1 comes before x2 if (x1 > x2) swap(x1,x2); if (y1 < 0 || y1 >= c.nr()) return; // this is a horizontal line for (long x = x1; x <= x2; ++x) { if (x < 0 || x >= c.nc()) continue; assign_pixel(c[y1][x] , val); } } else { const long rise = (((long)y2) - ((long)y1)); const long run = (((long)x2) - ((long)x1)); if (std::abs(rise) < std::abs(run)) { const double slope = ((double)rise)/run; double first, last; if (x1 > x2) { first = x2; last = x1; } else { first = x1; last = x2; } long y; long x; const double x1f = x1; const double y1f = y1; for (double i = first; i <= last; ++i) { y = static_cast<long>(slope*(i-x1f) + y1f); x = static_cast<long>(i); if (y < 0 || y >= c.nr()) continue; if (x < 0 || x >= c.nc()) continue; assign_pixel(c[y][x] , val); } } else { const double slope = ((double)run)/rise; double first, last; if (y1 > y2) { first = y2; last = y1; } else { first = y1; last = y2; } long x; long y; const double x1f = x1; const double y1f = y1; for (double i = first; i <= last; ++i) { x = static_cast<long>(slope*(i-y1f) + x1f); y = static_cast<long>(i); if (x < 0 || x >= c.nc()) continue; if (y < 0 || y >= c.nr()) continue; assign_pixel(c[y][x] , val); } } } } // ---------------------------------------------------------------------------------------- template < typename image_type, typename pixel_type > void draw_line ( image_type& c, const point& p1, const point& p2, const pixel_type& val ) { draw_line(p1.x(),p1.y(),p2.x(),p2.y(),c,val); } // ---------------------------------------------------------------------------------------- template < typename image_type, typename pixel_type > void fill_rect ( image_type& img, const rectangle& rect, const pixel_type& pixel ) { rectangle area = rect.intersect(get_rect(img)); for (long r = area.top(); r <= area.bottom(); ++r) { for (long c = area.left(); c <= area.right(); ++c) { assign_pixel(img[r][c], pixel); } } } // ---------------------------------------------------------------------------------------- } #endif // DLIB_DRAW_IMAGe_