Added YUYV Metadata

Apra-Labs · Nov 12, 2022 · bb491c9 · bb491c9
1 parent 7a410d9
commit bb491c9
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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
@@ -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: