summaryrefslogtreecommitdiffstats
path: root/source
diff options
context:
space:
mode:
Diffstat (limited to 'source')
-rw-r--r--source/slang/slang-ast-print.cpp4
-rw-r--r--source/slang/slang-language-server.cpp116
-rw-r--r--source/slang/slang-syntax.cpp11
-rw-r--r--source/slang/slang-syntax.h2
4 files changed, 132 insertions, 1 deletions
diff --git a/source/slang/slang-ast-print.cpp b/source/slang/slang-ast-print.cpp
index b38b58385..23430cd13 100644
--- a/source/slang/slang-ast-print.cpp
+++ b/source/slang/slang-ast-print.cpp
@@ -168,7 +168,9 @@ void ASTPrinter::_addDeclPathRec(const DeclRef<Decl>& declRef, Index depth)
// If the parent declaration is a generic, then we need to print out its
// signature
- if (parentGenericDeclRef)
+ if (parentGenericDeclRef &&
+ !declRef.as<GenericValueParamDecl>() &&
+ !declRef.as<GenericTypeParamDecl>())
{
auto genSubst = as<GenericSubstitution>(declRef.substitutions.substitutions);
if (genSubst)
diff --git a/source/slang/slang-language-server.cpp b/source/slang/slang-language-server.cpp
index a2cafa55a..b826d7cfe 100644
--- a/source/slang/slang-language-server.cpp
+++ b/source/slang/slang-language-server.cpp
@@ -216,6 +216,55 @@ static bool isBoolType(Type* t)
return basicType->baseType == BaseType::Bool;
}
+String getDeclKindString(DeclRef<Decl> declRef)
+{
+ if (declRef.as<ParamDecl>())
+ {
+ return "(parameter) ";
+ }
+ else if (declRef.as<GenericTypeParamDecl>())
+ {
+ return "(generic type parameter) ";
+ }
+ else if (declRef.as<GenericValueParamDecl>())
+ {
+ return "(generic value parameter) ";
+ }
+ else if (declRef.as<AttributeDecl>())
+ {
+ return "(attribute) ";
+ }
+ else if (auto varDecl = declRef.as<VarDeclBase>())
+ {
+ auto parent = declRef.getParent();
+ if (as<GenericDecl>(parent))
+ parent = parent.getParent();
+ if (parent.as<InterfaceDecl>())
+ {
+ return "(associated constant) ";
+ }
+ else if (parent.as<AggTypeDeclBase>())
+ {
+ return "(field) ";
+ }
+ const char* scopeKind = "";
+ if (parent.as<NamespaceDeclBase>())
+ scopeKind = "global ";
+ else if (getParentDecl(declRef.getDecl()))
+ scopeKind = "local ";
+ StringBuilder sb;
+ sb << "(";
+ sb << scopeKind;
+ if (varDecl.as<LetDecl>())
+ sb << "value";
+ else
+ sb << "variable";
+ sb << ") ";
+ return sb.produceString();
+ }
+ return String();
+}
+
String getDeclSignatureString(DeclRef<Decl> declRef, WorkspaceVersion* version)
{
if (declRef.getDecl())
@@ -225,6 +274,7 @@ String getDeclSignatureString(DeclRef<Decl> declRef, WorkspaceVersion* version)
astBuilder,
ASTPrinter::OptionFlag::ParamNames | ASTPrinter::OptionFlag::NoInternalKeywords |
ASTPrinter::OptionFlag::SimplifiedBuiltinType);
+ printer.getStringBuilder() << getDeclKindString(declRef);
printer.addDeclSignature(declRef);
if (auto varDecl = as<VarDeclBase>(declRef.getDecl()))
{
@@ -496,6 +546,72 @@ SlangResult LanguageServer::hover(
_tryGetDocumentation(sb, version, declRef.getDecl());
+ if (auto funcDecl = as<FunctionDeclBase>(declRef.getDecl()))
+ {
+ DiagnosticSink sink;
+ SharedSemanticsContext semanticContext(version->linkage, getModule(funcDecl), &sink);
+ SemanticsVisitor semanticsVisitor(&semanticContext);
+
+ auto assocDecls = semanticContext.getAssociatedDeclsForDecl(funcDecl);
+ Decl* bwdDiff = nullptr;
+ Decl* fwdDiff = nullptr;
+ Decl* primalSubst = nullptr;
+ auto getDeclFromExpr = [&](Expr* expr) -> Decl*
+ {
+ if (auto declRefExpr = as<DeclRefExpr>(expr))
+ return declRefExpr->declRef.getDecl();
+ return nullptr;
+ };
+ for (auto& assocDecl : assocDecls)
+ {
+ if (assocDecl->kind == DeclAssociationKind::ForwardDerivativeFunc)
+ fwdDiff = assocDecl->decl;
+ else if (assocDecl->kind == DeclAssociationKind::BackwardDerivativeFunc)
+ bwdDiff = assocDecl->decl;
+ else if (assocDecl->kind == DeclAssociationKind::PrimalSubstituteFunc)
+ primalSubst = assocDecl->decl;
+ }
+ bool isBackwardDifferentiable = false;
+ bool isForwardDifferentiable = false;
+ for (auto modifier : funcDecl->modifiers)
+ {
+ if (auto bwdDiffModifier = as<BackwardDerivativeAttribute>(modifier))
+ bwdDiff = getDeclFromExpr(bwdDiffModifier->funcExpr);
+ else if (auto fwdDiffModifier = as<ForwardDerivativeAttribute>(modifier))
+ fwdDiff = getDeclFromExpr(fwdDiffModifier->funcExpr);
+ else if (auto primalSubstModifier = as<PrimalSubstituteAttribute>(modifier))
+ primalSubst = getDeclFromExpr(primalSubstModifier->funcExpr);
+ else if (as<ForwardDifferentiableAttribute>(modifier))
+ isForwardDifferentiable = true;
+ else if (as<BackwardDifferentiableAttribute>(modifier))
+ isBackwardDifferentiable = true;
+ }
+ if (primalSubst)
+ {
+ for (auto modifier : primalSubst->modifiers)
+ {
+ if (as<ForwardDifferentiableAttribute>(modifier))
+ isForwardDifferentiable = true;
+ else if (as<BackwardDifferentiableAttribute>(modifier))
+ isBackwardDifferentiable = true;
+ }
+ }
+ if (isBackwardDifferentiable)
+ {
+ sb << "\nForward and backward differentiable\n\n";
+ }
+ if (isForwardDifferentiable)
+ {
+ sb << "\nForward differentiable\n\n";
+ }
+ if (fwdDiff && fwdDiff->getName())
+ sb << "Forward derivative: `" << fwdDiff->getName()->text << "`\n\n";
+ if (bwdDiff && bwdDiff->getName())
+ sb << "Backward derivative: `" << bwdDiff->getName()->text << "`\n\n";
+ if (primalSubst && primalSubst->getName())
+ sb << "Primal substitute: `" << primalSubst->getName()->text << "`\n\n";
+ }
+
auto humaneLoc = version->linkage->getSourceManager()->getHumaneLoc(
declRef.getLoc(), SourceLocType::Actual);
appendDefinitionLocation(sb, m_workspace, humaneLoc);
diff --git a/source/slang/slang-syntax.cpp b/source/slang/slang-syntax.cpp
index 11729800c..606c77096 100644
--- a/source/slang/slang-syntax.cpp
+++ b/source/slang/slang-syntax.cpp
@@ -1413,6 +1413,17 @@ Decl* getParentDecl(Decl* decl)
return decl;
}
+Decl* getParentFunc(Decl* decl)
+{
+ while (decl)
+ {
+ if (as<FunctionDeclBase>(decl))
+ return decl;
+ decl = decl->parentDecl;
+ }
+ return nullptr;
+}
+
static const ImageFormatInfo kImageFormatInfos[] =
{
#define SLANG_IMAGE_FORMAT_INFO(TYPE, COUNT, SIZE) SLANG_SCALAR_TYPE_##TYPE, uint8_t(COUNT), uint8_t(SIZE)
diff --git a/source/slang/slang-syntax.h b/source/slang/slang-syntax.h
index dd119cf3a..3ac258f08 100644
--- a/source/slang/slang-syntax.h
+++ b/source/slang/slang-syntax.h
@@ -337,6 +337,8 @@ namespace Slang
/// Get the parent decl, skipping any generic decls in between.
Decl* getParentDecl(Decl* decl);
+ Decl* getParentFunc(Decl* decl);
+
} // namespace Slang
#endif
generated by cgit v1.2.3 (git 2.39.1) at 2026-06-01 19:01:38 +0000