File: extractBinaryData.py

package info (click to toggle)
labplot 2.12.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 78,500 kB
  • sloc: cpp: 241,048; ansic: 6,324; python: 915; xml: 400; yacc: 237; sh: 221; awk: 35; makefile: 11
file content (48 lines) | stat: -rw-r--r-- 1,602 bytes parent folder | download | duplicates (3)
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
import sys
import struct

if __name__ == "__main__":
    if len(sys.argv) < 2:
        raise Exception("Pass the filename as second argument!")
    filename = sys.argv[1]

    with open(filename, "rb") as f:

        number_variables = 0
        number_points = 0
        line = f.readline()
        flags = b""
        while b"Binary:" not in line:
            if b"No. Variables: " in line:
                number_variables = int(line.split(b"No. Variables: ")[1])
            if b"No. Points: " in line:
                number_points = int(line.split(b"No. Points: ")[1])
            if b"Flags: " in line:
                flags = line.split(b"Flags: ")[1]
            line = f.readline()

        isComplex = b"complex" in flags
        #d = bytearray(f.read())
        data_str = ""
        for r in range(number_points):
            line = ''
            for c in range(number_variables):
                value = struct.unpack('d', f.read(8))[0]  # every data is a double (8byte)
                line += f'"{value:{1}.{15}e}"'
                if isComplex:
                    line += ", "
                    value = struct.unpack('d', f.read(8))[0]  # every data is a double (8byte)
                    line += f'"{value:{1}.{15}e}"'
                if c < number_variables - 1:
                    line += ", "


            line = f"{{{line}}}"
            if r < number_points - 1:
                line += ",\n"
            data_str += line
        data_str = f"QVector<QStringList> refData = {{{data_str}}};"

        with open(filename + ".data", "w") as f:
            f.write(data_str)