Add XYZ12LE to vapoursynth

HomeOfAviSynthPlusEvolution · Feb 28, 2024 · c6543b9 · c6543b9 · Asd-g · Feb 28, 2024
1 parent e24e13f
commit c6543b9
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/VapourSynth/lwlibav_source.c b/VapourSynth/lwlibav_source.c
@@ -290,6 +290,22 @@ static const VSFrameRef *VS_CC vs_filter_get_frame( int n, int activation_reason
         bottom = ( vohp->frame_order_list[n].bottom == vohp->frame_order_list[frame_number].bottom ) ? vohp->frame_order_list[n - 1].bottom :
             vohp->frame_order_list[n].bottom;
     }
+    if ( vohp->scaler.output_pixel_format == AV_PIX_FMT_XYZ12LE )
+    {
+        const int pitch = vsapi->getStride(vs_frame, output_index) / 2;
+        uint16_t* as_frame_ptr = (uint16_t*)(vsapi->getWritePtr(vs_frame, output_index));
+        for ( int y = 0; y < vsapi->getFrameHeight(vs_frame, output_index); ++y )
+        {
+            for ( int x = 0; x < pitch; x += 3 )
+            {
+                const uint16_t temp = as_frame_ptr[x];
+                as_frame_ptr[x] = as_frame_ptr[x + 2];
+                as_frame_ptr[x + 2] = temp;
+            }
+
+            as_frame_ptr += pitch;
+        }
+    }
     set_frame_properties( vi, av_frame, vdhp->format->streams[vdhp->stream_index], vs_frame, top, bottom, vsapi, n );
     return vs_frame;
 }

diff --git a/VapourSynth/video_output.c b/VapourSynth/video_output.c
@@ -441,6 +441,7 @@ static const component_reorder_t *get_component_reorder( enum AVPixelFormat av_o
             { AV_PIX_FMT_RGBA64LE,     {  0,  1,  2,  3 } },
             { AV_PIX_FMT_BGRA64LE,     {  2,  1,  0,  3 } },
             { AV_PIX_FMT_RGBA64BE,     {  0,  1,  2,  3 | component_reorder_bigendian } },
+            { AV_PIX_FMT_XYZ12LE,      {  0,  1,  2, -1 } },
             { AV_PIX_FMT_NONE,         {  0,  1,  2,  3 } }
         };
     int i = 0;
@@ -649,6 +650,7 @@ static int determine_colorspace_conversion
             { AV_PIX_FMT_RGBA64LE,     pfRGB48,     1 },
             { AV_PIX_FMT_BGRA64LE,     pfRGB48,     1 },
             { AV_PIX_FMT_RGBA64BE,     pfRGB48,     1 },
+            { AV_PIX_FMT_XYZ12LE,      pfRGB48,     1 },
             { AV_PIX_FMT_NONE,         pfNone,      1 }
         };
     if( vs_vohp->variable_info || vs_vohp->vs_output_pixel_format == pfNone )