File: sorting-permutation.py

package info (click to toggle)
segyio 1.8.3-1.2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 3,724 kB
  • sloc: cpp: 17,823; ansic: 4,637; python: 4,326; makefile: 38; sh: 23
file content (82 lines) | stat: -rw-r--r-- 3,123 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import sys
import os
import shutil
import segyio

def pathjoin( prefix, path):
    directory, base = os.path.split(path)
    pref, suff = os.path.splitext(base)
    filename = pref + '-' + prefix + suff
    return os.path.join(directory, filename)

# This script reverse the traces for a given file, by iline, xline and offset in
# all combinations of the three.

def main():
    if len(sys.argv) < 2:
        sys.exit("Usage: {} [source-file] [destination-file]".format(sys.argv[0]))

    sourcefile = sys.argv[1]
    destfile = sys.argv[2] if len(sys.argv) > 2 else sourcefile

    for suf in ['dec-il-xl-off',
                'dec-il-inc-xl-off',
                'dec-xl-inc-il-off',
                'dec-il-xl-inc-off',
                'dec-il-off-inc-xl',
                'dec-xl-off-inc-il',
                'dec-off-inc-il-xl']:
        fname = pathjoin(suf, destfile)
        shutil.copyfile(sourcefile, fname)

        with segyio.open(sourcefile) as src, segyio.open(fname, 'r+') as dst:
            for i in range(1 + src.ext_headers):
                dst.text[i] = src.text[i]

            dst.bin = src.bin
            dst.trace = src.trace
            dst.header = src.header

    with segyio.open(sourcefile) as src:
        with segyio.open(pathjoin("dec-il-xl-off", destfile), 'r+') as dst:
            dst.header = src.header[::-1]
            dst.trace = src.trace[::-1]

    with segyio.open(sourcefile) as src:
        with segyio.open(pathjoin("dec-il-inc-xl-off", destfile), 'r+') as dst:
            dst.header.iline[::, ::] = src.header.iline[::-1, ::]
            dst.iline[::, ::] = src.iline[::-1, ::]

    with segyio.open(sourcefile) as src:
        with segyio.open(pathjoin("dec-xl-inc-il-off", destfile), 'r+') as dst:
            dst.header.xline[::, ::] = src.header.xline[::-1, ::]
            dst.xline[::, ::] = src.xline[::-1, ::]

    with segyio.open(sourcefile) as src:
        with segyio.open(pathjoin("dec-il-xl-inc-off", destfile), 'r+') as dst:
            dst.header = src.header[::-1]
            dst.trace = src.trace[::-1]

            for i in range(0, len(dst.trace), 2):
                dst.trace[i], dst.trace[i + 1] = dst.trace[i + 1], dst.trace[i]
                dst.header[i], dst.header[i + 1] = dst.header[i + 1], dst.header[i]

    with segyio.open(sourcefile) as src:
        with segyio.open(pathjoin("dec-il-off-inc-xl", destfile), 'r+') as dst:
            dst.header.iline[::, ::] = src.header.iline[::-1, ::-1]
            dst.iline[::, ::] = src.iline[::-1, ::-1]

    with segyio.open(sourcefile) as src:
        with segyio.open(pathjoin("dec-xl-off-inc-il", destfile), 'r+') as dst:
            dst.header.iline[::, ::] = src.header.iline[::-1, ::-1]
            dst.header.xline[::, ::] = src.header.xline[::-1, ::-1]
            dst.iline[::, ::] = src.iline[::-1, ::]

    with segyio.open(sourcefile) as src:
        with segyio.open(pathjoin("dec-off-inc-il-xl", destfile), 'r+') as dst:
            dst.header.xline[::, ::] = src.header.xline[::, ::-1]
            dst.xline[::, ::] = src.xline[::, ::-1]


if __name__ == '__main__':
    main()