tplot translational and rotational energy separately - slidergrid - grid of ela… | |
git clone git://src.adamsgaard.dk/slidergrid | |
Log | |
Files | |
Refs | |
README | |
LICENSE | |
--- | |
commit 5a85affe06c9ef33083476a69ceddd03b0eb4155 | |
parent 5923f961afa2aca4593b8c4004ae6c5816e7250e | |
Author: Anders Damsgaard <[email protected]> | |
Date: Wed, 30 Mar 2016 14:13:35 -0700 | |
plot translational and rotational energy separately | |
Diffstat: | |
M postprocessing.py | 24 +++++++++++++++++------- | |
1 file changed, 17 insertions(+), 7 deletions(-) | |
--- | |
diff --git a/postprocessing.py b/postprocessing.py | |
t@@ -65,18 +65,25 @@ class sgvis: | |
return 0.0 | |
def current_kinetic_energy(self): | |
- E_kin = 0.0 | |
+ E_t = 0.0 | |
+ E_r = 0.0 | |
for idx in np.arange(np.size(self.mass)): | |
- E_kin += self.slider_linear_kinetic_energy(idx) | |
- return E_kin | |
+ E_t += self.slider_translational_kinetic_energy(idx) | |
+ E_r += self.slider_rotational_kinetic_energy(idx) | |
+ return E_t, E_r | |
- def slider_linear_kinetic_energy(self, idx): | |
+ def slider_translational_kinetic_energy(self, idx): | |
return 0.5*self.mass[idx] \ | |
* np.sqrt(np.dot(self.vel[idx, :], self.vel[idx, :]))**2 | |
+ def slider_rotational_kinetic_energy(self, idx): | |
+ return 0.5*self.moment_of_inertia[idx] \ | |
+ * np.sqrt(np.dot(self.angvel[idx, :], self.angvel[idx, :]))**2 | |
+ | |
def plot_kinetic_energy(self): | |
t = [] | |
- E_kin = [] | |
+ E_t_series = [] | |
+ E_r_series = [] | |
for filename in os.listdir(self.folder): | |
if 'sliders' in filename \ | |
and '.txt' in filename \ | |
t@@ -84,9 +91,12 @@ class sgvis: | |
and '.png' not in filename: | |
self.read_sliders(filename) | |
t.append(self.current_time) | |
- E_kin.append(self.current_kinetic_energy) | |
+ E_t, E_r = self.current_kinetic_energy() | |
+ E_t_series.append(E_t) | |
+ E_r_series.append(E_r) | |
- plt.plot(t, E_kin) | |
+ plt.plot(t, E_t_series) | |
+ plt.plot(t, E_r_series) | |
outfile = self.folder + '/E_kin.pdf' | |
print(outfile) | |
plt.savefig(outfile) |