164ab3302SCarolineConcatto //===-- lib/Semantics/expression.cpp --------------------------------------===//
264ab3302SCarolineConcatto //
364ab3302SCarolineConcatto // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
464ab3302SCarolineConcatto // See https://llvm.org/LICENSE.txt for license information.
564ab3302SCarolineConcatto // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
664ab3302SCarolineConcatto //
764ab3302SCarolineConcatto //===----------------------------------------------------------------------===//
864ab3302SCarolineConcatto 
964ab3302SCarolineConcatto #include "flang/Semantics/expression.h"
1064ab3302SCarolineConcatto #include "check-call.h"
1164ab3302SCarolineConcatto #include "pointer-assignment.h"
120c190575SPeter Klausler #include "resolve-names-utils.h"
139977b24aSpeter klausler #include "resolve-names.h"
14ae0d1d2eSPeter Steinfeld #include "flang/Common/Fortran.h"
1564ab3302SCarolineConcatto #include "flang/Common/idioms.h"
1664ab3302SCarolineConcatto #include "flang/Evaluate/common.h"
1764ab3302SCarolineConcatto #include "flang/Evaluate/fold.h"
1864ab3302SCarolineConcatto #include "flang/Evaluate/tools.h"
1964ab3302SCarolineConcatto #include "flang/Parser/characters.h"
2064ab3302SCarolineConcatto #include "flang/Parser/dump-parse-tree.h"
2164ab3302SCarolineConcatto #include "flang/Parser/parse-tree-visitor.h"
2264ab3302SCarolineConcatto #include "flang/Parser/parse-tree.h"
2364ab3302SCarolineConcatto #include "flang/Semantics/scope.h"
2464ab3302SCarolineConcatto #include "flang/Semantics/semantics.h"
2564ab3302SCarolineConcatto #include "flang/Semantics/symbol.h"
2664ab3302SCarolineConcatto #include "flang/Semantics/tools.h"
278670e499SCaroline Concatto #include "llvm/Support/raw_ostream.h"
2864ab3302SCarolineConcatto #include <algorithm>
2964ab3302SCarolineConcatto #include <functional>
3064ab3302SCarolineConcatto #include <optional>
3164ab3302SCarolineConcatto #include <set>
3264ab3302SCarolineConcatto 
3364ab3302SCarolineConcatto // Typedef for optional generic expressions (ubiquitous in this file)
3464ab3302SCarolineConcatto using MaybeExpr =
3564ab3302SCarolineConcatto     std::optional<Fortran::evaluate::Expr<Fortran::evaluate::SomeType>>;
3664ab3302SCarolineConcatto 
3764ab3302SCarolineConcatto // Much of the code that implements semantic analysis of expressions is
3864ab3302SCarolineConcatto // tightly coupled with their typed representations in lib/Evaluate,
3964ab3302SCarolineConcatto // and appears here in namespace Fortran::evaluate for convenience.
4064ab3302SCarolineConcatto namespace Fortran::evaluate {
4164ab3302SCarolineConcatto 
4264ab3302SCarolineConcatto using common::LanguageFeature;
4364ab3302SCarolineConcatto using common::NumericOperator;
4464ab3302SCarolineConcatto using common::TypeCategory;
4564ab3302SCarolineConcatto 
ToUpperCase(const std::string & str)4664ab3302SCarolineConcatto static inline std::string ToUpperCase(const std::string &str) {
4764ab3302SCarolineConcatto   return parser::ToUpperCaseLetters(str);
4864ab3302SCarolineConcatto }
4964ab3302SCarolineConcatto 
5064ab3302SCarolineConcatto struct DynamicTypeWithLength : public DynamicType {
DynamicTypeWithLengthFortran::evaluate::DynamicTypeWithLength5164ab3302SCarolineConcatto   explicit DynamicTypeWithLength(const DynamicType &t) : DynamicType{t} {}
5264ab3302SCarolineConcatto   std::optional<Expr<SubscriptInteger>> LEN() const;
5364ab3302SCarolineConcatto   std::optional<Expr<SubscriptInteger>> length;
5464ab3302SCarolineConcatto };
5564ab3302SCarolineConcatto 
LEN() const5664ab3302SCarolineConcatto std::optional<Expr<SubscriptInteger>> DynamicTypeWithLength::LEN() const {
5764ab3302SCarolineConcatto   if (length) {
5864ab3302SCarolineConcatto     return length;
59ac964175Speter klausler   } else {
60ac964175Speter klausler     return GetCharLength();
6164ab3302SCarolineConcatto   }
6264ab3302SCarolineConcatto }
6364ab3302SCarolineConcatto 
AnalyzeTypeSpec(const std::optional<parser::TypeSpec> & spec)6464ab3302SCarolineConcatto static std::optional<DynamicTypeWithLength> AnalyzeTypeSpec(
6564ab3302SCarolineConcatto     const std::optional<parser::TypeSpec> &spec) {
6664ab3302SCarolineConcatto   if (spec) {
6764ab3302SCarolineConcatto     if (const semantics::DeclTypeSpec * typeSpec{spec->declTypeSpec}) {
6864ab3302SCarolineConcatto       // Name resolution sets TypeSpec::declTypeSpec only when it's valid
6964ab3302SCarolineConcatto       // (viz., an intrinsic type with valid known kind or a non-polymorphic
7064ab3302SCarolineConcatto       // & non-ABSTRACT derived type).
7164ab3302SCarolineConcatto       if (const semantics::IntrinsicTypeSpec *
7264ab3302SCarolineConcatto           intrinsic{typeSpec->AsIntrinsic()}) {
7364ab3302SCarolineConcatto         TypeCategory category{intrinsic->category()};
7464ab3302SCarolineConcatto         if (auto optKind{ToInt64(intrinsic->kind())}) {
7564ab3302SCarolineConcatto           int kind{static_cast<int>(*optKind)};
7664ab3302SCarolineConcatto           if (category == TypeCategory::Character) {
7764ab3302SCarolineConcatto             const semantics::CharacterTypeSpec &cts{
7864ab3302SCarolineConcatto                 typeSpec->characterTypeSpec()};
7964ab3302SCarolineConcatto             const semantics::ParamValue &len{cts.length()};
8064ab3302SCarolineConcatto             // N.B. CHARACTER(LEN=*) is allowed in type-specs in ALLOCATE() &
8164ab3302SCarolineConcatto             // type guards, but not in array constructors.
8264ab3302SCarolineConcatto             return DynamicTypeWithLength{DynamicType{kind, len}};
8364ab3302SCarolineConcatto           } else {
8464ab3302SCarolineConcatto             return DynamicTypeWithLength{DynamicType{category, kind}};
8564ab3302SCarolineConcatto           }
8664ab3302SCarolineConcatto         }
8764ab3302SCarolineConcatto       } else if (const semantics::DerivedTypeSpec *
8864ab3302SCarolineConcatto           derived{typeSpec->AsDerived()}) {
8964ab3302SCarolineConcatto         return DynamicTypeWithLength{DynamicType{*derived}};
9064ab3302SCarolineConcatto       }
9164ab3302SCarolineConcatto     }
9264ab3302SCarolineConcatto   }
9364ab3302SCarolineConcatto   return std::nullopt;
9464ab3302SCarolineConcatto }
9564ab3302SCarolineConcatto 
96fa0443f2SPeter Klausler // Utilities to set a source location, if we have one, on an actual argument,
97fa0443f2SPeter Klausler // when it is statically present.
SetArgSourceLocation(ActualArgument & x,parser::CharBlock at)98fa0443f2SPeter Klausler static void SetArgSourceLocation(ActualArgument &x, parser::CharBlock at) {
99fa0443f2SPeter Klausler   x.set_sourceLocation(at);
100fa0443f2SPeter Klausler }
SetArgSourceLocation(std::optional<ActualArgument> & x,parser::CharBlock at)101fa0443f2SPeter Klausler static void SetArgSourceLocation(
102fa0443f2SPeter Klausler     std::optional<ActualArgument> &x, parser::CharBlock at) {
103fa0443f2SPeter Klausler   if (x) {
104fa0443f2SPeter Klausler     x->set_sourceLocation(at);
105fa0443f2SPeter Klausler   }
106fa0443f2SPeter Klausler }
SetArgSourceLocation(std::optional<ActualArgument> & x,std::optional<parser::CharBlock> at)107fa0443f2SPeter Klausler static void SetArgSourceLocation(
108fa0443f2SPeter Klausler     std::optional<ActualArgument> &x, std::optional<parser::CharBlock> at) {
109fa0443f2SPeter Klausler   if (x && at) {
110fa0443f2SPeter Klausler     x->set_sourceLocation(*at);
111fa0443f2SPeter Klausler   }
112fa0443f2SPeter Klausler }
113fa0443f2SPeter Klausler 
11464ab3302SCarolineConcatto class ArgumentAnalyzer {
11564ab3302SCarolineConcatto public:
ArgumentAnalyzer(ExpressionAnalyzer & context)11664ab3302SCarolineConcatto   explicit ArgumentAnalyzer(ExpressionAnalyzer &context)
117934b27a9Speter klausler       : context_{context}, source_{context.GetContextualMessages().at()},
118934b27a9Speter klausler         isProcedureCall_{false} {}
ArgumentAnalyzer(ExpressionAnalyzer & context,parser::CharBlock source,bool isProcedureCall=false)11964ab3302SCarolineConcatto   ArgumentAnalyzer(ExpressionAnalyzer &context, parser::CharBlock source,
120a3538b83STim Keith       bool isProcedureCall = false)
121a3538b83STim Keith       : context_{context}, source_{source}, isProcedureCall_{isProcedureCall} {}
fatalErrors() const12264ab3302SCarolineConcatto   bool fatalErrors() const { return fatalErrors_; }
GetActuals()12364ab3302SCarolineConcatto   ActualArguments &&GetActuals() {
12464ab3302SCarolineConcatto     CHECK(!fatalErrors_);
12564ab3302SCarolineConcatto     return std::move(actuals_);
12664ab3302SCarolineConcatto   }
GetExpr(std::size_t i) const12764ab3302SCarolineConcatto   const Expr<SomeType> &GetExpr(std::size_t i) const {
12864ab3302SCarolineConcatto     return DEREF(actuals_.at(i).value().UnwrapExpr());
12964ab3302SCarolineConcatto   }
MoveExpr(std::size_t i)13064ab3302SCarolineConcatto   Expr<SomeType> &&MoveExpr(std::size_t i) {
13164ab3302SCarolineConcatto     return std::move(DEREF(actuals_.at(i).value().UnwrapExpr()));
13264ab3302SCarolineConcatto   }
Analyze(const common::Indirection<parser::Expr> & x)13364ab3302SCarolineConcatto   void Analyze(const common::Indirection<parser::Expr> &x) {
13464ab3302SCarolineConcatto     Analyze(x.value());
13564ab3302SCarolineConcatto   }
Analyze(const parser::Expr & x)13664ab3302SCarolineConcatto   void Analyze(const parser::Expr &x) {
13764ab3302SCarolineConcatto     actuals_.emplace_back(AnalyzeExpr(x));
138fa0443f2SPeter Klausler     SetArgSourceLocation(actuals_.back(), x.source);
13964ab3302SCarolineConcatto     fatalErrors_ |= !actuals_.back();
14064ab3302SCarolineConcatto   }
14164ab3302SCarolineConcatto   void Analyze(const parser::Variable &);
14264ab3302SCarolineConcatto   void Analyze(const parser::ActualArgSpec &, bool isSubroutine);
143d9195d66Speter klausler   void ConvertBOZ(std::optional<DynamicType> &thisType, std::size_t i,
144d9195d66Speter klausler       std::optional<DynamicType> otherType);
14564ab3302SCarolineConcatto 
146d9195d66Speter klausler   bool IsIntrinsicRelational(
147d9195d66Speter klausler       RelationalOperator, const DynamicType &, const DynamicType &) const;
14864ab3302SCarolineConcatto   bool IsIntrinsicLogical() const;
14964ab3302SCarolineConcatto   bool IsIntrinsicNumeric(NumericOperator) const;
15064ab3302SCarolineConcatto   bool IsIntrinsicConcat() const;
15164ab3302SCarolineConcatto 
152d9195d66Speter klausler   bool CheckConformance();
1530fcda9aeSpeter klausler   bool CheckForNullPointer(const char *where = "as an operand here");
1541a633e72Speter klausler 
15564ab3302SCarolineConcatto   // Find and return a user-defined operator or report an error.
15664ab3302SCarolineConcatto   // The provided message is used if there is no such operator.
157d9195d66Speter klausler   MaybeExpr TryDefinedOp(const char *, parser::MessageFixedText,
158d9195d66Speter klausler       const Symbol **definedOpSymbolPtr = nullptr, bool isUserOp = false);
15964ab3302SCarolineConcatto   template <typename E>
TryDefinedOp(E opr,parser::MessageFixedText msg)160d9195d66Speter klausler   MaybeExpr TryDefinedOp(E opr, parser::MessageFixedText msg) {
16164ab3302SCarolineConcatto     return TryDefinedOp(
162d9195d66Speter klausler         context_.context().languageFeatures().GetNames(opr), msg);
16364ab3302SCarolineConcatto   }
16464ab3302SCarolineConcatto   // Find and return a user-defined assignment
16564ab3302SCarolineConcatto   std::optional<ProcedureRef> TryDefinedAssignment();
16664ab3302SCarolineConcatto   std::optional<ProcedureRef> GetDefinedAssignmentProc();
16783dca19cSPete Steinfeld   std::optional<DynamicType> GetType(std::size_t) const;
1688670e499SCaroline Concatto   void Dump(llvm::raw_ostream &);
16964ab3302SCarolineConcatto 
17064ab3302SCarolineConcatto private:
171d9195d66Speter klausler   MaybeExpr TryDefinedOp(std::vector<const char *>, parser::MessageFixedText);
17264ab3302SCarolineConcatto   MaybeExpr TryBoundOp(const Symbol &, int passIndex);
17364ab3302SCarolineConcatto   std::optional<ActualArgument> AnalyzeExpr(const parser::Expr &);
17488afb6e8Speter klausler   MaybeExpr AnalyzeExprOrWholeAssumedSizeArray(const parser::Expr &);
17564ab3302SCarolineConcatto   bool AreConformable() const;
176d9195d66Speter klausler   const Symbol *FindBoundOp(
177d9195d66Speter klausler       parser::CharBlock, int passIndex, const Symbol *&definedOp);
178b05c8c57STim Keith   void AddAssignmentConversion(
179b05c8c57STim Keith       const DynamicType &lhsType, const DynamicType &rhsType);
18064ab3302SCarolineConcatto   bool OkLogicalIntegerAssignment(TypeCategory lhs, TypeCategory rhs);
18164ab3302SCarolineConcatto   int GetRank(std::size_t) const;
IsBOZLiteral(std::size_t i) const18264ab3302SCarolineConcatto   bool IsBOZLiteral(std::size_t i) const {
183571673ceSPeter Steinfeld     return evaluate::IsBOZLiteral(GetExpr(i));
18464ab3302SCarolineConcatto   }
18564ab3302SCarolineConcatto   void SayNoMatch(const std::string &, bool isAssignment = false);
18664ab3302SCarolineConcatto   std::string TypeAsFortran(std::size_t);
1871bd083b5Speter klausler   bool AnyUntypedOrMissingOperand();
18864ab3302SCarolineConcatto 
18964ab3302SCarolineConcatto   ExpressionAnalyzer &context_;
19064ab3302SCarolineConcatto   ActualArguments actuals_;
19164ab3302SCarolineConcatto   parser::CharBlock source_;
19264ab3302SCarolineConcatto   bool fatalErrors_{false};
193a3538b83STim Keith   const bool isProcedureCall_; // false for user-defined op or assignment
19464ab3302SCarolineConcatto };
19564ab3302SCarolineConcatto 
19664ab3302SCarolineConcatto // Wraps a data reference in a typed Designator<>, and a procedure
19764ab3302SCarolineConcatto // or procedure pointer reference in a ProcedureDesignator.
Designate(DataRef && ref)19864ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Designate(DataRef &&ref) {
1993061334eSpeter klausler   const Symbol &last{ref.GetLastSymbol()};
2003cc5d4ffSpeter klausler   const Symbol &symbol{BypassGeneric(last).GetUltimate()};
20164ab3302SCarolineConcatto   if (semantics::IsProcedure(symbol)) {
20264ab3302SCarolineConcatto     if (auto *component{std::get_if<Component>(&ref.u)}) {
20364ab3302SCarolineConcatto       return Expr<SomeType>{ProcedureDesignator{std::move(*component)}};
20464ab3302SCarolineConcatto     } else if (!std::holds_alternative<SymbolRef>(ref.u)) {
20564ab3302SCarolineConcatto       DIE("unexpected alternative in DataRef");
20664ab3302SCarolineConcatto     } else if (!symbol.attrs().test(semantics::Attr::INTRINSIC)) {
2073cc5d4ffSpeter klausler       if (symbol.has<semantics::GenericDetails>()) {
2083cc5d4ffSpeter klausler         Say("'%s' is not a specific procedure"_err_en_US, symbol.name());
2093cc5d4ffSpeter klausler       } else {
21064ab3302SCarolineConcatto         return Expr<SomeType>{ProcedureDesignator{symbol}};
2113cc5d4ffSpeter klausler       }
21264ab3302SCarolineConcatto     } else if (auto interface{context_.intrinsics().IsSpecificIntrinsicFunction(
213848cca6cSEmil Kieri                    symbol.name().ToString())};
214848cca6cSEmil Kieri                interface && !interface->isRestrictedSpecific) {
21564ab3302SCarolineConcatto       SpecificIntrinsic intrinsic{
21664ab3302SCarolineConcatto           symbol.name().ToString(), std::move(*interface)};
21764ab3302SCarolineConcatto       intrinsic.isRestrictedSpecific = interface->isRestrictedSpecific;
21864ab3302SCarolineConcatto       return Expr<SomeType>{ProcedureDesignator{std::move(intrinsic)}};
21964ab3302SCarolineConcatto     } else {
220848cca6cSEmil Kieri       Say("'%s' is not an unrestricted specific intrinsic procedure"_err_en_US,
22164ab3302SCarolineConcatto           symbol.name());
22264ab3302SCarolineConcatto     }
22364ab3302SCarolineConcatto     return std::nullopt;
2243061334eSpeter klausler   } else if (MaybeExpr result{AsGenericExpr(std::move(ref))}) {
2253061334eSpeter klausler     return result;
2266829bd3eSpeter klausler   } else {
2273061334eSpeter klausler     if (!context_.HasError(last) && !context_.HasError(symbol)) {
2283061334eSpeter klausler       AttachDeclaration(
2293061334eSpeter klausler           Say("'%s' is not an object that can appear in an expression"_err_en_US,
2303061334eSpeter klausler               last.name()),
2313061334eSpeter klausler           symbol);
2323061334eSpeter klausler       context_.SetError(last);
2333061334eSpeter klausler     }
2343061334eSpeter klausler     return std::nullopt;
2356829bd3eSpeter klausler   }
23664ab3302SCarolineConcatto }
23764ab3302SCarolineConcatto 
23864ab3302SCarolineConcatto // Some subscript semantic checks must be deferred until all of the
23964ab3302SCarolineConcatto // subscripts are in hand.
CompleteSubscripts(ArrayRef && ref)24064ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::CompleteSubscripts(ArrayRef &&ref) {
24164ab3302SCarolineConcatto   const Symbol &symbol{ref.GetLastSymbol().GetUltimate()};
24264ab3302SCarolineConcatto   int symbolRank{symbol.Rank()};
24364ab3302SCarolineConcatto   int subscripts{static_cast<int>(ref.size())};
24464ab3302SCarolineConcatto   if (subscripts == 0) {
24593d10919Speter klausler     return std::nullopt; // error recovery
24664ab3302SCarolineConcatto   } else if (subscripts != symbolRank) {
2473495c782STim Keith     if (symbolRank != 0) {
24864ab3302SCarolineConcatto       Say("Reference to rank-%d object '%s' has %d subscripts"_err_en_US,
24964ab3302SCarolineConcatto           symbolRank, symbol.name(), subscripts);
2503495c782STim Keith     }
25164ab3302SCarolineConcatto     return std::nullopt;
25293d10919Speter klausler   } else if (const auto *object{
25393d10919Speter klausler                  symbol.detailsIf<semantics::ObjectEntityDetails>()}) {
25464ab3302SCarolineConcatto     // C928 & C1002
25564ab3302SCarolineConcatto     if (Triplet * last{std::get_if<Triplet>(&ref.subscript().back().u)}) {
25664ab3302SCarolineConcatto       if (!last->upper() && object->IsAssumedSize()) {
25764ab3302SCarolineConcatto         Say("Assumed-size array '%s' must have explicit final "
25864ab3302SCarolineConcatto             "subscript upper bound value"_err_en_US,
25964ab3302SCarolineConcatto             symbol.name());
26064ab3302SCarolineConcatto         return std::nullopt;
26164ab3302SCarolineConcatto       }
26264ab3302SCarolineConcatto     }
26393d10919Speter klausler   } else {
26493d10919Speter klausler     // Shouldn't get here from Analyze(ArrayElement) without a valid base,
26593d10919Speter klausler     // which, if not an object, must be a construct entity from
26693d10919Speter klausler     // SELECT TYPE/RANK or ASSOCIATE.
26793d10919Speter klausler     CHECK(symbol.has<semantics::AssocEntityDetails>());
26864ab3302SCarolineConcatto   }
26964ab3302SCarolineConcatto   return Designate(DataRef{std::move(ref)});
27064ab3302SCarolineConcatto }
27164ab3302SCarolineConcatto 
27264ab3302SCarolineConcatto // Applies subscripts to a data reference.
ApplySubscripts(DataRef && dataRef,std::vector<Subscript> && subscripts)27364ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::ApplySubscripts(
27464ab3302SCarolineConcatto     DataRef &&dataRef, std::vector<Subscript> &&subscripts) {
27593d10919Speter klausler   if (subscripts.empty()) {
27693d10919Speter klausler     return std::nullopt; // error recovery
27793d10919Speter klausler   }
278cd03e96fSPeter Klausler   return common::visit(
27964ab3302SCarolineConcatto       common::visitors{
28064ab3302SCarolineConcatto           [&](SymbolRef &&symbol) {
28164ab3302SCarolineConcatto             return CompleteSubscripts(ArrayRef{symbol, std::move(subscripts)});
28264ab3302SCarolineConcatto           },
28364ab3302SCarolineConcatto           [&](Component &&c) {
28464ab3302SCarolineConcatto             return CompleteSubscripts(
28564ab3302SCarolineConcatto                 ArrayRef{std::move(c), std::move(subscripts)});
28664ab3302SCarolineConcatto           },
28764ab3302SCarolineConcatto           [&](auto &&) -> MaybeExpr {
28864ab3302SCarolineConcatto             DIE("bad base for ArrayRef");
28964ab3302SCarolineConcatto             return std::nullopt;
29064ab3302SCarolineConcatto           },
29164ab3302SCarolineConcatto       },
29264ab3302SCarolineConcatto       std::move(dataRef.u));
29364ab3302SCarolineConcatto }
29464ab3302SCarolineConcatto 
29548a70ea1SPeixin-Qiao // C919a - only one part-ref of a data-ref may have rank > 0
CheckRanks(const DataRef & dataRef)29648a70ea1SPeixin-Qiao bool ExpressionAnalyzer::CheckRanks(const DataRef &dataRef) {
29748a70ea1SPeixin-Qiao   return common::visit(
29848a70ea1SPeixin-Qiao       common::visitors{
29948a70ea1SPeixin-Qiao           [this](const Component &component) {
30048a70ea1SPeixin-Qiao             const Symbol &symbol{component.GetLastSymbol()};
30148a70ea1SPeixin-Qiao             if (int componentRank{symbol.Rank()}; componentRank > 0) {
30248a70ea1SPeixin-Qiao               if (int baseRank{component.base().Rank()}; baseRank > 0) {
30348a70ea1SPeixin-Qiao                 Say("Reference to whole rank-%d component '%s' of rank-%d array of derived type is not allowed"_err_en_US,
30464ab3302SCarolineConcatto                     componentRank, symbol.name(), baseRank);
30548a70ea1SPeixin-Qiao                 return false;
30648a70ea1SPeixin-Qiao               }
30748a70ea1SPeixin-Qiao             } else {
30848a70ea1SPeixin-Qiao               return CheckRanks(component.base());
30948a70ea1SPeixin-Qiao             }
31048a70ea1SPeixin-Qiao             return true;
31148a70ea1SPeixin-Qiao           },
31248a70ea1SPeixin-Qiao           [this](const ArrayRef &arrayRef) {
31348a70ea1SPeixin-Qiao             if (const auto *component{arrayRef.base().UnwrapComponent()}) {
31448a70ea1SPeixin-Qiao               int subscriptRank{0};
31548a70ea1SPeixin-Qiao               for (const Subscript &subscript : arrayRef.subscript()) {
31648a70ea1SPeixin-Qiao                 subscriptRank += subscript.Rank();
31748a70ea1SPeixin-Qiao               }
31848a70ea1SPeixin-Qiao               if (subscriptRank > 0) {
31948a70ea1SPeixin-Qiao                 if (int componentBaseRank{component->base().Rank()};
32048a70ea1SPeixin-Qiao                     componentBaseRank > 0) {
32148a70ea1SPeixin-Qiao                   Say("Subscripts of component '%s' of rank-%d derived type array have rank %d but must all be scalar"_err_en_US,
32248a70ea1SPeixin-Qiao                       component->GetLastSymbol().name(), componentBaseRank,
32348a70ea1SPeixin-Qiao                       subscriptRank);
32448a70ea1SPeixin-Qiao                   return false;
32548a70ea1SPeixin-Qiao                 }
32648a70ea1SPeixin-Qiao               } else {
32748a70ea1SPeixin-Qiao                 return CheckRanks(component->base());
32864ab3302SCarolineConcatto               }
32964ab3302SCarolineConcatto             }
33048a70ea1SPeixin-Qiao             return true;
33148a70ea1SPeixin-Qiao           },
33248a70ea1SPeixin-Qiao           [](const SymbolRef &) { return true; },
33348a70ea1SPeixin-Qiao           [](const CoarrayRef &) { return true; },
33448a70ea1SPeixin-Qiao       },
33548a70ea1SPeixin-Qiao       dataRef.u);
33664ab3302SCarolineConcatto }
33764ab3302SCarolineConcatto 
33864ab3302SCarolineConcatto // Parse tree correction after a substring S(j:k) was misparsed as an
339e03664d4SPeter Klausler // array section.  Fortran substrings must have a range, not a
34064ab3302SCarolineConcatto // single index.
FixMisparsedSubstringDataRef(parser::DataRef & dataRef)341e03664d4SPeter Klausler static std::optional<parser::Substring> FixMisparsedSubstringDataRef(
342e03664d4SPeter Klausler     parser::DataRef &dataRef) {
343e03664d4SPeter Klausler   if (auto *ae{
344e03664d4SPeter Klausler           std::get_if<common::Indirection<parser::ArrayElement>>(&dataRef.u)}) {
345e03664d4SPeter Klausler     // ...%a(j:k) and "a" is a character scalar
34664ab3302SCarolineConcatto     parser::ArrayElement &arrElement{ae->value()};
347e03664d4SPeter Klausler     if (arrElement.subscripts.size() == 1) {
348e03664d4SPeter Klausler       if (auto *triplet{std::get_if<parser::SubscriptTriplet>(
349e03664d4SPeter Klausler               &arrElement.subscripts.front().u)}) {
350e03664d4SPeter Klausler         if (!std::get<2 /*stride*/>(triplet->t).has_value()) {
351e03664d4SPeter Klausler           if (const Symbol *
352e03664d4SPeter Klausler               symbol{parser::GetLastName(arrElement.base).symbol}) {
35364ab3302SCarolineConcatto             const Symbol &ultimate{symbol->GetUltimate()};
35464ab3302SCarolineConcatto             if (const semantics::DeclTypeSpec * type{ultimate.GetType()}) {
35564ab3302SCarolineConcatto               if (!ultimate.IsObjectArray() &&
35664ab3302SCarolineConcatto                   type->category() == semantics::DeclTypeSpec::Character) {
35764ab3302SCarolineConcatto                 // The ambiguous S(j:k) was parsed as an array section
35864ab3302SCarolineConcatto                 // reference, but it's now clear that it's a substring.
35964ab3302SCarolineConcatto                 // Fix the parse tree in situ.
360e03664d4SPeter Klausler                 return arrElement.ConvertToSubstring();
36164ab3302SCarolineConcatto               }
36264ab3302SCarolineConcatto             }
36364ab3302SCarolineConcatto           }
36464ab3302SCarolineConcatto         }
36564ab3302SCarolineConcatto       }
36664ab3302SCarolineConcatto     }
36764ab3302SCarolineConcatto   }
368e03664d4SPeter Klausler   return std::nullopt;
36964ab3302SCarolineConcatto }
370e03664d4SPeter Klausler 
371e03664d4SPeter Klausler // When a designator is a misparsed type-param-inquiry of a misparsed
372e03664d4SPeter Klausler // substring -- it looks like a structure component reference of an array
373e03664d4SPeter Klausler // slice -- fix the substring and then convert to an intrinsic function
374e03664d4SPeter Klausler // call to KIND() or LEN().  And when the designator is a misparsed
375e03664d4SPeter Klausler // substring, convert it into a substring reference in place.
FixMisparsedSubstring(const parser::Designator & d)376e03664d4SPeter Klausler MaybeExpr ExpressionAnalyzer::FixMisparsedSubstring(
377e03664d4SPeter Klausler     const parser::Designator &d) {
378e03664d4SPeter Klausler   auto &mutate{const_cast<parser::Designator &>(d)};
379e03664d4SPeter Klausler   if (auto *dataRef{std::get_if<parser::DataRef>(&mutate.u)}) {
380e03664d4SPeter Klausler     if (auto *sc{std::get_if<common::Indirection<parser::StructureComponent>>(
381e03664d4SPeter Klausler             &dataRef->u)}) {
382e03664d4SPeter Klausler       parser::StructureComponent &structComponent{sc->value()};
383e03664d4SPeter Klausler       parser::CharBlock which{structComponent.component.source};
384e03664d4SPeter Klausler       if (which == "kind" || which == "len") {
385e03664d4SPeter Klausler         if (auto substring{
386e03664d4SPeter Klausler                 FixMisparsedSubstringDataRef(structComponent.base)}) {
387e03664d4SPeter Klausler           // ...%a(j:k)%kind or %len and "a" is a character scalar
388e03664d4SPeter Klausler           mutate.u = std::move(*substring);
389e03664d4SPeter Klausler           if (MaybeExpr substringExpr{Analyze(d)}) {
390e03664d4SPeter Klausler             return MakeFunctionRef(which,
391e03664d4SPeter Klausler                 ActualArguments{ActualArgument{std::move(*substringExpr)}});
392e03664d4SPeter Klausler           }
393e03664d4SPeter Klausler         }
394e03664d4SPeter Klausler       }
395e03664d4SPeter Klausler     } else if (auto substring{FixMisparsedSubstringDataRef(*dataRef)}) {
396e03664d4SPeter Klausler       mutate.u = std::move(*substring);
397e03664d4SPeter Klausler     }
398e03664d4SPeter Klausler   }
399e03664d4SPeter Klausler   return std::nullopt;
40064ab3302SCarolineConcatto }
40164ab3302SCarolineConcatto 
Analyze(const parser::Designator & d)40264ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Designator &d) {
40364ab3302SCarolineConcatto   auto restorer{GetContextualMessages().SetLocation(d.source)};
404e03664d4SPeter Klausler   if (auto substringInquiry{FixMisparsedSubstring(d)}) {
405*7aa77c5aSKazu Hirata     return substringInquiry;
406e03664d4SPeter Klausler   }
40764ab3302SCarolineConcatto   // These checks have to be deferred to these "top level" data-refs where
40864ab3302SCarolineConcatto   // we can be sure that there are no following subscripts (yet).
40964ab3302SCarolineConcatto   if (MaybeExpr result{Analyze(d.u)}) {
41064ab3302SCarolineConcatto     if (std::optional<DataRef> dataRef{ExtractDataRef(std::move(result))}) {
41148a70ea1SPeixin-Qiao       if (!CheckRanks(std::move(*dataRef))) {
41248a70ea1SPeixin-Qiao         return std::nullopt;
41348a70ea1SPeixin-Qiao       }
41448a70ea1SPeixin-Qiao       return Designate(std::move(*dataRef));
41548a70ea1SPeixin-Qiao     } else if (std::optional<DataRef> dataRef{
41648a70ea1SPeixin-Qiao                    ExtractDataRef(std::move(result), /*intoSubstring=*/true)}) {
41748a70ea1SPeixin-Qiao       if (!CheckRanks(std::move(*dataRef))) {
41848a70ea1SPeixin-Qiao         return std::nullopt;
41948a70ea1SPeixin-Qiao       }
42048a70ea1SPeixin-Qiao     } else if (std::optional<DataRef> dataRef{ExtractDataRef(std::move(result),
42148a70ea1SPeixin-Qiao                    /*intoSubstring=*/false, /*intoComplexPart=*/true)}) {
42248a70ea1SPeixin-Qiao       if (!CheckRanks(std::move(*dataRef))) {
42348a70ea1SPeixin-Qiao         return std::nullopt;
42448a70ea1SPeixin-Qiao       }
42564ab3302SCarolineConcatto     }
42664ab3302SCarolineConcatto     return result;
42764ab3302SCarolineConcatto   }
42864ab3302SCarolineConcatto   return std::nullopt;
42964ab3302SCarolineConcatto }
43064ab3302SCarolineConcatto 
43164ab3302SCarolineConcatto // A utility subroutine to repackage optional expressions of various levels
43264ab3302SCarolineConcatto // of type specificity as fully general MaybeExpr values.
AsMaybeExpr(A && x)43364ab3302SCarolineConcatto template <typename A> common::IfNoLvalue<MaybeExpr, A> AsMaybeExpr(A &&x) {
4344171f80dSpeter klausler   return AsGenericExpr(std::move(x));
43564ab3302SCarolineConcatto }
AsMaybeExpr(std::optional<A> && x)43664ab3302SCarolineConcatto template <typename A> MaybeExpr AsMaybeExpr(std::optional<A> &&x) {
43764ab3302SCarolineConcatto   if (x) {
43864ab3302SCarolineConcatto     return AsMaybeExpr(std::move(*x));
43964ab3302SCarolineConcatto   }
44064ab3302SCarolineConcatto   return std::nullopt;
44164ab3302SCarolineConcatto }
44264ab3302SCarolineConcatto 
44364ab3302SCarolineConcatto // Type kind parameter values for literal constants.
AnalyzeKindParam(const std::optional<parser::KindParam> & kindParam,int defaultKind)44464ab3302SCarolineConcatto int ExpressionAnalyzer::AnalyzeKindParam(
44564ab3302SCarolineConcatto     const std::optional<parser::KindParam> &kindParam, int defaultKind) {
44664ab3302SCarolineConcatto   if (!kindParam) {
44764ab3302SCarolineConcatto     return defaultKind;
44864ab3302SCarolineConcatto   }
449cd03e96fSPeter Klausler   return common::visit(
45064ab3302SCarolineConcatto       common::visitors{
45164ab3302SCarolineConcatto           [](std::uint64_t k) { return static_cast<int>(k); },
45264ab3302SCarolineConcatto           [&](const parser::Scalar<
45364ab3302SCarolineConcatto               parser::Integer<parser::Constant<parser::Name>>> &n) {
45464ab3302SCarolineConcatto             if (MaybeExpr ie{Analyze(n)}) {
45564ab3302SCarolineConcatto               if (std::optional<std::int64_t> i64{ToInt64(*ie)}) {
45664ab3302SCarolineConcatto                 int iv = *i64;
45764ab3302SCarolineConcatto                 if (iv == *i64) {
45864ab3302SCarolineConcatto                   return iv;
45964ab3302SCarolineConcatto                 }
46064ab3302SCarolineConcatto               }
46164ab3302SCarolineConcatto             }
46264ab3302SCarolineConcatto             return defaultKind;
46364ab3302SCarolineConcatto           },
46464ab3302SCarolineConcatto       },
46564ab3302SCarolineConcatto       kindParam->u);
46664ab3302SCarolineConcatto }
46764ab3302SCarolineConcatto 
46864ab3302SCarolineConcatto // Common handling of parser::IntLiteralConstant and SignedIntLiteralConstant
46964ab3302SCarolineConcatto struct IntTypeVisitor {
47064ab3302SCarolineConcatto   using Result = MaybeExpr;
47164ab3302SCarolineConcatto   using Types = IntegerTypes;
TestFortran::evaluate::IntTypeVisitor47264ab3302SCarolineConcatto   template <typename T> Result Test() {
47364ab3302SCarolineConcatto     if (T::kind >= kind) {
47464ab3302SCarolineConcatto       const char *p{digits.begin()};
47573506256SPeter Klausler       using Int = typename T::Scalar;
47673506256SPeter Klausler       typename Int::ValueWithOverflow num{0, false};
47773506256SPeter Klausler       if (isNegated) {
47873506256SPeter Klausler         auto unsignedNum{Int::Read(p, 10, false /*unsigned*/)};
47973506256SPeter Klausler         num.value = unsignedNum.value.Negate().value;
48073506256SPeter Klausler         num.overflow = unsignedNum.overflow || num.value > Int{0};
48173506256SPeter Klausler         if (!num.overflow && num.value.Negate().overflow &&
48273506256SPeter Klausler             !analyzer.context().IsInModuleFile(digits)) {
48373506256SPeter Klausler           analyzer.Say(digits,
48473506256SPeter Klausler               "negated maximum INTEGER(KIND=%d) literal"_port_en_US, T::kind);
48573506256SPeter Klausler         }
48673506256SPeter Klausler       } else {
48773506256SPeter Klausler         num = Int::Read(p, 10, true /*signed*/);
48873506256SPeter Klausler       }
48973506256SPeter Klausler       if (!num.overflow) {
49064ab3302SCarolineConcatto         if (T::kind > kind) {
49164ab3302SCarolineConcatto           if (!isDefaultKind ||
49264ab3302SCarolineConcatto               !analyzer.context().IsEnabled(LanguageFeature::BigIntLiterals)) {
49364ab3302SCarolineConcatto             return std::nullopt;
49464ab3302SCarolineConcatto           } else if (analyzer.context().ShouldWarn(
49564ab3302SCarolineConcatto                          LanguageFeature::BigIntLiterals)) {
49664ab3302SCarolineConcatto             analyzer.Say(digits,
49764ab3302SCarolineConcatto                 "Integer literal is too large for default INTEGER(KIND=%d); "
498a53967cdSPeter Klausler                 "assuming INTEGER(KIND=%d)"_port_en_US,
49964ab3302SCarolineConcatto                 kind, T::kind);
50064ab3302SCarolineConcatto           }
50164ab3302SCarolineConcatto         }
50264ab3302SCarolineConcatto         return Expr<SomeType>{
50373506256SPeter Klausler             Expr<SomeInteger>{Expr<T>{Constant<T>{std::move(num.value)}}}};
50464ab3302SCarolineConcatto       }
50564ab3302SCarolineConcatto     }
50664ab3302SCarolineConcatto     return std::nullopt;
50764ab3302SCarolineConcatto   }
50864ab3302SCarolineConcatto   ExpressionAnalyzer &analyzer;
50964ab3302SCarolineConcatto   parser::CharBlock digits;
51064ab3302SCarolineConcatto   int kind;
51164ab3302SCarolineConcatto   bool isDefaultKind;
51273506256SPeter Klausler   bool isNegated;
51364ab3302SCarolineConcatto };
51464ab3302SCarolineConcatto 
51564ab3302SCarolineConcatto template <typename PARSED>
IntLiteralConstant(const PARSED & x,bool isNegated)51673506256SPeter Klausler MaybeExpr ExpressionAnalyzer::IntLiteralConstant(
51773506256SPeter Klausler     const PARSED &x, bool isNegated) {
51864ab3302SCarolineConcatto   const auto &kindParam{std::get<std::optional<parser::KindParam>>(x.t)};
51964ab3302SCarolineConcatto   bool isDefaultKind{!kindParam};
52064ab3302SCarolineConcatto   int kind{AnalyzeKindParam(kindParam, GetDefaultKind(TypeCategory::Integer))};
52164ab3302SCarolineConcatto   if (CheckIntrinsicKind(TypeCategory::Integer, kind)) {
52264ab3302SCarolineConcatto     auto digits{std::get<parser::CharBlock>(x.t)};
52364ab3302SCarolineConcatto     if (MaybeExpr result{common::SearchTypes(
52473506256SPeter Klausler             IntTypeVisitor{*this, digits, kind, isDefaultKind, isNegated})}) {
52564ab3302SCarolineConcatto       return result;
52664ab3302SCarolineConcatto     } else if (isDefaultKind) {
52764ab3302SCarolineConcatto       Say(digits,
52864ab3302SCarolineConcatto           "Integer literal is too large for any allowable "
52964ab3302SCarolineConcatto           "kind of INTEGER"_err_en_US);
53064ab3302SCarolineConcatto     } else {
53164ab3302SCarolineConcatto       Say(digits, "Integer literal is too large for INTEGER(KIND=%d)"_err_en_US,
53264ab3302SCarolineConcatto           kind);
53364ab3302SCarolineConcatto     }
53464ab3302SCarolineConcatto   }
53564ab3302SCarolineConcatto   return std::nullopt;
53664ab3302SCarolineConcatto }
53764ab3302SCarolineConcatto 
Analyze(const parser::IntLiteralConstant & x,bool isNegated)53873506256SPeter Klausler MaybeExpr ExpressionAnalyzer::Analyze(
53973506256SPeter Klausler     const parser::IntLiteralConstant &x, bool isNegated) {
5402a00c617STim Keith   auto restorer{
5412a00c617STim Keith       GetContextualMessages().SetLocation(std::get<parser::CharBlock>(x.t))};
54273506256SPeter Klausler   return IntLiteralConstant(x, isNegated);
54364ab3302SCarolineConcatto }
54464ab3302SCarolineConcatto 
Analyze(const parser::SignedIntLiteralConstant & x)54564ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(
54664ab3302SCarolineConcatto     const parser::SignedIntLiteralConstant &x) {
5472a00c617STim Keith   auto restorer{GetContextualMessages().SetLocation(x.source)};
54864ab3302SCarolineConcatto   return IntLiteralConstant(x);
54964ab3302SCarolineConcatto }
55064ab3302SCarolineConcatto 
55164ab3302SCarolineConcatto template <typename TYPE>
ReadRealLiteral(parser::CharBlock source,FoldingContext & context)55264ab3302SCarolineConcatto Constant<TYPE> ReadRealLiteral(
55364ab3302SCarolineConcatto     parser::CharBlock source, FoldingContext &context) {
55464ab3302SCarolineConcatto   const char *p{source.begin()};
55523c2bedfSPeter Klausler   auto valWithFlags{
55623c2bedfSPeter Klausler       Scalar<TYPE>::Read(p, context.targetCharacteristics().roundingMode())};
55764ab3302SCarolineConcatto   CHECK(p == source.end());
55864ab3302SCarolineConcatto   RealFlagWarnings(context, valWithFlags.flags, "conversion of REAL literal");
55964ab3302SCarolineConcatto   auto value{valWithFlags.value};
56023c2bedfSPeter Klausler   if (context.targetCharacteristics().areSubnormalsFlushedToZero()) {
56164ab3302SCarolineConcatto     value = value.FlushSubnormalToZero();
56264ab3302SCarolineConcatto   }
56364ab3302SCarolineConcatto   return {value};
56464ab3302SCarolineConcatto }
56564ab3302SCarolineConcatto 
56664ab3302SCarolineConcatto struct RealTypeVisitor {
56764ab3302SCarolineConcatto   using Result = std::optional<Expr<SomeReal>>;
56864ab3302SCarolineConcatto   using Types = RealTypes;
56964ab3302SCarolineConcatto 
RealTypeVisitorFortran::evaluate::RealTypeVisitor57064ab3302SCarolineConcatto   RealTypeVisitor(int k, parser::CharBlock lit, FoldingContext &ctx)
57164ab3302SCarolineConcatto       : kind{k}, literal{lit}, context{ctx} {}
57264ab3302SCarolineConcatto 
TestFortran::evaluate::RealTypeVisitor57364ab3302SCarolineConcatto   template <typename T> Result Test() {
57464ab3302SCarolineConcatto     if (kind == T::kind) {
57564ab3302SCarolineConcatto       return {AsCategoryExpr(ReadRealLiteral<T>(literal, context))};
57664ab3302SCarolineConcatto     }
57764ab3302SCarolineConcatto     return std::nullopt;
57864ab3302SCarolineConcatto   }
57964ab3302SCarolineConcatto 
58064ab3302SCarolineConcatto   int kind;
58164ab3302SCarolineConcatto   parser::CharBlock literal;
58264ab3302SCarolineConcatto   FoldingContext &context;
58364ab3302SCarolineConcatto };
58464ab3302SCarolineConcatto 
58564ab3302SCarolineConcatto // Reads a real literal constant and encodes it with the right kind.
Analyze(const parser::RealLiteralConstant & x)58664ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::RealLiteralConstant &x) {
58764ab3302SCarolineConcatto   // Use a local message context around the real literal for better
58864ab3302SCarolineConcatto   // provenance on any messages.
58964ab3302SCarolineConcatto   auto restorer{GetContextualMessages().SetLocation(x.real.source)};
590657aaf8bSPete Steinfeld   // If a kind parameter appears, it defines the kind of the literal and the
591657aaf8bSPete Steinfeld   // letter used in an exponent part must be 'E' (e.g., the 'E' in
592657aaf8bSPete Steinfeld   // "6.02214E+23").  In the absence of an explicit kind parameter, any
593657aaf8bSPete Steinfeld   // exponent letter determines the kind.  Otherwise, defaults apply.
59464ab3302SCarolineConcatto   auto &defaults{context_.defaultKinds()};
59564ab3302SCarolineConcatto   int defaultKind{defaults.GetDefaultKind(TypeCategory::Real)};
59664ab3302SCarolineConcatto   const char *end{x.real.source.end()};
59764ab3302SCarolineConcatto   char expoLetter{' '};
59864ab3302SCarolineConcatto   std::optional<int> letterKind;
59964ab3302SCarolineConcatto   for (const char *p{x.real.source.begin()}; p < end; ++p) {
60064ab3302SCarolineConcatto     if (parser::IsLetter(*p)) {
60164ab3302SCarolineConcatto       expoLetter = *p;
60264ab3302SCarolineConcatto       switch (expoLetter) {
60384a099dfSpeter klausler       case 'e':
60484a099dfSpeter klausler         letterKind = defaults.GetDefaultKind(TypeCategory::Real);
60584a099dfSpeter klausler         break;
60684a099dfSpeter klausler       case 'd':
60784a099dfSpeter klausler         letterKind = defaults.doublePrecisionKind();
60884a099dfSpeter klausler         break;
60984a099dfSpeter klausler       case 'q':
61084a099dfSpeter klausler         letterKind = defaults.quadPrecisionKind();
61184a099dfSpeter klausler         break;
61284a099dfSpeter klausler       default:
61384a099dfSpeter klausler         Say("Unknown exponent letter '%c'"_err_en_US, expoLetter);
61464ab3302SCarolineConcatto       }
61564ab3302SCarolineConcatto       break;
61664ab3302SCarolineConcatto     }
61764ab3302SCarolineConcatto   }
61864ab3302SCarolineConcatto   if (letterKind) {
61964ab3302SCarolineConcatto     defaultKind = *letterKind;
62064ab3302SCarolineConcatto   }
621f1983feaSEmil Kieri   // C716 requires 'E' as an exponent.
622f1983feaSEmil Kieri   // Extension: allow exponent-letter matching the kind-param.
62364ab3302SCarolineConcatto   auto kind{AnalyzeKindParam(x.kind, defaultKind)};
624f1983feaSEmil Kieri   if (letterKind && expoLetter != 'e') {
625f1983feaSEmil Kieri     if (kind != *letterKind) {
62684b2909dSPete Steinfeld       Say("Explicit kind parameter on real constant disagrees with "
627f1983feaSEmil Kieri           "exponent letter '%c'"_warn_en_US,
62884b2909dSPete Steinfeld           expoLetter);
629f1983feaSEmil Kieri     } else if (x.kind &&
630f1983feaSEmil Kieri         context_.ShouldWarn(
631f1983feaSEmil Kieri             common::LanguageFeature::ExponentMatchingKindParam)) {
632f1983feaSEmil Kieri       Say("Explicit kind parameter together with non-'E' exponent letter "
633f1983feaSEmil Kieri           "is not standard"_port_en_US);
634f1983feaSEmil Kieri     }
63564ab3302SCarolineConcatto   }
63664ab3302SCarolineConcatto   auto result{common::SearchTypes(
63764ab3302SCarolineConcatto       RealTypeVisitor{kind, x.real.source, GetFoldingContext()})};
638657aaf8bSPete Steinfeld   if (!result) { // C717
63964ab3302SCarolineConcatto     Say("Unsupported REAL(KIND=%d)"_err_en_US, kind);
64064ab3302SCarolineConcatto   }
64164ab3302SCarolineConcatto   return AsMaybeExpr(std::move(result));
64264ab3302SCarolineConcatto }
64364ab3302SCarolineConcatto 
Analyze(const parser::SignedRealLiteralConstant & x)64464ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(
64564ab3302SCarolineConcatto     const parser::SignedRealLiteralConstant &x) {
64664ab3302SCarolineConcatto   if (auto result{Analyze(std::get<parser::RealLiteralConstant>(x.t))}) {
64764ab3302SCarolineConcatto     auto &realExpr{std::get<Expr<SomeReal>>(result->u)};
64864ab3302SCarolineConcatto     if (auto sign{std::get<std::optional<parser::Sign>>(x.t)}) {
64964ab3302SCarolineConcatto       if (sign == parser::Sign::Negative) {
6504171f80dSpeter klausler         return AsGenericExpr(-std::move(realExpr));
65164ab3302SCarolineConcatto       }
65264ab3302SCarolineConcatto     }
65364ab3302SCarolineConcatto     return result;
65464ab3302SCarolineConcatto   }
65564ab3302SCarolineConcatto   return std::nullopt;
65664ab3302SCarolineConcatto }
65764ab3302SCarolineConcatto 
Analyze(const parser::SignedComplexLiteralConstant & x)6582a00c617STim Keith MaybeExpr ExpressionAnalyzer::Analyze(
6592a00c617STim Keith     const parser::SignedComplexLiteralConstant &x) {
6602a00c617STim Keith   auto result{Analyze(std::get<parser::ComplexLiteralConstant>(x.t))};
6612a00c617STim Keith   if (!result) {
6622a00c617STim Keith     return std::nullopt;
6632a00c617STim Keith   } else if (std::get<parser::Sign>(x.t) == parser::Sign::Negative) {
6642a00c617STim Keith     return AsGenericExpr(-std::move(std::get<Expr<SomeComplex>>(result->u)));
6652a00c617STim Keith   } else {
6662a00c617STim Keith     return result;
6672a00c617STim Keith   }
6682a00c617STim Keith }
6692a00c617STim Keith 
Analyze(const parser::ComplexPart & x)67064ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::ComplexPart &x) {
67164ab3302SCarolineConcatto   return Analyze(x.u);
67264ab3302SCarolineConcatto }
67364ab3302SCarolineConcatto 
Analyze(const parser::ComplexLiteralConstant & z)67464ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::ComplexLiteralConstant &z) {
67564ab3302SCarolineConcatto   return AsMaybeExpr(
67664ab3302SCarolineConcatto       ConstructComplex(GetContextualMessages(), Analyze(std::get<0>(z.t)),
67764ab3302SCarolineConcatto           Analyze(std::get<1>(z.t)), GetDefaultKind(TypeCategory::Real)));
67864ab3302SCarolineConcatto }
67964ab3302SCarolineConcatto 
68064ab3302SCarolineConcatto // CHARACTER literal processing.
AnalyzeString(std::string && string,int kind)68164ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::AnalyzeString(std::string &&string, int kind) {
68264ab3302SCarolineConcatto   if (!CheckIntrinsicKind(TypeCategory::Character, kind)) {
68364ab3302SCarolineConcatto     return std::nullopt;
68464ab3302SCarolineConcatto   }
68564ab3302SCarolineConcatto   switch (kind) {
68664ab3302SCarolineConcatto   case 1:
68764ab3302SCarolineConcatto     return AsGenericExpr(Constant<Type<TypeCategory::Character, 1>>{
68864ab3302SCarolineConcatto         parser::DecodeString<std::string, parser::Encoding::LATIN_1>(
68964ab3302SCarolineConcatto             string, true)});
69064ab3302SCarolineConcatto   case 2:
69164ab3302SCarolineConcatto     return AsGenericExpr(Constant<Type<TypeCategory::Character, 2>>{
69264ab3302SCarolineConcatto         parser::DecodeString<std::u16string, parser::Encoding::UTF_8>(
69364ab3302SCarolineConcatto             string, true)});
69464ab3302SCarolineConcatto   case 4:
69564ab3302SCarolineConcatto     return AsGenericExpr(Constant<Type<TypeCategory::Character, 4>>{
69664ab3302SCarolineConcatto         parser::DecodeString<std::u32string, parser::Encoding::UTF_8>(
69764ab3302SCarolineConcatto             string, true)});
69884a099dfSpeter klausler   default:
69984a099dfSpeter klausler     CRASH_NO_CASE;
70064ab3302SCarolineConcatto   }
70164ab3302SCarolineConcatto }
70264ab3302SCarolineConcatto 
Analyze(const parser::CharLiteralConstant & x)70364ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::CharLiteralConstant &x) {
70464ab3302SCarolineConcatto   int kind{
70564ab3302SCarolineConcatto       AnalyzeKindParam(std::get<std::optional<parser::KindParam>>(x.t), 1)};
70664ab3302SCarolineConcatto   auto value{std::get<std::string>(x.t)};
70764ab3302SCarolineConcatto   return AnalyzeString(std::move(value), kind);
70864ab3302SCarolineConcatto }
70964ab3302SCarolineConcatto 
Analyze(const parser::HollerithLiteralConstant & x)71064ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(
71164ab3302SCarolineConcatto     const parser::HollerithLiteralConstant &x) {
71264ab3302SCarolineConcatto   int kind{GetDefaultKind(TypeCategory::Character)};
71364ab3302SCarolineConcatto   auto value{x.v};
71464ab3302SCarolineConcatto   return AnalyzeString(std::move(value), kind);
71564ab3302SCarolineConcatto }
71664ab3302SCarolineConcatto 
71764ab3302SCarolineConcatto // .TRUE. and .FALSE. of various kinds
Analyze(const parser::LogicalLiteralConstant & x)71864ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::LogicalLiteralConstant &x) {
71964ab3302SCarolineConcatto   auto kind{AnalyzeKindParam(std::get<std::optional<parser::KindParam>>(x.t),
72064ab3302SCarolineConcatto       GetDefaultKind(TypeCategory::Logical))};
72164ab3302SCarolineConcatto   bool value{std::get<bool>(x.t)};
72264ab3302SCarolineConcatto   auto result{common::SearchTypes(
72364ab3302SCarolineConcatto       TypeKindVisitor<TypeCategory::Logical, Constant, bool>{
72464ab3302SCarolineConcatto           kind, std::move(value)})};
72564ab3302SCarolineConcatto   if (!result) {
726e17e7173SPete Steinfeld     Say("unsupported LOGICAL(KIND=%d)"_err_en_US, kind); // C728
72764ab3302SCarolineConcatto   }
72864ab3302SCarolineConcatto   return result;
72964ab3302SCarolineConcatto }
73064ab3302SCarolineConcatto 
73164ab3302SCarolineConcatto // BOZ typeless literals
Analyze(const parser::BOZLiteralConstant & x)73264ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::BOZLiteralConstant &x) {
73364ab3302SCarolineConcatto   const char *p{x.v.c_str()};
73464ab3302SCarolineConcatto   std::uint64_t base{16};
73564ab3302SCarolineConcatto   switch (*p++) {
73684a099dfSpeter klausler   case 'b':
73784a099dfSpeter klausler     base = 2;
73884a099dfSpeter klausler     break;
73984a099dfSpeter klausler   case 'o':
74084a099dfSpeter klausler     base = 8;
74184a099dfSpeter klausler     break;
74284a099dfSpeter klausler   case 'z':
74384a099dfSpeter klausler     break;
74484a099dfSpeter klausler   case 'x':
74584a099dfSpeter klausler     break;
74684a099dfSpeter klausler   default:
74784a099dfSpeter klausler     CRASH_NO_CASE;
74864ab3302SCarolineConcatto   }
74964ab3302SCarolineConcatto   CHECK(*p == '"');
75064ab3302SCarolineConcatto   ++p;
75164ab3302SCarolineConcatto   auto value{BOZLiteralConstant::Read(p, base, false /*unsigned*/)};
75264ab3302SCarolineConcatto   if (*p != '"') {
7538830f117Ssameeran joshi     Say("Invalid digit ('%c') in BOZ literal '%s'"_err_en_US, *p,
7548830f117Ssameeran joshi         x.v); // C7107, C7108
75564ab3302SCarolineConcatto     return std::nullopt;
75664ab3302SCarolineConcatto   }
75764ab3302SCarolineConcatto   if (value.overflow) {
75864ab3302SCarolineConcatto     Say("BOZ literal '%s' too large"_err_en_US, x.v);
75964ab3302SCarolineConcatto     return std::nullopt;
76064ab3302SCarolineConcatto   }
76164ab3302SCarolineConcatto   return AsGenericExpr(std::move(value.value));
76264ab3302SCarolineConcatto }
76364ab3302SCarolineConcatto 
76464ab3302SCarolineConcatto // Names and named constants
Analyze(const parser::Name & n)76564ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Name &n) {
766934b27a9Speter klausler   auto restorer{GetContextualMessages().SetLocation(n.source)};
7673a1afd8cSpeter klausler   if (std::optional<int> kind{IsImpliedDo(n.source)}) {
76864ab3302SCarolineConcatto     return AsMaybeExpr(ConvertToKind<TypeCategory::Integer>(
76964ab3302SCarolineConcatto         *kind, AsExpr(ImpliedDoIndex{n.source})));
7709e7eef99SPeter Klausler   }
7719e7eef99SPeter Klausler   if (context_.HasError(n.symbol)) { // includes case of no symbol
77264ab3302SCarolineConcatto     return std::nullopt;
77364ab3302SCarolineConcatto   } else {
77464ab3302SCarolineConcatto     const Symbol &ultimate{n.symbol->GetUltimate()};
77564ab3302SCarolineConcatto     if (ultimate.has<semantics::TypeParamDetails>()) {
77664ab3302SCarolineConcatto       // A bare reference to a derived type parameter (within a parameterized
77764ab3302SCarolineConcatto       // derived type definition)
7784cbfd93aSpeter klausler       return Fold(ConvertToType(
7794cbfd93aSpeter klausler           ultimate, AsGenericExpr(TypeParamInquiry{std::nullopt, ultimate})));
78064ab3302SCarolineConcatto     } else {
78164ab3302SCarolineConcatto       if (n.symbol->attrs().test(semantics::Attr::VOLATILE)) {
78264ab3302SCarolineConcatto         if (const semantics::Scope *
78364ab3302SCarolineConcatto             pure{semantics::FindPureProcedureContaining(
78464ab3302SCarolineConcatto                 context_.FindScope(n.source))}) {
78564ab3302SCarolineConcatto           SayAt(n,
78664ab3302SCarolineConcatto               "VOLATILE variable '%s' may not be referenced in pure subprogram '%s'"_err_en_US,
78764ab3302SCarolineConcatto               n.source, DEREF(pure->symbol()).name());
78864ab3302SCarolineConcatto           n.symbol->attrs().reset(semantics::Attr::VOLATILE);
78964ab3302SCarolineConcatto         }
79064ab3302SCarolineConcatto       }
79188afb6e8Speter klausler       if (!isWholeAssumedSizeArrayOk_ &&
79288afb6e8Speter klausler           semantics::IsAssumedSizeArray(*n.symbol)) { // C1002, C1014, C1231
79388afb6e8Speter klausler         AttachDeclaration(
79488afb6e8Speter klausler             SayAt(n,
79588afb6e8Speter klausler                 "Whole assumed-size array '%s' may not appear here without subscripts"_err_en_US,
79688afb6e8Speter klausler                 n.source),
79788afb6e8Speter klausler             *n.symbol);
79888afb6e8Speter klausler       }
79964ab3302SCarolineConcatto       return Designate(DataRef{*n.symbol});
80064ab3302SCarolineConcatto     }
80164ab3302SCarolineConcatto   }
80264ab3302SCarolineConcatto }
80364ab3302SCarolineConcatto 
Analyze(const parser::NamedConstant & n)80464ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::NamedConstant &n) {
805934b27a9Speter klausler   auto restorer{GetContextualMessages().SetLocation(n.v.source)};
80664ab3302SCarolineConcatto   if (MaybeExpr value{Analyze(n.v)}) {
80764ab3302SCarolineConcatto     Expr<SomeType> folded{Fold(std::move(*value))};
80864ab3302SCarolineConcatto     if (IsConstantExpr(folded)) {
809a20d48d7Speter klausler       return folded;
81064ab3302SCarolineConcatto     }
811657aaf8bSPete Steinfeld     Say(n.v.source, "must be a constant"_err_en_US); // C718
81264ab3302SCarolineConcatto   }
81364ab3302SCarolineConcatto   return std::nullopt;
81464ab3302SCarolineConcatto }
81564ab3302SCarolineConcatto 
Analyze(const parser::NullInit & n)8165349f991Speter klausler MaybeExpr ExpressionAnalyzer::Analyze(const parser::NullInit &n) {
8175349f991Speter klausler   if (MaybeExpr value{Analyze(n.v)}) {
8185349f991Speter klausler     // Subtle: when the NullInit is a DataStmtConstant, it might
8195349f991Speter klausler     // be a misparse of a structure constructor without parameters
8205349f991Speter klausler     // or components (e.g., T()).  Checking the result to ensure
8215349f991Speter klausler     // that a "=>" data entity initializer actually resolved to
8225349f991Speter klausler     // a null pointer has to be done by the caller.
8235349f991Speter klausler     return Fold(std::move(*value));
8245349f991Speter klausler   }
8255349f991Speter klausler   return std::nullopt;
8263a1afd8cSpeter klausler }
8273a1afd8cSpeter klausler 
Analyze(const parser::InitialDataTarget & x)8283a1afd8cSpeter klausler MaybeExpr ExpressionAnalyzer::Analyze(const parser::InitialDataTarget &x) {
8293a1afd8cSpeter klausler   return Analyze(x.value());
8303a1afd8cSpeter klausler }
8313a1afd8cSpeter klausler 
Analyze(const parser::DataStmtValue & x)8324171f80dSpeter klausler MaybeExpr ExpressionAnalyzer::Analyze(const parser::DataStmtValue &x) {
8334171f80dSpeter klausler   if (const auto &repeat{
8344171f80dSpeter klausler           std::get<std::optional<parser::DataStmtRepeat>>(x.t)}) {
835a20d48d7Speter klausler     x.repetitions = -1;
8364171f80dSpeter klausler     if (MaybeExpr expr{Analyze(repeat->u)}) {
8374171f80dSpeter klausler       Expr<SomeType> folded{Fold(std::move(*expr))};
8384171f80dSpeter klausler       if (auto value{ToInt64(folded)}) {
8394171f80dSpeter klausler         if (*value >= 0) { // C882
8404171f80dSpeter klausler           x.repetitions = *value;
8414171f80dSpeter klausler         } else {
8424171f80dSpeter klausler           Say(FindSourceLocation(repeat),
8434171f80dSpeter klausler               "Repeat count (%jd) for data value must not be negative"_err_en_US,
8444171f80dSpeter klausler               *value);
8454171f80dSpeter klausler         }
8464171f80dSpeter klausler       }
8474171f80dSpeter klausler     }
8484171f80dSpeter klausler   }
8494171f80dSpeter klausler   return Analyze(std::get<parser::DataStmtConstant>(x.t));
8504171f80dSpeter klausler }
8514171f80dSpeter klausler 
85264ab3302SCarolineConcatto // Substring references
GetSubstringBound(const std::optional<parser::ScalarIntExpr> & bound)85364ab3302SCarolineConcatto std::optional<Expr<SubscriptInteger>> ExpressionAnalyzer::GetSubstringBound(
85464ab3302SCarolineConcatto     const std::optional<parser::ScalarIntExpr> &bound) {
85564ab3302SCarolineConcatto   if (bound) {
85664ab3302SCarolineConcatto     if (MaybeExpr expr{Analyze(*bound)}) {
85764ab3302SCarolineConcatto       if (expr->Rank() > 1) {
85864ab3302SCarolineConcatto         Say("substring bound expression has rank %d"_err_en_US, expr->Rank());
85964ab3302SCarolineConcatto       }
86064ab3302SCarolineConcatto       if (auto *intExpr{std::get_if<Expr<SomeInteger>>(&expr->u)}) {
86164ab3302SCarolineConcatto         if (auto *ssIntExpr{std::get_if<Expr<SubscriptInteger>>(&intExpr->u)}) {
86264ab3302SCarolineConcatto           return {std::move(*ssIntExpr)};
86364ab3302SCarolineConcatto         }
86464ab3302SCarolineConcatto         return {Expr<SubscriptInteger>{
86564ab3302SCarolineConcatto             Convert<SubscriptInteger, TypeCategory::Integer>{
86664ab3302SCarolineConcatto                 std::move(*intExpr)}}};
86764ab3302SCarolineConcatto       } else {
86864ab3302SCarolineConcatto         Say("substring bound expression is not INTEGER"_err_en_US);
86964ab3302SCarolineConcatto       }
87064ab3302SCarolineConcatto     }
87164ab3302SCarolineConcatto   }
87264ab3302SCarolineConcatto   return std::nullopt;
87364ab3302SCarolineConcatto }
87464ab3302SCarolineConcatto 
Analyze(const parser::Substring & ss)87564ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Substring &ss) {
87664ab3302SCarolineConcatto   if (MaybeExpr baseExpr{Analyze(std::get<parser::DataRef>(ss.t))}) {
87764ab3302SCarolineConcatto     if (std::optional<DataRef> dataRef{ExtractDataRef(std::move(*baseExpr))}) {
87848a70ea1SPeixin-Qiao       if (MaybeExpr newBaseExpr{Designate(std::move(*dataRef))}) {
87964ab3302SCarolineConcatto         if (std::optional<DataRef> checked{
88064ab3302SCarolineConcatto                 ExtractDataRef(std::move(*newBaseExpr))}) {
88164ab3302SCarolineConcatto           const parser::SubstringRange &range{
88264ab3302SCarolineConcatto               std::get<parser::SubstringRange>(ss.t)};
88364ab3302SCarolineConcatto           std::optional<Expr<SubscriptInteger>> first{
88464ab3302SCarolineConcatto               GetSubstringBound(std::get<0>(range.t))};
88564ab3302SCarolineConcatto           std::optional<Expr<SubscriptInteger>> last{
88664ab3302SCarolineConcatto               GetSubstringBound(std::get<1>(range.t))};
88764ab3302SCarolineConcatto           const Symbol &symbol{checked->GetLastSymbol()};
88864ab3302SCarolineConcatto           if (std::optional<DynamicType> dynamicType{
88964ab3302SCarolineConcatto                   DynamicType::From(symbol)}) {
89064ab3302SCarolineConcatto             if (dynamicType->category() == TypeCategory::Character) {
89164ab3302SCarolineConcatto               return WrapperHelper<TypeCategory::Character, Designator,
89264ab3302SCarolineConcatto                   Substring>(dynamicType->kind(),
89364ab3302SCarolineConcatto                   Substring{std::move(checked.value()), std::move(first),
89464ab3302SCarolineConcatto                       std::move(last)});
89564ab3302SCarolineConcatto             }
89664ab3302SCarolineConcatto           }
89764ab3302SCarolineConcatto           Say("substring may apply only to CHARACTER"_err_en_US);
89864ab3302SCarolineConcatto         }
89964ab3302SCarolineConcatto       }
90064ab3302SCarolineConcatto     }
90164ab3302SCarolineConcatto   }
90264ab3302SCarolineConcatto   return std::nullopt;
90364ab3302SCarolineConcatto }
90464ab3302SCarolineConcatto 
90564ab3302SCarolineConcatto // CHARACTER literal substrings
Analyze(const parser::CharLiteralConstantSubstring & x)90664ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(
90764ab3302SCarolineConcatto     const parser::CharLiteralConstantSubstring &x) {
90864ab3302SCarolineConcatto   const parser::SubstringRange &range{std::get<parser::SubstringRange>(x.t)};
90964ab3302SCarolineConcatto   std::optional<Expr<SubscriptInteger>> lower{
91064ab3302SCarolineConcatto       GetSubstringBound(std::get<0>(range.t))};
91164ab3302SCarolineConcatto   std::optional<Expr<SubscriptInteger>> upper{
91264ab3302SCarolineConcatto       GetSubstringBound(std::get<1>(range.t))};
91364ab3302SCarolineConcatto   if (MaybeExpr string{Analyze(std::get<parser::CharLiteralConstant>(x.t))}) {
91464ab3302SCarolineConcatto     if (auto *charExpr{std::get_if<Expr<SomeCharacter>>(&string->u)}) {
91564ab3302SCarolineConcatto       Expr<SubscriptInteger> length{
916cd03e96fSPeter Klausler           common::visit([](const auto &ckExpr) { return ckExpr.LEN().value(); },
91764ab3302SCarolineConcatto               charExpr->u)};
91864ab3302SCarolineConcatto       if (!lower) {
91964ab3302SCarolineConcatto         lower = Expr<SubscriptInteger>{1};
92064ab3302SCarolineConcatto       }
92164ab3302SCarolineConcatto       if (!upper) {
92264ab3302SCarolineConcatto         upper = Expr<SubscriptInteger>{
92364ab3302SCarolineConcatto             static_cast<std::int64_t>(ToInt64(length).value())};
92464ab3302SCarolineConcatto       }
925cd03e96fSPeter Klausler       return common::visit(
92664ab3302SCarolineConcatto           [&](auto &&ckExpr) -> MaybeExpr {
92764ab3302SCarolineConcatto             using Result = ResultType<decltype(ckExpr)>;
92864ab3302SCarolineConcatto             auto *cp{std::get_if<Constant<Result>>(&ckExpr.u)};
92964ab3302SCarolineConcatto             CHECK(DEREF(cp).size() == 1);
93064ab3302SCarolineConcatto             StaticDataObject::Pointer staticData{StaticDataObject::Create()};
93164ab3302SCarolineConcatto             staticData->set_alignment(Result::kind)
93264ab3302SCarolineConcatto                 .set_itemBytes(Result::kind)
93323c2bedfSPeter Klausler                 .Push(cp->GetScalarValue().value(),
93423c2bedfSPeter Klausler                     foldingContext_.targetCharacteristics().isBigEndian());
93564ab3302SCarolineConcatto             Substring substring{std::move(staticData), std::move(lower.value()),
93664ab3302SCarolineConcatto                 std::move(upper.value())};
9374171f80dSpeter klausler             return AsGenericExpr(
9384171f80dSpeter klausler                 Expr<Result>{Designator<Result>{std::move(substring)}});
93964ab3302SCarolineConcatto           },
94064ab3302SCarolineConcatto           std::move(charExpr->u));
94164ab3302SCarolineConcatto     }
94264ab3302SCarolineConcatto   }
94364ab3302SCarolineConcatto   return std::nullopt;
94464ab3302SCarolineConcatto }
94564ab3302SCarolineConcatto 
946e03664d4SPeter Klausler // substring%KIND/LEN
Analyze(const parser::SubstringInquiry & x)947e03664d4SPeter Klausler MaybeExpr ExpressionAnalyzer::Analyze(const parser::SubstringInquiry &x) {
948e03664d4SPeter Klausler   if (MaybeExpr substring{Analyze(x.v)}) {
949e03664d4SPeter Klausler     CHECK(x.source.size() >= 8);
950e03664d4SPeter Klausler     int nameLen{x.source.end()[-1] == 'n' ? 3 /*LEN*/ : 4 /*KIND*/};
951e03664d4SPeter Klausler     parser::CharBlock name{
952e03664d4SPeter Klausler         x.source.end() - nameLen, static_cast<std::size_t>(nameLen)};
953e03664d4SPeter Klausler     CHECK(name == "len" || name == "kind");
954e03664d4SPeter Klausler     return MakeFunctionRef(
955e03664d4SPeter Klausler         name, ActualArguments{ActualArgument{std::move(*substring)}});
956e03664d4SPeter Klausler   } else {
957e03664d4SPeter Klausler     return std::nullopt;
958e03664d4SPeter Klausler   }
959e03664d4SPeter Klausler }
960e03664d4SPeter Klausler 
96164ab3302SCarolineConcatto // Subscripted array references
AsSubscript(MaybeExpr && expr)96264ab3302SCarolineConcatto std::optional<Expr<SubscriptInteger>> ExpressionAnalyzer::AsSubscript(
96364ab3302SCarolineConcatto     MaybeExpr &&expr) {
96464ab3302SCarolineConcatto   if (expr) {
96564ab3302SCarolineConcatto     if (expr->Rank() > 1) {
96664ab3302SCarolineConcatto       Say("Subscript expression has rank %d greater than 1"_err_en_US,
96764ab3302SCarolineConcatto           expr->Rank());
96864ab3302SCarolineConcatto     }
96964ab3302SCarolineConcatto     if (auto *intExpr{std::get_if<Expr<SomeInteger>>(&expr->u)}) {
97064ab3302SCarolineConcatto       if (auto *ssIntExpr{std::get_if<Expr<SubscriptInteger>>(&intExpr->u)}) {
97164ab3302SCarolineConcatto         return std::move(*ssIntExpr);
97264ab3302SCarolineConcatto       } else {
97364ab3302SCarolineConcatto         return Expr<SubscriptInteger>{
97464ab3302SCarolineConcatto             Convert<SubscriptInteger, TypeCategory::Integer>{
97564ab3302SCarolineConcatto                 std::move(*intExpr)}};
97664ab3302SCarolineConcatto       }
97764ab3302SCarolineConcatto     } else {
97864ab3302SCarolineConcatto       Say("Subscript expression is not INTEGER"_err_en_US);
97964ab3302SCarolineConcatto     }
98064ab3302SCarolineConcatto   }
98164ab3302SCarolineConcatto   return std::nullopt;
98264ab3302SCarolineConcatto }
98364ab3302SCarolineConcatto 
TripletPart(const std::optional<parser::Subscript> & s)98464ab3302SCarolineConcatto std::optional<Expr<SubscriptInteger>> ExpressionAnalyzer::TripletPart(
98564ab3302SCarolineConcatto     const std::optional<parser::Subscript> &s) {
98664ab3302SCarolineConcatto   if (s) {
98764ab3302SCarolineConcatto     return AsSubscript(Analyze(*s));
98864ab3302SCarolineConcatto   } else {
98964ab3302SCarolineConcatto     return std::nullopt;
99064ab3302SCarolineConcatto   }
99164ab3302SCarolineConcatto }
99264ab3302SCarolineConcatto 
AnalyzeSectionSubscript(const parser::SectionSubscript & ss)99364ab3302SCarolineConcatto std::optional<Subscript> ExpressionAnalyzer::AnalyzeSectionSubscript(
99464ab3302SCarolineConcatto     const parser::SectionSubscript &ss) {
995cd03e96fSPeter Klausler   return common::visit(
9967f5928efSTim Keith       common::visitors{
9977f5928efSTim Keith           [&](const parser::SubscriptTriplet &t) -> std::optional<Subscript> {
9987f5928efSTim Keith             const auto &lower{std::get<0>(t.t)};
9997f5928efSTim Keith             const auto &upper{std::get<1>(t.t)};
10007f5928efSTim Keith             const auto &stride{std::get<2>(t.t)};
10017f5928efSTim Keith             auto result{Triplet{
10027f5928efSTim Keith                 TripletPart(lower), TripletPart(upper), TripletPart(stride)}};
10037f5928efSTim Keith             if ((lower && !result.lower()) || (upper && !result.upper())) {
10047f5928efSTim Keith               return std::nullopt;
10057f5928efSTim Keith             } else {
10067f5928efSTim Keith               return std::make_optional<Subscript>(result);
10077f5928efSTim Keith             }
100864ab3302SCarolineConcatto           },
100964ab3302SCarolineConcatto           [&](const auto &s) -> std::optional<Subscript> {
101064ab3302SCarolineConcatto             if (auto subscriptExpr{AsSubscript(Analyze(s))}) {
101164ab3302SCarolineConcatto               return Subscript{std::move(*subscriptExpr)};
101264ab3302SCarolineConcatto             } else {
101364ab3302SCarolineConcatto               return std::nullopt;
101464ab3302SCarolineConcatto             }
101564ab3302SCarolineConcatto           },
101664ab3302SCarolineConcatto       },
101764ab3302SCarolineConcatto       ss.u);
101864ab3302SCarolineConcatto }
101964ab3302SCarolineConcatto 
102064ab3302SCarolineConcatto // Empty result means an error occurred
AnalyzeSectionSubscripts(const std::list<parser::SectionSubscript> & sss)102164ab3302SCarolineConcatto std::vector<Subscript> ExpressionAnalyzer::AnalyzeSectionSubscripts(
102264ab3302SCarolineConcatto     const std::list<parser::SectionSubscript> &sss) {
102364ab3302SCarolineConcatto   bool error{false};
102464ab3302SCarolineConcatto   std::vector<Subscript> subscripts;
102564ab3302SCarolineConcatto   for (const auto &s : sss) {
102664ab3302SCarolineConcatto     if (auto subscript{AnalyzeSectionSubscript(s)}) {
102764ab3302SCarolineConcatto       subscripts.emplace_back(std::move(*subscript));
102864ab3302SCarolineConcatto     } else {
102964ab3302SCarolineConcatto       error = true;
103064ab3302SCarolineConcatto     }
103164ab3302SCarolineConcatto   }
103264ab3302SCarolineConcatto   return !error ? subscripts : std::vector<Subscript>{};
103364ab3302SCarolineConcatto }
103464ab3302SCarolineConcatto 
Analyze(const parser::ArrayElement & ae)103564ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::ArrayElement &ae) {
103688afb6e8Speter klausler   MaybeExpr baseExpr;
103788afb6e8Speter klausler   {
103888afb6e8Speter klausler     auto restorer{AllowWholeAssumedSizeArray()};
103988afb6e8Speter klausler     baseExpr = Analyze(ae.base);
104088afb6e8Speter klausler   }
104188afb6e8Speter klausler   if (baseExpr) {
10423495c782STim Keith     if (ae.subscripts.empty()) {
10433495c782STim Keith       // will be converted to function call later or error reported
10443495c782STim Keith     } else if (baseExpr->Rank() == 0) {
10453495c782STim Keith       if (const Symbol * symbol{GetLastSymbol(*baseExpr)}) {
1046f9e24a56SPete Steinfeld         if (!context_.HasError(symbol)) {
1047a94b721dSpeter klausler           if (inDataStmtConstant_) {
1048a94b721dSpeter klausler             // Better error for NULL(X) with a MOLD= argument
1049a94b721dSpeter klausler             Say("'%s' must be an array or structure constructor if used with non-empty parentheses as a DATA statement constant"_err_en_US,
1050a94b721dSpeter klausler                 symbol->name());
1051a94b721dSpeter klausler           } else {
10523495c782STim Keith             Say("'%s' is not an array"_err_en_US, symbol->name());
1053a94b721dSpeter klausler           }
1054627e9007STim Keith           context_.SetError(*symbol);
1055f9e24a56SPete Steinfeld         }
105664ab3302SCarolineConcatto       }
10573495c782STim Keith     } else if (std::optional<DataRef> dataRef{
10583495c782STim Keith                    ExtractDataRef(std::move(*baseExpr))}) {
10593495c782STim Keith       return ApplySubscripts(
10603495c782STim Keith           std::move(*dataRef), AnalyzeSectionSubscripts(ae.subscripts));
106164ab3302SCarolineConcatto     } else {
106264ab3302SCarolineConcatto       Say("Subscripts may be applied only to an object, component, or array constant"_err_en_US);
106364ab3302SCarolineConcatto     }
106464ab3302SCarolineConcatto   }
10653495c782STim Keith   // error was reported: analyze subscripts without reporting more errors
10663495c782STim Keith   auto restorer{GetContextualMessages().DiscardMessages()};
10673495c782STim Keith   AnalyzeSectionSubscripts(ae.subscripts);
106864ab3302SCarolineConcatto   return std::nullopt;
106964ab3302SCarolineConcatto }
107064ab3302SCarolineConcatto 
107164ab3302SCarolineConcatto // Type parameter inquiries apply to data references, but don't depend
107264ab3302SCarolineConcatto // on any trailing (co)subscripts.
IgnoreAnySubscripts(Designator<SomeDerived> && designator)107364ab3302SCarolineConcatto static NamedEntity IgnoreAnySubscripts(Designator<SomeDerived> &&designator) {
1074cd03e96fSPeter Klausler   return common::visit(
107564ab3302SCarolineConcatto       common::visitors{
107664ab3302SCarolineConcatto           [](SymbolRef &&symbol) { return NamedEntity{symbol}; },
107764ab3302SCarolineConcatto           [](Component &&component) {
107864ab3302SCarolineConcatto             return NamedEntity{std::move(component)};
107964ab3302SCarolineConcatto           },
108064ab3302SCarolineConcatto           [](ArrayRef &&arrayRef) { return std::move(arrayRef.base()); },
108164ab3302SCarolineConcatto           [](CoarrayRef &&coarrayRef) {
108264ab3302SCarolineConcatto             return NamedEntity{coarrayRef.GetLastSymbol()};
108364ab3302SCarolineConcatto           },
108464ab3302SCarolineConcatto       },
108564ab3302SCarolineConcatto       std::move(designator.u));
108664ab3302SCarolineConcatto }
108764ab3302SCarolineConcatto 
108864ab3302SCarolineConcatto // Components of parent derived types are explicitly represented as such.
CreateComponent(DataRef && base,const Symbol & component,const semantics::Scope & scope)1089f2360e11Speter klausler std::optional<Component> ExpressionAnalyzer::CreateComponent(
109064ab3302SCarolineConcatto     DataRef &&base, const Symbol &component, const semantics::Scope &scope) {
1091f2360e11Speter klausler   if (IsAllocatableOrPointer(component) && base.Rank() > 0) { // C919b
1092f2360e11Speter klausler     Say("An allocatable or pointer component reference must be applied to a scalar base"_err_en_US);
1093f2360e11Speter klausler   }
109464ab3302SCarolineConcatto   if (&component.owner() == &scope) {
109564ab3302SCarolineConcatto     return Component{std::move(base), component};
109664ab3302SCarolineConcatto   }
109773f21db5SPeter Klausler   if (const Symbol * typeSymbol{scope.GetSymbol()}) {
109873f21db5SPeter Klausler     if (const Symbol *
109973f21db5SPeter Klausler         parentComponent{typeSymbol->GetParentComponent(&scope)}) {
110073f21db5SPeter Klausler       if (const auto *object{
110173f21db5SPeter Klausler               parentComponent->detailsIf<semantics::ObjectEntityDetails>()}) {
110273f21db5SPeter Klausler         if (const auto *parentType{object->type()}) {
110373f21db5SPeter Klausler           if (const semantics::Scope *
110473f21db5SPeter Klausler               parentScope{parentType->derivedTypeSpec().scope()}) {
110564ab3302SCarolineConcatto             return CreateComponent(
110673f21db5SPeter Klausler                 DataRef{Component{std::move(base), *parentComponent}},
110773f21db5SPeter Klausler                 component, *parentScope);
110873f21db5SPeter Klausler           }
110973f21db5SPeter Klausler         }
111073f21db5SPeter Klausler       }
111164ab3302SCarolineConcatto     }
111264ab3302SCarolineConcatto   }
111364ab3302SCarolineConcatto   return std::nullopt;
111464ab3302SCarolineConcatto }
111564ab3302SCarolineConcatto 
111664ab3302SCarolineConcatto // Derived type component references and type parameter inquiries
Analyze(const parser::StructureComponent & sc)111764ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::StructureComponent &sc) {
111864ab3302SCarolineConcatto   MaybeExpr base{Analyze(sc.base)};
111964ab3302SCarolineConcatto   Symbol *sym{sc.component.symbol};
1120934b27a9Speter klausler   if (!base || !sym || context_.HasError(sym)) {
112164ab3302SCarolineConcatto     return std::nullopt;
112264ab3302SCarolineConcatto   }
112364ab3302SCarolineConcatto   const auto &name{sc.component.source};
112464ab3302SCarolineConcatto   if (auto *dtExpr{UnwrapExpr<Expr<SomeDerived>>(*base)}) {
112564ab3302SCarolineConcatto     const auto *dtSpec{GetDerivedTypeSpec(dtExpr->GetType())};
112664ab3302SCarolineConcatto     if (sym->detailsIf<semantics::TypeParamDetails>()) {
112764ab3302SCarolineConcatto       if (auto *designator{UnwrapExpr<Designator<SomeDerived>>(*dtExpr)}) {
112864ab3302SCarolineConcatto         if (std::optional<DynamicType> dyType{DynamicType::From(*sym)}) {
112964ab3302SCarolineConcatto           if (dyType->category() == TypeCategory::Integer) {
1130934b27a9Speter klausler             auto restorer{GetContextualMessages().SetLocation(name)};
11314cbfd93aSpeter klausler             return Fold(ConvertToType(*dyType,
11324cbfd93aSpeter klausler                 AsGenericExpr(TypeParamInquiry{
11334cbfd93aSpeter klausler                     IgnoreAnySubscripts(std::move(*designator)), *sym})));
113464ab3302SCarolineConcatto           }
113564ab3302SCarolineConcatto         }
113664ab3302SCarolineConcatto         Say(name, "Type parameter is not INTEGER"_err_en_US);
113764ab3302SCarolineConcatto       } else {
113864ab3302SCarolineConcatto         Say(name,
113964ab3302SCarolineConcatto             "A type parameter inquiry must be applied to "
114064ab3302SCarolineConcatto             "a designator"_err_en_US);
114164ab3302SCarolineConcatto       }
114264ab3302SCarolineConcatto     } else if (!dtSpec || !dtSpec->scope()) {
114364ab3302SCarolineConcatto       CHECK(context_.AnyFatalError() || !foldingContext_.messages().empty());
114464ab3302SCarolineConcatto       return std::nullopt;
114564ab3302SCarolineConcatto     } else if (std::optional<DataRef> dataRef{
114664ab3302SCarolineConcatto                    ExtractDataRef(std::move(*dtExpr))}) {
114764ab3302SCarolineConcatto       if (auto component{
114864ab3302SCarolineConcatto               CreateComponent(std::move(*dataRef), *sym, *dtSpec->scope())}) {
114964ab3302SCarolineConcatto         return Designate(DataRef{std::move(*component)});
115064ab3302SCarolineConcatto       } else {
115164ab3302SCarolineConcatto         Say(name, "Component is not in scope of derived TYPE(%s)"_err_en_US,
115264ab3302SCarolineConcatto             dtSpec->typeSymbol().name());
115364ab3302SCarolineConcatto       }
115464ab3302SCarolineConcatto     } else {
115564ab3302SCarolineConcatto       Say(name,
115664ab3302SCarolineConcatto           "Base of component reference must be a data reference"_err_en_US);
115764ab3302SCarolineConcatto     }
115864ab3302SCarolineConcatto   } else if (auto *details{sym->detailsIf<semantics::MiscDetails>()}) {
115964ab3302SCarolineConcatto     // special part-ref: %re, %im, %kind, %len
11609f5f2eb2SPeter Klausler     // Type errors on the base of %re/%im/%len are detected and
11619f5f2eb2SPeter Klausler     // reported in name resolution.
116264ab3302SCarolineConcatto     using MiscKind = semantics::MiscDetails::Kind;
116364ab3302SCarolineConcatto     MiscKind kind{details->kind()};
116464ab3302SCarolineConcatto     if (kind == MiscKind::ComplexPartRe || kind == MiscKind::ComplexPartIm) {
116564ab3302SCarolineConcatto       if (auto *zExpr{std::get_if<Expr<SomeComplex>>(&base->u)}) {
11663a26596aSPeter Klausler         if (std::optional<DataRef> dataRef{ExtractDataRef(*zExpr)}) {
11673a26596aSPeter Klausler           // Represent %RE/%IM as a designator
1168cd03e96fSPeter Klausler           Expr<SomeReal> realExpr{common::visit(
116964ab3302SCarolineConcatto               [&](const auto &z) {
117064ab3302SCarolineConcatto                 using PartType = typename ResultType<decltype(z)>::Part;
117164ab3302SCarolineConcatto                 auto part{kind == MiscKind::ComplexPartRe
117264ab3302SCarolineConcatto                         ? ComplexPart::Part::RE
117364ab3302SCarolineConcatto                         : ComplexPart::Part::IM};
117464ab3302SCarolineConcatto                 return AsCategoryExpr(Designator<PartType>{
117564ab3302SCarolineConcatto                     ComplexPart{std::move(*dataRef), part}});
117664ab3302SCarolineConcatto               },
117764ab3302SCarolineConcatto               zExpr->u)};
11784171f80dSpeter klausler           return AsGenericExpr(std::move(realExpr));
117964ab3302SCarolineConcatto         }
118064ab3302SCarolineConcatto       }
118164ab3302SCarolineConcatto     } else if (kind == MiscKind::KindParamInquiry ||
118264ab3302SCarolineConcatto         kind == MiscKind::LenParamInquiry) {
1183fa0443f2SPeter Klausler       ActualArgument arg{std::move(*base)};
1184fa0443f2SPeter Klausler       SetArgSourceLocation(arg, name);
1185fa0443f2SPeter Klausler       return MakeFunctionRef(name, ActualArguments{std::move(arg)});
118664ab3302SCarolineConcatto     } else {
118764ab3302SCarolineConcatto       DIE("unexpected MiscDetails::Kind");
118864ab3302SCarolineConcatto     }
118964ab3302SCarolineConcatto   } else {
119064ab3302SCarolineConcatto     Say(name, "derived type required before component reference"_err_en_US);
119164ab3302SCarolineConcatto   }
119264ab3302SCarolineConcatto   return std::nullopt;
119364ab3302SCarolineConcatto }
119464ab3302SCarolineConcatto 
Analyze(const parser::CoindexedNamedObject & x)119564ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::CoindexedNamedObject &x) {
11960b130278STim Keith   if (auto maybeDataRef{ExtractDataRef(Analyze(x.base))}) {
11970b130278STim Keith     DataRef *dataRef{&*maybeDataRef};
119864ab3302SCarolineConcatto     std::vector<Subscript> subscripts;
119964ab3302SCarolineConcatto     SymbolVector reversed;
120064ab3302SCarolineConcatto     if (auto *aRef{std::get_if<ArrayRef>(&dataRef->u)}) {
120164ab3302SCarolineConcatto       subscripts = std::move(aRef->subscript());
120264ab3302SCarolineConcatto       reversed.push_back(aRef->GetLastSymbol());
120364ab3302SCarolineConcatto       if (Component * component{aRef->base().UnwrapComponent()}) {
12040b130278STim Keith         dataRef = &component->base();
120564ab3302SCarolineConcatto       } else {
12060b130278STim Keith         dataRef = nullptr;
120764ab3302SCarolineConcatto       }
120864ab3302SCarolineConcatto     }
120964ab3302SCarolineConcatto     if (dataRef) {
121064ab3302SCarolineConcatto       while (auto *component{std::get_if<Component>(&dataRef->u)}) {
121164ab3302SCarolineConcatto         reversed.push_back(component->GetLastSymbol());
12120b130278STim Keith         dataRef = &component->base();
121364ab3302SCarolineConcatto       }
121464ab3302SCarolineConcatto       if (auto *baseSym{std::get_if<SymbolRef>(&dataRef->u)}) {
121564ab3302SCarolineConcatto         reversed.push_back(*baseSym);
121664ab3302SCarolineConcatto       } else {
121764ab3302SCarolineConcatto         Say("Base of coindexed named object has subscripts or cosubscripts"_err_en_US);
121864ab3302SCarolineConcatto       }
121964ab3302SCarolineConcatto     }
122064ab3302SCarolineConcatto     std::vector<Expr<SubscriptInteger>> cosubscripts;
122164ab3302SCarolineConcatto     bool cosubsOk{true};
122264ab3302SCarolineConcatto     for (const auto &cosub :
122364ab3302SCarolineConcatto         std::get<std::list<parser::Cosubscript>>(x.imageSelector.t)) {
122464ab3302SCarolineConcatto       MaybeExpr coex{Analyze(cosub)};
122564ab3302SCarolineConcatto       if (auto *intExpr{UnwrapExpr<Expr<SomeInteger>>(coex)}) {
122664ab3302SCarolineConcatto         cosubscripts.push_back(
122764ab3302SCarolineConcatto             ConvertToType<SubscriptInteger>(std::move(*intExpr)));
122864ab3302SCarolineConcatto       } else {
122964ab3302SCarolineConcatto         cosubsOk = false;
123064ab3302SCarolineConcatto       }
123164ab3302SCarolineConcatto     }
123264ab3302SCarolineConcatto     if (cosubsOk && !reversed.empty()) {
123364ab3302SCarolineConcatto       int numCosubscripts{static_cast<int>(cosubscripts.size())};
123464ab3302SCarolineConcatto       const Symbol &symbol{reversed.front()};
123564ab3302SCarolineConcatto       if (numCosubscripts != symbol.Corank()) {
123664ab3302SCarolineConcatto         Say("'%s' has corank %d, but coindexed reference has %d cosubscripts"_err_en_US,
123764ab3302SCarolineConcatto             symbol.name(), symbol.Corank(), numCosubscripts);
123864ab3302SCarolineConcatto       }
123964ab3302SCarolineConcatto     }
124015fa287bSPete Steinfeld     for (const auto &imageSelSpec :
124115fa287bSPete Steinfeld         std::get<std::list<parser::ImageSelectorSpec>>(x.imageSelector.t)) {
1242cd03e96fSPeter Klausler       common::visit(
124315fa287bSPete Steinfeld           common::visitors{
124415fa287bSPete Steinfeld               [&](const auto &x) { Analyze(x.v); },
124515fa287bSPete Steinfeld           },
124615fa287bSPete Steinfeld           imageSelSpec.u);
124715fa287bSPete Steinfeld     }
124864ab3302SCarolineConcatto     // Reverse the chain of symbols so that the base is first and coarray
124964ab3302SCarolineConcatto     // ultimate component is last.
12509520b6c8SPete Steinfeld     if (cosubsOk) {
125164ab3302SCarolineConcatto       return Designate(
125264ab3302SCarolineConcatto           DataRef{CoarrayRef{SymbolVector{reversed.crbegin(), reversed.crend()},
125364ab3302SCarolineConcatto               std::move(subscripts), std::move(cosubscripts)}});
125464ab3302SCarolineConcatto     }
12559520b6c8SPete Steinfeld   }
125664ab3302SCarolineConcatto   return std::nullopt;
125764ab3302SCarolineConcatto }
125864ab3302SCarolineConcatto 
IntegerTypeSpecKind(const parser::IntegerTypeSpec & spec)125964ab3302SCarolineConcatto int ExpressionAnalyzer::IntegerTypeSpecKind(
126064ab3302SCarolineConcatto     const parser::IntegerTypeSpec &spec) {
126164ab3302SCarolineConcatto   Expr<SubscriptInteger> value{
126264ab3302SCarolineConcatto       AnalyzeKindSelector(TypeCategory::Integer, spec.v)};
126364ab3302SCarolineConcatto   if (auto kind{ToInt64(value)}) {
126464ab3302SCarolineConcatto     return static_cast<int>(*kind);
126564ab3302SCarolineConcatto   }
126664ab3302SCarolineConcatto   SayAt(spec, "Constant INTEGER kind value required here"_err_en_US);
126764ab3302SCarolineConcatto   return GetDefaultKind(TypeCategory::Integer);
126864ab3302SCarolineConcatto }
126964ab3302SCarolineConcatto 
127064ab3302SCarolineConcatto // Array constructors
127164ab3302SCarolineConcatto 
127264ab3302SCarolineConcatto // Inverts a collection of generic ArrayConstructorValues<SomeType> that
127364ab3302SCarolineConcatto // all happen to have the same actual type T into one ArrayConstructor<T>.
127464ab3302SCarolineConcatto template <typename T>
MakeSpecific(ArrayConstructorValues<SomeType> && from)127564ab3302SCarolineConcatto ArrayConstructorValues<T> MakeSpecific(
127664ab3302SCarolineConcatto     ArrayConstructorValues<SomeType> &&from) {
127764ab3302SCarolineConcatto   ArrayConstructorValues<T> to;
127864ab3302SCarolineConcatto   for (ArrayConstructorValue<SomeType> &x : from) {
1279cd03e96fSPeter Klausler     common::visit(
128064ab3302SCarolineConcatto         common::visitors{
128164ab3302SCarolineConcatto             [&](common::CopyableIndirection<Expr<SomeType>> &&expr) {
128264ab3302SCarolineConcatto               auto *typed{UnwrapExpr<Expr<T>>(expr.value())};
128364ab3302SCarolineConcatto               to.Push(std::move(DEREF(typed)));
128464ab3302SCarolineConcatto             },
128564ab3302SCarolineConcatto             [&](ImpliedDo<SomeType> &&impliedDo) {
128664ab3302SCarolineConcatto               to.Push(ImpliedDo<T>{impliedDo.name(),
128764ab3302SCarolineConcatto                   std::move(impliedDo.lower()), std::move(impliedDo.upper()),
128864ab3302SCarolineConcatto                   std::move(impliedDo.stride()),
128964ab3302SCarolineConcatto                   MakeSpecific<T>(std::move(impliedDo.values()))});
129064ab3302SCarolineConcatto             },
129164ab3302SCarolineConcatto         },
129264ab3302SCarolineConcatto         std::move(x.u));
129364ab3302SCarolineConcatto   }
129464ab3302SCarolineConcatto   return to;
129564ab3302SCarolineConcatto }
129664ab3302SCarolineConcatto 
129764ab3302SCarolineConcatto class ArrayConstructorContext {
129864ab3302SCarolineConcatto public:
ArrayConstructorContext(ExpressionAnalyzer & c,std::optional<DynamicTypeWithLength> && t)129964ab3302SCarolineConcatto   ArrayConstructorContext(
130064ab3302SCarolineConcatto       ExpressionAnalyzer &c, std::optional<DynamicTypeWithLength> &&t)
130164ab3302SCarolineConcatto       : exprAnalyzer_{c}, type_{std::move(t)} {}
130264ab3302SCarolineConcatto 
130364ab3302SCarolineConcatto   void Add(const parser::AcValue &);
130464ab3302SCarolineConcatto   MaybeExpr ToExpr();
130564ab3302SCarolineConcatto 
130664ab3302SCarolineConcatto   // These interfaces allow *this to be used as a type visitor argument to
130764ab3302SCarolineConcatto   // common::SearchTypes() to convert the array constructor to a typed
130864ab3302SCarolineConcatto   // expression in ToExpr().
130964ab3302SCarolineConcatto   using Result = MaybeExpr;
131064ab3302SCarolineConcatto   using Types = AllTypes;
Test()131164ab3302SCarolineConcatto   template <typename T> Result Test() {
131264ab3302SCarolineConcatto     if (type_ && type_->category() == T::category) {
131364ab3302SCarolineConcatto       if constexpr (T::category == TypeCategory::Derived) {
1314ac964175Speter klausler         if (!type_->IsUnlimitedPolymorphic()) {
13150c64cb6dSPete Steinfeld           return AsMaybeExpr(ArrayConstructor<T>{type_->GetDerivedTypeSpec(),
13160c64cb6dSPete Steinfeld               MakeSpecific<T>(std::move(values_))});
13170c64cb6dSPete Steinfeld         }
131864ab3302SCarolineConcatto       } else if (type_->kind() == T::kind) {
131964ab3302SCarolineConcatto         if constexpr (T::category == TypeCategory::Character) {
132064ab3302SCarolineConcatto           if (auto len{type_->LEN()}) {
132164ab3302SCarolineConcatto             return AsMaybeExpr(ArrayConstructor<T>{
132264ab3302SCarolineConcatto                 *std::move(len), MakeSpecific<T>(std::move(values_))});
132364ab3302SCarolineConcatto           }
132464ab3302SCarolineConcatto         } else {
132564ab3302SCarolineConcatto           return AsMaybeExpr(
132664ab3302SCarolineConcatto               ArrayConstructor<T>{MakeSpecific<T>(std::move(values_))});
132764ab3302SCarolineConcatto         }
132864ab3302SCarolineConcatto       }
132964ab3302SCarolineConcatto     }
133064ab3302SCarolineConcatto     return std::nullopt;
133164ab3302SCarolineConcatto   }
133264ab3302SCarolineConcatto 
133364ab3302SCarolineConcatto private:
1334e8f96899Speter klausler   using ImpliedDoIntType = ResultType<ImpliedDoIndex>;
1335e8f96899Speter klausler 
133664ab3302SCarolineConcatto   void Push(MaybeExpr &&);
1337e8f96899Speter klausler   void Add(const parser::AcValue::Triplet &);
1338e8f96899Speter klausler   void Add(const parser::Expr &);
1339e8f96899Speter klausler   void Add(const parser::AcImpliedDo &);
1340e8f96899Speter klausler   void UnrollConstantImpliedDo(const parser::AcImpliedDo &,
1341e8f96899Speter klausler       parser::CharBlock name, std::int64_t lower, std::int64_t upper,
1342e8f96899Speter klausler       std::int64_t stride);
134364ab3302SCarolineConcatto 
134464ab3302SCarolineConcatto   template <int KIND, typename A>
GetSpecificIntExpr(const A & x)134564ab3302SCarolineConcatto   std::optional<Expr<Type<TypeCategory::Integer, KIND>>> GetSpecificIntExpr(
134664ab3302SCarolineConcatto       const A &x) {
134764ab3302SCarolineConcatto     if (MaybeExpr y{exprAnalyzer_.Analyze(x)}) {
134864ab3302SCarolineConcatto       Expr<SomeInteger> *intExpr{UnwrapExpr<Expr<SomeInteger>>(*y)};
1349e8f96899Speter klausler       return Fold(exprAnalyzer_.GetFoldingContext(),
1350e8f96899Speter klausler           ConvertToType<Type<TypeCategory::Integer, KIND>>(
1351e8f96899Speter klausler               std::move(DEREF(intExpr))));
135264ab3302SCarolineConcatto     }
135364ab3302SCarolineConcatto     return std::nullopt;
135464ab3302SCarolineConcatto   }
135564ab3302SCarolineConcatto 
135664ab3302SCarolineConcatto   // Nested array constructors all reference the same ExpressionAnalyzer,
135764ab3302SCarolineConcatto   // which represents the nest of active implied DO loop indices.
135864ab3302SCarolineConcatto   ExpressionAnalyzer &exprAnalyzer_;
135964ab3302SCarolineConcatto   std::optional<DynamicTypeWithLength> type_;
136064ab3302SCarolineConcatto   bool explicitType_{type_.has_value()};
136164ab3302SCarolineConcatto   std::optional<std::int64_t> constantLength_;
136264ab3302SCarolineConcatto   ArrayConstructorValues<SomeType> values_;
1363e8f96899Speter klausler   std::uint64_t messageDisplayedSet_{0};
136464ab3302SCarolineConcatto };
136564ab3302SCarolineConcatto 
Push(MaybeExpr && x)136664ab3302SCarolineConcatto void ArrayConstructorContext::Push(MaybeExpr &&x) {
136764ab3302SCarolineConcatto   if (!x) {
136864ab3302SCarolineConcatto     return;
136964ab3302SCarolineConcatto   }
1370e7a53f1eSpeter klausler   if (!type_) {
1371e7a53f1eSpeter klausler     if (auto *boz{std::get_if<BOZLiteralConstant>(&x->u)}) {
1372e7a53f1eSpeter klausler       // Treat an array constructor of BOZ as if default integer.
1373e7a53f1eSpeter klausler       if (exprAnalyzer_.context().ShouldWarn(
1374e7a53f1eSpeter klausler               common::LanguageFeature::BOZAsDefaultInteger)) {
1375e7a53f1eSpeter klausler         exprAnalyzer_.Say(
1376a53967cdSPeter Klausler             "BOZ literal in array constructor without explicit type is assumed to be default INTEGER"_port_en_US);
1377e7a53f1eSpeter klausler       }
1378e7a53f1eSpeter klausler       x = AsGenericExpr(ConvertToKind<TypeCategory::Integer>(
1379e7a53f1eSpeter klausler           exprAnalyzer_.GetDefaultKind(TypeCategory::Integer),
1380e7a53f1eSpeter klausler           std::move(*boz)));
1381e7a53f1eSpeter klausler     }
1382e7a53f1eSpeter klausler   }
1383ec3049c7Speter klausler   std::optional<DynamicType> dyType{x->GetType()};
1384ec3049c7Speter klausler   if (!dyType) {
1385ec3049c7Speter klausler     if (auto *boz{std::get_if<BOZLiteralConstant>(&x->u)}) {
1386ec3049c7Speter klausler       if (!type_) {
1387ec3049c7Speter klausler         // Treat an array constructor of BOZ as if default integer.
1388ec3049c7Speter klausler         if (exprAnalyzer_.context().ShouldWarn(
1389ec3049c7Speter klausler                 common::LanguageFeature::BOZAsDefaultInteger)) {
1390ec3049c7Speter klausler           exprAnalyzer_.Say(
1391a53967cdSPeter Klausler               "BOZ literal in array constructor without explicit type is assumed to be default INTEGER"_port_en_US);
1392ec3049c7Speter klausler         }
1393ec3049c7Speter klausler         x = AsGenericExpr(ConvertToKind<TypeCategory::Integer>(
1394ec3049c7Speter klausler             exprAnalyzer_.GetDefaultKind(TypeCategory::Integer),
1395ec3049c7Speter klausler             std::move(*boz)));
1396ec3049c7Speter klausler         dyType = x.value().GetType();
1397ec3049c7Speter klausler       } else if (auto cast{ConvertToType(*type_, std::move(*x))}) {
1398ec3049c7Speter klausler         x = std::move(cast);
1399ec3049c7Speter klausler         dyType = *type_;
1400ec3049c7Speter klausler       } else {
1401ec3049c7Speter klausler         if (!(messageDisplayedSet_ & 0x80)) {
1402ec3049c7Speter klausler           exprAnalyzer_.Say(
1403ec3049c7Speter klausler               "BOZ literal is not suitable for use in this array constructor"_err_en_US);
1404ec3049c7Speter klausler           messageDisplayedSet_ |= 0x80;
1405ec3049c7Speter klausler         }
1406ec3049c7Speter klausler         return;
1407ec3049c7Speter klausler       }
1408ec3049c7Speter klausler     } else { // procedure name, &c.
1409ec3049c7Speter klausler       if (!(messageDisplayedSet_ & 0x40)) {
1410ec3049c7Speter klausler         exprAnalyzer_.Say(
1411ec3049c7Speter klausler             "Item is not suitable for use in an array constructor"_err_en_US);
1412ec3049c7Speter klausler         messageDisplayedSet_ |= 0x40;
1413ec3049c7Speter klausler       }
1414ec3049c7Speter klausler       return;
1415ec3049c7Speter klausler     }
1416ec3049c7Speter klausler   } else if (dyType->IsUnlimitedPolymorphic()) {
1417ec3049c7Speter klausler     if (!(messageDisplayedSet_ & 8)) {
1418ec3049c7Speter klausler       exprAnalyzer_.Say("Cannot have an unlimited polymorphic value in an "
1419ec3049c7Speter klausler                         "array constructor"_err_en_US); // C7113
1420ec3049c7Speter klausler       messageDisplayedSet_ |= 8;
1421ec3049c7Speter klausler     }
1422ec3049c7Speter klausler     return;
1423ec3049c7Speter klausler   }
1424ec3049c7Speter klausler   DynamicTypeWithLength xType{dyType.value()};
142564ab3302SCarolineConcatto   if (Expr<SomeCharacter> * charExpr{UnwrapExpr<Expr<SomeCharacter>>(*x)}) {
142664ab3302SCarolineConcatto     CHECK(xType.category() == TypeCategory::Character);
142764ab3302SCarolineConcatto     xType.length =
1428cd03e96fSPeter Klausler         common::visit([](const auto &kc) { return kc.LEN(); }, charExpr->u);
142964ab3302SCarolineConcatto   }
143064ab3302SCarolineConcatto   if (!type_) {
143164ab3302SCarolineConcatto     // If there is no explicit type-spec in an array constructor, the type
143264ab3302SCarolineConcatto     // of the array is the declared type of all of the elements, which must
143364ab3302SCarolineConcatto     // be well-defined and all match.
143464ab3302SCarolineConcatto     // TODO: Possible language extension: use the most general type of
143564ab3302SCarolineConcatto     // the values as the type of a numeric constructed array, convert all
143664ab3302SCarolineConcatto     // of the other values to that type.  Alternative: let the first value
143764ab3302SCarolineConcatto     // determine the type, and convert the others to that type.
143864ab3302SCarolineConcatto     CHECK(!explicitType_);
143964ab3302SCarolineConcatto     type_ = std::move(xType);
144064ab3302SCarolineConcatto     constantLength_ = ToInt64(type_->length);
144164ab3302SCarolineConcatto     values_.Push(std::move(*x));
144264ab3302SCarolineConcatto   } else if (!explicitType_) {
1443ec3049c7Speter klausler     if (type_->IsTkCompatibleWith(xType) && xType.IsTkCompatibleWith(*type_)) {
144464ab3302SCarolineConcatto       values_.Push(std::move(*x));
144564ab3302SCarolineConcatto       if (auto thisLen{ToInt64(xType.LEN())}) {
144664ab3302SCarolineConcatto         if (constantLength_) {
144764ab3302SCarolineConcatto           if (exprAnalyzer_.context().warnOnNonstandardUsage() &&
144864ab3302SCarolineConcatto               *thisLen != *constantLength_) {
1449e8f96899Speter klausler             if (!(messageDisplayedSet_ & 1)) {
145064ab3302SCarolineConcatto               exprAnalyzer_.Say(
145164ab3302SCarolineConcatto                   "Character literal in array constructor without explicit "
1452a53967cdSPeter Klausler                   "type has different length than earlier elements"_port_en_US);
1453e8f96899Speter klausler               messageDisplayedSet_ |= 1;
1454e8f96899Speter klausler             }
145564ab3302SCarolineConcatto           }
145664ab3302SCarolineConcatto           if (*thisLen > *constantLength_) {
145764ab3302SCarolineConcatto             // Language extension: use the longest literal to determine the
145864ab3302SCarolineConcatto             // length of the array constructor's character elements, not the
145964ab3302SCarolineConcatto             // first, when there is no explicit type.
146064ab3302SCarolineConcatto             *constantLength_ = *thisLen;
146164ab3302SCarolineConcatto             type_->length = xType.LEN();
146264ab3302SCarolineConcatto           }
146364ab3302SCarolineConcatto         } else {
146464ab3302SCarolineConcatto           constantLength_ = *thisLen;
146564ab3302SCarolineConcatto           type_->length = xType.LEN();
146664ab3302SCarolineConcatto         }
146764ab3302SCarolineConcatto       }
146864ab3302SCarolineConcatto     } else {
1469e8f96899Speter klausler       if (!(messageDisplayedSet_ & 2)) {
147064ab3302SCarolineConcatto         exprAnalyzer_.Say(
147164ab3302SCarolineConcatto             "Values in array constructor must have the same declared type "
147249660234Ssameeran joshi             "when no explicit type appears"_err_en_US); // C7110
1473e8f96899Speter klausler         messageDisplayedSet_ |= 2;
147449660234Ssameeran joshi       }
147564ab3302SCarolineConcatto     }
147664ab3302SCarolineConcatto   } else {
147764ab3302SCarolineConcatto     if (auto cast{ConvertToType(*type_, std::move(*x))}) {
147864ab3302SCarolineConcatto       values_.Push(std::move(*cast));
1479e8f96899Speter klausler     } else if (!(messageDisplayedSet_ & 4)) {
1480ec3049c7Speter klausler       exprAnalyzer_.Say("Value in array constructor of type '%s' could not "
148149660234Ssameeran joshi                         "be converted to the type of the array '%s'"_err_en_US,
148249660234Ssameeran joshi           x->GetType()->AsFortran(), type_->AsFortran()); // C7111, C7112
1483e8f96899Speter klausler       messageDisplayedSet_ |= 4;
148464ab3302SCarolineConcatto     }
148564ab3302SCarolineConcatto   }
148664ab3302SCarolineConcatto }
148764ab3302SCarolineConcatto 
Add(const parser::AcValue & x)148864ab3302SCarolineConcatto void ArrayConstructorContext::Add(const parser::AcValue &x) {
1489cd03e96fSPeter Klausler   common::visit(
149064ab3302SCarolineConcatto       common::visitors{
1491e8f96899Speter klausler           [&](const parser::AcValue::Triplet &triplet) { Add(triplet); },
1492e8f96899Speter klausler           [&](const common::Indirection<parser::Expr> &expr) {
1493e8f96899Speter klausler             Add(expr.value());
1494e8f96899Speter klausler           },
1495e8f96899Speter klausler           [&](const common::Indirection<parser::AcImpliedDo> &impliedDo) {
1496e8f96899Speter klausler             Add(impliedDo.value());
1497e8f96899Speter klausler           },
1498e8f96899Speter klausler       },
1499e8f96899Speter klausler       x.u);
1500e8f96899Speter klausler }
1501e8f96899Speter klausler 
1502e8f96899Speter klausler // Transforms l:u(:s) into (_,_=l,u(,s)) with an anonymous index '_'
Add(const parser::AcValue::Triplet & triplet)1503e8f96899Speter klausler void ArrayConstructorContext::Add(const parser::AcValue::Triplet &triplet) {
1504e8f96899Speter klausler   std::optional<Expr<ImpliedDoIntType>> lower{
1505e8f96899Speter klausler       GetSpecificIntExpr<ImpliedDoIntType::kind>(std::get<0>(triplet.t))};
1506e8f96899Speter klausler   std::optional<Expr<ImpliedDoIntType>> upper{
1507e8f96899Speter klausler       GetSpecificIntExpr<ImpliedDoIntType::kind>(std::get<1>(triplet.t))};
1508e8f96899Speter klausler   std::optional<Expr<ImpliedDoIntType>> stride{
1509e8f96899Speter klausler       GetSpecificIntExpr<ImpliedDoIntType::kind>(std::get<2>(triplet.t))};
151064ab3302SCarolineConcatto   if (lower && upper) {
151164ab3302SCarolineConcatto     if (!stride) {
1512e8f96899Speter klausler       stride = Expr<ImpliedDoIntType>{1};
151364ab3302SCarolineConcatto     }
151464ab3302SCarolineConcatto     if (!type_) {
1515e8f96899Speter klausler       type_ = DynamicTypeWithLength{ImpliedDoIntType::GetType()};
151664ab3302SCarolineConcatto     }
151764ab3302SCarolineConcatto     auto v{std::move(values_)};
151864ab3302SCarolineConcatto     parser::CharBlock anonymous;
151964ab3302SCarolineConcatto     Push(Expr<SomeType>{
1520e8f96899Speter klausler         Expr<SomeInteger>{Expr<ImpliedDoIntType>{ImpliedDoIndex{anonymous}}}});
152164ab3302SCarolineConcatto     std::swap(v, values_);
152264ab3302SCarolineConcatto     values_.Push(ImpliedDo<SomeType>{anonymous, std::move(*lower),
152364ab3302SCarolineConcatto         std::move(*upper), std::move(*stride), std::move(v)});
152464ab3302SCarolineConcatto   }
1525e8f96899Speter klausler }
1526e8f96899Speter klausler 
Add(const parser::Expr & expr)1527e8f96899Speter klausler void ArrayConstructorContext::Add(const parser::Expr &expr) {
1528e8f96899Speter klausler   auto restorer{exprAnalyzer_.GetContextualMessages().SetLocation(expr.source)};
1529ec3049c7Speter klausler   Push(exprAnalyzer_.Analyze(expr));
1530e8f96899Speter klausler }
1531e8f96899Speter klausler 
Add(const parser::AcImpliedDo & impliedDo)1532e8f96899Speter klausler void ArrayConstructorContext::Add(const parser::AcImpliedDo &impliedDo) {
1533e8f96899Speter klausler   const auto &control{std::get<parser::AcImpliedDoControl>(impliedDo.t)};
1534e8f96899Speter klausler   const auto &bounds{std::get<parser::AcImpliedDoControl::Bounds>(control.t)};
153564ab3302SCarolineConcatto   exprAnalyzer_.Analyze(bounds.name);
153664ab3302SCarolineConcatto   parser::CharBlock name{bounds.name.thing.thing.source};
153764ab3302SCarolineConcatto   const Symbol *symbol{bounds.name.thing.thing.symbol};
1538e8f96899Speter klausler   int kind{ImpliedDoIntType::kind};
153964ab3302SCarolineConcatto   if (const auto dynamicType{DynamicType::From(symbol)}) {
154064ab3302SCarolineConcatto     kind = dynamicType->kind();
154164ab3302SCarolineConcatto   }
1542e8f96899Speter klausler   std::optional<Expr<ImpliedDoIntType>> lower{
1543e8f96899Speter klausler       GetSpecificIntExpr<ImpliedDoIntType::kind>(bounds.lower)};
1544e8f96899Speter klausler   std::optional<Expr<ImpliedDoIntType>> upper{
1545e8f96899Speter klausler       GetSpecificIntExpr<ImpliedDoIntType::kind>(bounds.upper)};
1546e8f96899Speter klausler   if (lower && upper) {
1547e8f96899Speter klausler     std::optional<Expr<ImpliedDoIntType>> stride{
1548e8f96899Speter klausler         GetSpecificIntExpr<ImpliedDoIntType::kind>(bounds.step)};
1549e8f96899Speter klausler     if (!stride) {
1550e8f96899Speter klausler       stride = Expr<ImpliedDoIntType>{1};
1551e8f96899Speter klausler     }
15523265b933Speter klausler     if (exprAnalyzer_.AddImpliedDo(name, kind)) {
1553e8f96899Speter klausler       // Check for constant bounds; the loop may require complete unrolling
1554e8f96899Speter klausler       // of the parse tree if all bounds are constant in order to allow the
1555e8f96899Speter klausler       // implied DO loop index to qualify as a constant expression.
1556e8f96899Speter klausler       auto cLower{ToInt64(lower)};
1557e8f96899Speter klausler       auto cUpper{ToInt64(upper)};
1558e8f96899Speter klausler       auto cStride{ToInt64(stride)};
1559e8f96899Speter klausler       if (!(messageDisplayedSet_ & 0x10) && cStride && *cStride == 0) {
1560e8f96899Speter klausler         exprAnalyzer_.SayAt(bounds.step.value().thing.thing.value().source,
1561e8f96899Speter klausler             "The stride of an implied DO loop must not be zero"_err_en_US);
1562e8f96899Speter klausler         messageDisplayedSet_ |= 0x10;
1563e8f96899Speter klausler       }
1564e8f96899Speter klausler       bool isConstant{cLower && cUpper && cStride && *cStride != 0};
1565e8f96899Speter klausler       bool isNonemptyConstant{isConstant &&
1566e8f96899Speter klausler           ((*cStride > 0 && *cLower <= *cUpper) ||
1567e8f96899Speter klausler               (*cStride < 0 && *cLower >= *cUpper))};
1568e8f96899Speter klausler       bool unrollConstantLoop{false};
1569e8f96899Speter klausler       parser::Messages buffer;
1570e8f96899Speter klausler       auto saveMessagesDisplayed{messageDisplayedSet_};
1571e8f96899Speter klausler       {
1572e8f96899Speter klausler         auto messageRestorer{
1573e8f96899Speter klausler             exprAnalyzer_.GetContextualMessages().SetMessages(buffer)};
1574e8f96899Speter klausler         auto v{std::move(values_)};
1575e8f96899Speter klausler         for (const auto &value :
1576e8f96899Speter klausler             std::get<std::list<parser::AcValue>>(impliedDo.t)) {
1577e8f96899Speter klausler           Add(value);
1578e8f96899Speter klausler         }
1579e8f96899Speter klausler         std::swap(v, values_);
1580e8f96899Speter klausler         if (isNonemptyConstant && buffer.AnyFatalError()) {
1581e8f96899Speter klausler           unrollConstantLoop = true;
1582e8f96899Speter klausler         } else {
1583e8f96899Speter klausler           values_.Push(ImpliedDo<SomeType>{name, std::move(*lower),
1584e8f96899Speter klausler               std::move(*upper), std::move(*stride), std::move(v)});
1585e8f96899Speter klausler         }
1586e8f96899Speter klausler       }
1587e8f96899Speter klausler       if (unrollConstantLoop) {
1588e8f96899Speter klausler         messageDisplayedSet_ = saveMessagesDisplayed;
1589e8f96899Speter klausler         UnrollConstantImpliedDo(impliedDo, name, *cLower, *cUpper, *cStride);
1590e8f96899Speter klausler       } else if (auto *messages{
1591e8f96899Speter klausler                      exprAnalyzer_.GetContextualMessages().messages()}) {
1592e8f96899Speter klausler         messages->Annex(std::move(buffer));
1593e8f96899Speter klausler       }
1594e8f96899Speter klausler       exprAnalyzer_.RemoveImpliedDo(name);
15953265b933Speter klausler     } else if (!(messageDisplayedSet_ & 0x20)) {
15963265b933Speter klausler       exprAnalyzer_.SayAt(name,
15973265b933Speter klausler           "Implied DO index '%s' is active in a surrounding implied DO loop "
15983265b933Speter klausler           "and may not have the same name"_err_en_US,
15993265b933Speter klausler           name); // C7115
16003265b933Speter klausler       messageDisplayedSet_ |= 0x20;
16013265b933Speter klausler     }
16023265b933Speter klausler   }
1603e8f96899Speter klausler }
1604e8f96899Speter klausler 
1605e8f96899Speter klausler // Fortran considers an implied DO index of an array constructor to be
1606e8f96899Speter klausler // a constant expression if the bounds of the implied DO loop are constant.
1607e8f96899Speter klausler // Usually this doesn't matter, but if we emitted spurious messages as a
1608e8f96899Speter klausler // result of not using constant values for the index while analyzing the
1609e8f96899Speter klausler // items, we need to do it again the "hard" way with multiple iterations over
1610e8f96899Speter klausler // the parse tree.
UnrollConstantImpliedDo(const parser::AcImpliedDo & impliedDo,parser::CharBlock name,std::int64_t lower,std::int64_t upper,std::int64_t stride)1611e8f96899Speter klausler void ArrayConstructorContext::UnrollConstantImpliedDo(
1612e8f96899Speter klausler     const parser::AcImpliedDo &impliedDo, parser::CharBlock name,
1613e8f96899Speter klausler     std::int64_t lower, std::int64_t upper, std::int64_t stride) {
1614e8f96899Speter klausler   auto &foldingContext{exprAnalyzer_.GetFoldingContext()};
1615e8f96899Speter klausler   auto restorer{exprAnalyzer_.DoNotUseSavedTypedExprs()};
1616e8f96899Speter klausler   for (auto &at{foldingContext.StartImpliedDo(name, lower)};
1617e8f96899Speter klausler        (stride > 0 && at <= upper) || (stride < 0 && at >= upper);
1618e8f96899Speter klausler        at += stride) {
1619e8f96899Speter klausler     for (const auto &value :
1620e8f96899Speter klausler         std::get<std::list<parser::AcValue>>(impliedDo.t)) {
1621e8f96899Speter klausler       Add(value);
1622e8f96899Speter klausler     }
1623e8f96899Speter klausler   }
1624e8f96899Speter klausler   foldingContext.EndImpliedDo(name);
162564ab3302SCarolineConcatto }
162664ab3302SCarolineConcatto 
ToExpr()162764ab3302SCarolineConcatto MaybeExpr ArrayConstructorContext::ToExpr() {
162864ab3302SCarolineConcatto   return common::SearchTypes(std::move(*this));
162964ab3302SCarolineConcatto }
163064ab3302SCarolineConcatto 
Analyze(const parser::ArrayConstructor & array)163164ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::ArrayConstructor &array) {
163264ab3302SCarolineConcatto   const parser::AcSpec &acSpec{array.v};
163364ab3302SCarolineConcatto   ArrayConstructorContext acContext{*this, AnalyzeTypeSpec(acSpec.type)};
163464ab3302SCarolineConcatto   for (const parser::AcValue &value : acSpec.values) {
163564ab3302SCarolineConcatto     acContext.Add(value);
163664ab3302SCarolineConcatto   }
163764ab3302SCarolineConcatto   return acContext.ToExpr();
163864ab3302SCarolineConcatto }
163964ab3302SCarolineConcatto 
Analyze(const parser::StructureConstructor & structure)164064ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(
164164ab3302SCarolineConcatto     const parser::StructureConstructor &structure) {
164264ab3302SCarolineConcatto   auto &parsedType{std::get<parser::DerivedTypeSpec>(structure.t)};
16438b550af7SPeter Steinfeld   parser::Name structureType{std::get<parser::Name>(parsedType.t)};
16448b550af7SPeter Steinfeld   parser::CharBlock &typeName{structureType.source};
16458b550af7SPeter Steinfeld   if (semantics::Symbol * typeSymbol{structureType.symbol}) {
16468b550af7SPeter Steinfeld     if (typeSymbol->has<semantics::DerivedTypeDetails>()) {
16478b550af7SPeter Steinfeld       semantics::DerivedTypeSpec dtSpec{typeName, typeSymbol->GetUltimate()};
16488b550af7SPeter Steinfeld       if (!CheckIsValidForwardReference(dtSpec)) {
16498b550af7SPeter Steinfeld         return std::nullopt;
16508b550af7SPeter Steinfeld       }
16518b550af7SPeter Steinfeld     }
16528b550af7SPeter Steinfeld   }
165364ab3302SCarolineConcatto   if (!parsedType.derivedTypeSpec) {
165464ab3302SCarolineConcatto     return std::nullopt;
165564ab3302SCarolineConcatto   }
165664ab3302SCarolineConcatto   const auto &spec{*parsedType.derivedTypeSpec};
165764ab3302SCarolineConcatto   const Symbol &typeSymbol{spec.typeSymbol()};
165864ab3302SCarolineConcatto   if (!spec.scope() || !typeSymbol.has<semantics::DerivedTypeDetails>()) {
165964ab3302SCarolineConcatto     return std::nullopt; // error recovery
166064ab3302SCarolineConcatto   }
166164ab3302SCarolineConcatto   const auto &typeDetails{typeSymbol.get<semantics::DerivedTypeDetails>()};
166264ab3302SCarolineConcatto   const Symbol *parentComponent{typeDetails.GetParentComponent(*spec.scope())};
166364ab3302SCarolineConcatto 
166464ab3302SCarolineConcatto   if (typeSymbol.attrs().test(semantics::Attr::ABSTRACT)) { // C796
166564ab3302SCarolineConcatto     AttachDeclaration(Say(typeName,
166664ab3302SCarolineConcatto                           "ABSTRACT derived type '%s' may not be used in a "
166764ab3302SCarolineConcatto                           "structure constructor"_err_en_US,
166864ab3302SCarolineConcatto                           typeName),
166949660234Ssameeran joshi         typeSymbol); // C7114
167064ab3302SCarolineConcatto   }
167164ab3302SCarolineConcatto 
167264ab3302SCarolineConcatto   // This iterator traverses all of the components in the derived type and its
167364ab3302SCarolineConcatto   // parents.  The symbols for whole parent components appear after their
167464ab3302SCarolineConcatto   // own components and before the components of the types that extend them.
167564ab3302SCarolineConcatto   // E.g., TYPE :: A; REAL X; END TYPE
167664ab3302SCarolineConcatto   //       TYPE, EXTENDS(A) :: B; REAL Y; END TYPE
167764ab3302SCarolineConcatto   // produces the component list X, A, Y.
167864ab3302SCarolineConcatto   // The order is important below because a structure constructor can
167964ab3302SCarolineConcatto   // initialize X or A by name, but not both.
168064ab3302SCarolineConcatto   auto components{semantics::OrderedComponentIterator{spec}};
168164ab3302SCarolineConcatto   auto nextAnonymous{components.begin()};
168264ab3302SCarolineConcatto 
168364ab3302SCarolineConcatto   std::set<parser::CharBlock> unavailable;
168464ab3302SCarolineConcatto   bool anyKeyword{false};
168564ab3302SCarolineConcatto   StructureConstructor result{spec};
168664ab3302SCarolineConcatto   bool checkConflicts{true}; // until we hit one
16874171f80dSpeter klausler   auto &messages{GetContextualMessages()};
168864ab3302SCarolineConcatto 
168964ab3302SCarolineConcatto   for (const auto &component :
169064ab3302SCarolineConcatto       std::get<std::list<parser::ComponentSpec>>(structure.t)) {
169164ab3302SCarolineConcatto     const parser::Expr &expr{
169264ab3302SCarolineConcatto         std::get<parser::ComponentDataSource>(component.t).v.value()};
169364ab3302SCarolineConcatto     parser::CharBlock source{expr.source};
169464ab3302SCarolineConcatto     auto restorer{messages.SetLocation(source)};
169564ab3302SCarolineConcatto     const Symbol *symbol{nullptr};
169664ab3302SCarolineConcatto     MaybeExpr value{Analyze(expr)};
169764ab3302SCarolineConcatto     std::optional<DynamicType> valueType{DynamicType::From(value)};
169864ab3302SCarolineConcatto     if (const auto &kw{std::get<std::optional<parser::Keyword>>(component.t)}) {
169964ab3302SCarolineConcatto       anyKeyword = true;
170064ab3302SCarolineConcatto       source = kw->v.source;
170164ab3302SCarolineConcatto       symbol = kw->v.symbol;
170264ab3302SCarolineConcatto       if (!symbol) {
170364ab3302SCarolineConcatto         auto componentIter{std::find_if(components.begin(), components.end(),
170464ab3302SCarolineConcatto             [=](const Symbol &symbol) { return symbol.name() == source; })};
170564ab3302SCarolineConcatto         if (componentIter != components.end()) {
170664ab3302SCarolineConcatto           symbol = &*componentIter;
170764ab3302SCarolineConcatto         }
170864ab3302SCarolineConcatto       }
170964ab3302SCarolineConcatto       if (!symbol) { // C7101
171064ab3302SCarolineConcatto         Say(source,
171164ab3302SCarolineConcatto             "Keyword '%s=' does not name a component of derived type '%s'"_err_en_US,
171264ab3302SCarolineConcatto             source, typeName);
171364ab3302SCarolineConcatto       }
171464ab3302SCarolineConcatto     } else {
171564ab3302SCarolineConcatto       if (anyKeyword) { // C7100
171664ab3302SCarolineConcatto         Say(source,
171764ab3302SCarolineConcatto             "Value in structure constructor lacks a component name"_err_en_US);
171864ab3302SCarolineConcatto         checkConflicts = false; // stem cascade
171964ab3302SCarolineConcatto       }
172064ab3302SCarolineConcatto       // Here's a regrettably common extension of the standard: anonymous
172164ab3302SCarolineConcatto       // initialization of parent components, e.g., T(PT(1)) rather than
172264ab3302SCarolineConcatto       // T(1) or T(PT=PT(1)).
172364ab3302SCarolineConcatto       if (nextAnonymous == components.begin() && parentComponent &&
172464ab3302SCarolineConcatto           valueType == DynamicType::From(*parentComponent) &&
172564ab3302SCarolineConcatto           context().IsEnabled(LanguageFeature::AnonymousParents)) {
172664ab3302SCarolineConcatto         auto iter{
172764ab3302SCarolineConcatto             std::find(components.begin(), components.end(), *parentComponent)};
172864ab3302SCarolineConcatto         if (iter != components.end()) {
172964ab3302SCarolineConcatto           symbol = parentComponent;
173064ab3302SCarolineConcatto           nextAnonymous = ++iter;
173164ab3302SCarolineConcatto           if (context().ShouldWarn(LanguageFeature::AnonymousParents)) {
173264ab3302SCarolineConcatto             Say(source,
173364ab3302SCarolineConcatto                 "Whole parent component '%s' in structure "
1734a53967cdSPeter Klausler                 "constructor should not be anonymous"_port_en_US,
173564ab3302SCarolineConcatto                 symbol->name());
173664ab3302SCarolineConcatto           }
173764ab3302SCarolineConcatto         }
173864ab3302SCarolineConcatto       }
173964ab3302SCarolineConcatto       while (!symbol && nextAnonymous != components.end()) {
174064ab3302SCarolineConcatto         const Symbol &next{*nextAnonymous};
174164ab3302SCarolineConcatto         ++nextAnonymous;
174264ab3302SCarolineConcatto         if (!next.test(Symbol::Flag::ParentComp)) {
174364ab3302SCarolineConcatto           symbol = &next;
174464ab3302SCarolineConcatto         }
174564ab3302SCarolineConcatto       }
174664ab3302SCarolineConcatto       if (!symbol) {
174764ab3302SCarolineConcatto         Say(source, "Unexpected value in structure constructor"_err_en_US);
174864ab3302SCarolineConcatto       }
174964ab3302SCarolineConcatto     }
175064ab3302SCarolineConcatto     if (symbol) {
175147452b96STim Keith       if (const auto *currScope{context_.globalScope().FindScope(source)}) {
175247452b96STim Keith         if (auto msg{CheckAccessibleComponent(*currScope, *symbol)}) {
175347452b96STim Keith           Say(source, *msg);
175447452b96STim Keith         }
175547452b96STim Keith       }
175664ab3302SCarolineConcatto       if (checkConflicts) {
175764ab3302SCarolineConcatto         auto componentIter{
175864ab3302SCarolineConcatto             std::find(components.begin(), components.end(), *symbol)};
175964ab3302SCarolineConcatto         if (unavailable.find(symbol->name()) != unavailable.cend()) {
176064ab3302SCarolineConcatto           // C797, C798
176164ab3302SCarolineConcatto           Say(source,
176264ab3302SCarolineConcatto               "Component '%s' conflicts with another component earlier in "
176364ab3302SCarolineConcatto               "this structure constructor"_err_en_US,
176464ab3302SCarolineConcatto               symbol->name());
176564ab3302SCarolineConcatto         } else if (symbol->test(Symbol::Flag::ParentComp)) {
176664ab3302SCarolineConcatto           // Make earlier components unavailable once a whole parent appears.
176764ab3302SCarolineConcatto           for (auto it{components.begin()}; it != componentIter; ++it) {
176864ab3302SCarolineConcatto             unavailable.insert(it->name());
176964ab3302SCarolineConcatto           }
177064ab3302SCarolineConcatto         } else {
177164ab3302SCarolineConcatto           // Make whole parent components unavailable after any of their
177264ab3302SCarolineConcatto           // constituents appear.
177364ab3302SCarolineConcatto           for (auto it{componentIter}; it != components.end(); ++it) {
177464ab3302SCarolineConcatto             if (it->test(Symbol::Flag::ParentComp)) {
177564ab3302SCarolineConcatto               unavailable.insert(it->name());
177664ab3302SCarolineConcatto             }
177764ab3302SCarolineConcatto           }
177864ab3302SCarolineConcatto         }
177964ab3302SCarolineConcatto       }
178064ab3302SCarolineConcatto       unavailable.insert(symbol->name());
178164ab3302SCarolineConcatto       if (value) {
178264ab3302SCarolineConcatto         if (symbol->has<semantics::ProcEntityDetails>()) {
178364ab3302SCarolineConcatto           CHECK(IsPointer(*symbol));
178464ab3302SCarolineConcatto         } else if (symbol->has<semantics::ObjectEntityDetails>()) {
178564ab3302SCarolineConcatto           // C1594(4)
178664ab3302SCarolineConcatto           const auto &innermost{context_.FindScope(expr.source)};
178747452b96STim Keith           if (const auto *pureProc{FindPureProcedureContaining(innermost)}) {
178847452b96STim Keith             if (const Symbol * pointer{FindPointerComponent(*symbol)}) {
178964ab3302SCarolineConcatto               if (const Symbol *
179047452b96STim Keith                   object{FindExternallyVisibleObject(*value, *pureProc)}) {
179164ab3302SCarolineConcatto                 if (auto *msg{Say(expr.source,
179264ab3302SCarolineConcatto                         "Externally visible object '%s' may not be "
179364ab3302SCarolineConcatto                         "associated with pointer component '%s' in a "
179464ab3302SCarolineConcatto                         "pure procedure"_err_en_US,
179564ab3302SCarolineConcatto                         object->name(), pointer->name())}) {
179664ab3302SCarolineConcatto                   msg->Attach(object->name(), "Object declaration"_en_US)
179764ab3302SCarolineConcatto                       .Attach(pointer->name(), "Pointer declaration"_en_US);
179864ab3302SCarolineConcatto                 }
179964ab3302SCarolineConcatto               }
180064ab3302SCarolineConcatto             }
180164ab3302SCarolineConcatto           }
180264ab3302SCarolineConcatto         } else if (symbol->has<semantics::TypeParamDetails>()) {
180364ab3302SCarolineConcatto           Say(expr.source,
180464ab3302SCarolineConcatto               "Type parameter '%s' may not appear as a component "
180564ab3302SCarolineConcatto               "of a structure constructor"_err_en_US,
180664ab3302SCarolineConcatto               symbol->name());
180764ab3302SCarolineConcatto           continue;
180864ab3302SCarolineConcatto         } else {
180964ab3302SCarolineConcatto           Say(expr.source,
181064ab3302SCarolineConcatto               "Component '%s' is neither a procedure pointer "
181164ab3302SCarolineConcatto               "nor a data object"_err_en_US,
181264ab3302SCarolineConcatto               symbol->name());
181364ab3302SCarolineConcatto           continue;
181464ab3302SCarolineConcatto         }
181564ab3302SCarolineConcatto         if (IsPointer(*symbol)) {
181664ab3302SCarolineConcatto           semantics::CheckPointerAssignment(
181764ab3302SCarolineConcatto               GetFoldingContext(), *symbol, *value); // C7104, C7105
181864ab3302SCarolineConcatto           result.Add(*symbol, Fold(std::move(*value)));
181964ab3302SCarolineConcatto         } else if (MaybeExpr converted{
182064ab3302SCarolineConcatto                        ConvertToType(*symbol, std::move(*value))}) {
18214171f80dSpeter klausler           if (auto componentShape{GetShape(GetFoldingContext(), *symbol)}) {
18224171f80dSpeter klausler             if (auto valueShape{GetShape(GetFoldingContext(), *converted)}) {
18234171f80dSpeter klausler               if (GetRank(*componentShape) == 0 && GetRank(*valueShape) > 0) {
18244171f80dSpeter klausler                 AttachDeclaration(
18254171f80dSpeter klausler                     Say(expr.source,
18264171f80dSpeter klausler                         "Rank-%d array value is not compatible with scalar component '%s'"_err_en_US,
1827f862d858Speter klausler                         GetRank(*valueShape), symbol->name()),
18284171f80dSpeter klausler                     *symbol);
1829dfecbcaeSpeter klausler               } else {
1830dfecbcaeSpeter klausler                 auto checked{
1831dfecbcaeSpeter klausler                     CheckConformance(messages, *componentShape, *valueShape,
1832dfecbcaeSpeter klausler                         CheckConformanceFlags::RightIsExpandableDeferred,
1833dfecbcaeSpeter klausler                         "component", "value")};
1834dfecbcaeSpeter klausler                 if (checked && *checked && GetRank(*componentShape) > 0 &&
1835dfecbcaeSpeter klausler                     GetRank(*valueShape) == 0 &&
18364171f80dSpeter klausler                     !IsExpandableScalar(*converted)) {
18374171f80dSpeter klausler                   AttachDeclaration(
18384171f80dSpeter klausler                       Say(expr.source,
18394171f80dSpeter klausler                           "Scalar value cannot be expanded to shape of array component '%s'"_err_en_US,
18404171f80dSpeter klausler                           symbol->name()),
18414171f80dSpeter klausler                       *symbol);
1842dfecbcaeSpeter klausler                 }
1843dfecbcaeSpeter klausler                 if (checked.value_or(true)) {
184464ab3302SCarolineConcatto                   result.Add(*symbol, std::move(*converted));
18454171f80dSpeter klausler                 }
18464171f80dSpeter klausler               }
18474171f80dSpeter klausler             } else {
18484171f80dSpeter klausler               Say(expr.source, "Shape of value cannot be determined"_err_en_US);
18494171f80dSpeter klausler             }
18504171f80dSpeter klausler           } else {
18514171f80dSpeter klausler             AttachDeclaration(
18524171f80dSpeter klausler                 Say(expr.source,
18534171f80dSpeter klausler                     "Shape of component '%s' cannot be determined"_err_en_US,
18544171f80dSpeter klausler                     symbol->name()),
18554171f80dSpeter klausler                 *symbol);
18564171f80dSpeter klausler           }
18570fcda9aeSpeter klausler         } else if (IsAllocatable(*symbol) && IsBareNullPointer(&*value)) {
185896aa4810SPeter Klausler           // NULL() with no arguments allowed by 7.5.10 para 6 for ALLOCATABLE.
185996aa4810SPeter Klausler           result.Add(*symbol, Expr<SomeType>{NullPointer{}});
186064ab3302SCarolineConcatto         } else if (auto symType{DynamicType::From(symbol)}) {
1861bb59b38eSPeter Klausler           if (IsAllocatable(*symbol) && symType->IsUnlimitedPolymorphic() &&
1862bb59b38eSPeter Klausler               valueType) {
1863bb59b38eSPeter Klausler             // ok
1864bb59b38eSPeter Klausler           } else if (valueType) {
186564ab3302SCarolineConcatto             AttachDeclaration(
186664ab3302SCarolineConcatto                 Say(expr.source,
1867bca13174SPeter Klausler                     "Value in structure constructor of type '%s' is "
1868bca13174SPeter Klausler                     "incompatible with component '%s' of type '%s'"_err_en_US,
186964ab3302SCarolineConcatto                     valueType->AsFortran(), symbol->name(),
187064ab3302SCarolineConcatto                     symType->AsFortran()),
187164ab3302SCarolineConcatto                 *symbol);
187264ab3302SCarolineConcatto           } else {
187364ab3302SCarolineConcatto             AttachDeclaration(
187464ab3302SCarolineConcatto                 Say(expr.source,
187564ab3302SCarolineConcatto                     "Value in structure constructor is incompatible with "
187664ab3302SCarolineConcatto                     "component '%s' of type %s"_err_en_US,
187764ab3302SCarolineConcatto                     symbol->name(), symType->AsFortran()),
187864ab3302SCarolineConcatto                 *symbol);
187964ab3302SCarolineConcatto           }
188064ab3302SCarolineConcatto         }
188164ab3302SCarolineConcatto       }
188264ab3302SCarolineConcatto     }
188364ab3302SCarolineConcatto   }
188464ab3302SCarolineConcatto 
188564ab3302SCarolineConcatto   // Ensure that unmentioned component objects have default initializers.
188664ab3302SCarolineConcatto   for (const Symbol &symbol : components) {
188764ab3302SCarolineConcatto     if (!symbol.test(Symbol::Flag::ParentComp) &&
188896aa4810SPeter Klausler         unavailable.find(symbol.name()) == unavailable.cend()) {
188996aa4810SPeter Klausler       if (IsAllocatable(symbol)) {
189096aa4810SPeter Klausler         // Set all remaining allocatables to explicit NULL()
189196aa4810SPeter Klausler         result.Add(symbol, Expr<SomeType>{NullPointer{}});
189296aa4810SPeter Klausler       } else if (const auto *details{
189364ab3302SCarolineConcatto                      symbol.detailsIf<semantics::ObjectEntityDetails>()}) {
189464ab3302SCarolineConcatto         if (details->init()) {
189564ab3302SCarolineConcatto           result.Add(symbol, common::Clone(*details->init()));
189664ab3302SCarolineConcatto         } else { // C799
189764ab3302SCarolineConcatto           AttachDeclaration(Say(typeName,
189864ab3302SCarolineConcatto                                 "Structure constructor lacks a value for "
189964ab3302SCarolineConcatto                                 "component '%s'"_err_en_US,
190064ab3302SCarolineConcatto                                 symbol.name()),
190164ab3302SCarolineConcatto               symbol);
190264ab3302SCarolineConcatto         }
190364ab3302SCarolineConcatto       }
190464ab3302SCarolineConcatto     }
190564ab3302SCarolineConcatto   }
190664ab3302SCarolineConcatto 
190764ab3302SCarolineConcatto   return AsMaybeExpr(Expr<SomeDerived>{std::move(result)});
190864ab3302SCarolineConcatto }
190964ab3302SCarolineConcatto 
GetPassName(const semantics::Symbol & proc)191064ab3302SCarolineConcatto static std::optional<parser::CharBlock> GetPassName(
191164ab3302SCarolineConcatto     const semantics::Symbol &proc) {
1912cd03e96fSPeter Klausler   return common::visit(
191364ab3302SCarolineConcatto       [](const auto &details) {
191464ab3302SCarolineConcatto         if constexpr (std::is_base_of_v<semantics::WithPassArg,
191564ab3302SCarolineConcatto                           std::decay_t<decltype(details)>>) {
191664ab3302SCarolineConcatto           return details.passName();
191764ab3302SCarolineConcatto         } else {
191864ab3302SCarolineConcatto           return std::optional<parser::CharBlock>{};
191964ab3302SCarolineConcatto         }
192064ab3302SCarolineConcatto       },
192164ab3302SCarolineConcatto       proc.details());
192264ab3302SCarolineConcatto }
192364ab3302SCarolineConcatto 
GetPassIndex(const Symbol & proc)192464ab3302SCarolineConcatto static int GetPassIndex(const Symbol &proc) {
192564ab3302SCarolineConcatto   CHECK(!proc.attrs().test(semantics::Attr::NOPASS));
192664ab3302SCarolineConcatto   std::optional<parser::CharBlock> passName{GetPassName(proc)};
192764ab3302SCarolineConcatto   const auto *interface { semantics::FindInterface(proc) };
192864ab3302SCarolineConcatto   if (!passName || !interface) {
192964ab3302SCarolineConcatto     return 0; // first argument is passed-object
193064ab3302SCarolineConcatto   }
193164ab3302SCarolineConcatto   const auto &subp{interface->get<semantics::SubprogramDetails>()};
193264ab3302SCarolineConcatto   int index{0};
193364ab3302SCarolineConcatto   for (const auto *arg : subp.dummyArgs()) {
193464ab3302SCarolineConcatto     if (arg && arg->name() == passName) {
193564ab3302SCarolineConcatto       return index;
193664ab3302SCarolineConcatto     }
193764ab3302SCarolineConcatto     ++index;
193864ab3302SCarolineConcatto   }
193964ab3302SCarolineConcatto   DIE("PASS argument name not in dummy argument list");
194064ab3302SCarolineConcatto }
194164ab3302SCarolineConcatto 
194264ab3302SCarolineConcatto // Injects an expression into an actual argument list as the "passed object"
194364ab3302SCarolineConcatto // for a type-bound procedure reference that is not NOPASS.  Adds an
194464ab3302SCarolineConcatto // argument keyword if possible, but not when the passed object goes
194564ab3302SCarolineConcatto // before a positional argument.
194664ab3302SCarolineConcatto // e.g., obj%tbp(x) -> tbp(obj,x).
AddPassArg(ActualArguments & actuals,const Expr<SomeDerived> & expr,const Symbol & component,bool isPassedObject=true)194764ab3302SCarolineConcatto static void AddPassArg(ActualArguments &actuals, const Expr<SomeDerived> &expr,
194864ab3302SCarolineConcatto     const Symbol &component, bool isPassedObject = true) {
194964ab3302SCarolineConcatto   if (component.attrs().test(semantics::Attr::NOPASS)) {
195064ab3302SCarolineConcatto     return;
195164ab3302SCarolineConcatto   }
195264ab3302SCarolineConcatto   int passIndex{GetPassIndex(component)};
195364ab3302SCarolineConcatto   auto iter{actuals.begin()};
195464ab3302SCarolineConcatto   int at{0};
195564ab3302SCarolineConcatto   while (iter < actuals.end() && at < passIndex) {
195664ab3302SCarolineConcatto     if (*iter && (*iter)->keyword()) {
195764ab3302SCarolineConcatto       iter = actuals.end();
195864ab3302SCarolineConcatto       break;
195964ab3302SCarolineConcatto     }
196064ab3302SCarolineConcatto     ++iter;
196164ab3302SCarolineConcatto     ++at;
196264ab3302SCarolineConcatto   }
196364ab3302SCarolineConcatto   ActualArgument passed{AsGenericExpr(common::Clone(expr))};
196464ab3302SCarolineConcatto   passed.set_isPassedObject(isPassedObject);
196564ab3302SCarolineConcatto   if (iter == actuals.end()) {
196664ab3302SCarolineConcatto     if (auto passName{GetPassName(component)}) {
196764ab3302SCarolineConcatto       passed.set_keyword(*passName);
196864ab3302SCarolineConcatto     }
196964ab3302SCarolineConcatto   }
197064ab3302SCarolineConcatto   actuals.emplace(iter, std::move(passed));
197164ab3302SCarolineConcatto }
197264ab3302SCarolineConcatto 
197364ab3302SCarolineConcatto // Return the compile-time resolution of a procedure binding, if possible.
GetBindingResolution(const std::optional<DynamicType> & baseType,const Symbol & component)197464ab3302SCarolineConcatto static const Symbol *GetBindingResolution(
197564ab3302SCarolineConcatto     const std::optional<DynamicType> &baseType, const Symbol &component) {
197664ab3302SCarolineConcatto   const auto *binding{component.detailsIf<semantics::ProcBindingDetails>()};
197764ab3302SCarolineConcatto   if (!binding) {
197864ab3302SCarolineConcatto     return nullptr;
197964ab3302SCarolineConcatto   }
198064ab3302SCarolineConcatto   if (!component.attrs().test(semantics::Attr::NON_OVERRIDABLE) &&
198164ab3302SCarolineConcatto       (!baseType || baseType->IsPolymorphic())) {
198264ab3302SCarolineConcatto     return nullptr;
198364ab3302SCarolineConcatto   }
198464ab3302SCarolineConcatto   return &binding->symbol();
198564ab3302SCarolineConcatto }
198664ab3302SCarolineConcatto 
AnalyzeProcedureComponentRef(const parser::ProcComponentRef & pcr,ActualArguments && arguments,bool isSubroutine)198764ab3302SCarolineConcatto auto ExpressionAnalyzer::AnalyzeProcedureComponentRef(
19880c190575SPeter Klausler     const parser::ProcComponentRef &pcr, ActualArguments &&arguments,
19890c190575SPeter Klausler     bool isSubroutine) -> std::optional<CalleeAndArguments> {
199064ab3302SCarolineConcatto   const parser::StructureComponent &sc{pcr.v.thing};
199164ab3302SCarolineConcatto   if (MaybeExpr base{Analyze(sc.base)}) {
199264ab3302SCarolineConcatto     if (const Symbol * sym{sc.component.symbol}) {
1993b6b589caSPeter Steinfeld       if (context_.HasError(sym)) {
1994b6b589caSPeter Steinfeld         return std::nullopt;
1995b6b589caSPeter Steinfeld       }
1996e5813a68Speter klausler       if (!IsProcedure(*sym)) {
1997e5813a68Speter klausler         AttachDeclaration(
1998e5813a68Speter klausler             Say(sc.component.source, "'%s' is not a procedure"_err_en_US,
1999e5813a68Speter klausler                 sc.component.source),
2000e5813a68Speter klausler             *sym);
2001562bfe12Speter klausler         return std::nullopt;
2002e5813a68Speter klausler       }
200364ab3302SCarolineConcatto       if (auto *dtExpr{UnwrapExpr<Expr<SomeDerived>>(*base)}) {
200464ab3302SCarolineConcatto         if (sym->has<semantics::GenericDetails>()) {
200564ab3302SCarolineConcatto           AdjustActuals adjustment{
200664ab3302SCarolineConcatto               [&](const Symbol &proc, ActualArguments &actuals) {
200764ab3302SCarolineConcatto                 if (!proc.attrs().test(semantics::Attr::NOPASS)) {
200864ab3302SCarolineConcatto                   AddPassArg(actuals, std::move(*dtExpr), proc);
200964ab3302SCarolineConcatto                 }
201064ab3302SCarolineConcatto                 return true;
201164ab3302SCarolineConcatto               }};
20120c190575SPeter Klausler           auto pair{ResolveGeneric(*sym, arguments, adjustment, isSubroutine)};
20130fcda9aeSpeter klausler           sym = pair.first;
2014bd92bca5SPeter Klausler           if (sym) {
2015bd92bca5SPeter Klausler             // re-resolve the name to the specific binding
2016bd92bca5SPeter Klausler             sc.component.symbol = const_cast<Symbol *>(sym);
2017bd92bca5SPeter Klausler           } else {
20180fcda9aeSpeter klausler             EmitGenericResolutionError(*sc.component.symbol, pair.second);
201964ab3302SCarolineConcatto             return std::nullopt;
202064ab3302SCarolineConcatto           }
202164ab3302SCarolineConcatto         }
2022c6d8aa27SPeixin-Qiao         std::optional<DataRef> dataRef{ExtractDataRef(std::move(*dtExpr))};
2023c6d8aa27SPeixin-Qiao         if (dataRef.has_value() && !CheckRanks(std::move(*dataRef))) {
2024c6d8aa27SPeixin-Qiao           return std::nullopt;
2025c6d8aa27SPeixin-Qiao         }
202664ab3302SCarolineConcatto         if (const Symbol *
202764ab3302SCarolineConcatto             resolution{GetBindingResolution(dtExpr->GetType(), *sym)}) {
202864ab3302SCarolineConcatto           AddPassArg(arguments, std::move(*dtExpr), *sym, false);
202964ab3302SCarolineConcatto           return CalleeAndArguments{
203064ab3302SCarolineConcatto               ProcedureDesignator{*resolution}, std::move(arguments)};
2031c6d8aa27SPeixin-Qiao         } else if (dataRef.has_value()) {
203264ab3302SCarolineConcatto           if (sym->attrs().test(semantics::Attr::NOPASS)) {
203364ab3302SCarolineConcatto             return CalleeAndArguments{
203464ab3302SCarolineConcatto                 ProcedureDesignator{Component{std::move(*dataRef), *sym}},
203564ab3302SCarolineConcatto                 std::move(arguments)};
203664ab3302SCarolineConcatto           } else {
203764ab3302SCarolineConcatto             AddPassArg(arguments,
203864ab3302SCarolineConcatto                 Expr<SomeDerived>{Designator<SomeDerived>{std::move(*dataRef)}},
203964ab3302SCarolineConcatto                 *sym);
204064ab3302SCarolineConcatto             return CalleeAndArguments{
204164ab3302SCarolineConcatto                 ProcedureDesignator{*sym}, std::move(arguments)};
204264ab3302SCarolineConcatto           }
204364ab3302SCarolineConcatto         }
204464ab3302SCarolineConcatto       }
2045ce6dd973STim Keith       Say(sc.component.source,
204664ab3302SCarolineConcatto           "Base of procedure component reference is not a derived-type object"_err_en_US);
204764ab3302SCarolineConcatto     }
204864ab3302SCarolineConcatto   }
2049f411c1ddSPeter Klausler   CHECK(context_.AnyFatalError());
205064ab3302SCarolineConcatto   return std::nullopt;
205164ab3302SCarolineConcatto }
205264ab3302SCarolineConcatto 
205364ab3302SCarolineConcatto // Can actual be argument associated with dummy?
CheckCompatibleArgument(bool isElemental,const ActualArgument & actual,const characteristics::DummyArgument & dummy)205464ab3302SCarolineConcatto static bool CheckCompatibleArgument(bool isElemental,
205564ab3302SCarolineConcatto     const ActualArgument &actual, const characteristics::DummyArgument &dummy) {
20560fcda9aeSpeter klausler   const auto *expr{actual.UnwrapExpr()};
2057cd03e96fSPeter Klausler   return common::visit(
205864ab3302SCarolineConcatto       common::visitors{
205964ab3302SCarolineConcatto           [&](const characteristics::DummyDataObject &x) {
20600fcda9aeSpeter klausler             if (x.attrs.test(characteristics::DummyDataObject::Attr::Pointer) &&
20610fcda9aeSpeter klausler                 IsBareNullPointer(expr)) {
20620fcda9aeSpeter klausler               // NULL() without MOLD= is compatible with any dummy data pointer
20630fcda9aeSpeter klausler               // but cannot be allowed to lead to ambiguity.
20640fcda9aeSpeter klausler               return true;
20650fcda9aeSpeter klausler             } else if (!isElemental && actual.Rank() != x.type.Rank() &&
20666f3d322fSpeter klausler                 !x.type.attrs().test(
20676f3d322fSpeter klausler                     characteristics::TypeAndShape::Attr::AssumedRank)) {
206864ab3302SCarolineConcatto               return false;
206964ab3302SCarolineConcatto             } else if (auto actualType{actual.GetType()}) {
20706f3d322fSpeter klausler               return x.type.type().IsTkCompatibleWith(*actualType);
207164ab3302SCarolineConcatto             }
20720fcda9aeSpeter klausler             return false;
207364ab3302SCarolineConcatto           },
207464ab3302SCarolineConcatto           [&](const characteristics::DummyProcedure &) {
2075bbd0dc3dSpeter klausler             return expr && IsProcedurePointerTarget(*expr);
207664ab3302SCarolineConcatto           },
207764ab3302SCarolineConcatto           [&](const characteristics::AlternateReturn &) {
207864ab3302SCarolineConcatto             return actual.isAlternateReturn();
207964ab3302SCarolineConcatto           },
208064ab3302SCarolineConcatto       },
208164ab3302SCarolineConcatto       dummy.u);
208264ab3302SCarolineConcatto }
208364ab3302SCarolineConcatto 
208464ab3302SCarolineConcatto // Are the actual arguments compatible with the dummy arguments of procedure?
CheckCompatibleArguments(const characteristics::Procedure & procedure,const ActualArguments & actuals)208564ab3302SCarolineConcatto static bool CheckCompatibleArguments(
208664ab3302SCarolineConcatto     const characteristics::Procedure &procedure,
208764ab3302SCarolineConcatto     const ActualArguments &actuals) {
208864ab3302SCarolineConcatto   bool isElemental{procedure.IsElemental()};
208964ab3302SCarolineConcatto   const auto &dummies{procedure.dummyArguments};
209064ab3302SCarolineConcatto   CHECK(dummies.size() == actuals.size());
209164ab3302SCarolineConcatto   for (std::size_t i{0}; i < dummies.size(); ++i) {
209264ab3302SCarolineConcatto     const characteristics::DummyArgument &dummy{dummies[i]};
209364ab3302SCarolineConcatto     const std::optional<ActualArgument> &actual{actuals[i]};
209464ab3302SCarolineConcatto     if (actual && !CheckCompatibleArgument(isElemental, *actual, dummy)) {
209564ab3302SCarolineConcatto       return false;
209664ab3302SCarolineConcatto     }
209764ab3302SCarolineConcatto   }
209864ab3302SCarolineConcatto   return true;
209964ab3302SCarolineConcatto }
210064ab3302SCarolineConcatto 
21019977b24aSpeter klausler // Handles a forward reference to a module function from what must
21029977b24aSpeter klausler // be a specification expression.  Return false if the symbol is
21039977b24aSpeter klausler // an invalid forward reference.
ResolveForward(const Symbol & symbol)21049977b24aSpeter klausler bool ExpressionAnalyzer::ResolveForward(const Symbol &symbol) {
21059977b24aSpeter klausler   if (context_.HasError(symbol)) {
21069977b24aSpeter klausler     return false;
21079977b24aSpeter klausler   }
21089977b24aSpeter klausler   if (const auto *details{
21099977b24aSpeter klausler           symbol.detailsIf<semantics::SubprogramNameDetails>()}) {
21109977b24aSpeter klausler     if (details->kind() == semantics::SubprogramKind::Module) {
21119977b24aSpeter klausler       // If this symbol is still a SubprogramNameDetails, we must be
21129977b24aSpeter klausler       // checking a specification expression in a sibling module
21139977b24aSpeter klausler       // procedure.  Resolve its names now so that its interface
21149977b24aSpeter klausler       // is known.
21159977b24aSpeter klausler       semantics::ResolveSpecificationParts(context_, symbol);
21169977b24aSpeter klausler       if (symbol.has<semantics::SubprogramNameDetails>()) {
21179977b24aSpeter klausler         // When the symbol hasn't had its details updated, we must have
21189977b24aSpeter klausler         // already been in the process of resolving the function's
21199977b24aSpeter klausler         // specification part; but recursive function calls are not
21209977b24aSpeter klausler         // allowed in specification parts (10.1.11 para 5).
21219977b24aSpeter klausler         Say("The module function '%s' may not be referenced recursively in a specification expression"_err_en_US,
21229977b24aSpeter klausler             symbol.name());
2123627e9007STim Keith         context_.SetError(symbol);
21249977b24aSpeter klausler         return false;
21259977b24aSpeter klausler       }
21269977b24aSpeter klausler     } else { // 10.1.11 para 4
21279977b24aSpeter klausler       Say("The internal function '%s' may not be referenced in a specification expression"_err_en_US,
21289977b24aSpeter klausler           symbol.name());
2129627e9007STim Keith       context_.SetError(symbol);
21309977b24aSpeter klausler       return false;
21319977b24aSpeter klausler     }
21329977b24aSpeter klausler   }
21339977b24aSpeter klausler   return true;
21349977b24aSpeter klausler }
21359977b24aSpeter klausler 
213664ab3302SCarolineConcatto // Resolve a call to a generic procedure with given actual arguments.
213764ab3302SCarolineConcatto // adjustActuals is called on procedure bindings to handle pass arg.
ResolveGeneric(const Symbol & symbol,const ActualArguments & actuals,const AdjustActuals & adjustActuals,bool isSubroutine,bool mightBeStructureConstructor)21380fcda9aeSpeter klausler std::pair<const Symbol *, bool> ExpressionAnalyzer::ResolveGeneric(
21390fcda9aeSpeter klausler     const Symbol &symbol, const ActualArguments &actuals,
21400c190575SPeter Klausler     const AdjustActuals &adjustActuals, bool isSubroutine,
21410c190575SPeter Klausler     bool mightBeStructureConstructor) {
214264ab3302SCarolineConcatto   const Symbol *elemental{nullptr}; // matching elemental specific proc
21430fcda9aeSpeter klausler   const Symbol *nonElemental{nullptr}; // matching non-elemental specific
21440c190575SPeter Klausler   const Symbol &ultimate{symbol.GetUltimate()};
21450c190575SPeter Klausler   // Check for a match with an explicit INTRINSIC
21460c190575SPeter Klausler   if (ultimate.attrs().test(semantics::Attr::INTRINSIC)) {
21470c190575SPeter Klausler     parser::Messages buffer;
21480c190575SPeter Klausler     auto restorer{foldingContext_.messages().SetMessages(buffer)};
21490c190575SPeter Klausler     ActualArguments localActuals{actuals};
21500c190575SPeter Klausler     if (context_.intrinsics().Probe(
21510c190575SPeter Klausler             CallCharacteristics{ultimate.name().ToString(), isSubroutine},
21520c190575SPeter Klausler             localActuals, foldingContext_) &&
21530c190575SPeter Klausler         !buffer.AnyFatalError()) {
21540c190575SPeter Klausler       return {&ultimate, false};
21550c190575SPeter Klausler     }
21560c190575SPeter Klausler   }
21570c190575SPeter Klausler   if (const auto *details{ultimate.detailsIf<semantics::GenericDetails>()}) {
21580fcda9aeSpeter klausler     bool anyBareNullActual{
21590fcda9aeSpeter klausler         std::find_if(actuals.begin(), actuals.end(), [](auto iter) {
21600fcda9aeSpeter klausler           return IsBareNullPointer(iter->UnwrapExpr());
21610fcda9aeSpeter klausler         }) != actuals.end()};
21620c190575SPeter Klausler     for (const Symbol &specific : details->specificProcs()) {
21639977b24aSpeter klausler       if (!ResolveForward(specific)) {
21649977b24aSpeter klausler         continue;
21659977b24aSpeter klausler       }
216664ab3302SCarolineConcatto       if (std::optional<characteristics::Procedure> procedure{
216764ab3302SCarolineConcatto               characteristics::Procedure::Characterize(
2168641ede93Speter klausler                   ProcedureDesignator{specific}, context_.foldingContext())}) {
216964ab3302SCarolineConcatto         ActualArguments localActuals{actuals};
217064ab3302SCarolineConcatto         if (specific.has<semantics::ProcBindingDetails>()) {
217164ab3302SCarolineConcatto           if (!adjustActuals.value()(specific, localActuals)) {
217264ab3302SCarolineConcatto             continue;
217364ab3302SCarolineConcatto           }
217464ab3302SCarolineConcatto         }
217514b90d1fSpeter klausler         if (semantics::CheckInterfaceForGeneric(*procedure, localActuals,
217614b90d1fSpeter klausler                 GetFoldingContext(), false /* no integer conversions */) &&
21770fcda9aeSpeter klausler             CheckCompatibleArguments(*procedure, localActuals)) {
21780fcda9aeSpeter klausler           if ((procedure->IsElemental() && elemental) ||
21790fcda9aeSpeter klausler               (!procedure->IsElemental() && nonElemental)) {
21800fcda9aeSpeter klausler             // 16.9.144(6): a bare NULL() is not allowed as an actual
21810fcda9aeSpeter klausler             // argument to a generic procedure if the specific procedure
21820fcda9aeSpeter klausler             // cannot be unambiguously distinguished
21830fcda9aeSpeter klausler             return {nullptr, true /* due to NULL actuals */};
21840fcda9aeSpeter klausler           }
218564ab3302SCarolineConcatto           if (!procedure->IsElemental()) {
21861bd083b5Speter klausler             // takes priority over elemental match
21870fcda9aeSpeter klausler             nonElemental = &specific;
21880fcda9aeSpeter klausler             if (!anyBareNullActual) {
21890fcda9aeSpeter klausler               break; // unambiguous case
219064ab3302SCarolineConcatto             }
21910fcda9aeSpeter klausler           } else {
219264ab3302SCarolineConcatto             elemental = &specific;
219364ab3302SCarolineConcatto           }
219464ab3302SCarolineConcatto         }
219564ab3302SCarolineConcatto       }
219664ab3302SCarolineConcatto     }
21970fcda9aeSpeter klausler     if (nonElemental) {
21980fcda9aeSpeter klausler       return {&AccessSpecific(symbol, *nonElemental), false};
21990fcda9aeSpeter klausler     } else if (elemental) {
22000fcda9aeSpeter klausler       return {&AccessSpecific(symbol, *elemental), false};
220164ab3302SCarolineConcatto     }
220264ab3302SCarolineConcatto     // Check parent derived type
220364ab3302SCarolineConcatto     if (const auto *parentScope{symbol.owner().GetDerivedTypeParent()}) {
220464ab3302SCarolineConcatto       if (const Symbol * extended{parentScope->FindComponent(symbol.name())}) {
22050c190575SPeter Klausler         auto pair{ResolveGeneric(
22060c190575SPeter Klausler             *extended, actuals, adjustActuals, isSubroutine, false)};
22070fcda9aeSpeter klausler         if (pair.first) {
22080fcda9aeSpeter klausler           return pair;
220964ab3302SCarolineConcatto         }
221064ab3302SCarolineConcatto       }
221164ab3302SCarolineConcatto     }
22120c190575SPeter Klausler     if (mightBeStructureConstructor && details->derivedType()) {
22130c190575SPeter Klausler       return {details->derivedType(), false};
221464ab3302SCarolineConcatto     }
22150c190575SPeter Klausler   }
22160c190575SPeter Klausler   // Check for generic or explicit INTRINSIC of the same name in outer scopes.
22170c190575SPeter Klausler   // See 15.5.5.2 for details.
22180c190575SPeter Klausler   if (!symbol.owner().IsGlobal() && !symbol.owner().IsDerivedType()) {
22190c190575SPeter Klausler     for (const std::string &n : GetAllNames(context_, symbol.name())) {
22200c190575SPeter Klausler       if (const Symbol * outer{symbol.owner().parent().FindSymbol(n)}) {
22210c190575SPeter Klausler         auto pair{ResolveGeneric(*outer, actuals, adjustActuals, isSubroutine,
22220c190575SPeter Klausler             mightBeStructureConstructor)};
22230c190575SPeter Klausler         if (pair.first) {
22240c190575SPeter Klausler           return pair;
22250c190575SPeter Klausler         }
22260c190575SPeter Klausler       }
22270c190575SPeter Klausler     }
222864ab3302SCarolineConcatto   }
22290fcda9aeSpeter klausler   return {nullptr, false};
223064ab3302SCarolineConcatto }
223164ab3302SCarolineConcatto 
AccessSpecific(const Symbol & originalGeneric,const Symbol & specific)22321bd083b5Speter klausler const Symbol &ExpressionAnalyzer::AccessSpecific(
22331bd083b5Speter klausler     const Symbol &originalGeneric, const Symbol &specific) {
22341bd083b5Speter klausler   if (const auto *hosted{
22351bd083b5Speter klausler           originalGeneric.detailsIf<semantics::HostAssocDetails>()}) {
22361bd083b5Speter klausler     return AccessSpecific(hosted->symbol(), specific);
22371bd083b5Speter klausler   } else if (const auto *used{
22381bd083b5Speter klausler                  originalGeneric.detailsIf<semantics::UseDetails>()}) {
22391bd083b5Speter klausler     const auto &scope{originalGeneric.owner()};
22409895ba86Speter klausler     if (auto iter{scope.find(specific.name())}; iter != scope.end()) {
22419895ba86Speter klausler       if (const auto *useDetails{
22429895ba86Speter klausler               iter->second->detailsIf<semantics::UseDetails>()}) {
22439895ba86Speter klausler         const Symbol &usedSymbol{useDetails->symbol()};
22449895ba86Speter klausler         const auto *usedGeneric{
22459895ba86Speter klausler             usedSymbol.detailsIf<semantics::GenericDetails>()};
22469895ba86Speter klausler         if (&usedSymbol == &specific ||
22479895ba86Speter klausler             (usedGeneric && usedGeneric->specific() == &specific)) {
22481bd083b5Speter klausler           return specific;
22499895ba86Speter klausler         }
22509895ba86Speter klausler       }
22519895ba86Speter klausler     }
22521bd083b5Speter klausler     // Create a renaming USE of the specific procedure.
22531bd083b5Speter klausler     auto rename{context_.SaveTempName(
22541bd083b5Speter klausler         used->symbol().owner().GetName().value().ToString() + "$" +
2255bcad53e1SPeter Klausler         specific.owner().GetName().value().ToString() + "$" +
22561bd083b5Speter klausler         specific.name().ToString())};
22571bd083b5Speter klausler     return *const_cast<semantics::Scope &>(scope)
22581bd083b5Speter klausler                 .try_emplace(rename, specific.attrs(),
22591bd083b5Speter klausler                     semantics::UseDetails{rename, specific})
22601bd083b5Speter klausler                 .first->second;
22611bd083b5Speter klausler   } else {
22621bd083b5Speter klausler     return specific;
22631bd083b5Speter klausler   }
22641bd083b5Speter klausler }
22651bd083b5Speter klausler 
EmitGenericResolutionError(const Symbol & symbol,bool dueToNullActuals)22660fcda9aeSpeter klausler void ExpressionAnalyzer::EmitGenericResolutionError(
22670fcda9aeSpeter klausler     const Symbol &symbol, bool dueToNullActuals) {
22680fcda9aeSpeter klausler   Say(dueToNullActuals
22690fcda9aeSpeter klausler           ? "One or more NULL() actual arguments to the generic procedure '%s' requires a MOLD= for disambiguation"_err_en_US
22700fcda9aeSpeter klausler           : semantics::IsGenericDefinedOp(symbol)
22710fcda9aeSpeter klausler           ? "No specific procedure of generic operator '%s' matches the actual arguments"_err_en_US
22720fcda9aeSpeter klausler           : "No specific procedure of generic '%s' matches the actual arguments"_err_en_US,
227364ab3302SCarolineConcatto       symbol.name());
227464ab3302SCarolineConcatto }
227564ab3302SCarolineConcatto 
GetCalleeAndArguments(const parser::ProcedureDesignator & pd,ActualArguments && arguments,bool isSubroutine,bool mightBeStructureConstructor)227664ab3302SCarolineConcatto auto ExpressionAnalyzer::GetCalleeAndArguments(
227764ab3302SCarolineConcatto     const parser::ProcedureDesignator &pd, ActualArguments &&arguments,
227864ab3302SCarolineConcatto     bool isSubroutine, bool mightBeStructureConstructor)
227964ab3302SCarolineConcatto     -> std::optional<CalleeAndArguments> {
2280cd03e96fSPeter Klausler   return common::visit(
228164ab3302SCarolineConcatto       common::visitors{
228264ab3302SCarolineConcatto           [&](const parser::Name &name) {
228364ab3302SCarolineConcatto             return GetCalleeAndArguments(name, std::move(arguments),
228464ab3302SCarolineConcatto                 isSubroutine, mightBeStructureConstructor);
228564ab3302SCarolineConcatto           },
228664ab3302SCarolineConcatto           [&](const parser::ProcComponentRef &pcr) {
22870c190575SPeter Klausler             return AnalyzeProcedureComponentRef(
22880c190575SPeter Klausler                 pcr, std::move(arguments), isSubroutine);
228964ab3302SCarolineConcatto           },
229064ab3302SCarolineConcatto       },
229164ab3302SCarolineConcatto       pd.u);
229264ab3302SCarolineConcatto }
229364ab3302SCarolineConcatto 
GetCalleeAndArguments(const parser::Name & name,ActualArguments && arguments,bool isSubroutine,bool mightBeStructureConstructor)229464ab3302SCarolineConcatto auto ExpressionAnalyzer::GetCalleeAndArguments(const parser::Name &name,
229564ab3302SCarolineConcatto     ActualArguments &&arguments, bool isSubroutine,
229664ab3302SCarolineConcatto     bool mightBeStructureConstructor) -> std::optional<CalleeAndArguments> {
229764ab3302SCarolineConcatto   const Symbol *symbol{name.symbol};
229864ab3302SCarolineConcatto   if (context_.HasError(symbol)) {
229964ab3302SCarolineConcatto     return std::nullopt; // also handles null symbol
230064ab3302SCarolineConcatto   }
230164ab3302SCarolineConcatto   const Symbol &ultimate{DEREF(symbol).GetUltimate()};
2302eb14135eSPeter Klausler   CheckForBadRecursion(name.source, ultimate);
2303eb14135eSPeter Klausler   bool dueToNullActual{false};
2304eb14135eSPeter Klausler   bool isGenericInterface{ultimate.has<semantics::GenericDetails>()};
23050c190575SPeter Klausler   bool isExplicitIntrinsic{ultimate.attrs().test(semantics::Attr::INTRINSIC)};
2306eb14135eSPeter Klausler   const Symbol *resolution{nullptr};
23070c190575SPeter Klausler   if (isGenericInterface || isExplicitIntrinsic) {
2308eb14135eSPeter Klausler     ExpressionAnalyzer::AdjustActuals noAdjustment;
23090c190575SPeter Klausler     auto pair{ResolveGeneric(*symbol, arguments, noAdjustment, isSubroutine,
23100c190575SPeter Klausler         mightBeStructureConstructor)};
2311eb14135eSPeter Klausler     resolution = pair.first;
2312eb14135eSPeter Klausler     dueToNullActual = pair.second;
2313bd92bca5SPeter Klausler     if (resolution) {
2314bd92bca5SPeter Klausler       // re-resolve name to the specific procedure
2315bd92bca5SPeter Klausler       name.symbol = const_cast<Symbol *>(resolution);
2316bd92bca5SPeter Klausler     }
23170c190575SPeter Klausler   } else {
2318b8a929cbSPeter Klausler     resolution = symbol;
23190c190575SPeter Klausler   }
23200c190575SPeter Klausler   if (!resolution || resolution->attrs().test(semantics::Attr::INTRINSIC)) {
23210c190575SPeter Klausler     // Not generic, or no resolution; may be intrinsic
23220c190575SPeter Klausler     if (std::optional<SpecificCall> specificCall{context_.intrinsics().Probe(
23230c190575SPeter Klausler             CallCharacteristics{ultimate.name().ToString(), isSubroutine},
232464ab3302SCarolineConcatto             arguments, GetFoldingContext())}) {
2325b82a8c3fSpeter klausler       CheckBadExplicitType(*specificCall, *symbol);
232664ab3302SCarolineConcatto       return CalleeAndArguments{
232764ab3302SCarolineConcatto           ProcedureDesignator{std::move(specificCall->specificIntrinsic)},
232864ab3302SCarolineConcatto           std::move(specificCall->arguments)};
2329b8a929cbSPeter Klausler     } else {
2330eb14135eSPeter Klausler       if (isGenericInterface) {
2331eb14135eSPeter Klausler         EmitGenericResolutionError(*symbol, dueToNullActual);
2332b8a929cbSPeter Klausler       }
2333eb14135eSPeter Klausler       return std::nullopt;
233464ab3302SCarolineConcatto     }
2335eb14135eSPeter Klausler   }
2336eb14135eSPeter Klausler   if (resolution->GetUltimate().has<semantics::DerivedTypeDetails>()) {
233764ab3302SCarolineConcatto     if (mightBeStructureConstructor) {
233864ab3302SCarolineConcatto       return CalleeAndArguments{
2339eb14135eSPeter Klausler           semantics::SymbolRef{*resolution}, std::move(arguments)};
234064ab3302SCarolineConcatto     }
2341eb14135eSPeter Klausler   } else if (IsProcedure(*resolution)) {
234264ab3302SCarolineConcatto     return CalleeAndArguments{
2343eb14135eSPeter Klausler         ProcedureDesignator{*resolution}, std::move(arguments)};
234464ab3302SCarolineConcatto   }
2345eb14135eSPeter Klausler   if (!context_.HasError(*resolution)) {
2346e5813a68Speter klausler     AttachDeclaration(
2347e5813a68Speter klausler         Say(name.source, "'%s' is not a callable procedure"_err_en_US,
2348e5813a68Speter klausler             name.source),
2349eb14135eSPeter Klausler         *resolution);
235064ab3302SCarolineConcatto   }
235164ab3302SCarolineConcatto   return std::nullopt;
235264ab3302SCarolineConcatto }
235364ab3302SCarolineConcatto 
2354b82a8c3fSpeter klausler // Fortran 2018 expressly states (8.2 p3) that any declared type for a
2355b82a8c3fSpeter klausler // generic intrinsic function "has no effect" on the result type of a
2356b82a8c3fSpeter klausler // call to that intrinsic.  So one can declare "character*8 cos" and
2357b82a8c3fSpeter klausler // still get a real result from "cos(1.)".  This is a dangerous feature,
2358b82a8c3fSpeter klausler // especially since implementations are free to extend their sets of
2359b82a8c3fSpeter klausler // intrinsics, and in doing so might clash with a name in a program.
2360b82a8c3fSpeter klausler // So we emit a warning in this situation, and perhaps it should be an
2361b82a8c3fSpeter klausler // error -- any correctly working program can silence the message by
2362b82a8c3fSpeter klausler // simply deleting the pointless type declaration.
CheckBadExplicitType(const SpecificCall & call,const Symbol & intrinsic)2363b82a8c3fSpeter klausler void ExpressionAnalyzer::CheckBadExplicitType(
2364b82a8c3fSpeter klausler     const SpecificCall &call, const Symbol &intrinsic) {
2365b82a8c3fSpeter klausler   if (intrinsic.GetUltimate().GetType()) {
2366b82a8c3fSpeter klausler     const auto &procedure{call.specificIntrinsic.characteristics.value()};
2367b82a8c3fSpeter klausler     if (const auto &result{procedure.functionResult}) {
2368b82a8c3fSpeter klausler       if (const auto *typeAndShape{result->GetTypeAndShape()}) {
2369b82a8c3fSpeter klausler         if (auto declared{
2370b82a8c3fSpeter klausler                 typeAndShape->Characterize(intrinsic, GetFoldingContext())}) {
2371b82a8c3fSpeter klausler           if (!declared->type().IsTkCompatibleWith(typeAndShape->type())) {
2372b82a8c3fSpeter klausler             if (auto *msg{Say(
2373a53967cdSPeter Klausler                     "The result type '%s' of the intrinsic function '%s' is not the explicit declared type '%s'"_warn_en_US,
2374b82a8c3fSpeter klausler                     typeAndShape->AsFortran(), intrinsic.name(),
2375b82a8c3fSpeter klausler                     declared->AsFortran())}) {
2376b82a8c3fSpeter klausler               msg->Attach(intrinsic.name(),
2377b82a8c3fSpeter klausler                   "Ignored declaration of intrinsic function '%s'"_en_US,
2378b82a8c3fSpeter klausler                   intrinsic.name());
2379b82a8c3fSpeter klausler             }
2380b82a8c3fSpeter klausler           }
2381b82a8c3fSpeter klausler         }
2382b82a8c3fSpeter klausler       }
2383b82a8c3fSpeter klausler     }
2384b82a8c3fSpeter klausler   }
2385b82a8c3fSpeter klausler }
2386b82a8c3fSpeter klausler 
CheckForBadRecursion(parser::CharBlock callSite,const semantics::Symbol & proc)238764ab3302SCarolineConcatto void ExpressionAnalyzer::CheckForBadRecursion(
238864ab3302SCarolineConcatto     parser::CharBlock callSite, const semantics::Symbol &proc) {
238964ab3302SCarolineConcatto   if (const auto *scope{proc.scope()}) {
239064ab3302SCarolineConcatto     if (scope->sourceRange().Contains(callSite)) {
239164ab3302SCarolineConcatto       parser::Message *msg{nullptr};
239264ab3302SCarolineConcatto       if (proc.attrs().test(semantics::Attr::NON_RECURSIVE)) { // 15.6.2.1(3)
239364ab3302SCarolineConcatto         msg = Say("NON_RECURSIVE procedure '%s' cannot call itself"_err_en_US,
239464ab3302SCarolineConcatto             callSite);
2395c42f6314Speter klausler       } else if (IsAssumedLengthCharacter(proc) && IsExternal(proc)) {
2396c4286209SPeter Klausler         // TODO: Also catch assumed PDT type parameters
2397657aaf8bSPete Steinfeld         msg = Say( // 15.6.2.1(3)
239864ab3302SCarolineConcatto             "Assumed-length CHARACTER(*) function '%s' cannot call itself"_err_en_US,
239964ab3302SCarolineConcatto             callSite);
240064ab3302SCarolineConcatto       }
240164ab3302SCarolineConcatto       AttachDeclaration(msg, proc);
240264ab3302SCarolineConcatto     }
240364ab3302SCarolineConcatto   }
240464ab3302SCarolineConcatto }
240564ab3302SCarolineConcatto 
AssumedTypeDummy(const A & x)240664ab3302SCarolineConcatto template <typename A> static const Symbol *AssumedTypeDummy(const A &x) {
240764ab3302SCarolineConcatto   if (const auto *designator{
240864ab3302SCarolineConcatto           std::get_if<common::Indirection<parser::Designator>>(&x.u)}) {
240964ab3302SCarolineConcatto     if (const auto *dataRef{
241064ab3302SCarolineConcatto             std::get_if<parser::DataRef>(&designator->value().u)}) {
241164ab3302SCarolineConcatto       if (const auto *name{std::get_if<parser::Name>(&dataRef->u)}) {
241292d27b96SJean Perier         return AssumedTypeDummy(*name);
241392d27b96SJean Perier       }
241492d27b96SJean Perier     }
241592d27b96SJean Perier   }
241692d27b96SJean Perier   return nullptr;
241792d27b96SJean Perier }
241892d27b96SJean Perier template <>
AssumedTypeDummy(const parser::Name & name)241992d27b96SJean Perier const Symbol *AssumedTypeDummy<parser::Name>(const parser::Name &name) {
242092d27b96SJean Perier   if (const Symbol * symbol{name.symbol}) {
242164ab3302SCarolineConcatto     if (const auto *type{symbol->GetType()}) {
242264ab3302SCarolineConcatto       if (type->category() == semantics::DeclTypeSpec::TypeStar) {
242364ab3302SCarolineConcatto         return symbol;
242464ab3302SCarolineConcatto       }
242564ab3302SCarolineConcatto     }
242664ab3302SCarolineConcatto   }
242764ab3302SCarolineConcatto   return nullptr;
242864ab3302SCarolineConcatto }
242992d27b96SJean Perier template <typename A>
AssumedTypePointerOrAllocatableDummy(const A & object)243092d27b96SJean Perier static const Symbol *AssumedTypePointerOrAllocatableDummy(const A &object) {
243192d27b96SJean Perier   // It is illegal for allocatable of pointer objects to be TYPE(*), but at that
243292d27b96SJean Perier   // point it is is not guaranteed that it has been checked the object has
243392d27b96SJean Perier   // POINTER or ALLOCATABLE attribute, so do not assume nullptr can be directly
243492d27b96SJean Perier   // returned.
2435cd03e96fSPeter Klausler   return common::visit(
243692d27b96SJean Perier       common::visitors{
243792d27b96SJean Perier           [&](const parser::StructureComponent &x) {
243892d27b96SJean Perier             return AssumedTypeDummy(x.component);
243992d27b96SJean Perier           },
244092d27b96SJean Perier           [&](const parser::Name &x) { return AssumedTypeDummy(x); },
244192d27b96SJean Perier       },
244292d27b96SJean Perier       object.u);
244392d27b96SJean Perier }
244492d27b96SJean Perier template <>
AssumedTypeDummy(const parser::AllocateObject & x)244592d27b96SJean Perier const Symbol *AssumedTypeDummy<parser::AllocateObject>(
244692d27b96SJean Perier     const parser::AllocateObject &x) {
244792d27b96SJean Perier   return AssumedTypePointerOrAllocatableDummy(x);
244892d27b96SJean Perier }
244992d27b96SJean Perier template <>
AssumedTypeDummy(const parser::PointerObject & x)245092d27b96SJean Perier const Symbol *AssumedTypeDummy<parser::PointerObject>(
245192d27b96SJean Perier     const parser::PointerObject &x) {
245292d27b96SJean Perier   return AssumedTypePointerOrAllocatableDummy(x);
245392d27b96SJean Perier }
245464ab3302SCarolineConcatto 
CheckIsValidForwardReference(const semantics::DerivedTypeSpec & dtSpec)24558b550af7SPeter Steinfeld bool ExpressionAnalyzer::CheckIsValidForwardReference(
24568b550af7SPeter Steinfeld     const semantics::DerivedTypeSpec &dtSpec) {
24578b550af7SPeter Steinfeld   if (dtSpec.IsForwardReferenced()) {
24588b550af7SPeter Steinfeld     Say("Cannot construct value for derived type '%s' "
24598b550af7SPeter Steinfeld         "before it is defined"_err_en_US,
24608b550af7SPeter Steinfeld         dtSpec.name());
24618b550af7SPeter Steinfeld     return false;
24628b550af7SPeter Steinfeld   }
24638b550af7SPeter Steinfeld   return true;
24648b550af7SPeter Steinfeld }
24658b550af7SPeter Steinfeld 
Analyze(const parser::FunctionReference & funcRef,std::optional<parser::StructureConstructor> * structureConstructor)246664ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::FunctionReference &funcRef,
246764ab3302SCarolineConcatto     std::optional<parser::StructureConstructor> *structureConstructor) {
246864ab3302SCarolineConcatto   const parser::Call &call{funcRef.v};
246964ab3302SCarolineConcatto   auto restorer{GetContextualMessages().SetLocation(call.source)};
2470a3538b83STim Keith   ArgumentAnalyzer analyzer{*this, call.source, true /* isProcedureCall */};
247164ab3302SCarolineConcatto   for (const auto &arg : std::get<std::list<parser::ActualArgSpec>>(call.t)) {
247264ab3302SCarolineConcatto     analyzer.Analyze(arg, false /* not subroutine call */);
247364ab3302SCarolineConcatto   }
247464ab3302SCarolineConcatto   if (analyzer.fatalErrors()) {
247564ab3302SCarolineConcatto     return std::nullopt;
247664ab3302SCarolineConcatto   }
247764ab3302SCarolineConcatto   if (std::optional<CalleeAndArguments> callee{
247864ab3302SCarolineConcatto           GetCalleeAndArguments(std::get<parser::ProcedureDesignator>(call.t),
247964ab3302SCarolineConcatto               analyzer.GetActuals(), false /* not subroutine */,
248064ab3302SCarolineConcatto               true /* might be structure constructor */)}) {
248164ab3302SCarolineConcatto     if (auto *proc{std::get_if<ProcedureDesignator>(&callee->u)}) {
248264ab3302SCarolineConcatto       return MakeFunctionRef(
248364ab3302SCarolineConcatto           call.source, std::move(*proc), std::move(callee->arguments));
248473c3530fSpeter klausler     }
248564ab3302SCarolineConcatto     CHECK(std::holds_alternative<semantics::SymbolRef>(callee->u));
248673c3530fSpeter klausler     const Symbol &symbol{*std::get<semantics::SymbolRef>(callee->u)};
248773c3530fSpeter klausler     if (structureConstructor) {
248873c3530fSpeter klausler       // Structure constructor misparsed as function reference?
248964ab3302SCarolineConcatto       const auto &designator{std::get<parser::ProcedureDesignator>(call.t)};
249064ab3302SCarolineConcatto       if (const auto *name{std::get_if<parser::Name>(&designator.u)}) {
249164ab3302SCarolineConcatto         semantics::Scope &scope{context_.FindScope(name->source)};
249273c3530fSpeter klausler         semantics::DerivedTypeSpec dtSpec{name->source, symbol.GetUltimate()};
24938b550af7SPeter Steinfeld         if (!CheckIsValidForwardReference(dtSpec)) {
2494cdbfb479SPeter Steinfeld           return std::nullopt;
2495cdbfb479SPeter Steinfeld         }
249664ab3302SCarolineConcatto         const semantics::DeclTypeSpec &type{
24975091671cSpeter klausler             semantics::FindOrInstantiateDerivedType(scope, std::move(dtSpec))};
249864ab3302SCarolineConcatto         auto &mutableRef{const_cast<parser::FunctionReference &>(funcRef)};
249964ab3302SCarolineConcatto         *structureConstructor =
250064ab3302SCarolineConcatto             mutableRef.ConvertToStructureConstructor(type.derivedTypeSpec());
250164ab3302SCarolineConcatto         return Analyze(structureConstructor->value());
250264ab3302SCarolineConcatto       }
250364ab3302SCarolineConcatto     }
250473c3530fSpeter klausler     if (!context_.HasError(symbol)) {
250573c3530fSpeter klausler       AttachDeclaration(
250673c3530fSpeter klausler           Say("'%s' is called like a function but is not a procedure"_err_en_US,
250773c3530fSpeter klausler               symbol.name()),
250873c3530fSpeter klausler           symbol);
250973c3530fSpeter klausler       context_.SetError(symbol);
251073c3530fSpeter klausler     }
251164ab3302SCarolineConcatto   }
251264ab3302SCarolineConcatto   return std::nullopt;
251364ab3302SCarolineConcatto }
251464ab3302SCarolineConcatto 
HasAlternateReturns(const evaluate::ActualArguments & args)2515ae0d1d2eSPeter Steinfeld static bool HasAlternateReturns(const evaluate::ActualArguments &args) {
2516ae0d1d2eSPeter Steinfeld   for (const auto &arg : args) {
2517ae0d1d2eSPeter Steinfeld     if (arg && arg->isAlternateReturn()) {
2518ae0d1d2eSPeter Steinfeld       return true;
2519ae0d1d2eSPeter Steinfeld     }
2520ae0d1d2eSPeter Steinfeld   }
2521ae0d1d2eSPeter Steinfeld   return false;
2522ae0d1d2eSPeter Steinfeld }
2523ae0d1d2eSPeter Steinfeld 
Analyze(const parser::CallStmt & callStmt)252464ab3302SCarolineConcatto void ExpressionAnalyzer::Analyze(const parser::CallStmt &callStmt) {
252564ab3302SCarolineConcatto   const parser::Call &call{callStmt.v};
252664ab3302SCarolineConcatto   auto restorer{GetContextualMessages().SetLocation(call.source)};
2527a3538b83STim Keith   ArgumentAnalyzer analyzer{*this, call.source, true /* isProcedureCall */};
252805756e69STim Keith   const auto &actualArgList{std::get<std::list<parser::ActualArgSpec>>(call.t)};
252905756e69STim Keith   for (const auto &arg : actualArgList) {
253064ab3302SCarolineConcatto     analyzer.Analyze(arg, true /* is subroutine call */);
253164ab3302SCarolineConcatto   }
253264ab3302SCarolineConcatto   if (!analyzer.fatalErrors()) {
253364ab3302SCarolineConcatto     if (std::optional<CalleeAndArguments> callee{
253464ab3302SCarolineConcatto             GetCalleeAndArguments(std::get<parser::ProcedureDesignator>(call.t),
253564ab3302SCarolineConcatto                 analyzer.GetActuals(), true /* subroutine */)}) {
253664ab3302SCarolineConcatto       ProcedureDesignator *proc{std::get_if<ProcedureDesignator>(&callee->u)};
253764ab3302SCarolineConcatto       CHECK(proc);
253864ab3302SCarolineConcatto       if (CheckCall(call.source, *proc, callee->arguments)) {
25398a8bb078Speter klausler         callStmt.typedCall.Reset(
25408a8bb078Speter klausler             new ProcedureRef{std::move(*proc), std::move(callee->arguments),
2541f6ded53fSPeter Klausler                 HasAlternateReturns(callee->arguments)},
25428a8bb078Speter klausler             ProcedureRef::Deleter);
2543f6ded53fSPeter Klausler         return;
254464ab3302SCarolineConcatto       }
254564ab3302SCarolineConcatto     }
2546f6ded53fSPeter Klausler     if (!context_.AnyFatalError()) {
2547f6ded53fSPeter Klausler       std::string buf;
2548f6ded53fSPeter Klausler       llvm::raw_string_ostream dump{buf};
2549f6ded53fSPeter Klausler       parser::DumpTree(dump, callStmt);
2550f6ded53fSPeter Klausler       Say("Internal error: Expression analysis failed on CALL statement: %s"_err_en_US,
2551f6ded53fSPeter Klausler           dump.str());
2552f6ded53fSPeter Klausler     }
255364ab3302SCarolineConcatto   }
255464ab3302SCarolineConcatto }
255564ab3302SCarolineConcatto 
Analyze(const parser::AssignmentStmt & x)255664ab3302SCarolineConcatto const Assignment *ExpressionAnalyzer::Analyze(const parser::AssignmentStmt &x) {
255764ab3302SCarolineConcatto   if (!x.typedAssignment) {
255864ab3302SCarolineConcatto     ArgumentAnalyzer analyzer{*this};
255964ab3302SCarolineConcatto     analyzer.Analyze(std::get<parser::Variable>(x.t));
256064ab3302SCarolineConcatto     analyzer.Analyze(std::get<parser::Expr>(x.t));
2561d9195d66Speter klausler     std::optional<Assignment> assignment;
2562d9195d66Speter klausler     if (!analyzer.fatalErrors()) {
256364ab3302SCarolineConcatto       std::optional<ProcedureRef> procRef{analyzer.TryDefinedAssignment()};
2564d9195d66Speter klausler       if (!procRef) {
2565d9195d66Speter klausler         analyzer.CheckForNullPointer(
2566d9195d66Speter klausler             "in a non-pointer intrinsic assignment statement");
2567d9195d66Speter klausler       }
2568d9195d66Speter klausler       assignment.emplace(analyzer.MoveExpr(0), analyzer.MoveExpr(1));
256964ab3302SCarolineConcatto       if (procRef) {
2570d9195d66Speter klausler         assignment->u = std::move(*procRef);
257164ab3302SCarolineConcatto       }
2572d9195d66Speter klausler     }
2573d9195d66Speter klausler     x.typedAssignment.Reset(new GenericAssignmentWrapper{std::move(assignment)},
25748a8bb078Speter klausler         GenericAssignmentWrapper::Deleter);
257564ab3302SCarolineConcatto   }
257664ab3302SCarolineConcatto   return common::GetPtrFromOptional(x.typedAssignment->v);
257764ab3302SCarolineConcatto }
257864ab3302SCarolineConcatto 
Analyze(const parser::PointerAssignmentStmt & x)257964ab3302SCarolineConcatto const Assignment *ExpressionAnalyzer::Analyze(
258064ab3302SCarolineConcatto     const parser::PointerAssignmentStmt &x) {
258164ab3302SCarolineConcatto   if (!x.typedAssignment) {
258264ab3302SCarolineConcatto     MaybeExpr lhs{Analyze(std::get<parser::DataRef>(x.t))};
258364ab3302SCarolineConcatto     MaybeExpr rhs{Analyze(std::get<parser::Expr>(x.t))};
258464ab3302SCarolineConcatto     if (!lhs || !rhs) {
25858a8bb078Speter klausler       x.typedAssignment.Reset(
25868a8bb078Speter klausler           new GenericAssignmentWrapper{}, GenericAssignmentWrapper::Deleter);
258764ab3302SCarolineConcatto     } else {
258864ab3302SCarolineConcatto       Assignment assignment{std::move(*lhs), std::move(*rhs)};
2589cd03e96fSPeter Klausler       common::visit(
2590cd03e96fSPeter Klausler           common::visitors{
259164ab3302SCarolineConcatto               [&](const std::list<parser::BoundsRemapping> &list) {
259264ab3302SCarolineConcatto                 Assignment::BoundsRemapping bounds;
259364ab3302SCarolineConcatto                 for (const auto &elem : list) {
259464ab3302SCarolineConcatto                   auto lower{AsSubscript(Analyze(std::get<0>(elem.t)))};
259564ab3302SCarolineConcatto                   auto upper{AsSubscript(Analyze(std::get<1>(elem.t)))};
259664ab3302SCarolineConcatto                   if (lower && upper) {
2597cd03e96fSPeter Klausler                     bounds.emplace_back(
2598cd03e96fSPeter Klausler                         Fold(std::move(*lower)), Fold(std::move(*upper)));
259964ab3302SCarolineConcatto                   }
260064ab3302SCarolineConcatto                 }
260164ab3302SCarolineConcatto                 assignment.u = std::move(bounds);
260264ab3302SCarolineConcatto               },
260364ab3302SCarolineConcatto               [&](const std::list<parser::BoundsSpec> &list) {
260464ab3302SCarolineConcatto                 Assignment::BoundsSpec bounds;
260564ab3302SCarolineConcatto                 for (const auto &bound : list) {
260664ab3302SCarolineConcatto                   if (auto lower{AsSubscript(Analyze(bound.v))}) {
260764ab3302SCarolineConcatto                     bounds.emplace_back(Fold(std::move(*lower)));
260864ab3302SCarolineConcatto                   }
260964ab3302SCarolineConcatto                 }
261064ab3302SCarolineConcatto                 assignment.u = std::move(bounds);
261164ab3302SCarolineConcatto               },
261264ab3302SCarolineConcatto           },
261364ab3302SCarolineConcatto           std::get<parser::PointerAssignmentStmt::Bounds>(x.t).u);
26148a8bb078Speter klausler       x.typedAssignment.Reset(
26158a8bb078Speter klausler           new GenericAssignmentWrapper{std::move(assignment)},
26168a8bb078Speter klausler           GenericAssignmentWrapper::Deleter);
261764ab3302SCarolineConcatto     }
261864ab3302SCarolineConcatto   }
261964ab3302SCarolineConcatto   return common::GetPtrFromOptional(x.typedAssignment->v);
262064ab3302SCarolineConcatto }
262164ab3302SCarolineConcatto 
IsExternalCalledImplicitly(parser::CharBlock callSite,const ProcedureDesignator & proc)262264ab3302SCarolineConcatto static bool IsExternalCalledImplicitly(
262364ab3302SCarolineConcatto     parser::CharBlock callSite, const ProcedureDesignator &proc) {
262464ab3302SCarolineConcatto   if (const auto *symbol{proc.GetSymbol()}) {
262564ab3302SCarolineConcatto     return symbol->has<semantics::SubprogramDetails>() &&
262664ab3302SCarolineConcatto         symbol->owner().IsGlobal() &&
2627c42f6314Speter klausler         (!symbol->scope() /*ENTRY*/ ||
2628c42f6314Speter klausler             !symbol->scope()->sourceRange().Contains(callSite));
262964ab3302SCarolineConcatto   } else {
263064ab3302SCarolineConcatto     return false;
263164ab3302SCarolineConcatto   }
263264ab3302SCarolineConcatto }
263364ab3302SCarolineConcatto 
CheckCall(parser::CharBlock callSite,const ProcedureDesignator & proc,ActualArguments & arguments)263464ab3302SCarolineConcatto std::optional<characteristics::Procedure> ExpressionAnalyzer::CheckCall(
263564ab3302SCarolineConcatto     parser::CharBlock callSite, const ProcedureDesignator &proc,
263664ab3302SCarolineConcatto     ActualArguments &arguments) {
2637641ede93Speter klausler   auto chars{characteristics::Procedure::Characterize(
2638641ede93Speter klausler       proc, context_.foldingContext())};
263964ab3302SCarolineConcatto   if (chars) {
264064ab3302SCarolineConcatto     bool treatExternalAsImplicit{IsExternalCalledImplicitly(callSite, proc)};
264164ab3302SCarolineConcatto     if (treatExternalAsImplicit && !chars->CanBeCalledViaImplicitInterface()) {
264264ab3302SCarolineConcatto       Say(callSite,
264347995a0eSJean Perier           "References to the procedure '%s' require an explicit interface"_err_en_US,
264464ab3302SCarolineConcatto           DEREF(proc.GetSymbol()).name());
264564ab3302SCarolineConcatto     }
2646c7574188SPeter Steinfeld     // Checks for ASSOCIATED() are done in intrinsic table processing
2647c4286209SPeter Klausler     const SpecificIntrinsic *specificIntrinsic{proc.GetSpecificIntrinsic()};
2648c4286209SPeter Klausler     bool procIsAssociated{
2649c4286209SPeter Klausler         specificIntrinsic && specificIntrinsic->name == "associated"};
2650c7574188SPeter Steinfeld     if (!procIsAssociated) {
2651bbad981dSPeter Klausler       const Symbol *procSymbol{proc.GetSymbol()};
2652bbad981dSPeter Klausler       bool procIsDummy{procSymbol && IsDummy(*procSymbol)};
2653c4286209SPeter Klausler       if (chars->functionResult &&
2654c4286209SPeter Klausler           chars->functionResult->IsAssumedLengthCharacter() &&
2655bbad981dSPeter Klausler           !specificIntrinsic && !procIsDummy) {
2656c4286209SPeter Klausler         Say(callSite,
2657c4286209SPeter Klausler             "Assumed-length character function must be defined with a length to be called"_err_en_US);
2658c4286209SPeter Klausler       }
265964ab3302SCarolineConcatto       semantics::CheckArguments(*chars, arguments, GetFoldingContext(),
266029d1a494SJean Perier           context_.FindScope(callSite), treatExternalAsImplicit,
2661c4286209SPeter Klausler           specificIntrinsic);
26628d0c3c05SPete Steinfeld       if (procSymbol && !IsPureProcedure(*procSymbol)) {
266364ab3302SCarolineConcatto         if (const semantics::Scope *
266464ab3302SCarolineConcatto             pure{semantics::FindPureProcedureContaining(
266564ab3302SCarolineConcatto                 context_.FindScope(callSite))}) {
266664ab3302SCarolineConcatto           Say(callSite,
266764ab3302SCarolineConcatto               "Procedure '%s' referenced in pure subprogram '%s' must be pure too"_err_en_US,
26688d0c3c05SPete Steinfeld               procSymbol->name(), DEREF(pure->symbol()).name());
266964ab3302SCarolineConcatto         }
267064ab3302SCarolineConcatto       }
267164ab3302SCarolineConcatto     }
2672c7574188SPeter Steinfeld   }
267364ab3302SCarolineConcatto   return chars;
267464ab3302SCarolineConcatto }
267564ab3302SCarolineConcatto 
267664ab3302SCarolineConcatto // Unary operations
267764ab3302SCarolineConcatto 
Analyze(const parser::Expr::Parentheses & x)267864ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::Parentheses &x) {
267964ab3302SCarolineConcatto   if (MaybeExpr operand{Analyze(x.v.value())}) {
268064ab3302SCarolineConcatto     if (const semantics::Symbol * symbol{GetLastSymbol(*operand)}) {
268164ab3302SCarolineConcatto       if (const semantics::Symbol * result{FindFunctionResult(*symbol)}) {
268264ab3302SCarolineConcatto         if (semantics::IsProcedurePointer(*result)) {
268364ab3302SCarolineConcatto           Say("A function reference that returns a procedure "
268464ab3302SCarolineConcatto               "pointer may not be parenthesized"_err_en_US); // C1003
268564ab3302SCarolineConcatto         }
268664ab3302SCarolineConcatto       }
268764ab3302SCarolineConcatto     }
268864ab3302SCarolineConcatto     return Parenthesize(std::move(*operand));
268964ab3302SCarolineConcatto   }
269064ab3302SCarolineConcatto   return std::nullopt;
269164ab3302SCarolineConcatto }
269264ab3302SCarolineConcatto 
NumericUnaryHelper(ExpressionAnalyzer & context,NumericOperator opr,const parser::Expr::IntrinsicUnary & x)269364ab3302SCarolineConcatto static MaybeExpr NumericUnaryHelper(ExpressionAnalyzer &context,
269464ab3302SCarolineConcatto     NumericOperator opr, const parser::Expr::IntrinsicUnary &x) {
269564ab3302SCarolineConcatto   ArgumentAnalyzer analyzer{context};
269664ab3302SCarolineConcatto   analyzer.Analyze(x.v);
2697d9195d66Speter klausler   if (!analyzer.fatalErrors()) {
2698d9195d66Speter klausler     if (analyzer.IsIntrinsicNumeric(opr)) {
2699d9195d66Speter klausler       analyzer.CheckForNullPointer();
270064ab3302SCarolineConcatto       if (opr == NumericOperator::Add) {
270164ab3302SCarolineConcatto         return analyzer.MoveExpr(0);
270264ab3302SCarolineConcatto       } else {
270364ab3302SCarolineConcatto         return Negation(context.GetContextualMessages(), analyzer.MoveExpr(0));
270464ab3302SCarolineConcatto       }
270564ab3302SCarolineConcatto     } else {
270664ab3302SCarolineConcatto       return analyzer.TryDefinedOp(AsFortran(opr),
270764ab3302SCarolineConcatto           "Operand of unary %s must be numeric; have %s"_err_en_US);
270864ab3302SCarolineConcatto     }
270964ab3302SCarolineConcatto   }
2710d9195d66Speter klausler   return std::nullopt;
2711d9195d66Speter klausler }
271264ab3302SCarolineConcatto 
Analyze(const parser::Expr::UnaryPlus & x)271364ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::UnaryPlus &x) {
271464ab3302SCarolineConcatto   return NumericUnaryHelper(*this, NumericOperator::Add, x);
271564ab3302SCarolineConcatto }
271664ab3302SCarolineConcatto 
Analyze(const parser::Expr::Negate & x)271764ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::Negate &x) {
271873506256SPeter Klausler   if (const auto *litConst{
271973506256SPeter Klausler           std::get_if<parser::LiteralConstant>(&x.v.value().u)}) {
272073506256SPeter Klausler     if (const auto *intConst{
272173506256SPeter Klausler             std::get_if<parser::IntLiteralConstant>(&litConst->u)}) {
272273506256SPeter Klausler       return Analyze(*intConst, true);
272373506256SPeter Klausler     }
272473506256SPeter Klausler   }
272564ab3302SCarolineConcatto   return NumericUnaryHelper(*this, NumericOperator::Subtract, x);
272664ab3302SCarolineConcatto }
272764ab3302SCarolineConcatto 
Analyze(const parser::Expr::NOT & x)272864ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::NOT &x) {
272964ab3302SCarolineConcatto   ArgumentAnalyzer analyzer{*this};
273064ab3302SCarolineConcatto   analyzer.Analyze(x.v);
2731d9195d66Speter klausler   if (!analyzer.fatalErrors()) {
2732d9195d66Speter klausler     if (analyzer.IsIntrinsicLogical()) {
2733d9195d66Speter klausler       analyzer.CheckForNullPointer();
273464ab3302SCarolineConcatto       return AsGenericExpr(
273564ab3302SCarolineConcatto           LogicalNegation(std::get<Expr<SomeLogical>>(analyzer.MoveExpr(0).u)));
273664ab3302SCarolineConcatto     } else {
273764ab3302SCarolineConcatto       return analyzer.TryDefinedOp(LogicalOperator::Not,
273864ab3302SCarolineConcatto           "Operand of %s must be LOGICAL; have %s"_err_en_US);
273964ab3302SCarolineConcatto     }
274064ab3302SCarolineConcatto   }
2741d9195d66Speter klausler   return std::nullopt;
2742d9195d66Speter klausler }
274364ab3302SCarolineConcatto 
Analyze(const parser::Expr::PercentLoc & x)274464ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::PercentLoc &x) {
274564ab3302SCarolineConcatto   // Represent %LOC() exactly as if it had been a call to the LOC() extension
274664ab3302SCarolineConcatto   // intrinsic function.
274764ab3302SCarolineConcatto   // Use the actual source for the name of the call for error reporting.
274864ab3302SCarolineConcatto   std::optional<ActualArgument> arg;
274964ab3302SCarolineConcatto   if (const Symbol * assumedTypeDummy{AssumedTypeDummy(x.v.value())}) {
275064ab3302SCarolineConcatto     arg = ActualArgument{ActualArgument::AssumedType{*assumedTypeDummy}};
275164ab3302SCarolineConcatto   } else if (MaybeExpr argExpr{Analyze(x.v.value())}) {
275264ab3302SCarolineConcatto     arg = ActualArgument{std::move(*argExpr)};
275364ab3302SCarolineConcatto   } else {
275464ab3302SCarolineConcatto     return std::nullopt;
275564ab3302SCarolineConcatto   }
275664ab3302SCarolineConcatto   parser::CharBlock at{GetContextualMessages().at()};
275764ab3302SCarolineConcatto   CHECK(at.size() >= 4);
275864ab3302SCarolineConcatto   parser::CharBlock loc{at.begin() + 1, 3};
275964ab3302SCarolineConcatto   CHECK(loc == "loc");
276064ab3302SCarolineConcatto   return MakeFunctionRef(loc, ActualArguments{std::move(*arg)});
276164ab3302SCarolineConcatto }
276264ab3302SCarolineConcatto 
Analyze(const parser::Expr::DefinedUnary & x)276364ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::DefinedUnary &x) {
276464ab3302SCarolineConcatto   const auto &name{std::get<parser::DefinedOpName>(x.t).v};
276564ab3302SCarolineConcatto   ArgumentAnalyzer analyzer{*this, name.source};
276664ab3302SCarolineConcatto   analyzer.Analyze(std::get<1>(x.t));
276764ab3302SCarolineConcatto   return analyzer.TryDefinedOp(name.source.ToString().c_str(),
2768d9195d66Speter klausler       "No operator %s defined for %s"_err_en_US, nullptr, true);
276964ab3302SCarolineConcatto }
277064ab3302SCarolineConcatto 
277164ab3302SCarolineConcatto // Binary (dyadic) operations
277264ab3302SCarolineConcatto 
277364ab3302SCarolineConcatto template <template <typename> class OPR>
NumericBinaryHelper(ExpressionAnalyzer & context,NumericOperator opr,const parser::Expr::IntrinsicBinary & x)277464ab3302SCarolineConcatto MaybeExpr NumericBinaryHelper(ExpressionAnalyzer &context, NumericOperator opr,
277564ab3302SCarolineConcatto     const parser::Expr::IntrinsicBinary &x) {
277664ab3302SCarolineConcatto   ArgumentAnalyzer analyzer{context};
277764ab3302SCarolineConcatto   analyzer.Analyze(std::get<0>(x.t));
277864ab3302SCarolineConcatto   analyzer.Analyze(std::get<1>(x.t));
2779d9195d66Speter klausler   if (!analyzer.fatalErrors()) {
2780d9195d66Speter klausler     if (analyzer.IsIntrinsicNumeric(opr)) {
2781d9195d66Speter klausler       analyzer.CheckForNullPointer();
27821a633e72Speter klausler       analyzer.CheckConformance();
278364ab3302SCarolineConcatto       return NumericOperation<OPR>(context.GetContextualMessages(),
278464ab3302SCarolineConcatto           analyzer.MoveExpr(0), analyzer.MoveExpr(1),
278564ab3302SCarolineConcatto           context.GetDefaultKind(TypeCategory::Real));
278664ab3302SCarolineConcatto     } else {
278764ab3302SCarolineConcatto       return analyzer.TryDefinedOp(AsFortran(opr),
278864ab3302SCarolineConcatto           "Operands of %s must be numeric; have %s and %s"_err_en_US);
278964ab3302SCarolineConcatto     }
279064ab3302SCarolineConcatto   }
2791d9195d66Speter klausler   return std::nullopt;
2792d9195d66Speter klausler }
279364ab3302SCarolineConcatto 
Analyze(const parser::Expr::Power & x)279464ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::Power &x) {
279564ab3302SCarolineConcatto   return NumericBinaryHelper<Power>(*this, NumericOperator::Power, x);
279664ab3302SCarolineConcatto }
279764ab3302SCarolineConcatto 
Analyze(const parser::Expr::Multiply & x)279864ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::Multiply &x) {
279964ab3302SCarolineConcatto   return NumericBinaryHelper<Multiply>(*this, NumericOperator::Multiply, x);
280064ab3302SCarolineConcatto }
280164ab3302SCarolineConcatto 
Analyze(const parser::Expr::Divide & x)280264ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::Divide &x) {
280364ab3302SCarolineConcatto   return NumericBinaryHelper<Divide>(*this, NumericOperator::Divide, x);
280464ab3302SCarolineConcatto }
280564ab3302SCarolineConcatto 
Analyze(const parser::Expr::Add & x)280664ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::Add &x) {
280764ab3302SCarolineConcatto   return NumericBinaryHelper<Add>(*this, NumericOperator::Add, x);
280864ab3302SCarolineConcatto }
280964ab3302SCarolineConcatto 
Analyze(const parser::Expr::Subtract & x)281064ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::Subtract &x) {
281164ab3302SCarolineConcatto   return NumericBinaryHelper<Subtract>(*this, NumericOperator::Subtract, x);
281264ab3302SCarolineConcatto }
281364ab3302SCarolineConcatto 
Analyze(const parser::Expr::ComplexConstructor & x)281464ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(
281564ab3302SCarolineConcatto     const parser::Expr::ComplexConstructor &x) {
281664ab3302SCarolineConcatto   auto re{Analyze(std::get<0>(x.t).value())};
281764ab3302SCarolineConcatto   auto im{Analyze(std::get<1>(x.t).value())};
281864ab3302SCarolineConcatto   if (re && im) {
281964ab3302SCarolineConcatto     ConformabilityCheck(GetContextualMessages(), *re, *im);
282064ab3302SCarolineConcatto   }
282164ab3302SCarolineConcatto   return AsMaybeExpr(ConstructComplex(GetContextualMessages(), std::move(re),
282264ab3302SCarolineConcatto       std::move(im), GetDefaultKind(TypeCategory::Real)));
282364ab3302SCarolineConcatto }
282464ab3302SCarolineConcatto 
Analyze(const parser::Expr::Concat & x)282564ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::Concat &x) {
282664ab3302SCarolineConcatto   ArgumentAnalyzer analyzer{*this};
282764ab3302SCarolineConcatto   analyzer.Analyze(std::get<0>(x.t));
282864ab3302SCarolineConcatto   analyzer.Analyze(std::get<1>(x.t));
2829d9195d66Speter klausler   if (!analyzer.fatalErrors()) {
2830d9195d66Speter klausler     if (analyzer.IsIntrinsicConcat()) {
2831d9195d66Speter klausler       analyzer.CheckForNullPointer();
2832cd03e96fSPeter Klausler       return common::visit(
283364ab3302SCarolineConcatto           [&](auto &&x, auto &&y) -> MaybeExpr {
283464ab3302SCarolineConcatto             using T = ResultType<decltype(x)>;
283564ab3302SCarolineConcatto             if constexpr (std::is_same_v<T, ResultType<decltype(y)>>) {
283664ab3302SCarolineConcatto               return AsGenericExpr(Concat<T::kind>{std::move(x), std::move(y)});
283764ab3302SCarolineConcatto             } else {
283864ab3302SCarolineConcatto               DIE("different types for intrinsic concat");
283964ab3302SCarolineConcatto             }
284064ab3302SCarolineConcatto           },
284164ab3302SCarolineConcatto           std::move(std::get<Expr<SomeCharacter>>(analyzer.MoveExpr(0).u).u),
284264ab3302SCarolineConcatto           std::move(std::get<Expr<SomeCharacter>>(analyzer.MoveExpr(1).u).u));
284364ab3302SCarolineConcatto     } else {
284464ab3302SCarolineConcatto       return analyzer.TryDefinedOp("//",
284564ab3302SCarolineConcatto           "Operands of %s must be CHARACTER with the same kind; have %s and %s"_err_en_US);
284664ab3302SCarolineConcatto     }
284764ab3302SCarolineConcatto   }
2848d9195d66Speter klausler   return std::nullopt;
2849d9195d66Speter klausler }
285064ab3302SCarolineConcatto 
285164ab3302SCarolineConcatto // The Name represents a user-defined intrinsic operator.
285264ab3302SCarolineConcatto // If the actuals match one of the specific procedures, return a function ref.
285364ab3302SCarolineConcatto // Otherwise report the error in messages.
AnalyzeDefinedOp(const parser::Name & name,ActualArguments && actuals)285464ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::AnalyzeDefinedOp(
285564ab3302SCarolineConcatto     const parser::Name &name, ActualArguments &&actuals) {
285664ab3302SCarolineConcatto   if (auto callee{GetCalleeAndArguments(name, std::move(actuals))}) {
285764ab3302SCarolineConcatto     CHECK(std::holds_alternative<ProcedureDesignator>(callee->u));
285864ab3302SCarolineConcatto     return MakeFunctionRef(name.source,
285964ab3302SCarolineConcatto         std::move(std::get<ProcedureDesignator>(callee->u)),
286064ab3302SCarolineConcatto         std::move(callee->arguments));
286164ab3302SCarolineConcatto   } else {
286264ab3302SCarolineConcatto     return std::nullopt;
286364ab3302SCarolineConcatto   }
286464ab3302SCarolineConcatto }
286564ab3302SCarolineConcatto 
RelationHelper(ExpressionAnalyzer & context,RelationalOperator opr,const parser::Expr::IntrinsicBinary & x)286664ab3302SCarolineConcatto MaybeExpr RelationHelper(ExpressionAnalyzer &context, RelationalOperator opr,
286764ab3302SCarolineConcatto     const parser::Expr::IntrinsicBinary &x) {
286864ab3302SCarolineConcatto   ArgumentAnalyzer analyzer{context};
286964ab3302SCarolineConcatto   analyzer.Analyze(std::get<0>(x.t));
287064ab3302SCarolineConcatto   analyzer.Analyze(std::get<1>(x.t));
2871d9195d66Speter klausler   if (!analyzer.fatalErrors()) {
2872ea4973f2SPeter Steinfeld     std::optional<DynamicType> leftType{analyzer.GetType(0)};
2873ea4973f2SPeter Steinfeld     std::optional<DynamicType> rightType{analyzer.GetType(1)};
2874d9195d66Speter klausler     analyzer.ConvertBOZ(leftType, 0, rightType);
2875d9195d66Speter klausler     analyzer.ConvertBOZ(rightType, 1, leftType);
2876d9195d66Speter klausler     if (leftType && rightType &&
2877d9195d66Speter klausler         analyzer.IsIntrinsicRelational(opr, *leftType, *rightType)) {
2878d9195d66Speter klausler       analyzer.CheckForNullPointer("as a relational operand");
287964ab3302SCarolineConcatto       return AsMaybeExpr(Relate(context.GetContextualMessages(), opr,
288064ab3302SCarolineConcatto           analyzer.MoveExpr(0), analyzer.MoveExpr(1)));
288164ab3302SCarolineConcatto     } else {
288264ab3302SCarolineConcatto       return analyzer.TryDefinedOp(opr,
2883d9195d66Speter klausler           leftType && leftType->category() == TypeCategory::Logical &&
2884d9195d66Speter klausler                   rightType && rightType->category() == TypeCategory::Logical
2885d9195d66Speter klausler               ? "LOGICAL operands must be compared using .EQV. or .NEQV."_err_en_US
2886d9195d66Speter klausler               : "Operands of %s must have comparable types; have %s and %s"_err_en_US);
288764ab3302SCarolineConcatto     }
288864ab3302SCarolineConcatto   }
2889d9195d66Speter klausler   return std::nullopt;
289083dca19cSPete Steinfeld }
289164ab3302SCarolineConcatto 
Analyze(const parser::Expr::LT & x)289264ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::LT &x) {
289364ab3302SCarolineConcatto   return RelationHelper(*this, RelationalOperator::LT, x);
289464ab3302SCarolineConcatto }
289564ab3302SCarolineConcatto 
Analyze(const parser::Expr::LE & x)289664ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::LE &x) {
289764ab3302SCarolineConcatto   return RelationHelper(*this, RelationalOperator::LE, x);
289864ab3302SCarolineConcatto }
289964ab3302SCarolineConcatto 
Analyze(const parser::Expr::EQ & x)290064ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::EQ &x) {
290164ab3302SCarolineConcatto   return RelationHelper(*this, RelationalOperator::EQ, x);
290264ab3302SCarolineConcatto }
290364ab3302SCarolineConcatto 
Analyze(const parser::Expr::NE & x)290464ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::NE &x) {
290564ab3302SCarolineConcatto   return RelationHelper(*this, RelationalOperator::NE, x);
290664ab3302SCarolineConcatto }
290764ab3302SCarolineConcatto 
Analyze(const parser::Expr::GE & x)290864ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::GE &x) {
290964ab3302SCarolineConcatto   return RelationHelper(*this, RelationalOperator::GE, x);
291064ab3302SCarolineConcatto }
291164ab3302SCarolineConcatto 
Analyze(const parser::Expr::GT & x)291264ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::GT &x) {
291364ab3302SCarolineConcatto   return RelationHelper(*this, RelationalOperator::GT, x);
291464ab3302SCarolineConcatto }
291564ab3302SCarolineConcatto 
LogicalBinaryHelper(ExpressionAnalyzer & context,LogicalOperator opr,const parser::Expr::IntrinsicBinary & x)291664ab3302SCarolineConcatto MaybeExpr LogicalBinaryHelper(ExpressionAnalyzer &context, LogicalOperator opr,
291764ab3302SCarolineConcatto     const parser::Expr::IntrinsicBinary &x) {
291864ab3302SCarolineConcatto   ArgumentAnalyzer analyzer{context};
291964ab3302SCarolineConcatto   analyzer.Analyze(std::get<0>(x.t));
292064ab3302SCarolineConcatto   analyzer.Analyze(std::get<1>(x.t));
2921d9195d66Speter klausler   if (!analyzer.fatalErrors()) {
2922d9195d66Speter klausler     if (analyzer.IsIntrinsicLogical()) {
2923d9195d66Speter klausler       analyzer.CheckForNullPointer("as a logical operand");
292464ab3302SCarolineConcatto       return AsGenericExpr(BinaryLogicalOperation(opr,
292564ab3302SCarolineConcatto           std::get<Expr<SomeLogical>>(analyzer.MoveExpr(0).u),
292664ab3302SCarolineConcatto           std::get<Expr<SomeLogical>>(analyzer.MoveExpr(1).u)));
292764ab3302SCarolineConcatto     } else {
292864ab3302SCarolineConcatto       return analyzer.TryDefinedOp(
292964ab3302SCarolineConcatto           opr, "Operands of %s must be LOGICAL; have %s and %s"_err_en_US);
293064ab3302SCarolineConcatto     }
293164ab3302SCarolineConcatto   }
2932d9195d66Speter klausler   return std::nullopt;
2933d9195d66Speter klausler }
293464ab3302SCarolineConcatto 
Analyze(const parser::Expr::AND & x)293564ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::AND &x) {
293664ab3302SCarolineConcatto   return LogicalBinaryHelper(*this, LogicalOperator::And, x);
293764ab3302SCarolineConcatto }
293864ab3302SCarolineConcatto 
Analyze(const parser::Expr::OR & x)293964ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::OR &x) {
294064ab3302SCarolineConcatto   return LogicalBinaryHelper(*this, LogicalOperator::Or, x);
294164ab3302SCarolineConcatto }
294264ab3302SCarolineConcatto 
Analyze(const parser::Expr::EQV & x)294364ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::EQV &x) {
294464ab3302SCarolineConcatto   return LogicalBinaryHelper(*this, LogicalOperator::Eqv, x);
294564ab3302SCarolineConcatto }
294664ab3302SCarolineConcatto 
Analyze(const parser::Expr::NEQV & x)294764ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::NEQV &x) {
294864ab3302SCarolineConcatto   return LogicalBinaryHelper(*this, LogicalOperator::Neqv, x);
294964ab3302SCarolineConcatto }
295064ab3302SCarolineConcatto 
Analyze(const parser::Expr::DefinedBinary & x)295164ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr::DefinedBinary &x) {
295264ab3302SCarolineConcatto   const auto &name{std::get<parser::DefinedOpName>(x.t).v};
295364ab3302SCarolineConcatto   ArgumentAnalyzer analyzer{*this, name.source};
295464ab3302SCarolineConcatto   analyzer.Analyze(std::get<1>(x.t));
295564ab3302SCarolineConcatto   analyzer.Analyze(std::get<2>(x.t));
295664ab3302SCarolineConcatto   return analyzer.TryDefinedOp(name.source.ToString().c_str(),
2957d9195d66Speter klausler       "No operator %s defined for %s and %s"_err_en_US, nullptr, true);
295864ab3302SCarolineConcatto }
295964ab3302SCarolineConcatto 
2960a5679615SPeter Klausler // Returns true if a parsed function reference should be converted
2961a5679615SPeter Klausler // into an array element reference.
CheckFuncRefToArrayElement(semantics::SemanticsContext & context,const parser::FunctionReference & funcRef)2962a5679615SPeter Klausler static bool CheckFuncRefToArrayElement(semantics::SemanticsContext &context,
296364ab3302SCarolineConcatto     const parser::FunctionReference &funcRef) {
296464ab3302SCarolineConcatto   // Emit message if the function reference fix will end up an array element
2965a5679615SPeter Klausler   // reference with no subscripts, or subscripts on a scalar, because it will
2966a5679615SPeter Klausler   // not be possible to later distinguish in expressions between an empty
2967a5679615SPeter Klausler   // subscript list due to bad subscripts error recovery or because the
2968a5679615SPeter Klausler   // user did not put any.
296964ab3302SCarolineConcatto   auto &proc{std::get<parser::ProcedureDesignator>(funcRef.v.t)};
297064ab3302SCarolineConcatto   const auto *name{std::get_if<parser::Name>(&proc.u)};
297164ab3302SCarolineConcatto   if (!name) {
297264ab3302SCarolineConcatto     name = &std::get<parser::ProcComponentRef>(proc.u).v.thing.component;
297364ab3302SCarolineConcatto   }
2974a5679615SPeter Klausler   if (!name->symbol) {
2975a5679615SPeter Klausler     return false;
2976a5679615SPeter Klausler   } else if (name->symbol->Rank() == 0) {
2977a5679615SPeter Klausler     if (const Symbol *
2978a5679615SPeter Klausler         function{
2979a5679615SPeter Klausler             semantics::IsFunctionResultWithSameNameAsFunction(*name->symbol)}) {
298064ab3302SCarolineConcatto       auto &msg{context.Say(funcRef.v.source,
2981a5679615SPeter Klausler           "Recursive call to '%s' requires a distinct RESULT in its declaration"_err_en_US,
2982a5679615SPeter Klausler           name->source)};
2983a5679615SPeter Klausler       AttachDeclaration(&msg, *function);
2984a5679615SPeter Klausler       name->symbol = const_cast<Symbol *>(function);
2985a5679615SPeter Klausler     }
2986a5679615SPeter Klausler     return false;
2987a5679615SPeter Klausler   } else {
2988a5679615SPeter Klausler     if (std::get<std::list<parser::ActualArgSpec>>(funcRef.v.t).empty()) {
2989a5679615SPeter Klausler       auto &msg{context.Say(funcRef.v.source,
2990a5679615SPeter Klausler           "Reference to array '%s' with empty subscript list"_err_en_US,
299164ab3302SCarolineConcatto           name->source)};
299264ab3302SCarolineConcatto       if (name->symbol) {
299364ab3302SCarolineConcatto         AttachDeclaration(&msg, *name->symbol);
299464ab3302SCarolineConcatto       }
299564ab3302SCarolineConcatto     }
2996a5679615SPeter Klausler     return true;
299764ab3302SCarolineConcatto   }
299864ab3302SCarolineConcatto }
299964ab3302SCarolineConcatto 
300064ab3302SCarolineConcatto // Converts, if appropriate, an original misparse of ambiguous syntax like
300164ab3302SCarolineConcatto // A(1) as a function reference into an array reference.
3002d9195d66Speter klausler // Misparsed structure constructors are detected elsewhere after generic
300364ab3302SCarolineConcatto // function call resolution fails.
300464ab3302SCarolineConcatto template <typename... A>
FixMisparsedFunctionReference(semantics::SemanticsContext & context,const std::variant<A...> & constU)300564ab3302SCarolineConcatto static void FixMisparsedFunctionReference(
300664ab3302SCarolineConcatto     semantics::SemanticsContext &context, const std::variant<A...> &constU) {
300764ab3302SCarolineConcatto   // The parse tree is updated in situ when resolving an ambiguous parse.
300864ab3302SCarolineConcatto   using uType = std::decay_t<decltype(constU)>;
300964ab3302SCarolineConcatto   auto &u{const_cast<uType &>(constU)};
301064ab3302SCarolineConcatto   if (auto *func{
301164ab3302SCarolineConcatto           std::get_if<common::Indirection<parser::FunctionReference>>(&u)}) {
301264ab3302SCarolineConcatto     parser::FunctionReference &funcRef{func->value()};
301364ab3302SCarolineConcatto     auto &proc{std::get<parser::ProcedureDesignator>(funcRef.v.t)};
301464ab3302SCarolineConcatto     if (Symbol *
30151f879005STim Keith         origSymbol{
3016cd03e96fSPeter Klausler             common::visit(common::visitors{
301764ab3302SCarolineConcatto                               [&](parser::Name &name) { return name.symbol; },
301864ab3302SCarolineConcatto                               [&](parser::ProcComponentRef &pcr) {
301964ab3302SCarolineConcatto                                 return pcr.v.thing.component.symbol;
302064ab3302SCarolineConcatto                               },
302164ab3302SCarolineConcatto                           },
302264ab3302SCarolineConcatto                 proc.u)}) {
302364ab3302SCarolineConcatto       Symbol &symbol{origSymbol->GetUltimate()};
302464ab3302SCarolineConcatto       if (symbol.has<semantics::ObjectEntityDetails>() ||
302564ab3302SCarolineConcatto           symbol.has<semantics::AssocEntityDetails>()) {
302664ab3302SCarolineConcatto         // Note that expression in AssocEntityDetails cannot be a procedure
302764ab3302SCarolineConcatto         // pointer as per C1105 so this cannot be a function reference.
302864ab3302SCarolineConcatto         if constexpr (common::HasMember<common::Indirection<parser::Designator>,
302964ab3302SCarolineConcatto                           uType>) {
3030a5679615SPeter Klausler           if (CheckFuncRefToArrayElement(context, funcRef)) {
303164ab3302SCarolineConcatto             u = common::Indirection{funcRef.ConvertToArrayElementRef()};
3032a5679615SPeter Klausler           }
303364ab3302SCarolineConcatto         } else {
303464ab3302SCarolineConcatto           DIE("can't fix misparsed function as array reference");
303564ab3302SCarolineConcatto         }
303664ab3302SCarolineConcatto       }
303764ab3302SCarolineConcatto     }
303864ab3302SCarolineConcatto   }
303964ab3302SCarolineConcatto }
304064ab3302SCarolineConcatto 
30413a1afd8cSpeter klausler // Common handling of parse tree node types that retain the
30423a1afd8cSpeter klausler // representation of the analyzed expression.
304364ab3302SCarolineConcatto template <typename PARSED>
ExprOrVariable(const PARSED & x,parser::CharBlock source)3044934b27a9Speter klausler MaybeExpr ExpressionAnalyzer::ExprOrVariable(
3045934b27a9Speter klausler     const PARSED &x, parser::CharBlock source) {
3046e8f96899Speter klausler   if (useSavedTypedExprs_ && x.typedExpr) {
30473a1afd8cSpeter klausler     return x.typedExpr->v;
30483a1afd8cSpeter klausler   }
3049934b27a9Speter klausler   auto restorer{GetContextualMessages().SetLocation(source)};
30503a1afd8cSpeter klausler   if constexpr (std::is_same_v<PARSED, parser::Expr> ||
30513a1afd8cSpeter klausler       std::is_same_v<PARSED, parser::Variable>) {
305264ab3302SCarolineConcatto     FixMisparsedFunctionReference(context_, x.u);
30533a1afd8cSpeter klausler   }
305464ab3302SCarolineConcatto   if (AssumedTypeDummy(x)) { // C710
305564ab3302SCarolineConcatto     Say("TYPE(*) dummy argument may only be used as an actual argument"_err_en_US);
3056e712a92bSpeter klausler     ResetExpr(x);
3057e712a92bSpeter klausler     return std::nullopt;
3058e712a92bSpeter klausler   }
3059e712a92bSpeter klausler   MaybeExpr result;
3060e712a92bSpeter klausler   if constexpr (common::HasMember<parser::StructureConstructor,
3061e712a92bSpeter klausler                     std::decay_t<decltype(x.u)>> &&
3062e712a92bSpeter klausler       common::HasMember<common::Indirection<parser::FunctionReference>,
3063e712a92bSpeter klausler           std::decay_t<decltype(x.u)>>) {
3064e712a92bSpeter klausler     if (const auto *funcRef{
3065e712a92bSpeter klausler             std::get_if<common::Indirection<parser::FunctionReference>>(
3066e712a92bSpeter klausler                 &x.u)}) {
3067e712a92bSpeter klausler       // Function references in Exprs might turn out to be misparsed structure
3068e712a92bSpeter klausler       // constructors; we have to try generic procedure resolution
3069e712a92bSpeter klausler       // first to be sure.
3070e712a92bSpeter klausler       std::optional<parser::StructureConstructor> ctor;
3071e712a92bSpeter klausler       result = Analyze(funcRef->value(), &ctor);
3072e712a92bSpeter klausler       if (result && ctor) {
3073e712a92bSpeter klausler         // A misparsed function reference is really a structure
3074e712a92bSpeter klausler         // constructor.  Repair the parse tree in situ.
3075e712a92bSpeter klausler         const_cast<PARSED &>(x).u = std::move(*ctor);
3076e712a92bSpeter klausler       }
3077e712a92bSpeter klausler     } else {
3078e712a92bSpeter klausler       result = Analyze(x.u);
3079e712a92bSpeter klausler     }
3080e712a92bSpeter klausler   } else {
3081e712a92bSpeter klausler     result = Analyze(x.u);
3082e712a92bSpeter klausler   }
3083e712a92bSpeter klausler   if (result) {
3084934b27a9Speter klausler     SetExpr(x, Fold(std::move(*result)));
30853a1afd8cSpeter klausler     return x.typedExpr->v;
3086e712a92bSpeter klausler   } else {
30873a1afd8cSpeter klausler     ResetExpr(x);
308864ab3302SCarolineConcatto     if (!context_.AnyFatalError()) {
30898670e499SCaroline Concatto       std::string buf;
30908670e499SCaroline Concatto       llvm::raw_string_ostream dump{buf};
309164ab3302SCarolineConcatto       parser::DumpTree(dump, x);
309264ab3302SCarolineConcatto       Say("Internal error: Expression analysis failed on: %s"_err_en_US,
309364ab3302SCarolineConcatto           dump.str());
309464ab3302SCarolineConcatto     }
30953a1afd8cSpeter klausler     return std::nullopt;
309664ab3302SCarolineConcatto   }
3097e712a92bSpeter klausler }
309864ab3302SCarolineConcatto 
Analyze(const parser::Expr & expr)309964ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Expr &expr) {
3100934b27a9Speter klausler   return ExprOrVariable(expr, expr.source);
310164ab3302SCarolineConcatto }
310264ab3302SCarolineConcatto 
Analyze(const parser::Variable & variable)310364ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::Analyze(const parser::Variable &variable) {
3104934b27a9Speter klausler   return ExprOrVariable(variable, variable.GetSource());
310564ab3302SCarolineConcatto }
310664ab3302SCarolineConcatto 
Analyze(const parser::Selector & selector)3107e712a92bSpeter klausler MaybeExpr ExpressionAnalyzer::Analyze(const parser::Selector &selector) {
3108e712a92bSpeter klausler   if (const auto *var{std::get_if<parser::Variable>(&selector.u)}) {
3109e712a92bSpeter klausler     if (!useSavedTypedExprs_ || !var->typedExpr) {
3110e712a92bSpeter klausler       parser::CharBlock source{var->GetSource()};
3111e712a92bSpeter klausler       auto restorer{GetContextualMessages().SetLocation(source)};
3112e712a92bSpeter klausler       FixMisparsedFunctionReference(context_, var->u);
3113e712a92bSpeter klausler       if (const auto *funcRef{
3114e712a92bSpeter klausler               std::get_if<common::Indirection<parser::FunctionReference>>(
3115e712a92bSpeter klausler                   &var->u)}) {
3116e712a92bSpeter klausler         // A Selector that parsed as a Variable might turn out during analysis
3117e712a92bSpeter klausler         // to actually be a structure constructor.  In that case, repair the
3118e712a92bSpeter klausler         // Variable parse tree node into an Expr
3119e712a92bSpeter klausler         std::optional<parser::StructureConstructor> ctor;
3120e712a92bSpeter klausler         if (MaybeExpr result{Analyze(funcRef->value(), &ctor)}) {
3121e712a92bSpeter klausler           if (ctor) {
3122e712a92bSpeter klausler             auto &writable{const_cast<parser::Selector &>(selector)};
3123e712a92bSpeter klausler             writable.u = parser::Expr{std::move(*ctor)};
3124e712a92bSpeter klausler             auto &expr{std::get<parser::Expr>(writable.u)};
3125e712a92bSpeter klausler             expr.source = source;
3126e712a92bSpeter klausler             SetExpr(expr, Fold(std::move(*result)));
3127e712a92bSpeter klausler             return expr.typedExpr->v;
3128e712a92bSpeter klausler           } else {
3129e712a92bSpeter klausler             SetExpr(*var, Fold(std::move(*result)));
3130e712a92bSpeter klausler             return var->typedExpr->v;
3131e712a92bSpeter klausler           }
3132e712a92bSpeter klausler         } else {
3133e712a92bSpeter klausler           ResetExpr(*var);
3134e712a92bSpeter klausler           if (context_.AnyFatalError()) {
3135e712a92bSpeter klausler             return std::nullopt;
3136e712a92bSpeter klausler           }
3137e712a92bSpeter klausler         }
3138e712a92bSpeter klausler       }
3139e712a92bSpeter klausler     }
3140e712a92bSpeter klausler   }
3141e712a92bSpeter klausler   // Not a Variable -> FunctionReference; handle normally as Variable or Expr
3142e712a92bSpeter klausler   return Analyze(selector.u);
3143e712a92bSpeter klausler }
3144e712a92bSpeter klausler 
Analyze(const parser::DataStmtConstant & x)31453a1afd8cSpeter klausler MaybeExpr ExpressionAnalyzer::Analyze(const parser::DataStmtConstant &x) {
3146a94b721dSpeter klausler   auto restorer{common::ScopedSet(inDataStmtConstant_, true)};
3147934b27a9Speter klausler   return ExprOrVariable(x, x.source);
31483a1afd8cSpeter klausler }
31493a1afd8cSpeter klausler 
Analyze(const parser::AllocateObject & x)315092d27b96SJean Perier MaybeExpr ExpressionAnalyzer::Analyze(const parser::AllocateObject &x) {
3151e712a92bSpeter klausler   return ExprOrVariable(x, parser::FindSourceLocation(x));
315292d27b96SJean Perier }
315392d27b96SJean Perier 
Analyze(const parser::PointerObject & x)315492d27b96SJean Perier MaybeExpr ExpressionAnalyzer::Analyze(const parser::PointerObject &x) {
3155e712a92bSpeter klausler   return ExprOrVariable(x, parser::FindSourceLocation(x));
315692d27b96SJean Perier }
315792d27b96SJean Perier 
AnalyzeKindSelector(TypeCategory category,const std::optional<parser::KindSelector> & selector)315864ab3302SCarolineConcatto Expr<SubscriptInteger> ExpressionAnalyzer::AnalyzeKindSelector(
315964ab3302SCarolineConcatto     TypeCategory category,
316064ab3302SCarolineConcatto     const std::optional<parser::KindSelector> &selector) {
316164ab3302SCarolineConcatto   int defaultKind{GetDefaultKind(category)};
316264ab3302SCarolineConcatto   if (!selector) {
316364ab3302SCarolineConcatto     return Expr<SubscriptInteger>{defaultKind};
316464ab3302SCarolineConcatto   }
3165cd03e96fSPeter Klausler   return common::visit(
316664ab3302SCarolineConcatto       common::visitors{
316764ab3302SCarolineConcatto           [&](const parser::ScalarIntConstantExpr &x) {
316864ab3302SCarolineConcatto             if (MaybeExpr kind{Analyze(x)}) {
3169934b27a9Speter klausler               if (std::optional<std::int64_t> code{ToInt64(*kind)}) {
317064ab3302SCarolineConcatto                 if (CheckIntrinsicKind(category, *code)) {
317164ab3302SCarolineConcatto                   return Expr<SubscriptInteger>{*code};
317264ab3302SCarolineConcatto                 }
3173934b27a9Speter klausler               } else if (auto *intExpr{UnwrapExpr<Expr<SomeInteger>>(*kind)}) {
317464ab3302SCarolineConcatto                 return ConvertToType<SubscriptInteger>(std::move(*intExpr));
317564ab3302SCarolineConcatto               }
317664ab3302SCarolineConcatto             }
317764ab3302SCarolineConcatto             return Expr<SubscriptInteger>{defaultKind};
317864ab3302SCarolineConcatto           },
317964ab3302SCarolineConcatto           [&](const parser::KindSelector::StarSize &x) {
318064ab3302SCarolineConcatto             std::intmax_t size = x.v;
318164ab3302SCarolineConcatto             if (!CheckIntrinsicSize(category, size)) {
318264ab3302SCarolineConcatto               size = defaultKind;
318364ab3302SCarolineConcatto             } else if (category == TypeCategory::Complex) {
318464ab3302SCarolineConcatto               size /= 2;
318564ab3302SCarolineConcatto             }
318664ab3302SCarolineConcatto             return Expr<SubscriptInteger>{size};
318764ab3302SCarolineConcatto           },
318864ab3302SCarolineConcatto       },
318964ab3302SCarolineConcatto       selector->u);
319064ab3302SCarolineConcatto }
319164ab3302SCarolineConcatto 
GetDefaultKind(common::TypeCategory category)319264ab3302SCarolineConcatto int ExpressionAnalyzer::GetDefaultKind(common::TypeCategory category) {
319364ab3302SCarolineConcatto   return context_.GetDefaultKind(category);
319464ab3302SCarolineConcatto }
319564ab3302SCarolineConcatto 
GetDefaultKindOfType(common::TypeCategory category)319664ab3302SCarolineConcatto DynamicType ExpressionAnalyzer::GetDefaultKindOfType(
319764ab3302SCarolineConcatto     common::TypeCategory category) {
319864ab3302SCarolineConcatto   return {category, GetDefaultKind(category)};
319964ab3302SCarolineConcatto }
320064ab3302SCarolineConcatto 
CheckIntrinsicKind(TypeCategory category,std::int64_t kind)320164ab3302SCarolineConcatto bool ExpressionAnalyzer::CheckIntrinsicKind(
320264ab3302SCarolineConcatto     TypeCategory category, std::int64_t kind) {
320323c2bedfSPeter Klausler   if (foldingContext_.targetCharacteristics().IsTypeEnabled(
320423c2bedfSPeter Klausler           category, kind)) { // C712, C714, C715, C727
320523c2bedfSPeter Klausler     return true;
320623c2bedfSPeter Klausler   } else if (foldingContext_.targetCharacteristics().CanSupportType(
320723c2bedfSPeter Klausler                  category, kind)) {
320823c2bedfSPeter Klausler     Say("%s(KIND=%jd) is not an enabled type for this targe"_warn_en_US,
320923c2bedfSPeter Klausler         ToUpperCase(EnumToString(category)), kind);
321064ab3302SCarolineConcatto     return true;
321164ab3302SCarolineConcatto   } else {
321264ab3302SCarolineConcatto     Say("%s(KIND=%jd) is not a supported type"_err_en_US,
321364ab3302SCarolineConcatto         ToUpperCase(EnumToString(category)), kind);
321464ab3302SCarolineConcatto     return false;
321564ab3302SCarolineConcatto   }
321664ab3302SCarolineConcatto }
321764ab3302SCarolineConcatto 
CheckIntrinsicSize(TypeCategory category,std::int64_t size)321864ab3302SCarolineConcatto bool ExpressionAnalyzer::CheckIntrinsicSize(
321964ab3302SCarolineConcatto     TypeCategory category, std::int64_t size) {
322023c2bedfSPeter Klausler   std::int64_t kind{size};
322164ab3302SCarolineConcatto   if (category == TypeCategory::Complex) {
322264ab3302SCarolineConcatto     // COMPLEX*16 == COMPLEX(KIND=8)
322323c2bedfSPeter Klausler     if (size % 2 == 0) {
322423c2bedfSPeter Klausler       kind = size / 2;
322523c2bedfSPeter Klausler     } else {
322623c2bedfSPeter Klausler       Say("COMPLEX*%jd is not a supported type"_err_en_US, size);
322723c2bedfSPeter Klausler       return false;
322864ab3302SCarolineConcatto     }
322964ab3302SCarolineConcatto   }
323023c2bedfSPeter Klausler   if (foldingContext_.targetCharacteristics().IsTypeEnabled(
323123c2bedfSPeter Klausler           category, kind)) { // C712, C714, C715, C727
323223c2bedfSPeter Klausler     return true;
323323c2bedfSPeter Klausler   } else if (foldingContext_.targetCharacteristics().CanSupportType(
323423c2bedfSPeter Klausler                  category, kind)) {
323523c2bedfSPeter Klausler     Say("%s*%jd is not an enabled type for this target"_warn_en_US,
323623c2bedfSPeter Klausler         ToUpperCase(EnumToString(category)), size);
323723c2bedfSPeter Klausler     return true;
323823c2bedfSPeter Klausler   } else {
323964ab3302SCarolineConcatto     Say("%s*%jd is not a supported type"_err_en_US,
324064ab3302SCarolineConcatto         ToUpperCase(EnumToString(category)), size);
324164ab3302SCarolineConcatto     return false;
324264ab3302SCarolineConcatto   }
324323c2bedfSPeter Klausler }
324464ab3302SCarolineConcatto 
AddImpliedDo(parser::CharBlock name,int kind)32453a1afd8cSpeter klausler bool ExpressionAnalyzer::AddImpliedDo(parser::CharBlock name, int kind) {
32463a1afd8cSpeter klausler   return impliedDos_.insert(std::make_pair(name, kind)).second;
324764ab3302SCarolineConcatto }
324864ab3302SCarolineConcatto 
RemoveImpliedDo(parser::CharBlock name)32493a1afd8cSpeter klausler void ExpressionAnalyzer::RemoveImpliedDo(parser::CharBlock name) {
32503a1afd8cSpeter klausler   auto iter{impliedDos_.find(name)};
32513a1afd8cSpeter klausler   if (iter != impliedDos_.end()) {
32523a1afd8cSpeter klausler     impliedDos_.erase(iter);
325364ab3302SCarolineConcatto   }
325464ab3302SCarolineConcatto }
325564ab3302SCarolineConcatto 
IsImpliedDo(parser::CharBlock name) const32563a1afd8cSpeter klausler std::optional<int> ExpressionAnalyzer::IsImpliedDo(
325764ab3302SCarolineConcatto     parser::CharBlock name) const {
32583a1afd8cSpeter klausler   auto iter{impliedDos_.find(name)};
32593a1afd8cSpeter klausler   if (iter != impliedDos_.cend()) {
326064ab3302SCarolineConcatto     return {iter->second};
326164ab3302SCarolineConcatto   } else {
326264ab3302SCarolineConcatto     return std::nullopt;
326364ab3302SCarolineConcatto   }
326464ab3302SCarolineConcatto }
326564ab3302SCarolineConcatto 
EnforceTypeConstraint(parser::CharBlock at,const MaybeExpr & result,TypeCategory category,bool defaultKind)326664ab3302SCarolineConcatto bool ExpressionAnalyzer::EnforceTypeConstraint(parser::CharBlock at,
326764ab3302SCarolineConcatto     const MaybeExpr &result, TypeCategory category, bool defaultKind) {
326864ab3302SCarolineConcatto   if (result) {
326964ab3302SCarolineConcatto     if (auto type{result->GetType()}) {
327064ab3302SCarolineConcatto       if (type->category() != category) { // C885
327164ab3302SCarolineConcatto         Say(at, "Must have %s type, but is %s"_err_en_US,
327264ab3302SCarolineConcatto             ToUpperCase(EnumToString(category)),
327364ab3302SCarolineConcatto             ToUpperCase(type->AsFortran()));
327464ab3302SCarolineConcatto         return false;
327564ab3302SCarolineConcatto       } else if (defaultKind) {
327664ab3302SCarolineConcatto         int kind{context_.GetDefaultKind(category)};
327764ab3302SCarolineConcatto         if (type->kind() != kind) {
327864ab3302SCarolineConcatto           Say(at, "Must have default kind(%d) of %s type, but is %s"_err_en_US,
327964ab3302SCarolineConcatto               kind, ToUpperCase(EnumToString(category)),
328064ab3302SCarolineConcatto               ToUpperCase(type->AsFortran()));
328164ab3302SCarolineConcatto           return false;
328264ab3302SCarolineConcatto         }
328364ab3302SCarolineConcatto       }
328464ab3302SCarolineConcatto     } else {
328564ab3302SCarolineConcatto       Say(at, "Must have %s type, but is typeless"_err_en_US,
328664ab3302SCarolineConcatto           ToUpperCase(EnumToString(category)));
328764ab3302SCarolineConcatto       return false;
328864ab3302SCarolineConcatto     }
328964ab3302SCarolineConcatto   }
329064ab3302SCarolineConcatto   return true;
329164ab3302SCarolineConcatto }
329264ab3302SCarolineConcatto 
MakeFunctionRef(parser::CharBlock callSite,ProcedureDesignator && proc,ActualArguments && arguments)329364ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::MakeFunctionRef(parser::CharBlock callSite,
329464ab3302SCarolineConcatto     ProcedureDesignator &&proc, ActualArguments &&arguments) {
329564ab3302SCarolineConcatto   if (const auto *intrinsic{std::get_if<SpecificIntrinsic>(&proc.u)}) {
32963632e9f8SPeter Klausler     if (intrinsic->characteristics.value().attrs.test(
32973632e9f8SPeter Klausler             characteristics::Procedure::Attr::NullPointer) &&
32983632e9f8SPeter Klausler         arguments.empty()) {
329964ab3302SCarolineConcatto       return Expr<SomeType>{NullPointer{}};
330064ab3302SCarolineConcatto     }
330164ab3302SCarolineConcatto   }
33029977b24aSpeter klausler   if (const Symbol * symbol{proc.GetSymbol()}) {
33039977b24aSpeter klausler     if (!ResolveForward(*symbol)) {
33049977b24aSpeter klausler       return std::nullopt;
33059977b24aSpeter klausler     }
33069977b24aSpeter klausler   }
330764ab3302SCarolineConcatto   if (auto chars{CheckCall(callSite, proc, arguments)}) {
330864ab3302SCarolineConcatto     if (chars->functionResult) {
330964ab3302SCarolineConcatto       const auto &result{*chars->functionResult};
331064ab3302SCarolineConcatto       if (result.IsProcedurePointer()) {
331164ab3302SCarolineConcatto         return Expr<SomeType>{
331264ab3302SCarolineConcatto             ProcedureRef{std::move(proc), std::move(arguments)}};
331364ab3302SCarolineConcatto       } else {
331464ab3302SCarolineConcatto         // Not a procedure pointer, so type and shape are known.
331564ab3302SCarolineConcatto         return TypedWrapper<FunctionRef, ProcedureRef>(
331664ab3302SCarolineConcatto             DEREF(result.GetTypeAndShape()).type(),
331764ab3302SCarolineConcatto             ProcedureRef{std::move(proc), std::move(arguments)});
331864ab3302SCarolineConcatto       }
3319c1a02405Speter klausler     } else {
3320c1a02405Speter klausler       Say("Function result characteristics are not known"_err_en_US);
332164ab3302SCarolineConcatto     }
332264ab3302SCarolineConcatto   }
332364ab3302SCarolineConcatto   return std::nullopt;
332464ab3302SCarolineConcatto }
332564ab3302SCarolineConcatto 
MakeFunctionRef(parser::CharBlock intrinsic,ActualArguments && arguments)332664ab3302SCarolineConcatto MaybeExpr ExpressionAnalyzer::MakeFunctionRef(
332764ab3302SCarolineConcatto     parser::CharBlock intrinsic, ActualArguments &&arguments) {
332864ab3302SCarolineConcatto   if (std::optional<SpecificCall> specificCall{
332964ab3302SCarolineConcatto           context_.intrinsics().Probe(CallCharacteristics{intrinsic.ToString()},
333052711fb8Speter klausler               arguments, GetFoldingContext())}) {
333164ab3302SCarolineConcatto     return MakeFunctionRef(intrinsic,
333264ab3302SCarolineConcatto         ProcedureDesignator{std::move(specificCall->specificIntrinsic)},
333364ab3302SCarolineConcatto         std::move(specificCall->arguments));
333464ab3302SCarolineConcatto   } else {
333564ab3302SCarolineConcatto     return std::nullopt;
333664ab3302SCarolineConcatto   }
333764ab3302SCarolineConcatto }
333864ab3302SCarolineConcatto 
Analyze(const parser::Variable & x)333964ab3302SCarolineConcatto void ArgumentAnalyzer::Analyze(const parser::Variable &x) {
334064ab3302SCarolineConcatto   source_.ExtendToCover(x.GetSource());
334164ab3302SCarolineConcatto   if (MaybeExpr expr{context_.Analyze(x)}) {
3342a0a1f519STim Keith     if (!IsConstantExpr(*expr)) {
334364ab3302SCarolineConcatto       actuals_.emplace_back(std::move(*expr));
3344fa0443f2SPeter Klausler       SetArgSourceLocation(actuals_.back(), x.GetSource());
3345a0a1f519STim Keith       return;
334664ab3302SCarolineConcatto     }
3347d8713523STim Keith     const Symbol *symbol{GetLastSymbol(*expr)};
3348d8713523STim Keith     if (!symbol) {
3349d8713523STim Keith       context_.SayAt(x, "Assignment to constant '%s' is not allowed"_err_en_US,
3350d8713523STim Keith           x.GetSource());
3351d8713523STim Keith     } else if (auto *subp{symbol->detailsIf<semantics::SubprogramDetails>()}) {
3352d8713523STim Keith       auto *msg{context_.SayAt(x,
3353d8713523STim Keith           "Assignment to subprogram '%s' is not allowed"_err_en_US,
3354d8713523STim Keith           symbol->name())};
3355d8713523STim Keith       if (subp->isFunction()) {
3356d8713523STim Keith         const auto &result{subp->result().name()};
3357ef141aecSPeter Klausler         msg->Attach(result, "Function result is '%s'"_en_US, result);
3358d8713523STim Keith       }
3359d8713523STim Keith     } else {
3360d8713523STim Keith       context_.SayAt(x, "Assignment to constant '%s' is not allowed"_err_en_US,
3361d8713523STim Keith           symbol->name());
3362d8713523STim Keith     }
3363a0a1f519STim Keith   }
3364a0a1f519STim Keith   fatalErrors_ = true;
336564ab3302SCarolineConcatto }
336664ab3302SCarolineConcatto 
Analyze(const parser::ActualArgSpec & arg,bool isSubroutine)336764ab3302SCarolineConcatto void ArgumentAnalyzer::Analyze(
336864ab3302SCarolineConcatto     const parser::ActualArgSpec &arg, bool isSubroutine) {
336964ab3302SCarolineConcatto   // TODO: Actual arguments that are procedures and procedure pointers need to
337064ab3302SCarolineConcatto   // be detected and represented (they're not expressions).
337164ab3302SCarolineConcatto   // TODO: C1534: Don't allow a "restricted" specific intrinsic to be passed.
337264ab3302SCarolineConcatto   std::optional<ActualArgument> actual;
3373cd03e96fSPeter Klausler   common::visit(
3374df209b80SPeter Steinfeld       common::visitors{
337564ab3302SCarolineConcatto           [&](const common::Indirection<parser::Expr> &x) {
337664ab3302SCarolineConcatto             actual = AnalyzeExpr(x.value());
3377fa0443f2SPeter Klausler             SetArgSourceLocation(actual, x.value().source);
337864ab3302SCarolineConcatto           },
3379ae0d1d2eSPeter Steinfeld           [&](const parser::AltReturnSpec &label) {
338064ab3302SCarolineConcatto             if (!isSubroutine) {
3381df209b80SPeter Steinfeld               context_.Say("alternate return specification may not appear on"
338264ab3302SCarolineConcatto                            " function reference"_err_en_US);
338364ab3302SCarolineConcatto             }
3384ae0d1d2eSPeter Steinfeld             actual = ActualArgument(label.v);
338564ab3302SCarolineConcatto           },
338664ab3302SCarolineConcatto           [&](const parser::ActualArg::PercentRef &) {
3387d1344422SPeter Steinfeld             context_.Say("%REF() intrinsic for arguments"_todo_en_US);
338864ab3302SCarolineConcatto           },
338964ab3302SCarolineConcatto           [&](const parser::ActualArg::PercentVal &) {
3390d1344422SPeter Steinfeld             context_.Say("%VAL() intrinsic for arguments"_todo_en_US);
339164ab3302SCarolineConcatto           },
339264ab3302SCarolineConcatto       },
339364ab3302SCarolineConcatto       std::get<parser::ActualArg>(arg.t).u);
339464ab3302SCarolineConcatto   if (actual) {
339564ab3302SCarolineConcatto     if (const auto &argKW{std::get<std::optional<parser::Keyword>>(arg.t)}) {
339664ab3302SCarolineConcatto       actual->set_keyword(argKW->v.source);
339764ab3302SCarolineConcatto     }
339864ab3302SCarolineConcatto     actuals_.emplace_back(std::move(*actual));
3399ae0d1d2eSPeter Steinfeld   } else {
340064ab3302SCarolineConcatto     fatalErrors_ = true;
340164ab3302SCarolineConcatto   }
340264ab3302SCarolineConcatto }
340364ab3302SCarolineConcatto 
IsIntrinsicRelational(RelationalOperator opr,const DynamicType & leftType,const DynamicType & rightType) const3404d9195d66Speter klausler bool ArgumentAnalyzer::IsIntrinsicRelational(RelationalOperator opr,
3405d9195d66Speter klausler     const DynamicType &leftType, const DynamicType &rightType) const {
340664ab3302SCarolineConcatto   CHECK(actuals_.size() == 2);
340764ab3302SCarolineConcatto   return semantics::IsIntrinsicRelational(
3408d9195d66Speter klausler       opr, leftType, GetRank(0), rightType, GetRank(1));
340964ab3302SCarolineConcatto }
341064ab3302SCarolineConcatto 
IsIntrinsicNumeric(NumericOperator opr) const341164ab3302SCarolineConcatto bool ArgumentAnalyzer::IsIntrinsicNumeric(NumericOperator opr) const {
3412d9195d66Speter klausler   std::optional<DynamicType> leftType{GetType(0)};
341364ab3302SCarolineConcatto   if (actuals_.size() == 1) {
341464ab3302SCarolineConcatto     if (IsBOZLiteral(0)) {
3415d9195d66Speter klausler       return opr == NumericOperator::Add; // unary '+'
341664ab3302SCarolineConcatto     } else {
3417d9195d66Speter klausler       return leftType && semantics::IsIntrinsicNumeric(*leftType);
341864ab3302SCarolineConcatto     }
341964ab3302SCarolineConcatto   } else {
3420d9195d66Speter klausler     std::optional<DynamicType> rightType{GetType(1)};
3421d9195d66Speter klausler     if (IsBOZLiteral(0) && rightType) { // BOZ opr Integer/Real
3422d9195d66Speter klausler       auto cat1{rightType->category()};
342364ab3302SCarolineConcatto       return cat1 == TypeCategory::Integer || cat1 == TypeCategory::Real;
3424d9195d66Speter klausler     } else if (IsBOZLiteral(1) && leftType) { // Integer/Real opr BOZ
3425d9195d66Speter klausler       auto cat0{leftType->category()};
342664ab3302SCarolineConcatto       return cat0 == TypeCategory::Integer || cat0 == TypeCategory::Real;
342764ab3302SCarolineConcatto     } else {
3428d9195d66Speter klausler       return leftType && rightType &&
3429d9195d66Speter klausler           semantics::IsIntrinsicNumeric(
3430d9195d66Speter klausler               *leftType, GetRank(0), *rightType, GetRank(1));
343164ab3302SCarolineConcatto     }
343264ab3302SCarolineConcatto   }
343364ab3302SCarolineConcatto }
343464ab3302SCarolineConcatto 
IsIntrinsicLogical() const343564ab3302SCarolineConcatto bool ArgumentAnalyzer::IsIntrinsicLogical() const {
3436d9195d66Speter klausler   if (std::optional<DynamicType> leftType{GetType(0)}) {
343764ab3302SCarolineConcatto     if (actuals_.size() == 1) {
3438d9195d66Speter klausler       return semantics::IsIntrinsicLogical(*leftType);
3439d9195d66Speter klausler     } else if (std::optional<DynamicType> rightType{GetType(1)}) {
344064ab3302SCarolineConcatto       return semantics::IsIntrinsicLogical(
3441d9195d66Speter klausler           *leftType, GetRank(0), *rightType, GetRank(1));
344264ab3302SCarolineConcatto     }
344364ab3302SCarolineConcatto   }
3444d9195d66Speter klausler   return false;
3445d9195d66Speter klausler }
344664ab3302SCarolineConcatto 
IsIntrinsicConcat() const344764ab3302SCarolineConcatto bool ArgumentAnalyzer::IsIntrinsicConcat() const {
3448d9195d66Speter klausler   if (std::optional<DynamicType> leftType{GetType(0)}) {
3449d9195d66Speter klausler     if (std::optional<DynamicType> rightType{GetType(1)}) {
345064ab3302SCarolineConcatto       return semantics::IsIntrinsicConcat(
3451d9195d66Speter klausler           *leftType, GetRank(0), *rightType, GetRank(1));
3452d9195d66Speter klausler     }
3453d9195d66Speter klausler   }
3454d9195d66Speter klausler   return false;
345564ab3302SCarolineConcatto }
345664ab3302SCarolineConcatto 
CheckConformance()3457d9195d66Speter klausler bool ArgumentAnalyzer::CheckConformance() {
34581a633e72Speter klausler   if (actuals_.size() == 2) {
34591a633e72Speter klausler     const auto *lhs{actuals_.at(0).value().UnwrapExpr()};
34601a633e72Speter klausler     const auto *rhs{actuals_.at(1).value().UnwrapExpr()};
34611a633e72Speter klausler     if (lhs && rhs) {
34621a633e72Speter klausler       auto &foldingContext{context_.GetFoldingContext()};
34631a633e72Speter klausler       auto lhShape{GetShape(foldingContext, *lhs)};
34641a633e72Speter klausler       auto rhShape{GetShape(foldingContext, *rhs)};
34651a633e72Speter klausler       if (lhShape && rhShape) {
3466d9195d66Speter klausler         if (!evaluate::CheckConformance(foldingContext.messages(), *lhShape,
3467dfecbcaeSpeter klausler                 *rhShape, CheckConformanceFlags::EitherScalarExpandable,
3468dfecbcaeSpeter klausler                 "left operand", "right operand")
3469d9195d66Speter klausler                  .value_or(false /*fail when conformance is not known now*/)) {
3470d9195d66Speter klausler           fatalErrors_ = true;
3471d9195d66Speter klausler           return false;
3472d9195d66Speter klausler         }
34731a633e72Speter klausler       }
34741a633e72Speter klausler     }
34751a633e72Speter klausler   }
34761a633e72Speter klausler   return true; // no proven problem
34771a633e72Speter klausler }
34781a633e72Speter klausler 
CheckForNullPointer(const char * where)3479d9195d66Speter klausler bool ArgumentAnalyzer::CheckForNullPointer(const char *where) {
3480d9195d66Speter klausler   for (const std::optional<ActualArgument> &arg : actuals_) {
3481d9195d66Speter klausler     if (arg) {
3482d9195d66Speter klausler       if (const Expr<SomeType> *expr{arg->UnwrapExpr()}) {
3483d9195d66Speter klausler         if (IsNullPointer(*expr)) {
348464ab3302SCarolineConcatto           context_.Say(
3485d9195d66Speter klausler               source_, "A NULL() pointer is not allowed %s"_err_en_US, where);
3486d9195d66Speter klausler           fatalErrors_ = true;
3487d9195d66Speter klausler           return false;
3488d9195d66Speter klausler         }
3489d9195d66Speter klausler       }
3490d9195d66Speter klausler     }
3491d9195d66Speter klausler   }
3492d9195d66Speter klausler   return true;
3493d9195d66Speter klausler }
3494d9195d66Speter klausler 
TryDefinedOp(const char * opr,parser::MessageFixedText error,const Symbol ** definedOpSymbolPtr,bool isUserOp)3495d9195d66Speter klausler MaybeExpr ArgumentAnalyzer::TryDefinedOp(const char *opr,
3496d9195d66Speter klausler     parser::MessageFixedText error, const Symbol **definedOpSymbolPtr,
3497d9195d66Speter klausler     bool isUserOp) {
3498d9195d66Speter klausler   if (AnyUntypedOrMissingOperand()) {
3499d9195d66Speter klausler     context_.Say(error, ToUpperCase(opr), TypeAsFortran(0), TypeAsFortran(1));
3500d9195d66Speter klausler     return std::nullopt;
3501d9195d66Speter klausler   }
3502d9195d66Speter klausler   const Symbol *localDefinedOpSymbolPtr{nullptr};
3503d9195d66Speter klausler   if (!definedOpSymbolPtr) {
3504d9195d66Speter klausler     definedOpSymbolPtr = &localDefinedOpSymbolPtr;
3505d9195d66Speter klausler   }
350664ab3302SCarolineConcatto   {
350764ab3302SCarolineConcatto     auto restorer{context_.GetContextualMessages().DiscardMessages()};
350864ab3302SCarolineConcatto     std::string oprNameString{
350964ab3302SCarolineConcatto         isUserOp ? std::string{opr} : "operator("s + opr + ')'};
351064ab3302SCarolineConcatto     parser::CharBlock oprName{oprNameString};
351164ab3302SCarolineConcatto     const auto &scope{context_.context().FindScope(source_)};
351264ab3302SCarolineConcatto     if (Symbol * symbol{scope.FindSymbol(oprName)}) {
3513d9195d66Speter klausler       *definedOpSymbolPtr = symbol;
351464ab3302SCarolineConcatto       parser::Name name{symbol->name(), symbol};
351564ab3302SCarolineConcatto       if (auto result{context_.AnalyzeDefinedOp(name, GetActuals())}) {
351664ab3302SCarolineConcatto         return result;
351764ab3302SCarolineConcatto       }
351864ab3302SCarolineConcatto     }
351964ab3302SCarolineConcatto     for (std::size_t passIndex{0}; passIndex < actuals_.size(); ++passIndex) {
3520d9195d66Speter klausler       if (const Symbol *
3521d9195d66Speter klausler           symbol{FindBoundOp(oprName, passIndex, *definedOpSymbolPtr)}) {
352264ab3302SCarolineConcatto         if (MaybeExpr result{TryBoundOp(*symbol, passIndex)}) {
352364ab3302SCarolineConcatto           return result;
352464ab3302SCarolineConcatto         }
352564ab3302SCarolineConcatto       }
352664ab3302SCarolineConcatto     }
352764ab3302SCarolineConcatto   }
3528d9195d66Speter klausler   if (*definedOpSymbolPtr) {
3529d9195d66Speter klausler     SayNoMatch(ToUpperCase((*definedOpSymbolPtr)->name().ToString()));
353064ab3302SCarolineConcatto   } else if (actuals_.size() == 1 || AreConformable()) {
3531d9195d66Speter klausler     if (CheckForNullPointer()) {
3532d9195d66Speter klausler       context_.Say(error, ToUpperCase(opr), TypeAsFortran(0), TypeAsFortran(1));
3533d9195d66Speter klausler     }
353464ab3302SCarolineConcatto   } else {
353564ab3302SCarolineConcatto     context_.Say(
353664ab3302SCarolineConcatto         "Operands of %s are not conformable; have rank %d and rank %d"_err_en_US,
353764ab3302SCarolineConcatto         ToUpperCase(opr), actuals_[0]->Rank(), actuals_[1]->Rank());
353864ab3302SCarolineConcatto   }
353964ab3302SCarolineConcatto   return std::nullopt;
354064ab3302SCarolineConcatto }
354164ab3302SCarolineConcatto 
TryDefinedOp(std::vector<const char * > oprs,parser::MessageFixedText error)354264ab3302SCarolineConcatto MaybeExpr ArgumentAnalyzer::TryDefinedOp(
3543d9195d66Speter klausler     std::vector<const char *> oprs, parser::MessageFixedText error) {
3544d9195d66Speter klausler   const Symbol *definedOpSymbolPtr{nullptr};
354564ab3302SCarolineConcatto   for (std::size_t i{1}; i < oprs.size(); ++i) {
354664ab3302SCarolineConcatto     auto restorer{context_.GetContextualMessages().DiscardMessages()};
3547d9195d66Speter klausler     if (auto result{TryDefinedOp(oprs[i], error, &definedOpSymbolPtr)}) {
354864ab3302SCarolineConcatto       return result;
354964ab3302SCarolineConcatto     }
355064ab3302SCarolineConcatto   }
3551d9195d66Speter klausler   return TryDefinedOp(oprs[0], error, &definedOpSymbolPtr);
355264ab3302SCarolineConcatto }
355364ab3302SCarolineConcatto 
TryBoundOp(const Symbol & symbol,int passIndex)355464ab3302SCarolineConcatto MaybeExpr ArgumentAnalyzer::TryBoundOp(const Symbol &symbol, int passIndex) {
355564ab3302SCarolineConcatto   ActualArguments localActuals{actuals_};
355664ab3302SCarolineConcatto   const Symbol *proc{GetBindingResolution(GetType(passIndex), symbol)};
355764ab3302SCarolineConcatto   if (!proc) {
355864ab3302SCarolineConcatto     proc = &symbol;
355964ab3302SCarolineConcatto     localActuals.at(passIndex).value().set_isPassedObject();
356064ab3302SCarolineConcatto   }
35611a633e72Speter klausler   CheckConformance();
356264ab3302SCarolineConcatto   return context_.MakeFunctionRef(
356364ab3302SCarolineConcatto       source_, ProcedureDesignator{*proc}, std::move(localActuals));
356464ab3302SCarolineConcatto }
356564ab3302SCarolineConcatto 
TryDefinedAssignment()356664ab3302SCarolineConcatto std::optional<ProcedureRef> ArgumentAnalyzer::TryDefinedAssignment() {
356764ab3302SCarolineConcatto   using semantics::Tristate;
356864ab3302SCarolineConcatto   const Expr<SomeType> &lhs{GetExpr(0)};
356964ab3302SCarolineConcatto   const Expr<SomeType> &rhs{GetExpr(1)};
357064ab3302SCarolineConcatto   std::optional<DynamicType> lhsType{lhs.GetType()};
357164ab3302SCarolineConcatto   std::optional<DynamicType> rhsType{rhs.GetType()};
357264ab3302SCarolineConcatto   int lhsRank{lhs.Rank()};
357364ab3302SCarolineConcatto   int rhsRank{rhs.Rank()};
357464ab3302SCarolineConcatto   Tristate isDefined{
357564ab3302SCarolineConcatto       semantics::IsDefinedAssignment(lhsType, lhsRank, rhsType, rhsRank)};
357664ab3302SCarolineConcatto   if (isDefined == Tristate::No) {
3577b05c8c57STim Keith     if (lhsType && rhsType) {
3578b05c8c57STim Keith       AddAssignmentConversion(*lhsType, *rhsType);
3579b05c8c57STim Keith     }
358064ab3302SCarolineConcatto     return std::nullopt; // user-defined assignment not allowed for these args
358164ab3302SCarolineConcatto   }
358264ab3302SCarolineConcatto   auto restorer{context_.GetContextualMessages().SetLocation(source_)};
358364ab3302SCarolineConcatto   if (std::optional<ProcedureRef> procRef{GetDefinedAssignmentProc()}) {
35841c2e5fd6Speter klausler     if (context_.inWhereBody() && !procRef->proc().IsElemental()) { // C1032
35851c2e5fd6Speter klausler       context_.Say(
35861c2e5fd6Speter klausler           "Defined assignment in WHERE must be elemental, but '%s' is not"_err_en_US,
35871c2e5fd6Speter klausler           DEREF(procRef->proc().GetSymbol()).name());
35881c2e5fd6Speter klausler     }
358964ab3302SCarolineConcatto     context_.CheckCall(source_, procRef->proc(), procRef->arguments());
359064ab3302SCarolineConcatto     return std::move(*procRef);
359164ab3302SCarolineConcatto   }
359264ab3302SCarolineConcatto   if (isDefined == Tristate::Yes) {
359364ab3302SCarolineConcatto     if (!lhsType || !rhsType || (lhsRank != rhsRank && rhsRank != 0) ||
359464ab3302SCarolineConcatto         !OkLogicalIntegerAssignment(lhsType->category(), rhsType->category())) {
359564ab3302SCarolineConcatto       SayNoMatch("ASSIGNMENT(=)", true);
359664ab3302SCarolineConcatto     }
359764ab3302SCarolineConcatto   }
359864ab3302SCarolineConcatto   return std::nullopt;
359964ab3302SCarolineConcatto }
360064ab3302SCarolineConcatto 
OkLogicalIntegerAssignment(TypeCategory lhs,TypeCategory rhs)360164ab3302SCarolineConcatto bool ArgumentAnalyzer::OkLogicalIntegerAssignment(
360264ab3302SCarolineConcatto     TypeCategory lhs, TypeCategory rhs) {
360364ab3302SCarolineConcatto   if (!context_.context().languageFeatures().IsEnabled(
360464ab3302SCarolineConcatto           common::LanguageFeature::LogicalIntegerAssignment)) {
360564ab3302SCarolineConcatto     return false;
360664ab3302SCarolineConcatto   }
360764ab3302SCarolineConcatto   std::optional<parser::MessageFixedText> msg;
360864ab3302SCarolineConcatto   if (lhs == TypeCategory::Integer && rhs == TypeCategory::Logical) {
360964ab3302SCarolineConcatto     // allow assignment to LOGICAL from INTEGER as a legacy extension
361073506256SPeter Klausler     msg = "assignment of LOGICAL to INTEGER"_port_en_US;
361164ab3302SCarolineConcatto   } else if (lhs == TypeCategory::Logical && rhs == TypeCategory::Integer) {
361264ab3302SCarolineConcatto     // ... and assignment to LOGICAL from INTEGER
361373506256SPeter Klausler     msg = "assignment of INTEGER to LOGICAL"_port_en_US;
361464ab3302SCarolineConcatto   } else {
361564ab3302SCarolineConcatto     return false;
361664ab3302SCarolineConcatto   }
361764ab3302SCarolineConcatto   if (context_.context().languageFeatures().ShouldWarn(
361864ab3302SCarolineConcatto           common::LanguageFeature::LogicalIntegerAssignment)) {
361964ab3302SCarolineConcatto     context_.Say(std::move(*msg));
362064ab3302SCarolineConcatto   }
362164ab3302SCarolineConcatto   return true;
362264ab3302SCarolineConcatto }
362364ab3302SCarolineConcatto 
GetDefinedAssignmentProc()362464ab3302SCarolineConcatto std::optional<ProcedureRef> ArgumentAnalyzer::GetDefinedAssignmentProc() {
362564ab3302SCarolineConcatto   auto restorer{context_.GetContextualMessages().DiscardMessages()};
362664ab3302SCarolineConcatto   std::string oprNameString{"assignment(=)"};
362764ab3302SCarolineConcatto   parser::CharBlock oprName{oprNameString};
362864ab3302SCarolineConcatto   const Symbol *proc{nullptr};
362964ab3302SCarolineConcatto   const auto &scope{context_.context().FindScope(source_)};
363064ab3302SCarolineConcatto   if (const Symbol * symbol{scope.FindSymbol(oprName)}) {
363164ab3302SCarolineConcatto     ExpressionAnalyzer::AdjustActuals noAdjustment;
36320c190575SPeter Klausler     auto pair{context_.ResolveGeneric(*symbol, actuals_, noAdjustment, true)};
36330fcda9aeSpeter klausler     if (pair.first) {
36340fcda9aeSpeter klausler       proc = pair.first;
363564ab3302SCarolineConcatto     } else {
36360fcda9aeSpeter klausler       context_.EmitGenericResolutionError(*symbol, pair.second);
363764ab3302SCarolineConcatto     }
363864ab3302SCarolineConcatto   }
3639ce6dd973STim Keith   int passedObjectIndex{-1};
3640d9195d66Speter klausler   const Symbol *definedOpSymbol{nullptr};
3641ce6dd973STim Keith   for (std::size_t i{0}; i < actuals_.size(); ++i) {
3642d9195d66Speter klausler     if (const Symbol * specific{FindBoundOp(oprName, i, definedOpSymbol)}) {
3643ce6dd973STim Keith       if (const Symbol *
3644ce6dd973STim Keith           resolution{GetBindingResolution(GetType(i), *specific)}) {
3645ce6dd973STim Keith         proc = resolution;
364664ab3302SCarolineConcatto       } else {
3647ce6dd973STim Keith         proc = specific;
3648ce6dd973STim Keith         passedObjectIndex = i;
3649ce6dd973STim Keith       }
3650ce6dd973STim Keith     }
3651ce6dd973STim Keith   }
3652ce6dd973STim Keith   if (!proc) {
365364ab3302SCarolineConcatto     return std::nullopt;
365464ab3302SCarolineConcatto   }
3655ce6dd973STim Keith   ActualArguments actualsCopy{actuals_};
3656ce6dd973STim Keith   if (passedObjectIndex >= 0) {
3657ce6dd973STim Keith     actualsCopy[passedObjectIndex]->set_isPassedObject();
3658ce6dd973STim Keith   }
3659ce6dd973STim Keith   return ProcedureRef{ProcedureDesignator{*proc}, std::move(actualsCopy)};
366064ab3302SCarolineConcatto }
366164ab3302SCarolineConcatto 
Dump(llvm::raw_ostream & os)36628670e499SCaroline Concatto void ArgumentAnalyzer::Dump(llvm::raw_ostream &os) {
366364ab3302SCarolineConcatto   os << "source_: " << source_.ToString() << " fatalErrors_ = " << fatalErrors_
366464ab3302SCarolineConcatto      << '\n';
366564ab3302SCarolineConcatto   for (const auto &actual : actuals_) {
366664ab3302SCarolineConcatto     if (!actual.has_value()) {
366764ab3302SCarolineConcatto       os << "- error\n";
366864ab3302SCarolineConcatto     } else if (const Symbol * symbol{actual->GetAssumedTypeDummy()}) {
366964ab3302SCarolineConcatto       os << "- assumed type: " << symbol->name().ToString() << '\n';
367064ab3302SCarolineConcatto     } else if (const Expr<SomeType> *expr{actual->UnwrapExpr()}) {
367164ab3302SCarolineConcatto       expr->AsFortran(os << "- expr: ") << '\n';
367264ab3302SCarolineConcatto     } else {
367364ab3302SCarolineConcatto       DIE("bad ActualArgument");
367464ab3302SCarolineConcatto     }
367564ab3302SCarolineConcatto   }
367664ab3302SCarolineConcatto }
367788afb6e8Speter klausler 
AnalyzeExpr(const parser::Expr & expr)367864ab3302SCarolineConcatto std::optional<ActualArgument> ArgumentAnalyzer::AnalyzeExpr(
367964ab3302SCarolineConcatto     const parser::Expr &expr) {
368064ab3302SCarolineConcatto   source_.ExtendToCover(expr.source);
368164ab3302SCarolineConcatto   if (const Symbol * assumedTypeDummy{AssumedTypeDummy(expr)}) {
36828a8bb078Speter klausler     expr.typedExpr.Reset(new GenericExprWrapper{}, GenericExprWrapper::Deleter);
3683a3538b83STim Keith     if (isProcedureCall_) {
3684fa0443f2SPeter Klausler       ActualArgument arg{ActualArgument::AssumedType{*assumedTypeDummy}};
3685fa0443f2SPeter Klausler       SetArgSourceLocation(arg, expr.source);
3686fa0443f2SPeter Klausler       return std::move(arg);
368788afb6e8Speter klausler     }
368864ab3302SCarolineConcatto     context_.SayAt(expr.source,
368964ab3302SCarolineConcatto         "TYPE(*) dummy argument may only be used as an actual argument"_err_en_US);
369088afb6e8Speter klausler   } else if (MaybeExpr argExpr{AnalyzeExprOrWholeAssumedSizeArray(expr)}) {
369188afb6e8Speter klausler     if (isProcedureCall_ || !IsProcedure(*argExpr)) {
3692fa0443f2SPeter Klausler       ActualArgument arg{std::move(*argExpr)};
3693fa0443f2SPeter Klausler       SetArgSourceLocation(arg, expr.source);
3694fa0443f2SPeter Klausler       return std::move(arg);
369588afb6e8Speter klausler     }
369688afb6e8Speter klausler     context_.SayAt(expr.source,
369788afb6e8Speter klausler         IsFunction(*argExpr) ? "Function call must have argument list"_err_en_US
369888afb6e8Speter klausler                              : "Subroutine name is not allowed here"_err_en_US);
369988afb6e8Speter klausler   }
370064ab3302SCarolineConcatto   return std::nullopt;
370164ab3302SCarolineConcatto }
370288afb6e8Speter klausler 
AnalyzeExprOrWholeAssumedSizeArray(const parser::Expr & expr)370388afb6e8Speter klausler MaybeExpr ArgumentAnalyzer::AnalyzeExprOrWholeAssumedSizeArray(
370488afb6e8Speter klausler     const parser::Expr &expr) {
370588afb6e8Speter klausler   // If an expression's parse tree is a whole assumed-size array:
370688afb6e8Speter klausler   //   Expr -> Designator -> DataRef -> Name
370788afb6e8Speter klausler   // treat it as a special case for argument passing and bypass
370888afb6e8Speter klausler   // the C1002/C1014 constraint checking in expression semantics.
370988afb6e8Speter klausler   if (const auto *name{parser::Unwrap<parser::Name>(expr)}) {
371088afb6e8Speter klausler     if (name->symbol && semantics::IsAssumedSizeArray(*name->symbol)) {
371188afb6e8Speter klausler       auto restorer{context_.AllowWholeAssumedSizeArray()};
371288afb6e8Speter klausler       return context_.Analyze(expr);
371388afb6e8Speter klausler     }
371488afb6e8Speter klausler   }
371588afb6e8Speter klausler   return context_.Analyze(expr);
371664ab3302SCarolineConcatto }
371764ab3302SCarolineConcatto 
AreConformable() const371864ab3302SCarolineConcatto bool ArgumentAnalyzer::AreConformable() const {
3719d9195d66Speter klausler   CHECK(actuals_.size() == 2);
3720d9195d66Speter klausler   return actuals_[0] && actuals_[1] &&
3721d9195d66Speter klausler       evaluate::AreConformable(*actuals_[0], *actuals_[1]);
372264ab3302SCarolineConcatto }
372364ab3302SCarolineConcatto 
372464ab3302SCarolineConcatto // Look for a type-bound operator in the type of arg number passIndex.
FindBoundOp(parser::CharBlock oprName,int passIndex,const Symbol * & definedOp)372564ab3302SCarolineConcatto const Symbol *ArgumentAnalyzer::FindBoundOp(
3726d9195d66Speter klausler     parser::CharBlock oprName, int passIndex, const Symbol *&definedOp) {
372764ab3302SCarolineConcatto   const auto *type{GetDerivedTypeSpec(GetType(passIndex))};
372864ab3302SCarolineConcatto   if (!type || !type->scope()) {
372964ab3302SCarolineConcatto     return nullptr;
373064ab3302SCarolineConcatto   }
373164ab3302SCarolineConcatto   const Symbol *symbol{type->scope()->FindComponent(oprName)};
373264ab3302SCarolineConcatto   if (!symbol) {
373364ab3302SCarolineConcatto     return nullptr;
373464ab3302SCarolineConcatto   }
3735d9195d66Speter klausler   definedOp = symbol;
373664ab3302SCarolineConcatto   ExpressionAnalyzer::AdjustActuals adjustment{
373764ab3302SCarolineConcatto       [&](const Symbol &proc, ActualArguments &) {
373864ab3302SCarolineConcatto         return passIndex == GetPassIndex(proc);
373964ab3302SCarolineConcatto       }};
37400c190575SPeter Klausler   auto pair{context_.ResolveGeneric(*symbol, actuals_, adjustment, false)};
37410fcda9aeSpeter klausler   if (!pair.first) {
37420fcda9aeSpeter klausler     context_.EmitGenericResolutionError(*symbol, pair.second);
374364ab3302SCarolineConcatto   }
37440fcda9aeSpeter klausler   return pair.first;
374564ab3302SCarolineConcatto }
374664ab3302SCarolineConcatto 
3747b05c8c57STim Keith // If there is an implicit conversion between intrinsic types, make it explicit
AddAssignmentConversion(const DynamicType & lhsType,const DynamicType & rhsType)3748b05c8c57STim Keith void ArgumentAnalyzer::AddAssignmentConversion(
3749b05c8c57STim Keith     const DynamicType &lhsType, const DynamicType &rhsType) {
3750b05c8c57STim Keith   if (lhsType.category() == rhsType.category() &&
37510363a164SPeter Klausler       (lhsType.category() == TypeCategory::Derived ||
37520363a164SPeter Klausler           lhsType.kind() == rhsType.kind())) {
3753b05c8c57STim Keith     // no conversion necessary
3754b05c8c57STim Keith   } else if (auto rhsExpr{evaluate::ConvertToType(lhsType, MoveExpr(1))}) {
3755fa0443f2SPeter Klausler     std::optional<parser::CharBlock> source;
3756fa0443f2SPeter Klausler     if (actuals_[1]) {
3757fa0443f2SPeter Klausler       source = actuals_[1]->sourceLocation();
3758fa0443f2SPeter Klausler     }
3759b05c8c57STim Keith     actuals_[1] = ActualArgument{*rhsExpr};
3760fa0443f2SPeter Klausler     SetArgSourceLocation(actuals_[1], source);
3761b05c8c57STim Keith   } else {
3762b05c8c57STim Keith     actuals_[1] = std::nullopt;
3763b05c8c57STim Keith   }
3764b05c8c57STim Keith }
3765b05c8c57STim Keith 
GetType(std::size_t i) const376664ab3302SCarolineConcatto std::optional<DynamicType> ArgumentAnalyzer::GetType(std::size_t i) const {
376764ab3302SCarolineConcatto   return i < actuals_.size() ? actuals_[i].value().GetType() : std::nullopt;
376864ab3302SCarolineConcatto }
GetRank(std::size_t i) const376964ab3302SCarolineConcatto int ArgumentAnalyzer::GetRank(std::size_t i) const {
377064ab3302SCarolineConcatto   return i < actuals_.size() ? actuals_[i].value().Rank() : 0;
377164ab3302SCarolineConcatto }
377264ab3302SCarolineConcatto 
377383dca19cSPete Steinfeld // If the argument at index i is a BOZ literal, convert its type to match the
3774e7a53f1eSpeter klausler // otherType.  If it's REAL convert to REAL, otherwise convert to INTEGER.
377583dca19cSPete Steinfeld // Note that IBM supports comparing BOZ literals to CHARACTER operands.  That
377683dca19cSPete Steinfeld // is not currently supported.
ConvertBOZ(std::optional<DynamicType> & thisType,std::size_t i,std::optional<DynamicType> otherType)3777d9195d66Speter klausler void ArgumentAnalyzer::ConvertBOZ(std::optional<DynamicType> &thisType,
377883dca19cSPete Steinfeld     std::size_t i, std::optional<DynamicType> otherType) {
377983dca19cSPete Steinfeld   if (IsBOZLiteral(i)) {
378083dca19cSPete Steinfeld     Expr<SomeType> &&argExpr{MoveExpr(i)};
378183dca19cSPete Steinfeld     auto *boz{std::get_if<BOZLiteralConstant>(&argExpr.u)};
378283dca19cSPete Steinfeld     if (otherType && otherType->category() == TypeCategory::Real) {
3783d9195d66Speter klausler       int kind{context_.context().GetDefaultKind(TypeCategory::Real)};
3784d9195d66Speter klausler       MaybeExpr realExpr{
3785d9195d66Speter klausler           ConvertToKind<TypeCategory::Real>(kind, std::move(*boz))};
378683dca19cSPete Steinfeld       actuals_[i] = std::move(*realExpr);
3787d9195d66Speter klausler       thisType.emplace(TypeCategory::Real, kind);
378883dca19cSPete Steinfeld     } else {
3789d9195d66Speter klausler       int kind{context_.context().GetDefaultKind(TypeCategory::Integer)};
3790d9195d66Speter klausler       MaybeExpr intExpr{
3791d9195d66Speter klausler           ConvertToKind<TypeCategory::Integer>(kind, std::move(*boz))};
379283dca19cSPete Steinfeld       actuals_[i] = std::move(*intExpr);
3793d9195d66Speter klausler       thisType.emplace(TypeCategory::Integer, kind);
379483dca19cSPete Steinfeld     }
379583dca19cSPete Steinfeld   }
379683dca19cSPete Steinfeld }
379783dca19cSPete Steinfeld 
379864ab3302SCarolineConcatto // Report error resolving opr when there is a user-defined one available
SayNoMatch(const std::string & opr,bool isAssignment)379964ab3302SCarolineConcatto void ArgumentAnalyzer::SayNoMatch(const std::string &opr, bool isAssignment) {
380064ab3302SCarolineConcatto   std::string type0{TypeAsFortran(0)};
380164ab3302SCarolineConcatto   auto rank0{actuals_[0]->Rank()};
380264ab3302SCarolineConcatto   if (actuals_.size() == 1) {
380364ab3302SCarolineConcatto     if (rank0 > 0) {
380464ab3302SCarolineConcatto       context_.Say("No intrinsic or user-defined %s matches "
380564ab3302SCarolineConcatto                    "rank %d array of %s"_err_en_US,
380664ab3302SCarolineConcatto           opr, rank0, type0);
380764ab3302SCarolineConcatto     } else {
380864ab3302SCarolineConcatto       context_.Say("No intrinsic or user-defined %s matches "
380964ab3302SCarolineConcatto                    "operand type %s"_err_en_US,
381064ab3302SCarolineConcatto           opr, type0);
381164ab3302SCarolineConcatto     }
381264ab3302SCarolineConcatto   } else {
381364ab3302SCarolineConcatto     std::string type1{TypeAsFortran(1)};
381464ab3302SCarolineConcatto     auto rank1{actuals_[1]->Rank()};
381564ab3302SCarolineConcatto     if (rank0 > 0 && rank1 > 0 && rank0 != rank1) {
381664ab3302SCarolineConcatto       context_.Say("No intrinsic or user-defined %s matches "
381764ab3302SCarolineConcatto                    "rank %d array of %s and rank %d array of %s"_err_en_US,
381864ab3302SCarolineConcatto           opr, rank0, type0, rank1, type1);
381964ab3302SCarolineConcatto     } else if (isAssignment && rank0 != rank1) {
382064ab3302SCarolineConcatto       if (rank0 == 0) {
382164ab3302SCarolineConcatto         context_.Say("No intrinsic or user-defined %s matches "
382264ab3302SCarolineConcatto                      "scalar %s and rank %d array of %s"_err_en_US,
382364ab3302SCarolineConcatto             opr, type0, rank1, type1);
382464ab3302SCarolineConcatto       } else {
382564ab3302SCarolineConcatto         context_.Say("No intrinsic or user-defined %s matches "
382664ab3302SCarolineConcatto                      "rank %d array of %s and scalar %s"_err_en_US,
382764ab3302SCarolineConcatto             opr, rank0, type0, type1);
382864ab3302SCarolineConcatto       }
382964ab3302SCarolineConcatto     } else {
383064ab3302SCarolineConcatto       context_.Say("No intrinsic or user-defined %s matches "
383164ab3302SCarolineConcatto                    "operand types %s and %s"_err_en_US,
383264ab3302SCarolineConcatto           opr, type0, type1);
383364ab3302SCarolineConcatto     }
383464ab3302SCarolineConcatto   }
383564ab3302SCarolineConcatto }
383664ab3302SCarolineConcatto 
TypeAsFortran(std::size_t i)383764ab3302SCarolineConcatto std::string ArgumentAnalyzer::TypeAsFortran(std::size_t i) {
38381bd083b5Speter klausler   if (i >= actuals_.size() || !actuals_[i]) {
38391bd083b5Speter klausler     return "missing argument";
38401bd083b5Speter klausler   } else if (std::optional<DynamicType> type{GetType(i)}) {
38410363a164SPeter Klausler     return type->IsAssumedType()         ? "TYPE(*)"s
38420363a164SPeter Klausler         : type->IsUnlimitedPolymorphic() ? "CLASS(*)"s
38430363a164SPeter Klausler         : type->IsPolymorphic()          ? "CLASS("s + type->AsFortran() + ')'
38440363a164SPeter Klausler         : type->category() == TypeCategory::Derived
384564ab3302SCarolineConcatto         ? "TYPE("s + type->AsFortran() + ')'
384664ab3302SCarolineConcatto         : type->category() == TypeCategory::Character
384764ab3302SCarolineConcatto         ? "CHARACTER(KIND="s + std::to_string(type->kind()) + ')'
384864ab3302SCarolineConcatto         : ToUpperCase(type->AsFortran());
384964ab3302SCarolineConcatto   } else {
385064ab3302SCarolineConcatto     return "untyped";
385164ab3302SCarolineConcatto   }
385264ab3302SCarolineConcatto }
385364ab3302SCarolineConcatto 
AnyUntypedOrMissingOperand()38541bd083b5Speter klausler bool ArgumentAnalyzer::AnyUntypedOrMissingOperand() {
385564ab3302SCarolineConcatto   for (const auto &actual : actuals_) {
38560fcda9aeSpeter klausler     if (!actual ||
38570fcda9aeSpeter klausler         (!actual->GetType() && !IsBareNullPointer(actual->UnwrapExpr()))) {
385864ab3302SCarolineConcatto       return true;
385964ab3302SCarolineConcatto     }
386064ab3302SCarolineConcatto   }
386164ab3302SCarolineConcatto   return false;
386264ab3302SCarolineConcatto }
386364ab3302SCarolineConcatto } // namespace Fortran::evaluate
386464ab3302SCarolineConcatto 
386564ab3302SCarolineConcatto namespace Fortran::semantics {
AnalyzeKindSelector(SemanticsContext & context,common::TypeCategory category,const std::optional<parser::KindSelector> & selector)386664ab3302SCarolineConcatto evaluate::Expr<evaluate::SubscriptInteger> AnalyzeKindSelector(
386764ab3302SCarolineConcatto     SemanticsContext &context, common::TypeCategory category,
386864ab3302SCarolineConcatto     const std::optional<parser::KindSelector> &selector) {
386964ab3302SCarolineConcatto   evaluate::ExpressionAnalyzer analyzer{context};
387064ab3302SCarolineConcatto   auto restorer{
387164ab3302SCarolineConcatto       analyzer.GetContextualMessages().SetLocation(context.location().value())};
387264ab3302SCarolineConcatto   return analyzer.AnalyzeKindSelector(category, selector);
387364ab3302SCarolineConcatto }
387464ab3302SCarolineConcatto 
ExprChecker(SemanticsContext & context)387564ab3302SCarolineConcatto ExprChecker::ExprChecker(SemanticsContext &context) : context_{context} {}
387664ab3302SCarolineConcatto 
Pre(const parser::DataImpliedDo & ido)38774171f80dSpeter klausler bool ExprChecker::Pre(const parser::DataImpliedDo &ido) {
38784171f80dSpeter klausler   parser::Walk(std::get<parser::DataImpliedDo::Bounds>(ido.t), *this);
38794171f80dSpeter klausler   const auto &bounds{std::get<parser::DataImpliedDo::Bounds>(ido.t)};
38804171f80dSpeter klausler   auto name{bounds.name.thing.thing};
38814171f80dSpeter klausler   int kind{evaluate::ResultType<evaluate::ImpliedDoIndex>::kind};
38824171f80dSpeter klausler   if (const auto dynamicType{evaluate::DynamicType::From(*name.symbol)}) {
38834171f80dSpeter klausler     if (dynamicType->category() == TypeCategory::Integer) {
38844171f80dSpeter klausler       kind = dynamicType->kind();
38854171f80dSpeter klausler     }
38864171f80dSpeter klausler   }
38874171f80dSpeter klausler   exprAnalyzer_.AddImpliedDo(name.source, kind);
38884171f80dSpeter klausler   parser::Walk(std::get<std::list<parser::DataIDoObject>>(ido.t), *this);
38894171f80dSpeter klausler   exprAnalyzer_.RemoveImpliedDo(name.source);
38904171f80dSpeter klausler   return false;
38914171f80dSpeter klausler }
38924171f80dSpeter klausler 
Walk(const parser::Program & program)389364ab3302SCarolineConcatto bool ExprChecker::Walk(const parser::Program &program) {
389464ab3302SCarolineConcatto   parser::Walk(program, *this);
389564ab3302SCarolineConcatto   return !context_.AnyFatalError();
389664ab3302SCarolineConcatto }
389784a099dfSpeter klausler } // namespace Fortran::semantics
3898