resize_utils.cpp File Reference

#include <cstdlib>
#include <opencv2/imgproc.hpp>
#include <opencv2/imgproc/types_c.h>
#include "capture_filter/resize_utils.h"
#include "debug.h"
#include "utils/color_out.h"
#include "video.h"

Include dependency graph for resize_utils.cpp:

Go to the source code of this file.

Macros
#define	DEFAULT_ALGO   INTER_LINEAR
#define	MOD_NAME   "[resize] "

Functions
static int	get_out_cv_data_type (codec_t pixfmt)
int	resize_algo_from_string (const char *str)
static const char *	resize_algo_to_string (int algo)
void	resize_frame (char *indata, codec_t in_color, char *outdata, int width, int height, struct resize_param *resize_spec)
static void	resize_frame_dimensions (char *indata, codec_t in_color, char *outdata, int width, int height, int target_width, int target_height, int algo)
static Mat	ug_to_rgb_mat (codec_t codec, int width, int height, char *indata)

Variables
struct {
const char *   name
int   val
}	interp_map []

Detailed Description

Author

Gerard Castillo gerar.nosp@m.d.ca.nosp@m.still.nosp@m.o@i2.nosp@m.cat.n.nosp@m.et Marc Palau marc..nosp@m.pala.nosp@m.u@i2c.nosp@m.at.n.nosp@m.et Martin Pulec marti.nosp@m.n.pu.nosp@m.lec@c.nosp@m.esne.nosp@m.t.cz

Definition in file resize_utils.cpp.

Macro Definition Documentation

DEFAULT_ALGO

#define DEFAULT_ALGO   INTER_LINEAR

Definition at line 69 of file resize_utils.cpp.

Referenced by resize_algo_from_string(), and resize_frame().

MOD_NAME

#define MOD_NAME   "[resize] "

Definition at line 70 of file resize_utils.cpp.

Referenced by ug_to_rgb_mat().

Function Documentation

get_out_cv_data_type()

static int get_out_cv_data_type ( codec_t pixfmt )

static

Definition at line 130 of file resize_utils.cpp.

{
    return get_bits_per_component(pixfmt) == DEPTH16 ? CV_16UC3 : CV_8UC3;
}

References DEPTH16, and get_bits_per_component().

Referenced by resize_frame(), and resize_frame_dimensions().

Here is the call graph for this function:

resize_algo_from_string()

int resize_algo_from_string

(

const char *

str

)

Definition at line 224 of file resize_utils.cpp.

{
    if (strcmp(str, "help") == 0) {
        color_printf("Available resize algorithms:\n");
        for (auto const &i : interp_map) {
            color_printf("\t" TBOLD("%s") "%s\n", i.name,
                         i.val == DEFAULT_ALGO ? " (default)" : "");
        }
        return RESIZE_ALGO_HELP_SHOWN;
    }
    for (auto const &i : interp_map) {
        if (strcmp(i.name, str) == 0) {
            return i.val;
        }
    }
    return RESIZE_ALGO_UNKN;
}

References color_printf(), DEFAULT_ALGO, interp_map, RESIZE_ALGO_HELP_SHOWN, RESIZE_ALGO_UNKN, and TBOLD.

Referenced by parse_fmt().

Here is the call graph for this function:

resize_algo_to_string()

static const char * resize_algo_to_string ( int algo )

static

Definition at line 243 of file resize_utils.cpp.

{
    for (auto const &i : interp_map) {
        if (i.val == algo) {
            return i.name;
        }
    }
    return "(unknown algo!)";
}

References interp_map.

Referenced by resize_frame().

resize_frame()

void resize_frame	(	char *	indata,
		codec_t	in_color,
		char *	outdata,
		int	width,
		int	height,
		struct resize_param *	resize_spec )

Definition at line 184 of file resize_utils.cpp.

{
    if (resize_spec->algo == RESIZE_ALGO_DFL) {
        resize_spec->algo = DEFAULT_ALGO;
        MSG(NOTICE, "using resize algorithm: %s\n",
          resize_algo_to_string(DEFAULT_ALGO));
    }
 
    DEBUG_TIMER_START(resize);
    if (resize_spec->mode == resize_param::USE_FRACTION) {
        const double factor = resize_spec->factor;
        Mat rgb = ug_to_rgb_mat(in_color, (int) width, (int) height, indata);
        Mat out((int) (height * factor), (int) (width * factor),
                get_out_cv_data_type(in_color), outdata);
        resize(rgb, out, Size(0, 0), factor, factor, resize_spec->algo);
    } else if (resize_spec->mode == resize_param::USE_DIMENSIONS) {
        resize_frame_dimensions(indata, in_color, outdata, width, height,
                                resize_spec->target_width,
                                resize_spec->target_height, resize_spec->algo);
    } else {
        abort();
    }
    DEBUG_TIMER_STOP(resize);
}

