3 files changed, 21 insertions, 9 deletions

diff --git a/inc/demo.h b/inc/demo.h
index d06cf1d..c7cc32e 100644
--- a/inc/demo.h
+++ b/inc/demo.h
@@ -137,6 +137,7 @@ int readdemo(FILE *demofile, demo *demo);
 int writedemo(FILE *outfile, demo *demo);
 
 /* Prints given demo file */
-void printdemo(demo *demo);
+/* if printchunks then print all chunks */
+void printdemo(demo *demo, char printchunks);
 
 #endif // DEMO_H
diff --git a/src/demo.c b/src/demo.c
index fdf9e6e..d73b1a7 100644
--- a/src/demo.c
+++ b/src/demo.c
@@ -549,10 +549,14 @@ void printdemodelta(demodelta *delta)
     printf("DELTA={datasize: %d}\n", delta->datasize);
 }
 
-void printdemo(demo *demo)
+void printdemo(demo *demo, char printchunks)
 {
+    printf("HEADER:\n");
     printdemoheader(&demo->header);
-    printf("numchunks: %d\n\n", demo->data.numchunks);
+    printf("\n");
+
+    if (printchunks)
+        printf("CHUNKS:\n");
 
     int typecount[4] = {0, 0, 0, 0};
     for (int i = 0; i < demo->data.numchunks; i++)
@@ -561,19 +565,23 @@ void printdemo(demo *demo)
         switch (chunk->type)
         {
         case DEMOTICK:
-            printdemotick(chunk->data.tick);
+            if (printchunks)
+                printdemotick(chunk->data.tick);
             typecount[(int)DEMOTICK]++;
             break;
         case DEMOSNAP:
-            printdemosnap(chunk->data.snap);
+            if (printchunks)
+                printdemosnap(chunk->data.snap);
             typecount[(int)DEMOSNAP]++;
             break;
         case DEMOMESSAGE:
-            printdemomessage(chunk->data.message);
+            if (printchunks)
+                printdemomessage(chunk->data.message);
             typecount[(int)DEMOMESSAGE]++;
             break;
         case DEMODELTA:
-            printdemodelta(chunk->data.delta);
+            if (printchunks)
+                printdemodelta(chunk->data.delta);   
             typecount[(int)DEMODELTA]++;
             break;
         default:
@@ -581,8 +589,11 @@ void printdemo(demo *demo)
             break;
         }
     }
+    if (printchunks)
+        printf("\n");
 
-    printf("\nchunks: %d\n", demo->data.numchunks);
+    printf("CHUNK INFO:\n");
+    printf("chunk count: %d\n", demo->data.numchunks);
     printf("chunk counts: [\n");
     printf("  ticks: %d\n", typecount[0]);
     printf("  snaps: %d\n", typecount[1]);
diff --git a/src/main.c b/src/main.c
index 8c74b73..cb6f3d5 100644
--- a/src/main.c
+++ b/src/main.c
@@ -120,7 +120,7 @@ int main()
     demo d;
 
     readdemo(fp, &d);
-    printdemo(&d);
+    printdemo(&d, 0);
     writedemo(op, &d);
 
     fclose(fp);