import numpy as np
import matplotlib.pyplot as plt
import os

top_directory = "/Users/natalie/Documents/masters/GDP/code/tmp/transient"
relevant = []

for dir in os.listdir(top_directory):
    try:
        int(dir[0])
    except:
        continue
    relevant.append(dir)
relevant.sort()
relevant.reverse()
relevant.pop()
relevant.reverse()

forces = []

for dir in relevant:
    forces.append([0,0,0])
    with open(f"{top_directory}/{dir}/uniform/functionObjects/functionObjectProperties") as file:
        for line in file:
            content = line.strip().split(" ")
            if content[0] == "pressureForce":
                forces[-1][0] += float(content[4])
                forces[-1][1] += float(content[5])
                forces[-1][2] += float(content[6])

            if content[0] == "viscousForce":
                forces[-1][0] += float(content[5])
                forces[-1][1] += float(content[6])
                forces[-1][2] += float(content[7])


plt.scatter([float(d)-0.2 for d in relevant], [force[0] for force in forces])
plt.scatter([float(d) for d in relevant], [force[1] for force in forces])
plt.grid()
plt.show()