def test_targets(): test_data = \ """#name lon lat radius coord object_1 354.36 -63.26 1.0 CEL object_2 19.45 -17.46 1.0 CEL #object_3 18.94 -41.05 1.0 CEL """ with open('targets.txt','w') as f: f.write(test_data) parser = ugali.utils.parser.Parser() parser.add_coords(targets=True) args = parser.parse_args(['-t','targets.txt']) np.testing.assert_array_almost_equal(args.coords['lon'],[316.311,156.487], decimal=3) np.testing.assert_array_almost_equal(args.coords['lat'],[-51.903,-78.575], decimal=3) np.testing.assert_array_almost_equal(args.coords['radius'],[1.0,1.0], decimal=3) os.remove('targets.txt') return args
def test_targets(): test_data = \ """#name lon lat radius coord object_1 354.36 -63.26 1.0 CEL object_2 19.45 -17.46 1.0 CEL #object_3 18.94 -41.05 1.0 CEL """ with open('targets.txt', 'w') as f: f.write(test_data) parser = ugali.utils.parser.Parser() parser.add_coords(targets=True) args = parser.parse_args(['-t', 'targets.txt']) np.testing.assert_array_almost_equal(args.coords['lon'], [316.311, 156.487], decimal=3) np.testing.assert_array_almost_equal(args.coords['lat'], [-51.903, -78.575], decimal=3) np.testing.assert_array_almost_equal(args.coords['radius'], [1.0, 1.0], decimal=3) os.remove('targets.txt') return args
#!/usr/bin/env python import ugali.candidate.associate import ugali.utils.parser import numpy as np from ugali.utils.projector import gal2cel,ang2const,ang2iau #CATALOGS = ['McConnachie12','Rykoff14', 'Harris96', 'Corwen04', 'Nilson73', 'Webbink85', 'Kharchenko13', 'WEBDA14'] CATALOGS = ['McConnachie12', 'Harris96', 'Corwen04', 'Nilson73', 'Webbink85', 'Kharchenko13', 'Bica08', 'WEBDA14', 'ExtraDwarfs','ExtraClusters'] if __name__ == "__main__": import argparse description = "python script" parser = ugali.utils.parser.Parser(description=description) parser.add_coords(required=True,radius=True,targets=True) parser.add_argument('-n','--nnearest',default=1,type=int) opts = parser.parse_args() catalog = ugali.candidate.associate.SourceCatalog() for i in CATALOGS: catalog += ugali.candidate.associate.catalogFactory(i) for name,(glon,glat,radius) in zip(opts.names, opts.coords): ra,dec = gal2cel(glon,glat) iau = ang2iau(glon,glat) const = ang2const(glon,glat)[0] if radius <= 0: radius = None idx1,idx2,sep = catalog.match([glon],[glat],tol=radius,nnearest=opts.nnearest) match = catalog[idx2] if len(match) > 0:
import ugali.candidate.associate import ugali.utils.parser import numpy as np from ugali.utils.projector import gal2cel, ang2const, ang2iau #CATALOGS = ['McConnachie12','Rykoff14', 'Harris96', 'Corwen04', 'Nilson73', 'Webbink85', 'Kharchenko13', 'WEBDA14'] CATALOGS = [ 'McConnachie12', 'Harris96', 'Corwen04', 'Nilson73', 'Webbink85', 'Kharchenko13', 'Bica08', 'WEBDA14', 'ExtraDwarfs', 'ExtraClusters' ] if __name__ == "__main__": import argparse description = "python script" parser = ugali.utils.parser.Parser(description=description) parser.add_coords(required=True, radius=True, targets=True) parser.add_argument('-n', '--nnearest', default=1, type=int) opts = parser.parse_args() catalog = ugali.candidate.associate.SourceCatalog() for i in CATALOGS: catalog += ugali.candidate.associate.catalogFactory(i) for name, (glon, glat, radius) in zip(opts.names, opts.coords): ra, dec = gal2cel(glon, glat) iau = ang2iau(glon, glat) const = ang2const(glon, glat)[0] if radius <= 0: radius = None idx1, idx2, sep = catalog.match([glon], [glat], tol=radius,
clobber=True) fitsio.write(outfile, dict(DISTANCE_MODULUS=self.distance_modulus_array.astype( 'f4', copy=False)), extname='DISTANCE_MODULUS', clobber=False) if __name__ == "__main__": import ugali.utils.parser description = "Script for executing the likelihood scan." parser = ugali.utils.parser.Parser(description=description) parser.add_config() parser.add_argument('outfile', metavar='outfile.fits', help='Output fits file.') parser.add_debug() parser.add_verbose() parser.add_coords(required=True, radius=False) opts = parser.parse_args() if len(opts.coords) != 1: raise Exception('Must specify exactly one coordinate.') lon, lat, radius = opts.coords[0] grid = createGridSearch(opts.config, lon, lat) if not opts.debug: result = grid.search() grid.write(opts.outfile)
outfile, distance_modulus_array=self.distance_modulus_array, coordsys='NULL', ordering='NULL', header_dict=header_dict) ############################################################ if __name__ == "__main__": import ugali.utils.parser description = "Script for executing the likelihood scan." parser = ugali.utils.parser.Parser(description=description) parser.add_config() parser.add_argument('outfile',metavar='outfile.fits',help='Output fits file.') parser.add_debug() parser.add_verbose() parser.add_coords(required=True,radius=False) opts = parser.parse_args() if len(opts.coords) != 1: raise Exception('Must specify exactly one coordinate.') lon,lat,radius = opts.coords[0] grid = createGridSearch(opts.config,lon,lat) if not opts.debug: result = grid.search() grid.write(opts.outfile) ##print opts.coords #scan = Scan(opts.config,opts.coords) #if not opts.debug:
mcmc = MCMC(config,loglike) if samples is not None: mcmc.load_samples(samples) return mcmc if __name__ == "__main__": import ugali.utils.parser description = "Script for running MCMC followup" parser = ugali.utils.parser.Parser(description=description) parser.add_config() parser.add_debug() parser.add_verbose() #parser.add_coords(required=True) parser.add_coords() parser.add_name() parser.add_argument('--srcmdl',help='Source model file') parser.add_argument('--grid',action='store_true',help='Grid search for intial parameters') parser.add_argument('outfile',default=None,help="Output file name") opts = parser.parse_args() if opts.coords is not None and len(opts.coords) != 1: raise Exception('Must specify exactly one coordinate.') #try: # # Initialize the MPI-based pool used for parallelization. # from emcee.utils import MPIPool # pool = MPIPool(loadbalance=True) #except (ImportError,ValueError) as e:
mcmc = MCMC(config, loglike) if samples is not None: mcmc.load_samples(samples) return mcmc if __name__ == "__main__": import ugali.utils.parser description = "Script for running MCMC followup" parser = ugali.utils.parser.Parser(description=description) parser.add_config() parser.add_debug() parser.add_verbose() #parser.add_coords(required=True) parser.add_coords() parser.add_name() parser.add_argument('--srcmdl', help='Source model file') parser.add_argument('--grid', action='store_true', help='Grid search for intial parameters') parser.add_argument('outfile', default=None, help="Output file name") opts = parser.parse_args() if opts.coords is not None and len(opts.coords) != 1: raise Exception('Must specify exactly one coordinate.') #try: # # Initialize the MPI-based pool used for parallelization. # from emcee.utils import MPIPool