fibren/plot_script_1alpha.py at master · anirban-pal/fibren · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#!/usr/bin/python

import matplotlib.pyplot as plt
import matplotlib.pylab as pl
import numpy as np
import csv
import os
import subprocess
from itertools import cycle

plt.rcParams.update({'font.size': 12})
#cmd = "./main 0.00 0.00 1.0 1.0 0 gridfile.dat"
#returned_value = os.system(cmd)  # returns the exit code in unix
returned_value = os.system("seq 0.0 0.001 2.5 > gridfile.dat")

qmid = [1.1, 0.0, 0.0]; ymax = 2.5; ymin = -0.1;
#qmid = [1.1, 1.1, 0.0]; ymax = 4.0; ymin = -0.1;
#qmid = [0.0, 1.1, 0.0]; ymax = 25.0; ymin = -0.1;

tmax = 180
tmin = 0
interval = 1
theta_range = range(tmin,tmax,interval)
colors = pl.cm.Dark2(np.linspace(0.5,1,(tmax-tmin)/interval))

fig, ax = plt.subplots(1, 2, figsize=(15,15))
ip = 0;

alpha_range = range(30,90,100);

for alpha in alpha_range:
	i = 0;

	lines = ["-","--","-.",":"]
	linecycler = cycle(lines)

	for theta in theta_range:
		cmd = "./main "+str(theta)+" "+str(alpha)+" "+str(qmid[0])+" "+str(qmid[1])+" "+str(qmid[2])+" gridfile.dat | grep Energy"
		#print(cmd)
		#returned_value = os.system(cmd)
		#print(returned_value)

		hosts = subprocess.check_output(cmd, shell=True)
		error = float(hosts.split(' ')[1])-1
		print(cmd+" Error= "+str(error))
		thetalabel = r'$\theta$ = '+str(theta)+r'$\degree$ $\varepsilon$ = '+"{:.1e}".format(error)
		thetatitle = r'$\alpha$ = '+str(alpha)+r'$\degree$'

		x = []
		y = []

		with open('outfile.gr','r') as csvfile:
		    next(csvfile);
		    next(csvfile);
		    plots = csv.reader(csvfile, delimiter=' ')
		    for row in plots:
		        x.append(float(row[0]))
		        y.append(float(row[1]))

		ax[ip].plot(x,y, next(linecycler), label=thetalabel, color=colors[i])
		i+=1

	ax[ip].set_xlabel('r', fontsize=12)
	#ax[ip].set_ylabel('g(r)', fontsize=18)
	ax[ip].set_ylim(ymin, ymax)
	ax[ip].set_xlim(0.0, 2.5)
	ax[ip].set_title(thetatitle, fontsize=12)
	#ax[ip].legend(loc=9)


	#plt.show()
	#plt.plot(x,x, label='Loaded from file!')
	ip+=1

maintitle = "Distance distribution g(r) with "+"q_mid = "+str(qmid)
#plt.savefig('outfile.png')
plt.suptitle(maintitle);
fileid = qmid[0]*4+qmid[1]*2+qmid[0];
filename = "outfile_"+str(fileid)+".svg"
plt.savefig(filename, format="svg", bbox_inches='tight')