spectre-core 0.0.12__py3-none-any.whl → 0.0.13__py3-none-any.whl

spectre_core/wgetting/_callisto.py

@@ -7,145 +7,200 @@ import subprocess
 import shutil
 import gzip
 from datetime import datetime
-from typing import Optional
-
-from spectre_core.config import get_spectre_data_dir_path, get_batches_dir_path, TimeFormats
-
-CALLISTO_INSTRUMENT_CODES = [
-    "ALASKA-ANCHORAGE",
-    "ALASKA-COHOE",
-    "ALASKA-HAARP",
-    "ALGERIA-CRAAG",
-    "ALMATY",
-    "AUSTRIA-Krumbach",
-    "AUSTRIA-MICHELBACH",
-    "AUSTRIA-OE3FLB",
-    "AUSTRIA-UNIGRAZ",
-    "Australia-ASSA",
-    "BIR",
-    "Croatia-Visnjan",
-    "DENMARK",
-    "EGYPT-Alexandria",
-    "EGYPT-SpaceAgency",
-    "FINLAND-Siuntio",
-    "Finland-Kempele",
-    "GERMANY-DLR",
-    "GLASGOW",
-    "GREENLAND",
-    "HUMAIN",
-    "HURBANOVO",
-    "INDIA-GAURI",
-    "INDIA-OOTY",
-    "INDIA-UDAIPUR",
-    "JAPAN-IBARAKI",
-    "KASI",
-    "MEXART",
-    "MEXICO-FCFM-UANL",
-    "MEXICO-LANCE-A",
-    "MEXICO-LANCE-B",
-    "MONGOLIA-UB",
-    "MRO",
-    "MRT3",
-    "Malaysia-Banting",
-    "NORWAY-EGERSUND",
-    "NORWAY-NY-AALESUND",
-    "NORWAY-RANDABERG",
-    "POLAND-Grotniki",
-    "ROMANIA",
-    "ROSWELL-NM",
-    "SPAIN-PERALEJOS",
-    "SSRT",
-    "SWISS-HB9SCT",
-    "SWISS-HEITERSWIL",
-    "SWISS-IRSOL",
-    "SWISS-Landschlacht",
-    "SWISS-MUHEN",
-    "TRIEST",
-    "TURKEY",
-    "UNAM",
-    "URUGUAY",
-    "USA-BOSTON",
-]
-
-_temp_dir = os.path.join(get_spectre_data_dir_path(), "temp")
-
-
-def _get_batch_name(station: str, date: str, time: str, instrument_code: str) -> str:
-    dt = datetime.strptime(f"{date}T{time}", '%Y%m%dT%H%M%S')
-    formatted_time = dt.strftime(TimeFormats.DATETIME)
-    return f"{formatted_time}_callisto-{station.lower()}-{instrument_code}.fits"
-
-
-def _get_batch_components(gz_path: str):
+
+from spectre_core.config import get_spectre_data_dir_path, get_batches_dir_path, TimeFormat
+
+from enum import Enum
+
+class CallistoInstrumentCode(Enum):
+    """e-Callisto network station codes."""
+    ALASKA_ANCHORAGE     = "ALASKA-ANCHORAGE"
+    ALASKA_COHOE         = "ALASKA-COHOE"
+    ALASKA_HAARP         = "ALASKA-HAARP"
+    ALGERIA_CRAAG        = "ALGERIA-CRAAG"
+    ALMATY               = "ALMATY"
+    AUSTRIA_KRUMBACH     = "AUSTRIA-Krumbach"
+    AUSTRIA_MICHELBACH   = "AUSTRIA-MICHELBACH"
+    AUSTRIA_OE3FLB       = "AUSTRIA-OE3FLB"
+    AUSTRIA_UNIGRAZ      = "AUSTRIA-UNIGRAZ"
+    AUSTRALIA_ASSA       = "Australia-ASSA"
+    BIR                  = "BIR"
+    CROATIA_VISNJAN      = "Croatia-Visnjan"
+    DENMARK              = "DENMARK"
+    EGYPT_ALEXANDRIA     = "EGYPT-Alexandria"
+    EGYPT_SPACEAGENCY    = "EGYPT-SpaceAgency"
+    FINLAND_SIUNTIO      = "FINLAND-Siuntio"
+    FINLAND_KEMPELE      = "Finland-Kempele"
+    GERMANY_DLR          = "GERMANY-DLR"
+    GLASGOW              = "GLASGOW"
+    GREENLAND            = "GREENLAND"
+    HUMAIN               = "HUMAIN"
+    HURBANOVO            = "HURBANOVO"
+    INDIA_GAURI          = "INDIA-GAURI"
+    INDIA_OOTY           = "INDIA-OOTY"
+    INDIA_UDAIPUR        = "INDIA-UDAIPUR"
+    JAPAN_IBARAKI        = "JAPAN-IBARAKI"
+    KASI                 = "KASI"
+    MEXART               = "MEXART"
+    MEXICO_FCFM_UANL     = "MEXICO-FCFM-UANL"
+    MEXICO_LANCE_A       = "MEXICO-LANCE-A"
+    MEXICO_LANCE_B       = "MEXICO-LANCE-B"
+    MONGOLIA_UB          = "MONGOLIA-UB"
+    MRO                  = "MRO"
+    MRT3                 = "MRT3"
+    MALAYSIA_BANTING     = "Malaysia-Banting"
+    NORWAY_EGERSUND      = "NORWAY-EGERSUND"
+    NORWAY_NY_AALESUND   = "NORWAY-NY-AALESUND"
+    NORWAY_RANDABERG     = "NORWAY-RANDABERG"
+    POLAND_GROTNIKI      = "POLAND-Grotniki"
+    ROMANIA              = "ROMANIA"
+    ROSWELL_NM           = "ROSWELL-NM"
+    SPAIN_PERALEJOS      = "SPAIN-PERALEJOS"
+    SSRT                 = "SSRT"
+    SWISS_HB9SCT         = "SWISS-HB9SCT"
+    SWISS_HEITERSWIL     = "SWISS-HEITERSWIL"
+    SWISS_IRSOL          = "SWISS-IRSOL"
+    SWISS_LANDSCHLACHT   = "SWISS-Landschlacht"
+    SWISS_MUHEN          = "SWISS-MUHEN"
+    TRIEST               = "TRIEST"
+    TURKEY               = "TURKEY"
+    UNAM                 = "UNAM"
+    URUGUAY              = "URUGUAY"
+    USA_BOSTON           = "USA-BOSTON"
+
+
+def _get_batch_name(
+    station: str, 
+    date: str, 
+    time: str, 
+    code: str
+) -> str:
+    """
+    Create a standardised batch file name for a `spectre` batch file.
+
+    :param station: Station name.
+    :param date: Observation date in 'YYYYMMDD' format.
+    :param time: Observation time in 'HHMMSS' format.
+    :param code: Numeric instrument code.
+    :return: Formatted batch file name.
+    """
+    dt = datetime.strptime(f"{date}T{time}", "%Y%m%dT%H%M%S")
+    formatted_time = dt.strftime(TimeFormat.DATETIME)
+    return f"{formatted_time}_callisto-{station.lower()}-{code}.fits"
+
+
+def _get_batch_components(
+    gz_path: str
+) -> list[str]:
+    """
+    Extract station, date, time, and instrument code from the `.fit.gz` file name.
+
+    :param gz_path: Path to the `.fit.gz` file.
+    :return: List of [station, date, time, instrument_code].
+    :raises ValueError: If file format is invalid.
+    """
     file_name = os.path.basename(gz_path)
     if not file_name.endswith(".fit.gz"):
