12#ifndef FORTRAN_SEMANTICS_CHECK_DIRECTIVE_STRUCTURE_H_
13#define FORTRAN_SEMANTICS_CHECK_DIRECTIVE_STRUCTURE_H_
15#include "flang/Common/enum-set.h"
16#include "flang/Semantics/semantics.h"
17#include "flang/Semantics/tools.h"
18#include "llvm/ADT/iterator_range.h"
21#include <unordered_map>
23namespace Fortran::semantics {
35template <
typename D>
class NoBranchingEnforce {
39 std::string &&upperCaseDirName)
40 : context_{context}, sourcePosition_{sourcePosition},
41 upperCaseDirName_{std::move(upperCaseDirName)},
42 currentDirective_{directive}, numDoConstruct_{0} {}
43 template <
typename T>
bool Pre(
const T &) {
return true; }
44 template <
typename T>
void Post(
const T &) {}
47 currentStatementSourcePosition_ = statement.source;
56 void Post(
const parser::ReturnStmt &) { EmitBranchOutError(
"RETURN"); }
57 void Post(
const parser::GotoStmt &gotoStmt) {
58 if constexpr (std::is_same_v<D, llvm::acc::Directive>) {
59 switch ((llvm::acc::Directive)currentDirective_) {
60 case llvm::acc::Directive::ACCD_parallel:
61 case llvm::acc::Directive::ACCD_serial:
62 case llvm::acc::Directive::ACCD_kernels:
63 if (labelsInBlock_.count(gotoStmt.v) == 0)
64 EmitBranchOutOfComputeConstructError(
"GOTO");
71 void CollectLabel(parser::Label label) { labelsInBlock_.insert(label); }
72 void Post(
const parser::ExitStmt &exitStmt) {
73 if (
const auto &exitName{exitStmt.v}) {
74 CheckConstructNameBranching(
"EXIT", exitName.value());
76 CheckConstructNameBranching(
"EXIT");
79 void Post(
const parser::CycleStmt &cycleStmt) {
80 if (
const auto &cycleName{cycleStmt.v}) {
81 CheckConstructNameBranching(
"CYCLE", cycleName.value());
83 if constexpr (std::is_same_v<D, llvm::omp::Directive>) {
84 switch ((llvm::omp::Directive)currentDirective_) {
86 case llvm::omp::Directive::OMPD_do:
87 case llvm::omp::Directive::OMPD_simd:
88 case llvm::omp::Directive::OMPD_parallel_do:
89 case llvm::omp::Directive::OMPD_parallel_do_simd:
90 case llvm::omp::Directive::OMPD_distribute_parallel_do:
91 case llvm::omp::Directive::OMPD_distribute_parallel_do_simd:
92 case llvm::omp::Directive::OMPD_distribute_parallel_for:
93 case llvm::omp::Directive::OMPD_distribute_simd:
94 case llvm::omp::Directive::OMPD_distribute_parallel_for_simd:
95 case llvm::omp::Directive::OMPD_target_teams_distribute:
96 case llvm::omp::Directive::OMPD_target_teams_distribute_simd:
97 case llvm::omp::Directive::OMPD_target_teams_distribute_parallel_do:
98 case llvm::omp::Directive::
99 OMPD_target_teams_distribute_parallel_do_simd:
104 }
else if constexpr (std::is_same_v<D, llvm::acc::Directive>) {
105 switch ((llvm::acc::Directive)currentDirective_) {
108 case llvm::acc::Directive::ACCD_loop:
109 case llvm::acc::Directive::ACCD_kernels_loop:
110 case llvm::acc::Directive::ACCD_parallel_loop:
111 case llvm::acc::Directive::ACCD_serial_loop:
117 CheckConstructNameBranching(
"CYCLE");
123 return {
"Enclosing %s construct"_en_US, upperCaseDirName_};
126 void EmitBranchOutError(
const char *stmt)
const {
128 .Say(currentStatementSourcePosition_,
129 "%s statement is not allowed in a %s construct"_err_en_US, stmt,
131 .Attach(sourcePosition_, GetEnclosingMsg());
134 void EmitBranchOutOfComputeConstructError(
const char *stmt)
const {
136 .Say(currentStatementSourcePosition_,
137 "%s to a label outside of a %s construct is not allowed"_err_en_US,
138 stmt, upperCaseDirName_)
139 .Attach(sourcePosition_, GetEnclosingMsg());
142 inline void EmitUnlabelledBranchOutError(
const char *stmt) {
144 .Say(currentStatementSourcePosition_,
145 "%s to construct outside of %s construct is not allowed"_err_en_US,
146 stmt, upperCaseDirName_)
147 .Attach(sourcePosition_, GetEnclosingMsg());
150 void EmitBranchOutErrorWithName(
152 const std::string branchingToName{toName.ToString()};
154 .Say(currentStatementSourcePosition_,
155 "%s to construct '%s' outside of %s construct is not allowed"_err_en_US,
156 stmt, branchingToName, upperCaseDirName_)
157 .Attach(sourcePosition_, GetEnclosingMsg());
166 void CheckConstructNameBranching(
168 const ConstructStack &stack{context_.constructStack()};
169 for (
auto iter{stack.cend()}; iter-- != stack.cbegin();) {
170 const ConstructNode &construct{*iter};
171 const auto &constructName{MaybeGetNodeName(construct)};
173 if (stmtName.source == constructName->source) {
174 EmitBranchOutErrorWithName(stmt, stmtName);
182 void CheckConstructNameBranching(
const char *stmt) {
184 if (numDoConstruct_ > 0) {
189 EmitUnlabelledBranchOutError(stmt);
195 std::string upperCaseDirName_;
199 std::set<parser::Label> labelsInBlock_;
207template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
212 &directiveClausesMap)
213 : context_{context}, directiveClausesMap_(directiveClausesMap) {}
214 virtual ~DirectiveStructureChecker() {}
216 using ClauseMapTy = std::multimap<C, const PC *>;
217 struct DirectiveContext {
219 : directiveSource{source}, directive{d} {}
229 const PC *clause{
nullptr};
230 ClauseMapTy clauseInfo;
231 std::list<C> actualClauses;
232 std::list<C> endDirectiveClauses;
233 std::list<C> crtGroup;
238 CHECK(!dirContext_.empty());
239 return dirContext_.back();
242 DirectiveContext &GetContextParent() {
243 CHECK(dirContext_.size() >= 2);
244 return dirContext_[dirContext_.size() - 2];
247 void SetContextClause(
const PC &clause) {
248 GetContext().clauseSource = clause.source;
249 GetContext().clause = &clause;
252 void ResetPartialContext(
const parser::CharBlock &source) {
253 CHECK(!dirContext_.empty());
254 SetContextDirectiveSource(source);
255 GetContext().allowedClauses = {};
256 GetContext().allowedOnceClauses = {};
257 GetContext().allowedExclusiveClauses = {};
258 GetContext().requiredClauses = {};
259 GetContext().clauseInfo = {};
262 void SetContextDirectiveSource(
const parser::CharBlock &directive) {
263 GetContext().directiveSource = directive;
266 void SetContextDirectiveEnum(D dir) { GetContext().directive = dir; }
268 void SetContextAllowed(
const common::EnumSet<C, ClauseEnumSize> &allowed) {
269 GetContext().allowedClauses = allowed;
272 void SetContextAllowedOnce(
273 const common::EnumSet<C, ClauseEnumSize> &allowedOnce) {
274 GetContext().allowedOnceClauses = allowedOnce;
277 void SetContextAllowedExclusive(
278 const common::EnumSet<C, ClauseEnumSize> &allowedExclusive) {
279 GetContext().allowedExclusiveClauses = allowedExclusive;
282 void SetContextRequired(
const common::EnumSet<C, ClauseEnumSize> &required) {
283 GetContext().requiredClauses = required;
286 void SetContextClauseInfo(C type) {
287 GetContext().clauseInfo.emplace(type, GetContext().clause);
290 void AddClauseToCrtContext(C type) {
291 GetContext().actualClauses.push_back(type);
294 void AddClauseToCrtGroupInContext(C type) {
295 GetContext().crtGroup.push_back(type);
298 void ResetCrtGroup() { GetContext().crtGroup.clear(); }
301 const PC *FindClause(C type) {
return FindClause(GetContext(), type); }
305 auto it{context.clauseInfo.find(type)};
306 if (it != context.clauseInfo.end()) {
313 const PC *FindClauseParent(C type) {
314 auto it{GetContextParent().clauseInfo.find(type)};
315 if (it != GetContextParent().clauseInfo.end()) {
321 llvm::iterator_range<typename ClauseMapTy::iterator> FindClauses(C type) {
322 auto it{GetContext().clauseInfo.equal_range(type)};
323 return llvm::make_range(it);
327 CHECK(!dirContext_.empty());
328 auto it{dirContext_.rbegin()};
329 if (++it != dirContext_.rend()) {
335 void PushContext(
const parser::CharBlock &source, D dir) {
336 dirContext_.emplace_back(source, dir);
340 CHECK(!dirContext_.empty());
341 auto it{dirContext_.rbegin()};
342 while (++it != dirContext_.rend()) {
343 if (it->directive == dir) {
350 bool CurrentDirectiveIsNested() {
return dirContext_.size() > 1; };
352 void SetClauseSets(D dir) {
353 dirContext_.back().allowedClauses = directiveClausesMap_[dir].allowed;
354 dirContext_.back().allowedOnceClauses =
355 directiveClausesMap_[dir].allowedOnce;
356 dirContext_.back().allowedExclusiveClauses =
357 directiveClausesMap_[dir].allowedExclusive;
358 dirContext_.back().requiredClauses =
359 directiveClausesMap_[dir].requiredOneOf;
361 void PushContextAndClauseSets(
const parser::CharBlock &source, D dir) {
362 PushContext(source, dir);
366 void SayNotMatching(
const parser::CharBlock &,
const parser::CharBlock &);
368 template <
typename B>
void CheckMatching(
const B &beginDir,
const B &endDir) {
369 const auto &begin{beginDir.v};
370 const auto &end{endDir.v};
372 SayNotMatching(beginDir.source, endDir.source);
377 void CheckNoBranching(
const parser::Block &block, D directive,
378 const parser::CharBlock &directiveSource);
381 void CheckOnlyAllowedAfter(C clause, common::EnumSet<C, ClauseEnumSize> set);
383 void CheckRequireAtLeastOneOf(
bool warnInsteadOfError =
false);
387 bool CheckAllowed(C clause,
bool warnInsteadOfError =
false);
391 void CheckAllowedOncePerGroup(C clause, C separator);
393 void CheckMutuallyExclusivePerGroup(
394 C clause, C separator, common::EnumSet<C, ClauseEnumSize> set);
396 void CheckAtLeastOneClause();
398 void CheckNotAllowedIfClause(
399 C clause, common::EnumSet<C, ClauseEnumSize> set);
401 std::string ContextDirectiveAsFortran();
403 void RequiresConstantPositiveParameter(
404 const C &clause,
const parser::ScalarIntConstantExpr &i);
406 void RequiresPositiveParameter(
const C &clause,
407 const parser::ScalarIntExpr &i, llvm::StringRef paramName =
"parameter",
408 bool allowZero =
true);
410 void OptionalConstantPositiveParameter(
411 const C &clause,
const std::optional<parser::ScalarIntConstantExpr> &o);
413 virtual llvm::StringRef getClauseName(C clause) {
return ""; };
415 virtual llvm::StringRef getDirectiveName(D directive) {
return ""; };
417 SemanticsContext &context_;
418 std::vector<DirectiveContext> dirContext_;
419 std::unordered_map<D, DirectiveClauses<C, ClauseEnumSize>>
420 directiveClausesMap_;
422 std::string ClauseSetToString(
const common::EnumSet<C, ClauseEnumSize> set);
427 std::set<parser::Label> labels;
428 template <
typename T>
bool Pre(
const T &) {
return true; }
429 template <
typename T>
void Post(
const T &) {}
432 labels.insert(*stmt.label);
437template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
438void DirectiveStructureChecker<D, C, PC, ClauseEnumSize>::CheckNoBranching(
439 const parser::Block &block, D directive,
442 parser::Walk(block, labelCollector);
444 context_, directiveSource, directive, ContextDirectiveAsFortran()};
445 for (
auto label : labelCollector.labels)
446 noBranchingEnforce.CollectLabel(label);
447 parser::Walk(block, noBranchingEnforce);
452template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
453void DirectiveStructureChecker<D, C, PC, ClauseEnumSize>::CheckOnlyAllowedAfter(
454 C clause, common::EnumSet<C, ClauseEnumSize> set) {
455 bool enforceCheck =
false;
456 for (
auto cl : GetContext().actualClauses) {
460 }
else if (enforceCheck && !set.test(cl)) {
461 auto parserClause = GetContext().clauseInfo.find(cl);
462 context_.Say(parserClause->second->source,
463 "Clause %s is not allowed after clause %s on the %s "
464 "directive"_err_en_US,
465 parser::ToUpperCaseLetters(getClauseName(cl).str()),
466 parser::ToUpperCaseLetters(getClauseName(clause).str()),
467 ContextDirectiveAsFortran());
473template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
475 ClauseEnumSize>::CheckAtLeastOneClause() {
476 if (GetContext().actualClauses.empty()) {
477 context_.Say(GetContext().directiveSource,
478 "At least one clause is required on the %s directive"_err_en_US,
479 ContextDirectiveAsFortran());
483template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
485DirectiveStructureChecker<D, C, PC, ClauseEnumSize>::ClauseSetToString(
486 const common::EnumSet<C, ClauseEnumSize> set) {
488 set.IterateOverMembers([&](C o) {
491 list.append(parser::ToUpperCaseLetters(getClauseName(o).str()));
498template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
500 ClauseEnumSize>::CheckRequireAtLeastOneOf(
bool warnInsteadOfError) {
501 if (GetContext().requiredClauses.empty()) {
504 for (
auto cl : GetContext().actualClauses) {
505 if (GetContext().requiredClauses.test(cl)) {
510 if (warnInsteadOfError) {
511 context_.Warn(common::UsageWarning::Portability,
512 GetContext().directiveSource,
513 "At least one of %s clause should appear on the %s directive"_port_en_US,
514 ClauseSetToString(GetContext().requiredClauses),
515 ContextDirectiveAsFortran());
517 context_.Say(GetContext().directiveSource,
518 "At least one of %s clause must appear on the %s directive"_err_en_US,
519 ClauseSetToString(GetContext().requiredClauses),
520 ContextDirectiveAsFortran());
524template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
526 ClauseEnumSize>::ContextDirectiveAsFortran() {
527 return parser::ToUpperCaseLetters(
528 getDirectiveName(GetContext().directive).str());
532template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
533bool DirectiveStructureChecker<D, C, PC, ClauseEnumSize>::CheckAllowed(
534 C clause,
bool warnInsteadOfError) {
535 if (!GetContext().allowedClauses.test(clause) &&
536 !GetContext().allowedOnceClauses.test(clause) &&
537 !GetContext().allowedExclusiveClauses.test(clause) &&
538 !GetContext().requiredClauses.test(clause)) {
539 if (warnInsteadOfError) {
540 context_.Warn(common::UsageWarning::Portability,
541 GetContext().clauseSource,
542 "%s clause is not allowed on the %s directive and will be ignored"_port_en_US,
543 parser::ToUpperCaseLetters(getClauseName(clause).str()),
544 parser::ToUpperCaseLetters(GetContext().directiveSource.ToString()));
546 context_.Say(GetContext().clauseSource,
547 "%s clause is not allowed on the %s directive"_err_en_US,
548 parser::ToUpperCaseLetters(getClauseName(clause).str()),
549 parser::ToUpperCaseLetters(GetContext().directiveSource.ToString()));
553 if ((GetContext().allowedOnceClauses.test(clause) ||
554 GetContext().allowedExclusiveClauses.test(clause)) &&
555 FindClause(clause)) {
556 context_.Say(GetContext().clauseSource,
557 "At most one %s clause can appear on the %s directive"_err_en_US,
558 parser::ToUpperCaseLetters(getClauseName(clause).str()),
559 parser::ToUpperCaseLetters(GetContext().directiveSource.ToString()));
562 if (GetContext().allowedExclusiveClauses.test(clause)) {
563 std::vector<C> others;
564 GetContext().allowedExclusiveClauses.IterateOverMembers([&](C o) {
566 others.emplace_back(o);
569 for (
const auto &e : others) {
570 context_.Say(GetContext().clauseSource,
571 "%s and %s clauses are mutually exclusive and may not appear on the "
572 "same %s directive"_err_en_US,
573 parser::ToUpperCaseLetters(getClauseName(clause).str()),
574 parser::ToUpperCaseLetters(getClauseName(e).str()),
575 parser::ToUpperCaseLetters(GetContext().directiveSource.ToString()));
577 if (!others.empty()) {
581 SetContextClauseInfo(clause);
582 AddClauseToCrtContext(clause);
583 AddClauseToCrtGroupInContext(clause);
589template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
591 ClauseEnumSize>::CheckNotAllowedIfClause(C clause,
592 common::EnumSet<C, ClauseEnumSize> set) {
593 if (!llvm::is_contained(GetContext().actualClauses, clause)) {
597 for (
auto cl : GetContext().actualClauses) {
599 context_.Say(GetContext().directiveSource,
600 "Clause %s is not allowed if clause %s appears on the %s directive"_err_en_US,
601 parser::ToUpperCaseLetters(getClauseName(cl).str()),
602 parser::ToUpperCaseLetters(getClauseName(clause).str()),
603 ContextDirectiveAsFortran());
608template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
610 ClauseEnumSize>::CheckAllowedOncePerGroup(C clause, C separator) {
611 bool clauseIsPresent =
false;
612 for (
auto cl : GetContext().actualClauses) {
614 if (clauseIsPresent) {
615 context_.Say(GetContext().clauseSource,
616 "At most one %s clause can appear on the %s directive or in group separated by the %s clause"_err_en_US,
617 parser::ToUpperCaseLetters(getClauseName(clause).str()),
618 parser::ToUpperCaseLetters(GetContext().directiveSource.ToString()),
619 parser::ToUpperCaseLetters(getClauseName(separator).str()));
621 clauseIsPresent =
true;
625 clauseIsPresent =
false;
629template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
631 ClauseEnumSize>::CheckMutuallyExclusivePerGroup(C clause, C separator,
632 common::EnumSet<C, ClauseEnumSize> set) {
635 for (
auto cl : GetContext().actualClauses) {
636 if (cl == separator) {
640 context_.Say(GetContext().directiveSource,
641 "Clause %s is not allowed if clause %s appears on the %s directive"_err_en_US,
642 parser::ToUpperCaseLetters(getClauseName(clause).str()),
643 parser::ToUpperCaseLetters(getClauseName(cl).str()),
644 ContextDirectiveAsFortran());
649 for (
auto cl : GetContext().crtGroup) {
651 context_.Say(GetContext().directiveSource,
652 "Clause %s is not allowed if clause %s appears on the %s directive"_err_en_US,
653 parser::ToUpperCaseLetters(getClauseName(clause).str()),
654 parser::ToUpperCaseLetters(getClauseName(cl).str()),
655 ContextDirectiveAsFortran());
661template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
663 ClauseEnumSize>::RequiresConstantPositiveParameter(
const C &clause,
664 const parser::ScalarIntConstantExpr &i) {
665 if (
const auto v{GetIntValue(i)}) {
667 context_.Say(GetContext().clauseSource,
668 "The parameter of the %s clause must be "
669 "a constant positive integer expression"_err_en_US,
670 parser::ToUpperCaseLetters(getClauseName(clause).str()));
676template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
678 ClauseEnumSize>::OptionalConstantPositiveParameter(
const C &clause,
679 const std::optional<parser::ScalarIntConstantExpr> &o) {
680 if (o != std::nullopt) {
681 RequiresConstantPositiveParameter(clause, o.value());
685template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
686void DirectiveStructureChecker<D, C, PC, ClauseEnumSize>::SayNotMatching(
687 const parser::CharBlock &beginSource,
const parser::CharBlock &endSource) {
689 .Say(endSource,
"Unmatched %s directive"_err_en_US,
690 parser::ToUpperCaseLetters(endSource.ToString()))
691 .Attach(beginSource,
"Does not match directive"_en_US);
695template <
typename D,
typename C,
typename PC, std::
size_t ClauseEnumSize>
697 ClauseEnumSize>::RequiresPositiveParameter(
const C &clause,
698 const parser::ScalarIntExpr &i, llvm::StringRef paramName,
bool allowZero) {
699 if (
const auto v{GetIntValue(i)}) {
700 if (*v < (allowZero ? 0 : 1)) {
701 context_.Say(GetContext().clauseSource,
702 "The %s of the %s clause must be "
703 "a positive integer expression"_err_en_US,
705 parser::ToUpperCaseLetters(getClauseName(clause).str()));
Definition char-block.h:28
Definition check-directive-structure.h:208
Definition check-directive-structure.h:35
Definition semantics.h:67
Definition parse-tree.h:2325
Definition parse-tree.h:589
Definition parse-tree.h:361
Definition semantics.h:427
Definition check-directive-structure.h:25
Definition check-directive-structure.h:217
Definition check-directive-structure.h:426