Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def _output_order_gadget(all_keys):
out = []
out_dregs = copy.copy(all_keys)
for X in map(str.strip, config_parser.get('gadget-default-output', 'field-ordering').split(',')):
if X in out_dregs:
del out_dregs[out_dregs.index(X)]
out.append(X)
return out + out_dregs
ramses_hydro_header = np.dtype([('ncpu', 'i4'), ('nvarh', 'i4'), ('ndim', 'i4'), ('nlevelmax', 'i4'),
('nboundary', 'i4'), ('gamma', 'f8')])
ramses_grav_header = np.dtype([('ncpu', 'i4'), ('ndim', 'i4'), ('nlevelmax', 'i4'),
('nboundary', 'i4')])
ramses_rt_header = np.dtype([('ncpu','i4'), ('nrtvar', 'i4'), ('ndim', 'i4'),
('nlevelmax', 'i4'), ('nboundary', 'i4'), ('gamma', 'f8')])
particle_blocks = map(
str.strip, config_parser.get('ramses', "particle-blocks").split(","))
particle_format = map(
str.strip, config_parser.get('ramses', "particle-format").split(","))
hydro_blocks = map(
str.strip, config_parser.get('ramses', "hydro-blocks").split(","))
grav_blocks = map(
str.strip, config_parser.get('ramses', "gravity-blocks").split(","))
rt_blocks = map(
str.strip, config_parser.get('ramses', 'rt-blocks', raw=True).split(",")
)
particle_distinguisher = map(
str.strip, config_parser.get('ramses', 'particle-distinguisher').split(","))
positive_typemap = map(lambda x: family.get_family(str.strip(x)),
config_parser.get('ramses', 'type-mapping-positive').split(","))
negative_typemap = map(lambda x: family.get_family(str.strip(x)),
config_parser.get('ramses', 'type-mapping-negative').split(","))
particle_blocks = map(
str.strip, config_parser.get('ramses', "particle-blocks").split(","))
particle_format = map(
str.strip, config_parser.get('ramses', "particle-format").split(","))
hydro_blocks = map(
str.strip, config_parser.get('ramses', "hydro-blocks").split(","))
grav_blocks = map(
str.strip, config_parser.get('ramses', "gravity-blocks").split(","))
rt_blocks = map(
str.strip, config_parser.get('ramses', 'rt-blocks', raw=True).split(",")
)
particle_distinguisher = map(
str.strip, config_parser.get('ramses', 'particle-distinguisher').split(","))
positive_typemap = map(lambda x: family.get_family(str.strip(x)),
config_parser.get('ramses', 'type-mapping-positive').split(","))
negative_typemap = map(lambda x: family.get_family(str.strip(x)),
config_parser.get('ramses', 'type-mapping-negative').split(","))
class RamsesSnap(SimSnap):
reader_pool = None
def __init__(self, dirname, **kwargs):
"""Initialize a RamsesSnap. Extra kwargs supported:
*cpus* : a list of the CPU IDs to load. If not set, load all CPU's data.
*maxlevel* : the maximum refinement level to load. If not set, the deepest level is loaded.
if groupfinder == 'None':
for directory in os.environ["PATH"].split(os.pathsep):
ahfs = glob.glob(os.path.join(directory, "rockstar-galaxies"))
for iahf, ahf in enumerate(ahfs):
if ((len(ahfs) > 1) & (iahf != len(ahfs)-1) &
(os.path.basename(ahf) == 'rockstar-galaxies')):
continue
else:
groupfinder = ahf
break
if not os.path.exists(groupfinder):
raise RuntimeError("Path to Rockstar (%s) is invalid" % groupfinder)
f = open('quickstart.cfg', 'w')
print >>f, config_parser.get('RockstarCatalogue', 'Config', vars={
'format': fileformat,
'partmass': sim.d['mass'].in_units('Msol h^-1',**sim.conversion_context()).min(),
'expfac': sim.properties['a'],
'hub': sim.properties['h'],
'omega0': sim.properties['omegaM0'],
'lambda0': sim.properties['omegaL0'],
'boxsize': sim['pos'].units.ratio('Mpc a h^-1', **sim.conversion_context()),
'vunit': sim['vel'].units.ratio('km s^-1 a', **sim.conversion_context()),
'munit': sim['mass'].units.ratio('Msol h^-1', **sim.conversion_context()),
'softening': sim.s['eps'].in_units('Mpc a h^-1', **sim.conversion_context()).min()
})
f.close()
if (not os.path.exists(sim._filename)):
os.system("gunzip "+sim._filename+".gz")
def __init_ignorable_keys(self):
self.fof_ignore = map(str.strip,config_parser.get("SubfindHDF","FoF-ignore").split(","))
self.sub_ignore = map(str.strip,config_parser.get("SubfindHDF","Sub-ignore").split(","))
for t in self.base._family_to_group_map.values():
# Don't add SubFind particles ever as this list is actually spherical overdensity
self.sub_ignore.append(t[0])
self.fof_ignore.append(t[0])
def _output_order_gadget(all_keys):
out = []
out_dregs = copy.copy(all_keys)
for X in map(str.strip, config_parser.get('gadget-default-output', 'field-ordering').split(',')):
if X in out_dregs:
del out_dregs[out_dregs.index(X)]
out.append(X)
return out + out_dregs
'i4'), ('mstar', 'f8'),
('mstar_lost', 'f8'), ('nsink', 'i4')])
ramses_amr_header = np.dtype([('ncpu', 'i4'), ('ndim', 'i4'), ('ng', 'i4', (3,)),
('nlevelmax', 'i4'), ('ngridmax',
'i4'), ('nboundary', 'i4'),
('ngrid', 'i4'), ('boxlen', 'f8')])
ramses_hydro_header = np.dtype([('ncpu', 'i4'), ('nvarh', 'i4'), ('ndim', 'i4'), ('nlevelmax', 'i4'),
('nboundary', 'i4'), ('gamma', 'f8')])
ramses_grav_header = np.dtype([('ncpu', 'i4'), ('ndim', 'i4'), ('nlevelmax', 'i4'),
('nboundary', 'i4')])
particle_blocks = map(
str.strip, config_parser.get('ramses', "particle-blocks").split(","))
particle_format = map(
str.strip, config_parser.get('ramses', "particle-format").split(","))
hydro_blocks = map(
str.strip, config_parser.get('ramses', "hydro-blocks").split(","))
grav_blocks = map(
str.strip, config_parser.get('ramses', "gravity-blocks").split(","))
particle_distinguisher = map(
str.strip, config_parser.get('ramses', 'particle-distinguisher').split(","))
class RamsesSnap(SimSnap):
reader_pool = None
def __init__(self, dirname, **kwargs):