import sys 
import os
import yaml

class LogHandler(object):
    def __init__(self):
        self.fstr = ""

    def print(self):
        print(self.fstr)

    def dump(self):
        with open("inference_profile.log",'w') as fout:
            fout.write(self.fstr)

    def append(self, new_str):
        self.fstr += new_str + "\n"

if __name__ == "__main__":
    filename = sys.argv[1]
    f = open(filename, 'r')
    config = yaml.load(f)
    ## general info
    cuda_version = config["cuda_version"]
    cudnn_version = config["cudnn_version"]
    trt_version = config["cudnn_version"]
    python_version = config["python_version"]
    gcc_version = config["gcc_version"]
    paddle_version = config["paddle_servion"]
    cpu = config["cpu"]
    gpu = config["gpu"]
    xpu = config["xpu"]
    api = config["api"]
    owner = config["owner"]
    ## model info
    model_name = config["model_name"]
    model_type = config["model_type"]
    model_source = config["model_source"]
    model_url = config["model_url"]
    ## data info
    batch_size = config["batch_size"]
    num_of_samples = config["num_of_samples"]
    input_shape = config["input_shape"]
    ## conf info
    runtime_device = config["runtime_device"]
    ir_optim = config["ir_optim"]
    enable_memory_optim = config["enable_memory_optim"]
    enable_tensorrt = config["enable_tensorrt"]
    precision = config["precision"]
    enable_mkldnn = config["enable_mkldnn"]
    cpu_math_library_num_threads = config["cpu_math_library_num_threads"]
    ## acc info 
    acc1 = "Nan"
    acc5 = "Nan"
    ## perf info
    average_latency, QPS = "", ""
    process_latency = ""
    cpu_rss, vms, shared, dirty, cpu_usage = "", "", "", "", ""
    gpu_id, total, free, used, gpu_utilization_rate, gpu_mem_utilization_rate = "","","","","", ""

    fh = LogHandler()

    fh.append("cuda_version: {}".format(cuda_version))
    fh.append("cudnn_version: {}".format(cudnn_version))
    fh.append("trt_version: {} ".format(trt_version))
    fh.append("python_version: {}".format(python_version))
    fh.append("gcc_version: {}".format(gcc_version))
    fh.append("paddle_version: {}".format(paddle_version))
    fh.append("cpu: {}".format(cpu))
    fh.append("gpu: {}".format(gpu)) # p4, v100, 1080
    fh.append("xpu: {}".format(xpu)) 
    fh.append("api: {}".format(api))
    fh.append("owner: {}".format(owner))
    fh.append("----------------------- Model info ----------------------")
    fh.append("model_name: {}".format(model_name))
    fh.append("model_type: {}".format(model_type))
    fh.append("model_source: {}".format(model_source))
    fh.append("model_url: {}".format(model_url))
    fh.append("----------------------- Data info -----------------------")
    fh.append("batch_size: {}".format(batch_size))
    fh.append("num_of_samples: {}".format(num_of_samples))
    fh.append("input_shape: {}".format(input_shape))
    fh.append("----------------------- Conf info -----------------------")
    fh.append("runtime_device: {}".format(runtime_device))
    fh.append("ir_optim: {}".format(ir_optim))
    fh.append("enable_memory_optim: {}".format(enable_memory_optim))
    fh.append("enable_tensorrt: {}".format(enable_tensorrt))
    fh.append("precision: {}".format(precision))    # fp32, fp16, int8
    fh.append("enable_mkldnn: {}".format(enable_mkldnn))
    fh.append("cpu_math_library_num_threads: {}".format(cpu_math_library_num_threads))
    fh.append("----------------------- Acc info ------------------------")
    fh.append("acc1:".format(acc1))
    fh.append("acc5:".format(acc5))
    fh.append("----------------------- Perf info -----------------------")
    fh.append("average_latency(ms): {}, QPS: {}".format(average_latency, QPS))
    fh.append("process_latency(ms): {}".format(process_latency))
    fh.append("process_name: clas_benchmark, cpu_rss(MB): {}, vms(MB): {}, shared(MB): {}, dirty(MB): {}, cpu_usage(%): {}".format(cpu_rss, vms, shared, dirty, cpu_usage))
    fh.append("gpu_id: {}, total(MB): {}, free(MB): {}, used(MB): {}, gpu_utilization_rate(%): {}, gpu_mem_utilization_rate(%): {}".format(gpu_id, total, free, used, gpu_utilization_rate, gpu_mem_utilization_rate))

fh.dump()