• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python
2#
3# Copyright 2020 Google LLC
4#
5# Use of this source code is governed by a BSD-style license that can be
6# found in the LICENSE file.
7
8import os
9import shlex
10import subprocess
11import sys
12import tempfile
13
14batchCompile = True
15
16skslc = sys.argv[1]
17lang = sys.argv[2]
18settings = sys.argv[3]
19with open(sys.argv[4], 'r') as reader:
20    inputs = shlex.split(reader.read())
21
22def pairwise(iterable):
23    # Iterate over an array pairwise (two elements at a time).
24    a = iter(iterable)
25    return zip(a, a)
26
27def executeWorklist(input, worklist):
28    # Invoke skslc, passing in the worklist.
29    worklist.close()
30    try:
31        output = subprocess.check_output([
32            skslc, worklist.name], stderr=subprocess.STDOUT).decode('utf-8')
33    except subprocess.CalledProcessError as err:
34        if err.returncode != 1:
35            print("### " + input + " skslc error:\n")
36            print("\n".join(err.output.decode('utf-8').splitlines()))
37            sys.exit(err.returncode)
38        pass  # Compile errors (exit code 1) are expected and normal in test code
39
40    # Delete the worklist file now that execution is complete.
41    os.remove(worklist.name)
42
43def makeEmptyFile(path):
44    try:
45        open(path, 'wb').close()
46    except OSError:
47        pass
48
49def extensionForSpirvAsm(ext):
50    return ext if (ext == '.frag' or ext == '.vert') else '.frag'
51
52if settings != "--settings" and settings != "--nosettings":
53    sys.exit("### Expected --settings or --nosettings, got " + settings)
54
55targets = []
56worklist = tempfile.NamedTemporaryFile(suffix='.worklist', delete=False, mode='w')
57
58# The `inputs` array pairs off input files with their matching output directory, e.g.:
59#     //skia/tests/sksl/shared/test.sksl
60#     //skia/tests/sksl/shared/golden/
61#     //skia/tests/sksl/intrinsics/abs.sksl
62#     //skia/tests/sksl/intrinsics/golden/
63#     ... (etc) ...
64# Here we loop over these inputs and convert them into a worklist file for skslc.
65for input, targetDir in pairwise(inputs):
66    noExt, ext = os.path.splitext(input)
67    head, tail = os.path.split(noExt)
68    if not os.path.isdir(targetDir):
69        os.mkdir(targetDir)
70
71    target = os.path.join(targetDir, tail)
72    if settings == "--nosettings":
73        target += "StandaloneSettings"
74
75    targets.append(target)
76
77    if lang == "--glsl":
78        worklist.write(input + "\n")
79        worklist.write(target + ".glsl\n")
80        worklist.write(settings + "\n\n")
81    elif lang == "--metal":
82        worklist.write(input + "\n")
83        worklist.write(target + ".metal\n")
84        worklist.write(settings + "\n\n")
85    elif lang == "--spirv":
86        worklist.write(input + "\n")
87        worklist.write(target + ".asm" + extensionForSpirvAsm(ext) + "\n")
88        worklist.write(settings + "\n\n")
89    elif lang == "--skvm":
90        worklist.write(input + "\n")
91        worklist.write(target + ".skvm\n")
92        worklist.write(settings + "\n\n")
93    elif lang == "--stage":
94        worklist.write(input + "\n")
95        worklist.write(target + ".stage\n")
96        worklist.write(settings + "\n\n")
97    else:
98        sys.exit("### Expected one of: --glsl --metal --spirv --skvm --stage --dsl, got " + lang)
99
100    # Compile items one at a time.
101    if not batchCompile:
102        executeWorklist(input, worklist)
103        worklist = tempfile.NamedTemporaryFile(suffix='.worklist', delete=False, mode='w')
104
105# Compile everything all in one go.
106if batchCompile:
107    executeWorklist("", worklist)
108else:
109    worklist.close()
110    os.remove(worklist.name)
111