旋转双棱镜光线追迹代码

根据提供的引用内容，以下是一个旋转双棱镜光线追迹的示例代码：

import numpy as np

def rotate_prism(ray, angle):
    # 定义旋转矩阵
    rotation_matrix = np.array([[np.cos(angle), -np.sin(angle), 0],
                                [np.sin(angle), np.cos(angle), 0],
                                [0, 0, 1]])
    # 将光线向量转换为列向量
    ray_vector = np.array([[ray[0]],
                           [ray[1]],
                           [ray[2]]])
    # 旋转光线向量
    rotated_ray_vector = np.dot(rotation_matrix, ray_vector)
    # 将旋转后的光线向量转换为行向量
    rotated_ray = [rotated_ray_vector[0][0], rotated_ray_vector[1][0], rotated_ray_vector[2][0]]
    
    return rotated_ray

# 示例使用
ray = [0, -0.6, -0.8]
angle = np.pi/4  # 旋转角度为45度
rotated_ray = rotate_prism(ray, angle)
print("旋转后的光线向量：", rotated_ray)