Manipulation of image conversions.
#include <iomanip>
#include <stdlib.h>
#include <visp3/core/vpConfig.h>
#include <visp3/core/vpDebug.h>
#include <visp3/core/vpImage.h>
#include <visp3/core/vpImageConvert.h>
#include <visp3/core/vpIoTools.h>
#include <visp3/core/vpTime.h>
#include <visp3/io/vpImageIo.h>
#include <visp3/io/vpParseArgv.h>
#if defined(VISP_HAVE_OPENCV) && defined(HAVE_OPENCV_IMGCODECS) && defined(HAVE_OPENCV_IMGPROC)
#include <opencv2/imgcodecs.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#endif
#if defined(VISP_HAVE_YARP)
#include <yarp/sig/ImageFile.h>
#endif
#ifdef ENABLE_VISP_NAMESPACE
#endif
#define GETOPTARGS "cdi:o:n:h"
void usage(const char *name, const char *badparam, std::string ipath, std::string opath, std::string user, int nbiter)
{
fprintf(stdout, "\n\
Test image conversions.\n\
\n\
SYNOPSIS\n\
%s [-i <input image path>] [-o <output image path>] [-n <nb benchmark iterations>]\n\
[-h]\n\
",
name);
fprintf(stdout, "\n\
OPTIONS: Default\n\
-i <input image path> %s\n\
Set image input path.\n\
From this path read \"Klimt/Klimt.pgm\"\n\
and \"Klimt/Klimt.ppm\" images.\n\
Setting the VISP_INPUT_IMAGE_PATH environment\n\
variable produces the same behaviour than using\n\
this option.\n\
\n\
-o <output image path> %s\n\
Set image output path.\n\
From this directory, creates the \"%s\"\n\
subdirectory depending on the username, where \n\
Klimt_grey.pgm and Klimt_color.ppm output images\n\
are written.\n\
\n\
-n <nb benchmark iterations> %d\n\
Set the number of benchmark iterations.\n\
\n\
-h\n\
Print the help.\n\n",
ipath.c_str(), opath.c_str(), user.c_str(), nbiter);
if (badparam)
fprintf(stdout, "\nERROR: Bad parameter [%s]\n", badparam);
}
bool getOptions(int argc, const char **argv, std::string &ipath, std::string &opath, const std::string &user,
int &nbIterations)
{
const char *optarg_;
int c;
switch (c) {
case 'i':
ipath = optarg_;
break;
case 'o':
opath = optarg_;
break;
case 'n':
nbIterations = atoi(optarg_);
break;
case 'h':
usage(argv[0], nullptr, ipath, opath, user, nbIterations);
return false;
case 'c':
case 'd':
break;
default:
usage(argv[0], optarg_, ipath, opath, user, nbIterations);
return false;
}
}
if ((c == 1) || (c == -1)) {
usage(argv[0], nullptr, ipath, opath, user, nbIterations);
std::cerr << "ERROR: " << std::endl;
std::cerr << " Bad argument " << optarg_ << std::endl << std::endl;
return false;
}
return true;
}
int main(int argc, const char **argv)
{
try {
std::string env_ipath;
std::string opt_ipath;
std::string opt_opath;
std::string ipath;
std::string opath;
std::string filename;
std::string username;
int nbIterations = 1;
if (!env_ipath.empty())
ipath = env_ipath;
#if defined(_WIN32)
opt_opath = "C:/temp";
#else
opt_opath = "/tmp";
#endif
if (getOptions(argc, argv, opt_ipath, opt_opath, username, nbIterations) == false) {
return EXIT_FAILURE;
}
if (!opt_ipath.empty())
ipath = opt_ipath;
if (!opt_opath.empty())
opath = opt_opath;
try {
}
catch (...) {
usage(argv[0], nullptr, ipath, opt_opath, username, nbIterations);
std::cerr << std::endl << "ERROR:" << std::endl;
std::cerr << " Cannot create " << opath << std::endl;
std::cerr << " Check your -o " << opt_opath << " option " << std::endl;
return EXIT_FAILURE;
}
}
if (opt_ipath.empty()) {
if (ipath != env_ipath) {
std::cout << std::endl << "WARNING: " << std::endl;
std::cout << " Since -i <visp image path=" << ipath << "> "
<< " is different from VISP_IMAGE_PATH=" << env_ipath << std::endl
<< " we skip the environment variable." << std::endl;
}
}
if (opt_ipath.empty() && env_ipath.empty()) {
usage(argv[0], nullptr, ipath, opt_opath, username, nbIterations);
std::cerr << std::endl << "ERROR:" << std::endl;
std::cerr << " Use -i <visp image path> option or set VISP_INPUT_IMAGE_PATH " << std::endl
<< " environment variable to specify the location of the " << std::endl
<< " image path where test images are located." << std::endl
<< std::endl;
return EXIT_FAILURE;
}
std::cout << "** Convert a grey image (.pgm) to a color image (.ppm)" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "** Convert a color image (.ppm) to a grey image (.pgm)" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "** Convert YUV pixel value to a RGB value" << std::endl;
unsigned char y = 187, u = 10, v = 30;
unsigned char r, g, b;
std::cout << " y(" << (int)y << ") u(" << (int)u << ") v(" << (int)v << ") = r(" << (int)r << ") g(" << (int)g
<< ") b(" << (int)b << ")" << std::endl;
#if defined(VISP_HAVE_OPENCV) && defined(HAVE_OPENCV_IMGCODECS) && defined(HAVE_OPENCV_HIGHGUI) && defined(HAVE_OPENCV_IMGPROC)
std::cout << "** Convert a cv::Mat to a vpImage<vpRGBa>" << std::endl;
cv::Mat imageMat;
std::cout << " Reading the color image with c++ interface of opencv: " << filename << std::endl;
#if VISP_HAVE_OPENCV_VERSION >= 0x030000
int flags = cv::IMREAD_COLOR;
#else
int flags = CV_LOAD_IMAGE_COLOR;
#endif
imageMat = cv::imread(filename, flags);
if (imageMat.data == nullptr) {
std::cout << " Cannot read image: " << filename << std::endl;
return EXIT_FAILURE;
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Reading the grayscale image with opencv: " << filename << std::endl;
#if VISP_HAVE_OPENCV_VERSION >= 0x030000
flags = cv::IMREAD_GRAYSCALE;
#else
flags = CV_LOAD_IMAGE_GRAYSCALE;
#endif
imageMat = cv::imread(filename, flags);
if (imageMat.data == nullptr) {
std::cout << " Cannot read image: " << filename << std::endl;
return EXIT_FAILURE;
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "** Convert a cv::Mat to a vpImage<nsigned char>" << std::endl;
std::cout << " Reading the color image with opencv: " << filename << std::endl;
#if VISP_HAVE_OPENCV_VERSION >= 0x030000
flags = cv::IMREAD_COLOR;
#else
flags = CV_LOAD_IMAGE_COLOR;
#endif
imageMat = cv::imread(filename, flags);
if (imageMat.data == nullptr) {
std::cout << " Cannot read image: " << filename << std::endl;
return EXIT_FAILURE;
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Reading the greyscale image with opencv: " << filename << std::endl;
#if VISP_HAVE_OPENCV_VERSION >= 0x030000
flags = cv::IMREAD_GRAYSCALE;
#else
flags = CV_LOAD_IMAGE_GRAYSCALE;
#endif
imageMat = cv::imread(filename, flags);
if (imageMat.data == nullptr) {
std::cout << " Cannot read image: " << filename << std::endl;
return EXIT_FAILURE;
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << " Convert result in " << filename << std::endl;
std::cout << "** Convert a vpImage<vpRGBa> to a cv::Mat" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
if (!cv::imwrite(filename, imageMat)) {
std::cout << " Cannot write image: " << filename << std::endl;
return EXIT_FAILURE;
}
std::cout << " Convert result in " << filename << std::endl;
std::cout << "** Convert a vpImage<unsigned char> to a cv::Mat" << std::endl;
std::cout << " Load " << filename << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
if (!cv::imwrite(filename, imageMat)) {
std::cout << " Cannot write image: " << filename << std::endl;
return EXIT_FAILURE;
}
std::cout << " Convert result in " << filename << std::endl;
std::cout <<
"== Conversion c++ interface : " << chrono.
getDurationMs() <<
" ms" << std::endl;
#endif
std::cout << "** Split a vpImage<vpRGBa> to vpImage<unsigned char>" << std::endl;
std::cout << " Load " << filename << std::endl;
for (int iteration = 0; iteration < nbIterations; iteration++) {
}
std::cout <<
" Time for " << nbIterations <<
" split (ms): " << chrono.
getDurationMs() << std::endl;
std::cout << " Save Klimt R channel: " << filename << std::endl;
std::cout << " Save Klimt B channel: " << filename << std::endl;
std::cout << "** Merge 4 vpImage<unsigned char> (RGBa) to vpImage<vpRGBa>" << std::endl;
for (int iteration = 0; iteration < nbIterations; iteration++) {
}
std::cout <<
" Time for 1000 merge (ms): " << chrono.
getDurationMs() << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "** Convert a vpImage<vpRGBa> in RGB color space to a vpImage<vpRGBa> in HSV color" << std::endl;
std::vector<unsigned char> hue(size);
std::vector<unsigned char> saturation(size);
std::vector<unsigned char> value(size);
std::cout << " Resulting image saved in: " << filename << std::endl;
vpImageConvert::HSVToRGBa(&hue.front(), &saturation.front(), &value.front(),
reinterpret_cast<unsigned char *
>(Ic_from_hsv.bitmap), size);
for (
unsigned int i = 0; i < Ic.
getHeight(); i++) {
for (
unsigned int j = 0; j < Ic.
getWidth(); j++) {
double precision = 10.;
if ((!
vpMath::equal(
static_cast<double>(Ic[i][j].R),
static_cast<double>(Ic_from_hsv[i][j].R), precision))
|| (!
vpMath::equal(
static_cast<double>(Ic[i][j].G),
static_cast<double>(Ic_from_hsv[i][j].G), precision))
|| (!
vpMath::equal(
static_cast<double>(Ic[i][j].B),
static_cast<double>(Ic_from_hsv[i][j].B), precision))) {
std::cerr << "Ic[i][j].R=" << static_cast<unsigned int>(Ic[i][j].R)
<< " ; Ic_from_hsv[i][j].R=" << static_cast<unsigned int>(Ic_from_hsv[i][j].R) << " precision: " << precision << std::endl;
std::cerr << "Ic[i][j].G=" << static_cast<unsigned int>(Ic[i][j].G)
<< " ; Ic_from_hsv[i][j].G=" << static_cast<unsigned int>(Ic_from_hsv[i][j].G) << " precision: " << precision << std::endl;
std::cerr << "Ic[i][j].B=" << static_cast<unsigned int>(Ic[i][j].B)
<< " ; Ic_from_hsv[i][j].B=" << static_cast<unsigned int>(Ic_from_hsv[i][j].B) << " precision: " << precision << std::endl;
}
}
}
std::vector<double> hue2(size);
std::vector<double> saturation2(size);
std::vector<double> value2(size);
std::vector<unsigned char> rgba(size * 4);
std::cout << " Resulting image saved in: " << filename << std::endl;
for (
unsigned int i = 0; i < Ic.
getHeight(); i++) {
for (
unsigned int j = 0; j < Ic.
getWidth(); j++) {
if (Ic[i][j].R != I_HSV2RGBa[i][j].R || Ic[i][j].G != I_HSV2RGBa[i][j].G || Ic[i][j].B != I_HSV2RGBa[i][j].B) {
std::cerr << "Ic[i][j].R=" << static_cast<unsigned>(Ic[i][j].R)
<< " ; I_HSV2RGBa[i][j].R=" << static_cast<unsigned>(I_HSV2RGBa[i][j].R) << std::endl;
std::cerr << "Ic[i][j].G=" << static_cast<unsigned>(Ic[i][j].G)
<< " ; I_HSV2RGBa[i][j].G=" << static_cast<unsigned>(I_HSV2RGBa[i][j].G) << std::endl;
std::cerr << "Ic[i][j].B=" << static_cast<unsigned>(Ic[i][j].B)
<< " ; I_HSV2RGBa[i][j].B=" << static_cast<unsigned>(I_HSV2RGBa[i][j].B) << std::endl;
}
}
}
std::cout << "** Construction of a vpImage from an array with copyData==true" << std::endl;
std::vector<unsigned char> rgba2(size * 4);
std::fill(rgba2.begin(), rgba2.end(), 127);
std::cout << " Resulting image saved in: " << filename << std::endl;
if (I_copyData.getSize() > 0) {
I_copyData[0][0].R = 10;
}
{
std::cout << "** Test color conversion" << std::endl;
std::vector<unsigned char> rgb_array(I_color.
getSize() * 3);
#if defined(VISP_HAVE_OPENCV) && defined(HAVE_OPENCV_IMGCODECS) && defined(HAVE_OPENCV_HIGHGUI) && defined(HAVE_OPENCV_IMGPROC)
std::cout << "\n BGR cv::Mat to Grayscale" << std::endl;
cv::Mat colorMat = cv::imread(filename);
std::cout << " colorMat=" << colorMat.cols << "x" << colorMat.rows << std::endl;
std::cout << "\n RGB to Grayscale + Flip" << std::endl;
std::vector<unsigned char> rgb2gray_flip_array_sse(I_color.
getSize());
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n Conversion BGR to Grayscale + Flip" << std::endl;
std::vector<unsigned char> bgr2gray_flip_array_sse(I_color.
getSize());
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n RGB to Grayscale + Flip + Crop" << std::endl;
cv::Rect rect_roi(11, 17, 347, 449);
cv::Mat colorMat_crop = colorMat(rect_roi);
cv::Mat colorMat_crop_continuous = colorMat(rect_roi).clone();
std::cout << " colorMat_crop: " << colorMat_crop.cols << "x" << colorMat_crop.rows << " is continuous? "
<< colorMat_crop.isContinuous() << std::endl;
std::cout << " colorMat_crop_continuous: " << colorMat_crop_continuous.cols << "x" << colorMat_crop_continuous.rows
<< " is continuous? " << colorMat_crop_continuous.isContinuous() << std::endl;
(unsigned int)(rect_roi.width - rect_roi.x));
for (unsigned int i = (unsigned int)rect_roi.y; i < (unsigned int)rect_roi.height; i++) {
for (unsigned int j = (unsigned int)rect_roi.x; j < (unsigned int)rect_roi.width; j++) {
I_color_crop[(unsigned int)((int)i - rect_roi.y)][(unsigned int)((int)j - rect_roi.x)] = I_color[i][j];
}
}
std::cout << " Resulting image saved in: " << filename << std::endl;
std::vector<unsigned char> rgb_array_crop(I_color_crop.getSize() * 3);
std::vector<unsigned char> rgb2gray_flip_crop_array_sse(I_color_crop.getSize());
I_color_crop.getHeight(), true);
vpImage<unsigned char> I_rgb2gray_flip_crop_sse(&rgb2gray_flip_crop_array_sse.front(), I_color_crop.getHeight(),
I_color_crop.getWidth());
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n BGR to Grayscale + Flip + Crop" << std::endl;
std::cout << " Resulting image saved in: " << filename << std::endl;
std::cout << "\n BGR to Grayscale + Flip + Crop + No continuous Mat" << std::endl;
vpImage<unsigned char> I_bgr2gray_flip_crop_no_continuous_sse(I_color_crop.getHeight(), I_color_crop.getWidth());
std::cout << " Resulting image saved in: " << filename << std::endl;
#endif
std::cout << " Test succeed" << std::endl;
}
#if defined(VISP_HAVE_YARP)
std::cout << "** Test ViSP to Yarp image conversion by copy" << std::endl;
{
bool convert_by_copy = true;
std::cout << " Reading the gray image with ViSP: " << filename << std::endl;
yarp::sig::ImageOf< yarp::sig::PixelMono > *Iyarp = new yarp::sig::ImageOf<yarp::sig::PixelMono >();
std::cout << " Converted Yarp image saved in: " << filename << std::endl;
yarp::sig::file::write(*Iyarp, filename, yarp::sig::file::FORMAT_PGM);
std::cout << " Reading the gray image with Yarp: " << filename << std::endl;
yarp::sig::ImageOf< yarp::sig::PixelMono > *IIyarp = new yarp::sig::ImageOf<yarp::sig::PixelMono >();
yarp::sig::file::read(*IIyarp, filename, yarp::sig::file::FORMAT_PGM);
std::cout << " Converted image in ViSP saved in: " << filename << std::endl;
if (I != II) {
std::cout << " Yarp gray conversion test failed" << std::endl;
return EXIT_FAILURE;
}
std::cout << std::endl;
}
{
bool convert_by_copy = true;
std::cout << " Reading the color image with ViSP: " << filename << std::endl;
yarp::sig::ImageOf< yarp::sig::PixelRgba > *Iyarp = new yarp::sig::ImageOf<yarp::sig::PixelRgba >();
std::cout << " Converted image saved in: " << filename << std::endl;
yarp::sig::file::write(*Iyarp, filename, yarp::sig::file::FORMAT_PPM);
std::cout << " Reading the color image with Yarp: " << filename << std::endl;
yarp::sig::ImageOf< yarp::sig::PixelRgba > *IIyarp = new yarp::sig::ImageOf<yarp::sig::PixelRgba >();
yarp::sig::file::read(*IIyarp, filename, yarp::sig::file::FORMAT_PPM);
std::cout << " Converted image in ViSP saved in: " << filename << std::endl;
if (I != II) {
std::cout << " Yarp color conversion test failed" << std::endl;
return EXIT_FAILURE;
}
std::cout << std::endl;
}
std::cout << "** Test ViSP to Yarp image conversion without copy" << std::endl;
{
bool convert_by_copy = false;
std::cout << " Reading the gray image with ViSP: " << filename << std::endl;
yarp::sig::ImageOf< yarp::sig::PixelMono > *Iyarp = new yarp::sig::ImageOf<yarp::sig::PixelMono >();
std::cout << " Converted Yarp image saved in: " << filename << std::endl;
yarp::sig::file::write(*Iyarp, filename, yarp::sig::file::FORMAT_PGM);
std::cout << " Reading the gray image with Yarp: " << filename << std::endl;
yarp::sig::ImageOf< yarp::sig::PixelMono > *IIyarp = new yarp::sig::ImageOf<yarp::sig::PixelMono >();
yarp::sig::file::read(*IIyarp, filename, yarp::sig::file::FORMAT_PGM);
std::cout << " Converted image in ViSP saved in: " << filename << std::endl;
if (I != II) {
std::cout << " Yarp gray conversion test failed" << std::endl;
return EXIT_FAILURE;
}
else {
std::cout << " Yarp gray conversion test succeed" << std::endl;
}
std::cout << std::endl;
}
{
bool convert_by_copy = false;
std::cout << " Reading the color image with ViSP: " << filename << std::endl;
yarp::sig::ImageOf< yarp::sig::PixelRgba > *Iyarp = new yarp::sig::ImageOf<yarp::sig::PixelRgba >();
std::cout << " Converted image saved in: " << filename << std::endl;
yarp::sig::file::write(*Iyarp, filename, yarp::sig::file::FORMAT_PPM);
std::cout << " Reading the color image with Yarp: " << filename << std::endl;
yarp::sig::ImageOf< yarp::sig::PixelRgba > *IIyarp = new yarp::sig::ImageOf<yarp::sig::PixelRgba >();
yarp::sig::file::read(*IIyarp, filename, yarp::sig::file::FORMAT_PPM);
std::cout << " Converted image in ViSP saved in: " << filename << std::endl;
if (I != II) {
std::cout << " Yarp RGBa color conversion test failed" << std::endl;
return EXIT_FAILURE;
}
else {
std::cout << " Yarp RGBa color conversion test succeed" << std::endl;
}
yarp::sig::ImageOf< yarp::sig::PixelRgb > *IIIyarp = new yarp::sig::ImageOf<yarp::sig::PixelRgb >();
std::cout << " Converted RGB image saved in: " << filename << std::endl;
yarp::sig::file::write(*Iyarp, filename, yarp::sig::file::FORMAT_PPM);
std::cout << " Reading the RGB color image with Yarp: " << filename << std::endl;
yarp::sig::file::read(*IIIyarp, filename, yarp::sig::file::FORMAT_PPM);
std::cout << " Converted RGB image in ViSP saved in: " << filename << std::endl;
if (I != III) {
std::cout << " Yarp RGB color conversion test failed" << std::endl;
return EXIT_FAILURE;
}
else {
std::cout << " Yarp RGB color conversion test succeed" << std::endl;
}
std::cout << std::endl;
}
#endif
std::cout << "** All the tests succeed" << std::endl;
return EXIT_SUCCESS;
}
std::cout <<
"Catch an exception: " << e.
getMessage() << std::endl;
return EXIT_FAILURE;
}
}
void start(bool reset=true)
error that can be emitted by ViSP classes.
const char * getMessage() const
static void HSVToRGBa(const double *hue, const double *saturation, const double *value, unsigned char *rgba, unsigned int size)
static void merge(const vpImage< unsigned char > *R, const vpImage< unsigned char > *G, const vpImage< unsigned char > *B, const vpImage< unsigned char > *a, vpImage< vpRGBa > &RGBa)
static void YUVToRGB(unsigned char y, unsigned char u, unsigned char v, unsigned char &r, unsigned char &g, unsigned char &b)
static void split(const vpImage< vpRGBa > &src, vpImage< unsigned char > *pR, vpImage< unsigned char > *pG, vpImage< unsigned char > *pB, vpImage< unsigned char > *pa=nullptr)
static void RGBaToHSV(const unsigned char *rgba, double *hue, double *saturation, double *value, unsigned int size)
static void convert(const vpImage< unsigned char > &src, vpImage< vpRGBa > &dest)
static void RGBToGrey(unsigned char *rgb, unsigned char *grey, unsigned int width, unsigned int height, bool flip=false)
static void RGBaToRGB(unsigned char *rgba, unsigned char *rgb, unsigned int size)
static void read(vpImage< unsigned char > &I, const std::string &filename, int backend=IO_DEFAULT_BACKEND)
static void write(const vpImage< unsigned char > &I, const std::string &filename, int backend=IO_DEFAULT_BACKEND)
unsigned int getWidth() const
unsigned int getSize() const
Type * bitmap
points toward the bitmap
unsigned int getHeight() const
static bool equal(double x, double y, double threshold=0.001)
static bool parse(int *argcPtr, const char **argv, vpArgvInfo *argTable, int flags)