starfield 0.13.0

Astronomical data reduction toolkit with star catalogs, coordinate systems, and star finding algorithms (inspired by skyfield)
Documentation 
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

import numpy as np
import json, base64

def _bytes_to_str(data: bytes) -> str:
    return base64.b64encode(data).decode('utf-8')

class ResultCollector:
    def __init__(self):

        self.result = None

    def __call__(self, data):
        if isinstance(data, bytes):
            self.collect_bytes(data)
        elif isinstance(data, str):
            self.collect_string(data)
        elif isinstance(data, np.ndarray):
            self.collect_array(data)
        else:
            raise TypeError(f"Unsupported type: {type(data)}")

    def collect_bytes(self, data: bytes):
        self._set_result({
            "type": 'bytes',
            "data": _bytes_to_str(data)
        })

    def collect_string(self, data: str):
        self._set_result({
            "type": 'string',
            "data": data
        })

    def collect_array(self, array: np.ndarray):
        self._set_result({
            "type": 'array',
            "dtype": str(array.dtype),
            "shape": array.shape,
            "data": _bytes_to_str(array.tobytes())
        })

    def _set_result(self, result: dict):
        if self.result is not None:
            raise Exception("Result already collected")
        self.result = json.dumps(result)
    
    def get_result(self) -> str:
        if self.result is None:
            raise Exception("No result collected")
        return self.result
    
rust = ResultCollector()