Expose builtin models

libvmaf/include/libvmaf/model.h

@@ -70,6 +70,22 @@ typedef struct VmafModelCollectionScore {
     } bootstrap;
 } VmafModelCollectionScore;
 
+typedef struct VmafBuiltInModel {
+    const char *version;
+    const char *data;
+    const int *data_len;
+} VmafBuiltInModel;
+
+/**
+ * Iterate through all built in vmaf models
+ * 
+ * @param prev previous model. NULL to get the first model
+ * 
+ * @return next model or NULL after the last model
+ * 
+*/
+const VmafBuiltInModel *vmaf_built_in_model_next(const VmafBuiltInModel *prev);
+
 int vmaf_model_collection_load(VmafModel **model,
                                VmafModelCollection **model_collection,
                                VmafModelConfig *cfg,

libvmaf/src/model.c

@@ -13,12 +13,6 @@
 #include "read_json_model.h"
 #include "svm.h"
 
-typedef struct VmafBuiltInModel {
-    const char *version;
-    const char *data;
-    const int *data_len;
-} VmafBuiltInModel;
-
 #if VMAF_BUILT_IN_MODELS
 #if VMAF_FLOAT_FEATURES
 extern const char src_vmaf_float_v0_6_1neg_json;
@@ -316,3 +310,10 @@ int vmaf_model_collection_feature_overload(VmafModel *model,
     return err;
 }
 
+const VmafBuiltInModel *vmaf_built_in_model_next(const VmafBuiltInModel *prev){
+    if(!prev)
+        return &built_in_models[0];
+    if(prev - built_in_models < BUILT_IN_MODEL_CNT)
+        return prev + 1;
+    return NULL;
+}

libvmaf/src/model.h

@@ -83,6 +83,16 @@ typedef struct VmafModelCollection {
     const char *name;
 } VmafModelCollection;
 
+/**
+ * Iterate through all built in vmaf models
+ * 
+ * @param prev previous model. NULL to get the first model
+ * 
+ * @return next model or NULL after the last model
+ * 
+*/
+const VmafBuiltInModel *vmaf_built_in_model_next(const VmafBuiltInModel *prev);
+
 char *vmaf_model_generate_name(VmafModelConfig *cfg);
 
 int vmaf_model_collection_append(VmafModelCollection **model_collection,