F3 demo2

F3 demo, https://dataunderground.org/dataset/f3.

Interpretation data is provided by Xinming Wu, all data is big endian

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

import numpy as np
import cigvis
from cigvis import colormap


def load_wellLog(p):
    nlog = 4
    npoints = 2121
    x = [259, 619, 339, 141]
    y = [33, 545, 704, 84]
    z = np.arange(0, 0.2 * npoints, 0.2)
    v = np.fromfile(p, np.float32).reshape(nlog, npoints)
    v = 0.5 * np.log(v)

    nodes = []
    for i in range(nlog):
        points = np.c_[np.ones(npoints) * x[i], np.ones(npoints) * y[i], z]
        nodes += cigvis.create_well_logs(points,
                                         v[i],
                                         cyclinder=False,
                                         radius_tube=[2, 5],
                                         null_value=-999.25)

    return nodes


if __name__ == '__main__':

    root = '/Volumes/T7/DATA/cigvisdata/F3/'
    seisp = root + 'seis.dat'
    saltp = root + 'salt.dat'
    hz1p = root + 'hz1.dat'
    hz2p = root + 'hz.dat'
    unc1p = root + 'unc1.dat'
    unc2p = root + 'unc2.dat'
    ni, nx, nt = 591, 951, 362
    shape = (ni, nx, nt)
    step1 = 2
    step2 = 4

    # seismic
    seis = np.memmap(seisp, np.float32, 'c', shape=shape)
    # overlay
    inter = np.memmap(root + 'overlay.dat', np.float32, 'c', shape=shape)

    fg_cmap = colormap.set_alpha('jet', 0.6)
    fg_clim = [inter.max() * 0.15, inter.max() * 0.5]
    nodes = cigvis.create_slices(seis,
                                 pos=[ni - 2, 25, nt - 2],
                                 cmap='gray',
                                 clim=[-2.0, 1.5])
    nodes = cigvis.add_mask(nodes,
                            inter,
                            clims=fg_clim,
                            cmaps=fg_cmap,
                            interpolation='nearest')

    salt = np.memmap(saltp, np.float32, 'c', shape=shape)
    nodes += cigvis.create_bodys(salt, 0.0, 0.0, color='cyan')

    hz2 = np.fromfile(hz2p, np.float32).reshape(ni, nx)
    nodes += cigvis.create_surfaces([hz2],
                                    value_type='yellow',
                                    step1=step1,
                                    step2=step2)

    unc = np.fromfile(root + 'unc.dat', np.float32).reshape(shape)
    unc2 = np.fromfile(unc2p, np.float32).reshape(ni, nx)
    nodes += cigvis.create_surfaces([unc2],
                                    volume=unc,
                                    value_type='amp',
                                    step1=step1,
                                    step2=step2)

    nodes += load_wellLog(root + 'logs.dat')

    nodes += cigvis.create_fault_skin(root + 'skins/')

    nodes += cigvis.create_axis(seis.shape,
                                'axis',
                                'auto',
                                ticks_font_size=28,
                                labels_font_size=32,
                                ticks_length=6)
    nodes += cigvis.create_colorbar_from_nodes(nodes, 'Impedance', 'mask')

    cigvis.plot3D(nodes,
                  azimuth=-65.0,
                  elevation=22.0,
                  fov=15.0,
                  axis_scales=(1, 1, 1.7),
                  zoom_factor=1.4)