creator file modified

1 files changed, 5 insertions, 4 deletions

diff --git a/src/creator.c b/src/creator.c
index 71a981f..9990972 100644
--- a/src/creator.c
+++ b/src/creator.c
@@ -1,5 +1,5 @@
-#include <creator.h>
-#include <render.h>
+#include "creator.h"
+#include "render.h"
 
 static void encode(AVCodecContext *enc_ctx, AVFrame *frame, AVPacket *pkt,
                    FILE *outfile)
@@ -24,7 +24,7 @@ static void encode(AVCodecContext *enc_ctx, AVFrame *frame, AVPacket *pkt,
         av_packet_unref(pkt);
     }
 }
-int generateVideo(filename, int width, int height, int fps, int bitRate, Config *config, u32 (*sfunc) (long double, long double, u32))
+int generateVideo(filename, int length, int width, int height, int fps, int bitRate, Config *config, u32 (*sfunc) (long double, long double, u32))
 {
     const char *filename;
     const AVCodec *codec;
@@ -101,7 +101,8 @@ int generateVideo(filename, int width, int height, int fps, int bitRate, Config
     y_max = y_max_s;
     _config = config;
     _sfunc = sfunc;
-    for(i=0;i<25;i++) {
+
+    for(i=0;i<length*fps;i++) {
         fflush(stdout);
         /* make sure the frame data is writable */
         ret = av_frame_make_writable(picture);