# Copyright (c) 2023 Jintao Li.
# Computational and Interpretation Group (CIG),
# University of Science and Technology of China (USTC).
# All rights reserved.
"""
Overlay faults with different colors on x/y/z slices
====================================================

When the same fault volume is overlaid on several seismic slices, each slice
orientation can use a different colormap. Pass a dictionary to ``add_mask``:
``'x'`` controls inline slices, ``'y'`` controls crossline slices, and ``'z'``
controls time/depth slices.

.. image:: ../../_static/cigvis/3Dvispy/02b.png
    :alt: image
    :align: center

"""

# sphinx_gallery_thumbnail_path = '_static/cigvis/3Dvispy/02b.png'


import numpy as np
import cigvis
from pathlib import Path

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

sxp = root / 'data/rgt/sx.dat'
fxp = root / 'data/rgt/fx.dat'
ni, nx, nt = 128, 128, 128

sx = np.fromfile(sxp, np.float32).reshape(ni, nx, nt)
fx = np.fromfile(fxp, np.float32).reshape(ni, nx, nt)

nodes = cigvis.create_slices(sx, pos=[[36], [28], [84]], cmap='gray')

# Use one fault mask, but color it by the orientation of the slice plane.
fault_cmaps = {
    'x': 'Reds',    # inline slice
    'y': 'Greens',  # crossline slice
    'z': 'Blues',   # time/depth slice
}
nodes = cigvis.add_mask(
    nodes,
    fx,
    cmap=fault_cmaps,
    interpolation='nearest',
    alpha=1,
    excpt='min',
)

cigvis.plot3D(
    nodes,
    view=cigvis.Plot3DView(size=(700, 600)),
    save=cigvis.Plot3DSave(path='example.png'),
)