def plotChain(File1, **keywords):
""" Plot the trace of a single parameter between two chains """
#ftag = ''
ftag = 'mcmc'
silent = False
for keyw in keywords:
if keyw == 'ftag':
ftag = keywords[keyw]
if keyw == 'Silent':
silent = keywords[keyw]
hdrKeys = readMCMChdr(File1)
for i in hdrKeys.keys():
key = hdrKeys[i]
if isNonParam(key):
pass
else:
try:
d1 = read1parMCMC(File1,key)
i1 = d1['istep']
x1 = d1[key]
pp = plt.plot(i1,x1,'b.')
plt.title('Trace for '+key)
plt.savefig(ftag+'.TRACE.par_'+key+'.png')
if not silent: print 'plotting Trace for '+key
plt.clf()
except:
print 'key not found'
raise
def plotTrace(File1,File2, **keywords):
""" Plot the trace of a single parameter between two chains """
#ftag = ''
ftag = 'mcmc'
silent = False
for keyw in keywords:
if keyw == 'ftag':
ftag = keywords[keyw]
if keyw == 'Silent':
silent = keywords[keyw]
hdrKeys = readMCMChdr(File1)
for i in hdrKeys.keys():
key = hdrKeys[i]
if isNonParam(key):
pass
else:
try:
d1 = read1parMCMC(File1,key)
d2 = read1parMCMC(File2,key)
i1 = d1['istep']
i2 = d2['istep']
x1 = d1[key]
x2 = d2[key]
pp = plt.plot(i1,x1,'bo')
plt.setp(pp,'markersize',8)
plt.setp(pp,'markerfacecolor','k')
pp = plt.plot(i2,x2,'r.')
#pp = plt.title('Trace for '+key)
plt.savefig(ftag+'.TRACE.par'+key+'.png')
if not silent: print 'plotting Trace for '+key
plt.clf()
except:
print 'key not found'
raise
def GelmanRubinConvergence(FileList,ModelFile,OutFile, **kwargs):
"""The Gelman-Rubin convergence statistic.
When this is close to 1, the parameter chain has converged"""
smallM = len(FileList)
hdrKeys = readMCMChdr(FileList[0])
GRFile = open(OutFile,'w')
print >> GRFile, "# Gelman-Rubin Statistic per parameter"
lastN = 0
for keyw in kwargs:
if keyw == 'lastn':
lastN = kwargs[keyw]
for i1 in hdrKeys.keys():
key1 = hdrKeys[i1]
if isNonParam(key1):
pass
else:
xmean_vector = []
si2_vector = []
xi2_vector = []
allX = []
for iChain in range(smallM):
dp = read1parMCMC(FileList[iChain],key1)
x = dp[key1][-1*lastN:]
if iChain == 0:
smallN = len(x)
else:
alternateN = len(x)
if alternateN != smallN:
print 'Chain lengths are not the same for'
print FileList[0], ' and ', FileList[iChain]
print 'try using lastN keyword'
sys.exit()
xmean_vector.append(np.mean(x))
si2_vector.append(np.var(x))
xi2_vector.append(np.mean(x)**2)
allX.extend(x)
xmean_vector = np.array(xmean_vector)
si2_vector = np.array(si2_vector)
xi2_vector = np.array(xi2_vector)
allX = np.array(allX)
Bn = np.var(xmean_vector)
W = np.mean(si2_vector)
sig2hat = ((smallN-1)*W)/smallN + Bn
Vhat = sig2hat + Bn/smallM
term1 = (((smallN-1)/smallN)**2)*(1e0/smallM)*np.var(si2_vector)
term2 = (((smallM + 1)/smallM)**2)*(2e0/(smallM-1))*(Bn**2)
term3a = 2e0*((smallM + 1)*(smallN -1)/(smallM*(smallN**2)) )
cov_si2_xi2 = np.cov( si2_vector, xi2_vector)
cov_si2_x = np.cov(si2_vector, xmean_vector)
xdash = np.mean(allX)
term3b = (smallN/smallM)*(cov_si2_xi2[0][1] - 2e0*xdash*cov_si2_x[0][1])
varVhat = term1 + term2 + term3a*term3b
df = 2e0*(Vhat**2)/(varVhat)
Rhat = (Vhat/W)*(df/(df-2e0))
print key1, Rhat
print >> GRFile, key1+' = ',Rhat
GRFile.close()
def autocorMCMC_old(File, lowtol, jmax, OutStatFile, mkPlotsFlag, **keywords):
""" Compute the auto-correlation of parameters in a chain """
#ftag = ''
ftag = 'mcmc'
silent = False
for keyw in keywords:
if keyw == 'ftag':
ftag = keywords[keyw]
if keyw == 'Silent':
silent = keywords[keyw]
chain_stats = {}
OutFileObject = open(OutStatFile,'w')
hdrKeys = readMCMChdr(File)
for i in hdrKeys.keys():
key = hdrKeys[i]
# empty list for autocorrelation data
x = []
# skip the non-parameter data
if isNonParam(key):
pass
else:
data = read1parMCMC(File,key)
#ChainLength = len(data['istep'])
#print np.shape(data['istep'])
if not silent: print 'Par '+key
#x = np.array(data[key])
#for i in range(ChainLength):
#x.append(data[key][i])
median_x = np.median(x)
# removing median value
x1 = np.array(data[key])-median_x
cj = [] # auto-correlation
jarr = [] # lag
# starting auto-correlation, set to be much higher than tolerance
cval = 1e0
j = 0
szx1 = len(x1)
while cval > lowtol and j < jmax:
sli0 = 0
sli1 = szx1-j-1
slj0 = j
slj1 = szx1-1
next2 = x1[sli0:sli1]
x1i_x1ipj = x1[sli0:sli1]*x1[slj0:slj1]
x1i_sq = (x1[sli0:sli1]*x1[sli0:sli1])
x1i = (x1[sli0:sli1])
val = (np.mean(x1i_x1ipj) - (np.mean(x1i))**2)/\
(np.mean(x1i_sq) - (np.mean(x1i))**2)
cj.append(val)
jarr.append(j)
cval = val
j += 1
cj = np.array(cj)
jarr = np.array(jarr)
corlen_index = np.where( np.abs(cj-0.5e0) == np.min( np.abs(cj-0.5e0)) )[0]
corlen = jarr[corlen_index[0]]
efflen = long(float(ChainLength)/float(corlen))
if mkPlotsFlag:
print 'plotting acor', ftag+'.ACOR.par_'+key+'.png'
plt.plot(jarr,cj,'b.')
plt.plot([corlen,corlen],[0,1],'k--')
plt.plot([0,max(jarr)],[0.5,0.5],'k--')
plt.xlabel('j (Lag)')
plt.ylabel('C (Auto-Correlation)')
plt.title('Auto-Correlation for "'+key+'"')
plt.savefig(ftag+'.ACOR.par_'+key+'.png')
plt.clf()
print >> OutFileObject, '##'+key+'## Corr Length = '+format(corlen,'d')
print >> OutFileObject, '##'+key+'## Eff Length = '+format(efflen,'d')
chain_stats[key] = {'corlength':corlen,'efflength':efflen}
clen = []
elen = []
for key in chain_stats.keys():
clen.append(chain_stats[key]['corlength'])
elen.append(chain_stats[key]['efflength'])
if not silent:
print '## ALL ## Chain Length = '+format(ChainLength,'d')
print '## ALL ## Corr Length = '+format(max(clen),'d')
print '## ALL ## Eff Length = '+format(min(elen),'d')
print >> OutFileObject, '## ALL ## Chain Length = '+format(ChainLength,'d')
print >> OutFileObject, '## ALL ## Corr Length = '+format(max(clen),'d')
print >> OutFileObject, '## ALL ## Eff Length = '+format(min(elen),'d')
OutFileObject.close()
def autocorMCMC(File, lowtol, jmax, OutStatFile, mkPlotsFlag, **keywords):
""" Compute the auto-correlation of parameters in a chain """
#ftag = ''
ftag = 'mcmc'
silent = False
Nres = 1
Jstep = Nres
SetDynamic = False
for keyw in keywords:
if keyw == 'ftag':
ftag = keywords[keyw]
if keyw.lower() == 'silent':
silent = keywords[keyw]
if keyw.lower().startswith('res'):
Nres = keywords[keyw]
if keyw.lower().startswith('dyn'):
SetDynamic = keywords[keyw]
chain_stats = {}
OutFileObject = open(OutStatFile,'w')
hdrKeys = readMCMChdr(File)
ParList = []
for i in hdrKeys.keys():
key = hdrKeys[i]
if not isNonParam(key):
ParList.append(key)
for key in ParList:
data = read1parMCMC(File,key)
if not silent: print 'Par '+key
x1 = np.array(data[key])-np.median(data[key])
cj = [] # auto-correlation
jarr = [] # lag
# starting auto-correlation, set to be much higher than tolerance
val = 1e0
j = 0
szx1 = len(x1)
sli0 = 0
slj1 = szx1-1
while val > lowtol and j < jmax:
sli1 = szx1-j-1
slj0 = j
#t0 = time.time()
x1i_x1ipj = x1[sli0:sli1]*x1[slj0:slj1]
x1i_sq = (x1[sli0:sli1]*x1[sli0:sli1])
x1i = (x1[sli0:sli1])
mean_x1i_x1ipj = np.mean(x1i_x1ipj)
mean_x1i = np.mean(x1i)
mean_x1i_sq = np.mean(x1i_sq)
#mean_x1i_x1ipj = quickMean(x1i_x1ipj)
#mean_x1i = quickMean(x1i)
#mean_x1i_sq = quickMean(x1i_sq)
val = (mean_x1i_x1ipj - (mean_x1i)**2)/\
(mean_x1i_sq - (mean_x1i)**2)
#t1 = time.time()
if j > 0 and SetDynamic:
dc_dj = np.abs( (cj[-1]-val)/(jarr[-1]-j))
if (val > 0.45) and (val < 0.55):
scale = 0.005
else:
scale = 0.01
Jstep = long(round(scale/dc_dj))
#print t1-t0,j,val, dc_dj, 0.01/(dc_dj), Jstep
if Jstep < 1:
Jstep = 1
elif Jstep > 100:
Jstep = 100
cj.append(val)
jarr.append(j)
if SetDynamic:
j += Jstep
else:
j += Nres
cj = np.array(cj)
jarr = np.array(jarr)
ChainLength = szx1
corlen_index = np.where( np.abs(cj-0.5e0) == np.min( np.abs(cj-0.5e0)) )[0]
corlen = jarr[corlen_index[0]]
efflen = long(float(ChainLength)/float(corlen))
if mkPlotsFlag:
print 'plotting acor', ftag+'.ACOR.par_'+key+'.png'
plt.plot(jarr,cj,'b.')
plt.plot([corlen,corlen],[0,1],'k--')
plt.plot([0,max(jarr)],[0.5,0.5],'k--')
plt.xlabel('j (Lag)')
plt.ylabel('C (Auto-Correlation)')
plt.title('Auto-Correlation for "'+key+'"')
plt.savefig(ftag+'.ACOR.par_'+key+'.png')
plt.clf()
print >> OutFileObject, '##'+key+'## Corr Length = '+format(corlen,'d')
print >> OutFileObject, '##'+key+'## Eff Length = '+format(efflen,'d')
chain_stats[key] = {'corlength':corlen,'efflength':efflen}
clen = []
elen = []
for key in chain_stats.keys():
clen.append(chain_stats[key]['corlength'])
elen.append(chain_stats[key]['efflength'])
if not silent:
print '## ALL ## Chain Length = '+format(ChainLength,'d')
print '## ALL ## Corr Length = '+format(max(clen),'d')
print '## ALL ## Eff Length = '+format(min(elen),'d')
print >> OutFileObject, '## ALL ## Chain Length = '+format(ChainLength,'d')
print >> OutFileObject, '## ALL ## Corr Length = '+format(max(clen),'d')
print >> OutFileObject, '## ALL ## Eff Length = '+format(min(elen),'d')
OutFileObject.close()
def covcorStats(File, FileTag, **kwargs):
""" Given MCMC parameters, this function computes
the covariance between parameters, the pearson's correlation
coefficient and spearman's rank correlation.
"""
DFlag = False
for key in kwargs:
if key.lower().startswith('derive'):
DFlag = kwargs[key]
stats = {}
OutFileObject_COV = open(FileTag+'_COV.data','w')
OutFileObject_PR = open(FileTag+'_PEARSONSR.data','w')
OutFileObject_SR = open(FileTag+'_SPEARMANSR.data','w')
#print >> OutFileObject, '# par1 | par2 | cov | pearson\'s r | spearman\'s r '
topline = '#'
passCount = 0
hdrKeys = readMCMChdr(File)
parList1 = []
parList2 = []
for i in hdrKeys.keys():
key = hdrKeys[i]
if not isNonParam(key):
parList1.append(key)
parList2.append(key)
ComboList = []
ComboDict = {}
for par1 in parList1:
covline = ''
pcorline = ''
scorline = ''
for par2 in parList2:
t0 = time.time()
#x1 = np.arange(10)
Combo = (par1,par2)
if not Combo[::-1] in ComboList:
if par1 == par2:
d1 = read1parMCMC(File,par1,derived=DFlag)
x1 = np.array(d1[par1])
x2 = x1
else:
d1 = read1parMCMC(File,par1,derived=DFlag)
x1 = np.array(d1[par1])
d2 = read1parMCMC(File,par2,derived=DFlag)
x2 = np.array(d2[par2])
ComboList.append((par1,par2))
cov = np.cov(x1,x2)
pcor = scipy.stats.pearsonr(x1,x2)
scor = scipy.stats.spearmanr(x1,x2)
ComboDict[Combo] = {'cov':cov,'pcor':pcor,'scor':scor}
else:
cov = ComboDict[Combo[::-1]]['cov']
pcor = ComboDict[Combo[::-1]]['pcor']
scor = ComboDict[Combo[::-1]]['scor']
#x2 = np.arange(10)
t1 = time.time()
print par1, par2, t1-t0
covline = covline+' '+format(cov[0][1],'+.2e')
pcorline = pcorline+' '+format(pcor[0],'+.2e')
scorline = scorline+' '+format(scor[0],'+.2e')
if passCount == 0: topline = topline+5*' '+par2
if passCount == 0:
print >> OutFileObject_COV, topline
print >> OutFileObject_COV, '#'+88*'-'
print >> OutFileObject_PR, topline
print >> OutFileObject_PR, '#'+88*'-'
print >> OutFileObject_SR, topline
print >> OutFileObject_SR, '#'+88*'-'
lenkey = len(par1)
if lenkey < 10:
fac = 10-lenkey
else:
fac = 10
print >> OutFileObject_COV, par1+fac*' '+' | '+covline
print >> OutFileObject_PR, par1+fac*' '+' | '+pcorline
print >> OutFileObject_SR, par1+fac*' '+' | '+scorline
passCount += 1
OutFileObject_COV.close()
OutFileObject_PR.close()
OutFileObject_SR.close()
def printDerived_MTQ_2011(STARTFILE,MCMCfile,DerivedFile):
""" print a file with Derived parameters from the MCMC ensemble. """
hdrKeys = readMCMChdr(MCMCfile)
ModelParams = ReadStartParams(STARTFILE)
mcmcFile = open(MCMCfile,'r')
mcmcFile = mcmcFile.readlines()
OutFileObject = open(DerivedFile,'w')
for line in mcmcFile:
derived = {}
if not line.startswith('#'):
data_line = ReadMCMCline(line,hdrKeys)
print 'step/line = ',format(long(data_line['istep']),'n')
for key in ModelParams.keys():
if ModelParams[key]['open']:
ModelParams[key]['value'] = data_line[key]
Period = computePeriod(ModelParams)
derived['Period'] = {'value':Period,'printformat':'.9f'}
RefFilt = ModelParams['RefFilt']['printformat']
# get Filter and Transit time tags
Tags = getTags(ModelParams)
# compute parameters used to compute lightcurve using the reference filter
Dref, v1ref, v2ref = MTQ_2011.MTQ_FilterParams(RefFilt,Tags,ModelParams)
u1ref, u2ref = LDC_v2u(v1ref,v2ref)
tT = ModelParams['tT']['value']
tG = ModelParams['tG']['value']
TransitRef = MTQ_2011.MTQ_getDerivedParams(Dref,tT,tG,u1ref,u2ref,Period)
#print TransitRef
derived['inc'] = {'value':TransitRef['inc']*180e0/np.pi,'printformat':'.4f'}
derived['b'] = {'value':TransitRef['b'],'printformat':'.6f'}
derived['aRs'] = {'value':TransitRef['aRs'],'printformat':'.6f'}
derived['velRs'] = {'value':TransitRef['velRs'],'printformat':'.6f'}
derived['rho_star'] = {'value':TransitRef['rho_star'],'printformat':'.6f'}
derived['RpRs'+'.'+ModelParams['RefFilt']['printformat']] = \
{'value':TransitRef['RpRs'],'printformat':'.9f'}
for key in ModelParams.keys():
if key.startswith('T0'):
split_TT = map(str,key.split('.'))
transit_tag = split_TT[1].strip()
# compute D, v1, v2 and then u1 and u2 for a given transit tag
D, v1, v2 = MTQ_2011.MTQ_FilterParams(transit_tag,Tags,ModelParams)
u1, u2 = LDC_v2u(v1,v2)
RpRs = computeRpRs(u1,u2,tT,tG,D)
filterD = filterMatchD(transit_tag,Tags,ModelParams)
#print filterD
derived['RpRs'+'.'+filterD] = {'value':RpRs,'printformat':'.9f'}
DerivedLine = ''
keylist = derived.keys()
Nkeys = len(keylist)
if data_line['istep'] == 0:
keyorder = {}
for i in range(len(keylist)):
keyorder[i] = keylist[i]
for i in range(Nkeys):
if i == 0:
DerivedLine =\
str(format(derived[keylist[i]]['value'],\
derived[keylist[i]]['printformat']))
else:
DerivedLine =\
DerivedLine+'|'+str(format(derived[keylist[i]]['value'],\
derived[keylist[i]]['printformat']))
else:
for i in range(Nkeys):
if i == 0:
DerivedLine =\
str(format(derived[keylist[i]]['value'],\
derived[keylist[i]]['printformat']))
else:
DerivedLine = DerivedLine+'|'+str(format(derived[keylist[i]]['value'],\
derived[keylist[i]]['printformat']))
DerivedLine = DerivedLine+'|'+str(format(data_line['istep'],'.0f'))+'|:'
print >> OutFileObject, DerivedLine
keyline = ''
for i in range(Nkeys):
if i == 0:
keyline = '#'+keylist[i]
else:
keyline = keyline+'|'+keylist[i]
keyline = keyline+'|istep|:'
OutFileObject.close()
#print >> OutFileObject, keyline
opencopy = open(DerivedFile,'r')
opencopy = opencopy.readlines()
os.system('rm -v %s' % (DerivedFile))
rearranged = open(DerivedFile,'w')
print >> rearranged, keyline
for iline in range(len(opencopy)):
print >> rearranged, opencopy[iline].strip('\n')
rearranged.close()