52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
import os.path,subprocess
|
|
import matplotlib.pyplot as plt
|
|
|
|
from subprocess import STDOUT,PIPE
|
|
from typing import ByteString
|
|
|
|
def compile_java(java_file):
|
|
subprocess.check_call(['javac', java_file])
|
|
|
|
def execute_java(java_file, stdin, probability, width):
|
|
java_class,ext = os.path.splitext(java_file)
|
|
cmd = ['java', java_class, str(probability), str(width)]
|
|
proc = subprocess.Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=STDOUT)
|
|
stdout,stderr = proc.communicate(stdin)
|
|
return stdout.decode('utf-8')
|
|
|
|
filename = 'Main.java'
|
|
compile_java(filename)
|
|
|
|
# fixed width
|
|
width_list = [x for x in range(100,251,50)]
|
|
for w in width_list:
|
|
probabily_list = [x/100 for x in range(0,100,5)]
|
|
time_list = []
|
|
for probability in probabily_list:
|
|
time_list.append(float(execute_java(filename, '', probability, w).strip('\n')))
|
|
print(time_list)
|
|
plt.plot(probabily_list, time_list)
|
|
plt.legend(['width = ' + str(w) for w in width_list], loc='upper left')
|
|
plt.xlabel('Probability')
|
|
plt.ylabel('Time')
|
|
plt.title('Variable Probability')
|
|
plt.savefig('fixed_width.png')
|
|
plt.close()
|
|
|
|
# variable width
|
|
p_list = [0.2,0.5,0.8]
|
|
for p in p_list:
|
|
w_list = [x for x in range(2,201)]
|
|
time_list = []
|
|
for width in w_list:
|
|
time_list.append(float(execute_java(filename, '', p, width).strip('\n')))
|
|
print(time_list)
|
|
plt.plot(w_list, time_list)
|
|
plt.legend(['probability = ' + str(p) for p in p_list], loc='upper left')
|
|
plt.xlabel('Width')
|
|
plt.ylabel('Time')
|
|
plt.title('Variable width')
|
|
plt.savefig('variable_width.png')
|
|
|
|
|