force/vel/rho now selectable (for flowvis)

1 files changed, 13 insertions, 0 deletions

diff --git a/Smoke/heightplots.c b/Smoke/heightplots.c
index c4546df..4593455 100644
--- a/Smoke/heightplots.c
+++ b/Smoke/heightplots.c
@@ -6,6 +6,7 @@
 
 #include <rfftw.h>
 
+#include "fluids.h"
 #include "funcs.h"
 #include "palette.h"
 
@@ -20,6 +21,8 @@ static int heightplots_colormap = PALETTE_BLACKWHITE;
 
 static float heightplots_alpha = 1.0f;
 
+static int heightplots_dataset = DATASET_RHO;
+
 static fftw_real *heightplots_frame;
 
 
@@ -63,6 +66,16 @@ float heightplots_get_alpha(void)
   return heightplots_alpha;
 }
 
+void heightplots_set_dataset(int dataset)
+{
+  heightplots_dataset = dataset;
+}
+
+int heightplots_get_dataset(void)
+{
+  return heightplots_dataset;
+}
+
 void heightplots_set_frame(fftw_real *frame)
 {
   heightplots_frame = frame;