summaryrefslogtreecommitdiffstats
path: root/source/slang/slang.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'source/slang/slang.cpp')
-rw-r--r--source/slang/slang.cpp135
1 files changed, 83 insertions, 52 deletions
diff --git a/source/slang/slang.cpp b/source/slang/slang.cpp
index c0c7c4f5a..6515dc35d 100644
--- a/source/slang/slang.cpp
+++ b/source/slang/slang.cpp
@@ -10,6 +10,7 @@
// Artifact
#include "../compiler-core/slang-artifact-impl.h"
#include "../compiler-core/slang-artifact-desc-util.h"
+#include "../compiler-core/slang-artifact-util.h"
#include "slang-module-library.h"
@@ -3151,16 +3152,20 @@ SLANG_NO_THROW SlangResult SLANG_MCALL ComponentType::getEntryPointCode(
auto targetProgram = getTargetProgram(target);
DiagnosticSink sink(linkage->getSourceManager(), Lexer::sourceLocationLexer);
- auto& entryPointResult = targetProgram->getOrCreateEntryPointResult(entryPointIndex, &sink);
+
+ IArtifact* entryPointResult = targetProgram->getOrCreateEntryPointResult(entryPointIndex, &sink);
sink.getBlobIfNeeded(outDiagnostics);
- if(entryPointResult.format == ResultFormat::None )
+ if(entryPointResult == nullptr)
return SLANG_FAIL;
- ComPtr<ISlangBlob> blob;
- SLANG_RETURN_ON_FAIL(entryPointResult.getBlob(blob));
- *outCode = blob.detach();
- return SLANG_OK;
+ IArtifact* significantBlob = ArtifactUtil::findSignificant(entryPointResult);
+ if (!significantBlob)
+ {
+ return SLANG_FAIL;
+ }
+
+ return significantBlob->loadBlob(ArtifactKeep::Yes, outCode);
}
SLANG_NO_THROW SlangResult SLANG_MCALL ComponentType::getEntryPointHostCallable(
@@ -3177,17 +3182,13 @@ SLANG_NO_THROW SlangResult SLANG_MCALL ComponentType::getEntryPointHostCallable(
auto targetProgram = getTargetProgram(target);
DiagnosticSink sink(linkage->getSourceManager(), Lexer::sourceLocationLexer);
- auto& entryPointResult = targetProgram->getOrCreateEntryPointResult(entryPointIndex, &sink);
+ IArtifact* artifact = targetProgram->getOrCreateEntryPointResult(entryPointIndex, &sink);
sink.getBlobIfNeeded(outDiagnostics);
- if(entryPointResult.format == ResultFormat::None )
+ if(artifact == nullptr)
return SLANG_FAIL;
- ComPtr<ISlangSharedLibrary> sharedLibrary;
- SLANG_RETURN_ON_FAIL(entryPointResult.getSharedLibrary(sharedLibrary));
-
- *outSharedLibrary = sharedLibrary.detach();
- return SLANG_OK;
+ return artifact->loadSharedLibrary(ArtifactKeep::Yes, outSharedLibrary);
}
RefPtr<ComponentType> ComponentType::specialize(
@@ -4757,10 +4758,14 @@ void const* EndToEndCompileRequest::getEntryPointCode(int entryPointIndex, size_
auto targetProgram = program->getTargetProgram(targetReq);
if (!targetProgram)
return nullptr;
- CompileResult& result = targetProgram->getExistingEntryPointResult(entryPointIndex);
+ IArtifact* artifact = targetProgram->getExistingEntryPointResult(entryPointIndex);
+ if (!artifact)
+ {
+ return nullptr;
+ }
ComPtr<ISlangBlob> blob;
- SLANG_RETURN_NULL_ON_FAIL(result.getBlob(blob));
+ SLANG_RETURN_NULL_ON_FAIL(artifact->loadBlob(ArtifactKeep::Yes, blob.writeRef()));
if (outSize)
{
@@ -4774,7 +4779,7 @@ static SlangResult _getEntryPointResult(
EndToEndCompileRequest* req,
int entryPointIndex,
int targetIndex,
- Slang::CompileResult** outCompileResult)
+ ComPtr<IArtifact>& outArtifact)
{
auto linkage = req->getLinkage();
auto program = req->getSpecializedGlobalAndEntryPointsComponentType();
@@ -4793,18 +4798,18 @@ static SlangResult _getEntryPointResult(
}
auto entryPointReq = program->getEntryPoint(entryPointIndex);
-
auto targetProgram = program->getTargetProgram(targetReq);
if (!targetProgram)
return SLANG_FAIL;
- *outCompileResult = &targetProgram->getExistingEntryPointResult(entryPointIndex);
+
+ outArtifact = targetProgram->getExistingEntryPointResult(entryPointIndex);
return SLANG_OK;
}
static SlangResult _getWholeProgramResult(
EndToEndCompileRequest* req,
int targetIndex,
- Slang::CompileResult** outCompileResult)
+ ComPtr<IArtifact>& outArtifact)
{
auto linkage = req->getLinkage();
auto program = req->getSpecializedGlobalAndEntryPointsComponentType();
@@ -4824,34 +4829,34 @@ static SlangResult _getWholeProgramResult(
auto targetProgram = program->getTargetProgram(targetReq);
if (!targetProgram)
return SLANG_FAIL;
- *outCompileResult = &targetProgram->getExistingWholeProgramResult();
+ outArtifact = targetProgram->getExistingWholeProgramResult();
return SLANG_OK;
}
SlangResult EndToEndCompileRequest::getEntryPointCodeBlob(int entryPointIndex, int targetIndex, ISlangBlob** outBlob)
{
if (!outBlob) return SLANG_E_INVALID_ARG;
-
- CompileResult* compileResult = nullptr;
- SLANG_RETURN_ON_FAIL(_getEntryPointResult(this, entryPointIndex, targetIndex, &compileResult));
-
- ComPtr<ISlangBlob> blob;
- SLANG_RETURN_ON_FAIL(compileResult->getBlob(blob));
- *outBlob = blob.detach();
- return SLANG_OK;
+ ComPtr<IArtifact> artifact;
+ SLANG_RETURN_ON_FAIL(_getEntryPointResult(this, entryPointIndex, targetIndex, artifact));
+ if (auto significant = ArtifactUtil::findSignificant(artifact))
+ {
+ SLANG_RETURN_ON_FAIL(significant->loadBlob(ArtifactKeep::Yes, outBlob));
+ return SLANG_OK;
+ }
+ return SLANG_E_NOT_AVAILABLE;
}
SlangResult EndToEndCompileRequest::getEntryPointHostCallable(int entryPointIndex, int targetIndex, ISlangSharedLibrary** outSharedLibrary)
{
if (!outSharedLibrary) return SLANG_E_INVALID_ARG;
-
- CompileResult* compileResult = nullptr;
- SLANG_RETURN_ON_FAIL(_getEntryPointResult(this, entryPointIndex, targetIndex, &compileResult));
-
- ComPtr<ISlangSharedLibrary> sharedLibrary;
- SLANG_RETURN_ON_FAIL(compileResult->getSharedLibrary(sharedLibrary));
- *outSharedLibrary = sharedLibrary.detach();
- return SLANG_OK;
+ ComPtr<IArtifact> artifact;
+ SLANG_RETURN_ON_FAIL(_getEntryPointResult(this, entryPointIndex, targetIndex, artifact));
+ if (auto significant = ArtifactUtil::findSignificant(artifact))
+ {
+ SLANG_RETURN_ON_FAIL(significant->loadSharedLibrary(ArtifactKeep::Yes, outSharedLibrary));
+ return SLANG_OK;
+ }
+ return SLANG_E_NOT_AVAILABLE;
}
SlangResult EndToEndCompileRequest::getTargetCodeBlob(int targetIndex, ISlangBlob** outBlob)
@@ -4859,13 +4864,15 @@ SlangResult EndToEndCompileRequest::getTargetCodeBlob(int targetIndex, ISlangBlo
if (!outBlob)
return SLANG_E_INVALID_ARG;
- CompileResult* compileResult = nullptr;
- SLANG_RETURN_ON_FAIL(_getWholeProgramResult(this, targetIndex, &compileResult));
+ ComPtr<IArtifact> artifact;
+ SLANG_RETURN_ON_FAIL(_getWholeProgramResult(this, targetIndex, artifact));
- ComPtr<ISlangBlob> blob;
- SLANG_RETURN_ON_FAIL(compileResult->getBlob(blob));
- *outBlob = blob.detach();
- return SLANG_OK;
+ if (auto significant = ArtifactUtil::findSignificant(artifact))
+ {
+ SLANG_RETURN_ON_FAIL(significant->loadBlob(ArtifactKeep::Yes, outBlob));
+ return SLANG_OK;
+ }
+ return SLANG_E_NOT_AVAILABLE;
}
SlangResult EndToEndCompileRequest::getTargetHostCallable(int targetIndex,ISlangSharedLibrary** outSharedLibrary)
@@ -4873,12 +4880,9 @@ SlangResult EndToEndCompileRequest::getTargetHostCallable(int targetIndex,ISlang
if (!outSharedLibrary)
return SLANG_E_INVALID_ARG;
- CompileResult* compileResult = nullptr;
- SLANG_RETURN_ON_FAIL(_getWholeProgramResult(this, targetIndex, &compileResult));
-
- ComPtr<ISlangSharedLibrary> sharedLibrary;
- SLANG_RETURN_ON_FAIL(compileResult->getSharedLibrary(sharedLibrary));
- *outSharedLibrary = sharedLibrary.detach();
+ ComPtr<IArtifact> artifact;
+ SLANG_RETURN_ON_FAIL(_getWholeProgramResult(this, targetIndex, artifact));
+ SLANG_RETURN_ON_FAIL(artifact->loadSharedLibrary(ArtifactKeep::Yes, outSharedLibrary));
return SLANG_OK;
}
@@ -5019,12 +5023,39 @@ SlangResult EndToEndCompileRequest::getEntryPoint(SlangInt entryPointIndex, slan
SlangResult EndToEndCompileRequest::isParameterLocationUsed(Int entryPointIndex, Int targetIndex, SlangParameterCategory category, UInt spaceIndex, UInt registerIndex, bool& outUsed)
{
- CompileResult* compileResult = nullptr;
- if (_getEntryPointResult(this, static_cast<int>(entryPointIndex), static_cast<int>(targetIndex), &compileResult) != SLANG_OK)
+ if (!ShaderBindingRange::isUsageTracked((slang::ParameterCategory)category))
+ return SLANG_E_NOT_AVAILABLE;
+
+ ComPtr<IArtifact> artifact;
+ if (SLANG_FAILED(_getEntryPointResult(this, static_cast<int>(entryPointIndex), static_cast<int>(targetIndex), artifact)))
return SLANG_E_INVALID_ARG;
- return compileResult->isParameterLocationUsed(category, spaceIndex, registerIndex, outUsed);
-}
+ // We need to find the meta data
+ IArtifact* metadataArtifact = artifact->findArtifactByDerivedDesc(IArtifact::FindStyle::SelfOrChildren,
+ ArtifactDesc::make(ArtifactKind::Base, ArtifactPayload::PostEmitMetadata, ArtifactStyle::Base));
+ if (!metadataArtifact)
+ {
+ return SLANG_E_NOT_AVAILABLE;
+ }
+
+ // Find a rep
+ auto metadataRep = findRepresentation<IPostEmitMetadataArtifactRepresentation>(metadataArtifact);
+ if (!metadataRep)
+ return SLANG_E_NOT_AVAILABLE;
+
+ // TODO: optimize this with a binary search through a sorted list
+ for (const auto& range : metadataRep->getBindingRanges())
+ {
+ if (range.containsBinding((slang::ParameterCategory)category, spaceIndex, registerIndex))
+ {
+ outUsed = true;
+ return SLANG_OK;
+ }
+ }
+
+ outUsed = false;
+ return SLANG_OK;
+}
} // namespace Slang