PyEFIE-2D/plotting_utils.py at main · eattardo/PyEFIE-2D · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.tri as tri
from point2D import Point2D


def plot_complex_quantity_on_2D_grid(msh, F, ngridx1=50, ngridx2=50):
    """
    """

    numCells = len(msh)
    x1 = []
    x2 = []
    for i in range(numCells):
        x1.append(msh.cells[i].x)
        x2.append(msh.cells[i].y)

    # Create grid values first.
    x1m = np.min(x1)
    x1p = np.max(x1)
    x2m = np.min(x2)
    x2p = np.max(x2)

    # Start plotting
    fig, (ax1,ax2) = plt.subplots(nrows=2)
    # -----------------------
    # Interpolation on a grid
    # -----------------------
    # A contour plot of irregularly spaced data coordinates
    # via interpolation on a grid.
    x1i = np.linspace(x1m, x1p, ngridx1)
    x2i = np.linspace(x2m, x2p, ngridx2)

    # Perform linear interpolation of the data (x,y)
    # on a grid defined by (xi,yi)
    triang = tri.Triangulation(x1, x2)
    interpolatorReal = tri.LinearTriInterpolator(triang, F.real)
    interpolatorIm = tri.LinearTriInterpolator(triang, F.imag)
    grid1, grid2 = np.meshgrid(x1i, x2i)
    fiReal = interpolatorReal(grid1, grid2)
    fiIm = interpolatorIm(grid1, grid2)

    # Note that scipy.interpolate provides means to interpolate data on a grid
    # as well. The following would be an alternative to the four lines above:
    #from scipy.interpolate import griddata
    #zi = griddata((x, y), z, (xi[None,:], yi[:,None]), method='linear')
    #cmap="RdBu_r"
    ax1.contour(grid1, grid2, fiReal, levels=32, linewidths=0.5, colors='k')
    cntr1 = ax1.contourf(grid1, grid2, fiReal, levels=14, cmap="RdBu_r")

    ax2.contour(grid1, grid2, fiIm, levels=32, linewidths=0.5, colors='k')
    cntr2 = ax2.contourf(grid1, grid2, fiIm, levels=14, cmap="RdBu_r")

    fig.colorbar(cntr1, ax=ax1)
    ax1.set(xlim=(x1m, x1p), ylim=(x2m, x2p))
    strTitle = "Real(F)"

    ax1.set_title(strTitle)
    plt.subplots_adjust(hspace=0.5)
    plt.xlabel("(m)")
    plt.ylabel("(m)")
    plt.show()

    fig.colorbar(cntr2, ax=ax2)
    ax2.set(xlim=(x1m, x1p), ylim=(x2m, x2p))
    strTitle = "Im(F)"

    ax2.set_title(strTitle)
    plt.subplots_adjust(hspace=0.5)
    plt.xlabel("(m)")
    plt.ylabel("(m)")
    plt.show()
#-----------------------------------------------------------------------------------------

def plot_efield_in_points(xp, yp, msh, E):
    """ Plot the E-fields in the points (xp,yp)
    """
    numCells = len(msh)
    Np = len(xp)
    N_ = len(yp)
    if N_!=Np:
        raise RuntimeError("xp and yp must have same length")

    # Search in the list of points in the mesh
    Ei = []
    points = []
    for i in range(Np):
        pi = Point2D(xp[i], yp[i])
        points.append(pi)
        _,ci = msh.pointInCell(pi)
        Ei.append(E[ci])
    return points, Ei