avfilter/vf_v360: simplify allocating remap data

1 files changed, 3 insertions, 3 deletions

diff --git a/libavfilter/vf_v360.c b/libavfilter/vf_v360.c
index c8258cecc6..0f8aa2c34f 100644
--- a/libavfilter/vf_v360.c
+++ b/libavfilter/vf_v360.c
@@ -2454,16 +2454,16 @@ static int config_output(AVFilterLink *outlink)
     if (desc->log2_chroma_h == desc->log2_chroma_w && desc->log2_chroma_h == 0) {
         s->nb_allocated = 1;
         s->map[0] = s->map[1] = s->map[2] = s->map[3] = 0;
-        allocate_plane(s, sizeof_uv, sizeof_ker, 0);
     } else {
         s->nb_allocated = 2;
         s->map[0] = 0;
         s->map[1] = s->map[2] = 1;
         s->map[3] = 0;
-        allocate_plane(s, sizeof_uv, sizeof_ker, 0);
-        allocate_plane(s, sizeof_uv, sizeof_ker, 1);
     }
 
+    for (int i = 0; i < s->nb_allocated; i++)
+        allocate_plane(s, sizeof_uv, sizeof_ker, i);
+
     calculate_rotation_matrix(s->yaw, s->pitch, s->roll, s->rot_mat, s->rotation_order);
     set_mirror_modifier(s->h_flip, s->v_flip, s->d_flip, s->output_mirror_modifier);