-
Notifications
You must be signed in to change notification settings - Fork 0
/
lut.py
20 lines (17 loc) · 1 KB
/
lut.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import gzip
import numpy as np
import struct
class LUT:
def __init__(self, file):
print("LUT Reading Started")
with gzip.open(file, "rb") as f:
self.f = f
self.load()
def load(self):
self.m1LUT250 = np.frombuffer(self.f.read(4*2*40*4*2), dtype=np.dtype('>f4')).reshape((2, 40, 4, 2)) # noqa
self.m1LUT500 = np.frombuffer(self.f.read(4*5*20*2*2), dtype=np.dtype('>f4')).reshape((5, 20, 2, 2)) # noqa
self.m1LUT1000 = np.frombuffer(self.f.read(4*15*10*2), dtype=np.dtype('>f4')).reshape((15, 10, 2)) # noqa
self.f.seek((4*2*40*4*2 + 4*5*20*2*2 + 4*15*10*2), 1)
self.RVSrefLUT250 = np.frombuffer(self.f.read(4*2*40*2*1354), dtype=np.dtype('>f4')).reshape((2, 40, 2, 1354)) # noqa
self.RVSrefLUT500 = np.frombuffer(self.f.read(4*5*20*2*1354), dtype=np.dtype('>f4')).reshape((5, 20, 2, 1354)) # noqa
self.RVSrefLUT1000 = np.frombuffer(self.f.read(4*15*10*2*1354), dtype=np.dtype('>f4')).reshape((15, 10, 2, 1354)) # noqa