From 78c98e5ad66d2172645020f3076e09432b1737e6 Mon Sep 17 00:00:00 2001
From: Rowan Cockett <rowan@3ptscience.com>
Date: Sat, 21 Mar 2015 21:59:27 -0700
Subject: [PATCH] simple transmitter.

---
 simpegEM/FDEM/SurveyFDEM.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/simpegEM/FDEM/SurveyFDEM.py b/simpegEM/FDEM/SurveyFDEM.py
index f5fd96b1..bb173055 100644
--- a/simpegEM/FDEM/SurveyFDEM.py
+++ b/simpegEM/FDEM/SurveyFDEM.py
@@ -82,12 +82,13 @@ class RxFDEM(Survey.BaseRx):
 
 
 class TxFDEM(Survey.BaseTx):
+    #TODO: Break these out into Classes of Sources.
 
     freq = None #: Frequency (float)
 
     rxPair = RxFDEM
 
-    knownTxTypes = ['VMD', 'VMD_B', 'CircularLoop']
+    knownTxTypes = ['VMD', 'VMD_B', 'CircularLoop', 'Simple']
 
     radius = None
 
@@ -184,6 +185,16 @@ class TxFDEM(Survey.BaseTx):
         elif solType == 'e' or solType == 'j':
             return C.T*mui*b_0
 
+class SimpleTxFDEM(TxFDEM):
+
+    def __init__(self, vec, freq, rxList):
+        self.vec = vec
+        self.freq = float(freq)
+        TxFDEM.__init__(self, None, 'Simple', rxList)
+
+    def getSource(self, prob):
+        return self.vec
+
 
 
 class FieldsFDEM(Problem.Fields):