simulate and animate a soda bottle opening in python

main.py
import numpy as np
import matplotlib.pyplot as plt
from mayavi import mlab

# Define the initial shape of the bottle
bottle_height = 10
bottle_radius = 2
bottle_length = 2 * bottle_radius

bottle = np.zeros((bottle_height, bottle_length, bottle_length), dtype=np.float32)

# Define the opening function
def opening_function(x, y, z):
    return np.exp(-((x/bottle_radius)**2 + (y/bottle_radius)**2 + (z/bottle_height)**2))

# Define the simulation parameters
opening_radius = 0.5 * bottle_radius
opening_speed = 0.1
animation_frames = 50

# Simulate the opening process and animate it
for i in range(animation_frames):
    opening = opening_radius * opening_function(
        bottle_length * (i / animation_frames - 0.5),
        bottle_length * (i / animation_frames - 0.5),
        bottle_height * (i / animation_frames))
    bottle[:, :, bottle_radius:] = np.maximum(bottle[:, :, bottle_radius:], opening)
    mlab.clf()
    mlab.contour3d(bottle, contours=[0.5], opacity=0.5, colormap='cool')
    mlab.axes()
    mlab.show()

1030 chars
33 lines