Initial commit

getDistance.py

@@ -0,0 +1,30 @@
+import math
+import pandas as pd
+import numpy as np
+import matplotlib.pyplot as plt
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument("--density", "-d")
+parser.add_argument("--frame", "-f")
+
+args=parser.parse_args()
+
+columns = ["ID", "idx", "Mass", "Radius", "X", "Y", "Z", "vX", "vY", "vZ", "sX", "sY", "sZ", "Colour"]
+
+
+num = str(args.frame).rjust(5, '0')
+file = "BTs/"+str(args.density)+"/boom."+num+".bt"
+data = pd.read_csv(file, sep=' ', names=columns)
+data["X"] = data["X"]*1.5e8
+data["Y"] = data["Y"]*1.5e8
+data["Z"] = data["Z"]*1.5e8
+data["X2"] = data["X"]**2
+data["Y2"] = data["Y"]**2
+data["Z2"] = data["Z"]**2
+data["distance2"] = data["X2"]+data["Y2"]+data["Z2"]
+data["distance"] = data["distance2"]**0.5
+
+distance = data["distance"].mean()
+
+print(distance)