计算机图形学-三维投影变换

三维投影变换

import tkinter as tk
import random as rd
import time
import numpy as np
import math
#########################################################################
def rotateX3d(point,var):
    transMatrix=[[1,0,0,0],[0,math.cos(var),-math.sin(var),0],[0,math.sin(var),math.cos(var),0],[0,0,0,1]]
    point = np.matmul(np.array(point, dtype='float32'), transMatrix)
    return point.tolist()
def rotateY3d(point,var):
    transMatrix=[[math.cos(var),0,math.sin(var),0],[0,1,0,0],[-math.sin(var),0,math.cos(var),0],[0,0,0,1]]
    point = np.matmul(np.array(point, dtype='float32'), transMatrix)
    return point.tolist()
def rotateZ3d(point,var):
    transMatrix=[[math.cos(var),-math.sin(var),0,0],[math.sin(var),math.cos(var),0,0],[0,0,1,0],[0,0,0,1]]
    point = np.matmul(np.array(point, dtype='float32'), transMatrix)
    return point.tolist()

def projectPoints(points):
    # 一点透视参数
    L = 200  # 平移位置L, M, N
    M = 200
    N = 200
    D = 800  # 视距
    # 一点透视变换矩阵
    transMatrix = np.array(
        [[1, 0, 0, 0], [0, 1, 0, 0], [0, 0, 0, 1 / D], [L, M, 0, 1 + (N / D)]],
        dtype='float32')
    onepoint = [] 
    # 一点透视：将三维坐标映射到二维
    for e in range(0, len(points)):
        tmp = np.matmul(np.array(points[e], dtype='float32'), transMatrix)
        tmp = tmp / tmp[3]  # 齐次化
        onepoint.append(tmp.tolist()[:2])
    return onepoint

#########################################################################

window = tk.Tk()

canvas = tk.Canvas(window, width=800, height=800, background='white')
canvas.pack()

points = [[0, 0, 0, 1], [200, 0, 0, 1], [200, 200, 0, 1], [0, 200, 0, 1],
          [0, 0, 200, 1], [200, 0, 200, 1], [200, 200, 200, 1], [0, 200, 200, 1]]


points_2d = projectPoints(points)
lineList=[]
lineList.append(canvas.create_line(points_2d[0][:2], points_2d[1][:2]))
lineList.append(canvas.create_line(points_2d[1][:2], points_2d[2][:2]))
lineList.append(canvas.create_line(points_2d[2][:2], points_2d[3][:2]))
lineList.append(canvas.create_line(points_2d[3][:2], points_2d[0][:2]))

lineList.append(canvas.create_line(points_2d[4][:2], points_2d[5][:2]))
lineList.append(canvas.create_line(points_2d[5][:2], points_2d[6][:2]))
lineList.append(canvas.create_line(points_2d[6][:2], points_2d[7][:2]))
lineList.append(canvas.create_line(points_2d[7][:2], points_2d[4][:2]))

lineList.append(canvas.create_line(points_2d[0][:2], points_2d[4][:2]))
lineList.append(canvas.create_line(points_2d[1][:2], points_2d[5][:2]))
lineList.append(canvas.create_line(points_2d[2][:2], points_2d[6][:2]))
lineList.append(canvas.create_line(points_2d[3][:2], points_2d[7][:2]))


# pointIdxs=[]
per_deg=math.pi/180

var=0
while True:
    for i in range(len(points)):
        points[i]=rotateX3d(points[i], per_deg*5)
        points[i]=rotateY3d(points[i], per_deg*5)
        points[i]=rotateZ3d(points[i], per_deg*5)
    points_2d = projectPoints(points)

    for j in range(len(lineList)):
        canvas.delete(lineList[j])

    lineList[0]=canvas.create_line(points_2d[0][:2], points_2d[1][:2])
    lineList[1]=canvas.create_line(points_2d[1][:2], points_2d[2][:2])
    lineList[2]=canvas.create_line(points_2d[2][:2], points_2d[3][:2])
    lineList[3]=canvas.create_line(points_2d[3][:2], points_2d[0][:2])

    lineList[4]=canvas.create_line(points_2d[4][:2], points_2d[5][:2])
    lineList[5]=canvas.create_line(points_2d[5][:2], points_2d[6][:2])
    lineList[6]=canvas.create_line(points_2d[6][:2], points_2d[7][:2])
    lineList[7]=canvas.create_line(points_2d[7][:2], points_2d[4][:2])

    lineList[8]=canvas.create_line(points_2d[0][:2], points_2d[4][:2])
    lineList[9]=canvas.create_line(points_2d[1][:2], points_2d[5][:2])
    lineList[10]=canvas.create_line(points_2d[2][:2], points_2d[6][:2])
    lineList[11]=canvas.create_line(points_2d[3][:2], points_2d[7][:2])

    canvas.update()
    time.sleep(0.1)

window.mainloop()