void color_sycc_to_rgb(opj_image_t *img) { if(img->numcomps < 3) { img->color_space = OPJ_CLRSPC_GRAY; return; } if((img->comps[0].dx == 1) && (img->comps[1].dx == 2) && (img->comps[2].dx == 2) && (img->comps[0].dy == 1) && (img->comps[1].dy == 2) && (img->comps[2].dy == 2))/* horizontal and vertical sub-sample */ { sycc420_to_rgb(img); } else if((img->comps[0].dx == 1) && (img->comps[1].dx == 2) && (img->comps[2].dx == 2) && (img->comps[0].dy == 1) && (img->comps[1].dy == 1) && (img->comps[2].dy == 1))/* horizontal sub-sample only */ { sycc422_to_rgb(img); } else if((img->comps[0].dx == 1) && (img->comps[1].dx == 1) && (img->comps[2].dx == 1) && (img->comps[0].dy == 1) && (img->comps[1].dy == 1) && (img->comps[2].dy == 1))/* no sub-sample */ { sycc444_to_rgb(img); } else { fprintf(stderr,"%s:%d:color_sycc_to_rgb\n\tCAN NOT CONVERT\n", __FILE__,__LINE__); return; } img->color_space = OPJ_CLRSPC_SRGB; }/* color_sycc_to_rgb() */
void color_sycc_to_rgb(opj_image_t* img) { if (img->numcomps < 3) { img->color_space = OPJ_CLRSPC_GRAY; return; } if ((img->comps[0].dx == 1) && (img->comps[1].dx == 2) && (img->comps[2].dx == 2) && (img->comps[0].dy == 1) && (img->comps[1].dy == 2) && (img->comps[2].dy == 2)) { sycc420_to_rgb(img); } else if ((img->comps[0].dx == 1) && (img->comps[1].dx == 2) && (img->comps[2].dx == 2) && (img->comps[0].dy == 1) && (img->comps[1].dy == 1) && (img->comps[2].dy == 1)) { sycc422_to_rgb(img); } else if ((img->comps[0].dx == 1) && (img->comps[1].dx == 1) && (img->comps[2].dx == 1) && (img->comps[0].dy == 1) && (img->comps[1].dy == 1) && (img->comps[2].dy == 1)) { sycc444_to_rgb(img); } else { return; } img->color_space = OPJ_CLRSPC_SRGB; }
TEST(fxcodec, YUV420ToRGB) { opj_image_comp_t u; memset(&u, 0, sizeof(u)); u.dx = 1; u.dy = 1; u.w = 16; u.h = 16; u.prec = 8; u.bpp = 8; opj_image_comp_t v; memset(&v, 0, sizeof(v)); v.dx = 1; v.dy = 1; v.w = 16; v.h = 16; v.prec = 8; v.bpp = 8; opj_image_comp_t y; memset(&y, 0, sizeof(y)); y.dx = 1; y.dy = 1; y.prec = 8; y.bpp = 8; opj_image_t img; memset(&img, 0, sizeof(img)); img.numcomps = 3; img.color_space = OPJ_CLRSPC_SYCC; img.comps = FX_Alloc(opj_image_comp_t, 3); const struct { OPJ_UINT32 w; bool expected; } cases[] = {{0, false}, {1, false}, {30, false}, {31, true}, {32, true}, {33, false}, {34, false}, {UINT_MAX, false}}; for (size_t i = 0; i < sizeof(cases) / sizeof(cases[0]); ++i) { y.w = cases[i].w; y.h = y.w; img.x1 = y.w; img.y1 = y.h; y.data = FX_Alloc(OPJ_INT32, y.w * y.h); memset(y.data, 1, y.w * y.h * sizeof(OPJ_INT32)); u.data = FX_Alloc(OPJ_INT32, u.w * u.h); memset(u.data, 0, u.w * u.h * sizeof(OPJ_INT32)); v.data = FX_Alloc(OPJ_INT32, v.w * v.h); memset(v.data, 0, v.w * v.h * sizeof(OPJ_INT32)); img.comps[0] = y; img.comps[1] = u; img.comps[2] = v; sycc420_to_rgb(&img); if (cases[i].expected) { EXPECT_EQ(img.comps[0].w, img.comps[1].w); EXPECT_EQ(img.comps[0].h, img.comps[1].h); EXPECT_EQ(img.comps[0].w, img.comps[2].w); EXPECT_EQ(img.comps[0].h, img.comps[2].h); } else { EXPECT_NE(img.comps[0].w, img.comps[1].w); EXPECT_NE(img.comps[0].h, img.comps[1].h); EXPECT_NE(img.comps[0].w, img.comps[2].w); EXPECT_NE(img.comps[0].h, img.comps[2].h); } FX_Free(img.comps[0].data); FX_Free(img.comps[1].data); FX_Free(img.comps[2].data); } FX_Free(img.comps); }