pmasw/PassiveData.py at main · fosghen/pmasw · GitHub

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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
"""Класс пассивых сейсмических данных."""

import numpy as np


class PassiveData:
    """
    Класс пассивных сейсмических данных.

    Класс хранит сейсмическую запись и её
    время дескретизации. Используется совместно
    с DataLoader, PassiveProcesser, PMASW.

    ...

    Attributes
    ----------
    dt : float
        Время дескретизации сигнала.

    seismogram : ndarray[dtype: float64, dim = 2]
            Массив сейсмичсеской записи.
            [tempor_axis, spatial_axis]

    """

    def __init__(self, data, dt):
        """
        Определение сейсмических данных и врмени дескретизации.

        Parameters
        ----------
        data : ndarray[dtype: float64, dim = 2]
            Массив сейсмичсеской записи.
            [tempor_axis, spatial_axis]

        dt : float
            Время десретизации сигнала.

        fs : int
            Частота дескретизации сигнала.

        nt : int
            Количество отсчётов по времени.

        nx : int
            Количество трасс.

        """
        self.seismogram = data
        self.dt = dt
        self._fs = int(1 / dt)


    @property
    def dt(self):
        """
        Возвращает время дескретизации обрабатываемого сигнала.

        Returns
        -------
        dt : float
            Время дескретизации сигнала.

        """
        return self._dt


    @dt.setter
    def dt(self, value):
        """
        Устанавливает время дескретизации обрабатываемого сигнала.

        Parameters
        ----------
        value : float | int
            Время дескретизации сигнала.

        """
        if (not isinstance(value, float)) and (not isinstance(value, int)):
            raise ValueError("Время дескретизации должно быть числом")

        if value < 0:
            raise ValueError("Время дескретизации должно быть"
                             " положительным числом")

        self._dt = float(value)
        self._fs = int(1 / self._dt)


    @property
    def fs(self):
        """
        Возвращает частоту дескретизации.

        Returns
        -------
        fs : int
            Частота дескретизации.

        """
        return int(1 / self._dt)


    @fs.setter
    def fs(self, value):
        """
        Устанавливает частоту дескретизации.

        Parameters
        ----------
        value : int
            Частота дескретизации.

        """
        if not isinstance(value, int):
            raise ValueError("Частота дескретизации должна быть целым числом")

        if value < 0:
            raise ValueError("Частота дескретизации должна быть"
                             " положительным числом")

        self._fs = value
        self._dt = float(1 / self._fs)


    @property
    def nt(self):
        """
        Возвращает количество отсчётов по времени.

        Returns
        -------
        nt : int
            Количество отсчётов по времени.

        """
        return self._nt


    @property
    def nx(self):
        """
        Возвращает количество трасс.

        Returns
        -------
        nx : int
            Количество трасс.

        """
        return self._nx


    @property
    def seismogram(self):
        """
        Возвращает сейсмическую запись.

        Returns
        -------
        seismogram : ndarray[dtype: float64, dim = 2]
            Массив сейсмичсеской записи.
            [tempor_axis, spatial_axis]

        """
        return self._seismogram


    @seismogram.setter
    def seismogram(self, value):
        """
        Устанавливает сейсмическую запись.

        Parameters
        ----------
        value : ndarray[dtype: float64 | int32, dim = 2]
            Массив сейсмичкской записи.
            [tempor_axis, spatial_axis]

        """
        if not isinstance(value, np.ndarray):
            raise ValueError("Подаваемый массив должен быть типа"
                             "numpy.ndarray")

        if len(value) == 0:
            raise ValueError("Подаваемый массив не должен быть пустым")

        if (value.dtype != np.int32) and (value.dtype != np.float64):
            raise ValueError("Подаваемый массив должен состоять из типов"
                             "данных float64 или int32")

        if not isinstance(value[0], np.ndarray):
            raise ValueError("Подаваемый массив должен состоять из "
                             "массивов типа numpy.array")

        if len(value[0]) == 0:
            raise ValueError("Подаваемый массив не должен быть пустым")

        if len(value.shape) != 2:
            raise ValueError("Размер массива должен быть равен 2")

        self._seismogram = np.float64(value)
        self._nt, self._nx = value.shape