From 3221b3c7504df1ba613730b3923e0c8fc1de3ee3 Mon Sep 17 00:00:00 2001 From: rowanc1 Date: Thu, 5 Dec 2013 11:31:43 -0800 Subject: [PATCH] Updates to video function (skip frames) --- SimPEG/mesh/TensorView.py | 8 +++++--- SimPEG/utils/ModelBuilder.py | 2 +- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/SimPEG/mesh/TensorView.py b/SimPEG/mesh/TensorView.py index 9956565d..a745c1cd 100644 --- a/SimPEG/mesh/TensorView.py +++ b/SimPEG/mesh/TensorView.py @@ -372,7 +372,7 @@ class TensorView(object): return animate(fig, animateFrame, frames=mesh.nCv['xyz'.index(normal)]) - def video(mesh,var,function,figsize=(10,8),colorbar=True): + def video(mesh, var, function, figsize=(10, 8), colorbar=True, skip=1): """ Call a function for a list of models to create a video. @@ -393,9 +393,11 @@ class TensorView(object): if colorbar: plt.colorbar(function(var[0],ax,clim,tlt,0)) - def animateFrame(i): + frames = np.arange(0,len(var),skip) + def animateFrame(j): + i = frames[j] function(var[i],ax,clim,tlt,i) - return animate(fig, animateFrame, frames=len(var)) + return animate(fig, animateFrame, frames=len(frames)) diff --git a/SimPEG/utils/ModelBuilder.py b/SimPEG/utils/ModelBuilder.py index 56df6250..969551ed 100644 --- a/SimPEG/utils/ModelBuilder.py +++ b/SimPEG/utils/ModelBuilder.py @@ -172,7 +172,7 @@ def randomModel(shape, seed=None, anisotropy=None, its=100, bounds=[0,1]): if len(shape) is 1: smth = np.array([1,10.,1],dtype=float) elif len(shape) is 2: - smth = np.array([[1,2,1],[7,10,7],[1,2,1]],dtype=float) + smth = np.array([[1,7,1],[2,10,2],[1,7,1]],dtype=float) elif len(shape) is 3: kernal = np.array([1,4,1], dtype=float).reshape((1,3)) smth = np.array(sp.kron(sp.kron(kernal,kernal.T).todense()[:],kernal).todense()).reshape((3,3,3))