-        raise ValueError(f"Unexpected file extension in {file_name}. Expected .fit.gz")
-    
+        raise ValueError(f"Invalid file extension: {file_name}. Expected .fit.gz")
     file_base_name = file_name.rstrip(".fit.gz")
     parts = file_base_name.split('_')
     if len(parts) != 4:
-        raise ValueError("Filename does not conform to the expected format of [station]_[date]_[time]_[instrument_code]")
-    
+        raise ValueError("Invalid file name format. Expected '[station]_[date]_[time]_[code].fit.gz'")
     return parts
 
 
-def _get_batch_path(gz_path: str) -> str:
-    station, date, time, instrument_code = _get_batch_components(gz_path)
-    fits_batch_name = _get_batch_name(station, date, time, instrument_code)
-    batch_start_time = fits_batch_name.split('_')[0]
-    batch_start_datetime = datetime.strptime(batch_start_time, TimeFormats.DATETIME)
-    batch_parent_path = get_batches_dir_path(year = batch_start_datetime.year,
-                                            month = batch_start_datetime.month,
-                                            day = batch_start_datetime.day)
-    if not os.path.exists(batch_parent_path):
-        os.makedirs(batch_parent_path)
-    return os.path.join(batch_parent_path, fits_batch_name)
-
-
-def _unzip_file_to_batches(gz_path: str):
+def _get_batch_path(
+    gz_path: str
+) -> str:
+    """
+    Generate the full path for the `spectre` batch file.
+
+    :param gz_path: Path to the raw `.fit.gz` file.
+    :return: Path to the corresponding batch file.
+    """
+    station, date, time, code = _get_batch_components(gz_path)
+    batch_name = _get_batch_name(station, date, time, code)
+    batch_start_time = batch_name.split("_")[0]
+    dt = datetime.strptime(batch_start_time, TimeFormat.DATETIME)
+    batch_dir = get_batches_dir_path(year=dt.year, month=dt.month, day=dt.day)
+    os.makedirs(batch_dir, exist_ok=True)
+    return os.path.join(batch_dir, batch_name)
+
+
+def _unzip_file_to_batches(
+    gz_path: str
+) -> None:
+    """
+    Decompress a `.fit.gz` file and save it as a `.fits` batch file.
+
+    :param gz_path: Path to the `.fit.gz` file.
+    """
     fits_path = _get_batch_path(gz_path)
