From f838a6d6c55867f060db3c7b29c0528613d107a7 Mon Sep 17 00:00:00 2001
From: Raimundo Neto <raimndo.pereira@ufu.br>
Date: Wed, 14 Sep 2022 11:31:03 -0300
Subject: [PATCH] added plot method

---
 main.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/main.py b/main.py
index 1eacab2..db9633e 100644
--- a/main.py
+++ b/main.py
@@ -210,7 +210,23 @@ def solveProblem(self,KK,MM):
         return modo
 
 
+    def plot(self, qtd_modos=5):
+
+        if self.done:
+            
+            fig = go.Figure()
+
+            Lo = np.arange(0,self.L,self.L_ef)
+
+            for mode in range(qtd_modos):
+                fig.add_trace(go.Scatter(x=Lo,y=self.modo[:,mode],name=f"{mode + 1} modo de vibrar", showlegend=True))
+
+            fig.update_layout(title="Viga de Euler-Bernoulli e seus modos de vibrar")
+            fig.show()
+        else:
+            raise Exception("You need to run the vibration modes before plotting!")
 if __name__ == '__main__':
 
     viga1 = BeamEB(0.35,0.02,0.06,7e10,2780,100)
     modo = viga1.runVibrationModes()
+    viga1.plot()