]> git.sesse.net Git - mlt/blobdiff - src/framework/mlt_profile.c
Restrict auto-profile frame rate to a sane value.
[mlt] / src / framework / mlt_profile.c
index 3baab0efc3345689d6fd845c8eed86a15779e8fd..4385d7a789be5bf857c1d0b05f170256dce057f0 100644 (file)
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */
 
-#include "mlt_profile.h"
-#include "mlt_factory.h"
-#include "mlt_properties.h"
+#include "mlt.h"
 
 #include <stdlib.h>
 #include <string.h>
 #include <libgen.h>
+#include <math.h>
 
 
-/** the default subdirectory of the prefix for holding profiles */
-#define PROFILES_DIR "/share/mlt/profiles/"
+/** the default subdirectory of the datadir for holding profiles */
+#define PROFILES_DIR "/profiles/"
 
 /** Load a profile from the system folder.
  *
@@ -55,14 +54,12 @@ static mlt_profile mlt_profile_select( const char *name )
        {
                filename = calloc( 1, strlen( name ) + 1 );
        }
-       // Load from $prefix/share/mlt/profiles
+       // Load from $datadir/mlt/profiles
        else if ( prefix == NULL )
        {
-               prefix = PREFIX;
-               filename = calloc( 1, strlen( prefix ) + strlen( PROFILES_DIR ) + strlen( name ) + 2 );
+               prefix = mlt_environment( "MLT_DATA" );
+               filename = calloc( 1, strlen( prefix ) + strlen( PROFILES_DIR ) + strlen( name ) + 1 );
                strcpy( filename, prefix );
-               if ( filename[ strlen( filename ) - 1 ] != '/' )
-                       filename[ strlen( filename ) ] = '/';
                strcat( filename, PROFILES_DIR );
        }
        // Use environment variable instead
@@ -90,9 +87,9 @@ static mlt_profile mlt_profile_select( const char *name )
  * This will never return NULL as it uses the dv_pal settings as hard-coded fallback default.
  *
  * \public \memberof mlt_profile_s
- * @param name the name of a profile settings file located in the standard location or
+ * \param name the name of a profile settings file located in the standard location or
  * the full path name to a profile settings file
- * @return a profile
+ * \return a profile
  */
 
 mlt_profile mlt_profile_init( const char *name )
@@ -163,8 +160,8 @@ static void set_mlt_normalisation( const char *profile_name )
 /** Load a profile from specific file.
  *
  * \public \memberof mlt_profile_s
- * @param file the full path name to a properties file
- * @return a profile or NULL on error
+ * \param file the full path name to a properties file
+ * \return a profile or NULL on error
  */
 
 mlt_profile mlt_profile_load_file( const char *file )
@@ -198,8 +195,8 @@ mlt_profile mlt_profile_load_file( const char *file )
 /** Load a profile from a properties object.
  *
  * \public \memberof mlt_profile_s
- * @param properties a properties list
- * @return a profile or NULL if out of memory
+ * \param properties a properties list
+ * \return a profile or NULL if out of memory
  */
 
 mlt_profile mlt_profile_load_properties( mlt_properties properties )
@@ -228,13 +225,15 @@ mlt_profile mlt_profile_load_properties( mlt_properties properties )
 /** Load an anonymous profile from string.
  *
  * \public \memberof mlt_profile_s
- * @param string a newline-delimited list of properties as name=value pairs
- * @return a profile or NULL if out of memory
+ * \param string a newline-delimited list of properties as name=value pairs
+ * \return a profile or NULL if out of memory
  */
 
 mlt_profile mlt_profile_load_string( const char *string )
 {
        mlt_properties properties = mlt_properties_new();
+       mlt_profile profile = NULL;
+
        if ( properties )
        {
                const char *p = string;
@@ -245,21 +244,23 @@ mlt_profile mlt_profile_load_string( const char *string )
                        p = strchr( p, '\n' );
                        if ( p ) p++;
                }
+               profile = mlt_profile_load_properties( properties );
+               mlt_properties_close( properties );
        }
