Nicolas Cannasse 11 роки тому
батько
коміт
133e4851a7
2 змінених файлів з 4 додано та 2 видалено
  1. 2 1
      h3d/mat/Material.hx
  2. 2 1
      h3d/pass/Pass.hx

+ 2 - 1
h3d/mat/Material.hx

@@ -11,7 +11,7 @@ class Material {
 		this.passes = passes;
 	}
 	
-	public function addPass( p : Pass ) {
+	public function addPass<T:Pass>( p : T ) : T {
 		var prev = null, cur = passes;
 		while( cur != null ) {
 			prev = cur;
@@ -22,6 +22,7 @@ class Material {
 		else
 			prev.nextPass = p;
 		p.nextPass = null;
+		return p;
 	}
 	
 	public function removePass( p : Pass ) {

+ 2 - 1
h3d/pass/Pass.hx

@@ -55,8 +55,9 @@ class Pass {
 		this.colorMask = (r?1:0) | (g?2:0) | (b?4:0) | (a?8:0);
 	}
 	
-	public function addShader(s) {
+	public function addShader<T:hxsl.Shader>(s:T) : T {
 		shaders.push(s);
+		return s;
 	}
 	
 	public function removeShader(s) {