]> git.sesse.net Git - vlc/blobdiff - modules/codec/crystalhd.c
Clean up L20 decoding
[vlc] / modules / codec / crystalhd.c
index da25d3ef1a1ed5208cb430e2a919f25767d4ccce..ff60509d521041485685d71742e014cc7e390414 100644 (file)
@@ -15,8 +15,8 @@
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  *****************************************************************************/
 
 /*****************************************************************************
 # include "config.h"
 #endif
 
+/* TODO
+ * - pts = 0?
+ * - mpeg4-asp
+ * - win32 testing
+ */
+
 /* VLC includes */
 #include <vlc_common.h>
 #include <vlc_plugin.h>
 #include <vlc_codec.h>
 
+/* Workaround for some versions of libcrystalHD */
 #if !defined(_WIN32) && !defined(__APPLE__)
 #  define __LINUX_USER__
 #endif
@@ -87,7 +94,7 @@ vlc_module_end ()
  * Local prototypes
  *****************************************************************************/
 static picture_t *DecodeBlock   ( decoder_t *p_dec, block_t **pp_block );
-static void crystal_CopyPicture ( picture_t *, BC_DTS_PROC_OUT* );
+// static void crystal_CopyPicture ( picture_t *, BC_DTS_PROC_OUT* );
 static int crystal_insert_sps_pps(decoder_t *, uint8_t *, uint32_t);
 
 /*****************************************************************************
@@ -102,16 +109,22 @@ struct decoder_sys_t
 
     uint32_t i_nal_size;     /* NAL header size */
 
+    /* Callback */
+    picture_t       *p_pic;
+    BC_DTS_PROC_OUT *proc_out;
+
 #ifdef USE_DL_OPENING
     HINSTANCE p_bcm_dll;
     BC_STATUS (WINAPI *OurDtsCloseDecoder)( HANDLE hDevice );
     BC_STATUS (WINAPI *OurDtsDeviceClose)( HANDLE hDevice );
     BC_STATUS (WINAPI *OurDtsFlushInput)( HANDLE hDevice, U32 Mode );
     BC_STATUS (WINAPI *OurDtsStopDecoder)( HANDLE hDevice );
-    BC_STATUS (WINAPI *OurDtsGetDriverStatus)( HANDLE hDevice, BC_DTS_STATUS *pStatus );
+    BC_STATUS (WINAPI *OurDtsGetDriverStatus)( HANDLE hDevice,
+                            BC_DTS_STATUS *pStatus );
     BC_STATUS (WINAPI *OurDtsProcInput)( HANDLE hDevice, U8 *pUserData,
                             U32 ulSizeInBytes, U64 timeStamp, BOOL encrypted );
-    BC_STATUS (WINAPI *OurDtsProcOutput)( HANDLE hDevice, U32 milliSecWait, BC_DTS_PROC_OUT *pOut );
+    BC_STATUS (WINAPI *OurDtsProcOutput)( HANDLE hDevice, U32 milliSecWait,
+                            BC_DTS_PROC_OUT *pOut );
     BC_STATUS (WINAPI *OurDtsIsEndOfStream)( HANDLE hDevice, U8* bEOS );
 #endif
 };
@@ -149,6 +162,9 @@ static int OpenDecoder( vlc_object_t *p_this )
 /* Not ready for production yet
     case VLC_CODEC_MP4V:
         i_bcm_codec_subtype = BC_MSUBTYPE_DIVX;
+        break;
+    case VLC_CODEC_DIV3:
+        i_bcm_codec_subtype = BC_MSUBTYPE_DIVX311;
         break; */
     default:
         return VLC_EGENERIC;
@@ -160,28 +176,49 @@ static int OpenDecoder( vlc_object_t *p_this )
         return VLC_ENOMEM;
 
     /* Fill decoder_sys_t */
-    p_dec->p_sys          = p_sys;
-    p_sys->i_nal_size     = 4; // assume 4 byte start codes
-    p_sys->i_sps_pps_size = 0;
-    p_sys->p_sps_pps_buf  = NULL;
-
+    p_dec->p_sys            = p_sys;
+    p_sys->i_nal_size       = 4; // assume 4 byte start codes
+    p_sys->i_sps_pps_size   = 0;
+    p_sys->p_sps_pps_buf    = NULL;
+    p_dec->p_sys->p_pic     = NULL;
+    p_dec->p_sys->proc_out  = NULL;
+
+    /* Win32 code *
+     * We cannot link and ship BCM dll, even with LGPL license (too big)
+     * and if we don't ship it, the plugin would not work depending on the
+     * installation order => DLopen */
 #ifdef USE_DL_OPENING
