# -*- coding: utf-8 -*-
##########################################################################
# NSAp - Copyright (C) CEA, 2021
# Distributed under the terms of the CeCILL-B license, as published by
# the CEA-CNRS-INRIA. Refer to the LICENSE file or to
# http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
# for details.
##########################################################################

"""
Suface utilities.
"""

# Imports
import logging
import warnings
from .sampling import (
    interpolate, interpolate_data, downsample, downsample_data, downsample_ico,
    neighbors, neighbors_rec, get_rectangular_projection, find_neighbors,
    icosahedron, number_of_ico_vertices, order_of_ico_from_vertices,
    rotate_data, order_triangles, find_rotation_interpol_coefs,
    number_of_neighbors, min_depth_to_get_n_neighbors, patch_tri)
from .coord import (
    cart2sph, sph2cart, text2grid, grid2text, ico2ico, text2ico)

# Global parameters
LEVELS = {
    "debug": logging.DEBUG,
    "info": logging.INFO,
    "warning": logging.WARNING,
    "error": logging.ERROR,
    "critical": logging.CRITICAL
}



[docs]
def get_logger():
    """ Setup the logger.

    Returns
    -------
    logger: logging.Logger
        return a logger.
    """
    return logging.getLogger("surfify")




[docs]
def setup_logging(level="info", logfile=None):
    """ Setup the logging.

    Parameters
    ----------
    logfile: str, default None
        the log file.
    """
    logger = get_logger()
    logging_format = logging.Formatter(
        "[%(asctime)s] {%(pathname)s:%(lineno)d} %(levelname)s - "
        "%(message)s", "%Y-%m-%d %H:%M:%S")
    while len(logging.root.handlers) > 0:
        logging.root.removeHandler(logging.root.handlers[-1])
    while len(logger.handlers) > 0:
        logger.removeHandler(logger.handlers[-1])
    level = LEVELS.get(level)
    if level is None:
        raise ValueError("Unknown logging level.")
    logger.setLevel(level)
    stream_handler = logging.StreamHandler()
    stream_handler.setLevel(level)
    stream_handler.setFormatter(logging_format)
    logger.addHandler(stream_handler)
    if logfile is not None:
        file_handler = logging.FileHandler(logfile, mode="a")
        file_handler.setLevel(level)
        file_handler.setFormatter(logging_format)
        logger.addHandler(file_handler)
    if level != logging.DEBUG:
        warnings.simplefilter("ignore", DeprecationWarning)




[docs]
def debug_msg(name, tensor):
    """ Format a debug message.

    Parameters
    ----------
    name: str
        the tensor name in the displayed message.
    tensor: Tensor
        a pytorch tensor.

    Returns
    -------
    msg: str
        the formated debug message.
    """
    return "  {3}: {0} - {1} - {2}".format(
        tensor.shape, tensor.get_device(), tensor.dtype, name)