femagtools 1.7.8__py3-none-any.whl → 1.8.0__py3-none-any.whl

femagtools/plot/wdg.py

@@ -9,14 +9,38 @@ import numpy as np
 import matplotlib.pyplot as plt
 
 
+def currdist(wdg, title='', k='all', phi=0, ax=0):
+    """plot current distribution  of winding
+    Arguments:
+          k: (int) winding key (all if 0 or 'all', default all)
+          phi: (float) current angle (default 0)
+    """
+    if ax == 0:
+        ax = plt.gca()
+    if title:
+        ax.set_title(title)
+    cdist = wdg.currdist(k=k, phi=phi)
+    taus = 360/wdg.Q
+    c = np.sum([cdist[j] for j in cdist], axis=0)
+    pos = np.arange(0, len(cdist[1]))*taus + taus/2
+    cbars = ax.bar(pos, c/np.max(c), width=taus/4)
+    ax.set_xlabel("Position / Deg")
+    ax.grid()
+    return cbars
+
+
 def mmf(f, title='', ax=0):
-    """plot magnetomotive force (mmf) of winding"""
+    """plot magnetomotive force (mmf) of winding
+    Arguments
+      f: (dict) windings mmf
+      title: (str) plot title
+    """
     if ax == 0:
         ax = plt.gca()
     if title:
         ax.set_title(title)
-    ax.plot(np.array(f['pos'])/np.pi*180, f['mmf'])
-    ax.plot(np.array(f['pos_fft'])/np.pi*180, f['mmf_fft'])
+    yphi = ax.plot(np.array(f['pos'])/np.pi*180, f['mmf'])
+    yfft = ax.plot(np.array(f['pos_fft'])/np.pi*180, f['mmf_fft'])
     ax.set_xlabel('Position / Deg')
 
     phi = [f['alfa0']/np.pi*180, f['alfa0']/np.pi*180]
@@ -28,7 +52,7 @@ def mmf(f, title='', ax=0):
     ax.annotate(f"", xy=(phi[0], y[0]),
                 xytext=(0, y[0]), arrowprops=dict(arrowstyle="->"))
     ax.grid()
-
+    return yphi, yfft
 
 def mmf_fft(f, title='', mmfmin=1e-2, ax=0):
     """plot winding mmf harmonics"""
@@ -37,7 +61,16 @@ def mmf_fft(f, title='', mmfmin=1e-2, ax=0):
     if title:
         ax.set_title(title)
     else:
-        ax.set_title('MMF Harmonics (per phase)')
+        def single_phase(f):
+            try:
+                return len(f['currdens']) == 1
+            except KeyError:
+                return True
+        if single_phase(f):
+            ax.set_title('MMF Harmonics (single phase)')
+        else:
+            ax.set_title('MMF Harmonics')
+
     ax.grid(True)
     order, mmf = np.array([(n, m) for n, m in zip(f['nue'],
                                                   f['mmf_nue']) if m > mmfmin]).T
@@ -120,11 +153,11 @@ def winding_factors(wdg, n=8, ax=0):
                                   wdg.kwd(n),
                                   wdg.kw(n))]).T
     try:
-        markerline1, stemlines1, _ = ax.stem(order-1, kwp,
+        markerline1, stemlines1, _ = ax.stem(order-0.5, kwp,
                                              'C1:', basefmt=" ",
                                              markerfmt='C1.',
                                              label='Pitch')
-        markerline2, stemlines2, _ = ax.stem(order+1, kwd,
+        markerline2, stemlines2, _ = ax.stem(order+0.5, kwd,
                                              'C2:', basefmt=" ",
                                              markerfmt='C2.',
                                              label='Distribution')

femagtools/svgfsl/converter.py

@@ -20,6 +20,11 @@ def main():
                            help='stator without airgap in/out',
                            dest='stator',
                            default='')
+    argparser.add_argument('--EESM',
+                           help='Electric Excited Synchronous Motor',
+                           dest='EESM',
+                           action="store_true",
+                           default=False)
     argparser.add_argument('-p', '--plot',
                            help='show plots',
                            dest='show_plots',
@@ -61,6 +66,7 @@ def main():
 
     res = convert(args.svgfile,  # SVG-Filename
                   part=part,
+                  EESM=args.EESM,
                   view_only=args.view,
                   show_plots=args.show_plots,
                   show_areas=args.show_areas,

femagtools/templates/gen_hairpin_winding.mako

@@ -1,9 +1,13 @@
 -- hairpin winding template
 
-rcoil, phi_coil = c2pd(m.xcoil_1, m.ycoil_1)
-
 slot_div_angle = 360/m.tot_num_slot
 
+if m.xcoil_1 ~= nil and m.ycoil_1 ~= nil then 
+  rcoil, phi_coil = c2pd(m.xcoil_1, m.ycoil_1)
+else
+  rcoil, phi_coil = da1/2 + m.slot_height/2, slot_div_angle/2
+end 
+
 -- delete existing mesh in the slot
 for i = 1, m.num_sl_gen do
     xc, yc = pd2c(rcoil, phi_coil+(i-1)*slot_div_angle)
@@ -12,7 +16,7 @@ end
 -- wire params
 wire_height = ${model['wire']["wire_height"]*1e3}
 wire_width = ${model['wire']["wire_width"]*1e3}
-wire_corner_r = ${model['wire']["corner_radius"]*1e3}
+wire_corner_r = 0.0
 wire_gap = ${model['wire']["wire_separation"]*1e3}
 TH1 = ${model['wire']["wire_th1"]*1e3} -- distance to the bore radius
 nwires = ${model['wire']["num_layers"]}
@@ -128,7 +132,7 @@ for j = 1, m.num_sl_gen do
    end
 end
 
-ndt(agndst)
+ndt(agndst*1.5)
 for j = 1, m.num_sl_gen do
   for i = 1, nwires do
     dx, dy = x0+(wire_gap+wire_height)*(i-1), y0
@@ -142,8 +146,8 @@ create_mesh()
 
 -- create winding
 
-widfile = io.open("wid.fsl", 'r')
-
+--widfile = io.open("wid.fsl", 'r')
+widfile = io.open("wid.txt", 'r')
 nrows = 1
 winding = {}
 
@@ -158,52 +162,39 @@ for i in widfile:lines() do
 end
 
 --[[
-
 winding table
-
-1  slot_go pos_in_slot slot_ret pos_in_slot
-
+key slot layer dir
 ]]--
 dir = 'wi'
 cols = {"green", "yellow", "cyan"}
 wk = 1
 for i = 1, #winding do
-    idx = winding[i][2]
-    idxp = winding[i][3]
-    idy = winding[i][4]
-    idyp = winding[i][5]
-
-   if i == 1 then
-        if idx <= m.num_sl_gen then
-            wkey = def_new_wdg(wire_xy[idx][idxp].x, wire_xy[idx][idxp].y, cols[winding[i][1]], "Phase"..winding[i][1], 1, 0, 0, dir)
-        end
-
-        if math.abs(idy) <= m.num_sl_gen then
-            add_to_wdg (wire_xy[math.abs(idy)][idyp].x, wire_xy[math.abs(idy)][idyp].y, "wsamekey", "wo", "wser")
-        end
-   else
-        if winding[i][1] == winding[i-1][1] and winding[i][1] == wk then
-            if idx <= m.num_sl_gen then
-                add_to_wdg (wire_xy[idx][idxp].x, wire_xy[idx][idxp].y, "wsamekey", dir, "wser")
-            end
-            if math.abs(idy) <= m.num_sl_gen then
-                add_to_wdg (wire_xy[math.abs(idy)][idyp].x, wire_xy[math.abs(idy)][idyp].y, "wsamekey", "wo", "wser")
-            end
-        else
-            if idx <= m.num_sl_gen then
-                wkey = def_new_wdg(wire_xy[idx][idxp].x, wire_xy[idx][idxp].y, cols[winding[i][1]], "Phase"..winding[i][1], 1, 0, 0, dir)
-                wk = wk + 1
-            end
-            if math.abs(idy) <= m.num_sl_gen then
-                if wk == winding[i][1] then
-                  add_to_wdg (wire_xy[math.abs(idy)][idyp].x, wire_xy[math.abs(idy)][idyp].y, "wsamekey", "wo", "wser")
-                else
-                  wkey = def_new_wdg(wire_xy[math.abs(idy)][idyp].x, wire_xy[math.abs(idy)][idyp].y, cols[winding[i][1]], "Phase"..winding[i][1], 1, 0, 0, "wo")
+    windingskey = winding[i][1]
+    slot_nr = winding[i][2]
+    layer = winding[i][3]
+    direction = winding[i][4]
+    if direction < 0 then 
+      dir = 'wo'
+    else
+      dir = 'wi'
+    end 
+    if i == 1 then
+          if slot_nr <= m.num_sl_gen then
+              wkey = def_new_wdg(wire_xy[slot_nr][layer].x, wire_xy[slot_nr][layer].y, cols[winding[i][1]], "Phase"..winding[i][1], 1, 0, 0, dir)
+          end
+
+    else
+          if winding[i][1] == winding[i-1][1] and winding[i][1] == wk then
+              if slot_nr <= m.num_sl_gen then
+                  add_to_wdg (wire_xy[slot_nr][layer].x, wire_xy[slot_nr][layer].y, "wsamekey", dir, "wser")
+              end
+            
+          else
+              if slot_nr <= m.num_sl_gen then
+                  wkey = def_new_wdg(wire_xy[slot_nr][layer].x, wire_xy[slot_nr][layer].y, cols[winding[i][1]], "Phase"..winding[i][1], 1, 0, 0, dir)
                   wk = wk + 1
-                end
-            end
-        end
+              end
+          end
    end
 end
-
 m.num_par_wdgs = ${model.get('num_par_wdgs', 1)}

femagtools/templates/gen_winding.mako

@@ -88,6 +88,12 @@ m.num_poles       =  ${model.poles}
 
 % if 'thcap' in model.winding:
 -- Thermal Material properties
+
+conductor_density = ${model.winding['spmaweight']*1e3}
+conductor_thcond = ${model.winding['thcond']}
+conductor_thcap = ${model.winding['thcap']}
+
+--[[
 if m.slot_height ~= nil then
   -- FEMAG slot model
   -- TODO: slot model from user
@@ -114,4 +120,5 @@ if m.slot_height ~= nil then
     end
   end
 end
+]]--
 %endif

femagtools/templates/magnetIron.mako

@@ -1,4 +1,3 @@
-
 m.magn_rad       = da2/2
 m.yoke_rad       = dy2/2
 
@@ -39,58 +38,43 @@ for i = 0, m.npols_gen-1 do
     end
 end
 %endif
-%if model.get('thcond', 0) and model.get('thcap', 0):
+
+%if model.get('thcond', 0) and model.get('thcap', 0) and model.get('density', 0):
+if m.shaft_rad == nil then 
+    m.shaft_rad = dy2/2
+end 
+if m.shaft_rad > dy2/2 then 
+    m.shaft_rad = dy2/2
+end 
 beta = math.pi/m.num_poles
-xrb,yrb = pr2c(m.yoke_rad+0.1, beta)  -- rotor lamination
-thcond = ${model['thcond']}
-thcap = ${model['thcap']}
-density = ${model.get('density')*1e3}
-def_mat_therm(xrb,yrb,'blue',density,thcond,thcap,1)
+rotor_thcond = ${model['thcond']}
+rotor_thcap = ${model['thcap']}
+rotor_density = ${model.get('density')}
 
 %if model.get('thcond_shaft', 0) and model.get('thcap_shaft', 0):
 if m.shaft_rad < m.yoke_rad then
-   thcond = ${model['thcond_shaft']}
-   thcap = ${model['thcap_shaft']}
-   density = ${model['spmaweight_shaft']*1e3}
-   def_mat_therm(m.yoke_rad/2,0.1,'blue',density,thcond,thcap,1) -- Shaft
+   shaft_thcond = ${model['thcond_shaft']}
+   shaft_thcap = ${model['thcap_shaft']}
+   shaft_density = ${model['spmaweight_shaft']*1e3}
+   r_shaft = (m.shaft_rad + m.yoke_rad)/2
+   x0_shaft, y0_shaft = pd2c(r_shaft, beta/2)
 end
 %endif
-rm = m.magn_rad*math.cos(beta)
-density = ${model['spmaweight_magnet']*1e3}
-thcond = ${model.get('thcond_magnet', 8)}
-thcap = ${model.get('thcap_magnet', 440)}
-for i = 1,m.npols_gen do -- Magnete
-  alfa = (2*i-1) * beta
-  xmx,ymx = pr2c(rm,alfa)
-  def_mat_therm(xmx,ymx,darkgreen-i%2,density,thcond,thcap,1)
-end
-
-if m.air_triangle > 0 then
-  rs = math.sqrt(rm^2 + m.magn_width^2/4)
-  gam = math.atan(m.magn_width/2, rm)
-  for i = 1,m.npols_gen do -- air triangle
-    alfa = (2*i-1) * beta
-    xsx,ysx = pr2c(rs,alfa + gam + 1e-2)
-    def_mat_therm(xsx,ysx,skyblue,1.12,0.026,1007,1)
-    xsx,ysx = pr2c(rs,alfa - gam - 1e-2)
-    def_mat_therm(xsx,ysx,skyblue,1.12,0.026,1007,1)
-  end
-end
-
 
--- add air layer (inside) for heat transfer
-h = 3.8
-beta = 360*m.npols_gen/m.num_poles
+if x0_shaft == nil then
+   -- add air layer (inside) for heat transfer
+   h = 3.8
+   beta = 360*m.npols_gen/m.num_poles
 
-
-x0, y0 = pd2c(m.shaft_rad, m.zeroangl)
-x1, y1 = pd2c(m.shaft_rad-h, m.zeroangl)
-x2, y2 = pd2c(m.shaft_rad-h, beta+m.zeroangl)
-x3, y3 = pd2c(m.shaft_rad, beta+m.zeroangl)
-nc_line(x0, y0, x1, y1, 0)
-nc_circle(x1, y1, x2, y2, 0)
-nc_line(x2, y2, x3, y3, 0)
-x0, y0 = pd2c(m.shaft_rad-h/2, beta/2+m.zeroangl)
-create_mesh_se(x0, y0)
+   x0, y0 = pd2c(m.shaft_rad, m.zeroangl)
+   x1, y1 = pd2c(m.shaft_rad-h, m.zeroangl)
+   x2, y2 = pd2c(m.shaft_rad-h, beta+m.zeroangl)
+   x3, y3 = pd2c(m.shaft_rad, beta+m.zeroangl)
+   nc_line(x0, y0, x1, y1, 0)
+   nc_circle(x1, y1, x2, y2, 0)
+   nc_line(x2, y2, x3, y3, 0)
+   x0, y0 = pd2c(m.shaft_rad-h/2, beta/2+m.zeroangl)
+   create_mesh_se(x0, y0)
+end
 
 %endif

femagtools/templates/magnetIron2.mako

@@ -41,3 +41,42 @@ for i = 0, m.npols_gen-1 do
     end
 end
 %endif
+%if model.get('thcond', 0) and model.get('thcap', 0) and model.get('density', 0):
+if m.shaft_rad == nil then 
+    m.shaft_rad = dy2/2
+end 
+if m.shaft_rad > dy2/2 then 
+    m.shaft_rad = dy2/2
+end 
+beta = math.pi/m.num_poles
+rotor_thcond = ${model['thcond']}
+rotor_thcap = ${model['thcap']}
+rotor_density = ${model.get('density')}
+
+%if model.get('thcond_shaft', 0) and model.get('thcap_shaft', 0):
+if m.shaft_rad < m.yoke_rad then
+   shaft_thcond = ${model['thcond_shaft']}
+   shaft_thcap = ${model['thcap_shaft']}
+   shaft_density = ${model['spmaweight_shaft']*1e3}
+   r_shaft = (m.shaft_rad + m.yoke_rad)/2
+   x0_shaft, y0_shaft = pd2c(r_shaft, beta/2)
+end
+%endif
+
+if x0_shaft == nil then
+   -- add air layer (inside) for heat transfer
+   h = 3.8
+   beta = 360*m.npols_gen/m.num_poles
+
+   x0, y0 = pd2c(m.shaft_rad, m.zeroangl)
+   x1, y1 = pd2c(m.shaft_rad-h, m.zeroangl)
+   x2, y2 = pd2c(m.shaft_rad-h, beta+m.zeroangl)
+   x3, y3 = pd2c(m.shaft_rad, beta+m.zeroangl)
+   nc_line(x0, y0, x1, y1, 0)
+   nc_circle(x1, y1, x2, y2, 0)
+   nc_line(x2, y2, x3, y3, 0)
+   x0, y0 = pd2c(m.shaft_rad-h/2, beta/2+m.zeroangl)
+   create_mesh_se(x0, y0)
+end
+
+%endif

femagtools/templates/magnetIron3.mako

@@ -65,3 +65,42 @@ end
 end
 
 %endif
+%if model.get('thcond', 0) and model.get('thcap', 0) and model.get('density', 0):
+if m.shaft_rad == nil then 
+    m.shaft_rad = dy2/2
+end 
+if m.shaft_rad > dy2/2 then 
+    m.shaft_rad = dy2/2
+end 
+beta = math.pi/m.num_poles
+rotor_thcond = ${model['thcond']}
+rotor_thcap = ${model['thcap']}
+rotor_density = ${model.get('density')}
+
+%if model.get('thcond_shaft', 0) and model.get('thcap_shaft', 0):
+if m.shaft_rad < m.yoke_rad then
+   shaft_thcond = ${model['thcond_shaft']}
+   shaft_thcap = ${model['thcap_shaft']}
+   shaft_density = ${model['spmaweight_shaft']*1e3}
+   r_shaft = (m.shaft_rad + m.yoke_rad)/2
+   x0_shaft, y0_shaft = pd2c(r_shaft, beta/2)
+end
+%endif
+
+if x0_shaft == nil then
+   -- add air layer (inside) for heat transfer
+   h = 3.8
+   beta = 360*m.npols_gen/m.num_poles
+
+   x0, y0 = pd2c(m.shaft_rad, m.zeroangl)
+   x1, y1 = pd2c(m.shaft_rad-h, m.zeroangl)
+   x2, y2 = pd2c(m.shaft_rad-h, beta+m.zeroangl)
+   x3, y3 = pd2c(m.shaft_rad, beta+m.zeroangl)
+   nc_line(x0, y0, x1, y1, 0)
+   nc_circle(x1, y1, x2, y2, 0)
+   nc_line(x2, y2, x3, y3, 0)
+   x0, y0 = pd2c(m.shaft_rad-h/2, beta/2+m.zeroangl)
+   create_mesh_se(x0, y0)
+end
+
+%endif

femagtools/templates/magnetIron4.mako

@@ -40,3 +40,42 @@ for i = 0, m.npols_gen-1 do
     end
 end
 %endif
+%if model.get('thcond', 0) and model.get('thcap', 0):
+if m.shaft_rad == nil then 
+    m.shaft_rad = dy2/2
+end 
+if m.shaft_rad > dy2/2 then 
+    m.shaft_rad = dy2/2
+end 
+beta = math.pi/m.num_poles
+rotor_thcond = ${model['thcond']}
+rotor_thcap = ${model['thcap']}
+rotor_density = ${model.get('density')}
+
+%if model.get('thcond_shaft', 0) and model.get('thcap_shaft', 0) and model.get('density', 0):
+if m.shaft_rad < m.yoke_rad then
+   shaft_thcond = ${model['thcond_shaft']}
+   shaft_thcap = ${model['thcap_shaft']}
+   shaft_density = ${model['spmaweight_shaft']*1e3}
+   r_shaft = (m.shaft_rad + m.yoke_rad)/2
+   x0_shaft, y0_shaft = pd2c(r_shaft, beta/2)
+end
+%endif
+
+if x0_shaft == nil then
+   -- add air layer (inside) for heat transfer
+   h = 3.8
+   beta = 360*m.npols_gen/m.num_poles
+
+   x0, y0 = pd2c(m.shaft_rad, m.zeroangl)
+   x1, y1 = pd2c(m.shaft_rad-h, m.zeroangl)
+   x2, y2 = pd2c(m.shaft_rad-h, beta+m.zeroangl)
+   x3, y3 = pd2c(m.shaft_rad, beta+m.zeroangl)
+   nc_line(x0, y0, x1, y1, 0)
+   nc_circle(x1, y1, x2, y2, 0)
+   nc_line(x2, y2, x3, y3, 0)
+   x0, y0 = pd2c(m.shaft_rad-h/2, beta/2+m.zeroangl)
+   create_mesh_se(x0, y0)
+end
+
+%endif

femagtools/templates/magnetIron5.mako

@@ -40,3 +40,42 @@ for i = 0, m.npols_gen-1 do
     end
 end
 %endif
+%if model.get('thcond', 0) and model.get('thcap', 0) and model.get('density', 0):
+if m.shaft_rad == nil then 
+    m.shaft_rad = dy2/2
+end 
+if m.shaft_rad > dy2/2 then 
+    m.shaft_rad = dy2/2
+end 
+beta = math.pi/m.num_poles
+rotor_thcond = ${model['thcond']}
+rotor_thcap = ${model['thcap']}
+rotor_density = ${model.get('density')}
+
+%if model.get('thcond_shaft', 0) and model.get('thcap_shaft', 0):
+if m.shaft_rad < m.yoke_rad then
+   shaft_thcond = ${model['thcond_shaft']}
+   shaft_thcap = ${model['thcap_shaft']}
+   shaft_density = ${model['spmaweight_shaft']*1e3}
+   r_shaft = (m.shaft_rad + m.yoke_rad)/2
+   x0_shaft, y0_shaft = pd2c(r_shaft, beta/2)
+end
+%endif
+
+if x0_shaft == nil then
+   -- add air layer (inside) for heat transfer
+   h = 3.8
+   beta = 360*m.npols_gen/m.num_poles
+
+   x0, y0 = pd2c(m.shaft_rad, m.zeroangl)
+   x1, y1 = pd2c(m.shaft_rad-h, m.zeroangl)
+   x2, y2 = pd2c(m.shaft_rad-h, beta+m.zeroangl)
+   x3, y3 = pd2c(m.shaft_rad, beta+m.zeroangl)
+   nc_line(x0, y0, x1, y1, 0)
+   nc_circle(x1, y1, x2, y2, 0)
+   nc_line(x2, y2, x3, y3, 0)
+   x0, y0 = pd2c(m.shaft_rad-h/2, beta/2+m.zeroangl)
+   create_mesh_se(x0, y0)
+end
+
+%endif

femagtools/templates/magnetIronV.mako

@@ -68,62 +68,42 @@ for i = 0, m.npols_gen-1 do
 end
 %endif
 
-%if model.get('thcond', 0) and model.get('thcap', 0):
-beta = math.pi/m.num_poles + m.zeroangl/180*math.pi
-xrb,yrb = pr2c(m.yoke_rad+0.1, beta)  -- rotor lamination
-thcond = ${model['thcond']}
-thcap = ${model['thcap']}
-def_mat_therm(xrb,yrb,'blue',7700,thcond,thcap,1)
---def_mat_therm(m.yoke_rad/2,0.1,'blue',7700,thcond,thcap,1) -- Shaft
-
-gam = m.magn_angle/2/180*math.pi
-y = m.rotor_rad*math.sin(beta)
-x = m.rotor_rad*math.cos(beta)
-rr = y/math.sin(gam)
-xx = x - math.sqrt(rr^2 - y^2)
-rm = xx/(math.cos(beta/2) - math.sin(beta/2)/math.tan(gam))
-thcon = 8
-thcap = 440
-for i = 1,m.npols_gen do -- Magnets
-  alfa = (2*i-1) * beta
-  xmx,ymx = pr2c(rm,alfa-beta/2)
-  def_mat_therm(xmx,ymx,darkgreen-i%2,7500,thcond,thcap,1)
-  xmx,ymx = pr2c(rm,alfa+beta/2)
-  def_mat_therm(xmx,ymx,darkgreen-i%2,7500,thcond,thcap,1)
+%if model.get('thcond', 0) and model.get('thcap', 0) and model.get('density', 0):
+if m.shaft_rad == nil then 
+    m.shaft_rad = dy2/2
+end 
+if m.shaft_rad > dy2/2 then 
+    m.shaft_rad = dy2/2
+end 
+beta = math.pi/m.num_poles
+rotor_thcond = ${model['thcond']}
+rotor_thcap = ${model['thcap']}
+rotor_density = ${model.get('density')}
+
+%if model.get('thcond_shaft', 0) and model.get('thcap_shaft', 0):
+if m.shaft_rad < m.yoke_rad then
+   shaft_thcond = ${model['thcond_shaft']}
+   shaft_thcap = ${model['thcap_shaft']}
+   shaft_density = ${model['spmaweight_shaft']*1e3}
+   r_shaft = (m.shaft_rad + m.yoke_rad)/2
+   x0_shaft, y0_shaft = pd2c(r_shaft, beta/2)
 end
+%endif
 
-thcond = 0.026   -- air
-thcap = 1007
-get_spel_keys("sekeys")      -- Get all subregions of the model
-for i=1, #sekeys do
-  srkey =  get_spel_data("srkey", sekeys[i])
-  if srkey > 0 then
-    srname = get_sreg_data("name",srkey)
-    if srname == '    ' then
-      srkey = 0
-    end
-  end
-  if srkey == 0 then
-    elkeys = get_spel_data("elkeys", sekeys[i])
-    Ex, Ey = get_elem_data("xycp", elkeys[1])
-    r, phi = c2pr(Ex, Ey)
-    if r < m.rotor_rad then -- rotor only
-      def_mat_therm(Ex,Ey,'skyblue',1.12,thcond,thcap,1)
-    end
-  end
+if x0_shaft == nil then
+   -- add air layer (inside) for heat transfer
+   h = 3.8
+   beta = 360*m.npols_gen/m.num_poles
+
+   x0, y0 = pd2c(m.shaft_rad, m.zeroangl)
+   x1, y1 = pd2c(m.shaft_rad-h, m.zeroangl)
+   x2, y2 = pd2c(m.shaft_rad-h, beta+m.zeroangl)
+   x3, y3 = pd2c(m.shaft_rad, beta+m.zeroangl)
+   nc_line(x0, y0, x1, y1, 0)
+   nc_circle(x1, y1, x2, y2, 0)
+   nc_line(x2, y2, x3, y3, 0)
+   x0, y0 = pd2c(m.shaft_rad-h/2, beta/2+m.zeroangl)
+   create_mesh_se(x0, y0)
 end
 
--- add air layer (inside) for heat transfer
-h = 3.8
-beta = 360*m.npols_gen/m.num_poles
-x0, y0 = pd2c(dy2/2, m.zeroangl)
-x1, y1 = pd2c(dy2/2-h, m.zeroangl)
-x2, y2 = pd2c(dy2/2-h, beta+m.zeroangl)
-x3, y3 = pd2c(dy2/2, beta+m.zeroangl)
-nc_line(x0, y0, x1, y1, 0)
-nc_circle(x1, y1, x2, y2, 0)
-nc_line(x2, y2, x3, y3, 0)
-x0, y0 = pd2c(dy2/2-h/2, beta/2+m.zeroangl)
-create_mesh_se(x0, y0)
-
 %endif