References resize_param::algo, DEBUG_TIMER_START, DEBUG_TIMER_STOP, DEFAULT_ALGO, resize_param::factor, get_out_cv_data_type(), resize_param::mode, MSG, RESIZE_ALGO_DFL, resize_algo_to_string(), resize_frame_dimensions(), resize_param::target_height, resize_param::target_width, ug_to_rgb_mat(), resize_param::USE_DIMENSIONS, and resize_param::USE_FRACTION.

Referenced by filter().

Here is the call graph for this function:

resize_frame_dimensions()

static void resize_frame_dimensions ( char * indata, codec_t in_color, char * outdata, int width, int height, int target_width, int target_height, int algo )

static

Definition at line 136 of file resize_utils.cpp.

{
    const codec_t out_color =
        get_bits_per_component(in_color) == 16 ? RG48 : RGB;
    Mat rgb = ug_to_rgb_mat(in_color, (int) width, (int) height, indata);
 
    double in_aspect = (double) width / height;
    double out_aspect = (double) target_width / target_height;
    Rect r;
    if (in_aspect == out_aspect) {
        r.x = 0;
        r.y = 0;
        r.width = target_width;
        r.height = target_height;
    } else if (in_aspect > out_aspect) {
        r.x = 0;
        r.width = target_width;
        r.height = (int) (target_width / in_aspect);
        r.y = (target_height - r.height) / 2;
        // clear top and bottom margin
        size_t linesize = vc_get_linesize(target_width, out_color);
        size_t top_margin_size = r.y * linesize;
        size_t bottom_margin_size = (target_height - r.y - r.height) * linesize;
        memset(outdata, 0, top_margin_size);
        memset(outdata + linesize * target_height - bottom_margin_size, 0, bottom_margin_size);
    } else {
        r.y = 0;
        r.height = target_height;
        r.width = (int) (target_height * in_aspect);
        r.x = (target_width - r.width) / 2;
        // clear left and right margins
        size_t linesize = vc_get_linesize(target_width, out_color);
        size_t left_margin_size = vc_get_linesize(r.x, out_color);
        size_t right_margin_size = vc_get_linesize(target_width - r.x - r.width, out_color);
        for (int i = 0; i < target_height; ++i) {
            memset(outdata + i * linesize, 0, left_margin_size);
            memset(outdata + (i + 1) * linesize - right_margin_size, 0, right_margin_size);
        }
    }
 
    Mat out((int) target_height, (int) target_width,
            get_out_cv_data_type(in_color), outdata);
    resize(rgb, out(r), r.size(), 0, 0, algo);
}

References get_bits_per_component(), get_out_cv_data_type(), RG48, RGB, ug_to_rgb_mat(), and vc_get_linesize().

Referenced by resize_frame().

Here is the call graph for this function:

ug_to_rgb_mat()

static Mat ug_to_rgb_mat ( codec_t codec, int width, int height, char * indata )

static

Definition at line 85 of file resize_utils.cpp.

                                                                             {
    Mat yuv;
    Mat rgb;
    int pix_fmt = CV_8UC2;
    int cv_color = 0;
    int num = 1;
    int den = 1;
 
    switch (codec) {
    case RG48:
        rgb.create(height, width, CV_16UC3);
        rgb.data = (uchar*)indata;
        return rgb;
    case RGB:
        rgb.create(height, width, CV_8UC3);
        rgb.data = (uchar*)indata;
        return rgb;
    case RGBA:
        cv_color = CV_RGBA2RGB;
        pix_fmt = CV_8UC4;
        break;
    case I420:
        pix_fmt = CV_8U;
        num = 3;
        den = 2;
        cv_color = CV_YUV2RGB_I420;
        break;
    case UYVY:
        cv_color = CV_YUV2RGB_UYVY;
        break;
    case YUYV:
        cv_color = CV_YUV2RGB_YUYV;
        break;
    default:
        LOG(LOG_LEVEL_ERROR) << MOD_NAME "Unsupported codec: " << codec << "\n";
        abort();
    }
    yuv.create(height * num / den, width, pix_fmt);
    yuv.data = (uchar*)indata;
    cvtColor(yuv, rgb, cv_color);
 
    return rgb;
}

References I420, LOG, LOG_LEVEL_ERROR, MOD_NAME, RG48, RGB, RGBA, UYVY, and YUYV.

Referenced by resize_frame(), and resize_frame_dimensions().

Variable Documentation

[struct]

const struct { ... } interp_map[]

Initial value:

= {
        {INTER_NEAREST,        "nearest"      },
        { INTER_LINEAR,        "linear"       },
        { INTER_CUBIC,         "cubic"        },
        { INTER_AREA,          "area"         },
        { INTER_LANCZOS4,      "lanczos4"     },
        
        
}

Referenced by resize_algo_from_string(), and resize_algo_to_string().

name

const char* name

Definition at line 212 of file resize_utils.cpp.

val

int val

Definition at line 211 of file resize_utils.cpp.