89 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			89 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
| #!/bin/env python
 | |
| __author__ = 'dongyun.zdy'
 | |
| import getopt
 | |
| import sys
 | |
| import math
 | |
| 
 | |
| 
 | |
| 
 | |
| def array_model_form(args,
 | |
|                      params):
 | |
|     Nelem = args
 | |
| 
 | |
|     (
 | |
|         Telem_ence,
 | |
|         Telem_copy
 | |
|     ) = params
 | |
| 
 | |
|     ELEM_PER_PAGE = 1024
 | |
|     extend_cnt = math.ceil(math.log(float(Nelem)/ELEM_PER_PAGE, 2))
 | |
|     if extend_cnt < 0:
 | |
|         extend_cnt = 0
 | |
|     copy_cnt = ELEM_PER_PAGE * (math.pow(2, extend_cnt) - 1)
 | |
| 
 | |
|     total_cost = Telem_ence * Nelem
 | |
|     #total_cost += Tmem_alloc * extend_cnt
 | |
|     total_cost += Telem_copy * copy_cnt
 | |
| 
 | |
|     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])
 | |
|     time = case_param[1]
 | |
|     cost_val = array_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)
 | |
| 
 | |
|     new_line = ",".join([line.strip(), str(cost_val)])
 | |
|     new_line += "\n"
 | |
|     out_file.write(new_line)
 | |
| 
 | |
| out_file.close()
 | |
| 
 | |
| 
 | 
