def v_disp(self):
"""SPH-smoothed local velocity dispersion"""
import sph
sph.build_tree(self)
nsmooth = config['sph']['smooth-particles']
self['rho']
logger.info(
'Calculating velocity dispersion with %d nearest neighbours' % nsmooth)
sm = array.SimArray(np.empty(len(self['vel']),dtype=self['vel'].dtype),
self['vel'].units)
self.kdtree.set_array_ref('rho',self['rho'])
self.kdtree.set_array_ref('smooth',self['smooth'])
self.kdtree.set_array_ref('mass',self['mass'])
self.kdtree.set_array_ref('qty',self['vel'])
self.kdtree.set_array_ref('qty_sm',sm)
start = time.time()
self.kdtree.populate('qty_disp',nsmooth)
end = time.time()
logger.info('Velocity dispersion done in %5.3g s' % (end - start))
return sm
def v_mean(self):
"""SPH-smoothed mean velocity"""
import sph
sph.build_tree(self)
nsmooth = config['sph']['smooth-particles']
logger.info(
'Calculating mean velocity with %d nearest neighbours' % nsmooth)
sm = array.SimArray(np.empty_like(self['vel']),
self['vel'].units)
self.kdtree.set_array_ref('rho',self['rho'])
self.kdtree.set_array_ref('smooth',self['smooth'])
self.kdtree.set_array_ref('mass',self['mass'])
self.kdtree.set_array_ref('qty',self['vel'])
self.kdtree.set_array_ref('qty_sm',sm)
start = time.time()
self.kdtree.populate('qty_mean',nsmooth)
end = time.time()
logger.info('Mean velocity done in %5.3g s' % (end - start))
return sm
def v_disp(self):
"""SPH-smoothed local velocity dispersion"""
import sph
sph.build_tree(self)
nsmooth = config['sph']['smooth-particles']
self['rho']
logger.info('Calculating velocity dispersion with %d nearest neighbours' %
nsmooth)
sm = array.SimArray(np.empty(len(self['vel']), dtype=self['vel'].dtype),
self['vel'].units)
self.kdtree.set_array_ref('rho', self['rho'])
self.kdtree.set_array_ref('smooth', self['smooth'])
self.kdtree.set_array_ref('mass', self['mass'])
self.kdtree.set_array_ref('qty', self['vel'])
self.kdtree.set_array_ref('qty_sm', sm)
start = time.time()
self.kdtree.populate('qty_disp', nsmooth)
end = time.time()
logger.info('Velocity dispersion done in %5.3g s' % (end - start))
return sm
def v_mean(self):
"""SPH-smoothed mean velocity"""
import sph
sph.build_tree(self)
nsmooth = config['sph']['smooth-particles']
logger.info('Calculating mean velocity with %d nearest neighbours' %
nsmooth)
sm = array.SimArray(np.empty_like(self['vel']), self['vel'].units)
self.kdtree.set_array_ref('rho', self['rho'])
self.kdtree.set_array_ref('smooth', self['smooth'])
self.kdtree.set_array_ref('mass', self['mass'])
self.kdtree.set_array_ref('qty', self['vel'])
self.kdtree.set_array_ref('qty_sm', sm)
start = time.time()
self.kdtree.populate('qty_mean', nsmooth)
end = time.time()
logger.info('Mean velocity done in %5.3g s' % (end - start))
return sm
def v_disp(self):
import sph
sph.build_tree(self)
nsmooth = config['sph']['smooth-particles']
self['rho']
if config['verbose']: print 'Calculating velocity dispersion with %d nearest neighbours' % nsmooth
sm = array.SimArray(np.empty(len(self['pos'])), self['vel'].units)
self.kdtree.populate(sm, 'v_disp', nn=nsmooth, smooth=self['smooth'], rho=self['rho'])
if config['verbose']: print 'Velocity dispersion done.'
return sm
def v_mean(self):
import sph
sph.build_tree(self)
nsmooth = config['sph']['smooth-particles']
if config['verbose']: print 'Calculating mean velocity with %d nearest neighbours' % nsmooth
sm = array.SimArray(np.empty((len(self['pos']),3)), self['vel'])
self.kdtree.populate(sm, 'v_mean', nn=nsmooth, smooth=self['smooth'], rho=self['rho'])
if config['verbose']: print 'Mean velocity done.'
return sm
