 |
tsave plot as PDF - slidergrid - grid of elastic sliders on a frictional surface |
 |
git clone git://src.adamsgaard.dk/slidergrid |
 |
Log |
|
Files |
|
Refs |
|
README |
|
LICENSE |
|
--- |
|
commit 2186c59a2858b1645b8f09a0cea58e87358aeda0 |
|
parent 64ff0d3cc5fff3d9f05502d7db911e75ebc1731d |
|
Author: Anders Damsgaard <[email protected]> |
|
Date: Wed, 30 Mar 2016 13:56:15 -0700 |
|
|
|
save plot as PDF |
|
|
|
Diffstat: |
|
M postprocessing.py | 24 ++++++++++++++++++++++-- |
|
|
|
1 file changed, 22 insertions(+), 2 deletions(-) |
|
--- |
|
diff --git a/postprocessing.py b/postprocessing.py |
|
t@@ -37,7 +37,7 @@ class sgvis: |
|
|
|
def plot_sliders(self): |
|
plt.plot(self.sliders[:, 0], self.sliders[:, 1], '+') |
|
- outfile = self.folder + '/' + self.filename + '.png' |
|
+ outfile = self.folder + '/' + self.filename + '.pdf' |
|
print(outfile) |
|
plt.savefig(outfile) |
|
plt.clf() |
|
t@@ -46,12 +46,32 @@ class sgvis: |
|
for filename in os.listdir(self.folder): |
|
if 'sliders' in filename \ |
|
and '.txt' in filename \ |
|
+ and '.pdf' not in filename \ |
|
and '.png' not in filename: |
|
self.read_sliders(filename) |
|
self.plot_sliders() |
|
|
|
+ def current_time(self): |
|
+ return 0.0 |
|
+ |
|
+ def current_kinetic_energy(self): |
|
+ return 0.0 |
|
+ |
|
def plot_kinetic_energy(self): |
|
- pass |
|
+ t = [] |
|
+ E_kin = [] |
|
+ for filename in os.listdir(self.folder): |
|
+ if 'sliders' in filename \ |
|
+ and '.txt' in filename \ |
|
+ and '.png' not in filename: |
|
+ self.read_sliders(filename) |
|
+ E_kin.append(self.current_kinetic_energy) |
|
+ |
|
+ plt.plot(t, E_kin) |
|
+ outfile = self.folder + '/E_kin.pdf' |
|
+ print(outfile) |
|
+ plt.savefig(outfile) |
|
+ plt.clf() |
|
|
|
def iterate_over_folders_and_files(folders, |
|
plot_sliders, |
