#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Dataset
====================

This script is an example to process a DataSet class and save the output

"""

import pathlib

import pyhydrophone as pyhy
import pypam

# %%
# First it is necessary to create a csv with all the metadata. You can have a look at an example at
# examples/data_summary_example.csv
# Acoustic Data
summary_path = pathlib.Path("data_summary_example.csv")
include_dirs = False

# Output folder
output_folder = summary_path.parent.joinpath("data_exploration")

# Hydrophone Setup
# If Vpp is 2.0 then it means the wav is -1 to 1 directly related to V
model = "ST300HF"
name = "SoundTrap"
serial_number = 67416073
soundtrap = pyhy.soundtrap.SoundTrap(
    name=name, model=model, serial_number=serial_number
)

bk_model = "Nexus"
bk_name = "B&K"
preamp_gain = -170
bk_Vpp = 2.0
bk = pyhy.BruelKjaer(
    name=bk_name,
    model=bk_model,
    preamp_gain=preamp_gain,
    Vpp=bk_Vpp,
    serial_number=1,
    type_signal="ref",
)

upam_model = "uPam"
upam_name = "Seiche"
upam_serial_number = "SM7213"
upam_sensitivity = -196.0
upam_preamp_gain = 0.0
upam_Vpp = 20.0
upam = pyhy.uPam(
    name=upam_name,
    model=upam_name,
    serial_number=upam_serial_number,
    sensitivity=upam_sensitivity,
    preamp_gain=upam_preamp_gain,
    Vpp=upam_Vpp,
)


instruments = {"SoundTrap": soundtrap, "uPam": upam, "B&K": bk}

# Acoustic params. Reference pressure 1 uPa
REF_PRESSURE = 1e-6

# SURVEY PARAMETERS
nfft = 4096
binsize = 60.0
fft_overlap = 0.5
bin_overlap = 0.0
dc_subtract = False
band_lf = [50, 500]
band_mf = [500, 2000]
band_hf = [2000, 20000]
band_list = [band_lf]
temporal_features = ["rms", "sel", "aci"]
frequency_features = ["third_octaves_levels"]


# %%
# Create the dataset object
ds = pypam.dataset.DataSet(
    summary_path,
    output_folder,
    instruments,
    temporal_features=temporal_features,
    frequency_features=frequency_features,
    bands_list=band_list,
    binsize=binsize,
    bin_overlap=bin_overlap,
    nfft=nfft,
    fft_overlap=fft_overlap,
    dc_subtract=dc_subtract,
)
# Call the dataset creation. Will create the files in the corresponding folder
ds()