import sys
|
import matplotlib
|
|
from PySide6.QtWidgets import QApplication, QMainWindow, QVBoxLayout
|
|
from numint.input_parser import parse_function, parse_value
|
from numint.riemann_sum import numint_epsilon, numint_section
|
from numint.ui_mainwindow import Ui_MainWindow
|
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg, NavigationToolbar2QT as NavigationToolbar
|
from matplotlib.figure import Figure
|
|
matplotlib.use('QtAgg')
|
|
|
class PlotCanvas(FigureCanvasQTAgg):
|
def __init__(self, parent=None, width=5, height=4, dpi=100):
|
fig = Figure(figsize=(width, height), dpi=dpi)
|
self.axes = fig.add_subplot(111)
|
super(PlotCanvas, self).__init__(fig)
|
self.left = 0.0
|
self.right = 0.0
|
|
|
class MainWindow(QMainWindow):
|
def __init__(self):
|
super(MainWindow, self).__init__()
|
self.ui = Ui_MainWindow()
|
self.ui.setupUi(self)
|
self.canvas = PlotCanvas(self)
|
self._update_plot()
|
|
toolbar = NavigationToolbar(self.canvas, self)
|
layout = QVBoxLayout()
|
layout.addWidget(toolbar)
|
layout.addWidget(self.canvas)
|
self.ui.plotBox.setLayout(layout)
|
self.ui.plotBtn.clicked.connect(self._update_plot)
|
self.ui.breakMethod.buttonClicked.connect(self._activate_break_rule)
|
self.ui.numberOfSectionsMethod.click()
|
self.ui.plotBtn.click()
|
|
def _update_plot(self):
|
fn_expr = self.ui.function.text()
|
self.fn_expr = fn_expr
|
start = self.ui.startValue.text()
|
end = self.ui.endValue.text()
|
|
try:
|
f = parse_function(fn_expr)
|
(a, b) = (parse_value(start), parse_value(end))
|
method_id = self.ui.breakMethod.checkedId()
|
if method_id == self.ui.breakMethod.id(self.ui.minEpsilonMethod):
|
epsilon = self.ui.espilon.text()
|
eps = abs(parse_value(epsilon))
|
self.ui.espilon.setText(f"{eps}")
|
print(f"plot f(x) = {fn_expr}, {a}, {b}, epsilon = {eps}")
|
(self.x, self.y, self.left, self.right, self.num_of_iterations, *_) = numint_epsilon(f, a, b, eps)
|
self._plot_canvas()
|
elif method_id == self.ui.breakMethod.id(self.ui.numberOfSectionsMethod):
|
section = self.ui.section.text()
|
sec = int(abs(parse_value(section)))
|
self.ui.section.setText(f"{sec}")
|
print(f"plot f(x) = {fn_expr}, {a}, {b}, section = {sec}")
|
(self.x, self.y, self.left, self.right, self.num_of_iterations, *_) = numint_section(f, a, b, sec)
|
self._plot_canvas()
|
except Exception as ex:
|
print(ex)
|
pass
|
|
def _plot_canvas(self):
|
self.canvas.axes.cla() # Clear the canvas.
|
# self.canvas.axes.stairs(self.y[1:], self.x, baseline=0, fill=True, alpha=0.5, label=self.fn_expr)
|
# self.canvas.axes.stairs(self.y[0:-1], self.x, baseline=0, fill=True, alpha=0.5, label=self.fn_expr)
|
self.canvas.axes.step(self.x, self.y, color="blue", where="pre", label="right")
|
self.canvas.axes.step(self.x, self.y, color="orange", where="post", label="left")
|
|
self.canvas.axes.plot(self.x, self.y, color="black", label=self.fn_expr, linewidth=0.75)
|
self.canvas.draw()
|
|
self.ui.leftSum.setText(f"{self.left}")
|
self.ui.rightSum.setText(f"{self.right}")
|
self.ui.numOfSections.setText(f"{self.num_of_iterations}")
|
|
def _activate_break_rule(self):
|
method_id = self.ui.breakMethod.checkedId()
|
print(method_id)
|
if method_id == self.ui.breakMethod.id(self.ui.numberOfSectionsMethod):
|
print("activate num of sec")
|
self.ui.section.setEnabled(True)
|
self.ui.section.setFocus()
|
self.ui.espilon.setEnabled(False)
|
|
elif method_id == self.ui.breakMethod.id(self.ui.minEpsilonMethod):
|
print("activate minimum epsilon method")
|
self.ui.section.setEnabled(False)
|
self.ui.espilon.setEnabled(True)
|
self.ui.espilon.setFocus()
|
|
|
|
|
def main():
|
app = QApplication(sys.argv)
|
window = MainWindow()
|
window.show()
|
|
sys.exit(app.exec())
|
|
|
if __name__ == "__main__":
|
main()
|
