44                    const lsp::URIForFile *uri = nullptr) {  in getLocationFromLoc()  argument
52     if (sourceLoc && (!uri || sourceLoc->uri == *uri)) {  in getLocationFromLoc()
76                                     const lsp::URIForFile &uri) {  in collectLocationsFromLoc()  argument
84     if (sourceLoc && sourceLoc->uri != uri)  in collectLocationsFromLoc()
183                                                const lsp::URIForFile &uri) {  in getLspDiagnoticFromDiag()  argument
195       getLocationFromLoc(sourceMgr, diag.getLocation(), &uri);  in getLspDiagnoticFromDiag()
223       noteLoc.uri = uri;  in getLspDiagnoticFromDiag()
240   MLIRDocument(MLIRContext &context, const lsp::URIForFile &uri,
249   void getLocationsOf(const lsp::URIForFile &uri, const lsp::Position &defPos,
251   void findReferencesOf(const lsp::URIForFile &uri, const lsp::Position &pos,
258   Optional<lsp::Hover> findHover(const lsp::URIForFile &uri,
284   lsp::CompletionList getCodeCompletion(const lsp::URIForFile &uri,
292   void getCodeActionForDiagnostic(const lsp::URIForFile &uri,
313 MLIRDocument::MLIRDocument(MLIRContext &context, const lsp::URIForFile &uri,  in MLIRDocument()  argument
317     diagnostics.push_back(getLspDiagnoticFromDiag(sourceMgr, diag, uri));  in MLIRDocument()
321   auto memBuffer = llvm::MemoryBuffer::getMemBufferCopy(contents, uri.file());  in MLIRDocument()
323     lsp::Logger::error("Failed to create memory buffer for file", uri.file());  in MLIRDocument()
340 void MLIRDocument::getLocationsOf(const lsp::URIForFile &uri,  in getLocationsOf()  argument
349     locations.emplace_back(uri, sourceMgr, def.loc);  in getLocationsOf()
356       return collectLocationsFromLoc(op.op->getLoc(), locations, uri);  in getLocationsOf()
359         return collectLocationsFromLoc(op.op->getLoc(), locations, uri);  in getLocationsOf()
362         locations.emplace_back(uri, sourceMgr, op.loc);  in getLocationsOf()
363         return collectLocationsFromLoc(op.op->getLoc(), locations, uri);  in getLocationsOf()
378 void MLIRDocument::findReferencesOf(const lsp::URIForFile &uri,  in findReferencesOf()  argument
384     references.emplace_back(uri, sourceMgr, def.loc);  in findReferencesOf()
386       references.emplace_back(uri, sourceMgr, use);  in findReferencesOf()
398           references.emplace_back(uri, sourceMgr, symUse);  in findReferencesOf()
408         references.emplace_back(uri, sourceMgr, symUse);  in findReferencesOf()
428 Optional<lsp::Hover> MLIRDocument::findHover(const lsp::URIForFile &uri,  in findHover()  argument
784 MLIRDocument::getCodeCompletion(const lsp::URIForFile &uri,  in getCodeCompletion()  argument
811     const lsp::URIForFile &uri, lsp::Position &pos, StringRef severity,  in getCodeActionForDiagnostic()  argument
850                     const lsp::URIForFile &uri, StringRef contents,  in MLIRTextFileChunk()
852       : lineOffset(lineOffset), document(context, uri, contents, diagnostics) {}  in MLIRTextFileChunk()
879   MLIRTextFile(const lsp::URIForFile &uri, StringRef fileContents,
890   void getLocationsOf(const lsp::URIForFile &uri, lsp::Position defPos,
892   void findReferencesOf(const lsp::URIForFile &uri, lsp::Position pos,
894   Optional<lsp::Hover> findHover(const lsp::URIForFile &uri,
897   lsp::CompletionList getCodeCompletion(const lsp::URIForFile &uri,
899   void getCodeActions(const lsp::URIForFile &uri, const lsp::Range &pos,
927 MLIRTextFile::MLIRTextFile(const lsp::URIForFile &uri, StringRef fileContents,  in MLIRTextFile()  argument
941       context, /*lineOffset=*/0, uri, subContents.front(), diagnostics));  in MLIRTextFile()
946     auto chunk = std::make_unique<MLIRTextFileChunk>(context, lineOffset, uri,  in MLIRTextFile()
959         if (it.location.uri == uri)  in MLIRTextFile()
967 void MLIRTextFile::getLocationsOf(const lsp::URIForFile &uri,  in getLocationsOf()  argument
971   chunk.document.getLocationsOf(uri, defPos, locations);  in getLocationsOf()
977     if (loc.uri == uri)  in getLocationsOf()
981 void MLIRTextFile::findReferencesOf(const lsp::URIForFile &uri,  in findReferencesOf()  argument
985   chunk.document.findReferencesOf(uri, pos, references);  in findReferencesOf()
991     if (loc.uri == uri)  in findReferencesOf()
995 Optional<lsp::Hover> MLIRTextFile::findHover(const lsp::URIForFile &uri,  in findHover()  argument
998   Optional<lsp::Hover> hoverInfo = chunk.document.findHover(uri, hoverPos);  in findHover()
1045 lsp::CompletionList MLIRTextFile::getCodeCompletion(const lsp::URIForFile &uri,  in getCodeCompletion()  argument
1049       uri, completePos, context.getDialectRegistry());  in getCodeCompletion()
1061 void MLIRTextFile::getCodeActions(const lsp::URIForFile &uri,  in getCodeActions()  argument
1091     chunk.document.getCodeActionForDiagnostic(uri, diagPos, severity,  in getCodeActions()
1097         if (noteDiag.location.uri != uri)  in getCodeActions()
1101         chunk.document.getCodeActionForDiagnostic(uri, diagPos, "note",  in getCodeActions()
1110     action.edit->changes[uri.uri().str()] = std::move(edits);  in getCodeActions()
1156     const URIForFile &uri, StringRef contents, int64_t version,  in addOrUpdateDocument()  argument
1158   impl->files[uri.file()] = std::make_unique<MLIRTextFile>(  in addOrUpdateDocument()
1159       uri, contents, version, impl->registry, diagnostics);  in addOrUpdateDocument()
1162 Optional<int64_t> lsp::MLIRServer::removeDocument(const URIForFile &uri) {  in removeDocument()  argument
1163   auto it = impl->files.find(uri.file());  in removeDocument()
1172 void lsp::MLIRServer::getLocationsOf(const URIForFile &uri,  in getLocationsOf()  argument
1175   auto fileIt = impl->files.find(uri.file());  in getLocationsOf()
1177     fileIt->second->getLocationsOf(uri, defPos, locations);  in getLocationsOf()
1180 void lsp::MLIRServer::findReferencesOf(const URIForFile &uri,  in findReferencesOf()  argument
1183   auto fileIt = impl->files.find(uri.file());  in findReferencesOf()
1185     fileIt->second->findReferencesOf(uri, pos, references);  in findReferencesOf()
1188 Optional<lsp::Hover> lsp::MLIRServer::findHover(const URIForFile &uri,  in findHover()  argument
1190   auto fileIt = impl->files.find(uri.file());  in findHover()
1192     return fileIt->second->findHover(uri, hoverPos);  in findHover()
1197     const URIForFile &uri, std::vector<DocumentSymbol> &symbols) {  in findDocumentSymbols()  argument
1198   auto fileIt = impl->files.find(uri.file());  in findDocumentSymbols()
1204 lsp::MLIRServer::getCodeCompletion(const URIForFile &uri,  in getCodeCompletion()  argument
1206   auto fileIt = impl->files.find(uri.file());  in getCodeCompletion()
1208     return fileIt->second->getCodeCompletion(uri, completePos);  in getCodeCompletion()
1212 void lsp::MLIRServer::getCodeActions(const URIForFile &uri, const Range &pos,  in getCodeActions()  argument
1215   auto fileIt = impl->files.find(uri.file());  in getCodeActions()
1217     fileIt->second->getCodeActions(uri, pos, context, actions);  in getCodeActions()