minor changes

SimPEG/EM/Static/DC/SrcDC.py

@@ -51,3 +51,36 @@ class Pole(BaseSrc):
             q = self.current * mkvc(q)
         return q
 
+
+# class Dipole_ky(BaseSrc):
+
+#     def __init__(self, rxList, locA, locB, **kwargs):
+#         assert locA.shape == locB.shape, 'Shape of locA and locB should be the same'
+#         self.loc = [locA[[0,2]], locB[[0,2]]]
+#         BaseSrc.__init__(self, rxList, **kwargs)
+
+#     def eval(self, prob):
+#         if prob._formulation == 'HJ':
+#             inds = closestPoints(prob.mesh, self.loc, gridLoc='CC')
+#             q = np.zeros(prob.mesh.nC)
+#             q[inds] = self.current * np.r_[1., -1.]
+#         elif prob._formulation == 'EB':
+#             qa = prob.mesh.getInterpolationMat(self.loc[0], locType='N').todense()
+#             qb = -prob.mesh.getInterpolationMat(self.loc[1], locType='N').todense()
+#             q = self.current * mkvc(qa+qb)
+#         return q
+
+# class Pole_ky(BaseSrc):
+
+#     def __init__(self, rxList, loc, **kwargs):
+#         BaseSrc.__init__(self, rxList, loc=loc, **kwargs)
+
+#     def eval(self, prob):
+#         if prob._formulation == 'HJ':
+#             inds = closestPoints(prob.mesh, self.loc[[0,2]])
+#             q = np.zeros(prob.mesh.nC)
+#             q[inds] = self.current * np.r_[1.]
+#         elif prob._formulation == 'EB':
+#             q = prob.mesh.getInterpolationMat(self.loc[[0,2]], locType='N').todense()
+#             q = self.current * mkvc(q)
+#         return q

SimPEG/EM/Static/Utils/StaticUtils.py

@@ -213,6 +213,7 @@ def gen_DCIPsurvey(endl, mesh, stype, a, b, n):
     # Pole-dipole: Moving pole on one end -> [A a MN1 a MN2 ... MNn a B]
     SrcList = []
 
+
     if stype != 'gradient':
 
         for ii in range(0, int(nstn)-1):
@@ -261,7 +262,6 @@ def gen_DCIPsurvey(endl, mesh, stype, a, b, n):
                     srcClass = DC.Src.Pole([rxClass], M[ii,:])
             SrcList.append(srcClass)
 
-    # TODO: test gradient array!
     elif stype == 'gradient':
 
         # Gradient survey only requires Tx at end of line and creates a square