#!/usr/bin/env python3

from gettext import install
from logging import basicConfig, getLogger
from math import exp
from warnings import filterwarnings
from backscattering_analyzer.acquisition import AcquisitionType
from rich.console import Console
from rich.logging import RichHandler

install(__name__)
filterwarnings("error")
basicConfig(
    level="INFO",
    format="%(message)s",
    datefmt="[%X]",
    handlers=[RichHandler()],
)

logger = getLogger(__name__)
console = Console()

from backscattering_analyzer.display import show_projection
from backscattering_analyzer.experiment import Experiment
from matplotlib.pyplot import figure, show

date = "2024_10_31"
bench = "SDB1"
file = "values-coupling.toml"

experiment = Experiment(bench, date, file, 0.1)

"""
Figure = figure()
Figure.gca().plot(
    experiment.data.excited.movements.bench.x,
    experiment.data.excited.movements.bench.y,
    label=_("excited"),
)
Figure.gca().plot(
    experiment.data.reference.movements.bench.x,
    experiment.data.reference.movements.bench.y,
    label=_("reference"),
)
Figure.gca().legend()
Figure.gca().set_xlabel(_("time (s)"))
Figure.gca().set_ylabel(_("movement"))
Figure.gca().set_title(_("DEBUG"))
show()
"""

experiment.factors["pre"] = 1000

projection = experiment.compute_projection(AcquisitionType.EXCITED)

Figure = figure()
Figure.gca().loglog(
    experiment.data.excited.sensibility.psd().sqrt().x,
    experiment.data.excited.sensibility.psd().sqrt().y,
    label=_("measured"),
)
Figure.gca().loglog(
    projection.x,
    projection.y,
    label=_("projection"),
)
Figure.gca().legend()
show()