1 files changed, 113 insertions, 0 deletions
diff --git a/source/slang/slang-target-program.cpp b/source/slang/slang-target-program.cpp
new file mode 100644
index 000000000..ffb859b55
--- /dev/null
+++ b/source/slang/slang-target-program.cpp
@@ -0,0 +1,113 @@
+// slang-target-program.cpp
+#include "slang-target-program.h"
+
+#include "slang-compiler.h"
+#include "slang-type-layout.h"
+
+namespace Slang
+{
+
+//
+// TargetProgram
+//
+
+TargetProgram::TargetProgram(ComponentType* componentType, TargetRequest* targetReq)
+    : m_program(componentType), m_targetReq(targetReq)
+{
+    m_entryPointResults.setCount(componentType->getEntryPointCount());
+    m_optionSet.overrideWith(m_program->getOptionSet());
+    m_optionSet.inheritFrom(targetReq->getOptionSet());
+}
+
+IArtifact* TargetProgram::_createWholeProgramResult(
+    DiagnosticSink* sink,
+    EndToEndCompileRequest* endToEndReq)
+{
+    // We want to call `emitEntryPoints` function to generate code that contains
+    // all the entrypoints defined in `m_program`.
+    // The current logic of `emitEntryPoints` takes a list of entry-point indices to
+    // emit code for, so we construct such a list first.
+    List<Int> entryPointIndices;
+
+    m_entryPointResults.setCount(m_program->getEntryPointCount());
+    entryPointIndices.setCount(m_program->getEntryPointCount());
+    for (Index i = 0; i < entryPointIndices.getCount(); i++)
+        entryPointIndices[i] = i;
+
+    CodeGenContext::Shared sharedCodeGenContext(this, entryPointIndices, sink, endToEndReq);
+    CodeGenContext codeGenContext(&sharedCodeGenContext);
+
+    if (SLANG_FAILED(codeGenContext.emitEntryPoints(m_wholeProgramResult)))
+    {
+        return nullptr;
+    }
+
+    return m_wholeProgramResult;
+}
+
+IArtifact* TargetProgram::_createEntryPointResult(
+    Int entryPointIndex,
+    DiagnosticSink* sink,
+    EndToEndCompileRequest* endToEndReq)
+{
+    // It is possible that entry points got added to the `Program`
+    // *after* we created this `TargetProgram`, so there might be
+    // a request for an entry point that we didn't allocate space for.
+    //
+    // TODO: Change the construction logic so that a `Program` is
+    // constructed all at once rather than incrementally, to avoid
+    // this problem.
+    //
+    if (entryPointIndex >= m_entryPointResults.getCount())
+        m_entryPointResults.setCount(entryPointIndex + 1);
+
+
+    CodeGenContext::EntryPointIndices entryPointIndices;
+    entryPointIndices.add(entryPointIndex);
+
+    CodeGenContext::Shared sharedCodeGenContext(this, entryPointIndices, sink, endToEndReq);
+    CodeGenContext codeGenContext(&sharedCodeGenContext);
+
+    codeGenContext.emitEntryPoints(m_entryPointResults[entryPointIndex]);
+
+    return m_entryPointResults[entryPointIndex];
+}
+
+IArtifact* TargetProgram::getOrCreateWholeProgramResult(DiagnosticSink* sink)
+{
+    if (m_wholeProgramResult)
+        return m_wholeProgramResult;
+
+    // If we haven't yet computed a layout for this target
+    // program, we need to make sure that is done before
+    // code generation.
+    //
+    if (!getOrCreateIRModuleForLayout(sink))
+    {
+        return nullptr;
+    }
+
+    return _createWholeProgramResult(sink);
+}
+
+IArtifact* TargetProgram::getOrCreateEntryPointResult(Int entryPointIndex, DiagnosticSink* sink)
+{
+    if (entryPointIndex >= m_entryPointResults.getCount())
+        m_entryPointResults.setCount(entryPointIndex + 1);
+
+    if (IArtifact* artifact = m_entryPointResults[entryPointIndex])
+        return artifact;
+
+    // If we haven't yet computed a layout for this target
+    // program, we need to make sure that is done before
+    // code generation.
+    //
+    if (!getOrCreateIRModuleForLayout(sink))
+    {
+        return nullptr;
+    }
+
+    return _createEntryPointResult(entryPointIndex, sink);
+}
+
+} // namespace Slang