Formatting cleanups.

user_libs/antenna_patterns/initial_save.py

@@ -36,37 +36,37 @@ epsr = 5
 
 # Observation radii and angles
 radii = np.linspace(0.1, 0.3, 20)
-theta = np.linspace(3, 357, 60) * (180/np.pi)
+theta = np.linspace(3, 357, 60) * (180 / np.pi)
 
 # Scaling of time-domain field pattern values by material impedance
 impscaling = False
 
 # Centre frequency of modelled antenna
-f = 1.5e9 # GSSI 1.5GHz antenna model
+f = 1.5e9  # GSSI 1.5GHz antenna model
 
 # Largest dimension of antenna transmitting element
-D = 0.060 # GSSI 1.5GHz antenna model
+D = 0.060  # GSSI 1.5GHz antenna model
 
 # Traces to plot for sanity checking
-traceno = np.s_[:] # All traces
+traceno = np.s_[:]  # All traces
 ########################################
 
 # Critical angle and velocity
 if epsr:
     mr = 1
-    z1 = np.sqrt(mr/epsr) * z0
+    z1 = np.sqrt(mr / epsr) * z0
     v1 = c / np.sqrt(epsr)
-    thetac = np.round(np.arcsin(v1/c) * (180/np.pi))
-    wavelength = v1/f
+    thetac = np.round(np.arcsin(v1 / c) * (180 / np.pi))
+    wavelength = v1 / f
 
 # Print some useful information
-print('Centre frequency: {} GHz'.format(f/1e9))
+print('Centre frequency: {} GHz'.format(f / 1e9))
 if epsr:
     print('Critical angle for Er {} is {} degrees'.format(epsr, thetac))
     print('Wavelength: {:.3f} m'.format(wavelength))
-    print('Observation distance(s) from {:.3f} m ({:.1f} wavelengths) to {:.3f} m ({:.1f} wavelengths)'.format(radii[0], radii[0]/wavelength, radii[-1], radii[-1]/wavelength))
-    print('Theoretical boundary between reactive & radiating near-field (0.62*sqrt((D^3/wavelength): {:.3f} m'.format(0.62 * np.sqrt((D**3)/wavelength)))
-    print('Theoretical boundary between radiating near-field & far-field (2*D^2/wavelength): {:.3f} m'.format((2 * D**2)/wavelength))
+    print('Observation distance(s) from {:.3f} m ({:.1f} wavelengths) to {:.3f} m ({:.1f} wavelengths)'.format(radii[0], radii[0] / wavelength, radii[-1], radii[-1] / wavelength))
+    print('Theoretical boundary between reactive & radiating near-field (0.62*sqrt((D^3/wavelength): {:.3f} m'.format(0.62 * np.sqrt((D**3) / wavelength)))
+    print('Theoretical boundary between radiating near-field & far-field (2*D^2/wavelength): {:.3f} m'.format((2 * D**2) / wavelength))
 
 # Load text file with coordinates of pattern origin
 origin = np.loadtxt(os.path.splitext(outputfile)[0] + '_rxsorigin.txt')
@@ -77,13 +77,13 @@ iterations = f.attrs['Iterations']
 dt = f.attrs['dt']
 nrx = f.attrs['nrx']
 if antenna:
-    nrx = nrx - 1 # Ignore first receiver point with full antenna model
+    nrx = nrx - 1  # Ignore first receiver point with full antenna model
     start = 2
 else:
     start = 1
 time = np.arange(0, dt * iterations, dt)
 time = time / 1E-9
-fs = 1 / dt # Sampling frequency
+fs = 1 / dt  # Sampling frequency
 
 # Initialise arrays to store fields
 coords = np.zeros((nrx, 3), dtype=np.float32)
@@ -115,7 +115,7 @@ for rx in range(0, nrx):
     Hz[:, rx] = f[path + 'Hz'][:]
 f.close()
 
-## Plot traces for sanity checking
+# Plot traces for sanity checking
 #fig, ((ax1, ax2), (ax3, ax4), (ax5, ax6)) = plt.subplots(num=outputfile, nrows=3, ncols=2, sharex=False, sharey='col', subplot_kw=dict(xlabel='Time [ns]'), figsize=(20, 10), facecolor='w', edgecolor='w')
 #ax1.plot(time, Ex[:, traceno],'r', lw=2)
 #ax1.set_ylabel('$E_x$, field strength [V/m]')
@@ -129,12 +129,12 @@ f.close()
 #ax4.set_ylabel('$H_y$, field strength [A/m]')
 #ax6.plot(time, Hz[:, traceno],'b', lw=2)
 #ax6.set_ylabel('$H_z$, field strength [A/m]')
-## Turn on grid
+# Turn on grid
 #[ax.grid() for ax in fig.axes]
-#plt.show()
+# plt.show()
 
 # Calculate fields for patterns
-rxstart = 0 # Index for rx points
+rxstart = 0  # Index for rx points
 for radius in range(0, len(radii)):
     index = 0
     # Observation points

user_libs/antenna_patterns/plot_fields.py

@@ -35,13 +35,13 @@ epsr = 5
 # Observation radii and angles
 radii = np.linspace(0.1, 0.3, 20)
 theta = np.linspace(3, 357, 60)
-theta = np.deg2rad(np.append(theta, theta[0])) # Append start value to close circle
+theta = np.deg2rad(np.append(theta, theta[0]))  # Append start value to close circle
 
 # Centre frequency of modelled antenna
