femagtools 1.8.5__py3-none-any.whl → 1.8.7__py3-none-any.whl

femagtools/plot/bch.py

@@ -22,6 +22,26 @@ except ImportError:   # ModuleNotFoundError:
 logger = logging.getLogger("femagtools.plot.bch")
 
 
+def find_peaks_and_valleys(t, y):
+    """ return peaks and valleys of y with maximum amplitude
+    """
+    peaks = (np.diff(np.sign(np.diff(y))) < 0).nonzero()[0] + 1
+    if len(peaks > 0):
+        ip = np.argmax(y[peaks])
+        pv = {'yp': y[peaks][ip], 'tp': t[peaks][ip]}
+    else:
+        pv = {'yp': [], 'tp': []}
+    valleys = (np.diff(np.sign(np.diff(y))) > 0).nonzero()[0] + 1
+    if len(valleys > 0):
+        iv = np.argmin(y[valleys])
+        pv.update({'yv': y[valleys][iv], 'tv': t[valleys][iv]})
+    else:
+        pv.update({'yv': [], 'tv': []})
+    pv.update({'peaks': y[peaks], 'valleys': y[valleys],
+               'tpeaks': t[peaks], 'tvalleys': t[valleys]})
+    return pv
+
+
 def _create_3d_axis():
     """creates a subplot with 3d projection if one does not already exist"""
     from matplotlib.projections import get_projection_class
@@ -495,48 +515,151 @@ def cogging(bch, title=''):
         fig.subplots_adjust(top=0.92)
 
 
-def transientsc(bch, title=''):
-    """creates a transient short circuit plot"""
-    cols = 1
-    rows = 2
-    htitle = 1.5 if title else 0
-    fig, ax = plt.subplots(nrows=rows, ncols=cols,
-                           figsize=(10, 3*rows + htitle))
-    if title:
-        fig.suptitle(title, fontsize=16)
-
-    row = 1
-    plt.subplot(rows, cols, row)
-    ax = plt.gca()
+def demagnetization(demag, ax=0):
+    """plot rel. remanence vs. current"""
+    if ax == 0:
+        ax = plt.gca()
+    scale = 1
+    unit = 'A'
+    if np.max(demag['i1']) > 25e3:
+        scale = 1e-3
+        unit = 'kA'
+    i1 = [scale*x for x in demag['i1']]
+    ax.plot(i1, demag['rr'], 'o', color='C0')
+    ax.plot(i1, demag['rr'], color='C0')
+    rrmin = 0.6
+    if demag.get('i1c', 0):
+        Icrit = scale*demag['i1c']
+        Hk = demag['Hk']
+        Tmag = demag['Tmag']
+        di = 0.05*np.max(i1)
+        rrmin = min(0.6, np.min(demag['rr']))
+        ax.plot([Icrit, Icrit], [rrmin, 1], 'k--')
+        ax.annotate(
+            f'Icrit = {Icrit:.1f}{unit}\nHk = {Hk:.1f} kA/m\nTmag={Tmag:.1f} °C',
+            xy=(Icrit, rrmin),
+            xytext=(Icrit-di, rrmin+0.1*(1-rrmin)), ha='right',
+            bbox={'facecolor': 'white',
+                  'edgecolor': 'white'})
+    ax.set_ylim([rrmin, 1.01])
+    ax.set_ylabel('Rel. Remanence')
+    ax.set_xlabel(f'Phase Current / {unit}')
+    ax.grid()
+
+
+def transientsc_currents(scData, ax=0, title='', set_xlabel=True):
+    """plot transient shortcircuit currents vs time"""
+    if ax == 0:
+        ax = plt.gca()
     ax.grid(True)
