1 | from sas.sascalc.file_converter.core.bsl_loader import CLoader |
---|
2 | from copy import deepcopy |
---|
3 | import os |
---|
4 | import numpy as np |
---|
5 | |
---|
6 | class BSLLoader(CLoader): |
---|
7 | |
---|
8 | # TODO: Change to __init__(self, filename, frame) |
---|
9 | # and parse n_(pixels/rasters) from header file |
---|
10 | def __init__(self, filename, frame): |
---|
11 | header_file = open(filename, 'r') |
---|
12 | data_info = {} |
---|
13 | is_valid = True |
---|
14 | err_msg = "" |
---|
15 | |
---|
16 | [folder, filename] = os.path.split(filename) |
---|
17 | |
---|
18 | # First 2 lines are headers |
---|
19 | header_file.readline() |
---|
20 | header_file.readline() |
---|
21 | |
---|
22 | while True: |
---|
23 | import pdb; pdb.set_trace() |
---|
24 | metadata = header_file.readline().strip() |
---|
25 | metadata = metadata.split() |
---|
26 | data_filename = header_file.readline().strip() |
---|
27 | |
---|
28 | if len(metadata) != 10: |
---|
29 | is_valid = False |
---|
30 | err_msg = "Invalid header file: {}".format(filename) |
---|
31 | break |
---|
32 | # SAS data will be in file Xnn001.mdd |
---|
33 | if data_filename != filename.replace('0.', '1.'): |
---|
34 | continue |
---|
35 | try: |
---|
36 | data_info = { |
---|
37 | 'filename': os.path.join(folder, data_filename), |
---|
38 | 'pixels': int(metadata[0]), |
---|
39 | 'rasters': int(metadata[1]), |
---|
40 | 'frames': int(metadata[2]), |
---|
41 | 'swap_bytes': int(metadata[3]) |
---|
42 | } |
---|
43 | except: |
---|
44 | is_valid = False |
---|
45 | err_msg = "Invalid metadata in header file for {}".format(filename.replace('0.', '1.')) |
---|
46 | break |
---|
47 | |
---|
48 | if not is_valid: |
---|
49 | raise Exception(err_msg) |
---|
50 | |
---|
51 | if data_info['frames'] == 1: |
---|
52 | # File is actually in OTOKO (1D) format |
---|
53 | # Number of frames is 2nd indicator, |
---|
54 | data_info['frames'] = data_info['rasters'] |
---|
55 | data_info['rasters'] = data_info['pixels'] |
---|
56 | data_info['pixels'] = 1 |
---|
57 | |
---|
58 | CLoader.__init__(self, data_info['filename'], frame, |
---|
59 | data_info['pixels'], data_info['rasters'], data_info['swap_bytes']) |
---|
60 | |
---|
61 | def __setattr__(self, name, value): |
---|
62 | if name == 'filename': |
---|
63 | return self.set_filename(value) |
---|
64 | elif name == 'frame': |
---|
65 | return self.set_frame(value) |
---|
66 | elif name == 'n_pixels': |
---|
67 | return self.set_n_pixels(value) |
---|
68 | elif name == 'n_rasters': |
---|
69 | return self.set_n_rasters(value) |
---|
70 | elif name == 'swap_bytes': |
---|
71 | return self.set_swap_bytes(value) |
---|
72 | return CLoader.__setattr__(self, name, value) |
---|
73 | |
---|
74 | def __getattr__(self, name): |
---|
75 | if name == 'filename': |
---|
76 | return self.get_filename() |
---|
77 | elif name == 'frame': |
---|
78 | return self.get_frame() |
---|
79 | elif name == 'n_pixels': |
---|
80 | return self.get_n_pixels() |
---|
81 | elif name == 'n_rasters': |
---|
82 | return self.get_n_rasters() |
---|
83 | elif name == 'swap_bytes': |
---|
84 | return self.get_swap_bytes() |
---|
85 | return CLoader.__getattr__(self, name) |
---|
86 | |
---|
87 | def create_arr(self): |
---|
88 | return np.zeros((self.n_rasters, self.n_pixels)) |
---|