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