-    p_sys->p_bcm_dll = LoadLibrary( "bcmDIL.dll" );
+#  define DLL_NAME "bcmDIL.dll"
+#  define PATHS_NB 3
+    static const char *psz_paths[PATHS_NB] = {
+    DLL_NAME,
+    "C:\\Program Files\\Broadcom\\Broadcom CrystalHD Decoder\\" DLL_NAME,
+    "C:\\Program Files (x86)\\Broadcom\\Broadcom CrystalHD Decoder\\" DLL_NAME,
+    };
+    for( int i = 0; i < PATHS_NB; i++ )
+    {
+        HINSTANCE p_bcm_dll = LoadLibrary( psz_paths[i] );
+        if( p_bcm_dll )
+        {
+            p_sys->p_bcm_dll = p_bcm_dll;
+            break;
+        }
+    }
     if( !p_sys->p_bcm_dll )
     {
-        #ifdef DEBUG_CRYSTALHD
         msg_Dbg( p_dec, "Couldn't load the CrystalHD dll");
-        #endif
         return VLC_EGENERIC;
     }
 
 #define LOAD_SYM( a ) \
-    BC_FUNC( a )  = (void *)GetProcAddress(p_sys->p_bcm_dll, TEXT( #a ) ); \
-    if( !BC_FUNC( a ) ) { msg_Err( p_dec, "missing symbol " # a ); return VLC_EGENERIC; }
+    BC_FUNC( a )  = (void *)GetProcAddress( p_sys->p_bcm_dll, TEXT( #a ) ); \
+    if( !BC_FUNC( a ) ) { \
+        msg_Err( p_dec, "missing symbol " # a ); return VLC_EGENERIC; }
 
 #define LOAD_SYM_PSYS( a ) \
-    p_sys->BC_FUNC( a )  = (void *)GetProcAddress(p_sys->p_bcm_dll, #a ); \
-    if( !p_sys->BC_FUNC( a ) ) { msg_Err( p_dec, "missing symbol " # a ); return VLC_EGENERIC; }
+    p_sys->BC_FUNC( a )  = (void *)GetProcAddress( p_sys->p_bcm_dll, #a ); \
+    if( !p_sys->BC_FUNC( a ) ) { \
+        msg_Err( p_dec, "missing symbol " # a ); return VLC_EGENERIC; }
 
     BC_STATUS (WINAPI *OurDtsDeviceOpen)( HANDLE *hDevice, U32 mode );
     LOAD_SYM( DtsDeviceOpen );
@@ -215,9 +252,9 @@ static int OpenDecoder( vlc_object_t *p_this )
 
     /* Get the handle for the device */
     if( BC_FUNC(DtsDeviceOpen)( &p_sys->bcm_handle,
-                    (DTS_PLAYBACK_MODE | DTS_LOAD_FILE_PLAY_FW | DTS_SKIP_TX_CHK_CPB ) )
-                    // | DTS_DFLT_RESOLUTION(vdecRESOLUTION_720p29_97) ) )
-                    != BC_STS_SUCCESS )
+             (DTS_PLAYBACK_MODE | DTS_LOAD_FILE_PLAY_FW | DTS_SKIP_TX_CHK_CPB) )
+             // | DTS_DFLT_RESOLUTION(vdecRESOLUTION_720p29_97) ) )
+             != BC_STS_SUCCESS )
     {
         msg_Err( p_dec, "Couldn't find and open the BCM CrystalHD device" );
         free( p_sys );
@@ -229,11 +266,13 @@ static int OpenDecoder( vlc_object_t *p_this )
     if( BC_FUNC(DtsCrystalHDVersion)( p_sys->bcm_handle, &info ) == BC_STS_SUCCESS )
     {
         msg_Dbg( p_dec, "Using CrystalHD Driver version: %i.%i.%i, "
-            "Library version: %i.%i.%i, "
-            "Firmware version: %i.%i.%i",
-            info.drvVersion.drvRelease, info.drvVersion.drvMajor, info.drvVersion.drvMinor,
-            info.dilVersion.dilRelease, info.dilVersion.dilMajor, info.dilVersion.dilMinor,
-            info.fwVersion.fwRelease, info.fwVersion.fwMajor, info.fwVersion.fwMinor );
+            "Library version: %i.%i.%i, Firmware version: %i.%i.%i",
+            info.drvVersion.drvRelease, info.drvVersion.drvMajor,
+            info.drvVersion.drvMinor,
+            info.dilVersion.dilRelease, info.dilVersion.dilMajor,
+            info.dilVersion.dilMinor,
+            info.fwVersion.fwRelease,   info.fwVersion.fwMajor,
+            info.fwVersion.fwMinor );
     }
 #endif
 
@@ -254,8 +293,9 @@ static int OpenDecoder( vlc_object_t *p_this )
         }
     }
 
-    /* Always YUY2 color */
-    if( BC_FUNC(DtsSetColorSpace)( p_sys->bcm_handle, OUTPUT_MODE422_YUY2 ) != BC_STS_SUCCESS )
+    /* Always use YUY2 color */
+    if( BC_FUNC(DtsSetColorSpace)( p_sys->bcm_handle, OUTPUT_MODE422_YUY2 )
+            != BC_STS_SUCCESS )
     {
         msg_Err( p_dec, "Couldn't set the color space. Please report this!" );
         goto error;
@@ -280,7 +320,8 @@ static int OpenDecoder( vlc_object_t *p_this )
     }
 
     /* Open a decoder */
-    if( BC_FUNC(DtsOpenDecoder)( p_sys->bcm_handle, BC_STREAM_TYPE_ES ) != BC_STS_SUCCESS )
+    if( BC_FUNC(DtsOpenDecoder)( p_sys->bcm_handle, BC_STREAM_TYPE_ES )
+            != BC_STS_SUCCESS )
     {
         msg_Err( p_dec, "Couldn't open the CrystalHD decoder" );
         goto error;
@@ -345,6 +386,43 @@ error:
     free( p_sys );
 }
 
+static BC_STATUS ourCallback(void *shnd, uint32_t width, uint32_t height, uint32_t stride, void *pOut)
+{
+    VLC_UNUSED(width); VLC_UNUSED(height); VLC_UNUSED(stride);
+
+    decoder_t *p_dec          = (decoder_t *)shnd;
+    BC_DTS_PROC_OUT *proc_out = p_dec->p_sys->proc_out;
+    BC_DTS_PROC_OUT *proc_in  = (BC_DTS_PROC_OUT*)pOut;
+
+    /* Direct Rendering */
+    /* Do not allocate for the second-field in the pair, in interlaced */
+    if( !(proc_in->PicInfo.flags & VDEC_FLAG_INTERLACED_SRC) ||
+        !(proc_in->PicInfo.flags & VDEC_FLAG_FIELDPAIR) )
+        p_dec->p_sys->p_pic = decoder_NewPicture( p_dec );
+
+    /* */
+    picture_t *p_pic = p_dec->p_sys->p_pic;
+    if( !p_pic )
+        return BC_STS_ERROR;
+
+    /* Interlacing */
+    p_pic->b_progressive     = !(proc_in->PicInfo.flags & VDEC_FLAG_INTERLACED_SRC);
+    p_pic->b_top_field_first = !(proc_in->PicInfo.flags & VDEC_FLAG_BOTTOM_FIRST);
+    p_pic->i_nb_fields       = p_pic->b_progressive? 1: 2;
+
+    /* Filling out the struct */
+    proc_out->Ybuff      = !(proc_in->PicInfo.flags & VDEC_FLAG_FIELDPAIR) ?
+                             &p_pic->p[0].p_pixels[0] :
+                             &p_pic->p[0].p_pixels[p_pic->p[0].i_pitch];
+    proc_out->YbuffSz    = 2 * p_pic->p[0].i_pitch;
+    proc_out->StrideSz   = (proc_in->PicInfo.flags & VDEC_FLAG_INTERLACED_SRC)?
+                            2 * (p_pic->p[0].i_pitch/2) - p_dec->fmt_out.video.i_width:
+                            p_pic->p[0].i_pitch/2 - p_dec->fmt_out.video.i_width;
+    proc_out->PoutFlags |= BC_POUT_FLAGS_STRIDE;              /* Trust Stride info */
+
+    return BC_STS_SUCCESS;
+}
+
 /****************************************************************************
  * DecodeBlock: the whole thing
  ****************************************************************************/
@@ -356,8 +434,6 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
     BC_DTS_PROC_OUT proc_out;
     BC_DTS_STATUS driver_stat;
 
-    picture_t *p_pic;
-
     /* First check the status of the decode to produce pictures */
     if( BC_FUNC_PSYS(DtsGetDriverStatus)( p_sys->bcm_handle, &driver_stat ) != BC_STS_SUCCESS )
         return NULL;
@@ -397,16 +473,12 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
     memset( &proc_out, 0, sizeof(BC_DTS_PROC_OUT) );
     proc_out.PicInfo.width  = p_dec->fmt_out.video.i_width;
     proc_out.PicInfo.height = p_dec->fmt_out.video.i_height;
-    proc_out.YbuffSz        = p_dec->fmt_out.video.i_width * p_dec->fmt_out.video.i_height  / 2;
-    proc_out.Ybuff          = malloc( proc_out.YbuffSz  * 4);               // Allocate in bytes
-    proc_out.PoutFlags      = BC_POUT_FLAGS_SIZE;                           //FIXME why?
-
-#ifdef DEBUG_CRYSTALHD
-    msg_Dbg( p_dec, "%i, %i",  p_dec->fmt_out.video.i_width, p_dec->fmt_out.video.i_height );
-#endif
-    if( !proc_out.Ybuff )
-        return NULL;
+    proc_out.PoutFlags      = BC_POUT_FLAGS_SIZE;
+    proc_out.AppCallBack    = ourCallback;
+    proc_out.hnd            = p_dec;
+    p_sys->proc_out         = &proc_out;
 
+    /* */
     BC_STATUS sts = BC_FUNC_PSYS(DtsProcOutput)( p_sys->bcm_handle, 128, &proc_out );
 #ifdef DEBUG_CRYSTALHD
     if( sts != BC_STS_SUCCESS )
@@ -414,6 +486,7 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 #endif
 
     uint8_t b_eos;
+    picture_t *p_pic = p_sys->p_pic;
     switch( sts )
     {
         case BC_STS_SUCCESS:
@@ -423,17 +496,21 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
                 break;
             }
 
-            p_pic = decoder_NewPicture( p_dec );
             if( !p_pic )
                 break;
 
-            crystal_CopyPicture( p_pic, &proc_out );
-            p_pic->date = proc_out.PicInfo.timeStamp > 0 ? FROM_BC_PTS(proc_out.PicInfo.timeStamp) : VLC_TS_INVALID;
+            /* In interlaced mode, do not push the first field in the pipeline */
+            if( (proc_out.PicInfo.flags & VDEC_FLAG_INTERLACED_SRC) &&
+               !(proc_out.PicInfo.flags & VDEC_FLAG_FIELDPAIR) )
+                return NULL;
+
+            //  crystal_CopyPicture( p_pic, &proc_out );
+            p_pic->date = proc_out.PicInfo.timeStamp > 0 ?
+                          FROM_BC_PTS(proc_out.PicInfo.timeStamp) : VLC_TS_INVALID;
             //p_pic->date += 100 * 1000;
 #ifdef DEBUG_CRYSTALHD
             msg_Dbg( p_dec, "TS Output is %"PRIu64, p_pic->date);
 #endif
-            free( proc_out.Ybuff );
             return p_pic;
 
         case BC_STS_DEC_NOT_OPEN:
@@ -450,7 +527,10 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
                 break; */
             p_dec->fmt_out.video.i_width  = proc_out.PicInfo.width;
             p_dec->fmt_out.video.i_height = proc_out.PicInfo.height;
-#define setAR( a, b, c ) case a: p_dec->fmt_out.video.i_sar_num = b; p_dec->fmt_out.video.i_sar_den = c; break;
+            if( proc_out.PicInfo.height == 1088 )
+                p_dec->fmt_out.video.i_height = 1080;
+#define setAR( a, b, c ) case a: p_dec->fmt_out.video.i_sar_num = b; \
+                                 p_dec->fmt_out.video.i_sar_den = c; break;
             switch( proc_out.PicInfo.aspect_ratio )
             {
                 setAR( vdecAspectRatioSquare, 1, 1 )
@@ -474,12 +554,14 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 #undef setAR
             msg_Dbg( p_dec, "Format Change Detected [%i, %i], AR: %i/%i",
                     proc_out.PicInfo.width, proc_out.PicInfo.height,
-                    p_dec->fmt_out.video.i_sar_num, p_dec->fmt_out.video.i_sar_den );
+                    p_dec->fmt_out.video.i_sar_num,
+                    p_dec->fmt_out.video.i_sar_den );
             break;
 
         /* Nothing is documented here... */
         case BC_STS_NO_DATA:
-            if( BC_FUNC_PSYS(DtsIsEndOfStream)( p_sys->bcm_handle, &b_eos ) == BC_STS_SUCCESS )
+            if( BC_FUNC_PSYS(DtsIsEndOfStream)( p_sys->bcm_handle, &b_eos )
+                    == BC_STS_SUCCESS )
                 if( b_eos )
                     msg_Dbg( p_dec, "End of Stream" );
             break;
@@ -495,10 +577,12 @@ static picture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
             msg_Err( p_dec, "Unknown return status. Please report %i", sts );
             break;
     }
-    free( proc_out.Ybuff );
+    if( p_pic )
+        decoder_DeletePicture( p_dec, p_pic );
     return NULL;
 }
 
+#if 0
 /* Copy the data
  * FIXME: this should not exist */
 static void crystal_CopyPicture ( picture_t *p_pic, BC_DTS_PROC_OUT* p_out )
@@ -514,9 +598,12 @@ static void crystal_CopyPicture ( picture_t *p_pic, BC_DTS_PROC_OUT* p_out )
     for( ; p_dst < p_dst_end; p_dst += i_dst_stride, p_src += (p_out->PicInfo.width * 2))
         vlc_memcpy( p_dst, p_src, p_out->PicInfo.width * 2); // Copy in bytes
 }
+#endif
 
 /* Parse the SPS/PPS Metadata to feed the decoder for avc1 */
-static int crystal_insert_sps_pps(decoder_t *p_dec, uint8_t *p_buf, uint32_t i_buf_size)
+static int crystal_insert_sps_pps( decoder_t *p_dec,
+                                   uint8_t *p_buf,
+                                   uint32_t i_buf_size)
 {
     decoder_sys_t *p_sys = p_dec->p_sys;
     int i_profile;
@@ -539,14 +626,16 @@ static int crystal_insert_sps_pps(decoder_t *p_dec, uint8_t *p_buf, uint32_t i_b
     /* Read infos in first 6 bytes */
     i_profile         = (p_buf[1] << 16) | (p_buf[2] << 8) | p_buf[3];
     p_sys->i_nal_size = (p_buf[4] & 0x03) + 1;
-    p_buf += 5;
-    i_data_size -= 5;
+    p_buf            += 5;
+    i_data_size      -= 5;
 
     for ( unsigned int j = 0; j < 2; j++ )
     {
         /* First time is SPS, Second is PPS */
-        if (i_data_size < 1) {
-            msg_Err( p_dec, "PPS too small after processing SPS/PPS %u", i_data_size );
+        if( i_data_size < 1 )
+        {
+            msg_Err( p_dec, "PPS too small after processing SPS/PPS %u",
+                    i_data_size );
             goto error;
         }
         i_loop_end = p_buf[0] & (j == 0 ? 0x1f : 0xff);
@@ -554,7 +643,8 @@ static int crystal_insert_sps_pps(decoder_t *p_dec, uint8_t *p_buf, uint32_t i_b
 
         for ( unsigned int i = 0; i < i_loop_end; i++)
         {
-            if (i_data_size < 2 ) {
+            if( i_data_size < 2 )
+            {
                 msg_Err( p_dec, "SPS is too small %u", i_data_size );
                 goto error;
             }
@@ -563,8 +653,10 @@ static int crystal_insert_sps_pps(decoder_t *p_dec, uint8_t *p_buf, uint32_t i_b
             p_buf += 2;
             i_data_size -= 2;
 
-            if (i_data_size < i_nal_size ) {
-                msg_Err( p_dec, "SPS size does not match NAL specified size %u", i_data_size );
+            if( i_data_size < i_nal_size )
+            {
+                msg_Err( p_dec, "SPS size does not match NAL specified size %u",
+                        i_data_size );
                 goto error;
             }
 
@@ -573,7 +665,7 @@ static int crystal_insert_sps_pps(decoder_t *p_dec, uint8_t *p_buf, uint32_t i_b
             p_sys->p_sps_pps_buf[p_sys->i_sps_pps_size++] = 0;
             p_sys->p_sps_pps_buf[p_sys->i_sps_pps_size++] = 1;
 
-            memcpy(p_sys->p_sps_pps_buf + p_sys->i_sps_pps_size, p_buf, i_nal_size);
+            memcpy( p_sys->p_sps_pps_buf + p_sys->i_sps_pps_size, p_buf, i_nal_size );
             p_sys->i_sps_pps_size += i_nal_size;
 
             p_buf += i_nal_size;