# End performJob-function
if __name__ == '__main__':
#main()
#%%
# Load lags from file
fileName = '/Users/uqnsomme/Documents/Obelix/Asymm/Chains/MyChainSkewGauss0.dat'
data = np.loadtxt(fileName)
lags = data[:, 2]
# Create histogram of lags
plt.figure(figsize=(6,10), dpi=500)
n, bins, patches = plt.hist(lags, bins=100)
plt.xlabel('Time lag', fontsize='large')
plt.ylabel('Number of instances', fontsize='large')
maxheight = 1000 * np.ceil(np.max(n)*0.001)
ax = plt.axes()
ax.xaxis.set_major_locator(MultipleLocator(200))
ax.xaxis.set_minor_locator(MultipleLocator(100))
ax.yaxis.set_major_locator(MultipleLocator(1000))
ax.yaxis.set_minor_locator(MultipleLocator(500))
ax.set_aspect(np.max(bins)/maxheight)
#%%
conv = Convergence()
conv.plotGelmanRubin(fileName, 10, 50)
# Initialisation
AGN = createOzDESAGN(QSOtemplate)
stack = Stacking()
stats = Stats()
conv = Convergence()
plotOzDESAGN(QSOtemplate, plotIndividuals=False, binends=binends, name='MgII')
# Create AGN
AGN.generateAGN(limits, True, number, resDir, fName)
output = AGN.generateLightCurveParameters(section, resDir+fName)
AGN.printLightCurveParametersToFile(output, resDir+params)
lag = printLagValueInParameterBin(resDir+params, limits, emLine=emLine)
# Do the Javelin magic
runJavelinOnFakeData(resDir+params, limits, number, resDir, string, emLine=emLine)
# Stack and analyse
LLH, lags, lagMaxes = stack.stackData(resDir, string, number, resDir+params, limits, meanTrueLag=lag, binsize=10,
polydeg=20, plotLags=False, plotLLHs=False, plotStacked=True, indivStat=False)
LLHmax, LLHmin, LLHplus = stats.getLikelihoodEstimates(lags, LLH, printResults=True)
med1, med1min, med1max = stats.getMedianEstimates(LLH, lags, printResults=True)
##med2, med2min, med2max = stats.getMedianEstimates(lagMaxes, printResults=True)
# Analysis of results
plotAChain(chainDir+'MyChain'+string+'0.dat')
R = conv.GelmanRubin(chainDir+'MyChain'+string+'0.dat', 10, 10, False)
print 'Gelman-Rubin R =', R
conv.plotGelmanRubin(chainDir+'MyChain'+string+'0.dat', 10, 10)
