Sep 4, 2020 · Sep 4, 2020
diff --git a/libuavs3e.c b/libuavs3e.c

 static void uavs3e_image_copy_pic(void *dst[4], int i_dst[4], unsigned char *const src[4], const int i_src[4],  enum AVPixelFormat pix_fmts, int width, int height)
 {
    __imgb_cpy_plane(src[0], dst[0], width,      height,      i_src[0], i_dst[0]);
 __imgb_cpy_plane(src[1], dst[1], width >> 1, height >> 1, i_src[1], i_dst[1]);
 __imgb_cpy_plane(src[2], dst[2], width >> 1, height >> 1, i_src[2], i_dst[2]);
    if(sizeof(pel)>1 && pix_fmts==AV_PIX_FMT_YUV420P){    //sizeof(pel)==2 when BIT_DEPTH == 10
        //Expand YUV420P to 2-byte
 for(int plane=0;plane<3;plane++){
            int planeheight=(plane==0)?height:height>>1;
            int planewidth=(plane==0)?width:width>>1;

            pel *dstpel = (pel *)dst[plane];
            int ind_src = 0;
            int ind_dst = 0;

            for (int y = 0; y < planeheight; y++) {
               for (int x = 0; x < planewidth; x++) {
                   dstpel[ind_dst + x] = src[plane][ind_src + x];    //byte to short
               }//for (int x = 0; x < planewidth; x++)
               ind_src += i_src[plane];
               ind_dst += (i_dst[plane]>>1);
            }//for (int y = 0; y < planeheight; y++)
        }//for(int plane=0;plane<3;plane++)
    }
    else{
        width=(sizeof(pel)>1)?width<<1:width;
        __imgb_cpy_plane(src[0], dst[0], width,      height,      i_src[0], i_dst[0]);
        __imgb_cpy_plane(src[1], dst[1], width >> 1, height >> 1, i_src[1], i_dst[1]);
        __imgb_cpy_plane(src[2], dst[2], width >> 1, height >> 1, i_src[2], i_dst[2]);
    }
 }

 static int uavs3e_encode_frame(AVCodecContext *avctx, AVPacket *pkt,
Original file line number	Diff line number	Diff line change
Expand Up		@@ -153,9 +153,31 @@ static void __imgb_cpy_plane(void src, void dst, int bw, int h, int s_src, int

		static void uavs3e_image_copy_pic(void dst[4], int i_dst[4], unsigned char const src[4], const int i_src[4], enum AVPixelFormat pix_fmts, int width, int height)
		{
		__imgb_cpy_plane(src[0], dst[0], width, height, i_src[0], i_dst[0]);
		__imgb_cpy_plane(src[1], dst[1], width >> 1, height >> 1, i_src[1], i_dst[1]);
		__imgb_cpy_plane(src[2], dst[2], width >> 1, height >> 1, i_src[2], i_dst[2]);
		if(sizeof(pel)>1 && pix_fmts==AV_PIX_FMT_YUV420P){ //sizeof(pel)==2 when BIT_DEPTH == 10
		//Expand YUV420P to 2-byte
		for(int plane=0;plane<3;plane++){
		int planeheight=(plane==0)?height:height>>1;
		int planewidth=(plane==0)?width:width>>1;

		pel dstpel = (pel )dst[plane];
		int ind_src = 0;
		int ind_dst = 0;

		for (int y = 0; y < planeheight; y++) {
		for (int x = 0; x < planewidth; x++) {
		dstpel[ind_dst + x] = src[plane][ind_src + x]; //byte to short
		}//for (int x = 0; x < planewidth; x++)
		ind_src += i_src[plane];
		ind_dst += (i_dst[plane]>>1);
		}//for (int y = 0; y < planeheight; y++)
		}//for(int plane=0;plane<3;plane++)
		}
		else{
		width=(sizeof(pel)>1)?width<<1:width;
		__imgb_cpy_plane(src[0], dst[0], width, height, i_src[0], i_dst[0]);
		__imgb_cpy_plane(src[1], dst[1], width >> 1, height >> 1, i_src[1], i_dst[1]);
		__imgb_cpy_plane(src[2], dst[2], width >> 1, height >> 1, i_src[2], i_dst[2]);
		}
		}

		static int uavs3e_encode_frame(AVCodecContext avctx, AVPacket pkt,
Expand Down