2016-01-28 14:47:46 +01:00
|
|
|
#!/usr/bin/python
|
|
|
|
|
2016-07-20 17:33:17 +02:00
|
|
|
# Usage: 'python makevariants.py blender.shader.arm'
|
2016-01-28 14:47:46 +01:00
|
|
|
# Output: blender.vert/frag.glsl, blender_NormalMapping.vert/frag.glsl,...
|
|
|
|
|
|
|
|
import sys
|
|
|
|
import itertools
|
|
|
|
import os
|
|
|
|
import json
|
|
|
|
|
|
|
|
# Create variations
|
2016-02-07 23:03:52 +01:00
|
|
|
def writeFile(path, name, defs, lines):
|
2016-01-28 14:47:46 +01:00
|
|
|
# with open('out/' + name, "w") as f:
|
2016-02-07 23:03:52 +01:00
|
|
|
with open(path + '/' + name, "w") as f:
|
2016-01-28 23:26:10 +01:00
|
|
|
# Write variation
|
|
|
|
defs_written = False
|
2016-01-28 14:47:46 +01:00
|
|
|
for line in lines:
|
|
|
|
f.write(line + '\n')
|
2016-01-28 23:26:10 +01:00
|
|
|
# Append defs after #version
|
|
|
|
if defs_written == False and line.startswith('#version '):
|
|
|
|
for d in defs:
|
|
|
|
f.write('#define ' + d + '\n')
|
|
|
|
defs_written = True
|
2016-01-28 14:47:46 +01:00
|
|
|
|
2016-07-10 00:51:39 +02:00
|
|
|
def make(json_name, fp, defs=None):
|
2016-01-28 14:47:46 +01:00
|
|
|
vert_shaders = []
|
|
|
|
frag_shaders = []
|
|
|
|
shader_names = []
|
2016-05-19 22:22:41 +02:00
|
|
|
|
|
|
|
if defs != None:
|
|
|
|
parse_defs = False
|
|
|
|
else:
|
|
|
|
parse_defs = True
|
|
|
|
defs = []
|
2016-01-28 14:47:46 +01:00
|
|
|
|
|
|
|
base_name = json_name.split('.', 1)[0]
|
|
|
|
|
|
|
|
# Make out dir
|
2016-07-10 00:51:39 +02:00
|
|
|
path = fp + '/compiled/Shaders/' + base_name
|
2016-01-28 14:47:46 +01:00
|
|
|
if not os.path.exists(path):
|
|
|
|
os.makedirs(path)
|
|
|
|
|
2016-05-19 22:22:41 +02:00
|
|
|
# Open json file
|
|
|
|
#json_file = open(sys.argv[1]).read()
|
|
|
|
json_file = open(json_name).read()
|
|
|
|
json_data = json.loads(json_file)
|
2016-01-28 14:47:46 +01:00
|
|
|
|
2016-05-19 22:22:41 +02:00
|
|
|
# Go through every context shaders and gather ifdefs
|
|
|
|
for c in json_data['contexts']:
|
|
|
|
vs = open(c['vertex_shader']).read().splitlines()
|
|
|
|
fs = open(c['fragment_shader']).read().splitlines()
|
|
|
|
shader_names.append(c['vertex_shader'].split('.', 1)[0])
|
|
|
|
vert_shaders.append(vs)
|
|
|
|
frag_shaders.append(fs)
|
2016-01-28 14:47:46 +01:00
|
|
|
|
2016-05-19 22:22:41 +02:00
|
|
|
if parse_defs == True:
|
2016-05-06 11:13:46 +02:00
|
|
|
lines = vs + fs
|
|
|
|
for line in lines:
|
|
|
|
if line.startswith('#ifdef'):
|
|
|
|
d = line.split(' ')[1]
|
|
|
|
if d != 'GL_ES':
|
|
|
|
defs.append(d)
|
2016-01-28 14:47:46 +01:00
|
|
|
|
2016-05-19 22:22:41 +02:00
|
|
|
|
|
|
|
if parse_defs == True:
|
2016-05-06 11:13:46 +02:00
|
|
|
# Merge duplicates and sort
|
|
|
|
defs = sorted(list(set(defs)))
|
2016-01-28 14:47:46 +01:00
|
|
|
|
2016-05-06 11:13:46 +02:00
|
|
|
for i in range(0, len(vert_shaders)):
|
|
|
|
vert_lines = vert_shaders[i]
|
|
|
|
frag_lines = frag_shaders[i]
|
2016-01-28 14:47:46 +01:00
|
|
|
|
2016-05-06 11:13:46 +02:00
|
|
|
# Process #defines and output name + defines + (.vert/.frag).glsl
|
|
|
|
for L in range(0, len(defs)+1):
|
|
|
|
for subset in itertools.combinations(defs, L):
|
|
|
|
shader_name = shader_names[i]
|
|
|
|
for s in subset:
|
|
|
|
shader_name += s
|
|
|
|
writeFile(path, shader_name + '.vert.glsl', subset, vert_lines)
|
|
|
|
writeFile(path, shader_name + '.frag.glsl', subset, frag_lines)
|
2016-05-19 22:22:41 +02:00
|
|
|
|
|
|
|
# Defs specified
|
|
|
|
else:
|
|
|
|
for i in range(0, len(vert_shaders)):
|
|
|
|
vert_lines = vert_shaders[i]
|
|
|
|
frag_lines = frag_shaders[i]
|
|
|
|
shader_name = shader_names[i]
|
|
|
|
for s in defs:
|
|
|
|
shader_name += s
|
|
|
|
writeFile(path, shader_name + '.vert.glsl', defs, vert_lines)
|
|
|
|
writeFile(path, shader_name + '.frag.glsl', defs, frag_lines)
|