From 5169e0fe836d70b0346a4536b7aeff11c91ba12e Mon Sep 17 00:00:00 2001
From: Lynix <lynix680@gmail.com>
Date: Fri, 22 May 2020 19:21:56 +0200
Subject: [PATCH] ShaderNode: Add captions

---
 src/ShaderNode/DataModels/SampleTexture.cpp | 27 +++++++++++++++++++++
 src/ShaderNode/DataModels/SampleTexture.hpp |  5 ++++
 2 files changed, 32 insertions(+)

diff --git a/src/ShaderNode/DataModels/SampleTexture.cpp b/src/ShaderNode/DataModels/SampleTexture.cpp
index 6d222abb6..180989226 100644
--- a/src/ShaderNode/DataModels/SampleTexture.cpp
+++ b/src/ShaderNode/DataModels/SampleTexture.cpp
@@ -121,6 +121,33 @@ auto SampleTexture::dataType(QtNodes::PortType portType, QtNodes::PortIndex port
 	}
 }
 
+QString SampleTexture::portCaption(QtNodes::PortType portType, QtNodes::PortIndex portIndex) const
+{
+	switch (portType)
+	{
+		case QtNodes::PortType::In:
+		{
+			assert(portIndex == 0);
+			return tr("UV");
+		}
+
+		case QtNodes::PortType::Out:
+		{
+			assert(portIndex == 0);
+			return tr("Sample");
+		}
+
+		default:
+			assert(false);
+			throw std::runtime_error("Invalid PortType");
+	}
+}
+
+bool SampleTexture::portCaptionVisible(QtNodes::PortType portType, QtNodes::PortIndex portIndex) const
+{
+	return true;
+}
+
 std::shared_ptr<QtNodes::NodeData> SampleTexture::outData(QtNodes::PortIndex port)
 {
 	assert(port == 0);
diff --git a/src/ShaderNode/DataModels/SampleTexture.hpp b/src/ShaderNode/DataModels/SampleTexture.hpp
index f717797f2..af11d14ec 100644
--- a/src/ShaderNode/DataModels/SampleTexture.hpp
+++ b/src/ShaderNode/DataModels/SampleTexture.hpp
@@ -25,6 +25,11 @@ class SampleTexture : public ShaderNode
 		unsigned int nPorts(QtNodes::PortType portType) const override;
 
 		QtNodes::NodeDataType dataType(QtNodes::PortType portType, QtNodes::PortIndex portIndex) const override;
+
+		QString portCaption(QtNodes::PortType portType, QtNodes::PortIndex portIndex) const override;
+
+		bool portCaptionVisible(QtNodes::PortType portType, QtNodes::PortIndex portIndex) const override;
+
 		std::shared_ptr<QtNodes::NodeData> outData(QtNodes::PortIndex port) override;
 
 	protected: