Fix test_decode for monochrome video

Setting --test_decode=fatal no longer checks
U and V planes for monochrome video.

Change-Id: I4ff9f083855223d30bc1cd5d605d866cd5bd1cd7

aomenc.c

diff --git a/aomenc.c b/aomenc.c
index 2375510..6202e05 100644
--- a/aomenc.c
+++ b/aomenc.c
@@ -1005,6 +1005,22 @@
   return stream;
 }
 
+static void set_config_arg_ctrls(struct stream_config *config, int key,
+                                 const struct arg *arg) {
+  int j;
+  /* Point either to the next free element or the first instance of this
+   * control.
+   */
+  for (j = 0; j < config->arg_ctrl_cnt; j++)
+    if (config->arg_ctrls[j][0] == key) break;
+
+  /* Update/insert */
+  assert(j < (int)ARG_CTRL_CNT_MAX);
+  config->arg_ctrls[j][0] = key;
+  config->arg_ctrls[j][1] = arg_parse_enum_or_int(arg);
+  if (j == config->arg_ctrl_cnt) config->arg_ctrl_cnt++;
+}
+
 static int parse_stream_params(struct AvxEncoderConfig *global,
                                struct stream_state *stream, char **argv) {
   char **argi, **argj;
@@ -1177,27 +1193,19 @@
       config->cfg.tile_height_count =
           arg_parse_list(&arg, config->cfg.tile_heights, MAX_TILE_HEIGHTS);
 #endif
+#if CONFIG_MONO_VIDEO
+    } else if (arg_match(&arg, &input_color_space, argi)) {
+      aom_color_space_t color_space = arg_parse_enum_or_int(&arg);
+      config->cfg.monochrome = (color_space == AOM_CS_MONOCHROME);
+      set_config_arg_ctrls(config, AV1E_SET_COLOR_SPACE, &arg);
+#endif
     } else {
       int i, match = 0;
       for (i = 0; ctrl_args[i]; i++) {
         if (arg_match(&arg, ctrl_args[i], argi)) {
-          int j;
           match = 1;
-
-          /* Point either to the next free element or the first
-           * instance of this control.
-           */
-          for (j = 0; j < config->arg_ctrl_cnt; j++)
-            if (ctrl_args_map != NULL &&
-                config->arg_ctrls[j][0] == ctrl_args_map[i])
-              break;
-
-          /* Update/insert */
-          assert(j < (int)ARG_CTRL_CNT_MAX);
-          if (ctrl_args_map != NULL && j < (int)ARG_CTRL_CNT_MAX) {
-            config->arg_ctrls[j][0] = ctrl_args_map[i];
-            config->arg_ctrls[j][1] = arg_parse_enum_or_int(&arg);
-            if (j == config->arg_ctrl_cnt) config->arg_ctrl_cnt++;
+          if (ctrl_args_map) {
+            set_config_arg_ctrls(config, ctrl_args_map[i], &arg);
           }
         }
       }

av1/av1_cx_iface.c

diff --git a/av1/av1_cx_iface.c b/av1/av1_cx_iface.c
index 3838ab6..8f636dd 100644
--- a/av1/av1_cx_iface.c
+++ b/av1/av1_cx_iface.c
@@ -1518,6 +1518,9 @@
                                             va_list args) {
   struct av1_extracfg extra_cfg = ctx->extra_cfg;
   extra_cfg.color_space = CAST(AV1E_SET_COLOR_SPACE, args);
+#if CONFIG_MONO_VIDEO
+  ctx->cfg.monochrome = (extra_cfg.color_space == AOM_CS_MONOCHROME);
+#endif
   return update_extra_cfg(ctx, &extra_cfg);
 }