.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "examples/plot_dc_resistivity.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_examples_plot_dc_resistivity.py>`
        to download the full example code

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_examples_plot_dc_resistivity.py:


Basic Forward 2D DC Resistivity
===============================

2D DC forward modeling example with Tensor and Curvilinear Meshes

.. GENERATED FROM PYTHON SOURCE LINES 7-67



.. image-sg:: /examples/images/sphx_glr_plot_dc_resistivity_001.png
   :alt: TensorMesh, CurvilinearMesh
   :srcset: /examples/images/sphx_glr_plot_dc_resistivity_001.png
   :class: sphx-glr-single-img


.. rst-class:: sphx-glr-script-out

 .. code-block:: none

    /home/vsts/work/1/s/examples/plot_dc_resistivity.py:31: DeprecationWarning: The closest_points_index utilty function has been moved to be a method of a class object. Please access it as mesh.closest_points_index(). This will be removed in a future version of discretize
      txind = discretize.utils.closest_points_index(mesh, pts)






|

.. code-block:: default

    import discretize
    from pymatsolver import SolverLU
    import numpy as np
    import matplotlib.pyplot as plt


    def run(plotIt=True):
        # Step1: Generate Tensor and Curvilinear Mesh
        sz = [40, 40]
        tM = discretize.TensorMesh(sz)
        rM = discretize.CurvilinearMesh(
            discretize.utils.example_curvilinear_grid(sz, "rotate")
        )

        # Step2: Direct Current (DC) operator
        def DCfun(mesh, pts):
            D = mesh.face_divergence
            sigma = 1e-2 * np.ones(mesh.nC)
            MsigI = mesh.get_face_inner_product(
                sigma, invert_model=True, invert_matrix=True
            )
            A = -D * MsigI * D.T
            A[-1, -1] /= mesh.cell_volumes[-1]  # Remove null space
            rhs = np.zeros(mesh.nC)
            txind = discretize.utils.closest_points_index(mesh, pts)
            rhs[txind] = np.r_[1, -1]
            return A, rhs

        pts = np.vstack((np.r_[0.25, 0.5], np.r_[0.75, 0.5]))

        # Step3: Solve DC problem (LU solver)
        AtM, rhstM = DCfun(tM, pts)
        AinvtM = SolverLU(AtM)
        phitM = AinvtM * rhstM

        ArM, rhsrM = DCfun(rM, pts)
        AinvrM = SolverLU(ArM)
        phirM = AinvrM * rhsrM

        if not plotIt:
            return

        # Step4: Making Figure
        fig, axes = plt.subplots(1, 2, figsize=(12 * 1.2, 4 * 1.2))
        vmin, vmax = phitM.min(), phitM.max()

        dat = tM.plot_image(phitM, ax=axes[0], clim=(vmin, vmax), grid=True)
        cb0 = plt.colorbar(dat[0], ax=axes[0])
        cb0.set_label("Voltage (V)")
        axes[0].set_title("TensorMesh")

        dat = rM.plot_image(phirM, ax=axes[1], clim=(vmin, vmax), grid=True)
        cb1 = plt.colorbar(dat[0], ax=axes[1])
        cb1.set_label("Voltage (V)")
        axes[1].set_title("CurvilinearMesh")


    if __name__ == "__main__":
        run()
        plt.show()


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 0.376 seconds)


.. _sphx_glr_download_examples_plot_dc_resistivity.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example




    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: plot_dc_resistivity.py <plot_dc_resistivity.py>`

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: plot_dc_resistivity.ipynb <plot_dc_resistivity.ipynb>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_