python-terminal-3d/grass_donut.py

108 lines
3.3 KiB
Python
Raw Normal View History

2024-04-14 14:38:11 +00:00
# Alexander Bass
# Created 4/13/24
# Last Edited 4/13/24
from point import Point, Rotator
from util import to_rad
import ppm
from itertools import product
from entity import PointBlobEntity
import math
DIRT = ppm.load_ppm(open("dirt.ppm", "r"))
GRASS_TOP = ppm.load_ppm(open("grass_side.ppm", "r"))
GRASS_SIDE = ppm.load_ppm(open("grass_top.ppm", "r"))
def mix_color_transparency(left, right):
for a in left:
if a > 0:
return left
return right
def mul_color(a, b):
return (a[0] * b[0], a[1] * b[1], a[2] * b[2])
def mul_color_scalar(a, b):
return (a[0] * b, a[1] * b, a[2] * b)
def cube(dirt, g_sides, g_top, face_x, face_y):
GRASS_COLOR_MUL = (0.4, 1, 0.4)
out = []
image_size = 16
point_count = 20
world_width = 16
for x, y in product(range(point_count), range(point_count)):
# Scale x and y to be from 0 to 1 to be later scaled back for any respective coordinate system
x /= point_count
y /= point_count
dirt_color = dirt[round(y * image_size)][round(x * image_size)]
top_col = g_top[round(y * image_size)][round(x * image_size)]
grass_side_color = g_sides[int(y * image_size)][int(x * image_size)]
grass_side_color = mul_color(grass_side_color, GRASS_COLOR_MUL)
top_col = mul_color(top_col, GRASS_COLOR_MUL)
side_color = mix_color_transparency(grass_side_color, dirt_color)
b_red, b_green, b_blue = dirt_color
t_red, t_green, t_blue = top_col
s_red, s_green, s_blue = side_color
x_w = x * world_width
y_w = y * world_width
w_w = world_width
if face_x == -1 or face_x == 1 and face_y == 0:
out.append(Point().set_xyz(0, x_w, y_w).set_color(s_red, s_green, s_blue))
if face_x == 1 or face_x == -1 and face_y == 0:
out.append(Point().set_xyz(w_w, x_w, y_w).set_color(s_red, s_green, s_blue))
if face_y == -1 or face_y == 1 and face_x == 0:
out.append(Point().set_xyz(x_w, 0, y_w).set_color(s_red, s_green, s_blue))
if face_y == 1 or face_y == -1 and face_x == 0:
out.append(Point().set_xyz(x_w, w_w, y_w).set_color(s_red, s_green, s_blue))
# Top and bottom faces
out.append(Point().set_xyz(x_w, y_w, w_w).set_color(b_red, b_green, b_blue))
out.append(Point().set_xyz(x_w, y_w, 0).set_color(t_red, t_green, t_blue))
for p in out:
p.translate_x(face_x * world_width - world_width / 2)
p.translate_y(face_y * world_width - world_width / 2)
p.translate_z(-world_width / 2)
p.rotate_z(to_rad(90))
p.rotate_y(45 * 3)
pass
return out
class DonutEntity(PointBlobEntity):
def __init__(self):
super().__init__()
points = []
for x, y in product((-1, 0, 1), (-1, 0, 1)):
if x == y == 0:
continue
points.extend(cube(DIRT, GRASS_TOP, GRASS_SIDE, x, y))
self.set_points(points)
def update(self, t):
tf = Rotator()
tf.rotate_y(math.cos(t / 25) * 0.5)
# tf.rotate_z(0.02)
tf.rotate_x(math.sin(t / 20) * 0.5 + 0.02)
tf.build()
for p in self.points:
p.rotate(tf)
pass
def shift(self, x, y, z):
for p in self.points:
p.translate(x, y, z)