mplwidget.py

#! /usr/bin/python3
# coding: utf-8


"""Matplotlib QtDesigner Widget."""


import numpy as np
import matplotlib.image as image
from PyQt5 import QtWidgets
from PyQt5.QtCore import pyqtSignal
from matplotlib.figure import Figure
from matplotlib.collections import LineCollection
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg


def make_segments(x, y):
    """Make segments."""
    points = np.array([x, y]).T.reshape(-1, 1, 2)
    segments = np.concatenate([points[:-1], points[1:]], axis=1)
    return segments


class MplCanvas(FigureCanvasQTAgg):
    """MplCanvas."""

    depassement = pyqtSignal()

    def __init__(self):
        """Init."""
        self.fig = Figure()
        self.ax = self.fig.add_subplot(111)
        FigureCanvasQTAgg.__init__(self, self.fig)
        FigureCanvasQTAgg.setSizePolicy(
            self, QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Expanding)
        FigureCanvasQTAgg.updateGeometry(self)

    def graph(self, model, t=0):
        """Plot deformée."""
        # Initialise l'espace du graphique
        self.fig.clf()
        self.ax = self.fig.add_subplot(111)
        self.ax.set_title(model.legend['title'])

        # Afiche la courbe de deformee coloree en fonction du resulat selectionné
        if t == 0:
            lc = self.colorline(model.deformee[0], model.deformee[1], np.round(np.absolute(model.deplacements[:-1:]), 10))
            cbar = self.fig.colorbar(lc)
            cbar.ax.set_title(r"Déplacement en $mm$")
        elif t == 1:
            contraintes = model.contraintes
            lc = self.colorline(model.deformee[0], model.deformee[1], np.round(np.absolute(contraintes), 10))
            cbar = self.fig.colorbar(lc)
            cbar.ax.set_title(r"Contraintes en $MPa$")
            if max(contraintes) >= model.material.Re:
                self.depassement.emit()

        # Affiche l'effort et les liasons
        for effort in model.efforts:
            self.ax.arrow(effort[0], effort[1], effort[2], effort[3], head_width=effort[4], head_length=effort[5], fc='m', ec='m')

        encastrement_horiz = image.imread('ui/liaisons/encastrement_horiz.jpg')
        encastrement_vert = image.imread('ui/liaisons/encastrement_vert.jpg')
        glissiere = image.imread('ui/liaisons/glissiere.jpg')
        rotule = image.imread('ui/liaisons/rotule.jpg')
        ponctuelle = image.imread('ui/liaisons/ponctuelle.jpg')

        if model.__class__.__name__ == "PoutreEnTraction":
            self.ax.set_xlim([-0.1, 0.1])
            self.ax.set_ylim([-50, model._lenght + 50])
            self.ax.get_xaxis().set_visible(False)
            self.ax.imshow(encastrement_horiz, aspect='auto', extent=(-0.01, 0.01, -30, 5))
        elif model.__class__.__name__ == "PoutreEnFlexion":
            if model.selected == 0:
                self.ax.imshow(encastrement_vert, aspect='auto', extent=(-20, 10, -0.5, 0.5))
            if model.selected == 1:
                self.ax.imshow(encastrement_vert, aspect='auto', extent=(-20, 10, -0.005, 0.005))
                self.ax.imshow(glissiere, aspect='auto', extent=(model._lenght-20, model._lenght + 20, -0.01, 0))
            if model.selected == 2:
                self.ax.imshow(rotule, aspect='auto', extent=(-20, 20, -0.03, 0))
                self.ax.imshow(ponctuelle, aspect='auto', extent=(model._lenght-20, model._lenght + 20, -0.03, 0))
        elif model.__class__.__name__ == "TreilliSimple":
            pass

        # Affiche la poutre initiale
        self.ax.plot(model.initial[0], model.initial[1], linewidth=2, color='k', linestyle="-.")
        self.ax.set_xlabel(model.legend['xtitle'])
        self.ax.set_ylabel(model.legend['ytitle'])
        self.draw()

    def colorline(self, x, y, z):
        """Plot a colored line with coordinates x and y."""
        z = np.asarray(z)
        segments = make_segments(x, y)
        lc = LineCollection(segments, array=z, cmap='jet', linewidth=6, alpha=1)
        self.ax.add_collection(lc)
        return lc


class MplWidget(QtWidgets.QWidget):
    """QtDesigner QtWidget Promotion Class."""

    def __init__(self, parent=None):
        """Init."""
        QtWidgets.QWidget.__init__(self, parent)
        self.canvas = MplCanvas()
        self.vbl = QtWidgets.QVBoxLayout()
        self.vbl.addWidget(self.canvas)
        self.setLayout(self.vbl)