1 //===- FrontendOptions.cpp ------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #include "flang/Frontend/FrontendOptions.h"
14 
15 using namespace Fortran::frontend;
16 
isFixedFormSuffix(llvm::StringRef suffix)17 bool Fortran::frontend::isFixedFormSuffix(llvm::StringRef suffix) {
18   // Note: Keep this list in-sync with flang/test/lit.cfg.py
19   return suffix == "f77" || suffix == "f" || suffix == "F" || suffix == "ff" ||
20       suffix == "for" || suffix == "FOR" || suffix == "fpp" || suffix == "FPP";
21 }
22 
isFreeFormSuffix(llvm::StringRef suffix)23 bool Fortran::frontend::isFreeFormSuffix(llvm::StringRef suffix) {
24   // Note: Keep this list in-sync with flang/test/lit.cfg.py
25   // TODO: Add Cuda Fortan files (i.e. `*.cuf` and `*.CUF`).
26   return suffix == "f90" || suffix == "F90" || suffix == "ff90" ||
27       suffix == "f95" || suffix == "F95" || suffix == "ff95" ||
28       suffix == "f03" || suffix == "F03" || suffix == "f08" ||
29       suffix == "F08" || suffix == "f18" || suffix == "F18";
30 }
31 
isToBePreprocessed(llvm::StringRef suffix)32 bool Fortran::frontend::isToBePreprocessed(llvm::StringRef suffix) {
33   return suffix == "F" || suffix == "FOR" || suffix == "fpp" ||
34       suffix == "FPP" || suffix == "F90" || suffix == "F95" ||
35       suffix == "F03" || suffix == "F08" || suffix == "F18";
36 }
37 
getInputKindForExtension(llvm::StringRef extension)38 InputKind FrontendOptions::getInputKindForExtension(llvm::StringRef extension) {
39   if (isFixedFormSuffix(extension) || isFreeFormSuffix(extension)) {
40     return Language::Fortran;
41   }
42 
43   if (extension == "bc" || extension == "ll")
44     return Language::LLVM_IR;
45   if (extension == "fir" || extension == "mlir")
46     return Language::MLIR;
47 
48   return Language::Unknown;
49 }
50