mirror of
https://github.com/compiler-explorer/compiler-explorer.git
synced 2025-12-27 07:04:04 -05:00
101 lines
4.2 KiB
TypeScript
101 lines
4.2 KiB
TypeScript
// Copyright (c) 2022, Compiler Explorer Authors
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are met:
|
|
//
|
|
// * Redistributions of source code must retain the above copyright notice,
|
|
// this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above copyright
|
|
// notice, this list of conditions and the following disclaimer in the
|
|
// documentation and/or other materials provided with the distribution.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
|
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
// POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
import path from 'node:path';
|
|
|
|
import type {ExecutionOptionsWithEnv} from '../../types/compilation/compilation.interfaces.js';
|
|
import type {PreliminaryCompilerInfo} from '../../types/compiler.interfaces.js';
|
|
import type {ParseFiltersAndOutputOptions} from '../../types/features/filters.interfaces.js';
|
|
import {BaseCompiler} from '../base-compiler.js';
|
|
import {CompilationEnvironment} from '../compilation-env.js';
|
|
import {logger} from '../logger.js';
|
|
import {SPIRVAsmParser} from '../parsers/asm-parser-spirv.js';
|
|
import * as utils from '../utils.js';
|
|
|
|
export class CLSPVCompiler extends BaseCompiler {
|
|
disassemblerPath: any;
|
|
|
|
static get key() {
|
|
return 'clspv';
|
|
}
|
|
|
|
constructor(compilerInfo: PreliminaryCompilerInfo, env: CompilationEnvironment) {
|
|
super(compilerInfo, env);
|
|
|
|
this.asm = new SPIRVAsmParser(this.compilerProps);
|
|
|
|
this.disassemblerPath = this.compilerProps('disassemblerPath');
|
|
}
|
|
|
|
getPrimaryOutputFilename(dirPath: string, outputFilebase: string) {
|
|
return path.join(dirPath, `${outputFilebase}.spv`);
|
|
}
|
|
|
|
override optionsForFilter(filters: ParseFiltersAndOutputOptions, outputFilename: string) {
|
|
const sourceDir = path.dirname(outputFilename);
|
|
const spvBinFilename = this.getPrimaryOutputFilename(sourceDir, this.outputFilebase);
|
|
return ['-o', spvBinFilename, '-g'];
|
|
}
|
|
|
|
// TODO: Check this to see if it needs key
|
|
override getOutputFilename(dirPath: string, outputFilebase: string) {
|
|
return path.join(dirPath, `${outputFilebase}.spvasm`);
|
|
}
|
|
|
|
override async runCompiler(
|
|
compiler: string,
|
|
options: string[],
|
|
inputFilename: string,
|
|
execOptions: ExecutionOptionsWithEnv,
|
|
) {
|
|
const sourceDir = path.dirname(inputFilename);
|
|
const spvBinFilename = this.getPrimaryOutputFilename(sourceDir, this.outputFilebase);
|
|
|
|
if (!execOptions) {
|
|
execOptions = this.getDefaultExecOptions();
|
|
}
|
|
execOptions.customCwd = path.dirname(inputFilename);
|
|
|
|
const spvBin = await this.exec(compiler, options, execOptions);
|
|
const result = this.transformToCompilationResult(spvBin, inputFilename);
|
|
|
|
if (spvBin.code !== 0 || !(await utils.fileExists(spvBinFilename))) {
|
|
return result;
|
|
}
|
|
|
|
const spvasmFilename = this.getOutputFilename(sourceDir, this.outputFilebase);
|
|
const disassemblerFlags = [spvBinFilename, '-o', spvasmFilename, '--comment'];
|
|
|
|
const spvasmOutput = await this.exec(this.disassemblerPath, disassemblerFlags, execOptions);
|
|
if (spvasmOutput.code !== 0) {
|
|
logger.error('SPIR-V binary to text failed', spvasmOutput);
|
|
}
|
|
|
|
result.stdout = result.stdout.concat(utils.parseOutput(spvasmOutput.stdout));
|
|
result.stderr = result.stderr.concat(utils.parseOutput(spvasmOutput.stderr));
|
|
result.languageId = 'spirv';
|
|
return result;
|
|
}
|
|
}
|