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

.. only:: html

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

        :ref:`Go to the end <sphx_glr_download_gallery_more_demos_030-overlay2.py>`
        to download the full example code

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

.. _sphx_glr_gallery_more_demos_030-overlay2.py:


Overlay display of 3D data (continuous)
===========================================

.. image:: ../../_static/cigvis/more_demos/030.png
    :alt: image
    :align: center

.. GENERATED FROM PYTHON SOURCE LINES 10-101

.. code-block:: default


    # sphinx_gallery_thumbnail_path = '_static/cigvis/more_demos/030.png'

    import numpy as np
    import cigvis
    from cigvis import colormap
    from pathlib import Path

    root = Path(__file__).resolve().parent.parent.parent


    def show(bg, fg, fx):
        """
        seismic and RGT
        """
        cmap1 = colormap.set_alpha('jet', 0.5)
        nodes1 = cigvis.create_slices(bg, pos=[[36], [28], [84]], cmap='gray')
        nodes1 = cigvis.add_mask(nodes1, fg, cmaps=cmap1)
        nodes1 += cigvis.create_colorbar_from_nodes(nodes1,
                                                    label_str='RGT',
                                                    select='mask')
        """
        mask RGT 的底部 (mask fg > 120)
        """
        cmap2 = colormap.set_alpha_except_max('jet', 0.5)
        nodes2 = cigvis.create_slices(bg, pos=[[36], [28], [84]], cmap='gray')
        nodes2 = cigvis.add_mask(nodes2, fg, clims=[fg.min(), 120], cmaps=cmap2)
        nodes2 += cigvis.create_colorbar_from_nodes(nodes2,
                                                    label_str='RGT',
                                                    select='mask')
        """
        seismic and fault
        """
        # Note: set alpha = 1 is the best
        cmap3 = colormap.set_alpha_except_min('jet', 1)
        nodes3 = cigvis.create_slices(bg, pos=[[36], [28], [84]], cmap='gray')
        nodes3 = cigvis.add_mask(nodes3, [fg, fx],
                                 cmaps=[cmap2, cmap3],
                                 interpolation=['linear', 'nearest'])

        values = np.unique(fx).astype(int)
        values = values[1:]
        labels = ['fx' + str(i) for i in values]

        # set idx=1, means select the second mask (cmap3 in this case)
        nodes3 += cigvis.create_colorbar_from_nodes(
            nodes3,
            label_str='Fault',
            select='mask',
            idx=1,
            discrete=True,
            disc_ticks=[values, labels],
        )
        """
        seismic, masked RGT (mask fg > 120) and fault
        """
        cmap4 = [
            colormap.set_alpha_except_max('jet', 0.5),
            colormap.set_alpha_except_min('jet', 1)
        ]
        interp = ['cubic', 'nearest']
        fg_clim = [[fg.min(), 120], [fx.min(), fx.max()]]
        nodes4 = cigvis.create_slices(bg, pos=[[36], [28], [84]], cmap='gray')
        nodes4 = cigvis.add_mask(nodes4, [fg, fx],
                                 clims=fg_clim,
                                 cmaps=cmap4,
                                 interpolation=interp)

        nodes4 += cigvis.create_colorbar_from_nodes(nodes4,
                                                    label_str='RGT',
                                                    select='mask')

        cigvis.plot3D([nodes1, nodes2, nodes3, nodes4],
                      grid=(2, 2),
                      size=(1000, 700),
                      cbar_region_ratio=0.18,
                      share=True,
                      savename='example.png')


    if __name__ == '__main__':
        sxp = root / 'data/rgt/sx.dat'
        uxp = root / 'data/rgt/ux.dat'
        fxp = root / 'data/rgt/fx.dat'
        ni, nx, nt = 128, 128, 128

        sx = np.memmap(sxp, np.float32, 'c', shape=(ni, nx, nt))
        fx = np.memmap(fxp, np.float32, 'c', shape=(ni, nx, nt))
        ux = np.memmap(uxp, np.float32, 'c', shape=(ni, nx, nt))

        show(sx, ux, fx)


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

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


.. _sphx_glr_download_gallery_more_demos_030-overlay2.py:

.. only:: html

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




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

      :download:`Download Python source code: 030-overlay2.py <030-overlay2.py>`

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

      :download:`Download Jupyter notebook: 030-overlay2.ipynb <030-overlay2.ipynb>`


.. only:: html

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

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