import winfault

Turbine = winfault.WT_data()

scada = Turbine.scada_data

# This gets all the data EXCEPT the faults listed. Labels as nf for "no-fault"
nf = Turbine.filter(scada, Turbine.status_data_wec, "Main_Status",
                    'fault_case_1', True, 600, 600, [62, 9, 80])
# feeding fault
ff = Turbine.filter(scada, Turbine.status_data_wec, "Main_Status",
                    'fault_case_1', False, 600, 600, 62)
# mains failure fault
mf = Turbine.filter(scada, Turbine.status_data_wec, "Main_Status",
                    'fault_case_1', False, 600, 600, 60)

# generator heating fault
gf = Turbine.filter(scada, Turbine.status_data_wec, "Main_Status",
                    'fault_case_1', False, 600, 600, 9)

# aircooling fault
af = Turbine.filter(scada, Turbine.status_data_wec, "Main_Status",
                    'fault_case_1', False, 600, 600, 228)

# excitation fault
ef = Turbine.filter(scada, Turbine.status_data_wec, "Main_Status",
                    'fault_case_1', False, 600, 600, 80)

features = ['WEC_ava_windspeed',
            'WEC_ava_Rotation',
            'WEC_ava_Power',
            'WEC_ava_reactive_Power',
            'WEC_ava_blade_angle_A',
            'Inverter_averages',
            'Inverter_std_dev',
            'CS101__Spinner_temp',
            'CS101__Front_bearing_temp',
            'CS101__Rear_bearing_temp',
            'CS101__Pitch_cabinet_blade_A_temp',
            'CS101__Pitch_cabinet_blade_B_temp',
            'CS101__Pitch_cabinet_blade_C_temp',
            'CS101__Rotor_temp_1',
            'CS101__Rotor_temp_2',
            'CS101__Stator_temp_1',
            'CS101__Stator_temp_2',
            'CS101__Nacelle_ambient_temp_1',
            'CS101__Nacelle_ambient_temp_2',
            'CS101__Nacelle_temp',
            'CS101__Nacelle_cabinet_temp',
            'CS101__Main_carrier_temp',
            'CS101__Rectifier_cabinet_temp',
            'CS101__Yaw_inverter_cabinet_temp',
            'CS101__Fan_inverter_cabinet_temp',
            'CS101__Ambient_temp',
            'CS101__Tower_temp',
            'CS101__Control_cabinet_temp',
            'CS101__Transformer_temp']
# select the faults to include.
faults = [ff, ef, gf]

# with basic SVM, faults are found better in a 1v1 basis. For feature
# selection paper, this might be a better way to go:
# faults = [ff]
# If done this way, remember to update the labels for the confusion
# matrix as well!
# Also remember to update what faults are to be taken out for the "no-fault"
# data set:
# nf = Turbine.filter(scada,Turbine.status_data_wec, "Main_Status",
#                     'fault_case_1', True, 600,600,[62])

# label and split into train, test and balanced training data
xtrain, xtest, ytrain, ytest, xbaltrain, ybaltrain = \
    Turbine.get_test_train_data(features, faults, nf)
# labels for confusion matrix
labels = ['no-fault', 'feeding fault', 'excitation fault', 'generator fault']
# train and test svm
winfault.svm_class_and_score(xbaltrain, ybaltrain, xtest, ytest, labels)