-    with gzip.open(gz_path, 'rb') as f_in, open(fits_path, 'wb') as f_out:
+    with gzip.open(gz_path, "rb") as f_in, open(fits_path, "wb") as f_out:
         shutil.copyfileobj(f_in, f_out)
 
 
-def _unzip_to_batches():
-    for entry in os.scandir(_temp_dir):
-        if entry.is_file() and entry.name.endswith('.gz'):
+def _unzip_to_batches(
+    tmp_dir: str
+) -> None:
+    """
+    Decompress all `.gz` files in a temporary directory and save them as `spectre`
+    batch files.
+
+    :param tmp_dir: Path to the temporary directory containing `.gz` files.
+    """
+    for entry in os.scandir(tmp_dir):
+        if entry.is_file() and entry.name.endswith(".gz"):
             _unzip_file_to_batches(entry.path)
             os.remove(entry.path)
 
 
-def _wget_callisto_data(instrument_code: str, 
-                       year: int, 
-                       month: int, 
-                       day: int):
+def _wget_callisto_data(
+    instrument_code: str, 
+    year: int, 
+    month: int, 
+    day: int, 
+    tmp_dir: str
+) -> None:
+    """
+    Download raw `.fit.gz` files from the e-Callisto network using `wget`.
+
+    :param instrument_code: Instrument code for filtering files.
+    :param year: Observation year.
+    :param month: Observation month.
+    :param day: Observation day.
+    :param tmp_dir: Path to the temporary directory for downloads.
+    """
     date_str = f"{year:04d}/{month:02d}/{day:02d}"
     base_url = f"http://soleil.i4ds.ch/solarradio/data/2002-20yy_Callisto/{date_str}/"
     command = [
-        'wget', '-r', '-l1', '-nd', '-np', 
-        '-R', '.tmp',
-        '-A', f'{instrument_code}*.fit.gz',
-        '-P', _temp_dir,
-        base_url
+        "wget", "-r", "-l1", "-nd", "-np",
+        "-R", ".tmp", "-A", f"{instrument_code}*.fit.gz",
+        "-P", tmp_dir, base_url
     ]
     subprocess.run(command, check=True)
 
 
