streamlines stuffs

1 files changed, 14 insertions, 0 deletions

diff --git a/Smoke/streamlines.c b/Smoke/streamlines.c
index 9be9c38..a962180 100644
--- a/Smoke/streamlines.c
+++ b/Smoke/streamlines.c
@@ -6,18 +6,22 @@
 
 #include <rfftw.h>
 
+#include "fluids.h"
 #include "funcs.h"
 #include "palette.h"
 
 #include "streamlines.h"
 
 
+
 static int streamlines_render = FALSE;
 
 static int streamlines_num_colors = PALETTE_MAXCOLORS;
 
 static int streamlines_colormap = PALETTE_BLACKWHITE;
 
+static int streamlines_dataset = DATASET_RHO;
+
 static float streamlines_alpha = 1.0f;
 
 static fftw_real *streamlines_frame;
@@ -63,6 +67,16 @@ float streamlines_get_alpha(void)
   return streamlines_alpha;
 }
 
+void streamlines_set_dataset(int dataset)
+{
+  streamlines_dataset = dataset;
+}
+
+int streamlines_get_dataset(void)
+{
+  return streamlines_dataset;
+}
+
 void streamlines_set_frame(fftw_real *frame)
 {
   streamlines_frame = frame;