import datetime

from mongoengine import *

from .data import DataSheartest
from .material import Material


class DynamicShearTest(Document):

    date = DateTimeField(default=datetime.datetime.now,
                         wtf_options={"render_kw": {
                             "step": "60"
                         }})

    tags = ListField(StringField())

    standard = StringField(default='TP Asphalt Teil 24')

    lab = StringField(default='TU Dresden', required=True)
    auditor = StringField(default=None)
    machine = StringField(default=None)

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

    project = StringField(required=True)
    workpackage = StringField()

    material1 = LazyReferenceField(Material, required=True)
    material2 = LazyReferenceField(Material, required=True)
    bounding = LazyReferenceField(Material, required=True)

    gap_width = FloatField(default=1.0)

    meta = {
        'allow_inheritance': True,
        'index_opts': {},
        'index_background': True,
        'index_cls': False,
        'auto_create_index': True,
        'collection': 'sheartest',
    }


class DynamicShearTestExtension(DynamicShearTest):

    #metadata
    f = FloatField()
    sigma_normal = FloatField()
    T = FloatField()
    extension = FloatField()

    #results
    data = LazyReferenceField(DataSheartest,
                              required=True,
                              reverse_delete_rule=CASCADE)

    stiffness = FloatField()
    #fit parameter
    ## F
    fit_amp_F = FloatField()
    fit_freq_F = FloatField()
    fit_phase_F = FloatField()
    fit_offset_F = FloatField()
    fit_slope_F = FloatField()
    ## S1
    fit_amp_s_vert_1 = FloatField()
    fit_freq_s_vert_1 = FloatField()
    fit_phase_s_vert_1 = FloatField()
    fit_offset_s_vert_1 = FloatField()
    fit_slope_s_vert_1 = FloatField()
    r2_s_vert_1 = FloatField()
    ## S2
    fit_amp_s_vert_2 = FloatField()
    fit_freq_s_vert_2 = FloatField()
    fit_phase_s_vert_2 = FloatField()
    fit_offset_s_vert_2 = FloatField()
    fit_slope_s_vert_2 = FloatField()
    r2_s_vert_2 = FloatField()
    ## S-Sum
    fit_amp_s_vert_sum = FloatField()
    fit_freq_s_vert_sum = FloatField()
    fit_phase_s_vert_sum = FloatField()
    fit_offset_s_vert_sum = FloatField()
    fit_slope_s_vert_sum = FloatField()
    r2_s_vert_sum = FloatField()
    ## r2
    r2_F = FloatField()
    r2_s_vert_1 = FloatField()
    r2_s_vert_2 = FloatField()
    r2_s_vert_sum = FloatField()