78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
import armutils
|
|
import os
|
|
import exporter
|
|
from material.shader_data import ShaderData
|
|
import material.make_forward as make_forward
|
|
import material.state as state
|
|
|
|
def parse(self, material, mat_context, defs):
|
|
state.material = material
|
|
state.group = material.node_tree
|
|
state.nodes = state.group.nodes
|
|
state.links = state.group.links
|
|
state.mat_context = mat_context
|
|
state.defs = defs
|
|
|
|
state.path = armutils.get_fp() + '/build/compiled/ShaderRaws/' + material.name
|
|
if not os.path.exists(state.path):
|
|
os.makedirs(state.path)
|
|
|
|
state.data = ShaderData(material)
|
|
|
|
rid = exporter.ArmoryExporter.renderpath_id
|
|
if rid == 'forward':
|
|
parse_forward()
|
|
elif rid == 'deferred':
|
|
parse_deferred
|
|
|
|
# TODO: Merge finalize shader here..
|
|
armutils.write_arm(state.path + '/' + material.name + '_data.arm', state.data.get())
|
|
|
|
def parse_deferred():
|
|
# Merge with fwd..
|
|
rpasses = exporter.ArmoryExporter.renderpath_passes
|
|
mesh_context_id = exporter.ArmoryExporter.mesh_context
|
|
shadows_context_id = exporter.ArmoryExporter.shadows_context
|
|
|
|
for rp in rpasses:
|
|
if rp == mesh_context_id:
|
|
con = make_deferred.mesh(rp)
|
|
elif rp == shadows_context_id:
|
|
con = make_deferred.shadows(rp)
|
|
else:
|
|
continue
|
|
|
|
write_shaders(con, rp)
|
|
|
|
def parse_forward():
|
|
rpasses = exporter.ArmoryExporter.renderpath_passes
|
|
mesh_context_id = exporter.ArmoryExporter.mesh_context
|
|
shadows_context_id = exporter.ArmoryExporter.shadows_context
|
|
|
|
for rp in rpasses:
|
|
if rp == mesh_context_id:
|
|
con = make_forward.mesh(rp)
|
|
elif rp == shadows_context_id:
|
|
con = make_forward.shadows(rp)
|
|
else:
|
|
continue
|
|
|
|
write_shaders(con, rp)
|
|
|
|
def write_shaders(con, rp):
|
|
if con.vert != None:
|
|
with open(state.path + '/' + state.material.name + '_' + rp + '.vert.glsl', 'w') as f:
|
|
f.write(con.vert.get())
|
|
if con.frag != None:
|
|
with open(state.path + '/' + state.material.name + '_' + rp + '.frag.glsl', 'w') as f:
|
|
f.write(con.frag.get())
|
|
if con.geom != None:
|
|
with open(state.path + '/' + state.material.name + '_' + rp + '.geom.glsl', 'w') as f:
|
|
f.write(con.geom.get())
|
|
if con.tesc != None:
|
|
with open(state.path + '/' + state.material.name + '_' + rp + '.tesc.glsl', 'w') as f:
|
|
f.write(con.tesc.get())
|
|
if con.tese != None:
|
|
with open(state.path + '/' + state.material.name + '_' + rp + '.tese.glsl', 'w') as f:
|
|
f.write(con.tese.get())
|