llvm::DenseMap<unsigned, DiagnosticMappingInfo> DiagMap;
public:
- typedef llvm::DenseMap<unsigned, DiagnosticMappingInfo>::const_iterator
+ typedef llvm::DenseMap<unsigned, DiagnosticMappingInfo>::iterator
iterator;
+ typedef llvm::DenseMap<unsigned, DiagnosticMappingInfo>::const_iterator
+ const_iterator;
void setMappingInfo(diag::kind Diag, DiagnosticMappingInfo Info) {
DiagMap[Diag] = Info;
}
DiagnosticMappingInfo getMappingInfo(diag::kind Diag) const {
- iterator I = DiagMap.find(Diag);
+ const_iterator I = DiagMap.find(Diag);
if (I != DiagMap.end())
return I->second;
return DiagnosticMappingInfo::MakeUnset();
}
- iterator begin() const { return DiagMap.begin(); }
- iterator end() const { return DiagMap.end(); }
+ DiagnosticMappingInfo &getOrAddMappingInfo(diag::kind Diag);
+
+ const_iterator begin() const { return DiagMap.begin(); }
+ const_iterator end() const { return DiagMap.end(); }
};
/// \brief Keeps and automatically disposes all DiagStates that we create.
};
}
+// Unfortunately, the split between DiagnosticIDs and Diagnostic is not
+// particularly clean, but for now we just implement this method here so we can
+// access GetDefaultDiagMapping.
+DiagnosticMappingInfo &DiagnosticsEngine::DiagState::getOrAddMappingInfo(
+ diag::kind Diag)
+{
+ std::pair<iterator, bool> Result = DiagMap.insert(
+ std::make_pair(Diag, DiagnosticMappingInfo::MakeUnset()));
+
+ // Initialize the entry if we added it.
+ if (Result.second) {
+ assert(Result.first->second.isUnset() && "unexpected unset entry");
+ Result.first->second = DiagnosticMappingInfo::MakeInfo(
+ GetDefaultDiagMapping(Diag), /*IsUser=*/false, /*IsPragma=*/false);
+ }
+
+ return Result.first->second;
+}
+
static const StaticDiagCategoryRec CategoryNameTable[] = {
#define GET_CATEGORY_TABLE
#define CATEGORY(X, ENUM) { X, STR_SIZE(X, uint8_t) },
Pos = Diag.GetDiagStatePointForLoc(Loc);
DiagnosticsEngine::DiagState *State = Pos->State;
- // Get the mapping information, if unset, compute it lazily.
- DiagnosticMappingInfo MappingInfo = State->getMappingInfo((diag::kind)DiagID);
- if (MappingInfo.isUnset()) {
- MappingInfo = DiagnosticMappingInfo::MakeInfo(
- GetDefaultDiagMapping(DiagID), /*IsUser=*/false, /*IsPragma=*/false);
- State->setMappingInfo((diag::kind) DiagID, MappingInfo);
- }
+ // Get the mapping information, or compute it lazily.
+ DiagnosticMappingInfo &MappingInfo = State->getOrAddMappingInfo(
+ (diag::kind)DiagID);
bool ShouldEmitInSystemHeader = false;
continue;
Record.push_back(point.Loc.getRawEncoding());
- for (DiagnosticsEngine::DiagState::iterator
+ for (DiagnosticsEngine::DiagState::const_iterator
I = point.State->begin(), E = point.State->end(); I != E; ++I) {
if (I->second.isPragma()) {
Record.push_back(I->first);