falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17262 Commits
32 Branches
283MB
Tree: ad1eebe360
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'ad1eebe360'
${ noResults }
FFmpeg/libavcodec/ulti.c

417 lines
14KB
Raw Blame History 

  1. /*

  2.  * IBM Ultimotion Video Decoder

  3.  * Copyright (C) 2004 Konstantin Shishkov

  4.  *

  5.  * This file is part of FFmpeg.

  6.  *

  7.  * FFmpeg is free software; you can redistribute it and/or

  8.  * modify it under the terms of the GNU Lesser General Public

  9.  * License as published by the Free Software Foundation; either

  10.  * version 2.1 of the License, or (at your option) any later version.

  11.  *

  12.  * FFmpeg is distributed in the hope that it will be useful,

  13.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  14.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  15.  * Lesser General Public License for more details.

  16.  *

  17.  * You should have received a copy of the GNU Lesser General Public

  18.  * License along with FFmpeg; if not, write to the Free Software

  19.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

  20.  */

  21. 

  22. /**

  23.  * @file ulti.c

  24.  * IBM Ultimotion Video Decoder.

  25.  */

  26. 

  27. #include <stdio.h>

  28. #include <stdlib.h>

  29. #include <string.h>

  30. #include <unistd.h>

  31. 

  32. #include "avcodec.h"

  33. #include "bytestream.h"

  34. 

  35. #include "ulti_cb.h"

  36. 

  37. typedef struct UltimotionDecodeContext {

  38.     AVCodecContext *avctx;

  39.     int width, height, blocks;

  40.     AVFrame frame;

  41.     const uint8_t *ulti_codebook;

  42. } UltimotionDecodeContext;

  43. 

  44. static av_cold int ulti_decode_init(AVCodecContext *avctx)

  45. {

  46.     UltimotionDecodeContext *s = avctx->priv_data;

  47. 

  48.     s->avctx = avctx;

  49.     s->width = avctx->width;

  50.     s->height = avctx->height;

  51.     s->blocks = (s->width / 8) * (s->height / 8);

  52.     avctx->pix_fmt = PIX_FMT_YUV410P;

  53.     avctx->coded_frame = (AVFrame*) &s->frame;

  54.     s->ulti_codebook = ulti_codebook;

  55. 

  56.     return 0;

  57. }

  58. 

  59. static const int block_coords[8] = // 4x4 block coords in 8x8 superblock

  60.     { 0, 0, 0, 4, 4, 4, 4, 0};

  61. 

  62. static const int angle_by_index[4] = { 0, 2, 6, 12};

  63. 

  64. /* Lookup tables for luma and chroma - used by ulti_convert_yuv() */

  65. static const uint8_t ulti_lumas[64] =

  66.     { 0x10, 0x13, 0x17, 0x1A, 0x1E, 0x21, 0x25, 0x28,

  67.       0x2C, 0x2F, 0x33, 0x36, 0x3A, 0x3D, 0x41, 0x44,

  68.       0x48, 0x4B, 0x4F, 0x52, 0x56, 0x59, 0x5C, 0x60,

  69.       0x63, 0x67, 0x6A, 0x6E, 0x71, 0x75, 0x78, 0x7C,

  70.       0x7F, 0x83, 0x86, 0x8A, 0x8D, 0x91, 0x94, 0x98,

  71.       0x9B, 0x9F, 0xA2, 0xA5, 0xA9, 0xAC, 0xB0, 0xB3,

  72.       0xB7, 0xBA, 0xBE, 0xC1, 0xC5, 0xC8, 0xCC, 0xCF,

  73.       0xD3, 0xD6, 0xDA, 0xDD, 0xE1, 0xE4, 0xE8, 0xEB};

  74. 

  75. static const uint8_t ulti_chromas[16] =

  76.     { 0x60, 0x67, 0x6D, 0x73, 0x7A, 0x80, 0x86, 0x8D,

  77.       0x93, 0x99, 0xA0, 0xA6, 0xAC, 0xB3, 0xB9, 0xC0};

  78. 

  79. /* convert Ultimotion YUV block (sixteen 6-bit Y samples and

  80.  two 4-bit chroma samples) into standard YUV and put it into frame */

  81. static void ulti_convert_yuv(AVFrame *frame, int x, int y,

  82.                              uint8_t *luma,int chroma)

  83. {

  84.     uint8_t *y_plane, *cr_plane, *cb_plane;

  85.     int i;

  86. 

  87.     y_plane = frame->data[0] + x + y * frame->linesize[0];

  88.     cr_plane = frame->data[1] + (x / 4) + (y / 4) * frame->linesize[1];

  89.     cb_plane = frame->data[2] + (x / 4) + (y / 4) * frame->linesize[2];

  90. 

  91.     cr_plane[0] = ulti_chromas[chroma >> 4];

  92. 

  93.     cb_plane[0] = ulti_chromas[chroma & 0xF];

  94. 

  95. 

  96.     for(i = 0; i < 16; i++){

  97.         y_plane[i & 3] = ulti_lumas[luma[i]];

  98.         if((i & 3) == 3) { //next row

  99.             y_plane += frame->linesize[0];

  100.         }

  101.     }

  102. }

  103. 

  104. /* generate block like in MS Video1 */

  105. static void ulti_pattern(AVFrame *frame, int x, int y,

  106.                          int f0, int f1, int Y0, int Y1, int chroma)

  107. {

  108.     uint8_t Luma[16];

  109.     int mask, i;

  110.     for(mask = 0x80, i = 0; mask; mask >>= 1, i++) {

  111.         if(f0 & mask)

  112.             Luma[i] = Y1;

  113.         else

  114.             Luma[i] = Y0;

  115.     }

  116. 

  117.     for(mask = 0x80, i = 8; mask; mask >>= 1, i++) {

  118.         if(f1 & mask)

  119.             Luma[i] = Y1;

  120.         else

  121.             Luma[i] = Y0;

  122.     }

  123. 

  124.     ulti_convert_yuv(frame, x, y, Luma, chroma);

  125. }

  126. 

  127. /* fill block with some gradient */

  128. static void ulti_grad(AVFrame *frame, int x, int y, uint8_t *Y, int chroma, int angle)

  129. {

  130.     uint8_t Luma[16];

  131.     if(angle & 8) { //reverse order

  132.         int t;

  133.         angle &= 0x7;

  134.         t = Y[0];

  135.         Y[0] = Y[3];

  136.         Y[3] = t;

  137.         t = Y[1];

  138.         Y[1] = Y[2];

  139.         Y[2] = t;

  140.     }

  141.     switch(angle){

  142.     case 0:

  143.         Luma[0]  = Y[0]; Luma[1]  = Y[1]; Luma[2]  = Y[2]; Luma[3]  = Y[3];

  144.         Luma[4]  = Y[0]; Luma[5]  = Y[1]; Luma[6]  = Y[2]; Luma[7]  = Y[3];

  145.         Luma[8]  = Y[0]; Luma[9]  = Y[1]; Luma[10] = Y[2]; Luma[11] = Y[3];

  146.         Luma[12] = Y[0]; Luma[13] = Y[1]; Luma[14] = Y[2]; Luma[15] = Y[3];

  147.         break;

  148.     case 1:

  149.         Luma[0]  = Y[1]; Luma[1]  = Y[2]; Luma[2]  = Y[3]; Luma[3]  = Y[3];

  150.         Luma[4]  = Y[0]; Luma[5]  = Y[1]; Luma[6]  = Y[2]; Luma[7]  = Y[3];

  151.         Luma[8]  = Y[0]; Luma[9]  = Y[1]; Luma[10] = Y[2]; Luma[11] = Y[3];

  152.         Luma[12] = Y[0]; Luma[13] = Y[0]; Luma[14] = Y[1]; Luma[15] = Y[2];

  153.         break;

  154.     case 2:

  155.         Luma[0]  = Y[1]; Luma[1]  = Y[2]; Luma[2]  = Y[3]; Luma[3]  = Y[3];

  156.         Luma[4]  = Y[1]; Luma[5]  = Y[2]; Luma[6]  = Y[2]; Luma[7]  = Y[3];

  157.         Luma[8]  = Y[0]; Luma[9]  = Y[1]; Luma[10] = Y[1]; Luma[11] = Y[2];

  158.         Luma[12] = Y[0]; Luma[13] = Y[0]; Luma[14] = Y[1]; Luma[15] = Y[2];

  159.         break;

  160.     case 3:

  161.         Luma[0]  = Y[2]; Luma[1]  = Y[3]; Luma[2]  = Y[3]; Luma[3]  = Y[3];

  162.         Luma[4]  = Y[1]; Luma[5]  = Y[2]; Luma[6]  = Y[2]; Luma[7]  = Y[3];

  163.         Luma[8]  = Y[0]; Luma[9]  = Y[1]; Luma[10] = Y[1]; Luma[11] = Y[2];

  164.         Luma[12] = Y[0]; Luma[13] = Y[0]; Luma[14] = Y[0]; Luma[15] = Y[1];

  165.         break;

  166.     case 4:

  167.         Luma[0]  = Y[3]; Luma[1]  = Y[3]; Luma[2]  = Y[3]; Luma[3]  = Y[3];

  168.         Luma[4]  = Y[2]; Luma[5]  = Y[2]; Luma[6]  = Y[2]; Luma[7]  = Y[2];

  169.         Luma[8]  = Y[1]; Luma[9]  = Y[1]; Luma[10] = Y[1]; Luma[11] = Y[1];

  170.         Luma[12] = Y[0]; Luma[13] = Y[0]; Luma[14] = Y[0]; Luma[15] = Y[0];

  171.         break;

  172.     case 5:

  173.         Luma[0]  = Y[3]; Luma[1]  = Y[3]; Luma[2]  = Y[3]; Luma[3]  = Y[2];

  174.         Luma[4]  = Y[3]; Luma[5]  = Y[2]; Luma[6]  = Y[2]; Luma[7]  = Y[1];

  175.         Luma[8]  = Y[2]; Luma[9]  = Y[1]; Luma[10] = Y[1]; Luma[11] = Y[0];

  176.         Luma[12] = Y[1]; Luma[13] = Y[0]; Luma[14] = Y[0]; Luma[15] = Y[0];

  177.         break;

  178.     case 6:

  179.         Luma[0]  = Y[3]; Luma[1]  = Y[3]; Luma[2]  = Y[2]; Luma[3]  = Y[2];

  180.         Luma[4]  = Y[3]; Luma[5]  = Y[2]; Luma[6]  = Y[1]; Luma[7]  = Y[1];

  181.         Luma[8]  = Y[2]; Luma[9]  = Y[2]; Luma[10] = Y[1]; Luma[11] = Y[0];

  182.         Luma[12] = Y[1]; Luma[13] = Y[1]; Luma[14] = Y[0]; Luma[15] = Y[0];

  183.         break;

  184.     case 7:

  185.         Luma[0]  = Y[3]; Luma[1]  = Y[3]; Luma[2]  = Y[2]; Luma[3]  = Y[1];

  186.         Luma[4]  = Y[3]; Luma[5]  = Y[2]; Luma[6]  = Y[1]; Luma[7]  = Y[0];

  187.         Luma[8]  = Y[3]; Luma[9]  = Y[2]; Luma[10] = Y[1]; Luma[11] = Y[0];

  188.         Luma[12] = Y[2]; Luma[13] = Y[1]; Luma[14] = Y[0]; Luma[15] = Y[0];

  189.         break;

  190.     default:

  191.         Luma[0]  = Y[0]; Luma[1]  = Y[0]; Luma[2]  = Y[1]; Luma[3]  = Y[1];

  192.         Luma[4]  = Y[0]; Luma[5]  = Y[0]; Luma[6]  = Y[1]; Luma[7]  = Y[1];

  193.         Luma[8]  = Y[2]; Luma[9]  = Y[2]; Luma[10] = Y[3]; Luma[11] = Y[3];

  194.         Luma[12] = Y[2]; Luma[13] = Y[2]; Luma[14] = Y[3]; Luma[15] = Y[3];

  195.         break;

  196.     }

  197. 

  198.     ulti_convert_yuv(frame, x, y, Luma, chroma);

  199. }

  200. 

  201. static int ulti_decode_frame(AVCodecContext *avctx,

  202.                              void *data, int *data_size,

  203.                              const uint8_t *buf, int buf_size)

  204. {

  205.     UltimotionDecodeContext *s=avctx->priv_data;

  206.     int modifier = 0;

  207.     int uniq = 0;

  208.     int mode = 0;

  209.     int blocks = 0;

  210.     int done = 0;

  211.     int x = 0, y = 0;

  212.     int i;

  213.     int skip;

  214.     int tmp;

  215. 

  216.     if(s->frame.data[0])

  217.         avctx->release_buffer(avctx, &s->frame);

  218. 

  219.     s->frame.reference = 1;

  220.     s->frame.buffer_hints = FF_BUFFER_HINTS_VALID | FF_BUFFER_HINTS_PRESERVE | FF_BUFFER_HINTS_REUSABLE;

  221.     if(avctx->get_buffer(avctx, &s->frame) < 0) {

  222.         av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");

  223.         return -1;

  224.     }

  225. 

  226.     while(!done) {

  227.         int idx;

  228.         if(blocks >= s->blocks || y >= s->height)

  229.             break;//all blocks decoded

  230. 

  231.         idx = *buf++;

  232.         if((idx & 0xF8) == 0x70) {

  233.             switch(idx) {

  234.             case 0x70: //change modifier

  235.                 modifier = *buf++;

  236.                 if(modifier>1)

  237.                     av_log(avctx, AV_LOG_INFO, "warning: modifier must be 0 or 1, got %i\n", modifier);

  238.                 break;

  239.             case 0x71: // set uniq flag

  240.                 uniq = 1;

  241.                 break;

  242.             case 0x72: //toggle mode

  243.                 mode = !mode;

  244.                 break;

  245.             case 0x73: //end-of-frame

  246.                 done = 1;

  247.                 break;

  248.             case 0x74: //skip some blocks

  249.                 skip = *buf++;

  250.                 if ((blocks + skip) >= s->blocks)

  251.                     break;

  252.                 blocks += skip;

  253.                 x += skip * 8;

  254.                 while(x >= s->width) {

  255.                     x -= s->width;

  256.                     y += 8;

  257.                 }

  258.                 break;

  259.             default:

  260.                 av_log(avctx, AV_LOG_INFO, "warning: unknown escape 0x%02X\n", idx);

  261.             }

  262.         } else { //handle one block

  263.             int code;

  264.             int cf;

  265.             int angle = 0;

  266.             uint8_t Y[4]; // luma samples of block

  267.             int tx = 0, ty = 0; //coords of subblock

  268.             int chroma = 0;

  269.             if (mode || uniq) {

  270.                 uniq = 0;

  271.                 cf = 1;

  272.                 chroma = 0;

  273.             } else {

  274.                 cf = 0;

  275.                 if (idx)

  276.                     chroma = *buf++;

  277.             }

  278.             for (i = 0; i < 4; i++) { // for every subblock

  279.                 code = (idx >> (6 - i*2)) & 3; //extract 2 bits

  280.                 if(!code) //skip subblock

  281.                     continue;

  282.                 if(cf)

  283.                     chroma = *buf++;

  284.                 tx = x + block_coords[i * 2];

  285.                 ty = y + block_coords[(i * 2) + 1];

  286.                 switch(code) {

  287.                 case 1:

  288.                     tmp = *buf++;

  289. 

  290.                     angle = angle_by_index[(tmp >> 6) & 0x3];

  291. 

  292.                     Y[0] = tmp & 0x3F;

  293.                     Y[1] = Y[0];

  294. 

  295.                     if (angle) {

  296.                         Y[2] = Y[0]+1;

  297.                         if (Y[2] > 0x3F)

  298.                             Y[2] = 0x3F;

  299.                         Y[3] = Y[2];

  300.                     } else {

  301.                         Y[2] = Y[0];

  302.                         Y[3] = Y[0];

  303.                     }

  304.                     break;

  305. 

  306.                 case 2:

  307.                     if (modifier) { // unpack four luma samples

  308.                         tmp = bytestream_get_be24(&buf);

  309. 

  310.                         Y[0] = (tmp >> 18) & 0x3F;

  311.                         Y[1] = (tmp >> 12) & 0x3F;

  312.                         Y[2] = (tmp >> 6) & 0x3F;

  313.                         Y[3] = tmp & 0x3F;

  314.                         angle = 16;

  315.                     } else { // retrieve luma samples from codebook

  316.                         tmp = bytestream_get_be16(&buf);

  317. 

  318.                         angle = (tmp >> 12) & 0xF;

  319.                         tmp &= 0xFFF;

  320.                         tmp <<= 2;

  321.                         Y[0] = s->ulti_codebook[tmp];

  322.                         Y[1] = s->ulti_codebook[tmp + 1];

  323.                         Y[2] = s->ulti_codebook[tmp + 2];

  324.                         Y[3] = s->ulti_codebook[tmp + 3];

  325.                     }

  326.                     break;

  327. 

  328.                 case 3:

  329.                     if (modifier) { // all 16 luma samples

  330.                         uint8_t Luma[16];

  331. 

  332.                         tmp = bytestream_get_be24(&buf);

  333.                         Luma[0] = (tmp >> 18) & 0x3F;

  334.                         Luma[1] = (tmp >> 12) & 0x3F;

  335.                         Luma[2] = (tmp >> 6) & 0x3F;

  336.                         Luma[3] = tmp & 0x3F;

  337. 

  338.                         tmp = bytestream_get_be24(&buf);

  339.                         Luma[4] = (tmp >> 18) & 0x3F;

  340.                         Luma[5] = (tmp >> 12) & 0x3F;

  341.                         Luma[6] = (tmp >> 6) & 0x3F;

  342.                         Luma[7] = tmp & 0x3F;

  343. 

  344.                         tmp = bytestream_get_be24(&buf);

  345.                         Luma[8] = (tmp >> 18) & 0x3F;

  346.                         Luma[9] = (tmp >> 12) & 0x3F;

  347.                         Luma[10] = (tmp >> 6) & 0x3F;

  348.                         Luma[11] = tmp & 0x3F;

  349. 

  350.                         tmp = bytestream_get_be24(&buf);

  351.                         Luma[12] = (tmp >> 18) & 0x3F;

  352.                         Luma[13] = (tmp >> 12) & 0x3F;

  353.                         Luma[14] = (tmp >> 6) & 0x3F;

  354.                         Luma[15] = tmp & 0x3F;

  355. 

  356.                         ulti_convert_yuv(&s->frame, tx, ty, Luma, chroma);

  357.                     } else {

  358.                         tmp = *buf++;

  359.                         if(tmp & 0x80) {

  360.                             angle = (tmp >> 4) & 0x7;

  361.                             tmp = (tmp << 8) + *buf++;

  362.                             Y[0] = (tmp >> 6) & 0x3F;

  363.                             Y[1] = tmp & 0x3F;

  364.                             Y[2] = (*buf++) & 0x3F;

  365.                             Y[3] = (*buf++) & 0x3F;

  366.                             ulti_grad(&s->frame, tx, ty, Y, chroma, angle); //draw block

  367.                         } else { // some patterns

  368.                             int f0, f1;

  369.                             f0 = *buf++;

  370.                             f1 = tmp;

  371.                             Y[0] = (*buf++) & 0x3F;

  372.                             Y[1] = (*buf++) & 0x3F;

  373.                             ulti_pattern(&s->frame, tx, ty, f1, f0, Y[0], Y[1], chroma);

  374.                         }

  375.                     }

  376.                     break;

  377.                 }

  378.                 if(code != 3)

  379.                     ulti_grad(&s->frame, tx, ty, Y, chroma, angle); // draw block

  380.             }

  381.             blocks++;

  382.                 x += 8;

  383.             if(x >= s->width) {

  384.                 x = 0;

  385.                 y += 8;

  386.             }

  387.         }

  388.     }

  389. 

  390.     *data_size=sizeof(AVFrame);

  391.     *(AVFrame*)data= s->frame;

  392. 

  393.     return buf_size;

  394. }

  395. 

  396. static av_cold int ulti_decode_end(AVCodecContext *avctx)

  397. {

  398. /*    UltimotionDecodeContext *s = avctx->priv_data;*/

  399. 

  400.     return 0;

  401. }

  402. 

  403. AVCodec ulti_decoder = {

  404.     "ultimotion",

  405.     CODEC_TYPE_VIDEO,

  406.     CODEC_ID_ULTI,

  407.     sizeof(UltimotionDecodeContext),

  408.     ulti_decode_init,

  409.     NULL,

  410.     ulti_decode_end,

  411.     ulti_decode_frame,

  412.     CODEC_CAP_DR1,

  413.     NULL,

  414.     .long_name = NULL_IF_CONFIG_SMALL("IBM UltiMotion"),

  415. };