-       return mlt_profile_load_properties( properties );
+       return profile;
 }
 
 /** Get the video frame rate as a floating point value.
  *
  * \public \memberof mlt_profile_s
- * @param aprofile a profile
+ * @param profile a profile
  * @return the frame rate
  */
 
-double mlt_profile_fps( mlt_profile aprofile )
+double mlt_profile_fps( mlt_profile profile )
 {
-       if ( aprofile )
-               return ( double ) aprofile->frame_rate_num / aprofile->frame_rate_den;
+       if ( profile )
+               return ( double ) profile->frame_rate_num / profile->frame_rate_den;
        else
                return 0;
 }
@@ -267,14 +268,14 @@ double mlt_profile_fps( mlt_profile aprofile )
 /** Get the sample aspect ratio as a floating point value.
  *
  * \public \memberof mlt_profile_s
- * @param aprofile a profile
- * @return the pixel aspect ratio
+ * \param profile a profile
+ * \return the pixel aspect ratio
  */
 
-double mlt_profile_sar( mlt_profile aprofile )
+double mlt_profile_sar( mlt_profile profile )
 {
-       if ( aprofile )
-               return ( double ) aprofile->sample_aspect_num / aprofile->sample_aspect_den;
+       if ( profile )
+               return ( double ) profile->sample_aspect_num / profile->sample_aspect_den;
        else
                return 0;
 }
@@ -282,14 +283,14 @@ double mlt_profile_sar( mlt_profile aprofile )
 /** Get the display aspect ratio as floating point value.
  *
  * \public \memberof mlt_profile_s
- * @param aprofile a profile
- * @return the image aspect ratio
+ * \param profile a profile
+ * \return the image aspect ratio
  */
 
-double mlt_profile_dar( mlt_profile aprofile )
+double mlt_profile_dar( mlt_profile profile )
 {
-       if ( aprofile )
-               return ( double ) aprofile->display_aspect_num / aprofile->display_aspect_den;
+       if ( profile )
+               return ( double ) profile->display_aspect_num / profile->display_aspect_den;
        else
                return 0;
 }
@@ -297,7 +298,7 @@ double mlt_profile_dar( mlt_profile aprofile )
 /** Free up the global profile resources.
  *
  * \public \memberof mlt_profile_s
- * @param profile a profile
+ * \param profile a profile
  */
 
 void mlt_profile_close( mlt_profile profile )
@@ -334,3 +335,121 @@ mlt_profile mlt_profile_clone( mlt_profile profile )
        }
        return clone;
 }
