mirror of
https://github.com/kuhyx/WUT_Computer_Science.git
synced 2026-07-04 17:43:12 +02:00
b74883b796
* works for cornell now, will be extended to import others * works for all scenes * can load environments and blur them * minor cleanup * Example commands added to README.md
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
from .utils.constants import SKYBOX_DISTANCE
|
|
import numpy as np
|
|
from abc import abstractmethod
|
|
|
|
|
|
# lights only have effect on Glossy materials
|
|
class Light:
|
|
def __init__(self, pos, color):
|
|
self.pos = pos
|
|
self.color = color
|
|
|
|
@abstractmethod
|
|
def get_L(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_irradiance(self, dist_light, NdotL):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_distance(self, M):
|
|
pass
|
|
|
|
|
|
class PointLight(Light):
|
|
def __init__(self, pos, color):
|
|
self.pos = pos
|
|
self.color = color
|
|
def get_L(self):
|
|
return (self.pos - M)*(1./(dist_light))
|
|
|
|
def get_distance(self, M):
|
|
return np.sqrt((self.pos - M).dot(self.pos - M))
|
|
|
|
def get_irradiance(self,dist_light, NdotL):
|
|
return self.color * NdotL/(dist_light**2.) * 100
|
|
|
|
class DirectionalLight(Light):
|
|
def __init__(self, Ldir, color):
|
|
self.Ldir = Ldir
|
|
self.color = color
|
|
def get_L(self):
|
|
return self.Ldir
|
|
|
|
def get_distance(self, M):
|
|
return SKYBOX_DISTANCE
|
|
|
|
def get_irradiance(self, dist_light, NdotL):
|
|
return self.color * NdotL |