Ejemplo n.º 1
0
# filename is provided by command line
print 'Reading your commands from command line'
mode, store = rd.read_args (sys.argv)

print 'Welcome to cobra, the plotting utility for the radiative transfer code, Python.'
rd.setpars() #this just sets some standard parameters, e.g. tex

# if user asked for help we print help message and exit
if mode.help:
    sub.help_me_spec()

filename = store.filename

# read the spec file
spectrum = rd.read_spec_file (filename)



# we now want to smooth our spectrum
smooth = store.ibin
sub.smooth_spectrum(spectrum, smooth)


# now plot up
fig=plt.figure(figsize=(8.3,11.7),dpi=80)
fig.suptitle(fname,fontsize=24,fontweight='bold')
fig.subplots_adjust(hspace=0.3,wspace=0.2)

ax.plot(spectrum.wavelength, spectrum.spec[0])
plt.show()
Ejemplo n.º 2
0



import matplotlib.pyplot as plt
import pylab
import numpy as np
import classes as cls
import disk as d
import sys, os
import read_output as rd
import bal
from constants import *
from math import fabs

spec = rd.read_spec_file("run110e")

spec_array = bal.flambda_to_fnu (spec.spec[4], spec.freq, spec.wavelength)

BI = bal.BALnicity( C/(1549.0*ANGSTROM), spec.freq, spec_array)

print BI



ls_file = sys.argv[1]
rd.setpars()
filenames = np.loadtxt(ls_file, dtype='string')

bheight = []
bhseven =[]
Ejemplo n.º 3
0
def plot_spec (filename, lmin, lmax, smooth = 1,  nobs = 0, use = [], \
	           savename = "fig", yscale = "linear", xscale = "linear" , \
	           sources = False, Fnu = False):

	''' 
	Function for plotting a spec file outputted from the radiative transfer code PYTHONRT

	:INPUT:
		filename 			string
							name of file

		lmin, lmax 			float
							wavelength range in ANGSTROMS

		nobs 				int
							number of observes

		smooth				int
							smoothing factor

		use					array
							which observations to use

		savename			string

		yscale, xscale 		string 
							lin or log scale 

		sources 			Bool
		 					Plot sources or not 

		Fnu 				Bool
		 					Is it an Fnu plot?


	:OUTPUT:
		Creates plot and opens in preview 
	'''


	# default savename is filename
	if savename == "fig":
		savename = filename + ".png"


	# create spec class from spec file
	spec = rd.read_spec_file(filename)

	if nobs == 0:
		nobs = len(spec.spec)

	# strip filenames of funny characters that TeX complains about
	savename = strip("_", savename)
	filename = strip("_", filename)


	# default argument is to plot all observations
	if len(use) == 0:
		use = np.arange(nobs)

	nuse = int(len(use))

	# work out the dimensions of the plot
	if nuse < 3:
		ny = nuse
		nx = 1
	else:
		nx = 2
		ny = (len(use) + 1) / 2


	# do we want to smooth? if so, do it!
	if smooth > 1:
		for i in use:
			sub.smooth_spectrum( spec, smooth )


	# now create figure
	fig=plt.figure(figsize=(8.3,11.7),dpi=80)
	fig.suptitle(filename,fontsize=24,fontweight='bold')
	fig.subplots_adjust(hspace=0.3,wspace=0.2)


	for i in range(nuse):

		ax = fig.add_subplot( ny, nx, i)

		if Fnu:
			ax.plot(spec.freq, spec.spec[use[i]])
		else:
			ax.plot(spec.wavelength, spec.spec[use[i]])

		ax.set_yscale(yscale)
		ax.set_xscale(xscale)

		plt.xlim(lmin, lmax)


	plt.savefig(savename)

	command = "open -a preview %s" % savename

	os.system(command)

	if sources:
		fig=plt.figure(figsize=(8.3,11.7),dpi=80)
		fig.suptitle(filename,fontsize=24,fontweight='bold')
		fig.subplots_adjust(hspace=0.3,wspace=0.2)


	return 0