[转] MoleHill操作AGAL的Helper类之一

本帖最后由 sleep2death 于 2011-4-20 16:49 编辑


应7yue大大的要求,贴写小东西.

package shader {
        import flash.display3D.*;
        import flash.display3D.textures.*;
        import flash.geom.Matrix3D;

        import com.adobe.utils.*;

        public class AGALHelper {
                public  static var context : Context3D;
                public static var program : Program3D;

                public static var vaNames    : Vector.<String> = new Vector.<String>(8); // MAX 8
                public static var fsNames    : Vector.<String> = new Vector.<String>(8); // MAX 8

                public static var vcNames    : Vector.<String> = new Vector.<String>(128); // MAX 128
                public static var fcNames    : Vector.<String> = new Vector.<String>(28); // MAX 28

                public static var vtNames    : Vector.<String> = new Vector.<String>(8); // MAX 8
                public static var ftNames    : Vector.<String> = new Vector.<String>(8); // MAX 8

                public static var vNames    : Vector.<String> = new Vector.<String>(8); // MAX 8

                public static const FLOAT  : uint = 0;
                public static const FLOAT2 : uint = 1;
                public static const FLOAT3 : uint = 2;
                public static const FLOAT4 : uint = 3;

                public static const VERTEX  : uint = 0;
                public static const FRAMENT : uint = 1;

                public static function getAttributeIndex(name : String) : int {
                        var insertIndex : int = vaNames.indexOf(name);

                        if(insertIndex < 0) {
                                insertIndex = vaNames.indexOf(null);
                                vaNames[insertIndex] = name;

                                trace("assign " + name + " to va" + insertIndex);
                        }

                        return insertIndex;

                }

                public static function setAttribute(name : String, buffer : VertexBuffer3D, offset : uint = 0, format : String = "float4") : void {
                        context.setVertexBufferAt(getAttributeIndex(name), buffer, offset, format);
                }

                public static function getVertexConstantIndex(name : String) : int {
                        var insertIndex : int = vcNames.indexOf(name);

                        if(insertIndex < 0) {
                                insertIndex = vcNames.indexOf(null);
                                vcNames[insertIndex] = name;

                                trace("assign " + name + " to vc" + insertIndex);
                        }

                        return insertIndex;
                }

                public static function setVertexConstant(name : String, value : *) : void {
                        if(value is Vector.<Number>){
                                context.setProgramConstantsFromVector(Context3DProgramType.VERTEX, getVertexConstantIndex(name), value);
                        }else if(value is Matrix3D){
                                context.setProgramConstantsFromMatrix(Context3DProgramType.VERTEX, getVertexConstantIndex(name), value, true);
                        }
                }

                public static function getFragmentConstantIndex(name : String) : int {
                        var insertIndex : int = fcNames.indexOf(name);

                        if(insertIndex < 0) {
                                insertIndex = fcNames.indexOf(null);
                                fcNames[insertIndex] = name;

                                trace("assign " + name + " to fc" + insertIndex);
                        }

                        return insertIndex;

                }

                public static function setFragmentConstant(name : String, value : *) : void {
                        if(value is Vector.<Number>){
                                context.setProgramConstantsFromVector(Context3DProgramType.FRAGMENT, getVertexConstantIndex(name), value);
                        }else if(value is Matrix3D){
                                context.setProgramConstantsFromMatrix(Context3DProgramType.FRAGMENT, getVertexConstantIndex(name), value, true);
                        }
                }

                public static function getSampleIndex(name : String) : int {
                        var insertIndex : int = fsNames.indexOf(name);

                        if(insertIndex < 0) {
                                insertIndex = fsNames.indexOf(null);
                                fsNames[insertIndex] = name;

                                trace("assign " + name + " to fs" + insertIndex);
                        }

                        return insertIndex;
                }

                public static function setSample(name : String, value : Texture) : void {
                        context.setTextureAt(getSampleIndex(name), value);
                }

                public static function getIndexByName(vf : String, type : String, name : String) : uint {
                        
                        var first : String = type == "v" ? "v" : vf + type;        
                        var names : Vector.<String> = AGALHelper[first + "Names"];

                        var insertIndex : int = names.indexOf(type + "_" + name);

                        if(insertIndex < 0) {
                                insertIndex = names.indexOf(null);
                                names[insertIndex] = type + "_" + name;

                                trace("assign " + type + "_" + name + " to " + first  + insertIndex);
                        }

                        return insertIndex;
                }

                public static var v_code : String = new String();

                public static function addVertexCode(op_code : String, out : String, in1 : String ,in2 : String = null, extra : String = null) : void {
                        if(op_code == "len"){
                                ExtraCode_len(out, in1);
                        }else{
                                v_code += op_code + " " + parseParam(out, "v")  + ", " + parseParam(in1, "v");
                                if(in2 != null) v_code += " " + parseParam(in2, "v");
                                if(extra != null) v_code +=" " + extra;
                                
                                v_code += " \n";
                        }
                }

                public static var f_code : String = new String();

                public static function addFragmentCode(op_code : String, out : String, in1 : String ,in2 : String = null, extra : String = null) : void {
                        f_code += op_code + " " + parseParam(out, "f")  + ", " + parseParam(in1, "f");
                        if(in2 != null) f_code += " " + parseParam(in2, "f");
                        if(extra != null) f_code +=" " + extra;
                        
                        f_code += " \n";
                }

                private static var v_assembler : AGALMiniAssembler = new AGALMiniAssembler();
                private static var f_assembler : AGALMiniAssembler = new AGALMiniAssembler();

                public static function uploadCode() : void {
                        v_assembler.assemble(Context3DProgramType.VERTEX, v_code);
                        f_assembler.assemble( Context3DProgramType.FRAGMENT, f_code);

                        if(!program){
                                program = context.createProgram();
                                context.setProgram(program);
                        }

                        program.upload(v_assembler.agalcode, f_assembler.agalcode);
                }

                public static function printCode() : void {
                        trace("========  VERTEX     ========");
                        trace(v_code);
                        trace("========  FRAGMENT   ========");
                        trace(f_code);
                }

                private static function parseParam( code : String, vf : String ) : String {
                        var prop : Array = code.split(".");
                        var prop_str : String;

                        if(prop.length > 1) {
                                code = prop[0];
                                prop_str = "."+prop[1];
                        }else{
                                prop_str = "";
                        }

                        var codes : Array = code.split("_");

                        if(codes.length > 1)
                                return (codes[0] == "v" ?"" : vf) + codes[0] + getIndexByName(vf, codes[0], codes[1]) + prop_str;

                        return code + prop_str;
                }

                private static function ExtraCode_len(out : String, in1 : String) : void {
                        addVertexCode("dp3", out, in1, in1);
                        addVertexCode("sqr", out, out);
                }
        }
}

