]> git.sesse.net Git - mlt/blobdiff - src/framework/mlt_frame.c
Add mlt_frame_write_ppm to visualize debugging.
[mlt] / src / framework / mlt_frame.c
index e174160ae375df935992f51b20100f5d703dec13..76afc2208c806fab12a65cf05014a65c46d333ba 100644 (file)
 #include <stdlib.h>
 #include <string.h>
 
-/** Constructor for a frame.
-*/
+/** Construct a frame object.
+ *
+ * \public \memberof mlt_frame_s
+ * \param service the pointer to any service that can provide access to the profile
+ * \return a frame object on success or NULL if there was an allocation error
+ */
 
 mlt_frame mlt_frame_init( mlt_service service )
 {
@@ -67,48 +71,73 @@ mlt_frame mlt_frame_init( mlt_service service )
        return this;
 }
 
-/** Fetch the frames properties.
-*/
+/** Get a frame's properties.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return the frame's properties or NULL if an invalid frame is supplied
+ */
 
 mlt_properties mlt_frame_properties( mlt_frame this )
 {
        return this != NULL ? &this->parent : NULL;
 }
 
-/** Check if we have a way to derive something other than a test card.
-*/
+/** Determine if the frame will produce a test card image.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return true (non-zero) if this will produce from a test card
+ */
 
 int mlt_frame_is_test_card( mlt_frame this )
 {
        return mlt_deque_count( this->stack_image ) == 0 || mlt_properties_get_int( MLT_FRAME_PROPERTIES( this ), "test_image" );
 }
 
-/** Check if we have a way to derive something other than test audio.
-*/
+/** Determine if the frame will produce audio from a test card.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return true (non-zero) if this will produce from a test card
+ */
 
 int mlt_frame_is_test_audio( mlt_frame this )
 {
        return mlt_deque_count( this->stack_audio ) == 0 || mlt_properties_get_int( MLT_FRAME_PROPERTIES( this ), "test_audio" );
 }
 
-/** Get the aspect ratio of the frame.
-*/
+/** Get the sample aspect ratio of the frame.
+ *
+ * \public \memberof  mlt_frame_s
+ * \param this a frame
+ * \return the aspect ratio
+ */
 
 double mlt_frame_get_aspect_ratio( mlt_frame this )
 {
        return mlt_properties_get_double( MLT_FRAME_PROPERTIES( this ), "aspect_ratio" );
 }
 
-/** Set the aspect ratio of the frame.
-*/
+/** Set the sample aspect ratio of the frame.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param value the new image sample aspect ratio
+ * \return true if error
+ */
 
 int mlt_frame_set_aspect_ratio( mlt_frame this, double value )
 {
        return mlt_properties_set_double( MLT_FRAME_PROPERTIES( this ), "aspect_ratio", value );
 }
 
-/** Get the position of this frame.
-*/
+/** Get the time position of this frame.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return the position
+ */
 
 mlt_position mlt_frame_get_position( mlt_frame this )
 {
@@ -116,8 +145,13 @@ mlt_position mlt_frame_get_position( mlt_frame this )
        return pos < 0 ? 0 : pos;
 }
 
-/** Set the position of this frame.
-*/
+/** Set the time position of this frame.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param value the position
+ * \return true if error
+ */
 
 int mlt_frame_set_position( mlt_frame this, mlt_position value )
 {
@@ -125,7 +159,12 @@ int mlt_frame_set_position( mlt_frame this, mlt_position value )
 }
 
 /** Stack a get_image callback.
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param the get_image callback
+ * \return true if error
+ */
 
 int mlt_frame_push_get_image( mlt_frame this, mlt_get_image get_image )
 {
@@ -133,7 +172,11 @@ int mlt_frame_push_get_image( mlt_frame this, mlt_get_image get_image )
 }
 
 /** Pop a get_image callback.
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return the get_image callback
+ */
 
 mlt_get_image mlt_frame_pop_get_image( mlt_frame this )
 {
@@ -141,7 +184,12 @@ mlt_get_image mlt_frame_pop_get_image( mlt_frame this )
 }
 
 /** Push a frame.
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param that the frame to push onto \p this
+ * \return true if error
+ */
 
 int mlt_frame_push_frame( mlt_frame this, mlt_frame that )
 {
@@ -149,7 +197,11 @@ int mlt_frame_push_frame( mlt_frame this, mlt_frame that )
 }
 
 /** Pop a frame.
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return a frame that was previously pushed
+ */
 
 mlt_frame mlt_frame_pop_frame( mlt_frame this )
 {
@@ -157,7 +209,12 @@ mlt_frame mlt_frame_pop_frame( mlt_frame this )
 }
 
 /** Push a service.
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param that an opaque pointer
+ * \return true if error
+ */
 
 int mlt_frame_push_service( mlt_frame this, void *that )
 {
@@ -165,23 +222,36 @@ int mlt_frame_push_service( mlt_frame this, void *that )
 }
 
 /** Pop a service.
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return an opaque pointer to something previously pushed
+ */
 
 void *mlt_frame_pop_service( mlt_frame this )
 {
        return mlt_deque_pop_back( this->stack_image );
 }
 
