Source code for exotools.db.tic_db

import numpy as np
from astropy.table import QTable
from typing_extensions import Self

from .base_db import BaseDB

_ID_FIELD = "tic_id"




class TicDB(BaseDB):
    """
    Dtypes:
    -------------------
    tic_id        int64
    gaia_id       int64
    priority    float64
    ra          float64
    dec         float64
    -------------------
    """

    def __init__(self, dataset: QTable):
        super().__init__(dataset, id_field="tic_id")

    def _factory(self, dataset: QTable) -> Self:
        return TicDB(dataset)

    @property
    def tic_ids(self) -> np.ndarray:
        return self.view["tic_id"].value

    @property
    def gaia_ids(self) -> np.ndarray:
        return self.view["gaia_id"].value

    @property
    def unique_tic_ids(self) -> np.ndarray:
        return np.unique(self.tic_ids)

    @property
    def unique_gaia_ids(self) -> np.ndarray:
        return np.unique(self.gaia_ids)