From 4372a378edb0e2aa0b8a33e6537e77d415d90630 Mon Sep 17 00:00:00 2001
From: Miikka Vaisala <mvaisala@asiaa.sinica.edu.tw>
Date: Mon, 14 Sep 2020 15:46:46 +0800
Subject: [PATCH] Plot Alfven speeds.

---
 analysis/python/astar/data/read.py       |  1 +
 analysis/python/astar/visual/lineplot.py | 42 +++++++++++++++++++++++-
 2 files changed, 42 insertions(+), 1 deletion(-)

diff --git a/analysis/python/astar/data/read.py b/analysis/python/astar/data/read.py
index 95f606b..3ae6ff8 100644
--- a/analysis/python/astar/data/read.py
+++ b/analysis/python/astar/data/read.py
@@ -20,6 +20,7 @@
 # This module is for reading data.
 
 import numpy as np
+import os
 
 #Optional YT interface
 try:
diff --git a/analysis/python/astar/visual/lineplot.py b/analysis/python/astar/visual/lineplot.py
index e793a6f..b76674a 100644
--- a/analysis/python/astar/visual/lineplot.py
+++ b/analysis/python/astar/visual/lineplot.py
@@ -39,7 +39,7 @@ def plot_min_man_rms(ts, xaxis, yaxis1, yaxis2, yaxis3):
 def plot_ts(ts, isotherm=False, show_all=False, lnrho=False, uutot=False, 
             uux=False, uuy=False, uuz=False, 
             aax=False, aay=False, aaz=False, 
-            ss=False, acc=False, sink=False, rho=False):
+            ss=False, acc=False, sink=False, rho=False, bb=False, alfven=False):
 
     if show_all:
         lnrho=True
@@ -51,6 +51,8 @@ def plot_ts(ts, isotherm=False, show_all=False, lnrho=False, uutot=False,
         ss=True
         acc=True
         sink=True
+        bb=True
+        alfven=True
 
     if isotherm:
         lnrho=True
@@ -143,6 +145,44 @@ def plot_ts(ts, isotherm=False, show_all=False, lnrho=False, uutot=False,
         yaxis3 = 'ss_max'
         plot_min_man_rms(ts, xaxis, yaxis1, yaxis2, yaxis3)
 
+    if bb:   
+        plt.figure()
+        xaxis = 't_step'
+        yaxis1 = 'bbtot_rms'
+        yaxis2 = 'bbtot_min'
+        yaxis3 = 'bbtot_max'
+        plot_min_man_rms(ts, xaxis, yaxis1, yaxis2, yaxis3)
+
+        plt.figure()
+        xaxis = 't_step'
+        yaxis1 = 'bbx_rms'
+        yaxis2 = 'bbx_min'
+        yaxis3 = 'bbx_max'
+        plot_min_man_rms(ts, xaxis, yaxis1, yaxis2, yaxis3)
+
+        plt.figure()
+        xaxis = 't_step'
+        yaxis1 = 'bby_rms'
+        yaxis2 = 'bby_min'
+        yaxis3 = 'bby_max'
+        plot_min_man_rms(ts, xaxis, yaxis1, yaxis2, yaxis3)
+
+        plt.figure()
+        xaxis = 't_step'
+        yaxis1 = 'bbz_rms'
+        yaxis2 = 'bbz_min'
+        yaxis3 = 'bbz_max'
+        plot_min_man_rms(ts, xaxis, yaxis1, yaxis2, yaxis3)
+
+    if alfven:
+        plt.figure()
+        xaxis = 't_step'
+        yaxis1 = 'vAtot_rms'
+        yaxis2 = 'vAtot_min'
+        yaxis3 = 'vAtot_max'
+        plot_min_man_rms(ts, xaxis, yaxis1, yaxis2, yaxis3)
+
+
     if acc:   
         plt.figure()
         xaxis = 't_step'