-/** Push a service.
-*/
+/** Push a number.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param that an integer
+ * \return true if error
+ */
 
 int mlt_frame_push_service_int( mlt_frame this, int that )
 {
        return mlt_deque_push_back_int( this->stack_image, that );
 }
 
-/** Pop a service.
-*/
+/** Pop a number.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return an integer that was previously pushed
+ */
 
 int mlt_frame_pop_service_int( mlt_frame this )
 {
@@ -189,7 +259,12 @@ int mlt_frame_pop_service_int( mlt_frame this )
 }
 
 /** Push an audio item on the stack.
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param that an opaque pointer
+ * \return true if error
+ */
 
 int mlt_frame_push_audio( mlt_frame this, void *that )
 {
@@ -197,7 +272,11 @@ int mlt_frame_push_audio( mlt_frame this, void *that )
 }
 
 /** Pop an audio item from the stack
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return an opaque pointer to something that was pushed onto the frame's audio stack
+ */
 
 void *mlt_frame_pop_audio( mlt_frame this )
 {
@@ -205,7 +284,11 @@ void *mlt_frame_pop_audio( mlt_frame this )
 }
 
 /** Return the service stack
-*/
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return the service stack
+ */
 
 mlt_deque mlt_frame_service_stack( mlt_frame this )
 {
@@ -213,23 +296,30 @@ mlt_deque mlt_frame_service_stack( mlt_frame this )
 }
 
 /** Replace image stack with the information provided.
-
-       This might prove to be unreliable and restrictive - the idea is that a transition
-       which normally uses two images may decide to only use the b frame (ie: in the case
-       of a composite where the b frame completely obscures the a frame).
-
-       The image must be writable and the destructor for the image itself must be taken
-       care of on another frame and that frame cannot have a replace applied to it...
-       Further it assumes that no alpha mask is in use.
-
-       For these reasons, it can only be used in a specific situation - when you have
-       multiple tracks each with their own transition and these transitions are applied
-       in a strictly reversed order (ie: highest numbered [lowest track] is processed
-       first).
-
-       More reliable approach - the cases should be detected during the process phase
-       and the upper tracks should simply not be invited to stack...
-*/
+ *
+ * This might prove to be unreliable and restrictive - the idea is that a transition
+ * which normally uses two images may decide to only use the b frame (ie: in the case
+ * of a composite where the b frame completely obscures the a frame).
+ *
+ * The image must be writable and the destructor for the image itself must be taken
+ * care of on another frame and that frame cannot have a replace applied to it...
+ * Further it assumes that no alpha mask is in use.
+ *
+ * For these reasons, it can only be used in a specific situation - when you have
+ * multiple tracks each with their own transition and these transitions are applied
+ * in a strictly reversed order (ie: highest numbered [lowest track] is processed
+ * first).
+ *
+ * More reliable approach - the cases should be detected during the process phase
+ * and the upper tracks should simply not be invited to stack...
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param image a new image
+ * \param format the image format
+ * \param width the width of the new image
+ * \param height the height of the new image
+ */
 
 void mlt_frame_replace_image( mlt_frame this, uint8_t *image, mlt_image_format format, int width, int height )
 {
@@ -244,6 +334,13 @@ void mlt_frame_replace_image( mlt_frame this, uint8_t *image, mlt_image_format f
        this->get_alpha_mask = NULL;
 }
 
+/** Get the short name for an image format.
+ *
+ * \public \memberof mlt_frame_s
+ * \param format the image format
+ * \return a string
+ */
+
 const char * mlt_image_format_name( mlt_image_format format )
 {
        switch ( format )
@@ -259,7 +356,24 @@ const char * mlt_image_format_name( mlt_image_format format )
 }
 
 /** Get the image associated to the frame.
-*/
+ *
+ * You should express the desired format, width, and height as inputs. As long
+ * as the loader producer was used to generate this or the imageconvert filter
+ * was attached, then you will get the image back in the format you desire.
+ * However, you do not always get the width and height you request depending
+ * on properties and filters. You do not need to supply a pre-allocated
+ * buffer, but you should always supply the desired image format.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param[out] buffer an image buffer
+ * \param[in,out] format the image format
+ * \param[in,out] width the horizontal size in pixels
+ * \param[in,out] height the vertical size in pixels
+ * \param writable whether or not you will need to be able to write to the memory returned in \p buffer
+ * \return true if error
+ * \todo Better describe the width and height as inputs.
+ */
 
 int mlt_frame_get_image( mlt_frame this, uint8_t **buffer, mlt_image_format *format, int *width, int *height, int writable )
 {
@@ -273,11 +387,19 @@ int mlt_frame_get_image( mlt_frame this, uint8_t **buffer, mlt_image_format *for
        {
                mlt_properties_set_int( properties, "image_count", mlt_properties_get_int( properties, "image_count" ) - 1 );
                error = get_image( this, buffer, format, width, height, writable );
-               mlt_properties_set_int( properties, "width", *width );
-               mlt_properties_set_int( properties, "height", *height );
-               mlt_properties_set_int( properties, "format", *format );
-               if ( this->convert_image )
-                       this->convert_image( this, buffer, format, requested_format );
+               if ( !error && *buffer )
+               {
+                       mlt_properties_set_int( properties, "width", *width );
+                       mlt_properties_set_int( properties, "height", *height );
+                       mlt_properties_set_int( properties, "format", *format );
+                       if ( this->convert_image )
+                               this->convert_image( this, buffer, format, requested_format );
+               }
+               else
+               {
+                       // Cause the image to be loaded from test card or fallback (white) below.
+                       mlt_frame_get_image( this, buffer, format, width, height, writable );
+               }
        }
        else if ( mlt_properties_get_data( properties, "image", NULL ) )
        {
@@ -285,7 +407,7 @@ int mlt_frame_get_image( mlt_frame this, uint8_t **buffer, mlt_image_format *for
                *buffer = mlt_properties_get_data( properties, "image", NULL );
                *width = mlt_properties_get_int( properties, "width" );
                *height = mlt_properties_get_int( properties, "height" );
-               if ( this->convert_image )
+               if ( this->convert_image && *buffer )
                        this->convert_image( this, buffer, format, requested_format );
        }
        else if ( producer )
@@ -371,12 +493,16 @@ int mlt_frame_get_image( mlt_frame this, uint8_t **buffer, mlt_image_format *for
                mlt_properties_set_int( properties, "test_image", 1 );
        }
 
-       mlt_properties_set_int( properties, "scaled_width", *width );
-       mlt_properties_set_int( properties, "scaled_height", *height );
-
        return error;
 }
 
+/** Get the alpha channel associated to the frame.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return the alpha channel
+ */
+
 uint8_t *mlt_frame_get_alpha_mask( mlt_frame this )
 {
        uint8_t *alpha = NULL;
@@ -388,7 +514,7 @@ uint8_t *mlt_frame_get_alpha_mask( mlt_frame this )
                        alpha = mlt_properties_get_data( &this->parent, "alpha", NULL );
                if ( alpha == NULL )
                {
-                       int size = mlt_properties_get_int( &this->parent, "scaled_width" ) * mlt_properties_get_int( &this->parent, "scaled_height" );
+                       int size = mlt_properties_get_int( &this->parent, "width" ) * mlt_properties_get_int( &this->parent, "height" );
                        alpha = mlt_pool_alloc( size );
                        memset( alpha, 255, size );
                        mlt_properties_set_data( &this->parent, "alpha", alpha, size, mlt_pool_release, NULL );
@@ -397,6 +523,15 @@ uint8_t *mlt_frame_get_alpha_mask( mlt_frame this )
        return alpha;
 }
 
+/** Get the short name for an audio format.
+ *
+ * You do not need to deallocate the returned string.
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param format an image format enum
+ * \return a string for the name of the image format
+ */
+
 const char * mlt_audio_format_name( mlt_audio_format format )
 {
        switch ( format )
@@ -409,6 +544,27 @@ const char * mlt_audio_format_name( mlt_audio_format format )
        return "invalid";
 }
 
+/** Get the audio associated to the frame.
+ *
+ * You should express the desired format, frequency, channels, and samples as inputs. As long
+ * as the loader producer was used to generate this or the audioconvert filter
+ * was attached, then you will get the audio back in the format you desire.
+ * However, you do not always get the channels and samples you request depending
+ * on properties and filters. You do not need to supply a pre-allocated
+ * buffer, but you should always supply the desired audio format.
+ * The audio is always in interleaved format.
+ * You should use the \p mlt_sample_calculator to determine the number of samples you want.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param[out] buffer an audio buffer
+ * \param[in,out] format the audio format
+ * \param[in,out] frequency the sample rate
+ * \param[in,out] channels
+ * \param[in,out] samples the number of samples per frame
+ * \return true if error
+ */
+
 int mlt_frame_get_audio( mlt_frame this, void **buffer, mlt_audio_format *format, int *frequency, int *channels, int *samples )
 {
        mlt_get_audio get_audio = mlt_frame_pop_audio( this );
@@ -497,22 +653,55 @@ int mlt_frame_get_audio( mlt_frame this, void **buffer, mlt_audio_format *format
        return 0;
 }
 
+/** Set the audio on a frame.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param buffer an buffer containing audio samples
+ * \param format the format of the audio in the \p buffer
+ * \param size the total size of the buffer (optional)
+ * \param destructor a function that releases or deallocates the \p buffer
+ * \return true if error
+ */
+
 int mlt_frame_set_audio( mlt_frame this, void *buffer, mlt_audio_format format, int size, mlt_destructor destructor )
 {
        mlt_properties_set_int( MLT_FRAME_PROPERTIES( this ), "audio_format", format );
        return mlt_properties_set_data( MLT_FRAME_PROPERTIES( this ), "audio", buffer, size, destructor, NULL );
 }
 
+/** Get audio on a frame as a waveform image.
+ *
+ * This generates an 8-bit grayscale image representation of the audio in a
+ * frame. Currently, this only really works for 2 channels.
+ * This allocates the bitmap using mlt_pool so you should release the return
+ * value with \p mlt_pool_release.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \param w the width of the image
+ * \param h the height of the image to create
+ * \return a pointer to a new bitmap
+ */
+
 unsigned char *mlt_frame_get_waveform( mlt_frame this, int w, int h )
 {
        int16_t *pcm = NULL;
        mlt_properties properties = MLT_FRAME_PROPERTIES( this );
        mlt_audio_format format = mlt_audio_s16;
-       int frequency = 32000; // lower frequency available?
+       int frequency = 16000;
        int channels = 2;
-       double fps = mlt_profile_fps( NULL );
+       mlt_producer producer = mlt_frame_get_original_producer( this );
+       double fps = mlt_producer_get_fps( mlt_producer_cut_parent( producer ) );
        int samples = mlt_sample_calculator( fps, frequency, mlt_frame_get_position( this ) );
 
+       // Increase audio resolution proportional to requested image size
+       while ( samples < w )
+       {
+               frequency += 16000;
+               samples = mlt_sample_calculator( fps, frequency, mlt_frame_get_position( this ) );
+       }
+
        // Get the pcm data
        mlt_frame_get_audio( this, (void**)&pcm, &format, &frequency, &channels, &samples );
 
@@ -525,38 +714,49 @@ unsigned char *mlt_frame_get_waveform( mlt_frame this, int w, int h )
 
        // Render vertical lines
        int16_t *ubound = pcm + samples * channels;
-       int skip = samples / w - 1;
+       int skip = samples / w;
+       skip = !skip ? 1 : skip;
+       unsigned char gray = 0xFF / skip;
        int i, j, k;
 
        // Iterate sample stream and along x coordinate
-       for ( i = 0; i < w && pcm < ubound; i++ )
+       for ( i = 0; pcm < ubound; i++ )
        {
                // pcm data has channels interleaved
-               for ( j = 0; j < channels; j++ )
+               for ( j = 0; j < channels; j++, pcm++ )
                {
                        // Determine sample's magnitude from 2s complement;
                        int pcm_magnitude = *pcm < 0 ? ~(*pcm) + 1 : *pcm;
                        // The height of a line is the ratio of the magnitude multiplied by
-                       // half the vertical resolution
-                       int height = ( int )( ( double )( pcm_magnitude ) / 32768 * h / 2 );
-                       // Determine the starting y coordinate - left channel above center,
-                       // right channel below - currently assumes 2 channels
-                       int displacement = ( h / 2 ) - ( 1 - j ) * height;
+                       // the vertical resolution of a single channel
+                               int height = h * pcm_magnitude / channels / 2 / 32768;
+                       // Determine the starting y coordinate - left top, right bottom
+                       int displacement = h * (j * 2 + 1) / channels / 2 - ( *pcm < 0 ? 0 : height );
                        // Position buffer pointer using y coordinate, stride, and x coordinate
-                       unsigned char *p = &bitmap[ i + displacement * w ];
+                       unsigned char *p = bitmap + i / skip + displacement * w;
 
                        // Draw vertical line
-                       for ( k = 0; k < height; k++ )
-                               p[ w * k ] = 0xFF;
-
-                       pcm++;
+                       for ( k = 0; k < height + 1; k++ )
+                               if ( *pcm < 0 )
+                                       p[ w * k ] = ( k == 0 ) ? 0xFF : p[ w * k ] + gray;
+                               else
+                                       p[ w * k ] = ( k == height ) ? 0xFF : p[ w * k ] + gray;
                }
-               pcm += skip * channels;
        }
 
        return bitmap;
 }
 
+/** Get the end service that produced this frame.
+ *
+ * This fetches the first producer of the frame and not any producers that
+ * encapsulate it.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ * \return a producer
+ */
+
 mlt_producer mlt_frame_get_original_producer( mlt_frame this )
 {
        if ( this != NULL )
@@ -564,6 +764,12 @@ mlt_producer mlt_frame_get_original_producer( mlt_frame this )
        return NULL;
 }
 
+/** Destroy the frame.
+ *
+ * \public \memberof mlt_frame_s
+ * \param this a frame
+ */
+
 void mlt_frame_close( mlt_frame this )
 {
        if ( this != NULL && mlt_properties_dec_ref( MLT_FRAME_PROPERTIES( this ) ) <= 0 )
@@ -580,42 +786,69 @@ void mlt_frame_close( mlt_frame this )
 
 /***** convenience functions *****/
 
-/* Will this break when mlt_position is converted to double? -Zach */
+/** Determine the number of samples that belong in a frame at a time position.
+ *
+ * \public \memberof mlt_frame_s
+ * \param fps the frame rate
+ * \param frequency the sample rate
+ * \param position the time position
+ * \return the number of samples per channel
+ */
+
 int mlt_sample_calculator( float fps, int frequency, int64_t position )
 {
-       int samples = 0;
+       /* Compute the cumulative number of samples until the start of this frame and the
+       cumulative number of samples until the start of the next frame. Round each to the
+       nearest integer and take the difference to determine the number of samples in
+       this frame.
+
+       This approach should prevent rounding errors that can accumulate over a large number
+       of frames causing A/V sync problems. */
+       return mlt_sample_calculator_to_now( fps, frequency, position + 1 )
+                - mlt_sample_calculator_to_now( fps, frequency, position );
+}
+
+/** Determine the number of samples that belong before a time position.
+ *
+ * \public \memberof mlt_frame_s
+ * \param fps the frame rate
+ * \param frequency the sample rate
+ * \param position the time position
+ * \return the number of samples per channel
+ * \bug Will this break when mlt_position is converted to double?
+ */
+
+int64_t mlt_sample_calculator_to_now( float fps, int frequency, int64_t position )
+{
+       int64_t samples = 0;
 
        if ( fps )
        {
-               /* Compute the cumulative number of samples until the start of this frame and the
-               cumulative number of samples until the start of the next frame. Round each to the
-               nearest integer and take the difference to determine the number of samples in
-               this frame.
-
-               This approach should prevent rounding errors that can accumulate over a large number
-               of frames causing A/V sync problems. */
-
-               int64_t samples_at_this =
-                       (int64_t)( (double) position * (double) frequency / (double) fps +
-                       ( position < 0 ? -0.5 : 0.5 ) );
-               int64_t samples_at_next =
-                       (int64_t)( (double) (position + 1) * (double) frequency / (double) fps +
+               samples = (int64_t)( (double) position * (double) frequency / (double) fps +
                        ( position < 0 ? -0.5 : 0.5 ) );
-               samples = (int)( samples_at_next - samples_at_this );
        }
 
        return samples;
 }
 
-int64_t mlt_sample_calculator_to_now( float fps, int frequency, int64_t frame )
+void mlt_frame_write_ppm( mlt_frame frame )
 {
-       int64_t samples = 0;
-
-       if ( fps )
+       int width;
+       int height;
+       mlt_image_format format = mlt_image_rgb24;
+       uint8_t *image;
+       
+       if ( mlt_frame_get_image( frame, &image, &format, &width, &height, 0 ) == 0 )
        {
-               samples = (int64_t)( (double) frame * (double) frequency / (double) fps +
-                       ( frame < 0 ? -0.5 : 0.5 ) );
+               FILE *file;
+               char filename[16];
+               
+               sprintf( filename, "frame-%05d.ppm", mlt_frame_get_position( frame ) );
+               file = fopen( filename, "wb" );
+               if ( !file )
+                       return;
+               fprintf( file, "P6\n%d %d\n255\n", width, height);
+               fwrite( image, width * height * 3, 1, file );
+               fclose( file );
        }
-
-       return samples;
 }