+
+
+/** Get the list of profiles.
+ *
+ * The caller MUST close the returned properties object!
+ * Each entry in the list is keyed on its name, and its value is another
+ * properties object that contains the attributes of the profile.
+ * \public \memberof mlt_profile_s
+ * \return a list of profiles
+ */
+
+mlt_properties mlt_profile_list( )
+{
+       char *filename = NULL;
+       const char *prefix = getenv( "MLT_PROFILES_PATH" );
+       mlt_properties properties = mlt_properties_new();
+       mlt_properties dir = mlt_properties_new();
+       int sort = 1;
+       const char *wildcard = NULL;
+       int i;
+
+       // Load from $datadir/mlt/profiles if no env var
+       if ( prefix == NULL )
+       {
+               prefix = mlt_environment( "MLT_DATA" );
+               filename = calloc( 1, strlen( prefix ) + strlen( PROFILES_DIR ) + 1 );
+               strcpy( filename, prefix );
+               strcat( filename, PROFILES_DIR );
+               prefix = filename;
+       }
+
+       mlt_properties_dir_list( dir, prefix, wildcard, sort );
+
+       for ( i = 0; i < mlt_properties_count( dir ); i++ )
+       {
+               char *filename = mlt_properties_get_value( dir, i );
+               char *profile_name = basename( filename );
+               if ( profile_name[0] != '.' && strcmp( profile_name, "Makefile" ) &&
+                    profile_name[ strlen( profile_name ) - 1 ] != '~' )
+               {
+                       mlt_properties profile = mlt_properties_load( filename );
+                       if ( profile )
+                       {
+                               mlt_properties_set_data( properties, profile_name, profile, 0,
+                                       (mlt_destructor) mlt_properties_close, NULL );
+                       }
+               }
+       }
+       mlt_properties_close( dir );
+       if ( filename )
+               free( filename );
+
+       return properties;
+}
+
+/** Update the profile using the attributes of a producer.
+ *
+ * Use this to make an "auto-profile." Typically, you need to re-open the producer
+ * after you use this because some producers (e.g. avformat) adjust their framerate
+ * to that of the profile used when you created it.
+ * \public \memberof mlt_profile_s
+ * \param profile the profile to update
+ * \param producer the producer to inspect
+ */
+
+void mlt_profile_from_producer( mlt_profile profile, mlt_producer producer )
+{
+       mlt_frame fr = NULL;
+       uint8_t *buffer = NULL;
+       mlt_image_format fmt = mlt_image_none;
+       mlt_properties p;
+       int w = profile->width;
+       int h = profile->height;
+
+       if ( ! mlt_service_get_frame( MLT_PRODUCER_SERVICE(producer), &fr, 0 ) && fr )
+       {
+               if ( ! mlt_frame_get_image( fr, &buffer, &fmt, &w, &h, 0 ) )
+               {
+                       // Some source properties are not exposed until after the first get_image call.
+                       mlt_frame_close( fr );
+                       mlt_service_get_frame( MLT_PRODUCER_SERVICE(producer), &fr, 0 );
+                       p = MLT_FRAME_PROPERTIES( fr );
+//                     mlt_properties_dump(p, stderr);
+                       if ( mlt_properties_get_int( p, "meta.media.frame_rate_den" ) &&
+                                mlt_properties_get_int( p, "meta.media.sample_aspect_den" ) )
+                       {
+                               profile->width = mlt_properties_get_int( p, "meta.media.width" );
+                               profile->height = mlt_properties_get_int( p, "meta.media.height" );
+                               profile->progressive = mlt_properties_get_int( p, "meta.media.progressive" );
+                               if ( 1000 > mlt_properties_get_double( p, "meta.media.frame_rate_num" )
+                                         / mlt_properties_get_double( p, "meta.media.frame_rate_den" ) )
+                               {
+                                       profile->frame_rate_num = mlt_properties_get_int( p, "meta.media.frame_rate_num" );
+                                       profile->frame_rate_den = mlt_properties_get_int( p, "meta.media.frame_rate_den" );
+                               } else {
+                                       profile->frame_rate_num = 60;
+                                       profile->frame_rate_den = 1;
+                               }
+                               // AVCHD is mis-reported as double frame rate.
+                               if ( profile->progressive == 0 && (
+                                    profile->frame_rate_num / profile->frame_rate_den == 50 ||
+                                    profile->frame_rate_num / profile->frame_rate_den == 59 ) )
+                                       profile->frame_rate_num /= 2;
+                               profile->sample_aspect_num = mlt_properties_get_int( p, "meta.media.sample_aspect_num" );
+                               profile->sample_aspect_den = mlt_properties_get_int( p, "meta.media.sample_aspect_den" );
+                               profile->colorspace = mlt_properties_get_int( p, "meta.media.colorspace" );
+                               profile->display_aspect_num = lrint( (double) profile->sample_aspect_num * profile->width
+                                       / profile->sample_aspect_den );
+                               profile->display_aspect_den = profile->height;
+                               free( profile->description );
+                               profile->description = strdup( "automatic" );
+                               profile->is_explicit = 0;
+                       }
+               }
+       }
+       mlt_frame_close( fr );
+       mlt_producer_seek( producer, 0 );
+}