From 1050e0eb96d6c8e7a6cfb253458155e1014625c3 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Tue, 5 Dec 2023 10:06:19 -0800
Subject: Support `include` for pulling file into the current module. (#3377)

* Support `include` for pulling file into the current module.

* Add auto-completion, hover info and goto-def support.

* Disable warning for missing `module` declaration for now.

---------

Co-authored-by: Yong He <yhe@nvidia.com>
---
 tests/language-feature/modules/include/a.slang    |  8 ++++++++
 tests/language-feature/modules/include/b.slang    |  6 ++++++
 tests/language-feature/modules/include/c.slang    |  4 ++++
 tests/language-feature/modules/include/main.slang | 20 ++++++++++++++++++++
 4 files changed, 38 insertions(+)
 create mode 100644 tests/language-feature/modules/include/a.slang
 create mode 100644 tests/language-feature/modules/include/b.slang
 create mode 100644 tests/language-feature/modules/include/c.slang
 create mode 100644 tests/language-feature/modules/include/main.slang

(limited to 'tests/language-feature/modules/include')

diff --git a/tests/language-feature/modules/include/a.slang b/tests/language-feature/modules/include/a.slang
new file mode 100644
index 000000000..5e01a15c5
--- /dev/null
+++ b/tests/language-feature/modules/include/a.slang
@@ -0,0 +1,8 @@
+//TEST_IGNORE_FILE:
+
+implementing main;
+
+// Cyclic include, OK.
+__include a;
+
+int f_a() { return 1; }
diff --git a/tests/language-feature/modules/include/b.slang b/tests/language-feature/modules/include/b.slang
new file mode 100644
index 000000000..737178d54
--- /dev/null
+++ b/tests/language-feature/modules/include/b.slang
@@ -0,0 +1,6 @@
+// TEST_IGNORE_FILE:
+implementing main;
+
+__include a;
+
+int f_b() { return 2 + f_a(); }
diff --git a/tests/language-feature/modules/include/c.slang b/tests/language-feature/modules/include/c.slang
new file mode 100644
index 000000000..9588b37be
--- /dev/null
+++ b/tests/language-feature/modules/include/c.slang
@@ -0,0 +1,4 @@
+// TEST_IGNORE_FILE:
+implementing main;
+
+int f_c() { return f_a() + f_b(); }
diff --git a/tests/language-feature/modules/include/main.slang b/tests/language-feature/modules/include/main.slang
new file mode 100644
index 000000000..e036adc3c
--- /dev/null
+++ b/tests/language-feature/modules/include/main.slang
@@ -0,0 +1,20 @@
+//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -shaderobj -output-using-type
+//TEST(compute):COMPARE_COMPUTE(filecheck-buffer=CHECK): -vk -shaderobj -output-using-type
+
+module main;
+
+__include c;
+__include c; // Duplicate include, OK.
+__include b;
+
+
+//TEST_INPUT:ubuffer(data=[0 0 0 0], stride=4):out,name=outputBuffer
+RWStructuredBuffer<int> outputBuffer;
+
+[numthreads(4, 1, 1)]
+void computeMain(int3 dispatchThreadID: SV_DispatchThreadID)
+{
+    int tid = dispatchThreadID.x;
+    outputBuffer[tid] = f_c();
+    // CHECK: 4 
+}
-- 
cgit v1.2.3