LNXSDK/leenkx/blender/lnx/material/make_overlay.py
2025-01-22 16:18:30 +01:00

52 lines
1.8 KiB
Python

import lnx
import lnx.material.make_finalize as make_finalize
import lnx.material.make_mesh as make_mesh
import lnx.material.mat_state as mat_state
import lnx.material.mat_utils as mat_utils
if lnx.is_reload(__name__):
make_finalize = lnx.reload_module(make_finalize)
make_mesh = lnx.reload_module(make_mesh)
mat_state = lnx.reload_module(mat_state)
mat_utils = lnx.reload_module(mat_utils)
else:
lnx.enable_reload(__name__)
def make(context_id):
con = { 'name': context_id, 'depth_write': True, 'compare_mode': 'less', 'cull_mode': 'clockwise' }
mat = mat_state.material
blend = mat.lnx_blending
if blend:
con['blend_source'] = mat.lnx_blending_source
con['blend_destination'] = mat.lnx_blending_destination
con['blend_operation'] = mat.lnx_blending_operation
con['alpha_blend_source'] = mat.lnx_blending_source_alpha
con['alpha_blend_destination'] = mat.lnx_blending_destination_alpha
con['alpha_blend_operation'] = mat.lnx_blending_operation_alpha
con_overlay = mat_state.data.add_context(con)
lnx_discard = mat.lnx_discard
is_transluc = mat_utils.is_transluc(mat)
parse_opacity = (blend and is_transluc) or lnx_discard
make_mesh.make_base(con_overlay, parse_opacity=parse_opacity)
frag = con_overlay.frag
if lnx_discard:
opac = mat.lnx_discard_opacity
frag.write('if (opacity < {0}) discard;'.format(opac))
frag.add_out('vec4 fragColor')
if blend and parse_opacity:
frag.write('fragColor = vec4(basecol + emissionCol, opacity);')
else:
frag.write('fragColor = vec4(basecol + emissionCol, 1.0);')
frag.write('fragColor.rgb = pow(fragColor.rgb, vec3(1.0 / 2.2));')
make_finalize.make(con_overlay)
return con_overlay