added new functions to default import __init__.py

renamed fieldmapy -> fieldmap
split out example for fieldmap, added map files for testing
some minor  modifications of the functions

ess/__init__.py

-__all__ = ["TraceWin", "installed", "lib_tw"]
-
+__all__ = ["fieldmap", "installed", "lib_tw", "SP_Relativity", "TraceWin" "TTF"]
 from . import TraceWin
 from . import installed
 from . import lib_tw
+from . import fieldmap

examples/fieldmap/example.py

 from ess import fieldmap
+import os
 
 '''
 Example
@@ -8,28 +9,61 @@ scans the folder and converts all files, assumes all are binary fieldmaps, to AS
 The dimension of the field is also needed.
 If you need to cut the field, the n_enter and n_exit should be adjusted.
 '''
-mypath = 'Data/FM/'          # where the fieldmaps are stored
-fieldmap_dim = 3
-#fieldmap_dim = 2
-#fieldmap_dim = 1
+plot = False                # Make comparison plots
+mypath = 'maps'             # where the fieldmaps are stored
 n_enter = 0                 #Number of lines to be removed from entrance of fieldmap
 n_exit = 0                  #Number of lines to be removed from exit of fieldmap
 
-onlyfiles = [f for f in os.listdir(mypath) if os.path.isfile(os.path.join(mypath, f))]
 
-for f in onlyfiles:
-    if f[0]!=".":
+if plot: 
+    from matplotlib import pyplot as plt
 
-        if (n_enter+n_exit>0):
-            Header, Field = fieldmap.cut_fieldmap_length(f, fieldmap_dim, n_enter, n_exit, 'b')
-        else:
-            Header, Field = fieldmap.read_binary_fieldmap(mypath+f, fieldmap_dim)
+for f in os.listdir(mypath):
+
+    print(f)
+    fullpath=os.path.join(mypath,f)
+
+    ftype=''
+    if 'ascii' in f:
+        ftype='a'
+    elif 'binary' in f:
+        ftype='b'
+    fdim=0
+    if '1D' in f:
+        fdim=1
+    elif '2D' in f:
+        fdim=2
+    elif '3D' in f:
+        fdim=3
 
-        save_to_ascii(Header, Field, fieldmap.fieldmap_dim, 'Data/FM/ASCII/', f)
+    if (n_enter or n_exit):
+	    Header, Field = fieldmap.cut_fieldmap_length(fullpath, fdim, n_enter, n_exit, ftype)
+    else:
+        Header, Field = fieldmap.read_fieldmap(fullpath, fdim, ftype)
+
+    fieldmap.save_to_ascii(Header, Field, 'new_'+f)
+    oldfield = fieldmap.field_on_axis(Header,Field, fdim)
+    origfield = fieldmap.field_on_axis_from_file(fullpath, fdim, 'a')
+    newfield = fieldmap.field_on_axis_from_file('new_'+f, fdim, 'a')
+    if plot:
+        # Plot for comparison
+        plt.figure()
+        plt.title(f)
+        if fdim==1:
+            plt.plot(oldfield,label='old')
+            plt.plot(newfield,label='new')
+            plt.plot(origfield,label='orig')
+        else:
+            plt.plot(oldfield[0,:],  oldfield[1,:], label='old')
+            plt.plot(newfield[0,:],  newfield[1,:], label='new')
+            plt.plot(origfield[0,:], origfield[1,:],label='orig')
+        plt.legend()
+    else:
+        if fdim==1:
+            avg_diff=sum(newfield-oldfield)/len(newfield)
+        else:
+            avg_diff=sum(newfield[1,:]-oldfield[1,:])/len(newfield[1,:])
+        print("Average diff",avg_diff)
 
-test = False
-#%pylab inline
-if test == True:
-    #field = field_on_axis_from_file('Data/FM/MB_W_coupler.edz', 3, 'b')
-    field = field_on_axis_from_file('Data/FM/ASCII/Spoke_W_coupler.edz', 3, 'a')
-    plot(field[0,:], field[1,:]) 
+if plot:
+    plt.show()