from experiment import *


Process('pressure', './ArN2_AirAge_Trace_Gases.csv', 
	'''
	sgloadcsv in.select=pressure,altitude
	| sgsort key=group,altitude
	| sgfieldout fields=pressure,altitude
	''')

for model in ['linear', 'poly']:
	lst=['pressure']
	for optim in ['mae', 'mse']:
		suffix="%s-%s"%(model, optim)
		lst.append('pressure-%s'%suffix)
		Process('pressure-%s model-%s'%(suffix, suffix), 'pressure',
			'''
			sgfieldmath amean:f=altitude.mean altitude=altitude-amean
			| sgfit --default=pressure,altitude
			  model=%s model.qc=${TARGETS[1]} model.nf=3
			  optim=%s optim.niter=20 
			'''%(model, optim))
	Figure('./fit-%s.png'%model, lst,
		'''
		sgcat -mode=vector ${SOURCES[1:]}
		| sgfieldmath pressure1.rename=mae pressure2.rename=mse 
		| sggraphps x=altitude y=pressure,mae,mse
		  xmin=6200 xmax=14500  ymin=-15200 ymax=47100
		  pressure.style=. pressure.label=observation
		  mae.color=FF0000 mse.color=00FF00
		  legend=topright left.label=Pressure
		''')