.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "gallery/more_demos/011-disc_cbar.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_011-disc_cbar.py>`
        to download the full example code

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

.. _sphx_glr_gallery_more_demos_011-disc_cbar.py:


Discrete colorbar
======================

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

.. GENERATED FROM PYTHON SOURCE LINES 10-93

.. code-block:: default


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

    import numpy as np
    import cigvis
    from cigvis import colormap
    from pathlib import Path
    root = Path(__file__).resolve().parent.parent.parent


    def show_discrete_cbar(d):
        """
        添加离散的colorbar

        离散数据建议将interpolation设置为 'nearest'
        """
        nodes1 = cigvis.create_slices(d, cmap='jet', interpolation='nearest')
        cbar1 = cigvis.create_colorbar(
            cmap='jet',
            clim=[d.min(), d.max()],
            discrete=True,
            disc_ticks=[np.unique(d), ['CA', 'CB', 'CC', 'CD', '100']],
            label_str='Facies')
        nodes1.append(cbar1)
        """
        离散的数据, 当数据中有outline时, 会导致其他值的颜色很接近
        """
        d[d == 1] = 100
        nodes2 = cigvis.create_slices(d, cmap='jet', interpolation='nearest')
        nodes2 += cigvis.create_colorbar_from_nodes(
            nodes2,
            select='slices',
            discrete=True,
            disc_ticks=[np.unique(d).astype(int)],
            label_str='Facies')
        """
        自定义 colormap 可以解决 cbar2 的问题
        """
        values = np.unique(d)
        colors = ['gray', 'green', '#7f6589', 'blue', (0, 0.5, 0.5)]
        cmap = colormap.custom_disc_cmap(values, colors)
        nodes3 = cigvis.create_slices(d, cmap=cmap, interpolation='nearest')
        nodes3 += cigvis.create_colorbar_from_nodes(
            nodes3,
            label_str='Facies',
            select='slices',
            discrete=True,
            disc_ticks=[values.astype(int)])
        """
        可以使用 get_colors_from_cmap 获取颜色

        然后自定义 colormap 解决 cbar2 的问题
        """
        values = np.unique(d)

        # 获取均匀分布的颜色
        colors = colormap.get_colors_from_cmap('jet',
                                               clim=[0, len(values) - 1],
                                               values=np.arange(0, len(values)))

        cmap = colormap.custom_disc_cmap(values, colors)
        nodes4 = cigvis.create_slices(d, cmap=cmap, interpolation='nearest')
        nodes4 += cigvis.create_colorbar_from_nodes(
            nodes3,
            label_str='Facies',
            select='slices',
            discrete=True,
            disc_ticks=[values.astype(int)])

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


    if __name__ == '__main__':
        lxp = root / 'data/label_h360x600x400.dat'
        ni, nx, nt = 400, 600, 360

        lx = np.fromfile(lxp, np.float32).reshape(ni, nx, nt)

        show_discrete_cbar(lx)


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

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


.. _sphx_glr_download_gallery_more_demos_011-disc_cbar.py:

.. only:: html

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




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

      :download:`Download Python source code: 011-disc_cbar.py <011-disc_cbar.py>`

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

      :download:`Download Jupyter notebook: 011-disc_cbar.ipynb <011-disc_cbar.ipynb>`


.. only:: html

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

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