-    istat = np.array([bch.scData[i]
+    if title:
+        ax.set_title(title)
+    istat = np.array([scData[i]
                       for i in ('ia', 'ib', 'ic')])
+    pv = [find_peaks_and_valleys(
+        np.array(scData['time']), i1)
+          for i1 in istat]
+    try:
+        ipvmax = np.argmax(
+            [y['yp'] if np.abs(y['yp']) > np.abs(y['yv']) else y['yv']
+             for y in pv if y['yp']])
+        imax = pv[ipvmax]['yp'] if np.abs(pv[ipvmax]['yp']) > np.abs(pv[ipvmax]['yv']) else pv[ipvmax]['yv']
+        iac = [pv[ipvmax]['tpeaks'][-1], pv[ipvmax]['peaks'][-1]]
+    except KeyError:
+        pass
     if np.max(istat) > 4000:
         istat *= 1e-3
-        ax.set_title('Currents / kA')
+        try:
+            imax *= 1e-3
+            iac[1] *= 1e-3
+        except NameError:
+            pass
+        ax.set_ylabel('Currents / kA')
     else:
-        ax.set_title('Currents / A')
+        ax.set_ylabel('Currents / A')
 
     for i, iph in zip(('ia', 'ib', 'ic'), istat):
-        ax.plot(bch.scData['time'], iph, label=i)
-    ax.set_xlabel('Time / s')
+        ax.plot(scData['time'], iph, label=i)
+    try:
+        ax.plot([pv[ipvmax]['tp']], [imax], '.')
+        ax.plot([iac[0]], [iac[1]], '.')
+        dtx = (scData['time'][-1]-scData['time'][0])/75
+        dy = imax/25
+        ax.annotate(f'Imax = {imax:.1f}',
+                    xy=(pv[ipvmax]['tp'], imax),
+                    xytext=(pv[ipvmax]['tp']+dtx, imax-dy))
+        dy = iac[1]/25
+        ax.annotate(f'I = {iac[1]:.1f}',
+                    xy=iac,
+                    xytext=(iac[0]+dtx, iac[1]-dy))
+    except NameError:
+        pass
+    if set_xlabel:
+        ax.set_xlabel('Time / s')
     ax.legend()
 
-    row = 2
-    plt.subplot(rows, cols, row)
-    ax = plt.gca()
-    scale = 1
-    torque = np.array(bch.scData['torque'])
+
+def transientsc_torque(scData, ax=0, title='', set_xlabel=True):
+    """plot transient shortcircuit torque vs time"""
+    if ax == 0:
+        ax = plt.gca()
+    if title:
+        ax.set_title(title)
+    pv = find_peaks_and_valleys(
+        np.array(scData['time']), np.array(scData['torque']))
+    try:
+        tqmax = pv['yp'] if np.abs(pv['yp']) > np.abs(pv['yv']) else pv['yv']
+        tp = pv['tp'] if np.abs(pv['yp']) > np.abs(pv['yv']) else pv['tv']
+        tc = [pv['tpeaks'][-1], pv['peaks'][-1]]
+    except (KeyError, ValueError):
+        pass
+    torque = np.array(scData['torque'])
     if np.max(torque) > 4000:
         torque *= 1e-3
-        ax.set_title('Torque / kNm')
+        try:
+            tqmax *= 1e-3
+            tc[1] *= 1e-3
+        except NameError:
+            pass
+        ax.set_ylabel('Torque / kNm')
     else:
-        ax.set_title('Torque / Nm')
+        ax.set_ylabel('Torque / Nm')
 
     ax.grid(True)
-    ax.plot(bch.scData['time'], torque)
-    ax.set_xlabel('Time / s')
+    ax.plot(scData['time'], torque)
+    try:
+        ax.plot([tp], [tqmax], '.')
+        ax.plot([tc[0]], [tc[1]], '.')
+        dtx = (scData['time'][-1]-scData['time'][0])/75
+        dy = tqmax/25
+        ax.annotate(f'Tmax = {tqmax:.1f}',
+                    xy=(tp, tqmax),
+                    xytext=(tp+dtx, tqmax-dy))
+        dy = tc[1]/25
+        ax.annotate(f'T = {tc[1]:.1f}',
+                    xy=tc,
+                    xytext=(tc[0]+dtx, tc[1]))
+    except NameError:
+        pass
+    if set_xlabel:
+        ax.set_xlabel('Time / s')
+
+def transientsc(bch, title=''):
+    """creates a transient short circuit plot"""
+    try:
+        scData = bch.scData
+    except AttributeError:
+        scData = bch
+    cols = 1
+    rows = 2
+    htitle = 1.5 if title else 0
+    fig, axs = plt.subplots(nrows=rows, ncols=cols, sharex=True,
+                            figsize=(10, 3*rows + htitle))
+    if title:
+        fig.suptitle(title, fontsize=16)
 
+    transientsc_currents(scData, axs[0], set_xlabel=False)
+    transientsc_torque(scData, axs[1])
     fig.tight_layout(h_pad=2)
     if title:
         fig.subplots_adjust(top=0.92)
@@ -550,22 +673,35 @@ def transientsc_demag(demag, magnet=0, title='', ax=0):
     """
     if ax == 0:
         ax = plt.gca()
-    pos = [d['displ'] for d in demag if 'displ' in d]
-    hmax = [-d['H_max'] for d in demag if 'H_max' in d]
-    havg = [-d['H_av'] for d in demag if 'H_av' in d]
-    hclim = [-d['lim_hc'] for d in demag if 'lim_hc' in d]*2
-
+    if type(d) == list:
+        pos = [d['displ'] for d in demag if 'displ' in d]
+        hmax = [-d['H_max'] for d in demag if 'H_max' in d]
+        havg = [-d['H_av'] for d in demag if 'H_av' in d]
+        hclim = [-d['lim_hc'] for d in demag if 'lim_hc' in d][0]
+    else:
+        pos = demag['displ']
+        hmax = demag['H_max']
+        havg = demag['H_av']
+        hclim = demag['Hk']
     ax.set_title('Transient Short Circuit Demagnetization [kA/m]')
     ax.plot(pos, hmax,
             label='H Max {:4.2f} kA/m'.format(max(hmax)))
     ax.plot(pos, havg,
             label='H Avg {:4.2f} kA/m'.format(max(havg)))
-    ax.plot([pos[0], pos[-1]], hclim, color='C3', linestyle='dashed',
-            label='Hc {:4.2f} kA/m'.format(hclim[0]))
+    if len(hclim) > 1:
+        ax.plot([pos[0], pos[-1]], [hclim,hclim], color='C3',
+                linestyle='dashed',
+                label='Hc {:4.2f} kA/m'.format(hclim))
+    if 'Tmag' in demag:
+        Tmag = demag['Tmag']
+    elif magnet:
+        Tmag = magnet['Tmag']
+    else:
+        Tmag = ''
     ax.set_xlabel('Rotor Position / °')
     ax.grid(True)
-    if magnet:
-        ax.legend(title=f"Magnet Temperature {magnet['Tmag']}°C")
+    if Tmag:
+        ax.legend(title=f"Magnet Temperature {Tmag}°C")
     else:
         ax.legend()
 

femagtools/plot/machine.py

@@ -0,0 +1,100 @@
+"""
+Create longitudinal drawing of radial flux machine
+"""
+import matplotlib.pyplot as plt
+import matplotlib.patches as pch
+
+def _draw_shaft(ax, dy2, lfe):
+    xx = (0.0, lfe, lfe, 0.0)
+    yy = (dy2/2, dy2/2, -dy2/2, -dy2/2)
+    ax.fill(xx, yy,
+            facecolor='lightgrey',
+            edgecolor='black',
+            linewidth=0)
+    xx = (-lfe/4, lfe+lfe/4, lfe+lfe/4, -lfe/4)
+    yy = (dy2/4, dy2/4, -dy2/4, -dy2/4)
+    ax.fill(xx, yy,
+            facecolor='lightgrey',
+            edgecolor='black',
+            linewidth=0)
+
+def _draw_rotor(ax, da1, dy2, lfe):
+    ag = 0.02*da1
+    xx = (0.0, lfe, lfe, 0.0)
+    yy = (dy2/2, dy2/2, da1/2-ag, da1/2-ag)
+    ax.fill(xx, yy,
+            facecolor='skyblue',
+            edgecolor='black',
+            linewidth=0)
+    yy = (-dy2/2, -dy2/2, -da1/2+ag, -da1/2+ag)
+    ax.fill(xx, yy,
+            facecolor='skyblue',
+            edgecolor='black',
+            linewidth=0)
+
+def _draw_stator(ax, da1, dy1, lfe):
+    xx = (0.0, lfe, lfe, 0.0)
+    yy = (da1/2, da1/2, dy1/2, dy1/2)
+    # yoke
+    ax.fill(xx, yy,
+            facecolor='skyblue',
+            edgecolor='black',
+            linewidth=0)
+
+    yy = (-da1/2, -da1/2, -dy1/2, -dy1/2)
+    ax.fill(xx, yy,
+            facecolor='skyblue',
+            edgecolor='black',
+            linewidth=0)
+
+    # winding
+    yh = (dy1-da1)/2
+    xx = (-yh/2, 0, 0, -yh/2)
+    yy = (da1/2, da1/2, dy1/2-yh/2, dy1/2-yh/2)
+    ax.fill(xx, yy, facecolor='gold',
+                edgecolor='black',
+                linewidth=0)
+
+    xx = (lfe, lfe+yh/2, lfe+yh/2, lfe)
+    ax.fill(xx, yy, facecolor='gold',
+                edgecolor='black',
+                linewidth=0)
+
+    yy = (-da1/2, -da1/2, -dy1/2+yh/2, -dy1/2+yh/2)
+    ax.fill(xx, yy, facecolor='gold',
+                edgecolor='black',
+                linewidth=0)
+
+    xx = (-yh/2, 0, 0, -yh/2)
+    ax.fill(xx, yy, facecolor='gold',
+                edgecolor='black',
+                linewidth=0)
+
+def machine(machine, ax):
+    dy2 = machine['inner_diam']*1e3
+    dy1 = machine['outer_diam']*1e3
+    da1 = machine['bore_diam']*1e3
+    lfe = machine['lfe']*1e3
+
+    _draw_rotor(ax, da1, dy2, lfe)
+    _draw_stator(ax, da1, dy1, lfe)
+    _draw_shaft(ax, dy2, lfe)
+    ax.set_aspect('equal')
+
+    for loc, spine in ax.spines.items():
+        spine.set_color('none')  # don't draw spine
+    #ax.yaxis.set_ticks([])
+    #ax.xaxis.set_ticks([])
+
+
+if __name__ == '__main__':
+    machine1 = {
+        "outer_diam": 0.2442,
+        "bore_diam": 0.179,
+        "inner_diam": 0.06,
+        "airgap": 0.7e-3,
+        "lfe": 0.083,
+    }
+    fig, ax = plt.subplots()
+    machine(machine1, ax)
+    plt.show()

femagtools/plot/nc.py

@@ -130,6 +130,19 @@ def demag(isa, cmap=DEFAULT_CMAP, ax=0):
     logger.info("Max demagnetization %f", np.max(demag))
 
 
+def remanence(isa, cmap=DEFAULT_CMAP, ax=0):
+    """plot remanence of NC/I7/ISA7 model
+    Args:
+      isa: Isa7/NC object
+    """
+    emag = [e for e in isa.elements if e.is_magnet()]
+    rem = np.linalg.norm([e.remanence(isa.MAGN_TEMPERATURE)
+                          for e in emag], axis=1)
+    _contour(ax, f'Remanence at {isa.MAGN_TEMPERATURE} °C (min {np.min(rem):.1f} T)',
+             emag, rem, 'T', cmap, isa)
+    logger.info("Min remanence %f", np.min(rem))
+
+
 def demag_pos(isa, pos=-1, icur=-1, ibeta=-1, cmap=DEFAULT_CMAP, ax=0):
     """plot demag of NC/I7/ISA7 model at rotor position
     Args:

femagtools/poc.py

@@ -75,6 +75,7 @@ class Poc:
         if self.pole_pitch:
             content.append("{0}".format(self.pole_pitch))
         if self.pocType in ['fun', 'har', 'hsp']:
+            self.data_check()
             func_steps = len(self.func_current)
             content += [f"{self.pocType}", f"{func_steps}"]
             if (self.pocType == 'fun' and
@@ -122,6 +123,7 @@ class Poc:
             import re
             for i in range(func_steps-1):
                 l = re.split(';|\t|,| ', pocfile.readline().strip())
+                l = [i for i in l if i]  # remove empty items
                 if len(l) > 2:
                     self.harmonic_id.append(int(l[0]))
                     self.func_current.append(float(l[1]))
@@ -129,6 +131,7 @@ class Poc:
                 else:
                     self.func_current.append(float(l[0]))
                     self.func_phi.append(float(l[1]))
+            self.data_check()
         else:
             self.shape_current=self.pocType
             self.pocType='Function'
@@ -139,6 +142,13 @@ class Poc:
         except ValueError:
             pass
 
+    def data_check( self ):
+        """simple phi data increasing check"""
+        import numpy as np
+        if (not np.all(np.diff(self.func_phi) > 0)
+            and np.all(np.diff(self.func_current) > 0)):
+            self.func_phi, self.func_current = self.func_current, self.func_phi
+
     def getProps( self ):
         keys=['key_winding',
               'phi_voltage_winding',