cdg.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:13k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * cdg.c: CDG decoder module
  3.  *****************************************************************************
  4.  * Copyright (C) 2007 Laurent Aimar
  5.  * $Id: c1a3e754a014daecd5def969713962d1d3070999 $
  6.  *
  7.  * Authors: Laurent Aimar <fenrir # via.ecp.fr>
  8.  *
  9.  * This program is free software; you can redistribute it and/or modify
  10.  * it under the terms of the GNU General Public License as published by
  11.  * the Free Software Foundation; either version 2 of the License, or
  12.  * (at your option) any later version.
  13.  *
  14.  * This program is distributed in the hope that it will be useful,
  15.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17.  * GNU General Public License for more details.
  18.  *
  19.  * You should have received a copy of the GNU General Public License
  20.  * along with this program; if not, write to the Free Software
  21.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #ifdef HAVE_CONFIG_H
  27. # include "config.h"
  28. #endif
  29. #include <vlc_common.h>
  30. #include <vlc_plugin.h>
  31. #include <vlc_codec.h>
  32. #include <vlc_vout.h>
  33. /*****************************************************************************
  34.  * decoder_sys_t : decoder descriptor
  35.  *****************************************************************************/
  36. /* The screen size */
  37. #define CDG_SCREEN_WIDTH (300)
  38. #define CDG_SCREEN_HEIGHT (216)
  39. /* The border of the screen size */
  40. #define CDG_SCREEN_BORDER_WIDTH (6)
  41. #define CDG_SCREEN_BORDER_HEIGHT (12)
  42. /* The display part */
  43. #define CDG_DISPLAY_WIDTH  (CDG_SCREEN_WIDTH-2*CDG_SCREEN_BORDER_WIDTH)
  44. #define CDG_DISPLAY_HEIGHT (CDG_SCREEN_HEIGHT-2*CDG_SCREEN_BORDER_HEIGHT)
  45. #define CDG_SCREEN_PITCH (CDG_SCREEN_WIDTH)
  46. struct decoder_sys_t
  47. {
  48.     uint8_t color[16][3];
  49.     int     i_offseth;
  50.     int     i_offsetv;
  51.     uint8_t screen[CDG_SCREEN_PITCH*CDG_SCREEN_HEIGHT];
  52.     uint8_t *p_screen;
  53.     int     i_packet;
  54. };
  55. #define CDG_PACKET_SIZE (24)
  56. #define CDG_COLOR_R_SHIFT ( 0)
  57. #define CDG_COLOR_G_SHIFT ( 8)
  58. #define CDG_COLOR_B_SHIFT (16)
  59. /*****************************************************************************
  60.  * Local prototypes
  61.  *****************************************************************************/
  62. static int  Open ( vlc_object_t * );
  63. static void Close( vlc_object_t * );
  64. static picture_t *Decode( decoder_t *, block_t ** );
  65. static int DecodePacket( decoder_sys_t *p_cdg, uint8_t *p_buffer, int i_buffer );
  66. static int Render( decoder_sys_t *p_cdg, picture_t *p_picture );
  67. /*****************************************************************************
  68.  * Module descriptor
  69.  *****************************************************************************/
  70. vlc_module_begin ()
  71.     set_category( CAT_INPUT )
  72.     set_subcategory( SUBCAT_INPUT_VCODEC )
  73.     set_description( N_("CDG video decoder") )
  74.     set_capability( "decoder", 1000 )
  75.     set_callbacks( Open, Close )
  76.     add_shortcut( "cdg" )
  77. vlc_module_end ()
  78. /*****************************************************************************
  79.  * Open: probe the decoder and return score
  80.  *****************************************************************************/
  81. static int Open( vlc_object_t *p_this )
  82. {
  83.     decoder_t *p_dec = (decoder_t*)p_this;
  84.     decoder_sys_t *p_sys;
  85.     if( p_dec->fmt_in.i_codec != VLC_FOURCC('C','D','G',' ') )
  86.         return VLC_EGENERIC;
  87.     /* Allocate the memory needed to store the decoder's structure */
  88.     p_dec->p_sys = p_sys = calloc( 1, sizeof(decoder_sys_t) );
  89.     if( !p_sys )
  90.         return VLC_ENOMEM;
  91.     /* Init */
  92.     p_sys->p_screen = p_sys->screen;
  93.     p_sys->i_packet = 0;
  94.     /* Set output properties
  95.      * TODO maybe it would be better to use RV16 or RV24 ? */
  96.     p_dec->fmt_out.i_cat = VIDEO_ES;
  97.     p_dec->fmt_out.i_codec = VLC_FOURCC('R','V','3','2');
  98.     p_dec->fmt_out.video.i_width = CDG_DISPLAY_WIDTH;
  99.     p_dec->fmt_out.video.i_height = CDG_DISPLAY_HEIGHT;
  100.     p_dec->fmt_out.video.i_aspect =
  101.         VOUT_ASPECT_FACTOR * p_dec->fmt_out.video.i_width / p_dec->fmt_out.video.i_height;
  102.     p_dec->fmt_out.video.i_rmask = 0xff << CDG_COLOR_R_SHIFT;
  103.     p_dec->fmt_out.video.i_gmask = 0xff << CDG_COLOR_G_SHIFT;
  104.     p_dec->fmt_out.video.i_bmask = 0xff << CDG_COLOR_B_SHIFT;
  105.     /* Set callbacks */
  106.     p_dec->pf_decode_video = Decode;
  107.     return VLC_SUCCESS;
  108. }
  109. /****************************************************************************
  110.  * Decode: the whole thing
  111.  ****************************************************************************
  112.  * This function must be fed with a complete compressed frame.
  113.  ****************************************************************************/
  114. static picture_t *Decode( decoder_t *p_dec, block_t **pp_block )
  115. {
  116.     decoder_sys_t *p_sys = p_dec->p_sys;
  117.     block_t *p_block;
  118.     picture_t *p_pic = NULL;
  119.     if( !pp_block || !*pp_block )
  120.         return NULL;
  121.     p_block = *pp_block;
  122.     if( p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
  123.     {
  124.         p_sys->i_packet = 0;
  125.         goto exit;
  126.     }
  127.     /* Decode packet */
  128.     while( p_block->i_buffer >= CDG_PACKET_SIZE )
  129.     {
  130.         DecodePacket( p_sys, p_block->p_buffer, CDG_PACKET_SIZE );
  131.         p_block->i_buffer -= CDG_PACKET_SIZE;
  132.         p_block->p_buffer += CDG_PACKET_SIZE;
  133.     }
  134.     /* Only display 25 frame per second (there is 75 packets per second) */
  135.     if( (p_sys->i_packet%3) == 1 )
  136.     {
  137.         /* Get a new picture */
  138.         p_pic = decoder_NewPicture( p_dec );
  139.         if( !p_pic )
  140.             goto exit;
  141.         Render( p_sys, p_pic );
  142.         p_pic->date = p_block->i_pts > 0 ? p_block->i_pts : p_block->i_dts;
  143.     }
  144. exit:
  145.     block_Release( p_block ); *pp_block = NULL;
  146.     return p_pic;
  147. }
  148. /*****************************************************************************
  149.  * Close: decoder destruction
  150.  *****************************************************************************/
  151. static void Close( vlc_object_t *p_this )
  152. {
  153.     decoder_t *p_dec = (decoder_t *)p_this;
  154.     decoder_sys_t *p_sys = p_dec->p_sys;
  155.     free( p_sys );
  156. }
  157. /*****************************************************************************
  158.  * Decoder
  159.  *****************************************************************************/
  160. static void ScreenFill( decoder_sys_t *p_cdg, int sx, int sy, int dx, int dy, int c )
  161. {
  162.     int x, y;
  163.     for( y = sy; y < sy+dy; y++ )
  164.         for( x = sx; x < sx+dx; x++ )
  165.             p_cdg->p_screen[y*CDG_SCREEN_PITCH+x] = c;
  166. }
  167. static int DecodeMemoryPreset( decoder_sys_t *p_cdg, const uint8_t *p_data )
  168. {
  169.     const int i_color = p_data[0]&0x0f;
  170. #if 0
  171.     const int i_repeat= p_data[1]&0x0f;
  172. #endif
  173.     /* if i_repeat > 0 we could ignore it if we have a reliable stream */
  174.     ScreenFill( p_cdg, 0, 0, CDG_SCREEN_WIDTH, CDG_SCREEN_HEIGHT, i_color );
  175.     return 0;
  176. }
  177. static int DecodeBorderPreset( decoder_sys_t *p_cdg, const uint8_t *p_data )
  178. {
  179.     const int i_color = p_data[0]&0x0f;
  180.     /* */
  181.     ScreenFill( p_cdg,   0,   0,
  182.                 CDG_SCREEN_WIDTH, CDG_SCREEN_BORDER_HEIGHT, i_color );
  183.     ScreenFill( p_cdg,   0, CDG_SCREEN_HEIGHT-CDG_SCREEN_BORDER_HEIGHT,
  184.                 CDG_SCREEN_WIDTH, CDG_SCREEN_BORDER_HEIGHT, i_color );
  185.     ScreenFill( p_cdg,   0, CDG_SCREEN_BORDER_HEIGHT,
  186.                 CDG_SCREEN_BORDER_WIDTH, CDG_SCREEN_HEIGHT-CDG_SCREEN_BORDER_HEIGHT, i_color );
  187.     ScreenFill( p_cdg,   CDG_SCREEN_WIDTH-CDG_SCREEN_BORDER_WIDTH, CDG_SCREEN_BORDER_HEIGHT,
  188.                 CDG_SCREEN_BORDER_WIDTH, CDG_SCREEN_HEIGHT-CDG_SCREEN_BORDER_HEIGHT, i_color );
  189.     return 0;
  190. }
  191. static int DecodeLoadColorTable( decoder_sys_t *p_cdg, const uint8_t *p_data, int i_base )
  192. {
  193.     int n;
  194.     for( n = 0; n < 8; n++ )
  195.     {
  196.         const int c = ((p_data[2*n+0] << 8) | p_data[2*n+1]);
  197.         const int r = (c >> 10)&0x0f;
  198.         const int g = ((c >> 6)&0x0c) | ((c >> 4)&0x03);
  199.         const int b = c &0x0f;
  200.         p_cdg->color[i_base+n][0] = r << 4;
  201.         p_cdg->color[i_base+n][1] = g << 4;
  202.         p_cdg->color[i_base+n][2] = b << 4;
  203.     }
  204.     return 0;
  205. }
  206. static int DecodeTileBlock( decoder_sys_t *p_cdg, const uint8_t *p_data, int doXor )
  207. {
  208.     int p_color[2];
  209.     int sx, sy;
  210.     int x, y;
  211.     p_color[0] = p_data[0] & 0x0f;
  212.     p_color[1] = p_data[1] & 0x0f;
  213.     sy = (p_data[2] & 0x1f)*12;
  214.     sx = (p_data[3] & 0x3f)*6;
  215.     for( y = 0; y < 12; y++ )
  216.     {
  217.         for( x = 0; x < 6; x++ )
  218.         {
  219.             const int idx = ( p_data[4+y] >> (5-x) ) & 0x01;
  220.             uint8_t *p = &p_cdg->p_screen[(sy+y)*CDG_SCREEN_PITCH+(sx+x)];
  221.             if( doXor )
  222.                 *p ^= p_color[idx];
  223.             else
  224.                 *p = p_color[idx];
  225.         }
  226.     }
  227.     return 0;
  228. }
  229. static int DecodeScroll( decoder_sys_t *p_cdg, const uint8_t *p_data, int b_copy )
  230. {
  231.     uint8_t copy[CDG_SCREEN_PITCH*CDG_SCREEN_HEIGHT];
  232.     uint8_t color = p_data[0]&0x0f;
  233.     int i_shifth;
  234.     int i_shiftv;
  235.     int x, y;
  236.     /* */
  237.     p_cdg->i_offseth = p_data[1]&0x7;
  238.     if( p_cdg->i_offseth >= CDG_SCREEN_BORDER_WIDTH )
  239.         p_cdg->i_offseth = CDG_SCREEN_BORDER_WIDTH-1;
  240.     p_cdg->i_offsetv = p_data[2]&0xf;
  241.     if( p_cdg->i_offsetv >= CDG_SCREEN_BORDER_HEIGHT )
  242.         p_cdg->i_offsetv = CDG_SCREEN_BORDER_HEIGHT-1;
  243.     /* */
  244.     switch( (p_data[1] >> 4)&0x3 )
  245.     {
  246.     case 0x01: i_shifth =  6; break;
  247.     case 0x02: i_shifth = -6; break;
  248.     default:
  249.         i_shifth = 0;
  250.         break;
  251.     }
  252.     switch( (p_data[2] >> 4)&0x3 )
  253.     {
  254.     case 0x01: i_shiftv = 12; break;
  255.     case 0x02: i_shiftv =-12; break;
  256.     default:
  257.         i_shiftv = 0;
  258.         break;
  259.     }
  260.     if( i_shifth == 0 && i_shiftv == 0 )
  261.         return 0;
  262.     /* Make a copy of the screen */
  263.     memcpy( copy, p_cdg->screen, sizeof(p_cdg->screen) );
  264.     /* Fill the uncovered part XXX way too much */
  265.     ScreenFill( p_cdg, 0, 0, CDG_SCREEN_WIDTH, CDG_SCREEN_HEIGHT, color );
  266.     /* Copy back */
  267.     for( y = 0; y < CDG_SCREEN_HEIGHT; y++ )
  268.     {
  269.         int dy = i_shiftv + y;
  270.         for( x = 0; x < CDG_SCREEN_WIDTH; x++ )
  271.         {
  272.             int dx = i_shifth + x;
  273.             if( b_copy )
  274.             {
  275.                 dy = ( dy + CDG_SCREEN_HEIGHT ) % CDG_SCREEN_HEIGHT;
  276.                 dy = ( dy + CDG_SCREEN_WIDTH  ) % CDG_SCREEN_WIDTH;
  277.             }
  278.             else
  279.             {
  280.                 if( dy < 0 || dy >= CDG_SCREEN_HEIGHT ||
  281.                     dx < 0 || dx >= CDG_SCREEN_WIDTH )
  282.                     continue;
  283.             }
  284.             p_cdg->screen[dy*CDG_SCREEN_PITCH+dx] = copy[y*CDG_SCREEN_PITCH+x];
  285.         }
  286.     }
  287.     /* */
  288.     //CdgDebug( CDG_LOG_WARNING, "DecodeScroll: color=%d ch=%d oh=%d cv=%d ov=%dn copy=%dn", color, i_shifth, p_cdg->i_offseth, i_shiftv, p_cdg->i_offsetv, b_copy );
  289.     return 0;
  290. }
  291. static int DecodePacket( decoder_sys_t *p_cdg, uint8_t *p_buffer, int i_buffer )
  292. {
  293.     const int i_cmd = p_buffer[0] & 0x3f;
  294.     const int i_instruction = p_buffer[1] & 0x3f;
  295.     const uint8_t *p_data = &p_buffer[4];
  296.     if( i_buffer != CDG_PACKET_SIZE )
  297.         return -1;
  298.     p_cdg->i_packet++;
  299.     /* Handle CDG command only */
  300.     if( i_cmd != 0x09 )
  301.         return 0;
  302.     switch( i_instruction )
  303.     {
  304.         case 1:
  305.             DecodeMemoryPreset( p_cdg, p_data );
  306.             break;
  307.         case 2:
  308.             DecodeBorderPreset( p_cdg, p_data );
  309.             break;
  310.         case 6:
  311.             DecodeTileBlock( p_cdg, p_data, 0 );
  312.             break;
  313.         case 20:
  314.             DecodeScroll( p_cdg, p_data, 0 );
  315.             break;
  316.         case 24:
  317.             DecodeScroll( p_cdg, p_data, 1 );
  318.             break;
  319.         case 28:
  320.             /* FIXME what to do about that one ? */
  321.             //CdgDebug( CDG_LOG_WARNING, "DecodeDefineTransparentColor not implementedn" );
  322.             //DecodeDefineTransparentColor( p_cdg, p_data );
  323.             break;
  324.         case 30:
  325.             DecodeLoadColorTable( p_cdg, p_data, 0 );
  326.             break;
  327.         case 31:
  328.             DecodeLoadColorTable( p_cdg, p_data, 8 );
  329.             break;
  330.         case 38:
  331.             DecodeTileBlock( p_cdg, p_data, 1 );
  332.             break;
  333.         default:
  334.             break;
  335.     }
  336.     return 0;
  337. }
  338. static void RenderPixel( picture_t *p_picture, int x, int y, uint32_t c )
  339. {
  340.     const int i_plane = 0;
  341.     const int i_pitch = p_picture->p[i_plane].i_pitch;
  342.     uint32_t *s = (uint32_t*)&p_picture->p[i_plane].p_pixels[y*i_pitch + x*sizeof(uint32_t)];
  343.     *s = c;
  344. }
  345. static uint32_t RenderRGB( int r, int g, int b )
  346. {
  347.     return ( r << CDG_COLOR_R_SHIFT ) | ( g << CDG_COLOR_G_SHIFT ) | ( b << CDG_COLOR_B_SHIFT );
  348. }
  349. static int Render( decoder_sys_t *p_cdg, picture_t *p_picture )
  350. {
  351.     int x, y;
  352.     for( y = 0; y < CDG_DISPLAY_HEIGHT; y++ )
  353.     {
  354.         for( x = 0; x < CDG_DISPLAY_WIDTH; x++ )
  355.         {
  356.             const int sx = x + p_cdg->i_offseth + CDG_SCREEN_BORDER_WIDTH;
  357.             const int sy = y + p_cdg->i_offsetv + CDG_SCREEN_BORDER_HEIGHT;
  358.             uint8_t cidx = p_cdg->p_screen[sy*CDG_SCREEN_PITCH +sx];
  359.             uint8_t r = p_cdg->color[cidx][0];
  360.             uint8_t g = p_cdg->color[cidx][1];
  361.             uint8_t b = p_cdg->color[cidx][2];
  362.             RenderPixel( p_picture, x, y, RenderRGB( r, g, b ) );
  363.         }
  364.     }
  365.     return 0;
  366. }