phiflow2blender/tutorial/TUM.py at main · aristotelis-economides/phiflow2blender · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
##############################################
# Example of a 3D fluid simulation with phiflow - https://github.com/tum-pbs/PhiFlow
# It creates a 3D TUM logo.
# Author: Aristotelis
# Date: 02 Aug 2021
#############################################

from phi.flow import *
import time

RES = 64
RADIUS = 2
T_STEP = 1
DOMAIN = dict(x=2*RES, y=RES, z=3*RES)
NUM_FRAMES = 500

#TODO: Change this path to where you want to save the simulation data
SCENE_PATH = "/home/teemps/FILES/UNI/TUM/HiWi-Thurey/TUM_Logo/tum-logo/tutorial/TUM64"
scene = Scene(SCENE_PATH)


OBSTACLE_GEOMETRIES = [
    Box[0:20*  RES           , 0:RES, 0.00     :0.20*  RES],
    Box[0:20*  RES           , 0:RES, 2.80* RES:3.00*  RES],
    Box[0:1.45*RES           , 0:RES, 0.20* RES:0.43*  RES],
    Box[0:0.30*RES           , 0:RES, 0.66* RES:2.56*  RES],
    Box[1.70*  RES:2*     RES, 0:RES, 0.20* RES:2.80*  RES],
    Box[0.30*  RES:1.45*  RES, 0:RES, 0.66* RES:0.90*  RES],
    Box[0.30*  RES:1.45*  RES, 0:RES, 1.70* RES:1.97*  RES],
    Box[0.30*  RES:1.45*  RES, 0:RES, 2.30* RES:2.55*  RES],
    Box[0.55*  RES:1.7*   RES, 0:RES, 1.15* RES:1.42*  RES]
]
OBSTACLE = Obstacle(union(OBSTACLE_GEOMETRIES))
OBSTACLE_MASK = HardGeometryMask(OBSTACLE.geometry) >> CenteredGrid(0, extrapolation.BOUNDARY, **DOMAIN)


INFLOW = CenteredGrid(Box[0.31*RES:0.45*RES, 0.35*RES:0.75*RES, 0.5* RES:0.65*RES ], extrapolation.BOUNDARY, **DOMAIN) * 0.2 + \
         CenteredGrid(Box[0.31*RES:0.45*RES, 0.35*RES:0.75*RES, 1.1* RES:1.5* RES ], extrapolation.BOUNDARY, **DOMAIN) * 0.2 + \
         CenteredGrid(Box[0.31*RES:0.45*RES, 0.35*RES:0.75*RES, 2.05*RES:2.2* RES ], extrapolation.BOUNDARY, **DOMAIN) * 0.2 + \
         CenteredGrid(Box[0.31*RES:0.45*RES, 0.35*RES:0.75*RES, 2.6* RES:2.75*RES ], extrapolation.BOUNDARY, **DOMAIN) * 0.2
velocity = StaggeredGrid(Noise(scale=1), extrapolation.ZERO, **DOMAIN)  # or use CenteredGrid
smoke = CenteredGrid(0, extrapolation.BOUNDARY, **DOMAIN)

total_time = 0

for step in range(NUM_FRAMES):

    start = time.time()
    smoke = advect.mac_cormack(smoke, velocity, dt=T_STEP) + INFLOW
    buoyancy_force = smoke * (0.1, 0, 0) >> velocity  # resamples smoke to velocity sample points
    velocity = advect.semi_lagrangian(velocity, velocity, 1) + buoyancy_force
    with math.SolveTape() as solves:
        velocity, pressure = fluid.make_incompressible(velocity, (OBSTACLE,), Solve('CG-adaptive', 1e-5, 0, max_iterations=1500, x0=None))
    scene.write({'smoke': smoke, 'velocity': velocity}, frame = step )
    end = time.time()

    total_time += end-start
    print(f'Step {step}:\t {end-start}seconds \tTotal time:{total_time}')