oceanbase/unittest/sql/optimizer/cost_model_utils/apply_material_model.py
wangzelin.wzl 93a1074b0c patch 4.0
2022-10-24 17:57:12 +08:00

83 lines
1.6 KiB
Python
Executable File

#!/bin/env python
__author__ = 'dongyun.zdy'
import getopt
import sys
import math
def material_model_form(args,
params):
(
Nrow,
Ncol,
) = args
(
# Tstartup,
Trow_once,
Trow_col
) = params
total_cost = 0 #Tstartup
total_cost += Nrow * (Trow_once + Ncol * Trow_col)
return total_cost
def extract_info_from_line(line):
splited = line.split(",")
line_info = []
for item in splited:
line_info.append(float(item))
return line_info
file_name = "get_total.data.prep"
output_fit_res = False
wrong_arg = False
opts,args = getopt.getopt(sys.argv[1:],"i:o:m:")
for op, value in opts:
if "-i" == op:
file_name = value
elif "-o" == op:
output_fit_res = True
out_file_name = value
elif "-m" == op:
model_file_name = value
else:
wrong_arg = True
if wrong_arg:
print "wrong arg"
sys.exit(1)
input_file = open(file_name, "r")
model_file = open(model_file_name, "r")
out_file = open(out_file_name, "w")
line = model_file.readline()
model_params = [float(p) for p in line.split(",")]
for line in input_file:
if line.startswith('#'):
out_file.write(line)
continue
case_param = extract_info_from_line(line)
args = (case_param[0],
case_param[1])
time = case_param[3]
cost_val = material_model_form(args, model_params)
percent = (cost_val - time) / time
new_line = ",".join([line.strip(),"\t" ,str(cost_val),"\t" , str(time),"\t\t" , str(percent * 100)])
new_line += "\n"
out_file.write(new_line)
out_file.close()