From eafa2bfbe8239a520a25c3d3472e211eaff4c7b0 Mon Sep 17 00:00:00 2001
From: Rowan Cockett <rowan@3ptscience.com>
Date: Sun, 15 Nov 2015 19:08:40 -0800
Subject: [PATCH] Minor performance updates.

---
 SimPEG/Mesh/TreeMesh.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/SimPEG/Mesh/TreeMesh.py b/SimPEG/Mesh/TreeMesh.py
index 60410607..2de403ce 100644
--- a/SimPEG/Mesh/TreeMesh.py
+++ b/SimPEG/Mesh/TreeMesh.py
@@ -380,7 +380,7 @@ class TreeMesh(BaseMesh, InnerProducts):
             if p[-1] >= self.levels: continue
             do = function(Cell(self, cell, p)) > p[-1]
             if do:
-                recurse += self._refineCell(cell)
+                recurse += self._refineCell(cell, p)
 
         if verbose: print '   ', time.time() - tic
 
@@ -391,9 +391,9 @@ class TreeMesh(BaseMesh, InnerProducts):
             self.balance()
         return recurse
 
-    def _refineCell(self, pointer):
-        pointer = self._asPointer(pointer)
-        ind = self._asIndex(pointer)
+    def _refineCell(self, ind, pointer=None):
+        ind = self._asIndex(ind)
+        pointer = self._asPointer(pointer if pointer is not None else ind)
         assert ind in self
         h = self._levelWidth(pointer[-1])/2 # halfWidth
         nL = pointer[-1] + 1 # new level