From 09fd81eac3f5861b043e00d4865fa205deba5319 Mon Sep 17 00:00:00 2001
From: Marius Peter <blendoit@gmail.com>
Date: Wed, 12 Jun 2019 15:13:23 -0700
Subject: debug messages for plotting and saving coordinates

---
 creator.py | 18 +++++++++++-------
 main.py    |  9 +++++----
 2 files changed, 16 insertions(+), 11 deletions(-)

diff --git a/creator.py b/creator.py
index aa8001f..a2626d0 100644
--- a/creator.py
+++ b/creator.py
@@ -92,12 +92,16 @@ class Coordinates:
 
         file_name = '{}_{}.txt'.format(self, number)
         full_path = os.path.join(save_dir_path, file_name)
-        # sys.stdout = open(full_path, 'w')
-        # self.print_coord(2)
-        with open(full_path, 'w') as sys.stdout:
-            self.print_coord(2)
-        # Following line required to reset value of sys.stdout
-        sys.stdout = sys.__stdout__
+        try:
+            with open(full_path, 'w') as sys.stdout:
+                self.print_coord(2)
+                # This line required to reset behavior of sys.stdout
+                sys.stdout = sys.__stdout__
+                print('Successfully wrote to file {}'.format(full_path))
+        except:
+            print('Unable to write {} to specified directory.\n'
+                  .format(file_name),
+                  'Was the full path passed to the function?')
         # It is cleaner to use this context guard to ensure file is closed
 
         return None
@@ -385,7 +389,7 @@ def plot(airfoil, spar, stringer):
             x = (spar.x_u[_], spar.x_l[_])
             y = (spar.y_u[_], spar.y_l[_])
             plt.plot(x, y, '.-', color='b')
-            plt.legend()
+            # plt.legend()
     except:
         print('Did not plot spars. Were they added?')
 
diff --git a/main.py b/main.py
index 9278eeb..b842e9e 100644
--- a/main.py
+++ b/main.py
@@ -34,28 +34,29 @@ def main():
 
     # Interate through all wings in population.
     for _ in range(1, POP_SIZE + 1):
+
         # Create airfoil instance
         af = creator.Airfoil()
         # Define NACA airfoil coordinates
         af.add_naca(2412)
-        af.print_coord(2)
+        # af.print_coord(2)
 
         # Create spar instance
         af.spar = creator.Spar()
         # Define the spar coordinates, stored in single spar object
         af.spar.add(af.coord, 0.15)
         af.spar.add(af.coord, 0.55)
-        af.spar.print_coord(2)
+        # af.spar.print_coord(2)
 
         # Create stringer instance
         af.stringer = creator.Stringer()
         # Define the stringer coordinates from their amount
         af.stringer.add(af.coord, af.spar.coord, 4, 7, 5, 6)
         # Print coordinates of af.stringer to terminal
-        af.stringer.print_coord(2)
+        # af.stringer.print_coord(2)
 
         # Plot components with matplotlib
-        creator.plot(af, af.spar, af.stringer)
+        # creator.plot(af, af.spar, af.stringer)
 
         # Save component coordinates
         af.save_coord(SAVE_PATH, _)
-- 
cgit v1.2.3