96 lines
1.9 KiB
Python
Executable File
96 lines
1.9 KiB
Python
Executable File
#!/bin/env python
|
|
__author__ = 'dongyun.zdy'
|
|
import getopt
|
|
import sys
|
|
import math
|
|
|
|
STARTUP = 0.0
|
|
|
|
def get_model_form(args,
|
|
params
|
|
):
|
|
(
|
|
Nrow,
|
|
Ncol,
|
|
) = args
|
|
|
|
(
|
|
# Tstartup,
|
|
Trow_once,
|
|
Trow_col,
|
|
Tlog1,
|
|
Tlog2,
|
|
# Tlog3
|
|
) = params
|
|
|
|
global STARTUP
|
|
total_cost = STARTUP#Tstartup
|
|
total_cost += Nrow * (Trow_once + Ncol * Trow_col)
|
|
total_cost += Tlog1 * math.log(Tlog2 * Nrow , 2)
|
|
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 file_name.find('rc') != -1:
|
|
STARTUP = 178.0
|
|
elif file_name.find('bc') != -1:
|
|
STARTUP = 210.0
|
|
else:
|
|
STARTUP = 520.0
|
|
|
|
|
|
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[4]
|
|
cost_val = get_model_form(args, model_params)
|
|
percent = (cost_val - time) / time
|
|
|
|
new_line = ",".join([line.strip(),str(cost_val),str(percent * 100)])
|
|
new_line += "\n"
|
|
out_file.write(new_line)
|
|
|
|
out_file.close()
|
|
|
|
|