omxh264enc: factor out string to profile/level enum conversion

https://bugzilla.gnome.org/show_bug.cgi?id=783862

1 files changed, 67 insertions, 50 deletions

diff --git a/omx/gstomxh264enc.c b/omx/gstomxh264enc.c
index 9172daf..7409c12 100644
--- a/omx/gstomxh264enc.c
+++ b/omx/gstomxh264enc.c
@@ -217,6 +217,69 @@ gst_omx_h264_enc_stop (GstVideoEncoder * enc)
   return GST_VIDEO_ENCODER_CLASS (parent_class)->stop (enc);
 }
 
+static OMX_VIDEO_AVCPROFILETYPE
+get_profile_from_str (const gchar * profile)
+{
+  if (g_str_equal (profile, "baseline")) {
+    return OMX_VIDEO_AVCProfileBaseline;
+  } else if (g_str_equal (profile, "main")) {
+    return OMX_VIDEO_AVCProfileMain;
+  } else if (g_str_equal (profile, "extended")) {
+    return OMX_VIDEO_AVCProfileExtended;
+  } else if (g_str_equal (profile, "high")) {
+    return OMX_VIDEO_AVCProfileHigh;
+  } else if (g_str_equal (profile, "high-10")) {
+    return OMX_VIDEO_AVCProfileHigh10;
+  } else if (g_str_equal (profile, "high-4:2:2")) {
+    return OMX_VIDEO_AVCProfileHigh422;
+  } else if (g_str_equal (profile, "high-4:4:4")) {
+    return OMX_VIDEO_AVCProfileHigh444;
+  }
+
+  return OMX_VIDEO_AVCProfileMax;
+}
+
+
+static OMX_VIDEO_AVCLEVELTYPE
+get_level_from_str (const gchar * level)
+{
+  if (g_str_equal (level, "1")) {
+    return OMX_VIDEO_AVCLevel1;
+  } else if (g_str_equal (level, "1b")) {
+    return OMX_VIDEO_AVCLevel1b;
+  } else if (g_str_equal (level, "1.1")) {
+    return OMX_VIDEO_AVCLevel11;
+  } else if (g_str_equal (level, "1.2")) {
+    return OMX_VIDEO_AVCLevel12;
+  } else if (g_str_equal (level, "1.3")) {
+    return OMX_VIDEO_AVCLevel13;
+  } else if (g_str_equal (level, "2")) {
+    return OMX_VIDEO_AVCLevel2;
+  } else if (g_str_equal (level, "2.1")) {
+    return OMX_VIDEO_AVCLevel21;
+  } else if (g_str_equal (level, "2.2")) {
+    return OMX_VIDEO_AVCLevel22;
+  } else if (g_str_equal (level, "3")) {
+    return OMX_VIDEO_AVCLevel3;
+  } else if (g_str_equal (level, "3.1")) {
+    return OMX_VIDEO_AVCLevel31;
+  } else if (g_str_equal (level, "3.2")) {
+    return OMX_VIDEO_AVCLevel32;
+  } else if (g_str_equal (level, "4")) {
+    return OMX_VIDEO_AVCLevel4;
+  } else if (g_str_equal (level, "4.1")) {
+    return OMX_VIDEO_AVCLevel41;
+  } else if (g_str_equal (level, "4.2")) {
+    return OMX_VIDEO_AVCLevel42;
+  } else if (g_str_equal (level, "5")) {
+    return OMX_VIDEO_AVCLevel5;
+  } else if (g_str_equal (level, "5.1")) {
+    return OMX_VIDEO_AVCLevel51;
+  }
+
+  return OMX_VIDEO_AVCLevelMax;
+}
+
 static gboolean
 set_avc_intra_perdiod (GstOMXH264Enc * self)
 {
@@ -403,61 +466,15 @@ gst_omx_h264_enc_set_format (GstOMXVideoEnc * enc, GstOMXPort * port,
     s = gst_caps_get_structure (peercaps, 0);
     profile_string = gst_structure_get_string (s, "profile");
     if (profile_string) {
-      if (g_str_equal (profile_string, "baseline")) {
-        param.eProfile = OMX_VIDEO_AVCProfileBaseline;
-      } else if (g_str_equal (profile_string, "main")) {
-        param.eProfile = OMX_VIDEO_AVCProfileMain;
-      } else if (g_str_equal (profile_string, "extended")) {
-        param.eProfile = OMX_VIDEO_AVCProfileExtended;
-      } else if (g_str_equal (profile_string, "high")) {
-        param.eProfile = OMX_VIDEO_AVCProfileHigh;
-      } else if (g_str_equal (profile_string, "high-10")) {
-        param.eProfile = OMX_VIDEO_AVCProfileHigh10;
-      } else if (g_str_equal (profile_string, "high-4:2:2")) {
-        param.eProfile = OMX_VIDEO_AVCProfileHigh422;
-      } else if (g_str_equal (profile_string, "high-4:4:4")) {
-        param.eProfile = OMX_VIDEO_AVCProfileHigh444;
-      } else {
+      param.eProfile = get_profile_from_str (profile_string);
+      if (param.eProfile == OMX_VIDEO_AVCProfileMax)
         goto unsupported_profile;
-      }
     }
     level_string = gst_structure_get_string (s, "level");
     if (level_string) {
-      if (g_str_equal (level_string, "1")) {
-        param.eLevel = OMX_VIDEO_AVCLevel1;
-      } else if (g_str_equal (level_string, "1b")) {
-        param.eLevel = OMX_VIDEO_AVCLevel1b;
-      } else if (g_str_equal (level_string, "1.1")) {
-        param.eLevel = OMX_VIDEO_AVCLevel11;
-      } else if (g_str_equal (level_string, "1.2")) {
-        param.eLevel = OMX_VIDEO_AVCLevel12;
-      } else if (g_str_equal (level_string, "1.3")) {
-        param.eLevel = OMX_VIDEO_AVCLevel13;
-      } else if (g_str_equal (level_string, "2")) {
-        param.eLevel = OMX_VIDEO_AVCLevel2;
-      } else if (g_str_equal (level_string, "2.1")) {
-        param.eLevel = OMX_VIDEO_AVCLevel21;
-      } else if (g_str_equal (level_string, "2.2")) {
-        param.eLevel = OMX_VIDEO_AVCLevel22;
-      } else if (g_str_equal (level_string, "3")) {
-        param.eLevel = OMX_VIDEO_AVCLevel3;
-      } else if (g_str_equal (level_string, "3.1")) {
-        param.eLevel = OMX_VIDEO_AVCLevel31;
-      } else if (g_str_equal (level_string, "3.2")) {
-        param.eLevel = OMX_VIDEO_AVCLevel32;
-      } else if (g_str_equal (level_string, "4")) {
-        param.eLevel = OMX_VIDEO_AVCLevel4;
-      } else if (g_str_equal (level_string, "4.1")) {
-        param.eLevel = OMX_VIDEO_AVCLevel41;
-      } else if (g_str_equal (level_string, "4.2")) {
-        param.eLevel = OMX_VIDEO_AVCLevel42;
-      } else if (g_str_equal (level_string, "5")) {
-        param.eLevel = OMX_VIDEO_AVCLevel5;
-      } else if (g_str_equal (level_string, "5.1")) {
-        param.eLevel = OMX_VIDEO_AVCLevel51;
-      } else {
+      param.eLevel = get_level_from_str (level_string);
+      if (param.eLevel == OMX_VIDEO_AVCLevelMax)
         goto unsupported_level;
-      }
     }
     gst_caps_unref (peercaps);
   }