readData.py

#!/usr/bin/env python3
# Launch the Euler-Bernoulli comparison for the beam
#%% 
import numpy as np
import matplotlib.pyplot as plt
from readIni import IniReader
import subprocess as sp

ini = IniReader("Gebt.ini");
nkp = ini.getValue("Beam","nkp",typeCast=int);
L   = ini.getValue("Beam","L"  ,typeCast=float);
dt  = ini.getValue("Beam","dt" ,typeCast=float);
nmemb = nkp-1
display = 0

#out = sp.run(["./gebt"],check=True);

u2 = {}; 
u3 = {};
phi1 = {};

for f in ["output"]:
    u2[f] = []
    u3[f] = []
    phi1[f] = []
    for k in range(100):
        for j in range(10):
            print("%s_%d_%03d.dat"%(f,k,j))
            data_pts = np.loadtxt("%s_%d_%03d.dat"%(f,k,j),max_rows=nkp);
            data_mem = np.loadtxt("%s_%d_%03d.dat"%(f,k,j),skiprows=nkp);

            pts = data_pts;
            mem = data_mem;

            if(display):
                #plt.plot(mem[:,0]+mem[:,3],mem[:,4]+mem[:,1]);
                plt.plot(pts[:,0]+pts[:,3],pts[:,5]+pts[:,2],'.-k');
                #plt.plot(mem[:,0]+mem[:,3],mem[:,5]+mem[:,2],'.r');

                plt.xlabel('x');
                plt.ylabel('w(x)');
                
                #plt.axis('scaled');

                plt.ylim([-10,10]);

                plt.pause(0.01);
                plt.clf();

                print("u2 = %f m"%(pts[-1,4]))
                print("u3 = %f m"%(pts[-1,5]))
                


            u2[f].append(pts[-1,4]);
            u3[f].append(pts[-1,5]);
            phi1[f].append(pts[-1,6]);

    u2[f] = np.array(u2[f]);
    u3[f] = np.array(u3[f]);
    phi1[f] = np.array(phi1[f]);
    t = np.linspace(0,u3[f].size*dt,u3[f].size);

for f in []:
    u2[f] = []
    u3[f] = []
    for k in range(100):
        for j in range(100):
            data_pts = np.loadtxt("output%s_%d_%03d.dat"%(f,k,j),max_rows=nkp);
            data_mem = np.loadtxt("output%s_%d_%03d.dat"%(f,k,j),skiprows=nkp);

            pts = data_pts;
            mem = data_mem;

            if(display):
                #plt.plot(mem[:,0]+mem[:,3],mem[:,4]+mem[:,1]);
                plt.plot(pts[:,0]+pts[:,3],pts[:,5]+pts[:,2],'.-k');
                #plt.plot(mem[:,0]+mem[:,3],mem[:,5]+mem[:,2],'.r');

                plt.xlabel('x');
                plt.ylabel('w(x)');
                
                #plt.axis('scaled');

                plt.ylim([-10,10]);

                plt.pause(0.01);
                plt.clf();

                print("u2 = %f m"%(pts[-1,4]))
                print("u3 = %f m"%(pts[-1,5]))


            u2[f].append(pts[-1,4]);
            u3[f].append(pts[-1,5]);

    u2[f] = np.array(u2[f]);
    u3[f] = np.array(u3[f]);
    t = np.linspace(0,u3[f].size*dt,u3[f].size);

plt.figure()
for f in ["output"]:
    plt.plot(t,u2[f]);
    plt.plot(t,u3[f]);
    plt.plot(t,phi1[f]);
    plt.grid(True);
    plt.title("GEBT");

print("Done!");

plt.show();