Update.

[flatland.git] / flatland.c

diff --git a/flatland.c b/flatland.c
index 77f89aa..df5075b 100644 (file)
--- a/flatland.c
+++ b/flatland.c
@@ -18,9 +18,11 @@
 
 #include "sequence_generator.h"
 
-THByteTensor *generate_sequence(long nb_sequences,
+THByteTensor *generate_sequence(int pulling,
+                                long nb_sequences,
                                 long nb_images,
                                 long image_height, long image_width,
+                                long nb_shapes,
                                 int random_shape_size, int random_colors) {
 
   long nb_channels = 3;
@@ -46,8 +48,9 @@ THByteTensor *generate_sequence(long nb_sequences,
 
   for(s = 0; s < nb_sequences; s++) {
     a = THByteTensor_storage(result)->data + THByteTensor_storageOffset(result) + s * st0;
-    fl_generate_sequence(nb_images, image_width, image_height,
+    fl_generate_sequence(nb_images, image_width, image_height, nb_shapes,
                          random_shape_size, random_colors,
+                         pulling,
                          tmp_buffer);
     unsigned char *r = tmp_buffer;
     for(k = 0; k < nb_images; k++) {