forked from LeenkxTeam/LNXSDK
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
|
import bpy
|
||
|
|
|
||
|
|
import lnx.material.cycles as cycles
|
||
|
|
import lnx.material.mat_state as mat_state
|
||
|
|
import lnx.material.make_finalize as make_finalize
|
||
|
|
import lnx.utils
|
||
|
|
|
||
|
|
if lnx.is_reload(__name__):
|
||
|
|
cycles = lnx.reload_module(cycles)
|
||
|
|
mat_state = lnx.reload_module(mat_state)
|
||
|
|
make_finalize = lnx.reload_module(make_finalize)
|
||
|
|
lnx.utils = lnx.reload_module(lnx.utils)
|
||
|
|
else:
|
||
|
|
lnx.enable_reload(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
def make(context_id):
|
||
|
|
wrd = bpy.data.worlds['Lnx']
|
||
|
|
|
||
|
|
vs = [{'name': 'pos', 'data': 'float3'}]
|
||
|
|
con_decal = mat_state.data.add_context({ 'name': context_id, 'vertex_elements': vs, 'depth_write': False, 'compare_mode': 'less', 'cull_mode': 'clockwise',
|
||
|
|
'blend_source': 'source_alpha',
|
||
|
|
'blend_destination': 'inverse_source_alpha',
|
||
|
|
'blend_operation': 'add',
|
||
|
|
'color_writes_alpha': [False, False]
|
||
|
|
})
|
||
|
|
|
||
|
|
vert = con_decal.make_vert()
|
||
|
|
frag = con_decal.make_frag()
|
||
|
|
geom = None
|
||
|
|
tesc = None
|
||
|
|
tese = None
|
||
|
|
|
||
|
|
vert.add_uniform('mat4 WVP', '_worldViewProjectionMatrix')
|
||
|
|
vert.add_uniform('mat3 N', '_normalMatrix')
|
||
|
|
vert.add_out('vec4 wvpposition')
|
||
|
|
vert.add_out('vec3 wnormal')
|
||
|
|
|
||
|
|
vert.write('wnormal = N * vec3(0.0, 0.0, 1.0);')
|
||
|
|
vert.write('wvpposition = WVP * vec4(pos.xyz, 1.0);')
|
||
|
|
vert.write('gl_Position = wvpposition;')
|
||
|
|
|
||
|
|
frag.add_include('compiled.inc')
|
||
|
|
frag.add_include('std/gbuffer.glsl')
|
||
|
|
frag.ins = vert.outs
|
||
|
|
frag.add_uniform('sampler2D gbufferD')
|
||
|
|
frag.add_uniform('mat4 invVP', '_inverseViewProjectionMatrix')
|
||
|
|
frag.add_uniform('mat4 invW', '_inverseWorldMatrix')
|
||
|
|
frag.add_out('vec4 fragColor[2]')
|
||
|
|
|
||
|
|
frag.write_attrib(' vec3 n = normalize(wnormal);')
|
||
|
|
|
||
|
|
frag.write_attrib(' vec2 screenPosition = wvpposition.xy / wvpposition.w;')
|
||
|
|
frag.write_attrib(' vec2 depthCoord = screenPosition * 0.5 + 0.5;')
|
||
|
|
frag.write_attrib('#ifdef _InvY')
|
||
|
|
frag.write_attrib(' depthCoord.y = 1.0 - depthCoord.y;')
|
||
|
|
frag.write_attrib('#endif')
|
||
|
|
frag.write_attrib(' float depth = texture(gbufferD, depthCoord).r * 2.0 - 1.0;')
|
||
|
|
|
||
|
|
frag.write_attrib(' vec3 wpos = getPos2(invVP, depth, depthCoord);')
|
||
|
|
frag.write_attrib(' vec4 mpos = invW * vec4(wpos, 1.0);')
|
||
|
|
frag.write_attrib(' if (abs(mpos.x) > 1.0) discard;')
|
||
|
|
frag.write_attrib(' if (abs(mpos.y) > 1.0) discard;')
|
||
|
|
frag.write_attrib(' if (abs(mpos.z) > 1.0) discard;')
|
||
|
|
frag.write_attrib(' vec2 texCoord = mpos.xy * 0.5 + 0.5;')
|
||
|
|
|
||
|
|
frag.write('vec3 basecol;')
|
||
|
|
frag.write('float roughness;')
|
||
|
|
frag.write('float metallic;')
|
||
|
|
frag.write('float occlusion;')
|
||
|
|
frag.write('float specular;')
|
||
|
|
frag.write('float opacity;')
|
||
|
|
frag.write('vec3 emissionCol;') # Declared to prevent compiler errors, but decals currently don't output any emission
|
||
|
|
frag.write('float ior;')
|
||
|
|
cycles.parse(mat_state.nodes, con_decal, vert, frag, geom, tesc, tese)
|
||
|
|
|
||
|
|
frag.write('n /= (abs(n.x) + abs(n.y) + abs(n.z));')
|
||
|
|
frag.write('n.xy = n.z >= 0.0 ? n.xy : octahedronWrap(n.xy);')
|
||
|
|
frag.write('fragColor[0] = vec4(n.xy, roughness, opacity);')
|
||
|
|
frag.write('fragColor[1] = vec4(basecol.rgb, opacity);')
|
||
|
|
|
||
|
|
make_finalize.make(con_decal)
|
||
|
|
|
||
|
|
return con_decal
|