forked from sPaMFouR/RedPipe
-
Notifications
You must be signed in to change notification settings - Fork 0
/
PlotFitNagy.py
172 lines (135 loc) · 8.86 KB
/
PlotFitNagy.py
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
#!/usr/bin/env python
# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx #
# xxxxxxxxxxxxxxxxxxxxxxxxxxxxx-------------------Nagy LIGHT CURVE MODEL-----------------xxxxxxxxxxxxxxxxxxxxxxxxxxxx #
# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx #
# ------------------------------------------------------------------------------------------------------------------- #
# Import Required Libraries
# ------------------------------------------------------------------------------------------------------------------- #
import os
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from scipy.stats import chisquare
from matplotlib.ticker import MultipleLocator
from scipy.interpolate import InterpolatedUnivariateSpline
# ------------------------------------------------------------------------------------------------------------------- #
# ------------------------------------------------------------------------------------------------------------------- #
# Data Files And Paths of Directories
# ------------------------------------------------------------------------------------------------------------------- #
core_file = 'core.out'
shell_file = 'shell.out'
comb_file = 'comb.out'
comb_file2 = 'comb2.out'
comb_file3 = 'comb3.out'
comb_file4 = 'comb4.out'
obsbol_file = 'OUTPUT_DateWiseSNBolFlux'
DIR_CURNT = os.getcwd()
DIR_Model = "/home/avinash/Dropbox/ModelSNe/Nagy_LC2/"
# ------------------------------------------------------------------------------------------------------------------- #
# ------------------------------------------------------------------------------------------------------------------- #
# Global Variables To Be Used In The Code
# ------------------------------------------------------------------------------------------------------------------- #
precision = 3
date_explosion = 2457644.60
out_cols = ['Phase', 'Lum', 'LogLum', 'LDiff', 'LRec', 'Xi']
comb_cols = ['Phase', 'LumCore', 'LumShell']
# ------------------------------------------------------------------------------------------------------------------- #
# ------------------------------------------------------------------------------------------------------------------- #
# Function To Calculate Chi-Square Statistic
# ------------------------------------------------------------------------------------------------------------------- #
def calc_chisquare(obs_data, model_data):
"""
Calculates the goodness-of-fit through the chi-square parameter. Returns the chi-square statistic.
Args:
obs_data : Observed Data
model_data : Model Data
Returns:
fit_chisq.statistic : The sum of squared-errors between the model and the observed data
"""
fit_interp = InterpolatedUnivariateSpline(model_data['Phase'], model_data['Lum'], k=1)
fit_chisq = chisquare(obs_data['Lum'], fit_interp(obs_data['Phase']))
return fit_chisq.statistic
# ------------------------------------------------------------------------------------------------------------------- #
# ------------------------------------------------------------------------------------------------------------------- #
# Function For Plotting
# ------------------------------------------------------------------------------------------------------------------- #
def set_plotparams(ax_obj):
"""
Sets the plot parameters for displaying the Light Curves and Expansion Velocity Evolution.
Args:
ax_obj : Axes object to be used for plotting and setting plot parameters
Returns:
None
"""
ax_obj.set_ylim(4e40, 5e42)
ax_obj.set_xlim(-10, 210)
ax_obj.yaxis.set_ticks_position('both')
ax_obj.xaxis.set_ticks_position('both')
ax_obj.xaxis.set_major_locator(MultipleLocator(50))
ax_obj.xaxis.set_minor_locator(MultipleLocator(10))
ax_obj.tick_params(which='both', direction='in', width=1, labelsize=14)
# ------------------------------------------------------------------------------------------------------------------- #
# ------------------------------------------------------------------------------------------------------------------- #
# Reads I/O Data Files
# ------------------------------------------------------------------------------------------------------------------- #
# os.system('./run')
obsbol_df = pd.read_csv(obsbol_file, sep='\s+', engine='python')
obsbol_df = obsbol_df[obsbol_df['Phase'] < 200]
comb_df = pd.read_csv(comb_file, sep='\s+', header=None, names=comb_cols, comment='#', engine='python')
comb_df['Lum'] = comb_df['LumCore'] + comb_df['LumShell']
comb2_df = pd.read_csv(comb_file2, sep='\s+', header=None, names=comb_cols, comment='#', engine='python')
comb2_df['Lum'] = comb2_df['LumCore'] + comb2_df['LumShell']
comb3_df = pd.read_csv(comb_file3, sep='\s+', header=None, names=comb_cols, comment='#', engine='python')
comb3_df['Lum'] = comb3_df['LumCore'] + comb3_df['LumShell']
comb4_df = pd.read_csv(comb_file4, sep='\s+', header=None, names=comb_cols, comment='#', engine='python')
comb4_df['Lum'] = comb4_df['LumCore'] + comb4_df['LumShell']
print calc_chisquare(obsbol_df, comb_df)
print calc_chisquare(obsbol_df, comb2_df)
print calc_chisquare(obsbol_df, comb3_df)
print calc_chisquare(obsbol_df, comb4_df)
# ------------------------------------------------------------------------------------------------------------------- #
# ------------------------------------------------------------------------------------------------------------------- #
# Plots The Fit To The Bolometric Light Curve
# ------------------------------------------------------------------------------------------------------------------- #
fig = plt.figure(figsize=(12, 8))
ax1 = fig.add_subplot(221)
ax1.semilogy(obsbol_df['Phase'], obsbol_df['Lum'], c='b', marker='o', ls='', label='Observed LC')
ax1.semilogy(comb_df['Phase'], comb_df['LumCore'], c='r', ls='--', lw=0.7, label='Model Core LC (a=0)')
ax1.semilogy(comb_df['Phase'], comb_df['LumShell'], c='g', ls='--', lw=0.7, label='Model Shell LC (n=2)')
ax1.semilogy(comb_df['Phase'], comb_df['Lum'], c='k', ls='-', lw=0.7, label='Model Combined LC')
set_plotparams(ax1)
ax1.text(120, 2e41, s=r'$\rm T_{rec}\ =\ 5500\ K$' + '\n' + r'$\rm \kappa_{shell}\ =\ 0.4\ cm^2\ g^{-1}$', fontsize=13)
ax2 = fig.add_subplot(222, sharey=ax1)
ax2.semilogy(obsbol_df['Phase'], obsbol_df['Lum'], c='b', marker='o', ls='', label='Observed LC')
ax2.semilogy(comb2_df['Phase'], comb2_df['LumCore'], c='r', ls='--', lw=0.7, label='Model Core LC (a=0)')
ax2.semilogy(comb2_df['Phase'], comb2_df['LumShell'], c='g', ls='--', lw=0.7, label='Model Shell LC (n=2)')
ax2.semilogy(comb2_df['Phase'], comb2_df['Lum'], c='k', ls='-', lw=0.7, label='Model Combined LC')
set_plotparams(ax2)
ax2.legend(fontsize=12, frameon=False)
ax2.tick_params(axis='y', which='both', direction='in', labelleft='off')
ax2.text(120, 2e41, s=r'$\rm T_{rec}\ =\ 5500\ K$' + '\n' + r'$\rm \kappa_{shell}\ =\ 0.3\ cm^2\ g^{-1}$', fontsize=13)
ax3 = fig.add_subplot(223)
ax3.semilogy(obsbol_df['Phase'], obsbol_df['Lum'], c='b', marker='o', ls='', label='Observed LC')
ax3.semilogy(comb3_df['Phase'], comb3_df['LumCore'], c='r', ls='--', lw=0.7, label='Model Core LC (a=0)')
ax3.semilogy(comb3_df['Phase'], comb3_df['LumShell'], c='g', ls='--', lw=0.7, label='Model Shell LC (n=2)')
ax3.semilogy(comb3_df['Phase'], comb3_df['Lum'], c='k', ls='-', lw=0.7, label='Model Combined LC')
set_plotparams(ax3)
ax3.text(120, 2e41, s=r'$\rm T_{rec}\ =\ 7000\ K$' + '\n' + r'$\rm \kappa_{shell}\ =\ 0.4\ cm^2\ g^{-1}$', fontsize=13)
ax4 = fig.add_subplot(224, sharey=ax3)
ax4.semilogy(obsbol_df['Phase'], obsbol_df['Lum'], c='b', marker='o', ls='', label='Observed LC')
ax4.semilogy(comb4_df['Phase'], comb4_df['LumCore'], c='r', ls='--', lw=0.7, label='Model Core LC (a=0)')
ax4.semilogy(comb4_df['Phase'], comb4_df['LumShell'], c='g', ls='--', lw=0.7, label='Model Shell LC (n=2)')
ax4.semilogy(comb4_df['Phase'], comb4_df['Lum'], c='k', ls='-', lw=0.7, label='Model Combined LC')
set_plotparams(ax4)
ax4.tick_params(axis='y', which='both', direction='in', labelleft='off')
ax4.text(120, 2e41, s=r'$\rm T_{rec}\ =\ 7000\ K$' + '\n' + r'$\rm \kappa_{shell}\ =\ 0.3\ cm^2\ g^{-1}$', fontsize=13)
fig.text(0.06, 0.5, r'Quasi-Bolometric Luminosity [$\rm erg\ s^{-1}$]', va='center', rotation='vertical', fontsize=14)
fig.text(0.5, 0.07, 'Time Since Explosion [Days]', ha='center', fontsize=14)
# plt.tick_params(labelcolor='none', top='off', bottom='off', left='off', right='off')
# plt.ylabel(r'Bolometric Luminosity [$\rm erg\ s^{-1}$]', fontsize=14)
# plt.xlabel('Time Since Explosion [Days]', fontsize=14)
fig.subplots_adjust(wspace=0.001, hspace=0.001, top=0.9, right=0.95)
fig.savefig('PLOT_FitNagy.pdf', format='pdf', dpi=2000, bbox_inches='tight')
plt.show()
plt.close(fig)
# ------------------------------------------------------------------------------------------------------------------- #