ilikecats/stk-code_catmod
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Algorithm for diffuse coeff implemented in py

Browse Source
This commit is contained in:
Vincent Lejeune
2014-10-22 00:17:32 +02:00
parent 7d22a5961d
commit de09937f88
1 changed files with 40 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
tools/ibl.py
View File

@@ -3,6 +3,8 @@ import numpy as np
import pylab as pl
n = 3
GridI, GridJ = np.meshgrid(np.linspace(-1, 1, n), np.linspace(-1, 1, n))
img = np.ones((n,n))
# constant factor of Ylm
c00 = 0.282095
@@ -32,21 +34,6 @@ def computeYmlOnGrid(Xgrid, Ygrid, Zgrid):
Y22 = c22 * (Xg * Xg - Yg * Yg)
return (Y00, Y1minus1, Y10, Y11, Y2minus2, Y2minus1, Y20, Y21, Y22)
GridI, GridJ = np.meshgrid(np.linspace(-1, 1, n), np.linspace(-1, 1, n))
FaceGrid = [(np.ones((n,n)), -GridI, -GridJ), #GL_TEXTURE_CUBE_MAP_POSITIVE_X
(-1 * np.ones((n,n)), -GridI, GridJ), #GL_TEXTURE_CUBE_MAP_NEGATIVE_X
(GridJ, np.ones((n,n)), GridI), #GL_TEXTURE_CUBE_MAP_POSITIVE_Y
(GridJ, -1 * np.ones((n,n)), -GridI), #GL_TEXTURE_CUBE_MAP_NEGATIVE_Y
(GridJ, GridI, np.ones((n,n))), #GL_TEXTURE_CUBE_MAP_POSITIVE_Z
(GridJ, -GridI, -1 * np.ones((n,n)))] #GL_TEXTURE_CUBE_MAP_NEGATIVE_Z
res = []
for (Xd, Yd, Zd) in FaceGrid:
res.append(computeYmlOnGrid(Xd, Yd, Zd))
# From http://www.rorydriscoll.com/2012/01/15/cubemap-texel-solid-angle/
def areaToPoint(x, y):
return np.arctan2(x * y, np.sqrt(x * x + y * y + 1))
@@ -60,8 +47,44 @@ def getSolidAngleGrid(Xgrid, Ygrid):
y1 = Ygrid + (1. / step)
return areaToPoint(x0,y0) - areaToPoint(x1, y0) - areaToPoint(x0, y1) + areaToPoint(x1, y1)
print(getSolidAngleGrid(GridI, GridJ))
def computeCoefficients():
"Compute coefficient SH00 to SH22 for monochromatic img"
FaceGrid = [(np.ones((n,n)), -GridI, -GridJ), #GL_TEXTURE_CUBE_MAP_POSITIVE_X
(-1 * np.ones((n,n)), -GridI, GridJ), #GL_TEXTURE_CUBE_MAP_NEGATIVE_X
(GridJ, np.ones((n,n)), GridI), #GL_TEXTURE_CUBE_MAP_POSITIVE_Y
(GridJ, -1 * np.ones((n,n)), -GridI), #GL_TEXTURE_CUBE_MAP_NEGATIVE_Y
(GridJ, GridI, np.ones((n,n))), #GL_TEXTURE_CUBE_MAP_POSITIVE_Z
(GridJ, -GridI, -1 * np.ones((n,n)))] #GL_TEXTURE_CUBE_MAP_NEGATIVE_Z
incomingIrradiance = getSolidAngleGrid(GridI, GridJ)
SH00 = 0
SH1minus1 = 0
SH10 = 0
SH11 = 0
SH2minus2 = 0
SH2minus1 = 0
SH20 = 0
SH21 = 0
SH22 = 0
for (Xgrid, Ygrid, Zgrid) in FaceGrid:
(Y00, Y1minus1, Y10, Y11, Y2minus2, Y2minus1, Y20, Y21, Y22) = computeYmlOnGrid(Xgrid, Ygrid, Zgrid)
SH00 += np.sum(np.sum(Y00 * incomingIrradiance * img))
SH1minus1 += np.sum(np.sum(Y1minus1 * incomingIrradiance * img))
SH10 += np.sum(np.sum(Y10 * incomingIrradiance * img))
SH11 += np.sum(np.sum(Y11 * incomingIrradiance * img))
SH2minus2 += np.sum(np.sum(Y2minus2 * incomingIrradiance * img))
SH2minus1 += np.sum(np.sum(Y2minus1 * incomingIrradiance * img))
SH20 += np.sum(np.sum(Y20 * incomingIrradiance * img))
SH21 += np.sum(np.sum(Y21 * incomingIrradiance * img))
SH22 += np.sum(np.sum(Y22 * incomingIrradiance * img))
return (SH00, SH1minus1, SH10, SH2minus2, SH2minus1, SH20, SH21, SH22)
print(computeCoefficients())
#res = []
#for (Xd, Yd, Zd) in FaceGrid:
# res.append(computeYmlOnGrid(Xd, Yd, Zd))
#I = img.open("C:/Users/vljn_000/Documents/GitHub/stk-assets/textures/ants.png")
#m = np.array(I)