os.environ['DISPLAY'] except KeyError as Err: warnings.warn('No display environment! Using matplotlib backend "Agg"') import matplotlib matplotlib.use('Agg') try: import carmcmc as cmcmc except ImportError: carma_pack = False else: carma_pack = True fhgt = 10 fwid = 16 set_plot_params(useTex=True) parser = argparse.ArgumentParser() parser.add_argument('-pwd', '--pwd', type=str, default=os.path.join(os.environ['KALI'], 'examples/data'), help=r'Path to working directory') parser.add_argument('-n', '--name', type=str, default='rand', help=r'SDSS ID') parser.add_argument('-b', '--band', type=str, default='g', help=r'SDSS bandpass') parser.add_argument('-libcarmaChain', '--lC', type=str, default='libcarmaChain', help=r'kali.carma Chain Filename') parser.add_argument('-cmcmcChain', '--cC', type=str, default='cmcmcChain', help=r'carma_pack Chain Filename') parser.add_argument('-nsteps', '--nsteps', type=int, default=250, help=r'Number of steps per walker') parser.add_argument('-nwalkers', '--nwalkers', type=int, default=25*psutil.cpu_count(logical=True), help=r'Number of walkers') parser.add_argument('-p', '--p', type=int, default=2, help=r'C-AR order') parser.add_argument('-q', '--q', type=int, default=1, help=r'C-MA order') parser.add_argument('--plot', dest='plot', action='store_true', help=r'Show plot?')
import kali.carma from kali.util.mpl_settings import set_plot_params import kali.util.mcmcviz as mcmcviz import kali.util.triangle as triangle except ImportError as e: print 'kali not found! Try setting up kali if you have it installed. Unable to proceed!!' print e sys.exit(0) b_ = dict(u=1.4e-10, g=0.9e-10, r=1.2e-10, i=1.8e-10, z=7.4e-10) # band softening parameter f0 = 3631.0 # Jy fhgt = 10 fwid = 16 set_plot_params(useTex=True) @np.vectorize def luptitude_to_flux(mag, err, band): # Converts a Luptitude to an SDSS flux flux = math.sinh(math.log(10.0) / -2.5 * mag - math.log(b_[band])) * 2 * b_[band] * f0 error = err * math.log(10) / 2.5 * 2 * b_[band] * math.sqrt( 1 + (flux / (2 * b_[band] * f0))**2) * f0 return flux, error def time_to_restFrame(time, z): # Converts a cadence to the rest frame t = np.zeros(time.shape[0])