如何使用呢?

先用helper设置一些属性:
AGALHelper.setVertexConstant("c_a1", Vector.<Number>([1.025]));
AGALHelper.setAttribute("a_vb", vertexBuffer));
然后写汇编
                        AGALHelper.addVertexCode("mov", "t_vb", "a_vb");
                        AGALHelper.addVertexCode("mov", "t_uvb", "a_uvb");
                        AGALHelper.addVertexCode("mov", "t_nb", "a_nb");

                        AGALHelper.addVertexCode("sub", "t_v3Ray", "t_vb", "c_v3CameraPos.xyz");
                        AGALHelper.addVertexCode("len", "t_fFar.xyz", "t_v3Ray");

                       
                        AGALHelper.addVertexCode("m44", "op", "t_vb", "c_proj");

                        AGALHelper.addVertexCode("mov", "v_uvb", "t_uvb");
                        AGALHelper.addVertexCode("mov", "v_nb", "t_nb");

                        AGALHelper.addFragmentCode("mov", "t_uvb", "v_uvb");
                        AGALHelper.addFragmentCode("tex", "t_day", "t_uvb", "s_day", "<2d, clamp, linear>");
                        AGALHelper.addFragmentCode("mov", "oc", "t_day");

                        AGALHelper.printCode();
                        AGALHelper.uploadCode();

好处是你再也不要去记ft0, fc1到底是啥了...另外可以新增一些方法,比如length等等