#5565: Export spectrum

codereader · Mar 25, 2021 · e0fb7ce · e0fb7ce
1 parent fec4baa
commit e0fb7ce
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/radiantcore/shaders/MaterialSourceGenerator.cpp b/radiantcore/shaders/MaterialSourceGenerator.cpp
@@ -89,6 +89,12 @@ std::ostream& operator<<(std::ostream& stream, ShaderTemplate& shaderTemplate)
         }
     }
 
+    // Spectrum
+    if (shaderTemplate.getSpectrum() != 0)
+    {
+        stream << "\tspectrum " << shaderTemplate.getSpectrum() << "\n";
+    }
+
     for (const auto& layer : shaderTemplate.getLayers())
     {
         stream << "\t{\n";

diff --git a/test/MaterialExport.cpp b/test/MaterialExport.cpp
@@ -194,4 +194,25 @@ TEST_F(MaterialExportTest, Sort)
     expectDefinitionDoesNotContain(material, "sort");
 }
 
+TEST_F(MaterialExportTest, Spectrum)
+{
+    auto material = GlobalMaterialManager().getMaterial("textures/exporttest/empty");
+
+    EXPECT_EQ(string::trim_copy(material->getDefinition()), "");
+
+    for (int i = -50; i < 50; ++i)
+    {
+        material->setSpectrum(i);
+
+        if (i != 0)
+        {
+            expectDefinitionContains(material, fmt::format("spectrum {0}", i));
+        }
+        else // spectrum 0 is the default, doesn't need to be declared
+        {
+            expectDefinitionDoesNotContain(material, "spectrum");
+        }
+    }
+}
+
 }