IonogramViewer2/ionogram_tester.py at master · Albom/IonogramViewer2 · 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
from os import path
from datetime import datetime
import fnmatch
from ionogram_visrc2t import IonogramVisrc2t
from ionogram_shigaraki import IonogramShigaraki
from ionogram_dps_amp import IonogramDpsAmp
from ionogram_ips42 import IonogramIps42
from ionogram_bazis import IonogramBazis
from ionogram_rinan import IonogramRinan
from ionogram_karazin import IonogramKarazin


class IonogramTester:
    def __init__(self):
        self.FILE_FORMATS = {
            "Unknown": {"class_name": "Unknown"},
            "IPS42": {
                "class_name": "IonogramIps42",
                "sizes": [36928],
                "patterns": ["??h??m.ion"],
            },
            "DPS_AMP": {"class_name": "IonogramDpsAmp"},
            "KARAZIN": {"class_name": "IonogramKarazin", "patterns": ["??-??.dat"]},
            "RINAN": {
                "class_name": "IonogramRinan",
                "patterns": [
                    "????????_????_iono.ion",
                    "????????_????_c?_iono.ion",
                    "????????_????_iono.pion",
                ],
            },
            "IION": {
                "class_name": "IonogramBazis",
                "patterns": ["NF??????.??", "B1??????.??", "B2??????.??"],
                "sizes": [1200021, 1600021],
            },
            "Shigaraki": {
                "class_name": "IonogramShigaraki",
                "patterns": ["????????????_ionogram.txt"],
            },
            "VISRC2T": {
                "class_name": "IonogramVisrc2t",
                "patterns": [
                    "??????????????.rad",
                    "??????????????.rad.bz2",
                    "??????????????.ig",
                    "??????????????.ig.bz2",
                ],
            },
        }
        self.class_name = ""
        self.probability = 0
        self.points = {x: 0 for x in self.FILE_FORMATS.keys()}

    def examine(self, filename):
        self.__init__()

        file_size = path.getsize(filename)

        try:
            with open(filename) as f:
                first_line = f.readline()
        except UnicodeDecodeError:
            first_line = ""

        keys = self.FILE_FORMATS.keys()
        for key in keys:
            if "sizes" in self.FILE_FORMATS[key]:
                for size in self.FILE_FORMATS[key]["sizes"]:
                    if file_size == size:
                        self.points[key] += 1
            if "patterns" in self.FILE_FORMATS[key]:
                for pattern in self.FILE_FORMATS[key]["patterns"]:
                    (_, name) = path.split(filename)
                    if fnmatch.fnmatch(name, pattern):
                        self.points[key] += 1

        try:
            datetime.strptime(first_line[:-1], "%Y.%m.%d (%j) %H:%M:%S.%f")
        except ValueError:
            pass
        else:
            self.points["DPS_AMP"] += 1

        max_points = 0
        all_points = 0
        file_format = "Unknown"
        for key in keys:
            all_points += self.points[key]
            if self.points[key] > max_points:
                max_points = self.points[key]
                file_format = key

        self.class_name = self.FILE_FORMATS[file_format]["class_name"]
        if all_points != 0:
            self.probability = max_points / all_points

        return self.probability > 0

    def get_iono(self):
        class_ = globals()[self.class_name]
        iono = class_()
        return iono