1 files changed, 15 insertions, 1 deletions
diff --git a/lib/libdv_wrapper.h b/lib/libdv_wrapper.h
index d98cae6..16a7d94 100644
--- a/lib/libdv_wrapper.h
+++ b/lib/libdv_wrapper.h
@@ -78,6 +78,19 @@ namespace DV {
     YUV_420 = e_dv_sample_420,
     YUV_422 = e_dv_sample_422
   } Sampling;
+
+  /*
+    typedef enum color_space_e { 
+    e_dv_color_yuv, 
+    e_dv_color_rgb, 
+    e_dv_color_bgr0, 
+    } dv_color_space_t;
+  */
+  typedef enum {
+    YUV = e_dv_color_yuv,
+    RGB = e_dv_color_rgb,
+    BGR0 = e_dv_color_bgr0
+  } ColorSpace;
 };
 
 class LibDVWrapper {
@@ -91,7 +104,7 @@ public:
   void setSystem(DV::System system);
   void setSampling(DV::Sampling sampling);
 
-  void setOutputBuffer(char *output);
+  void setOutputBuffer(char *output, DV::ColorSpace colorspace = DV::YUV);
 
   void decode(char *input);
 
@@ -103,6 +116,7 @@ private:
   unsigned char* yuv[3];
 
   dv_decoder_t *decoder;
+  DV::ColorSpace colorspace;
 };
 
 #endif/*__MIAV_LIBDV_WRAPPER_H__*/