python仿真光照射光盘的图像代码

以下是一个简单的Python代码示例，用于实现光照射光盘的图像仿真：

import numpy as np
from PIL import Image

# 加载光盘图像
image = Image.open('disk_image.jpg')
image_array = np.array(image)

# 图像尺寸
width, height = image.size

# 创建空白图像，用于存储光照效果
output_image = Image.new('RGB', (width, height))

# 光源位置
light_source = np.array([100, 100, 100])

# 光强度
light_intensity = 100

# 光盘表面属性（反射率）
disk_reflectivity = 0.8

# 迭代处理每个像素点
for y in range(height):
    for x in range(width):
        # 当前像素的颜色值
        pixel_color = image_array[y][x]
        
        # 计算光线方向（从光源指向当前像素）
        light_direction = np.array([x, y, 0]) - light_source
        
        # 归一化光线方向向量
        light_direction_normalized = light_direction / np.linalg.norm(light_direction)
        
        # 计算光线强度
        light_intensity_at_pixel = light_intensity * disk_reflectivity * np.dot(pixel_color, light_direction_normalized)
        
        # 更新像素颜色值
        new_pixel_color = np.clip(pixel_color + light_intensity_at_pixel, 0, 255).astype(np.uint8)
        
        # 在输出图像上设置像素颜色值
        output_image.putpixel((x, y), tuple(new_pixel_color))

# 显示输出图像
output_image.show()

请注意，上述代码仅提供了一个简单的光照射光盘的图像仿真示例。实际应用中，可能需要更复杂的光照模型和算法来模拟真实的光照效果。此外，需要根据实际情况调整光源位置、光强度、光盘表面属性等参数，以获得所需的光照效果。