LLVM Optimization Pass Viewer (#3769)gh-3514

1 files changed, 43 insertions, 3 deletions

diff --git a/lib/compilers/argument-parsers.js b/lib/compilers/argument-parsers.js
index bec330b87..ce80e74b7 100644
--- a/lib/compilers/argument-parsers.js
+++ b/lib/compilers/argument-parsers.js
@@ -22,9 +22,13 @@
 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 // POSSIBILITY OF SUCH DAMAGE.
 
+import path from 'path';
+import process from 'process';
+
 import _ from 'underscore';
 
 import {logger} from '../logger';
+import * as props from '../properties';
 import * as utils from '../utils';
 
 export class BaseParser {
@@ -143,15 +147,51 @@ export class ClangParser extends BaseParser {
             compiler.compiler.irArg = ['-Xclang', '-emit-llvm', '-fsyntax-only'];
         }
 
+        if (
+            BaseParser.hasSupport(options, '-mllvm') &&
+            this.mllvmOptions.has('--print-before-all') &&
+            this.mllvmOptions.has('--print-after-all')
+        ) {
+            compiler.compiler.supportsLLVMOptPipelineView = true;
+            compiler.compiler.llvmOptArg = ['-mllvm', '--print-before-all', '-mllvm', '--print-after-all'];
+            compiler.compiler.llvmOptModuleScopeArg = [];
+            if (this.mllvmOptions.has('--print-module-scope')) {
+                compiler.compiler.llvmOptModuleScopeArg = ['-mllvm', '-print-module-scope'];
+            }
+        }
+
         if (BaseParser.hasSupport(options, '-fcolor-diagnostics')) compiler.compiler.options += ' -fcolor-diagnostics';
         if (BaseParser.hasSupport(options, '-fno-crash-diagnostics'))
             compiler.compiler.options += ' -fno-crash-diagnostics';
     }
 
     static async parse(compiler) {
-        const options = await ClangParser.getOptions(compiler, '--help');
-        this.setCompilerSettingsFromOptions(compiler, options);
-        return compiler;
+        try {
+            const options = await ClangParser.getOptions(compiler, '--help');
+
+            const EXAMPLES_PATH = props.get('builtin', 'sourcePath', './examples/');
+            let filename = path.join(EXAMPLES_PATH, 'c++/default.cpp');
+            if (!path.isAbsolute(filename)) filename = path.join(process.env.PWD, filename);
+
+            this.mllvmOptions = new Set(
+                _.keys(await ClangParser.getOptions(compiler, `-mllvm --help-list-hidden ${filename} -c`, false)),
+            );
+            this.setCompilerSettingsFromOptions(compiler, options);
+            return compiler;
+        } catch (error) {
+            logger.error('Error while trying to generate llvm backend arguments');
+            logger.debug(error);
+        }
+    }
+
+    static async getOptions(compiler, helpArg, populate = true) {
+        const optionFinder = /^\s*(--?[\d+,<=>[\]a-z|-]*)\s*(.*)/i;
+        const result = await compiler.execCompilerCached(compiler.compiler.exe, helpArg.split(' '));
+        const options = result.code === 0 ? BaseParser.parseLines(result.stdout + result.stderr, optionFinder) : {};
+        if (populate) {
+            compiler.possibleArguments.populateOptions(options);
+        }
+        return options;
     }
 }