def theta(self): if self.__theta is None: self.__theta = np.zeros_like(self.r) oldsettings = np.seterr(all='ignore') self.__theta = np.squeeze(arcsin(self.z / self.r)) * self._unit_coeff misc.revert_seterr(oldsettings) self.__theta[self.r == 0.] = 0. return self.__theta
def _poly(self, params): tmp = 0. max_r = np.max(self.r) for i, param in enumerate(params): tmp += param * (self.r / max_r)**i oldsettings = np.seterr(all='ignore') tmp = tmp / self.r misc.revert_seterr(oldsettings) tt = (self.r == 0.) tmp[tt] = 0 vel_x = tmp * self.x vel_y = tmp * self.y vel_z = tmp * self.z return vel_x, vel_y, vel_z
def _poly(self, params): tmp = 0. max_r = np.max(self.r) for i, param in enumerate(params): tmp += param * (self.r/max_r)**i oldsettings = np.seterr(all='ignore') tmp = tmp / self.r misc.revert_seterr(oldsettings) tt = (self.r == 0.) tmp[tt] = 0 vel_x = tmp * self.x vel_y = tmp * self.y vel_z = tmp * self.z return vel_x, vel_y, vel_z