From 3dce65b6084ff208b7a91472a64118e636559502 Mon Sep 17 00:00:00 2001
From: Fernanda Rossi <fernandarossi95@ufu.br>
Date: Wed, 14 Sep 2022 11:32:30 -0300
Subject: [PATCH] added plot method

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

diff --git a/main.py b/main.py
index 1eacab2..ec2d9ce 100644
--- a/main.py
+++ b/main.py
@@ -209,8 +209,24 @@ 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()