-f = 1.5e9 # GSSI 1.5GHz antenna model
+f = 1.5e9  # GSSI 1.5GHz antenna model
 
 # Largest dimension of antenna transmitting element
-D = 0.060 # GSSI 1.5GHz antenna model
+D = 0.060  # GSSI 1.5GHz antenna model
 
 # Minimum value for plotting energy and ring steps (dB)
 min = -72
@@ -51,19 +51,19 @@ step = 12
 # Critical angle and velocity
 if epsr:
     mr = 1
-    z1 = np.sqrt(mr/epsr) * z0
+    z1 = np.sqrt(mr / epsr) * z0
     v1 = c / np.sqrt(epsr)
-    thetac = np.round(np.rad2deg(np.arcsin(v1/c)))
-    wavelength = v1/f
+    thetac = np.round(np.rad2deg(np.arcsin(v1 / c)))
+    wavelength = v1 / f
 
 # Print some useful information
-print('Centre frequency: {} GHz'.format(f/1e9))
+print('Centre frequency: {} GHz'.format(f / 1e9))
 if epsr:
     print('Critical angle for Er {} is {} degrees'.format(epsr, thetac))
     print('Wavelength: {:.3f} m'.format(wavelength))
-    print('Observation distance(s) from {:.3f} m ({:.1f} wavelengths) to {:.3f} m ({:.1f} wavelengths)'.format(radii[0], radii[0]/wavelength, radii[-1], radii[-1]/wavelength))
-    print('Theoretical boundary between reactive & radiating near-field (0.62*sqrt((D^3/wavelength): {:.3f} m'.format(0.62 * np.sqrt((D**3)/wavelength)))
-    print('Theoretical boundary between radiating near-field & far-field (2*D^2/wavelength): {:.3f} m'.format((2 * D**2)/wavelength))
+    print('Observation distance(s) from {:.3f} m ({:.1f} wavelengths) to {:.3f} m ({:.1f} wavelengths)'.format(radii[0], radii[0] / wavelength, radii[-1], radii[-1] / wavelength))
+    print('Theoretical boundary between reactive & radiating near-field (0.62*sqrt((D^3/wavelength): {:.3f} m'.format(0.62 * np.sqrt((D**3) / wavelength)))
+    print('Theoretical boundary between radiating near-field & far-field (2*D^2/wavelength): {:.3f} m'.format((2 * D**2) / wavelength))
 
 # Setup figure
 fig = plt.figure(num=args.numpyfile, figsize=(8, 8), facecolor='w', edgecolor='w')
@@ -81,15 +81,15 @@ ax.annotate('Ground', xy=(np.deg2rad(270), 0), xytext=(8, -15), textcoords='offs
 
 # Plot patterns
 for patt in range(0, len(radii)):
-    pattplot = np.append(patterns[patt, :], patterns[patt, 0]) # Append start value to close circle
-    pattplot = pattplot / np.max(np.max(patterns)) # Normalise, based on set of patterns
+    pattplot = np.append(patterns[patt, :], patterns[patt, 0])  # Append start value to close circle
+    pattplot = pattplot / np.max(np.max(patterns))  # Normalise, based on set of patterns
     ax.plot(theta, 10 * np.log10(pattplot), label='{:.2f}m'.format(radii[patt]), marker='.', ms=6, lw=1.5)
 
 # Add Hertzian dipole plot
-#hertzplot1 = np.append(hertzian[0, :], hertzian[0, 0]) # Append start value to close circle
+# hertzplot1 = np.append(hertzian[0, :], hertzian[0, 0]) # Append start value to close circle
 #hertzplot1 = hertzplot1 / np.max(np.max(hertzian))
 #ax.plot(theta, 10 * np.log10(hertzplot1), label='Inf. dipole, 0.1m', color='black', ls='-.', lw=3)
-#hertzplot2 = np.append(hertzian[-1, :], hertzian[-1, 0]) # Append start value to close circle
+# hertzplot2 = np.append(hertzian[-1, :], hertzian[-1, 0]) # Append start value to close circle
 #hertzplot2 = hertzplot2 / np.max(np.max(hertzian))
 #ax.plot(theta, 10 * np.log10(hertzplot2), label='Inf. dipole, 0.58m', color='black', ls='--', lw=3)
 
@@ -103,13 +103,13 @@ ax.set_rmax(0)
 ax.set_rlabel_position(45)
 ax.set_yticks(np.arange(min, step, step))
 yticks = ax.get_yticks().tolist()
-yticks[-1]='0 dB'
+yticks[-1] = '0 dB'
 ax.set_yticklabels(yticks)
 
 # Grid and legend
 ax.grid(True)
 handles, existlabels = ax.get_legend_handles_labels()
-leg = ax.legend([handles[0], handles[-1]], [existlabels[0], existlabels[-1]], ncol=2, loc=(0.27,-0.12), frameon=False) # Plot just first and last legend entries
+leg = ax.legend([handles[0], handles[-1]], [existlabels[0], existlabels[-1]], ncol=2, loc=(0.27, -0.12), frameon=False)  # Plot just first and last legend entries
 #leg = ax.legend([handles[0], handles[-3], handles[-2], handles[-1]], [existlabels[0], existlabels[-3], existlabels[-2], existlabels[-1]], ncol=4, loc=(-0.13,-0.12), frameon=False)
 [legobj.set_linewidth(2) for legobj in leg.legendHandles]
 
@@ -120,4 +120,3 @@ fig.savefig(savename, dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.
 #fig.savefig(savename, dpi=150, format='png', bbox_inches='tight', pad_inches=0.1)
 
 plt.show()
-