-def download_callisto_data(instrument_code: Optional[str], 
-                           year: Optional[int], 
-                           month: Optional[int], 
-                           day: Optional[int]):
-
-
-    if (year is None) or (month is None) or (day is None):
-        raise ValueError(f"All of year, month and day should be specified")
+def download_callisto_data(
+    instrument_code: CallistoInstrumentCode, 
+    year: int, 
+    month: int, 
+    day: int
+) -> None:
+    """
+    Download and decompress e-Callisto FITS files, saving them as `spectre` batch files.
+
+    :param instrument_code: e-Callisto station instrument code.
+    :param year: Year of the observation.
+    :param month: Month of the observation.
+    :param day: Day of the observation.
+    """
+    tmp_dir = os.path.join(get_spectre_data_dir_path(), "tmp")
+    # if there are any residual files in the temporary directory, remove them.
+    if os.path.exists(tmp_dir):
+        shutil.rmtree(tmp_dir)
+    os.makedirs(tmp_dir, exist_ok=True)
     
-    if not os.path.exists(_temp_dir):
-        os.mkdir(_temp_dir)
-
-    if instrument_code not in CALLISTO_INSTRUMENT_CODES:
-        raise ValueError(f"No match found for '{instrument_code}'. Expected one of {CALLISTO_INSTRUMENT_CODES}")
-
-    _wget_callisto_data(instrument_code, year, month, day)
-    _unzip_to_batches()
-    shutil.rmtree(_temp_dir)
+    _wget_callisto_data(instrument_code.value, year, month, day, tmp_dir)
+    _unzip_to_batches(tmp_dir)
+    shutil.rmtree(tmp_dir)

