llvm/flang/lib/parser/instrumented-parser.h
2018-04-19 15:46:02 -07:00

71 lines
2 KiB
C++

#ifndef FORTRAN_PARSER_INSTRUMENTED_PARSER_H_
#define FORTRAN_PARSER_INSTRUMENTED_PARSER_H_
#include "message.h"
#include "parse-state.h"
#include "provenance.h"
#include "user-state.h"
#include <cstddef>
#include <map>
#include <ostream>
namespace Fortran {
namespace parser {
class ParsingLog {
public:
bool Fails(const char *at, const MessageFixedText &tag, Messages &);
void Note(
const char *at, const MessageFixedText &tag, bool pass, const Messages &);
void Dump(std::ostream &, const CookedSource &) const;
private:
struct LogForPosition {
struct Entry {
Entry() {}
bool pass{true};
int count{0};
Messages messages;
};
std::map<MessageFixedText, Entry> perTag;
};
std::map<std::size_t, LogForPosition> perPos_;
};
template<typename PA> class InstrumentedParser {
public:
using resultType = typename PA::resultType;
constexpr InstrumentedParser(const InstrumentedParser &) = default;
constexpr InstrumentedParser(const MessageFixedText &tag, const PA &parser)
: tag_{tag}, parser_{parser} {}
std::optional<resultType> Parse(ParseState *state) const {
if (UserState * ustate{state->userState()}) {
if (ParsingLog * log{ustate->log()}) {
const char *at{state->GetLocation()};
if (log->Fails(at, tag_, state->messages())) {
return {};
}
Messages messages{std::move(state->messages())};
std::optional<resultType> result{parser_.Parse(state)};
log->Note(at, tag_, result.has_value(), state->messages());
messages.Annex(state->messages());
state->messages() = std::move(messages);
return result;
}
}
return parser_.Parse(state);
}
private:
const MessageFixedText tag_;
const PA parser_;
};
template<typename PA>
inline constexpr auto instrumented(
const MessageFixedText &tag, const PA &parser) {
return InstrumentedParser{tag, parser};
}
} // namespace parser
} // namespace Fortran
#endif // FORTRAN_PARSER_INSTRUMENTED_PARSER_H_