Add script to compute and display projection
This commit is contained in:
commit
22ae5f192b
3 changed files with 287 additions and 0 deletions
179
display.py
Executable file
179
display.py
Executable file
|
|
@ -0,0 +1,179 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
from gettext import install
|
||||||
|
from backscattering_analyzer.display import show_projection # type: ignore[reportMissingTypeStubs]
|
||||||
|
from backscattering_analyzer.experiment import Experiment # type: ignore[reportMissingTypeStubs]
|
||||||
|
from logging import basicConfig, getLogger
|
||||||
|
from rich.console import Console
|
||||||
|
from rich.theme import Theme
|
||||||
|
from rich.logging import RichHandler
|
||||||
|
from warnings import filterwarnings
|
||||||
|
from command_parser.flag import Flag # type: ignore[reportMissingTypeStubs]
|
||||||
|
from command_parser.value import Value # type: ignore[reportMissingTypeStubs]
|
||||||
|
from command_parser.parser import parse # type: ignore[reportMissingTypeStubs]
|
||||||
|
from sys import argv
|
||||||
|
from importlib.metadata import version
|
||||||
|
|
||||||
|
install(__name__)
|
||||||
|
|
||||||
|
filterwarnings("error")
|
||||||
|
|
||||||
|
flags = [
|
||||||
|
Flag("help"),
|
||||||
|
Flag("version", single_char="V"),
|
||||||
|
Flag("verbose"),
|
||||||
|
Flag("full"),
|
||||||
|
]
|
||||||
|
values = [
|
||||||
|
Value("date", "2023_03_24"),
|
||||||
|
Value("bench", "SDB1"),
|
||||||
|
Value("file", "values.toml", single_char="F"),
|
||||||
|
Value("vibration-frequency", "0.2"),
|
||||||
|
]
|
||||||
|
parse(" ".join(argv[1:]), flags, values)
|
||||||
|
|
||||||
|
if flags[0].value:
|
||||||
|
theme = Theme(
|
||||||
|
{
|
||||||
|
"main": "white bold",
|
||||||
|
"option": "grey50 italic",
|
||||||
|
"argument": "red",
|
||||||
|
"description": "green italic",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
console = Console(theme=theme)
|
||||||
|
help = (
|
||||||
|
"[main]display[/main] [option]\\[options][/option]"
|
||||||
|
"\n [argument]-b --bench[/argument] [description]bench of the experiment[/description]"
|
||||||
|
"\n [argument]-d --date[/argument] [description]date of the experiment[/description]"
|
||||||
|
"\n [argument]-h --help[/argument] [description]print this help and exit[/description]"
|
||||||
|
"\n [argument]-f --full[/argument] [description]run experiment with all benches with all dates[/description]"
|
||||||
|
"\n [argument]-F --file[/argument] [description]file containing all the parameters values (toml file)[/decription]"
|
||||||
|
"\n [argument]-v --verbose[/argument] [description]be verbose[/description]"
|
||||||
|
"\n [argument]-V --version[/argument] [description]print version number and exit[/description]"
|
||||||
|
)
|
||||||
|
console.print(help)
|
||||||
|
exit(0)
|
||||||
|
|
||||||
|
if flags[1].value:
|
||||||
|
version = (
|
||||||
|
"Versions\n"
|
||||||
|
"numpy {numpy}\n"
|
||||||
|
"scipy {scipy}\n"
|
||||||
|
"matplotlib {matplotlib}\n"
|
||||||
|
"science_signal {science_signal}\n"
|
||||||
|
"backscattering_analyzer {backscattering_analyzer}\n"
|
||||||
|
"command_parser {command_parser}\n"
|
||||||
|
"script 0.1.0\n"
|
||||||
|
).format(
|
||||||
|
numpy=version("numpy"),
|
||||||
|
scipy=version("scipy"),
|
||||||
|
matplotlib=version("matplotlib"),
|
||||||
|
science_signal=version("science_signal"),
|
||||||
|
backscattering_analyzer=version("backscattering_analyzer"),
|
||||||
|
command_parser=version("command_parser"),
|
||||||
|
)
|
||||||
|
console = Console()
|
||||||
|
console.print(version)
|
||||||
|
exit(0)
|
||||||
|
|
||||||
|
if flags[2].value:
|
||||||
|
level = "DEBUG"
|
||||||
|
else:
|
||||||
|
level = "INFO"
|
||||||
|
|
||||||
|
|
||||||
|
basicConfig(
|
||||||
|
level=level,
|
||||||
|
format="%(message)s",
|
||||||
|
datefmt="[%X]",
|
||||||
|
handlers=[RichHandler()],
|
||||||
|
)
|
||||||
|
|
||||||
|
logger = getLogger(__name__)
|
||||||
|
|
||||||
|
if flags[3].value:
|
||||||
|
logger.debug(_("loading every experiments"))
|
||||||
|
experiments: list[Experiment] = []
|
||||||
|
results: dict[str, float | None] = dict({})
|
||||||
|
for bench in ["SDB1", "SDB2", "SNEB", "SWEB"]:
|
||||||
|
if bench in ["SDB1", "SDB2"]:
|
||||||
|
experiment = Experiment(
|
||||||
|
bench,
|
||||||
|
"2023_03_24",
|
||||||
|
values[2].value,
|
||||||
|
float(values[3].value),
|
||||||
|
)
|
||||||
|
if bench == "SDB1":
|
||||||
|
start_frequency, end_frequency = 12.5, 30
|
||||||
|
else:
|
||||||
|
start_frequency, end_frequency = 40, 50
|
||||||
|
experiment.factors = experiment.fit_factors(
|
||||||
|
start_frequency=start_frequency,
|
||||||
|
end_frequency=end_frequency,
|
||||||
|
)
|
||||||
|
results[
|
||||||
|
"{bench}-{date}".format(
|
||||||
|
bench=bench,
|
||||||
|
date="2023_03_24",
|
||||||
|
)
|
||||||
|
] = experiment.factors["true"]
|
||||||
|
elif bench == "SNEB":
|
||||||
|
for date in ["2023_03_24", "2024_01_21"]:
|
||||||
|
experiment = Experiment(
|
||||||
|
bench,
|
||||||
|
date,
|
||||||
|
values[2].value,
|
||||||
|
float(values[3].value),
|
||||||
|
)
|
||||||
|
experiment.factors = experiment.fit_factors(
|
||||||
|
start_frequency = 15,
|
||||||
|
end_frequency = 100,
|
||||||
|
)
|
||||||
|
results[
|
||||||
|
"{bench}-{date}".format(
|
||||||
|
bench=bench,
|
||||||
|
date=date
|
||||||
|
)
|
||||||
|
] = experiment.factors["true"]
|
||||||
|
else: # bench == SWEB
|
||||||
|
for date in ["2023_03_24", "2023_12_20"]:
|
||||||
|
experiment = Experiment(
|
||||||
|
bench,
|
||||||
|
date,
|
||||||
|
values[2].value,
|
||||||
|
float(values[3].value),
|
||||||
|
)
|
||||||
|
experiment.factors = experiment.fit_factors(
|
||||||
|
start_frequency = 15,
|
||||||
|
end_frequency = 100,
|
||||||
|
)
|
||||||
|
results[
|
||||||
|
"{bench}-{date}".format(
|
||||||
|
bench=bench,
|
||||||
|
date=date
|
||||||
|
)
|
||||||
|
] = experiment.factors["true"]
|
||||||
|
console = Console()
|
||||||
|
console.print(results)
|
||||||
|
|
||||||
|
else:
|
||||||
|
experiment = Experiment(
|
||||||
|
values[1].value, values[0].value, values[2].value, float(values[3].value)
|
||||||
|
)
|
||||||
|
|
||||||
|
if experiment.bench == "SDB1":
|
||||||
|
start_frequency, end_frequency = 12.5, 30
|
||||||
|
elif experiment.bench == "SDB2":
|
||||||
|
start_frequency, end_frequency = 40, 50
|
||||||
|
else:
|
||||||
|
start_frequency, end_frequency = 15, 100
|
||||||
|
|
||||||
|
experiment.factors = experiment.fit_factors(
|
||||||
|
start_frequency=start_frequency,
|
||||||
|
end_frequency=end_frequency,
|
||||||
|
)
|
||||||
|
|
||||||
|
show_projection(experiment, start_frequency, end_frequency)
|
||||||
|
|
||||||
|
exit(0)
|
||||||
52
values-coupling.toml
Normal file
52
values-coupling.toml
Normal file
|
|
@ -0,0 +1,52 @@
|
||||||
|
[general]
|
||||||
|
data_folder = "/home/demagny/data"
|
||||||
|
[interferometer]
|
||||||
|
wavelength = 1.064e-6
|
||||||
|
[benches]
|
||||||
|
[benches.SDB1]
|
||||||
|
scatter_factor = [3.1e-10, 0]
|
||||||
|
[benches.SDB1.calib]
|
||||||
|
bench = 1.15
|
||||||
|
mirror = 1.15
|
||||||
|
|
||||||
|
[benches.SDB2]
|
||||||
|
scatter_factor = [4.1e-13, 0]
|
||||||
|
[benches.SDB2.calib]
|
||||||
|
bench = 1.3
|
||||||
|
mirror = 1.3
|
||||||
|
[benches.SIB2]
|
||||||
|
scatter_factor = [0, 0]
|
||||||
|
[benches.SIB2.calib]
|
||||||
|
bench = 1.0
|
||||||
|
mirror = 1.0
|
||||||
|
[benches.SNEB]
|
||||||
|
scatter_factor = [4.7e-9, 0]
|
||||||
|
[benches.SNEB.calib]
|
||||||
|
bench = 1.01
|
||||||
|
mirror = 1.01
|
||||||
|
[benches.SPRB]
|
||||||
|
scatter_factor = [0, 0]
|
||||||
|
[benches.SPRB.calib]
|
||||||
|
bench = 1.0
|
||||||
|
mirror = 1.0
|
||||||
|
[benches.SWEB]
|
||||||
|
scatter_factor = [1.3e-9, 0]
|
||||||
|
[benches.SWEB.calib]
|
||||||
|
bench = 0.98
|
||||||
|
mirror = 0.98
|
||||||
|
[dates]
|
||||||
|
[dates.2023_03_24]
|
||||||
|
laser = 23
|
||||||
|
dark_fringe = 8e-3
|
||||||
|
modelisation = "scatterCoupling_march.mat"
|
||||||
|
correct-power = false
|
||||||
|
[dates.2023_12_20]
|
||||||
|
laser = 12.1
|
||||||
|
dark_fringe = 4.445e-3
|
||||||
|
modelisation = "scatterCoupling_dec_jan.mat"
|
||||||
|
correct-power = false
|
||||||
|
[dates.2024_01_21]
|
||||||
|
laser = 12.1
|
||||||
|
dark_fringe = 4.445e-3
|
||||||
|
modelisation = "scatterCoupling_dec_jan.mat"
|
||||||
|
correct-power = false
|
||||||
56
values.toml
Normal file
56
values.toml
Normal file
|
|
@ -0,0 +1,56 @@
|
||||||
|
[general]
|
||||||
|
data_folder = "/home/demagny/data"
|
||||||
|
[interferometer]
|
||||||
|
wavelength = 1.064e-6
|
||||||
|
[benches]
|
||||||
|
[benches.SDB1]
|
||||||
|
scatter_factor = [1.5e-9, 0]
|
||||||
|
[benches.SDB1.calib]
|
||||||
|
bench = 1.15
|
||||||
|
mirror = 1.15
|
||||||
|
|
||||||
|
[benches.SDB2]
|
||||||
|
scatter_factor = [2.4e-11, 0]
|
||||||
|
[benches.SDB2.calib]
|
||||||
|
bench = 1.3
|
||||||
|
mirror = 1.3
|
||||||
|
[benches.SIB2]
|
||||||
|
scatter_factor = [0, 0]
|
||||||
|
[benches.SIB2.calib]
|
||||||
|
bench = 1.0
|
||||||
|
mirror = 1.0
|
||||||
|
[benches.SNEB]
|
||||||
|
scatter_factor = [9.5e-9, 0]
|
||||||
|
[benches.SNEB.calib]
|
||||||
|
bench = 1.01
|
||||||
|
mirror = 1.01
|
||||||
|
[benches.SPRB]
|
||||||
|
scatter_factor = [0, 0]
|
||||||
|
[benches.SPRB.calib]
|
||||||
|
bench = 1.0
|
||||||
|
mirror = 1.0
|
||||||
|
[benches.SWEB]
|
||||||
|
scatter_factor = [4.7e-9, 0]
|
||||||
|
[benches.SWEB.calib]
|
||||||
|
bench = -0.98
|
||||||
|
mirror = 0.98
|
||||||
|
[dates]
|
||||||
|
[dates.2023_03_24]
|
||||||
|
laser = 23
|
||||||
|
dark_fringe = 8e-3
|
||||||
|
#modelisation = "scatterCouplingO4.mat"
|
||||||
|
modelisation = "scatterCoupling_40.mat"
|
||||||
|
correct-power = true
|
||||||
|
[dates.2023_03_24.coupling]
|
||||||
|
laser = 40
|
||||||
|
dark_fringe = 5e-3
|
||||||
|
[dates.2023_12_20]
|
||||||
|
laser = 12.1
|
||||||
|
dark_fringe = 4.445e-3
|
||||||
|
modelisation = "scatterCouplingMisalignSR.mat"
|
||||||
|
correct-power = false
|
||||||
|
[dates.2024_01_21]
|
||||||
|
laser = 12.1
|
||||||
|
dark_fringe = 4.445e-3
|
||||||
|
modelisation = "scatterCouplingMisalignSR.mat"
|
||||||
|
correct-power = false
|
||||||
Loading…
Reference in a new issue