diff --git a/base/include/DMAAllocator.h b/base/include/DMAAllocator.h index 0675cdf07..660c2a9a5 100755 --- a/base/include/DMAAllocator.h +++ b/base/include/DMAAllocator.h @@ -26,6 +26,9 @@ class DMAAllocator : public HostAllocator case ImageMetadata::UYVY: colorFormat = NvBufferColorFormat_UYVY; break; + case ImageMetadata::YUYV: + colorFormat = NvBufferColorFormat_YUYV; + break; case ImageMetadata::RGBA: colorFormat = NvBufferColorFormat_ABGR32; break; @@ -131,6 +134,9 @@ class DMAAllocator : public HostAllocator case ImageMetadata::ImageType::UYVY: type = CV_8UC3; break; + case ImageMetadata::ImageType::YUYV: + type = CV_8UC3; + break; default: throw AIPException(AIP_FATAL, "Only Image Type accepted are UYVY or ARGB found " + std::to_string(imageType)); } diff --git a/base/include/ImageMetadata.h b/base/include/ImageMetadata.h index 655d7c33d..12440f74a 100644 --- a/base/include/ImageMetadata.h +++ b/base/include/ImageMetadata.h @@ -24,7 +24,8 @@ class ImageMetadata BAYERBG8, // BGGR BAYERGB8, // GBRG BAYERGR8, // GRBG - BAYERRG8 // RGGB + BAYERRG8, // RGGB + YUYV }; static size_t getElemSize(int depth) diff --git a/base/include/RawImageMetadata.h b/base/include/RawImageMetadata.h index a4ae08b41..f6069af3e 100644 --- a/base/include/RawImageMetadata.h +++ b/base/include/RawImageMetadata.h @@ -51,6 +51,9 @@ class RawImageMetadata : public FrameMetadata case ImageMetadata::UYVY: _step = _width * 2; break; + case ImageMetadata::YUYV: + _step = _width * 2; + break; case ImageMetadata::BGR: case ImageMetadata::RGB: _step = _width * 3; @@ -139,6 +142,9 @@ class RawImageMetadata : public FrameMetadata case ImageMetadata::MONO: multiple = 1; break; + case ImageMetadata::YUYV: + multiple = 2; + break; case ImageMetadata::UYVY: multiple = 2; break; @@ -206,6 +212,7 @@ class RawImageMetadata : public FrameMetadata case ImageMetadata::MONO: channels = 1; break; + case ImageMetadata::YUYV: case ImageMetadata::UYVY: case ImageMetadata::BGR: case ImageMetadata::RGB: diff --git a/base/src/DMAFDToHostCopy.cpp b/base/src/DMAFDToHostCopy.cpp index 5a6bcd7ad..3b6ec671b 100644 --- a/base/src/DMAFDToHostCopy.cpp +++ b/base/src/DMAFDToHostCopy.cpp @@ -157,6 +157,8 @@ bool DMAFDToHostCopy::processSOS(frame_sp &frame) { case ImageMetadata::ImageType::RGBA: case ImageMetadata::ImageType::BGRA: + case ImageMetadata::ImageType::YUYV: + break; case ImageMetadata::ImageType::UYVY: break; case ImageMetadata::ImageType::YUV420: