From 1dbbec55fc54cdb78ad6bd3278c3ba4c059bc077 Mon Sep 17 00:00:00 2001
From: Nathan Sweet <nathan.sweet@gmail.com>
Date: Mon, 5 Oct 2020 23:08:26 -0700
Subject: [PATCH] Added setDefaultShader and a bind method for customization
 when a texture is bound (eg to set shader values).

---
 .../spine/utils/TwoColorPolygonBatch.java     | 27 +++++++++++++++----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/spine-libgdx/spine-libgdx/src/com/esotericsoftware/spine/utils/TwoColorPolygonBatch.java b/spine-libgdx/spine-libgdx/src/com/esotericsoftware/spine/utils/TwoColorPolygonBatch.java
index eac334566..8f5a005f1 100644
--- a/spine-libgdx/spine-libgdx/src/com/esotericsoftware/spine/utils/TwoColorPolygonBatch.java
+++ b/spine-libgdx/spine-libgdx/src/com/esotericsoftware/spine/utils/TwoColorPolygonBatch.java
@@ -68,8 +68,8 @@ public class TwoColorPolygonBatch implements PolygonBatch {
 	private final Matrix4 combinedMatrix = new Matrix4();
 	private boolean blendingDisabled;
 
-	private final ShaderProgram defaultShader;
-	private final boolean ownsDefaultShader;
+	private ShaderProgram defaultShader;
+	private boolean ownsDefaultShader;
 	private ShaderProgram shader;
 
 	private int vertexIndex, triangleIndex;
@@ -1317,7 +1317,7 @@ public class TwoColorPolygonBatch implements PolygonBatch {
 
 		totalRenderCalls++;
 
-		lastTexture.bind();
+		bind(lastTexture);
 		Mesh mesh = this.mesh;
 		mesh.setVertices(vertices, 0, vertexIndex);
 		mesh.setIndices(triangles, 0, triangleIndex);
@@ -1325,8 +1325,7 @@ public class TwoColorPolygonBatch implements PolygonBatch {
 			Gdx.gl.glDisable(GL20.GL_BLEND);
 		else {
 			Gdx.gl.glEnable(GL20.GL_BLEND);
-			if (blendSrcFunc != -1) 
-				Gdx.gl.glBlendFuncSeparate(blendSrcFunc, blendDstFunc, blendSrcFuncAlpha, blendDstFuncAlpha);
+			if (blendSrcFunc != -1) Gdx.gl.glBlendFuncSeparate(blendSrcFunc, blendDstFunc, blendSrcFuncAlpha, blendDstFuncAlpha);
 		}
 		mesh.render(shader, GL20.GL_TRIANGLES, 0, triangleIndex);
 
@@ -1334,6 +1333,10 @@ public class TwoColorPolygonBatch implements PolygonBatch {
 		triangleIndex = 0;
 	}
 
+	protected void bind (Texture texture) {
+		texture.bind();
+	}
+
 	@Override
 	public void disableBlending () {
 		flush();
@@ -1401,6 +1404,20 @@ public class TwoColorPolygonBatch implements PolygonBatch {
 		invTexHeight = 1.0f / texture.getHeight();
 	}
 
+	/** Flushes the batch if the shader was changed. */
+	public void setDefaultShader (ShaderProgram newDefaultShader) {
+		boolean current = shader == defaultShader;
+		boolean flush = current && drawing;
+		if (flush) flush();
+		if (ownsDefaultShader) defaultShader.dispose();
+		defaultShader = newDefaultShader;
+		if (current) shader = newDefaultShader;
+		if (flush) {
+			newDefaultShader.bind();
+			setupMatrices();
+		}
+	}
+
 	/** Flushes the batch if the shader was changed.
 	 * @param newShader If null, the default shader is used. */
 	@Override