/*
|
* Rockchip App
|
*
|
* Copyright (C) 2017 Rockchip Electronics Co., Ltd.
|
* author: hogan.wang@rock-chips.com
|
*
|
* This software is available to you under a choice of one of two
|
* licenses. You may choose to be licensed under the terms of the GNU
|
* General Public License (GPL), available from the file
|
* COPYING in the main directory of this source tree, or the
|
* OpenIB.org BSD license below:
|
*
|
* Redistribution and use in source and binary forms, with or
|
* without modification, are permitted provided that the following
|
* conditions are met:
|
*
|
* - Redistributions of source code must retain the above
|
* copyright notice, this list of conditions and the following
|
* disclaimer.
|
*
|
* - Redistributions in binary form must reproduce the above
|
* copyright notice, this list of conditions and the following
|
* disclaimer in the documentation and/or other materials
|
* provided with the distribution.
|
*
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
* SOFTWARE.
|
*/
|
|
#include "draw_rect.h"
|
|
#ifndef ABS
|
#define ABS(x) (((x)<0) ? -(x) : (x))
|
#endif
|
|
static void YUV_SWAP(int *x, int *y)
|
{
|
int tmp = 0;
|
|
tmp = *x;
|
*x = *y;
|
*y = tmp;
|
}
|
|
YUV_Color set_yuv_color(COLOR_Type color_type)
|
{
|
YUV_Color color;
|
|
switch (color_type) {
|
case COLOR_Y:
|
color.Y = 226;
|
color.U = 0;
|
color.V = 149;
|
break;
|
|
case COLOR_R:
|
color.Y = 76;
|
color.U = 85;
|
color.V = 255;
|
break;
|
|
case COLOR_G:
|
color.Y = 150;
|
color.U = 44;
|
color.V = 21;
|
break;
|
|
case COLOR_B:
|
color.Y = 29;
|
color.U = 255;
|
color.V = 107;
|
break;
|
|
case COLOR_BK:
|
color.Y = 16;
|
color.U = 128;
|
color.V = 128;
|
break;
|
|
default:
|
color.Y = 128;
|
color.U = 128;
|
color.V = 128;
|
break;
|
}
|
|
return color;
|
}
|
|
static int yuv420_draw_point(unsigned char* imgdata,
|
int width,
|
int height,
|
YUV_Point Point,
|
YUV_Color color)
|
{
|
int imgSize;
|
int x, y;
|
|
if (!imgdata)
|
return -1;
|
|
if (width < 0 || height < 0)
|
return -1;
|
|
if (Point.x < 0 || Point.x > width ||
|
Point.y < 0 || Point.y > height)
|
return -1;
|
|
if (Point.x % 2)
|
Point.x--;
|
if (Point.y % 2)
|
Point.y--;
|
|
imgSize = width * height;
|
x = Point.x;
|
y = Point.y;
|
|
imgdata[y * width + x] = color.Y;
|
imgdata[imgSize + y * width / 2 + x] = color.U;
|
imgdata[imgSize + y * width / 2 + x + 1] = color.V;
|
|
return 0;
|
}
|
|
static int yuv420_draw_vline(void* imgdata,
|
int width,
|
int height,
|
YUV_Point startPoint,
|
YUV_Point endPoint,
|
YUV_Color color)
|
{
|
int ret = 0;
|
int x0, y0, y1;
|
YUV_Point Point;
|
|
x0 = startPoint.x;
|
y0 = startPoint.y;
|
y1 = endPoint.y;
|
|
if (y0 > y1)
|
YUV_SWAP(&y0, &y1);
|
|
Point.x = x0;
|
Point.y = y0;
|
|
while (Point.y < y1) {
|
Point.y++;
|
ret = yuv420_draw_point((unsigned char *)imgdata,
|
width, height, Point, color);
|
if (ret)
|
return -1;
|
}
|
|
return 0;
|
}
|
|
static int yuv420_draw_hline(void* imgdata,
|
int width,
|
int height,
|
YUV_Point startPoint,
|
YUV_Point endPoint,
|
YUV_Color color)
|
{
|
int ret = 0;
|
int x0, y0, x1;
|
YUV_Point Point;
|
|
x0 = startPoint.x;
|
y0 = startPoint.y;
|
x1 = endPoint.x;
|
|
if (x0 > x1)
|
YUV_SWAP(&x0, &x1);
|
|
Point.x = x0;
|
Point.y = y0;
|
|
while (Point.x < x1) {
|
Point.x++;
|
ret = yuv420_draw_point((unsigned char *)imgdata,
|
width, height, Point, color);
|
if (ret)
|
return -1;
|
}
|
|
return 0;
|
}
|
|
static int yuv420_draw_tline(void* imgdata,
|
int width,
|
int height,
|
YUV_Point startPoint,
|
YUV_Point endPoint,
|
YUV_Color color)
|
{
|
int x0, y0, x1, y1;
|
int dx, dy, error;
|
YUV_Point Point;
|
|
x0 = startPoint.x;
|
y0 = startPoint.y;
|
x1 = endPoint.x;
|
y1 = endPoint.y;
|
|
if (x0 > x1) {
|
YUV_SWAP(&x0, &x1);
|
YUV_SWAP(&y0, &y1);
|
}
|
|
dx = x1 - x0;
|
dy = y1 - y0;
|
Point.x = x0;
|
Point.y = y0;
|
|
if (dx == 0) {
|
yuv420_draw_vline(imgdata, width,
|
height, startPoint, endPoint, color);
|
return 0;
|
}
|
if (dy == 0){
|
yuv420_draw_hline(imgdata, width,
|
height, startPoint, endPoint, color);
|
return 0;
|
}
|
|
yuv420_draw_point((unsigned char *)imgdata,
|
width, height, Point, color);
|
|
if ((dx >= dy) && (dy > 0)) {/* 0<k<=1 */
|
error = (dy*2) - dx;
|
while (Point.x < x1) {
|
if (error > 0) {
|
error += (dy - dx) * 2;
|
Point.x++;
|
Point.y++;
|
} else {
|
error += dy * 2;
|
Point.x++;
|
}
|
yuv420_draw_point((unsigned char *)imgdata,
|
width, height, Point, color);
|
}
|
return 0;
|
}
|
|
if ((dy > dx) && (dy > 0)) {/* k>1 */
|
error = dy - (dx * 2);
|
while (Point.y < y1) {
|
if (error < 0) {
|
error += (dy - dx) * 2;
|
Point.x++;
|
Point.y++;
|
} else {
|
error += (-dx) * 2;
|
Point.y++;
|
}
|
yuv420_draw_point((unsigned char *)imgdata,
|
width, height, Point, color);
|
}
|
return 0;
|
}
|
|
if ((dx >= ABS(dy)) && (dy < 0)) { /* -1=<k<0 */
|
error = (dy * 2) + dx;
|
while (Point.x < x1) {
|
if (error < 0) {
|
error += (dy + dx) * 2;
|
Point.x++;
|
Point.y--;
|
} else {
|
error += dy * 2;
|
Point.x++;
|
}
|
yuv420_draw_point((unsigned char *)imgdata,
|
width, height, Point, color);
|
}
|
return 0;
|
}
|
|
if ((ABS(dy) > dx) && (dy < 0)) { /* k<-1 */
|
error = dy + (dx * 2);
|
while (Point.y > y1) {
|
if (error > 0) {
|
error += (dy + dx) * 2;
|
Point.x++;
|
Point.y--;
|
} else {
|
error += dx * 2;
|
Point.y--;
|
}
|
yuv420_draw_point((unsigned char *)imgdata,
|
width, height, Point, color);
|
}
|
return 0;
|
}
|
|
return 0;
|
}
|
|
void yuv420_draw_line(void* imgdata,
|
int width,
|
int height,
|
YUV_Point startPoint,
|
YUV_Point endPoint,
|
YUV_Color color)
|
{
|
YUV_Point sPoint;
|
YUV_Point ePoint;
|
|
sPoint.x = startPoint.x - 2;
|
sPoint.y = startPoint.y - 2;
|
ePoint.x = endPoint.x - 2;
|
ePoint.y = endPoint.y - 2;
|
yuv420_draw_tline(imgdata, width, height, sPoint, ePoint, color);
|
|
sPoint.x = startPoint.x;
|
sPoint.y = startPoint.y;
|
ePoint.x = endPoint.x;
|
ePoint.y = endPoint.y;
|
yuv420_draw_tline(imgdata, width, height, sPoint, ePoint, color);
|
|
sPoint.x = startPoint.x + 2;
|
sPoint.y = startPoint.y + 2;
|
ePoint.x = endPoint.x + 2;
|
ePoint.y = endPoint.y + 2;
|
yuv420_draw_tline(imgdata, width, height, sPoint, ePoint, color);
|
}
|
|
void yuv420_draw_rectangle(void* imgdata,
|
int width,
|
int height,
|
YUV_Rect rect_rio,
|
YUV_Color color)
|
{
|
int RoiWidth = rect_rio.width;
|
int RoiHeight = rect_rio.height;
|
int x = rect_rio.x;
|
int y = rect_rio.y;
|
YUV_Point Point[4];
|
|
Point[0].x = x;
|
Point[0].y = y;
|
Point[1].x = x + RoiWidth - 2;
|
Point[1].y = y;
|
Point[2].x = x + RoiWidth - 2;
|
Point[2].y = y + RoiHeight - 2;
|
Point[3].x = x;
|
Point[3].y = y + RoiHeight - 2;
|
|
yuv420_draw_line(imgdata, width, height, Point[0], Point[1], color);
|
yuv420_draw_line(imgdata, width, height, Point[1], Point[2], color);
|
yuv420_draw_line(imgdata, width, height, Point[3], Point[2], color);
|
yuv420_draw_line(imgdata, width, height, Point[0], Point[3], color);
|
}
|
