Update tools/parabola_plotter_tool.py

tools/parabola_plotter_tool.py

@@ -19,36 +19,53 @@ def plot_parabola_graph(a: float, b: float, c: float) -> str:
         return "Erro: O coeficiente 'a' não pode ser zero para uma parábola."
     
     try:
-        # Cria uma gama de valores para x
-        x = np.linspace(-10, 10, 400)
+        # Calcula as raízes 
+        delta = b**2 - 4*a*c
+        
+        if delta >= 0:
+            raiz1 = (-b - np.sqrt(delta)) / (2*a)
+            raiz2 = (-b + np.sqrt(delta)) / (2*a)
+            # Define o intervalo X para ir um pouco além das raízes
+            x_min_plot = min(raiz1, raiz2) - 1.5
+            x_max_plot = max(raiz1, raiz2) + 1.5
+        else: # Se não houver raízes reais, usa o vértice como centro
+            x_min_plot = -5
+            x_max_plot = 5
         
-        # 2. Calcula 'y' usando a fórmula
-        y = a * x**2 + b * x + c
         
-        # Monta a string da função para o título
-        function_string = f"{a}x² + {b}x + {c}".replace("+ -", "- ")
+        x_vertice = -b / (2*a)
+        y_vertice = a * x_vertice**2 + b * x_vertice + c
+
+        # Aumenta o intervalo para garantir que o vértice e as raízes estejam visíveis
+        x_min_plot = min(x_min_plot, x_vertice - 2)
+        x_max_plot = max(x_max_plot, x_vertice + 2)
 
-        # Configura e cria o gráfico
-        plt.figure(figsize=(8, 6))
-        plt.plot(x, y)
         
+        x = np.linspace(x_min_plot, x_max_plot, 400)
         
-        # Para o eixo X, de -10 a 10, de 2 em 2
-        plt.xticks(np.arange(-10, 11, 2))
+        y = a * x**2 + b * x + c
+
         
-        y_min, y_max = plt.ylim() 
-        plt.yticks(np.arange(round(y_min / 10) * 10, y_max + 1, 10))
+        plt.figure(figsize=(8, 6))
+        plt.plot(x, y)
         
-       
-        plt.title(f"Gráfico da Parábola")
+        plt.title("Gráfico da Parábola")
         plt.xlabel("x")
         plt.ylabel("y")
         plt.grid(True)
         plt.axhline(0, color='black', linewidth=0.5)
         plt.axvline(0, color='black', linewidth=0.5)
+
+        # Adiciona marcadores para as raízes, se existirem
+        if delta >= 0:
+            plt.plot([raiz1, raiz2], [0, 0], 'o', color='red', label='Raízes')
+            plt.legend()
         
-        # USA time.time_ns() (nanossegundos) para garantir um nome de arquivo único
-        
+        # Ajusta os limites do eixo Y para focar no vértice
+        y_min_plot = min(y_vertice, 0) - abs(y_vertice) * 0.5 - 2
+        y_max_plot = max(y_vertice, y.max()) + abs(y_vertice) * 0.5 + 2
+        plt.ylim(y_min_plot, y_max_plot)
+
         file_path = f"parabola_{int(time.time_ns())}.png"
         plt.savefig(file_path)
         plt.close()