llvm/flang/lib/parser/user-state.h
2018-03-30 16:21:12 -07:00

54 lines
1.5 KiB
C++

#ifndef FORTRAN_PARSER_USER_STATE_H_
#define FORTRAN_PARSER_USER_STATE_H_
// Instances of ParseState (parse-state.h) incorporate instances of this
// UserState class, which encapsulates any semantic information necessary for
// parse tree construction so as to avoid any need for representing
// state in static data.
#include "char-block.h"
#include <cinttypes>
#include <set>
#include <unordered_set>
namespace Fortran {
namespace parser {
class UserState {
public:
void NewSubprogram() {
doLabels_.clear();
nonlabelDoConstructNestingDepth_ = 0;
oldStructureComponents_.clear();
}
using Label = std::uint64_t;
bool IsDoLabel(Label label) const {
return doLabels_.find(label) != doLabels_.end();
}
bool InNonlabelDoConstruct() const {
return nonlabelDoConstructNestingDepth_ > 0;
}
void NewDoLabel(Label label) { doLabels_.insert(label); }
void EnterNonlabelDoConstruct() { ++nonlabelDoConstructNestingDepth_; }
void LeaveDoConstruct() {
if (nonlabelDoConstructNestingDepth_ > 0) {
--nonlabelDoConstructNestingDepth_;
}
}
void NoteOldStructureComponent(const CharBlock &name) {
oldStructureComponents_.insert(name);
}
bool IsOldStructureComponent(const CharBlock &name) const {
return oldStructureComponents_.find(name) != oldStructureComponents_.end();
}
private:
std::unordered_set<Label> doLabels_;
int nonlabelDoConstructNestingDepth_{0};
std::set<CharBlock> oldStructureComponents_;
};
} // namespace parser
} // namespace Fortran
#endif // FORTRAN_PARSER_USER_STATE_H_