import datetime

from mongoengine import *

from .taskmanager import TaskManagerBase


class DynamicShearTest(Document):

    date = DateTimeField(default=datetime.datetime.now,
                         wtf_options={"render_kw": {
                             "step": "60"
                         }})
    
    task_id = LazyReferenceField(TaskManagerBase, required=True)

    gap_width = FloatField(default=1.0)

    tags = ListField(StringField())

    standard = StringField(default='TP Asphalt Teil 48 C')

    filehash = StringField(required=True)
    speciment_name = StringField()

    meta = {
        'allow_inheritance':
        True,
        'index_opts': {},
        'index_background':
        True,
        'index_cls':
        False,
        'auto_create_index':
        True,
        "db_alias":
        'dblabtests',
        'collection':
        'sheartest',
        'indexes': [
            [("lab", 1)],
            [("speciment_name", 1)],
            [("project", 1)],
            [("bruch", 1)],
            [("lab", 1), ("project", 1)],
            [("lab", 1), ("project", 1), ("workpackage", 1)],
            [("lab", 1), ("project", 1), ("bounding", 1)],
        ]
    }


class DynamicShearTestExtension(DynamicShearTest):

    #metadata
    f_set = FloatField(required=True)
    sigma_normal = FloatField(required=True)
    T_set = FloatField(required=True)
    extension = FloatField(required=True)

    N_from = IntField()
    N_to = IntField()
    N_tot = IntField()
    n_samples_per_cycle = IntField()

    G = FloatField(required=True)
    broken = BooleanField(required=True)
    phase = FloatField()
    
    #fit parameter
    ## required parameters

    ## F
    F_amp = FloatField(required=True)
    F_freq = FloatField(required=True)
    F_phase = FloatField(required=True)
    F_offset = FloatField(required=True)
    F_slope = FloatField(required=True)
    F_r2 = FloatField(required=True)
    F_cycle_min = ListField(FloatField())
    F_min = FloatField()
    F_min_std = FloatField()
    F_min_diff_rel = FloatField()
    F_cycle_max = ListField(FloatField())
    F_max = FloatField()
    F_max_std = FloatField()
    F_max_diff_rel = FloatField()
    F_cycle_mean = ListField(FloatField())
    F_mean = FloatField()
    F_mean_std = FloatField()
    F_mean_diff_rel = FloatField()
    F_cycle_diff = ListField(FloatField())
    F_diff = FloatField()
    F_diff_std = FloatField()
    F_diff_diff_rel= FloatField()

    ## S1
    s_vert_1_amp = FloatField()
    s_vert_1_freq = FloatField()
    s_vert_1_phase = FloatField()
    s_vert_1_offset = FloatField()
    s_vert_1_slope = FloatField()
    s_vert_1_r2 = FloatField()
    s_vert_1_cycle_min = ListField(FloatField())
    s_vert_1_min = FloatField()
    s_vert_1_min_std = FloatField()
    s_vert_1_min_diff_rel = FloatField()
    s_vert_1_cycle_max = ListField(FloatField())
    s_vert_1_max = FloatField()
    s_vert_1_max_std = FloatField()
    s_vert_1_max_diff_rel = FloatField()
    s_vert_1_cycle_mean = ListField(FloatField())
    s_vert_1_mean = FloatField()
    s_vert_1_mean_std = FloatField()
    s_vert_1_mean_diff_rel = FloatField()
    s_vert_1_cycle_diff = ListField(FloatField())
    s_vert_1_diff = FloatField()
    s_vert_1_diff_std = FloatField()
    s_vert_1_diff_diff_rel = FloatField()
    ## S2
    s_vert_2_amp = FloatField()
    s_vert_2_freq = FloatField()
    s_vert_2_phase = FloatField()
    s_vert_2_offset = FloatField()
    s_vert_2_slope = FloatField()
    s_vert_2_r2 = FloatField()
    s_vert_2_cycle_min = ListField(FloatField())
    s_vert_2_min = FloatField()
    s_vert_2_min_std = FloatField()
    s_vert_2_min_diff_rel = FloatField()
    s_vert_2_cycle_max = ListField(FloatField())
    s_vert_2_max = FloatField()
    s_vert_2_max_std = FloatField()
    s_vert_2_max_diff_rel = FloatField()
    s_vert_2_cycle_mean = ListField(FloatField())
    s_vert_2_mean = FloatField()
    s_vert_2_mean_std = FloatField()
    s_vert_2_mean_diff_rel = FloatField()
    s_vert_2_cycle_diff = ListField(FloatField())
    s_vert_2_diff = FloatField()
    s_vert_2_diff_std = FloatField()
    s_vert_2_diff_diff_rel = FloatField()
    ## S-Sum
    s_vert_sum_amp = FloatField()
    s_vert_sum_freq = FloatField()
    s_vert_sum_phase = FloatField()
    s_vert_sum_offset = FloatField()
    s_vert_sum_slope = FloatField()
    s_vert_sum_r2 = FloatField()
    s_vert_sum_cycle_min = ListField(FloatField())
    s_vert_sum_min = FloatField()
    s_vert_sum_min_std = FloatField()
    s_vert_sum_min_diff_rel = FloatField()
    s_vert_sum_cycle_max = ListField(FloatField())
    s_vert_sum_max = FloatField()
    s_vert_sum_max_std = FloatField()
    s_vert_sum_max_diff_rel = FloatField()
    s_vert_sum_cycle_mean = ListField(FloatField())
    s_vert_sum_mean = FloatField()
    s_vert_sum_mean_std = FloatField()
    s_vert_sum_mean_diff_rel = FloatField()
    s_vert_sum_cycle_diff = ListField(FloatField())
    s_vert_sum_diff = FloatField()
    s_vert_sum_diff_std = FloatField()
    s_vert_sum_diff_diff_rel = FloatField()
    
    ## optional parameters
    ## S1
    s_hor_1_amp = FloatField(required=False)
    s_hor_1_freq = FloatField(required=False)
    s_hor_1_phase = FloatField(required=False)
    s_hor_1_offset = FloatField(required=False)
    s_hor_1_slope = FloatField(required=False)
    s_hor_1_r2 = FloatField(required=False)
    s_hor_1_cycle_min = ListField(FloatField(),required=False)
    s_hor_1_min = FloatField(required=False)
    s_hor_1_min_std = FloatField(required=False)
    s_hor_1_min_diff_rel = FloatField(required=False)
    s_hor_1_cycle_max = ListField(FloatField(),required=False)
    s_hor_1_max = FloatField(required=False)
    s_hor_1_max_std = FloatField(required=False)
    s_hor_1_max_diff_rel = FloatField(required=False)
    s_hor_1_cycle_mean = ListField(FloatField(),required=False)
    s_hor_1_mean = FloatField(required=False)
    s_hor_1_mean_std = FloatField(required=False)
    s_hor_1_mean_diff_rel = FloatField(required=False)
    s_hor_1_cycle_diff = ListField(FloatField(),required=False)
    s_hor_1_diff = FloatField(required=False)
    s_hor_1_diff_std = FloatField(required=False)
    s_hor_1_diff_diff_rel = FloatField(required=False)
    ## S2
    s_hor_2_amp = FloatField(required=False)
    s_hor_2_freq = FloatField(required=False)
    s_hor_2_phase = FloatField(required=False)
    s_hor_2_offset = FloatField(required=False)
    s_hor_2_slope = FloatField(required=False)
    s_hor_2_r2 = FloatField(required=False)
    s_hor_2_cycle_min = ListField(FloatField(),required=False)
    s_hor_2_min = FloatField(required=False)
    s_hor_2_min_std = FloatField(required=False)
    s_hor_2_min_diff_rel = FloatField(required=False)
    s_hor_2_cycle_max = ListField(FloatField(),required=False)
    s_hor_2_max = FloatField(required=False)
    s_hor_2_max_std = FloatField(required=False)
    s_hor_2_max_diff_rel = FloatField(required=False)
    s_hor_2_cycle_mean = ListField(FloatField(), required=False)
    s_hor_2_mean = FloatField(required=False)
    s_hor_2_mean_std = FloatField(required=False)
    s_hor_2_mean_diff_rel = FloatField(required=False)
    s_hor_2_cycle_diff = ListField(FloatField(), required=False)
    s_hor_2_diff = FloatField(required=False)
    s_hor_2_diff_std = FloatField(required=False)
    s_hor_2_diff_diff_rel = FloatField(required=False)
    ## Piston
    s_hor_piston_amp = FloatField(required=False)
    s_hor_piston_freq = FloatField(required=False)
    s_hor_piston_phase = FloatField(required=False)
    s_hor_piston_offset = FloatField(required=False)
    s_hor_piston_slope = FloatField(required=False)
    s_hor_piston_r2 = FloatField(required=False)
    s_hor_piston_cycle_min = ListField(FloatField(),required=False)
    s_hor_piston_min = FloatField(required=False)
    s_hor_piston_min_std = FloatField(required=False)
    s_hor_piston_min_diff_rel = FloatField(required=False)
    s_hor_piston_cycle_max = ListField(FloatField(),required=False)
    s_hor_piston_max = FloatField(required=False)
    s_hor_piston_max_std = FloatField(required=False)
    s_hor_piston_max_dif_rel = FloatField(required=False)
    s_hor_piston_cycle_mean = ListField(FloatField(),required=False)
    s_hor_piston_mean = FloatField(required=False)
    s_hor_piston_mean_std = FloatField(required=False)
    s_hor_piston_mean_diff_rel = FloatField(required=False)
    s_hor_piston_cycle_diff = ListField(FloatField(),required=False)
    s_hor_piston_diff = FloatField(required=False)
    s_hor_piston_diff_std = FloatField(required=False)
    s_hor_piston_diff_diff_rel = FloatField(required=False)