1257b2971SCaroline Concatto //===- CompilerInvocation.cpp ---------------------------------------------===// 2257b2971SCaroline Concatto // 3257b2971SCaroline Concatto // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4257b2971SCaroline Concatto // See https://llvm.org/LICENSE.txt for license information. 5257b2971SCaroline Concatto // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6257b2971SCaroline Concatto // 7257b2971SCaroline Concatto //===----------------------------------------------------------------------===// 8257b2971SCaroline Concatto 9257b2971SCaroline Concatto #include "flang/Frontend/CompilerInvocation.h" 106d48a1a5SFaris Rehman #include "flang/Common/Fortran-features.h" 117809fa20SFaris Rehman #include "flang/Frontend/PreprocessorOptions.h" 126d48a1a5SFaris Rehman #include "flang/Semantics/semantics.h" 13197d9a55SFaris Rehman #include "flang/Version.inc" 14257b2971SCaroline Concatto #include "clang/Basic/AllDiagnostics.h" 15257b2971SCaroline Concatto #include "clang/Basic/DiagnosticDriver.h" 16257b2971SCaroline Concatto #include "clang/Basic/DiagnosticOptions.h" 17257b2971SCaroline Concatto #include "clang/Driver/DriverDiagnostic.h" 18257b2971SCaroline Concatto #include "clang/Driver/Options.h" 19257b2971SCaroline Concatto #include "llvm/ADT/StringRef.h" 20257b2971SCaroline Concatto #include "llvm/ADT/StringSwitch.h" 21257b2971SCaroline Concatto #include "llvm/Option/Arg.h" 22257b2971SCaroline Concatto #include "llvm/Option/ArgList.h" 23257b2971SCaroline Concatto #include "llvm/Option/OptTable.h" 24cd4abc52SArnamoy Bhattacharyya #include "llvm/Support/FileSystem.h" 25cd4abc52SArnamoy Bhattacharyya #include "llvm/Support/FileUtilities.h" 268d51d37eSAndrzej Warzynski #include "llvm/Support/Process.h" 27257b2971SCaroline Concatto #include "llvm/Support/raw_ostream.h" 286d48a1a5SFaris Rehman #include <memory> 29257b2971SCaroline Concatto 30257b2971SCaroline Concatto using namespace Fortran::frontend; 31257b2971SCaroline Concatto 32257b2971SCaroline Concatto //===----------------------------------------------------------------------===// 33257b2971SCaroline Concatto // Initialization. 34257b2971SCaroline Concatto //===----------------------------------------------------------------------===// 35257b2971SCaroline Concatto CompilerInvocationBase::CompilerInvocationBase() 367809fa20SFaris Rehman : diagnosticOpts_(new clang::DiagnosticOptions()), 377809fa20SFaris Rehman preprocessorOpts_(new PreprocessorOptions()) {} 38257b2971SCaroline Concatto 39257b2971SCaroline Concatto CompilerInvocationBase::CompilerInvocationBase(const CompilerInvocationBase &x) 407809fa20SFaris Rehman : diagnosticOpts_(new clang::DiagnosticOptions(x.GetDiagnosticOpts())), 417809fa20SFaris Rehman preprocessorOpts_(new PreprocessorOptions(x.preprocessorOpts())) {} 42257b2971SCaroline Concatto 43257b2971SCaroline Concatto CompilerInvocationBase::~CompilerInvocationBase() = default; 44257b2971SCaroline Concatto 45257b2971SCaroline Concatto //===----------------------------------------------------------------------===// 46257b2971SCaroline Concatto // Deserialization (from args) 47257b2971SCaroline Concatto //===----------------------------------------------------------------------===// 488d51d37eSAndrzej Warzynski static bool parseShowColorsArgs( 498d51d37eSAndrzej Warzynski const llvm::opt::ArgList &args, bool defaultColor) { 508d51d37eSAndrzej Warzynski // Color diagnostics default to auto ("on" if terminal supports) in the driver 518d51d37eSAndrzej Warzynski // but default to off in cc1, needing an explicit OPT_fdiagnostics_color. 528d51d37eSAndrzej Warzynski // Support both clang's -f[no-]color-diagnostics and gcc's 538d51d37eSAndrzej Warzynski // -f[no-]diagnostics-colors[=never|always|auto]. 548d51d37eSAndrzej Warzynski enum { 558d51d37eSAndrzej Warzynski Colors_On, 568d51d37eSAndrzej Warzynski Colors_Off, 578d51d37eSAndrzej Warzynski Colors_Auto 588d51d37eSAndrzej Warzynski } ShowColors = defaultColor ? Colors_Auto : Colors_Off; 598d51d37eSAndrzej Warzynski 608d51d37eSAndrzej Warzynski for (auto *a : args) { 618d51d37eSAndrzej Warzynski const llvm::opt::Option &O = a->getOption(); 628d51d37eSAndrzej Warzynski if (O.matches(clang::driver::options::OPT_fcolor_diagnostics) || 638d51d37eSAndrzej Warzynski O.matches(clang::driver::options::OPT_fdiagnostics_color)) { 648d51d37eSAndrzej Warzynski ShowColors = Colors_On; 658d51d37eSAndrzej Warzynski } else if (O.matches(clang::driver::options::OPT_fno_color_diagnostics) || 668d51d37eSAndrzej Warzynski O.matches(clang::driver::options::OPT_fno_diagnostics_color)) { 678d51d37eSAndrzej Warzynski ShowColors = Colors_Off; 688d51d37eSAndrzej Warzynski } else if (O.matches(clang::driver::options::OPT_fdiagnostics_color_EQ)) { 698d51d37eSAndrzej Warzynski llvm::StringRef value(a->getValue()); 708d51d37eSAndrzej Warzynski if (value == "always") 718d51d37eSAndrzej Warzynski ShowColors = Colors_On; 728d51d37eSAndrzej Warzynski else if (value == "never") 738d51d37eSAndrzej Warzynski ShowColors = Colors_Off; 748d51d37eSAndrzej Warzynski else if (value == "auto") 758d51d37eSAndrzej Warzynski ShowColors = Colors_Auto; 768d51d37eSAndrzej Warzynski } 778d51d37eSAndrzej Warzynski } 788d51d37eSAndrzej Warzynski 798d51d37eSAndrzej Warzynski return ShowColors == Colors_On || 808d51d37eSAndrzej Warzynski (ShowColors == Colors_Auto && llvm::sys::Process::StandardErrHasColors()); 818d51d37eSAndrzej Warzynski } 828d51d37eSAndrzej Warzynski 838d51d37eSAndrzej Warzynski bool Fortran::frontend::ParseDiagnosticArgs(clang::DiagnosticOptions &opts, 848d51d37eSAndrzej Warzynski llvm::opt::ArgList &args, bool defaultDiagColor) { 858d51d37eSAndrzej Warzynski opts.ShowColors = parseShowColorsArgs(args, defaultDiagColor); 868d51d37eSAndrzej Warzynski 878d51d37eSAndrzej Warzynski return true; 888d51d37eSAndrzej Warzynski } 898d51d37eSAndrzej Warzynski 90523d7bc6SAndrzej Warzynski // Tweak the frontend configuration based on the frontend action 91523d7bc6SAndrzej Warzynski static void setUpFrontendBasedOnAction(FrontendOptions &opts) { 92523d7bc6SAndrzej Warzynski assert(opts.programAction_ != Fortran::frontend::InvalidAction && 93523d7bc6SAndrzej Warzynski "Fortran frontend action not set!"); 94523d7bc6SAndrzej Warzynski 95523d7bc6SAndrzej Warzynski if (opts.programAction_ == DebugDumpParsingLog) 96523d7bc6SAndrzej Warzynski opts.instrumentedParse_ = true; 97*85b86c6fSAndrzej Warzynski 98*85b86c6fSAndrzej Warzynski if (opts.programAction_ == DebugDumpProvenance) 99*85b86c6fSAndrzej Warzynski opts.needProvenanceRangeToCharBlockMappings_ = true; 100523d7bc6SAndrzej Warzynski } 101523d7bc6SAndrzej Warzynski 102257b2971SCaroline Concatto static InputKind ParseFrontendArgs(FrontendOptions &opts, 103257b2971SCaroline Concatto llvm::opt::ArgList &args, clang::DiagnosticsEngine &diags) { 104760e6c4cSAndrzej Warzynski 105760e6c4cSAndrzej Warzynski // By default the frontend driver creates a ParseSyntaxOnly action. 106760e6c4cSAndrzej Warzynski opts.programAction_ = ParseSyntaxOnly; 107760e6c4cSAndrzej Warzynski 108257b2971SCaroline Concatto // Identify the action (i.e. opts.ProgramAction) 109257b2971SCaroline Concatto if (const llvm::opt::Arg *a = 110257b2971SCaroline Concatto args.getLastArg(clang::driver::options::OPT_Action_Group)) { 111257b2971SCaroline Concatto switch (a->getOption().getID()) { 112257b2971SCaroline Concatto default: { 113257b2971SCaroline Concatto llvm_unreachable("Invalid option in group!"); 114257b2971SCaroline Concatto } 1154c5906cfSCaroline Concatto case clang::driver::options::OPT_test_io: 1164c5906cfSCaroline Concatto opts.programAction_ = InputOutputTest; 1174c5906cfSCaroline Concatto break; 118d28de0d7SCaroline Concatto case clang::driver::options::OPT_E: 119d28de0d7SCaroline Concatto opts.programAction_ = PrintPreprocessedInput; 120d28de0d7SCaroline Concatto break; 1217d246cb1SAndrzej Warzynski case clang::driver::options::OPT_fsyntax_only: 1227d246cb1SAndrzej Warzynski opts.programAction_ = ParseSyntaxOnly; 1237d246cb1SAndrzej Warzynski break; 124e5cdb6c5SAndrzej Warzynski case clang::driver::options::OPT_emit_obj: 125e5cdb6c5SAndrzej Warzynski opts.programAction_ = EmitObj; 126e5cdb6c5SAndrzej Warzynski break; 12796d229c9SAndrzej Warzynski case clang::driver::options::OPT_fdebug_unparse: 12896d229c9SAndrzej Warzynski opts.programAction_ = DebugUnparse; 12996d229c9SAndrzej Warzynski break; 13096d229c9SAndrzej Warzynski case clang::driver::options::OPT_fdebug_unparse_with_symbols: 13196d229c9SAndrzej Warzynski opts.programAction_ = DebugUnparseWithSymbols; 13296d229c9SAndrzej Warzynski break; 1334bd08dabSFaris Rehman case clang::driver::options::OPT_fdebug_dump_symbols: 1344bd08dabSFaris Rehman opts.programAction_ = DebugDumpSymbols; 1354bd08dabSFaris Rehman break; 1364bd08dabSFaris Rehman case clang::driver::options::OPT_fdebug_dump_parse_tree: 1374bd08dabSFaris Rehman opts.programAction_ = DebugDumpParseTree; 1384bd08dabSFaris Rehman break; 1394bd08dabSFaris Rehman case clang::driver::options::OPT_fdebug_dump_provenance: 1404bd08dabSFaris Rehman opts.programAction_ = DebugDumpProvenance; 1414bd08dabSFaris Rehman break; 142523d7bc6SAndrzej Warzynski case clang::driver::options::OPT_fdebug_dump_parsing_log: 143523d7bc6SAndrzej Warzynski opts.programAction_ = DebugDumpParsingLog; 144523d7bc6SAndrzej Warzynski break; 145529f7181SFaris Rehman case clang::driver::options::OPT_fdebug_measure_parse_tree: 146529f7181SFaris Rehman opts.programAction_ = DebugMeasureParseTree; 147529f7181SFaris Rehman break; 148529f7181SFaris Rehman case clang::driver::options::OPT_fdebug_pre_fir_tree: 149529f7181SFaris Rehman opts.programAction_ = DebugPreFIRTree; 150529f7181SFaris Rehman break; 151eefda605SAndrzej Warzynski case clang::driver::options::OPT_fget_symbols_sources: 152eefda605SAndrzej Warzynski opts.programAction_ = GetSymbolsSources; 153eefda605SAndrzej Warzynski break; 154d28de0d7SCaroline Concatto 155257b2971SCaroline Concatto // TODO: 156257b2971SCaroline Concatto // case calng::driver::options::OPT_emit_llvm: 157257b2971SCaroline Concatto // case clang::driver::options::OPT_emit_llvm_only: 158257b2971SCaroline Concatto // case clang::driver::options::OPT_emit_codegen_only: 159257b2971SCaroline Concatto // case clang::driver::options::OPT_emit_module: 160257b2971SCaroline Concatto // (...) 161257b2971SCaroline Concatto } 162257b2971SCaroline Concatto } 163257b2971SCaroline Concatto 1644c5906cfSCaroline Concatto opts.outputFile_ = args.getLastArgValue(clang::driver::options::OPT_o); 165257b2971SCaroline Concatto opts.showHelp_ = args.hasArg(clang::driver::options::OPT_help); 166257b2971SCaroline Concatto opts.showVersion_ = args.hasArg(clang::driver::options::OPT_version); 167257b2971SCaroline Concatto 168257b2971SCaroline Concatto // Get the input kind (from the value passed via `-x`) 169257b2971SCaroline Concatto InputKind dashX(Language::Unknown); 170257b2971SCaroline Concatto if (const llvm::opt::Arg *a = 171257b2971SCaroline Concatto args.getLastArg(clang::driver::options::OPT_x)) { 172257b2971SCaroline Concatto llvm::StringRef XValue = a->getValue(); 173257b2971SCaroline Concatto // Principal languages. 174257b2971SCaroline Concatto dashX = llvm::StringSwitch<InputKind>(XValue) 175257b2971SCaroline Concatto .Case("f90", Language::Fortran) 176257b2971SCaroline Concatto .Default(Language::Unknown); 177257b2971SCaroline Concatto 178257b2971SCaroline Concatto // Some special cases cannot be combined with suffixes. 179257b2971SCaroline Concatto if (dashX.IsUnknown()) 180257b2971SCaroline Concatto dashX = llvm::StringSwitch<InputKind>(XValue) 181257b2971SCaroline Concatto .Case("ir", Language::LLVM_IR) 182257b2971SCaroline Concatto .Default(Language::Unknown); 183257b2971SCaroline Concatto 184257b2971SCaroline Concatto if (dashX.IsUnknown()) 185257b2971SCaroline Concatto diags.Report(clang::diag::err_drv_invalid_value) 186257b2971SCaroline Concatto << a->getAsString(args) << a->getValue(); 187257b2971SCaroline Concatto } 188257b2971SCaroline Concatto 1894c5906cfSCaroline Concatto // Collect the input files and save them in our instance of FrontendOptions. 1904c5906cfSCaroline Concatto std::vector<std::string> inputs = 1914c5906cfSCaroline Concatto args.getAllArgValues(clang::driver::options::OPT_INPUT); 1924c5906cfSCaroline Concatto opts.inputs_.clear(); 1934c5906cfSCaroline Concatto if (inputs.empty()) 1944c5906cfSCaroline Concatto // '-' is the default input if none is given. 1954c5906cfSCaroline Concatto inputs.push_back("-"); 1964c5906cfSCaroline Concatto for (unsigned i = 0, e = inputs.size(); i != e; ++i) { 1974c5906cfSCaroline Concatto InputKind ik = dashX; 1984c5906cfSCaroline Concatto if (ik.IsUnknown()) { 1994c5906cfSCaroline Concatto ik = FrontendOptions::GetInputKindForExtension( 2004c5906cfSCaroline Concatto llvm::StringRef(inputs[i]).rsplit('.').second); 2014c5906cfSCaroline Concatto if (ik.IsUnknown()) 2024c5906cfSCaroline Concatto ik = Language::Unknown; 2034c5906cfSCaroline Concatto if (i == 0) 2044c5906cfSCaroline Concatto dashX = ik; 2054c5906cfSCaroline Concatto } 2064c5906cfSCaroline Concatto 2074c5906cfSCaroline Concatto opts.inputs_.emplace_back(std::move(inputs[i]), ik); 2084c5906cfSCaroline Concatto } 2093a1513c1SFaris Rehman 2103a1513c1SFaris Rehman // Set fortranForm_ based on options -ffree-form and -ffixed-form. 2113a1513c1SFaris Rehman if (const auto *arg = args.getLastArg(clang::driver::options::OPT_ffixed_form, 2123a1513c1SFaris Rehman clang::driver::options::OPT_ffree_form)) { 2133a1513c1SFaris Rehman opts.fortranForm_ = 2143a1513c1SFaris Rehman arg->getOption().matches(clang::driver::options::OPT_ffixed_form) 2153a1513c1SFaris Rehman ? FortranForm::FixedForm 2163a1513c1SFaris Rehman : FortranForm::FreeForm; 2173a1513c1SFaris Rehman } 2183a1513c1SFaris Rehman 2193a1513c1SFaris Rehman // Set fixedFormColumns_ based on -ffixed-line-length=<value> 2203a1513c1SFaris Rehman if (const auto *arg = 2213a1513c1SFaris Rehman args.getLastArg(clang::driver::options::OPT_ffixed_line_length_EQ)) { 2223a1513c1SFaris Rehman llvm::StringRef argValue = llvm::StringRef(arg->getValue()); 2233a1513c1SFaris Rehman std::int64_t columns = -1; 2243a1513c1SFaris Rehman if (argValue == "none") { 2253a1513c1SFaris Rehman columns = 0; 2263a1513c1SFaris Rehman } else if (argValue.getAsInteger(/*Radix=*/10, columns)) { 2273a1513c1SFaris Rehman columns = -1; 2283a1513c1SFaris Rehman } 2293a1513c1SFaris Rehman if (columns < 0) { 2303a1513c1SFaris Rehman diags.Report(clang::diag::err_drv_invalid_value_with_suggestion) 2313a1513c1SFaris Rehman << arg->getOption().getName() << arg->getValue() 2323a1513c1SFaris Rehman << "value must be 'none' or a non-negative integer"; 2333a1513c1SFaris Rehman } else if (columns == 0) { 2343a1513c1SFaris Rehman opts.fixedFormColumns_ = 1000000; 2353a1513c1SFaris Rehman } else if (columns < 7) { 2363a1513c1SFaris Rehman diags.Report(clang::diag::err_drv_invalid_value_with_suggestion) 2373a1513c1SFaris Rehman << arg->getOption().getName() << arg->getValue() 2383a1513c1SFaris Rehman << "value must be at least seven"; 2393a1513c1SFaris Rehman } else { 2403a1513c1SFaris Rehman opts.fixedFormColumns_ = columns; 2413a1513c1SFaris Rehman } 2423a1513c1SFaris Rehman } 2436d48a1a5SFaris Rehman 24410826ea7SFaris Rehman if (const llvm::opt::Arg *arg = 24510826ea7SFaris Rehman args.getLastArg(clang::driver::options::OPT_fimplicit_none, 24610826ea7SFaris Rehman clang::driver::options::OPT_fno_implicit_none)) { 24710826ea7SFaris Rehman opts.features_.Enable( 24810826ea7SFaris Rehman Fortran::common::LanguageFeature::ImplicitNoneTypeAlways, 24910826ea7SFaris Rehman arg->getOption().matches(clang::driver::options::OPT_fimplicit_none)); 25010826ea7SFaris Rehman } 25110826ea7SFaris Rehman if (const llvm::opt::Arg *arg = 25210826ea7SFaris Rehman args.getLastArg(clang::driver::options::OPT_fbackslash, 25310826ea7SFaris Rehman clang::driver::options::OPT_fno_backslash)) { 25410826ea7SFaris Rehman opts.features_.Enable(Fortran::common::LanguageFeature::BackslashEscapes, 25510826ea7SFaris Rehman arg->getOption().matches(clang::driver::options::OPT_fbackslash)); 25610826ea7SFaris Rehman } 25710826ea7SFaris Rehman if (const llvm::opt::Arg *arg = 25810826ea7SFaris Rehman args.getLastArg(clang::driver::options::OPT_flogical_abbreviations, 25910826ea7SFaris Rehman clang::driver::options::OPT_fno_logical_abbreviations)) { 26010826ea7SFaris Rehman opts.features_.Enable( 26110826ea7SFaris Rehman Fortran::common::LanguageFeature::LogicalAbbreviations, 26210826ea7SFaris Rehman arg->getOption().matches( 26310826ea7SFaris Rehman clang::driver::options::OPT_flogical_abbreviations)); 26410826ea7SFaris Rehman } 26510826ea7SFaris Rehman if (const llvm::opt::Arg *arg = 26610826ea7SFaris Rehman args.getLastArg(clang::driver::options::OPT_fxor_operator, 26710826ea7SFaris Rehman clang::driver::options::OPT_fno_xor_operator)) { 26810826ea7SFaris Rehman opts.features_.Enable(Fortran::common::LanguageFeature::XOROperator, 26910826ea7SFaris Rehman arg->getOption().matches(clang::driver::options::OPT_fxor_operator)); 27010826ea7SFaris Rehman } 27110826ea7SFaris Rehman if (args.hasArg( 27210826ea7SFaris Rehman clang::driver::options::OPT_falternative_parameter_statement)) { 27310826ea7SFaris Rehman opts.features_.Enable(Fortran::common::LanguageFeature::OldStyleParameter); 27410826ea7SFaris Rehman } 27510826ea7SFaris Rehman if (const llvm::opt::Arg *arg = 27610826ea7SFaris Rehman args.getLastArg(clang::driver::options::OPT_finput_charset_EQ)) { 27710826ea7SFaris Rehman llvm::StringRef argValue = arg->getValue(); 27810826ea7SFaris Rehman if (argValue == "utf-8") { 27910826ea7SFaris Rehman opts.encoding_ = Fortran::parser::Encoding::UTF_8; 28010826ea7SFaris Rehman } else if (argValue == "latin-1") { 28110826ea7SFaris Rehman opts.encoding_ = Fortran::parser::Encoding::LATIN_1; 28210826ea7SFaris Rehman } else { 28310826ea7SFaris Rehman diags.Report(clang::diag::err_drv_invalid_value) 28410826ea7SFaris Rehman << arg->getAsString(args) << argValue; 28510826ea7SFaris Rehman } 28610826ea7SFaris Rehman } 287523d7bc6SAndrzej Warzynski 288523d7bc6SAndrzej Warzynski setUpFrontendBasedOnAction(opts); 289523d7bc6SAndrzej Warzynski 290257b2971SCaroline Concatto return dashX; 291257b2971SCaroline Concatto } 292257b2971SCaroline Concatto 293cd4abc52SArnamoy Bhattacharyya // Generate the path to look for intrinsic modules 294cd4abc52SArnamoy Bhattacharyya static std::string getIntrinsicDir() { 295cd4abc52SArnamoy Bhattacharyya // TODO: Find a system independent API 296cd4abc52SArnamoy Bhattacharyya llvm::SmallString<128> driverPath; 297cd4abc52SArnamoy Bhattacharyya driverPath.assign(llvm::sys::fs::getMainExecutable(nullptr, nullptr)); 298cd4abc52SArnamoy Bhattacharyya llvm::sys::path::remove_filename(driverPath); 299cd4abc52SArnamoy Bhattacharyya driverPath.append("/../include/flang/"); 300cd4abc52SArnamoy Bhattacharyya return std::string(driverPath); 301cd4abc52SArnamoy Bhattacharyya } 302cd4abc52SArnamoy Bhattacharyya 3037809fa20SFaris Rehman /// Parses all preprocessor input arguments and populates the preprocessor 3047809fa20SFaris Rehman /// options accordingly. 3057809fa20SFaris Rehman /// 3067809fa20SFaris Rehman /// \param [in] opts The preprocessor options instance 3077809fa20SFaris Rehman /// \param [out] args The list of input arguments 3087809fa20SFaris Rehman static void parsePreprocessorArgs( 3097809fa20SFaris Rehman Fortran::frontend::PreprocessorOptions &opts, llvm::opt::ArgList &args) { 3107809fa20SFaris Rehman // Add macros from the command line. 3117809fa20SFaris Rehman for (const auto *currentArg : args.filtered( 3127809fa20SFaris Rehman clang::driver::options::OPT_D, clang::driver::options::OPT_U)) { 3137809fa20SFaris Rehman if (currentArg->getOption().matches(clang::driver::options::OPT_D)) { 3147809fa20SFaris Rehman opts.addMacroDef(currentArg->getValue()); 3157809fa20SFaris Rehman } else { 3167809fa20SFaris Rehman opts.addMacroUndef(currentArg->getValue()); 3177809fa20SFaris Rehman } 3187809fa20SFaris Rehman } 31987dfd5e0SFaris Rehman 32087dfd5e0SFaris Rehman // Add the ordered list of -I's. 32187dfd5e0SFaris Rehman for (const auto *currentArg : args.filtered(clang::driver::options::OPT_I)) 32287dfd5e0SFaris Rehman opts.searchDirectoriesFromDashI.emplace_back(currentArg->getValue()); 323cd4abc52SArnamoy Bhattacharyya 324cd4abc52SArnamoy Bhattacharyya // Prepend the ordered list of -intrinsic-modules-path 325cd4abc52SArnamoy Bhattacharyya // to the default location to search. 326cd4abc52SArnamoy Bhattacharyya for (const auto *currentArg : 327cd4abc52SArnamoy Bhattacharyya args.filtered(clang::driver::options::OPT_fintrinsic_modules_path)) 328cd4abc52SArnamoy Bhattacharyya opts.searchDirectoriesFromIntrModPath.emplace_back(currentArg->getValue()); 329b83a4450SAndrzej Warzynski 330b83a4450SAndrzej Warzynski // -cpp/-nocpp 331b83a4450SAndrzej Warzynski if (const auto *currentArg = args.getLastArg( 332b83a4450SAndrzej Warzynski clang::driver::options::OPT_cpp, clang::driver::options::OPT_nocpp)) 333b83a4450SAndrzej Warzynski opts.macrosFlag_ = 334b83a4450SAndrzej Warzynski (currentArg->getOption().matches(clang::driver::options::OPT_cpp)) 335b83a4450SAndrzej Warzynski ? PPMacrosFlag::Include 336b83a4450SAndrzej Warzynski : PPMacrosFlag::Exclude; 3377809fa20SFaris Rehman } 3387809fa20SFaris Rehman 339985a42fdSArnamoy Bhattacharyya /// Parses all semantic related arguments and populates the variables 340985a42fdSArnamoy Bhattacharyya /// options accordingly. 3411fd4beecSArnamoy Bhattacharyya static void parseSemaArgs(CompilerInvocation &res, llvm::opt::ArgList &args, 342985a42fdSArnamoy Bhattacharyya clang::DiagnosticsEngine &diags) { 343985a42fdSArnamoy Bhattacharyya 3441fd4beecSArnamoy Bhattacharyya // -J/module-dir option 345985a42fdSArnamoy Bhattacharyya auto moduleDirList = 346985a42fdSArnamoy Bhattacharyya args.getAllArgValues(clang::driver::options::OPT_module_dir); 347985a42fdSArnamoy Bhattacharyya // User can only specify -J/-module-dir once 348985a42fdSArnamoy Bhattacharyya // https://gcc.gnu.org/onlinedocs/gfortran/Directory-Options.html 349985a42fdSArnamoy Bhattacharyya if (moduleDirList.size() > 1) { 350985a42fdSArnamoy Bhattacharyya const unsigned diagID = 351985a42fdSArnamoy Bhattacharyya diags.getCustomDiagID(clang::DiagnosticsEngine::Error, 352985a42fdSArnamoy Bhattacharyya "Only one '-module-dir/-J' option allowed"); 353985a42fdSArnamoy Bhattacharyya diags.Report(diagID); 354985a42fdSArnamoy Bhattacharyya } 355985a42fdSArnamoy Bhattacharyya if (moduleDirList.size() == 1) 3561fd4beecSArnamoy Bhattacharyya res.SetModuleDir(moduleDirList[0]); 3571fd4beecSArnamoy Bhattacharyya 3581fd4beecSArnamoy Bhattacharyya // -fdebug-module-writer option 3591fd4beecSArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_fdebug_module_writer)) { 3601fd4beecSArnamoy Bhattacharyya res.SetDebugModuleDir(true); 3611fd4beecSArnamoy Bhattacharyya } 362985a42fdSArnamoy Bhattacharyya } 363985a42fdSArnamoy Bhattacharyya 3647416e8a8SArnamoy Bhattacharyya /// Parses all diagnostics related arguments and populates the variables 3657416e8a8SArnamoy Bhattacharyya /// options accordingly. 3667416e8a8SArnamoy Bhattacharyya static void parseDiagArgs(CompilerInvocation &res, llvm::opt::ArgList &args, 3677416e8a8SArnamoy Bhattacharyya clang::DiagnosticsEngine &diags) { 3687416e8a8SArnamoy Bhattacharyya // -Werror option 3697416e8a8SArnamoy Bhattacharyya // TODO: Currently throws a Diagnostic for anything other than -W<error>, 3707416e8a8SArnamoy Bhattacharyya // this has to change when other -W<opt>'s are supported. 3717416e8a8SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_W_Joined)) { 3727416e8a8SArnamoy Bhattacharyya if (args.getLastArgValue(clang::driver::options::OPT_W_Joined) 3737416e8a8SArnamoy Bhattacharyya .equals("error")) { 3747416e8a8SArnamoy Bhattacharyya res.SetWarnAsErr(true); 3757416e8a8SArnamoy Bhattacharyya } else { 3767416e8a8SArnamoy Bhattacharyya const unsigned diagID = 3777416e8a8SArnamoy Bhattacharyya diags.getCustomDiagID(clang::DiagnosticsEngine::Error, 3787416e8a8SArnamoy Bhattacharyya "Only `-Werror` is supported currently."); 3797416e8a8SArnamoy Bhattacharyya diags.Report(diagID); 3807416e8a8SArnamoy Bhattacharyya } 3817416e8a8SArnamoy Bhattacharyya } 3827416e8a8SArnamoy Bhattacharyya } 3837416e8a8SArnamoy Bhattacharyya 384ab971c29SArnamoy Bhattacharyya /// Parses all Dialect related arguments and populates the variables 385ab971c29SArnamoy Bhattacharyya /// options accordingly. 386ab971c29SArnamoy Bhattacharyya static void parseDialectArgs(CompilerInvocation &res, llvm::opt::ArgList &args, 387ab971c29SArnamoy Bhattacharyya clang::DiagnosticsEngine &diags) { 388ab971c29SArnamoy Bhattacharyya 389ab971c29SArnamoy Bhattacharyya // -fdefault* family 390ab971c29SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_fdefault_real_8)) { 391ab971c29SArnamoy Bhattacharyya res.defaultKinds().set_defaultRealKind(8); 392ab971c29SArnamoy Bhattacharyya res.defaultKinds().set_doublePrecisionKind(16); 393ab971c29SArnamoy Bhattacharyya } 394ab971c29SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_fdefault_integer_8)) { 395ab971c29SArnamoy Bhattacharyya res.defaultKinds().set_defaultIntegerKind(8); 396ab971c29SArnamoy Bhattacharyya res.defaultKinds().set_subscriptIntegerKind(8); 397ab971c29SArnamoy Bhattacharyya res.defaultKinds().set_sizeIntegerKind(8); 398ab971c29SArnamoy Bhattacharyya } 399ab971c29SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_fdefault_double_8)) { 400ab971c29SArnamoy Bhattacharyya if (!args.hasArg(clang::driver::options::OPT_fdefault_real_8)) { 401ab971c29SArnamoy Bhattacharyya // -fdefault-double-8 has to be used with -fdefault-real-8 402ab971c29SArnamoy Bhattacharyya // to be compatible with gfortran 403ab971c29SArnamoy Bhattacharyya const unsigned diagID = 404ab971c29SArnamoy Bhattacharyya diags.getCustomDiagID(clang::DiagnosticsEngine::Error, 405ab971c29SArnamoy Bhattacharyya "Use of `-fdefault-double-8` requires `-fdefault-real-8`"); 406ab971c29SArnamoy Bhattacharyya diags.Report(diagID); 407ab971c29SArnamoy Bhattacharyya } 408ab971c29SArnamoy Bhattacharyya // https://gcc.gnu.org/onlinedocs/gfortran/Fortran-Dialect-Options.html 409ab971c29SArnamoy Bhattacharyya res.defaultKinds().set_doublePrecisionKind(8); 410ab971c29SArnamoy Bhattacharyya } 411ab971c29SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_flarge_sizes)) 412ab971c29SArnamoy Bhattacharyya res.defaultKinds().set_sizeIntegerKind(8); 413ab971c29SArnamoy Bhattacharyya 414ab971c29SArnamoy Bhattacharyya // -fopenmp and -fopenacc 415ab971c29SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_fopenacc)) { 416ab971c29SArnamoy Bhattacharyya res.frontendOpts().features_.Enable( 417ab971c29SArnamoy Bhattacharyya Fortran::common::LanguageFeature::OpenACC); 418ab971c29SArnamoy Bhattacharyya } 419ab971c29SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_fopenmp)) { 420ab971c29SArnamoy Bhattacharyya res.frontendOpts().features_.Enable( 421ab971c29SArnamoy Bhattacharyya Fortran::common::LanguageFeature::OpenMP); 422ab971c29SArnamoy Bhattacharyya } 4234c7ebf79SArnamoy Bhattacharyya 424fcf629d7SAndrzej Warzynski // -pedantic 4254c7ebf79SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_pedantic)) { 4264c7ebf79SArnamoy Bhattacharyya res.set_EnableConformanceChecks(); 4274c7ebf79SArnamoy Bhattacharyya } 428fcf629d7SAndrzej Warzynski // -std=f2018 (currently this implies -pedantic) 4294c7ebf79SArnamoy Bhattacharyya // TODO: Set proper options when more fortran standards 4304c7ebf79SArnamoy Bhattacharyya // are supported. 4314c7ebf79SArnamoy Bhattacharyya if (args.hasArg(clang::driver::options::OPT_std_EQ)) { 4324c7ebf79SArnamoy Bhattacharyya auto standard = args.getLastArgValue(clang::driver::options::OPT_std_EQ); 4334c7ebf79SArnamoy Bhattacharyya // We only allow f2018 as the given standard 4344c7ebf79SArnamoy Bhattacharyya if (standard.equals("f2018")) { 4354c7ebf79SArnamoy Bhattacharyya res.set_EnableConformanceChecks(); 4364c7ebf79SArnamoy Bhattacharyya } else { 4374c7ebf79SArnamoy Bhattacharyya const unsigned diagID = 4384c7ebf79SArnamoy Bhattacharyya diags.getCustomDiagID(clang::DiagnosticsEngine::Error, 4394c7ebf79SArnamoy Bhattacharyya "Only -std=f2018 is allowed currently."); 4404c7ebf79SArnamoy Bhattacharyya diags.Report(diagID); 4414c7ebf79SArnamoy Bhattacharyya } 4424c7ebf79SArnamoy Bhattacharyya } 443ab971c29SArnamoy Bhattacharyya return; 444ab971c29SArnamoy Bhattacharyya } 445ab971c29SArnamoy Bhattacharyya 446257b2971SCaroline Concatto bool CompilerInvocation::CreateFromArgs(CompilerInvocation &res, 447257b2971SCaroline Concatto llvm::ArrayRef<const char *> commandLineArgs, 448257b2971SCaroline Concatto clang::DiagnosticsEngine &diags) { 449257b2971SCaroline Concatto 450257b2971SCaroline Concatto bool success = true; 451257b2971SCaroline Concatto 452257b2971SCaroline Concatto // Parse the arguments 453257b2971SCaroline Concatto const llvm::opt::OptTable &opts = clang::driver::getDriverOptTable(); 4548e3adda8SFaris Rehman const unsigned includedFlagsBitmask = clang::driver::options::FC1Option; 455257b2971SCaroline Concatto unsigned missingArgIndex, missingArgCount; 456257b2971SCaroline Concatto llvm::opt::InputArgList args = opts.ParseArgs( 457257b2971SCaroline Concatto commandLineArgs, missingArgIndex, missingArgCount, includedFlagsBitmask); 458257b2971SCaroline Concatto 459257b2971SCaroline Concatto // Issue errors on unknown arguments 460257b2971SCaroline Concatto for (const auto *a : args.filtered(clang::driver::options::OPT_UNKNOWN)) { 461257b2971SCaroline Concatto auto argString = a->getAsString(args); 462257b2971SCaroline Concatto std::string nearest; 463257b2971SCaroline Concatto if (opts.findNearest(argString, nearest, includedFlagsBitmask) > 1) 464257b2971SCaroline Concatto diags.Report(clang::diag::err_drv_unknown_argument) << argString; 465257b2971SCaroline Concatto else 466257b2971SCaroline Concatto diags.Report(clang::diag::err_drv_unknown_argument_with_suggestion) 467257b2971SCaroline Concatto << argString << nearest; 468257b2971SCaroline Concatto success = false; 469257b2971SCaroline Concatto } 470257b2971SCaroline Concatto 471257b2971SCaroline Concatto // Parse the frontend args 4729ffb5b04SAndrzej Warzynski ParseFrontendArgs(res.frontendOpts(), args, diags); 4737809fa20SFaris Rehman // Parse the preprocessor args 4747809fa20SFaris Rehman parsePreprocessorArgs(res.preprocessorOpts(), args); 475985a42fdSArnamoy Bhattacharyya // Parse semantic args 4761fd4beecSArnamoy Bhattacharyya parseSemaArgs(res, args, diags); 477ab971c29SArnamoy Bhattacharyya // Parse dialect arguments 478ab971c29SArnamoy Bhattacharyya parseDialectArgs(res, args, diags); 4797416e8a8SArnamoy Bhattacharyya // Parse diagnostic arguments 4807416e8a8SArnamoy Bhattacharyya parseDiagArgs(res, args, diags); 481257b2971SCaroline Concatto 482257b2971SCaroline Concatto return success; 483257b2971SCaroline Concatto } 484d28de0d7SCaroline Concatto 485b83a4450SAndrzej Warzynski void CompilerInvocation::collectMacroDefinitions() { 486b83a4450SAndrzej Warzynski auto &ppOpts = this->preprocessorOpts(); 487b83a4450SAndrzej Warzynski 4887809fa20SFaris Rehman for (unsigned i = 0, n = ppOpts.macros.size(); i != n; ++i) { 4897809fa20SFaris Rehman llvm::StringRef macro = ppOpts.macros[i].first; 4907809fa20SFaris Rehman bool isUndef = ppOpts.macros[i].second; 4917809fa20SFaris Rehman 4927809fa20SFaris Rehman std::pair<llvm::StringRef, llvm::StringRef> macroPair = macro.split('='); 4937809fa20SFaris Rehman llvm::StringRef macroName = macroPair.first; 4947809fa20SFaris Rehman llvm::StringRef macroBody = macroPair.second; 4957809fa20SFaris Rehman 4967809fa20SFaris Rehman // For an #undef'd macro, we only care about the name. 4977809fa20SFaris Rehman if (isUndef) { 498b83a4450SAndrzej Warzynski parserOpts_.predefinitions.emplace_back( 4997809fa20SFaris Rehman macroName.str(), std::optional<std::string>{}); 5007809fa20SFaris Rehman continue; 5017809fa20SFaris Rehman } 5027809fa20SFaris Rehman 5037809fa20SFaris Rehman // For a #define'd macro, figure out the actual definition. 5047809fa20SFaris Rehman if (macroName.size() == macro.size()) 5057809fa20SFaris Rehman macroBody = "1"; 5067809fa20SFaris Rehman else { 5077809fa20SFaris Rehman // Note: GCC drops anything following an end-of-line character. 5087809fa20SFaris Rehman llvm::StringRef::size_type End = macroBody.find_first_of("\n\r"); 5097809fa20SFaris Rehman macroBody = macroBody.substr(0, End); 5107809fa20SFaris Rehman } 511b83a4450SAndrzej Warzynski parserOpts_.predefinitions.emplace_back( 5127809fa20SFaris Rehman macroName, std::optional<std::string>(macroBody.str())); 5137809fa20SFaris Rehman } 5147809fa20SFaris Rehman } 5157809fa20SFaris Rehman 516d28de0d7SCaroline Concatto void CompilerInvocation::SetDefaultFortranOpts() { 5176bbbe4a5SAndrzej Warzynski auto &fortranOptions = fortranOpts(); 518d28de0d7SCaroline Concatto 519d28de0d7SCaroline Concatto // These defaults are based on the defaults in f18/f18.cpp. 520d28de0d7SCaroline Concatto std::vector<std::string> searchDirectories{"."s}; 521d28de0d7SCaroline Concatto fortranOptions.searchDirectories = searchDirectories; 522d28de0d7SCaroline Concatto fortranOptions.isFixedForm = false; 5230feff71eSAndrzej Warzynski } 5240feff71eSAndrzej Warzynski 5250feff71eSAndrzej Warzynski // TODO: When expanding this method, consider creating a dedicated API for 5260feff71eSAndrzej Warzynski // this. Also at some point we will need to differentiate between different 5270feff71eSAndrzej Warzynski // targets and add dedicated predefines for each. 5280feff71eSAndrzej Warzynski void CompilerInvocation::setDefaultPredefinitions() { 5290feff71eSAndrzej Warzynski auto &fortranOptions = fortranOpts(); 5300feff71eSAndrzej Warzynski const auto &frontendOptions = frontendOpts(); 531197d9a55SFaris Rehman 532197d9a55SFaris Rehman // Populate the macro list with version numbers and other predefinitions. 533197d9a55SFaris Rehman fortranOptions.predefinitions.emplace_back("__flang__", "1"); 534197d9a55SFaris Rehman fortranOptions.predefinitions.emplace_back( 535197d9a55SFaris Rehman "__flang_major__", FLANG_VERSION_MAJOR_STRING); 536197d9a55SFaris Rehman fortranOptions.predefinitions.emplace_back( 537197d9a55SFaris Rehman "__flang_minor__", FLANG_VERSION_MINOR_STRING); 538197d9a55SFaris Rehman fortranOptions.predefinitions.emplace_back( 539197d9a55SFaris Rehman "__flang_patchlevel__", FLANG_VERSION_PATCHLEVEL_STRING); 5406d48a1a5SFaris Rehman 5416d48a1a5SFaris Rehman // Add predefinitions based on extensions enabled 5426d48a1a5SFaris Rehman if (frontendOptions.features_.IsEnabled( 5436d48a1a5SFaris Rehman Fortran::common::LanguageFeature::OpenACC)) { 5446d48a1a5SFaris Rehman fortranOptions.predefinitions.emplace_back("_OPENACC", "202011"); 5456d48a1a5SFaris Rehman } 5466d48a1a5SFaris Rehman if (frontendOptions.features_.IsEnabled( 5476d48a1a5SFaris Rehman Fortran::common::LanguageFeature::OpenMP)) { 5486d48a1a5SFaris Rehman fortranOptions.predefinitions.emplace_back("_OPENMP", "201511"); 5496d48a1a5SFaris Rehman } 5506d48a1a5SFaris Rehman } 5516d48a1a5SFaris Rehman 5527809fa20SFaris Rehman void CompilerInvocation::setFortranOpts() { 5537809fa20SFaris Rehman auto &fortranOptions = fortranOpts(); 5543a1513c1SFaris Rehman const auto &frontendOptions = frontendOpts(); 5557809fa20SFaris Rehman const auto &preprocessorOptions = preprocessorOpts(); 556985a42fdSArnamoy Bhattacharyya auto &moduleDirJ = moduleDir(); 5577809fa20SFaris Rehman 5583a1513c1SFaris Rehman if (frontendOptions.fortranForm_ != FortranForm::Unknown) { 5593a1513c1SFaris Rehman fortranOptions.isFixedForm = 5603a1513c1SFaris Rehman frontendOptions.fortranForm_ == FortranForm::FixedForm; 5613a1513c1SFaris Rehman } 5623a1513c1SFaris Rehman fortranOptions.fixedFormColumns = frontendOptions.fixedFormColumns_; 5633a1513c1SFaris Rehman 5646d48a1a5SFaris Rehman fortranOptions.features = frontendOptions.features_; 56510826ea7SFaris Rehman fortranOptions.encoding = frontendOptions.encoding_; 5666d48a1a5SFaris Rehman 567cd4abc52SArnamoy Bhattacharyya // Adding search directories specified by -I 56887dfd5e0SFaris Rehman fortranOptions.searchDirectories.insert( 56987dfd5e0SFaris Rehman fortranOptions.searchDirectories.end(), 57087dfd5e0SFaris Rehman preprocessorOptions.searchDirectoriesFromDashI.begin(), 57187dfd5e0SFaris Rehman preprocessorOptions.searchDirectoriesFromDashI.end()); 572985a42fdSArnamoy Bhattacharyya 573cd4abc52SArnamoy Bhattacharyya // Add the ordered list of -intrinsic-modules-path 574cd4abc52SArnamoy Bhattacharyya fortranOptions.searchDirectories.insert( 575cd4abc52SArnamoy Bhattacharyya fortranOptions.searchDirectories.end(), 576cd4abc52SArnamoy Bhattacharyya preprocessorOptions.searchDirectoriesFromIntrModPath.begin(), 577cd4abc52SArnamoy Bhattacharyya preprocessorOptions.searchDirectoriesFromIntrModPath.end()); 578cd4abc52SArnamoy Bhattacharyya 579cd4abc52SArnamoy Bhattacharyya // Add the default intrinsic module directory at the end 580cd4abc52SArnamoy Bhattacharyya fortranOptions.searchDirectories.emplace_back(getIntrinsicDir()); 581cd4abc52SArnamoy Bhattacharyya 582985a42fdSArnamoy Bhattacharyya // Add the directory supplied through -J/-module-dir to the list of search 583985a42fdSArnamoy Bhattacharyya // directories 584985a42fdSArnamoy Bhattacharyya if (moduleDirJ.compare(".") != 0) 585985a42fdSArnamoy Bhattacharyya fortranOptions.searchDirectories.emplace_back(moduleDirJ); 586523d7bc6SAndrzej Warzynski 587523d7bc6SAndrzej Warzynski if (frontendOptions.instrumentedParse_) 588523d7bc6SAndrzej Warzynski fortranOptions.instrumentedParse = true; 5894c7ebf79SArnamoy Bhattacharyya 590*85b86c6fSAndrzej Warzynski if (frontendOptions.needProvenanceRangeToCharBlockMappings_) 591*85b86c6fSAndrzej Warzynski fortranOptions.needProvenanceRangeToCharBlockMappings = true; 592*85b86c6fSAndrzej Warzynski 5934c7ebf79SArnamoy Bhattacharyya if (enableConformanceChecks()) { 5944c7ebf79SArnamoy Bhattacharyya fortranOptions.features.WarnOnAllNonstandard(); 5954c7ebf79SArnamoy Bhattacharyya } 596985a42fdSArnamoy Bhattacharyya } 597985a42fdSArnamoy Bhattacharyya 598985a42fdSArnamoy Bhattacharyya void CompilerInvocation::setSemanticsOpts( 5996d48a1a5SFaris Rehman Fortran::parser::AllCookedSources &allCookedSources) { 6006d48a1a5SFaris Rehman const auto &fortranOptions = fortranOpts(); 6016d48a1a5SFaris Rehman 6026d48a1a5SFaris Rehman semanticsContext_ = std::make_unique<semantics::SemanticsContext>( 603ab971c29SArnamoy Bhattacharyya defaultKinds(), fortranOptions.features, allCookedSources); 6046d48a1a5SFaris Rehman 6051fd4beecSArnamoy Bhattacharyya semanticsContext_->set_moduleDirectory(moduleDir()) 6064c7ebf79SArnamoy Bhattacharyya .set_searchDirectories(fortranOptions.searchDirectories) 6077416e8a8SArnamoy Bhattacharyya .set_warnOnNonstandardUsage(enableConformanceChecks()) 6087416e8a8SArnamoy Bhattacharyya .set_warningsAreErrors(warnAsErr()); 6097809fa20SFaris Rehman } 610