From 7d295dc57e9ad5a47ffd53f66c915e3677ee8c4c Mon Sep 17 00:00:00 2001 From: Dave Marchant Date: Mon, 18 Nov 2013 11:24:27 -0800 Subject: [PATCH] Fix in example notebook. --- SimPEG/visulize/vtk/vtkView.py | 6 +- notebooks/3DRenderingWithvtkTools.ipynb | 95 ++++++++++++------------- 2 files changed, 47 insertions(+), 54 deletions(-) diff --git a/SimPEG/visulize/vtk/vtkView.py b/SimPEG/visulize/vtk/vtkView.py index 11e83ed4..2ce984df 100644 --- a/SimPEG/visulize/vtk/vtkView.py +++ b/SimPEG/visulize/vtk/vtkView.py @@ -75,11 +75,7 @@ class vtkView(object): def Show(self): """ - Open the VTK figure window and show the mesh. - - Inputs: - param: str imageType: type of image {'cell','face','edge'} - + Open the VTK figure window and show the mesh. """ #vtkSP = simpeg.visulize.vtk.vtkTools import SimPEG.visulize.vtk.vtkTools as vtkSP diff --git a/notebooks/3DRenderingWithvtkTools.ipynb b/notebooks/3DRenderingWithvtkTools.ipynb index c6f1663d..b0daa408 100644 --- a/notebooks/3DRenderingWithvtkTools.ipynb +++ b/notebooks/3DRenderingWithvtkTools.ipynb @@ -1,6 +1,6 @@ { "metadata": { - "name": "3D rendering with vtkTools" + "name": "" }, "nbformat": 3, "nbformat_minor": 0, @@ -10,64 +10,61 @@ { "cell_type": "code", "collapsed": false, - "input": "import numpy as np, vtk\nimport SimPEG as simpeg", - "language": "python", - "metadata": {}, - "outputs": [], - "prompt_number": 1 - }, - { - "cell_type": "code", - "collapsed": false, - "input": "#Make a mesh and model\nx0 = np.zeros(3)\nh1 = np.ones(20)*50\nh2 = np.ones(10)*100\nh3 = np.ones(5)*200\n\nmesh = simpeg.mesh.TensorMesh([h1,h2,h3],x0)\n", - "language": "python", - "metadata": {}, - "outputs": [], - "prompt_number": 2 - }, - { - "cell_type": "code", - "collapsed": false, - "input": "# Make a models that correspond to the cells, faces and edges.\nmodels = {'cell':{'Test':np.arange(0,mesh.nC),'AllOnce':np.ones(mesh.nC)},'face':{'Test':np.arange(0,np.sum(mesh.nF)),'AllOnce':np.ones(np.sum(mesh.nF))},'edge':{'Test':np.arange(0,np.sum(mesh.nE)),'AllOnce':np.ones(np.sum(mesh.nE))}}\n# Make the vtk viewer object.\nvtkViewer = simpeg.visulize.vtk.vtkView(mesh,models) \n ", - "language": "python", - "metadata": {}, - "outputs": [], - "prompt_number": 3 - }, - { - "cell_type": "code", - "collapsed": false, - "input": "# Show the image \nvtkViewer.Show(imageType='cell')\n", - "language": "python", - "metadata": {}, - "outputs": [], - "prompt_number": 4 - }, - { - "cell_type": "code", - "collapsed": false, - "input": "mesh.nCxx", - "language": "python", - "metadata": {}, - "outputs": [ - { - "output_type": "pyout", - "prompt_number": 7, - "text": "20" - } + "input": [ + "import numpy as np, vtk\n", + "import SimPEG as simpeg" ], + "language": "python", + "metadata": {}, + "outputs": [], + "prompt_number": 5 + }, + { + "cell_type": "code", + "collapsed": false, + "input": [ + "#Make a mesh and model\n", + "x0 = np.zeros(3)\n", + "h1 = np.ones(20)*50\n", + "h2 = np.ones(10)*100\n", + "h3 = np.ones(5)*200\n", + "\n", + "mesh = simpeg.mesh.TensorMesh([h1,h2,h3],x0)\n" + ], + "language": "python", + "metadata": {}, + "outputs": [], + "prompt_number": 6 + }, + { + "cell_type": "code", + "collapsed": false, + "input": [ + "# Make a models that correspond to the cells, faces and edges.\n", + "models = {'cell':{'Test':np.arange(0,mesh.nC),'AllOnce':np.ones(mesh.nC)},'face':{'Test':np.arange(0,np.sum(mesh.nF)),'AllOnce':np.ones(np.sum(mesh.nF))},'edge':{'Test':np.arange(0,np.sum(mesh.nE)),'AllOnce':np.ones(np.sum(mesh.nE))}}\n", + "# Make the vtk viewer object.\n", + "vtkViewer = simpeg.visulize.vtk.vtkView(mesh,models) \n", + " " + ], + "language": "python", + "metadata": {}, + "outputs": [], "prompt_number": 7 }, { "cell_type": "code", "collapsed": false, - "input": "", + "input": [ + "# Show the image \n", + "vtkViewer.Show()\n" + ], "language": "python", "metadata": {}, - "outputs": [] + "outputs": [], + "prompt_number": "*" } ], "metadata": {} } ] -} +} \ No newline at end of file