{spectre_core-0.0.12.dist-info → spectre_core-0.0.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: spectre-core
-Version: 0.0.12
+Version: 0.0.13
 Summary: The core Python package used by the spectre program. 
 Maintainer-email: Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
 License:                     GNU GENERAL PUBLIC LICENSE
@@ -691,6 +691,7 @@ Requires-Dist: scipy==1.12.0
 Requires-Dist: astropy==6.0.1
 Requires-Dist: matplotlib==3.5.0
 Requires-Dist: watchdog==4.0.0
+Requires-Dist: mypy==1.14.1
 
 # spectre-core
 
@@ -698,18 +699,24 @@ Requires-Dist: watchdog==4.0.0
   
 :loudspeaker: **This project is under active development. Contributors welcome.** :loudspeaker:  
 
-```spectre-core``` provides a SDR receiver-agnostic digital signal processing framework. It is the core Python package used by the [`spectre`](https://github.com/jcfitzpatrick12/spectre.git) program.  
+```spectre-core``` provides an extensible, receiver-agnostic toolkit for generating radio spectrograms in real-time. It is the core Python package used by the [`spectre`](https://github.com/jcfitzpatrick12/spectre.git) program.   
 
 
 ## Installation
-Typically, installation of ```spectre-core``` is managed in the ```spectre``` back-end. In any case, if you would like, simply call:  
-```pip install spectre-core```. 
+Simply call:  
+```bash
+pip install spectre-core
+```
 
 Alternatively, you can clone the repository in your preferred directory:  
-```git clone https://github.com/jcfitzpatrick12/spectre-core.git && cd spectre-core```  
+```bash
+git clone https://github.com/jcfitzpatrick12/spectre-core.git && cd spectre-core
+```  
 
 Then run:  
-```pip install -e .```
+```
+pip install -e .
+```
 
 
 ## Contributing

spectre_core-0.0.13.dist-info/RECORD

@@ -0,0 +1,75 @@
+spectre_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+spectre_core/exceptions.py,sha256=ccr-n88W0c_DKcsKMAe09QHX1iG2LUEqOMYHcGB-_do,451
+spectre_core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+spectre_core/_file_io/__init__.py,sha256=RZAtX8pFtdtTlxoxxttxJf1Wcg8lDFXgKMe1rcgaFP8,351
+spectre_core/_file_io/file_handlers.py,sha256=dJ3rMcTGbNnEXbYZgVoaos4p6qKGs-rWNeY8nmluQCQ,6402
+spectre_core/batches/__init__.py,sha256=nEEGLB9TnCKrP1VHQxYlU6U5MBt_4CHBcrRDnB-9GIU,685
+spectre_core/batches/_base.py,sha256=KI3f5ZDKC_HrLtMDqqXiVLOF5InAQnk5QkyUuCXyAGI,7237
+spectre_core/batches/_batches.py,sha256=CrfCjEhBO16SVfKbAvpxRsl0cjBxkiVOCSXegCuG0JM,7877
+spectre_core/batches/_factory.py,sha256=ApshWhbhZ-CSoQBrlcYaC5W5p_w_v4pxOdOdlrF7YIA,2277
+spectre_core/batches/_register.py,sha256=dSQC8KXj_jG8EiPwmKPdV0HSSalIZLaWt-8E29sczJM,1085
+spectre_core/batches/plugins/_batch_keys.py,sha256=8v0KE1n0NAQX0i9SwwB4Lgkct7Q_jna-H5S0Gs6p1qg,544
+spectre_core/batches/plugins/_callisto.py,sha256=ijm-VzGGlLQJjB2eQWw-04R6belCJ_NzSL9Jr7VTu2Q,6259
+spectre_core/batches/plugins/_iq_stream.py,sha256=DSto5ZzPk7A4VAe_HUAKNd2M9NuWqxcL--_Vneo08Bc,12583
+spectre_core/capture_configs/__init__.py,sha256=sUgwilZvQHAexAw_wao49qoBnu0CEBHIicI1kNfvCHg,1645
+spectre_core/capture_configs/_capture_config.py,sha256=ngbIzySgjsgBRJyrfgYZrJtr0wqcKDn8xBH0T1vi-yY,4546
+spectre_core/capture_configs/_capture_modes.py,sha256=uFBMwHYJCDqqQfYJvAUKzKXTmSBANRZMLlRSKV8WRb8,898
+spectre_core/capture_configs/_capture_templates.py,sha256=nSAxhOh1DeBzYY16evgOTkDuxRugGG5weJsICUouLPQ,10353
+spectre_core/capture_configs/_parameters.py,sha256=9KoNuwzDKtnyeju53fkImi1SeUjDn89cNwDL8zxX-YU,5563
+spectre_core/capture_configs/_pconstraints.py,sha256=bnl1m6M9iyo5KOsPKT_arwrrAZbxRKXVwTHQACzvs2g,5227
+spectre_core/capture_configs/_pnames.py,sha256=05mrrGfYHudaD2dYyukijbW-YtUnifZRn05xcBGv-fI,1852
+spectre_core/capture_configs/_ptemplates.py,sha256=p4lOi6W5B9gWv0d6ryNK9f_zxLHsj5PWW8Lmdco9ofg,24508
+spectre_core/capture_configs/_pvalidators.py,sha256=0W-OMDY1lifRPonwgEAwoMYkA0PWbqlK5nE0zsYVcTw,9596
+spectre_core/config/__init__.py,sha256=CVddibAIWU4cNerXCxfeUn_cvnWWDBm4pWeUifqM6Ko,502
+spectre_core/config/_paths.py,sha256=Qn0KVe27P6Bo65MfHe4LpH_ryPvLgbTxKjc4MTw1BB8,4090
+spectre_core/config/_time_formats.py,sha256=gS0j5zIvBhnV7KMYvTloloIbVwmCYn8MMKn3zNeQ4Xc,857
+spectre_core/jobs/__init__.py,sha256=WKTvxvpciedm6tsKjU02iXJhIdNsMDt-BnMVwVme2Bo,412
+spectre_core/jobs/_jobs.py,sha256=gGpxsLZZ7EdXBYGH-r_pKnRGWSapr78E5SK_VnulaGg,3844
+spectre_core/jobs/_workers.py,sha256=9GPEJVqIFuOAXpY9gfJxX-0_UPJ6RBtUe5qC5JLiYNw,5023
+spectre_core/logs/__init__.py,sha256=BHZTru9KyOUDPBnanM1EJK5E1BN6YAGrjhdKFZ1ysUM,484
+spectre_core/logs/_configure.py,sha256=NlR3BEwhj5k0qlTooOLb6OFRbYcDT05eWN-9Sl9Hkbk,2208
+spectre_core/logs/_decorators.py,sha256=u_Ixpr6aeKiSjM0mtqHuGXVGawNjgJPRvIpAj0zPSdQ,1169
+spectre_core/logs/_logs.py,sha256=M6sIsb2xxos9Ytj509qqp2jXZtvFU5uon9jIwEHLEI8,6578
+spectre_core/logs/_process_types.py,sha256=qnKfeXPk-6NOlNf4XtD8D3tYbA_ZVhHSZAKSd2MiVmY,488
+spectre_core/plotting/__init__.py,sha256=9zhmCphR4DSHlWfJhHeWPbvIrh8Kycov5Ta6woSrUZg,517
+spectre_core/plotting/_base.py,sha256=51125B7OwvVvDXU2utZ_z0xuUkaKMeJ55Jsp2I-SORg,8226
+spectre_core/plotting/_format.py,sha256=gWnj12PfgODQRTCCEyH2_mY7GY0brFbmolBEJfDtO6c,1393
+spectre_core/plotting/_panel_names.py,sha256=xeXUXAhzDoPYCgtjy_6zbwADHBtB0-jARnFOW-YTcAc,830
+spectre_core/plotting/_panel_stack.py,sha256=ZlVhAzb5Go4PdjR3-TeeaRA8T2-spqQHvFHLJVx1Weg,8732
+spectre_core/plotting/_panels.py,sha256=IaJ8EDLu5pvsN6kklNWc1ectCfgUsxwN0EKwT2wiFNo,16030
+spectre_core/post_processing/__init__.py,sha256=PVr7AhKK5at0o3BzNlJeC_Z_J8ZNATjGVY7PYg7fSew,624
+spectre_core/post_processing/_base.py,sha256=gtycVjtAStN5sW4KuzxVQngLzYCx3k-9B5IfmMZnu_g,5815
+spectre_core/post_processing/_factory.py,sha256=fnabYFy7Qe3EkJcR7baj1VyPdikw-xTSuNC01w7VupU,2193
+spectre_core/post_processing/_post_processor.py,sha256=0WLOwtx-FHK4t6dv-I6_1K9sRNn-625wPS11lqJ8Kho,1300
+spectre_core/post_processing/_register.py,sha256=chK5bCQ-0bpMhplbKT3cdS5anLhIA5Tt_ONzThJHrw0,1249
+spectre_core/post_processing/plugins/_event_handler_keys.py,sha256=LPA71kKsaLPyT2vpPJwm88-RWfX-AU0_T9H8-GdjINQ,614
+spectre_core/post_processing/plugins/_fixed_center_frequency.py,sha256=24LB3iDMa2KvgM7jyRnEr7Y45VK46bUT7Nj8bXGiGwc,5231
+spectre_core/post_processing/plugins/_swept_center_frequency.py,sha256=fzPAZ4zK1DRAHushUB-JS2IgKAuXyP4hdBLZJX6jBmo,20971
+spectre_core/receivers/__init__.py,sha256=A-BuCrnIRdRZe6HGXlLzKHGm8Zjfb8I9dzslBTeUHn4,660
+spectre_core/receivers/_base.py,sha256=E2aSYmveSMYxLrm5Uv235KlsehOyq82DvSolCIF6xug,10096
+spectre_core/receivers/_factory.py,sha256=5NdwZzBmUEHZQHltggDmuEoi-DRS6fHPSBzxvmq50zs,1843
+spectre_core/receivers/_register.py,sha256=jWS3Q_ZOZcqUFunxJwR5VRZN4_eS3LzcP-5edQdmj4c,1366
+spectre_core/receivers/_spec_names.py,sha256=pLYM9rMoZkT3JmRvSNfWKFiIJmj07vrphDbw-fNhpO8,1611
+spectre_core/receivers/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+spectre_core/receivers/plugins/_receiver_names.py,sha256=GZi4xznxscLBJXbfkZLtQ-m0U3GH362HiI5-y_FQO1k,426
+spectre_core/receivers/plugins/_rsp1a.py,sha256=LAvjxcxIkZRPpc96wqkco7rQ-H3AvUowsp0WG8vgAkw,2438
+spectre_core/receivers/plugins/_rspduo.py,sha256=ypr4xiEMOq4EJKFECz01PZjUJ2UcsWJVZRou8Ln5Ogw,3049
+spectre_core/receivers/plugins/_sdrplay_receiver.py,sha256=QpQ2sRc9Tz2IZRL_LyBX3UPCQ270RQMHH2mP6Y_n1-o,6899
+spectre_core/receivers/plugins/_test.py,sha256=t8oE4zQLYYWJV6SMIZAlNHs9mig263QJmP6XRLyF0RU,8277
+spectre_core/receivers/plugins/gr/__init__.py,sha256=oFSWmGoXQLK5X5xHvWzTdNr9amuaiiGjZirXZVogACU,154
+spectre_core/receivers/plugins/gr/_base.py,sha256=woFF6498aLIDf4EC7aD-TolY9LtZBqlLy-Vai_gfIvc,2571
+spectre_core/receivers/plugins/gr/_rsp1a.py,sha256=Nq0VNWjzfTd4Mruz4FSk6bfso6zh1cMw-C-qn5LpPv8,6614
+spectre_core/receivers/plugins/gr/_rspduo.py,sha256=s5g5nOQHiVHCESaHdj2VmgXl_LaiMaqXhTJfUbC7Iew,9811
+spectre_core/receivers/plugins/gr/_test.py,sha256=YA3JKdJ_CT3J93pqHUjltKrQUoaDXzUloEJEXVHH718,5780
+spectre_core/spectrograms/__init__.py,sha256=AsiOmn9XrAAHUvK-fdhRddAxX4M1Wd6TCtdmxGkl3FA,763
+spectre_core/spectrograms/_analytical.py,sha256=Axnt9JOJnWXRRuVU5nHPz5QU09KoWqNZkR5NnTX6kMY,11356
+spectre_core/spectrograms/_array_operations.py,sha256=rhiHjnqFGT1bffJygMbeYWb1H9VGBK6-uAgFh8I9owc,7547
+spectre_core/spectrograms/_spectrogram.py,sha256=WhHEt_QpmzspDqYlzdZcJ8CAXxRfs8-JfP0T3NHpjLQ,28205
+spectre_core/spectrograms/_transform.py,sha256=WZ5jAe3bOpNldxHDSHPf8Q_1ifBdWqXB_mlF6DL1VuE,11734
+spectre_core/wgetting/__init__.py,sha256=UkS0Z0wuuqpoZ1EL35wJcDpjBiAaZgdZ7064yGESxNE,341
+spectre_core/wgetting/_callisto.py,sha256=m9II6ayswrY4h_ST3TodioZ4C865lN_uOF-BozvjZAg,6952
+spectre_core-0.0.13.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+spectre_core-0.0.13.dist-info/METADATA,sha256=1PBd2ql6_9g9BGrwb1bWhnY4MSqRV1hdkXne6uE184o,42100
+spectre_core-0.0.13.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+spectre_core-0.0.13.dist-info/top_level.txt,sha256=-UsyjpFohXgZpgcZ9QbVeXhsIyF3Am8RxNFNDV_Ta2Y,13
+spectre_core-0.0.13.dist-info/RECORD,,

{spectre_core-0.0.12.dist-info → spectre_core-0.0.13.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.6.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

spectre_core/batches/library/_callisto.py

@@ -1,96 +0,0 @@
-# SPDX-FileCopyrightText: © 2024 Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
-# This file is part of SPECTRE
-# SPDX-License-Identifier: GPL-3.0-or-later
-
-from datetime import datetime, timedelta
-from typing import Tuple
-
-import numpy as np
-from astropy.io import fits
-from astropy.io.fits.hdu.image import PrimaryHDU
-from astropy.io.fits.hdu.table import BinTableHDU
-from astropy.io.fits.hdu.hdulist import HDUList
-
-from spectre_core.spectrograms import Spectrogram, SpectrumTypes
-from .._register import register_batch
-from .._base import BaseBatch, BatchFile
-
-
-@register_batch('callisto')
-class _Batch(BaseBatch):
-    def __init__(self,
-                 start_time: str,
-                 tag: str):
-        super().__init__(start_time, tag) 
-        self.add_file( FitsFile(self.parent_dir_path, self.name) )
-
-
-class FitsFile(BatchFile):
-    def __init__(self, 
-                 parent_dir_path: str, 
-                 base_file_name: str):
-        super().__init__(parent_dir_path, base_file_name, "fits")
-
-
-    def read(self) -> Spectrogram:
-        with fits.open(self.file_path, mode='readonly') as hdulist:
-            primary_hdu                = self._get_primary_hdu(hdulist)
-            dynamic_spectra            = self._get_dynamic_spectra(primary_hdu)
-            spectrogram_start_datetime = self._get_spectrogram_start_datetime(primary_hdu)
-            bintable_hdu               = self._get_bintable_hdu(hdulist)
-            times, frequencies         = self._get_time_and_frequency(bintable_hdu)
-            spectrum_type              = self._get_spectrum_type(primary_hdu)
-
-            if spectrum_type == SpectrumTypes.DIGITS:
-                dynamic_spectra_linearised = self._convert_units_to_linearised(dynamic_spectra)
-                return Spectrogram(dynamic_spectra_linearised[::-1, :], # reverse the spectra along the frequency axis
-                        times, 
-                        frequencies[::-1], # sort the frequencies in ascending order
-                        self.tag, 
-                        spectrogram_start_datetime,
-                        spectrum_type)
-            else:
-                raise NotImplementedError(f"SPECTRE does not currently support spectrum type with BUNITS '{spectrum_type}'")
-
-
-    @property
-    def datetimes(self) -> np.ndarray:
-        with fits.open(self.file_path, mode='readonly') as hdulist:
-            bintable_data = hdulist[1].data
-            times = bintable_data['TIME'][0]
-            return [self.start_datetime + timedelta(seconds=t) for t in times]        
-
-
-    def _get_primary_hdu(self, hdulist: HDUList) -> PrimaryHDU:
-        return hdulist['PRIMARY']
-
-
-    def _get_dynamic_spectra(self, primary_hdu: PrimaryHDU) -> np.ndarray:
-        return primary_hdu.data
-
-
-    def _get_spectrogram_start_datetime(self, primary_hdu: PrimaryHDU) -> datetime:
-        date_obs = primary_hdu.header['DATE-OBS']
-        time_obs = primary_hdu.header['TIME-OBS']
-        return datetime.strptime(f"{date_obs}T{time_obs}", "%Y-%m-%dT%H:%M:%S.%f")
-
-
-    def _get_bintable_hdu(self, hdulist: HDUList) -> BinTableHDU:
-        return hdulist[1]
-
-
-    def _get_time_and_frequency(self, bintable_hdu: BinTableHDU) -> Tuple[np.ndarray, np.ndarray]:
-        data = bintable_hdu.data
-        times = data['TIME'][0]
-        frequencies_MHz = data['FREQUENCY'][0]
-        frequencies = frequencies_MHz * 1e6 # convert to Hz
-        return times, frequencies
-
-
-    def _convert_units_to_linearised(self, dynamic_spectra: np.ndarray) -> np.ndarray:
-        digits_floats = np.array(dynamic_spectra, dtype='float')
-        # conversion as per ADC specs [see email from C. Monstein]
-        dB = (digits_floats / 255) * (2500 / 25)
-        return 10 ** (dB / 10)
-
-