replace >> with at() method

This commit is contained in:
topsy404 2021-09-16 15:51:57 +08:00 committed by GitHub
parent 6b82e5fd50
commit aa9658ea2b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -117,7 +117,7 @@
"source": [
"def step(velocity, smoke, pressure, dt=1.0, buoyancy_factor=1.0):\n",
" smoke = advect.semi_lagrangian(smoke, velocity, dt) + INFLOW\n",
" buoyancy_force = smoke * (0, buoyancy_factor) >> velocity # resamples smoke to velocity sample points\n",
" buoyancy_force = (smoke * (0, buoyancy_factor)).at(velocity) # resamples smoke to velocity sample points\n",
" velocity = advect.semi_lagrangian(velocity, velocity, dt) + dt * buoyancy_force\n",
" velocity = diffuse.explicit(velocity, NU, dt)\n",
" velocity, pressure = fluid.make_incompressible(velocity)\n",
@ -527,4 +527,4 @@
},
"nbformat": 4,
"nbformat_minor": 1
}
}