Source code for galaxia_ananke.photometry.Photometry

#!/usr/bin/env python
#
# Author: Adrien CR Thob
# Copyright (C) 2022  Adrien CR Thob
#
# This file is part of the py-Galaxia-ananke project,
# <https://github.com/athob/py-Galaxia-ananke>, which is licensed
# under the GNU Affero General Public License v3.0 (AGPL-3.0).
# 
# The full copyright notice, including terms governing use, modification,
# and redistribution, is contained in the files LICENSE and COPYRIGHT,
# which can be found at the root of the source code distribution tree:
# - LICENSE <https://github.com/athob/py-Galaxia-ananke/blob/main/LICENSE>
# - COPYRIGHT <https://github.com/athob/py-Galaxia-ananke/blob/main/COPYRIGHT>
#
"""
Docstring
"""
import shutil
from glob import glob

import pandas as pd

from .._constants import *
from ..utils import Singleton
from .Isochrone import Isochrone
from .PhotoSystem import PhotoSystem

__all__ = ['Photometry']




class nested_dict(dict):
    def __getitem__(self, __key):
        _key = __key.split('/')
        _temp = super().__getitem__(_key[0])
        return _temp if len(_key) == 1 else nested_dict(_temp)['/'.join(_key[1:])]
    
    def __setitem__(self, __key, __value):
        _key = __key.split('/')
        if len(_key) == 1:
            return super().__setitem__(_key[0], __value)
        else:
            if _key[0] not in self.keys():
                super().__setitem__(_key[0], {})
            return nested_dict(super().__getitem__(_key[0])).__setitem__('/'.join(_key[1:]), __value)

    
    # def __delitem__(self, __key):
    #     return super().__delitem__(__key)




class Photometry(nested_dict, metaclass=Singleton):
    _category = 'category'
    _isochrone = 'isochrone'
    def __new__(cls):
        cls.instance = super(Photometry, cls).__new__(cls)
        _temp = pd.DataFrame([[iso_path.parent.name, Isochrone(iso_path)]
                               for iso_path in ISOCHRONES_PATH.glob('*/*')
                               if iso_path.is_dir() and iso_path.parent.name != 'BolCorr'],
                             columns=[cls._category, cls._isochrone])
        _temp = {key: _temp[cls._isochrone].loc[item].to_list()
                 for key, item in _temp.groupby(cls._category).groups.items()}
        for key, item in _temp.items():
            cls.instance[key] = {iso.name: PhotoSystem(iso)
                                 for iso in item if iso.has_file_descriptor}
        if not CUSTOM_PHOTOCAT in cls.instance.keys(): cls.instance[CUSTOM_PHOTOCAT] = {}
        return cls.instance
    
    def __setitem__(self, __key, __value) -> None:
        __key_split = __key.split('/')
        if __key_split[0] is LEGACY_PHOTOCAT and LEGACY_PHOTOCAT in self.keys():
            raise ValueError(f"The '{LEGACY_PHOTOCAT}' entry is protected")
        # elif len(__key_split) == 1:
        #         __key = f"{CUSTOM_CATEGORY}/{__key}"
        return super().__setitem__(__key, __value)

    def __delitem__(self, __key):
        raise NotImplementedError()
        # return super().__delitem__(__key)
        


    def add_isochrone(self, name, isochrone_data, **kwargs):
        if '/' in name:
            raise ValueError()
        self[CUSTOM_PHOTOCAT][name] = Isochrone(name, isochrone_data, **kwargs)


    def _purge(self, hard=False, yes=False):
        if hard and (True if yes else (input("WARNING!! This will remove all the isochrones cached in your installation. Are you certain (y/[n])? ").lower() in ['y', 'yes'])):
            for dir in ISOCHRONES_PATH.glob('*'):
                shutil.rmtree(dir)



if __name__ == '__main__':
    raise NotImplementedError()