def load_src(name, fpath):
    import os, imp
    return imp.load_source(name, os.path.join(os.path.dirname(__file__), fpath))

load_src("torch", "../torchpack/torch.py")
load_src("hgspy", "../torchpack/hgspy.py")

import torch
import hgspy

DPI = 300
figformat = 'png'
plot_size = 5
fontsize = 16
torch.set_font_sizes(fontsize=16)

inputfile1 = "data/galsim/starpop-hm-a.txt"

data1 = np.genfromtxt(inputfile1, skip_header=1)

def plot_scatter(ax, dat, colormap):
    x = dat[:,12]
    y = dat[:,13]

    xy = np.vstack([x,y])
    z = scipy.stats.gaussian_kde(xy)(xy)

    idx = z.argsort()
    x, y, z = x[idx], y[idx], z[idx]

def load_src(name, fpath):
    import os, imp
    return imp.load_source(name, os.path.join(os.path.dirname(__file__), fpath))

load_src("torch", "../torchpack/torch.py")
load_src("hgspy", "../torchpack/hgspy.py")

import torch
import hgspy

DPI = 300
figformat = 'png'
plot_size = 5
fontsize = 16
outputfile_qlyc = "mass-vs-qlyc.png"
torch.set_font_sizes(fontsize)

data = np.genfromtxt("refdata/zams.txt", skip_header=1)

### Data set up.

mass = data[:,0]
qlyc = data[:,6]

### Plotting.
plotter = torch.Plotter(1, 1, plot_size, figformat, DPI)

###	Axes.
grid = plotter.axes1D((1,1), aspect_ratio=0.75)
grid[0].set_xlabel(plotter.format_label(torch.VarType('M_\star', units='M_{\odot}')))
grid[0].set_ylabel(plotter.format_label(torch.VarType('Q_\\mathrm{Lyc}', units='s^{-1}', isLog10=True)))

def plot():
	import warnings
	import sys
	def load_src(name, fpath):
		import os, imp
		return imp.load_source(name, os.path.join(os.path.dirname(__file__), fpath))

	load_src("torch", "../torchpack/torch.py")
	load_src("hgspy", "../torchpack/hgspy.py")

	import torch
	import hgspy

	DPI = 300
	figformat = 'png'
	plot_size = 5
	fontsize = 16
	plot_type = "ssw" # ["if", "hii", "ssw"]
	outputfile_qlyc = "henney-" + plot_type + ".png"

	torch.set_font_sizes(fontsize)

	### Data
	lmp = []
	lmn = []
	lhp = []
	lhn = []
	lcie = []
	lpdr = []
	ltot = []

	if plot_type == "ssw":
		nh = 1
	elif plot_type == "hii":
		nh = 1000
	else:
		nh = 10000

	hii = 1.0
	if plot_type == "if":
		hii = 0.5
	N = 1001
	minlogT = 4.0
	maxlogT = 8.0
	if plot_type == "if" or plot_type == "hii":
		minlogT = 3.8
		maxlogT = 4.0
	isLog = True

	nhii = hii * nh
	nhi = (1.0 - hii) * nh

	if not isLog:
		tem = np.linspace(10.0**minlogT, 10.0**maxlogT, N, endpoint=True)
	else:
		tem = np.linspace(minlogT, maxlogT, N, endpoint=True)

	for i in range(len(tem)):
		itemp = tem[i]
		if isLog:
			itemp = math.pow(10.0, itemp)
		lmp.append(coolCollExcIonMetals(nhii, itemp))
		lmn.append(coolCollExcNeuMetals(nhii, nhi, itemp))
		lhp.append(coolFreeFree(nhii, itemp))
		lhn.append(coolCollExcNeuHydrogen(nhii, nhi, itemp))
		lcie.append(coolCIE(nhii, itemp))
		lpdr.append(coolPDR(nhi, itemp))
		ltot.append(lmp[i] + lmn[i] + lhp[i] + lhn[i] + lcie[i] + lpdr[i])

	### Plotting.
	plotter = torch.Plotter(1, 1, plot_size, figformat, DPI)

	###	Axes.
	grid = plotter.axes1D((1,1), aspect_ratio=0.75)
	grid[0].set_xlabel(plotter.format_label(torch.VarType('T', units='K', isLog10=True)))
	grid[0].set_ylabel(plotter.format_label(torch.VarType('C', units='erg\\:s^{-1}\\:cm^{-3}', isLog10=False)))

	### Plot.
	if plot_type == "if":
		grid[0].plot(tem, lmn, label=r"$C_\mathrm{M^0}$", color='m')
		grid[0].plot(tem, lhn, label=r"$C_\mathrm{H^0}$", color='y')
		grid[0].plot(tem, lmp, label=r"$C_\mathrm{M^+}$", color='g')
		grid[0].plot(tem, lhp, label=r"$C_\mathrm{H^+}$", color='r')
		grid[0].plot(tem, ltot, label=r"$C_\mathrm{tot}$", color='k')
	elif plot_type == "hii":
		grid[0].plot(tem, lmp, label=r"$C_\mathrm{M^+}$", color='g')
		grid[0].plot(tem, lhp, label=r"$C_\mathrm{H^+}$", color='r')
		grid[0].plot(tem, ltot, label=r"$C_\mathrm{tot}$", color='k')
	elif plot_type == "ssw":
		grid[0].plot(tem, lmp, label=r"$C_\mathrm{M^+}$", color='g')
		grid[0].plot(tem, lhp, label=r"$C_\mathrm{H^+}$", color='r')
		grid[0].plot(tem, lcie, label=r"$C_\mathrm{CIE}$", color='b')
		grid[0].plot(tem, ltot, label=r"$C_\mathrm{tot}$", color='k')

	grid[0].set_xlim([minlogT, maxlogT])
	if plot_type == "ssw":
		grid[0].legend(loc='upper right')
	else:
		grid[0].legend(loc='upper left')

#.........这里部分代码省略.........