##############################################################################
# Copyright (c) 2017 ZTE Corporation and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
from index_calculation import generic_index as get_index
from index_calculation import get_reference
from result_accum import result_concat as concat


def fio_index():
    fio_dict = concat('results/fio/')
    fio_r_bm_ref = get_reference('storage', 'fio_bm', 'read', 'IOPS')
    fio_r_bm_index = get_index(fio_dict, 'fio_bm', fio_r_bm_ref, 'details', 'job_0', 'read', 'io_ps')
    fio_w_bm_ref = get_reference('storage', 'fio_bm', 'write', 'IOPS')
    fio_w_bm_index = get_index(fio_dict, 'fio_bm', fio_w_bm_ref, 'details', 'job_0', 'write', 'io_ps')

    fio_bm_index = (fio_r_bm_index + fio_w_bm_index) / 2

    fio_r_vm_ref = get_reference('storage', 'fio_vm', 'read', 'IOPS')
    fio_r_vm_index = get_index(fio_dict, 'fio_vm', fio_r_vm_ref, 'details', 'job_0', 'read', 'io_ps')

    fio_w_vm_ref = get_reference('storage', 'fio_vm', 'write', 'IOPS')
    fio_w_vm_index = get_index(fio_dict, 'fio_vm', fio_w_vm_ref, 'details', 'job_0', 'write', 'io_ps')

    fio_vm_index = (fio_r_vm_index + fio_w_vm_index) / 2

    fio_index = (fio_bm_index + fio_vm_index) / 2
    print fio_index

    fio_dict_i = {}
    fio_dict_i['index'] = fio_index
    fio_dict_i['results'] = fio_dict
    return fio_dict_i