#!/usr/bin/python

##
## Copyright (c) 2020 Intel Corporation
##
## Licensed under the Apache License, Version 2.0 (the "License");
## you may not use this file except in compliance with the License.
## You may obtain a copy of the License at
##
##
##     http://www.apache.org/licenses/LICENSE-2.0
##
## Unless required by applicable law or agreed to in writing, software
## distributed under the License is distributed on an "AS IS" BASIS,
## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
## See the License for the specific language governing permissions and
## limitations under the License.
##

from past.utils import old_div
import sys
import time
import requests
from rapid_log import RapidLog
from rapid_test import RapidTest

class IrqTest(RapidTest):
    """
    Class to manage the irq testing
    """
    def __init__(self, test_param, runtime, pushgateway, environment_file,
            machines):
        super().__init__(test_param, runtime, pushgateway, environment_file)
        self.machines = machines

    def run(self):
        RapidLog.info("+----------------------------------------------------------------------------------------------------------------------------")
        RapidLog.info("| Measuring time probably spent dealing with an interrupt. Interrupting DPDK cores for more than 50us might be problematic   ")
        RapidLog.info("| and result in packet loss. The first row shows the interrupted time buckets: first number is the bucket between 0us and    ")
        RapidLog.info("| that number expressed in us and so on. The numbers in the other rows show how many times per second, the program was       ")
        RapidLog.info("| interrupted for a time as specified by its bucket. '0' is printed when there are no interrupts in this bucket throughout   ")
        RapidLog.info("| the duration of the test. 0.00 means there were interrupts in this bucket but very few. Due to rounding this shows as 0.00 ") 
        RapidLog.info("+----------------------------------------------------------------------------------------------------------------------------")
        sys.stdout.flush()
        for machine in self.machines:
            buckets=machine.socket.show_irq_buckets(1)
            print('Measurement ongoing ... ',end='\r')
            machine.stop()
            old_irq = [[0 for x in range(len(buckets)+1)] for y in range(len(machine.get_cores())+1)] 
            irq = [[0 for x in range(len(buckets)+1)] for y in range(len(machine.get_cores())+1)]
            irq[0][0] = 'bucket us' 
            for j,bucket in enumerate(buckets,start=1):
                irq[0][j] = '<'+ bucket
            irq[0][-1] = '>'+ buckets [-2]
            machine.start()
            time.sleep(2)
            for j,bucket in enumerate(buckets,start=1):
                for i,irqcore in enumerate(machine.get_cores(),start=1):
                    old_irq[i][j] = machine.socket.irq_stats(irqcore,j-1)
            time.sleep(float(self.test['runtime']))
            machine.stop()
            for i,irqcore in enumerate(machine.get_cores(),start=1):
                irq[i][0]='core %s'%irqcore
                for j,bucket in enumerate(buckets,start=1):
                    diff =  machine.socket.irq_stats(irqcore,j-1) - old_irq[i][j]
                    if diff == 0:
                        irq[i][j] = '0'
                    else:
                        irq[i][j] = str(round(old_div(diff,float(self.test['runtime'])), 2))
            RapidLog.info('Results for PROX instance %s'%machine.name)
            for row in irq:
                RapidLog.info(''.join(['{:>12}'.format(item) for item in row]))
            if self.test['pushgateway']:
                URL = self.test['pushgateway'] + self.test['test']+ '/instance/' + self.test['environment_file']
                HEADERS = {'X-Requested-With': 'Python requests', 'Content-type': 'text/xml'}
                #DATA = 'Machine {}\n'.format(machine.name)
                for i,irqcore in enumerate(machine.get_cores(),start=1):
                    DATA = '{}\n'.format(irq[i][0])
                    for j,bucket in enumerate(buckets,start=1):
                        DATA = DATA + 'B{} {}\n'.format(irq[0][j].replace(">","M").replace("<","").replace(" ",""),irq[i][j])
                    response = requests.post(url=URL, data=DATA,headers=HEADERS)
                    if (response.status_code != 202) and (response.status_code != 200):
                        RapidLog.info('Cannot send metrics to {}'.format(URL))
                        RapidLog.info(DATA)
        return (True)