RawImage CaptureFlycap::getFrame() {
mutex.lock();
Error error;
Image image;
RawImage result;
result.setColorFormat(capture_format);
result.setWidth(0);
result.setHeight(0);
result.setTime(0.0);
result.setData(0);
error = camera->RetrieveBuffer(&image);
if (error != PGRERROR_OK) {
error.PrintErrorTrace();
mutex.unlock();
return result;
}
stored_image = new Image;
stored_image->DeepCopy(&image);
result.setData(stored_image->GetData());
struct timeval tv;
gettimeofday(&tv,NULL);
result.setTime((double)tv.tv_sec + tv.tv_usec*(1.0E-6));
result.setHeight(stored_image->GetRows());
result.setWidth(stored_image->GetCols());
result.setColorFormat(pixelFormatToColorFormat(stored_image->GetPixelFormat()));
mutex.unlock();
return result;
}
RawImage CaptureFromFile::getFrame()
{
#ifndef VDATA_NO_QT
mutex.lock();
#endif
RawImage result;
result.setColorFormat(COLOR_RGB8);
result.setTime(0.0);
rgba* rgba_img = 0;
int width;
int height;
if(images.size())
{
rgba_img = images[currentImageIndex];
width = widths[currentImageIndex];
height = heights[currentImageIndex];
currentImageIndex = (currentImageIndex + 1) % images.size();
}
if (rgba_img == 0)
{
fprintf (stderr, "CaptureFromFile Error, no images available");
is_capturing=false;
result.setData(0);
result.setWidth(640);
result.setHeight(480);
frame = 0;
}
else
{
frame = new unsigned char[width*height*3];
unsigned char* p = &frame[0];
for (int i=0; i < width * height; i++)
{
*p = rgba_img[i].r;
p++;
*p = rgba_img[i].g;
p++;
*p = rgba_img[i].b;
p++;
}
result.setWidth(width);
result.setHeight(height);
result.setData(frame);
tval tv;
gettimeofday(&tv,0);
result.setTime((double)tv.tv_sec + tv.tv_usec*(1.0E-6));
}
#ifndef VDATA_NO_QT
mutex.unlock();
#endif
return result;
}
bool CaptureFromFile::copyAndConvertFrame(const RawImage & src, RawImage & target)
{
#ifndef VDATA_NO_QT
mutex.lock();
#endif
ColorFormat output_fmt = Colors::stringToColorFormat(v_colorout->getSelection().c_str());
ColorFormat src_fmt=src.getColorFormat();
if (target.getData()==0)
target.allocate(output_fmt, src.getWidth(), src.getHeight());
else
target.ensure_allocation(output_fmt, src.getWidth(), src.getHeight());
target.setTime(src.getTime());
if (output_fmt == src_fmt)
{
if (src.getData() != 0)
memcpy(target.getData(),src.getData(),src.getNumBytes());
}
else if (src_fmt == COLOR_RGB8 && output_fmt == COLOR_YUV422_UYVY)
{
if (src.getData() != 0)
dc1394_convert_to_YUV422(src.getData(), target.getData(), src.getWidth(), src.getHeight(),
DC1394_BYTE_ORDER_UYVY, DC1394_COLOR_CODING_RGB8, 8);
}
else if (src_fmt == COLOR_YUV422_UYVY && output_fmt == COLOR_RGB8)
{
if (src.getData() != 0)
dc1394_convert_to_RGB8(src.getData(),target.getData(), src.getWidth(), src.getHeight(),
DC1394_BYTE_ORDER_UYVY, DC1394_COLOR_CODING_YUV422, 8);
}
else
{
fprintf(stderr,"Cannot copy and convert frame...unknown conversion selected from: %s to %s\n",
Colors::colorFormatToString(src_fmt).c_str(),
Colors::colorFormatToString(output_fmt).c_str());
#ifndef VDATA_NO_QT
mutex.unlock();
#endif
return false;
}
#ifndef VDATA_NO_QT
mutex.unlock();
#endif
return true;
}
bool CaptureFlycap::convertFrame(const RawImage & src,
RawImage & target,
ColorFormat output_fmt,
bool debayer,
dc1394color_filter_t bayer_format,
dc1394bayer_method_t bayer_method,
int y16bits) {
mutex.lock();
int width = v_width->getInt();
int height = v_height->getInt();
ColorFormat src_fmt=src.getColorFormat();
if (target.getData()==0) {
//allocate target, if it does not exist yet
target.allocate(output_fmt,src.getWidth(),src.getHeight());
} else {
target.ensure_allocation(output_fmt,src.getWidth(),src.getHeight());
}
target.setTime(src.getTime());
if (output_fmt==src_fmt) {
//just do a memcpy
memcpy(target.getData(),src.getData(),src.getNumBytes());
} else {
//do some more fancy conversion
if ((src_fmt==COLOR_MONO8 || src_fmt==COLOR_RAW8) && output_fmt==COLOR_RGB8) {
//check whether to debayer or simply average to a grey rgb image
if (debayer) {
//de-bayer
if ( dc1394_bayer_decoding_8bit( src.getData(), target.getData(), src.getWidth(), src.getHeight(), bayer_format, bayer_method) != DC1394_SUCCESS ) {
mutex.unlock();
return false;
}
} else {
dc1394_convert_to_RGB8(src.getData(),target.getData(), width, height, 0,
DC1394_COLOR_CODING_MONO8, 8);
//Conversions::y2rgb (src.getData(), target.getData(), src.getNumPixels());
}
} else if ((src_fmt==COLOR_MONO16 || src_fmt==COLOR_RAW16)) {
//check whether to debayer or simply average to a grey rgb image
if (debayer && output_fmt==COLOR_RGB16) {
//de-bayer
if ( dc1394_bayer_decoding_16bit( (uint16_t *)src.getData(), (uint16_t *)target.getData(), src.getWidth(), src.getHeight(), bayer_format, bayer_method, y16bits) != DC1394_SUCCESS ) {
fprintf(stderr,"Error in 16bit Bayer Conversion");
mutex.unlock();
return false;
}
} else if (debayer==false && output_fmt==COLOR_RGB8) {
dc1394_convert_to_RGB8(src.getData(),target.getData(), width, height, 0,
((src_fmt==COLOR_MONO16) ? DC1394_COLOR_CODING_MONO16 : DC1394_COLOR_CODING_RAW16), y16bits);
//Conversions::y162rgb (src.getData(), target.getData(), src.getNumPixels(), y16bits);
} else {
fprintf(stderr,"Cannot copy and convert frame...unknown conversion selected from: %s to %s\n",Colors::colorFormatToString(src_fmt).c_str(),Colors::colorFormatToString(output_fmt).c_str());
mutex.unlock();
return false;
}
} else if (src_fmt==COLOR_YUV411 && output_fmt==COLOR_RGB8) {
dc1394_convert_to_RGB8(src.getData(),target.getData(), width, height, 0,
DC1394_COLOR_CODING_YUV411, 8);
//Conversions::uyyvyy2rgb (src.getData(), target.getData(), src.getNumPixels());
} else if (src_fmt==COLOR_YUV422_UYVY && output_fmt==COLOR_RGB8) {
dc1394_convert_to_RGB8(src.getData(),target.getData(), width, height, DC1394_BYTE_ORDER_UYVY,
DC1394_COLOR_CODING_YUV422, 8);
} else if (src_fmt==COLOR_YUV422_YUYV && output_fmt==COLOR_RGB8) {
dc1394_convert_to_RGB8(src.getData(),target.getData(), width, height, DC1394_BYTE_ORDER_YUYV,
DC1394_COLOR_CODING_YUV422, 8);
} else if (src_fmt==COLOR_YUV444 && output_fmt==COLOR_RGB8) {
dc1394_convert_to_RGB8(src.getData(),target.getData(), width, height, 0,
DC1394_COLOR_CODING_YUV444, 8);
} else {
fprintf(stderr,"Cannot copy and convert frame...unknown conversion selected from: %s to %s\n",Colors::colorFormatToString(src_fmt).c_str(),Colors::colorFormatToString(output_fmt).c_str());
mutex.unlock();
return false;
}
}
mutex.unlock();
return true;
}