Página Principal Explorar Ajuda
Iniciar Sessão
haxe
/
armory3d.armortools
mirror de https://github.com/armory3d/armortools.git
2
0
Fork 0
Ficheiros Problemas 0 Wiki
Ramo: lab
Ramos Etiquetas
armory3d.armort...
/
paint
/
sources
/
make_bake.ts

make_bake.ts 7.4 KB
Link permanente Histórico Em bruto

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123 
  1. 

  2. function make_bake_run(con: node_shader_context_t, kong: node_shader_t) {
    3. 

  3. 	if (context_raw.bake_type == bake_type_t.CURVATURE) {
    4. 

  4. 		let pass: bool = parser_material_bake_passthrough;
    5. 

  5. 		let strength: string = pass ? parser_material_bake_passthrough_strength : context_raw.bake_curv_strength + "";
    6. 

  6. 		let radius: string = pass ? parser_material_bake_passthrough_radius : context_raw.bake_curv_radius + "";
    7. 

  7. 		let offset: string = pass ? parser_material_bake_passthrough_offset : context_raw.bake_curv_offset + "";
    8. 

  8. 		strength = "float(" + strength + ")";
    9. 

  9. 		radius = "float(" + radius + ")";
    10. 

  10. 		offset = "float(" + offset + ")";
    11. 

  11. 		kong.frag_n = true;
    12. 

  12. 		node_shader_write_frag(kong, "var dx: float3 = ddx3(n);");
    13. 

  13. 		node_shader_write_frag(kong, "var dy: float3 = ddy3(n);");
    14. 

  14. 		node_shader_write_frag(kong, "var curvature: float = max(dot(dx, dx), dot(dy, dy));");
    15. 

  15. 		node_shader_write_frag(kong, "curvature = clamp(pow(curvature, (1.0 / " + radius + ") * 0.25) * " + strength + " * 2.0 + " + offset + " / 10.0, 0.0, 1.0);");
    16. 

  16. 		if (context_raw.bake_axis != bake_axis_t.XYZ) {
    17. 

  17. 			let axis: string = make_bake_axis_string(context_raw.bake_axis);
    18. 

  18. 			node_shader_write_frag(kong, "curvature *= dot(n, " + axis + ");");
    19. 

  19. 		}
    20. 

  20. 		node_shader_write_frag(kong, "output[0] = float4(curvature, curvature, curvature, 1.0);");
    21. 

  21. 	}
    22. 

  22. 	else if (context_raw.bake_type == bake_type_t.NORMAL) { // Tangent
    23. 

  23. 		kong.frag_n = true;
    24. 

  24. 		node_shader_add_texture(kong, "texpaint_undo", "_texpaint_undo"); // Baked high-poly normals
    25. 

  25. 		node_shader_write_frag(kong, "var n0: float3 = sample_lod(texpaint_undo, sampler_linear, tex_coord, 0.0).rgb * float3(2.0, 2.0, 2.0) - float3(1.0, 1.0, 1.0);");
    26. 

  26. 		node_shader_add_function(kong, str_cotangent_frame);
    27. 

  27. 		node_shader_write_frag(kong, "var invTBN: float3x3 = transpose(cotangent_frame(n, n, tex_coord));");
    28. 

  28. 		node_shader_write_frag(kong, "var res: float3 = normalize(invTBN * n0) * float3(0.5, 0.5, 0.5) + float3(0.5, 0.5, 0.5);");
    29. 

  29. 		node_shader_write_frag(kong, "output[0] = float4(res, 1.0);");
    30. 

  30. 	}
    31. 

  31. 	else if (context_raw.bake_type == bake_type_t.NORMAL_OBJECT) {
    32. 

  32. 		kong.frag_n = true;
    33. 

  33. 		node_shader_write_frag(kong, "output[0] = float4(n * float3(0.5, 0.5, 0.5) + float3(0.5, 0.5, 0.5), 1.0);");
    34. 

  34. 		if (context_raw.bake_up_axis == bake_up_axis_t.Y) {
    35. 

  35. 			node_shader_write_frag(kong, "output[0].rgb = float3(output[0].r, output[0].b, 1.0 - output[0].g);");
    36. 

  36. 		}
    37. 

  37. 	}
    38. 

  38. 	else if (context_raw.bake_type == bake_type_t.HEIGHT) {
    39. 

  39. 		kong.frag_wposition = true;
    40. 

  40. 		node_shader_add_texture(kong, "texpaint_undo", "_texpaint_undo"); // Baked high-poly positions
    41. 

  41. 		node_shader_write_frag(kong, "var wpos0: float3 = sample_lod(texpaint_undo, sampler_linear, tex_coord, 0.0).rgb * float3(2.0, 2.0, 2.0) - float3(1.0, 1.0, 1.0);");
    42. 

  42. 		node_shader_write_frag(kong, "var res: float = distance(wpos0, input.wposition) * 10.0;");
    43. 

  43. 		node_shader_write_frag(kong, "output[0] = float4(res, res, res, 1.0);");
    44. 

  44. 	}
    45. 

  45. 	else if (context_raw.bake_type == bake_type_t.DERIVATIVE) {
    46. 

  46. 		node_shader_add_texture(kong, "texpaint_undo", "_texpaint_undo"); // Baked height
    47. 

  47. 		node_shader_write_frag(kong, "var tex_dx: float2 = ddx2(tex_coord);");
    48. 

  48. 		node_shader_write_frag(kong, "var tex_dy: float2 = ddy2(tex_coord);");
    49. 

  49. 		node_shader_write_frag(kong, "var h0: float = sample_lod(texpaint_undo, sampler_linear, tex_coord, 0.0).r * 100.0;");
    50. 

  50. 		node_shader_write_frag(kong, "var h1: float = sample_lod(texpaint_undo, sampler_linear, tex_coord + tex_dx, 0.0).r * 100.0;");
    51. 

  51. 		node_shader_write_frag(kong, "var h2: float = sample_lod(texpaint_undo, sampler_linear, tex_coord + tex_dy, 0.0).r * 100.0;");
    52. 

  52. 		node_shader_write_frag(kong, "output[0] = float4((h1 - h0) * 0.5 + 0.5, (h2 - h0) * 0.5 + 0.5, 0.0, 1.0);");
    53. 

  53. 	}
    54. 

  54. 	else if (context_raw.bake_type == bake_type_t.POSITION) {
    55. 

  55. 		kong.frag_wposition = true;
    56. 

  56. 		node_shader_write_frag(kong, "output[0] = float4(input.wposition * float3(0.5, 0.5, 0.5) + float3(0.5, 0.5, 0.5), 1.0);");
    57. 

  57. 		if (context_raw.bake_up_axis == bake_up_axis_t.Y) {
    58. 

  58. 			node_shader_write_frag(kong, "output[0].rgb = float3(output[0].r, output[0].b, 1.0 - output[0].g);");
    59. 

  59. 		}
    60. 

  60. 	}
    61. 

  61. 	else if (context_raw.bake_type == bake_type_t.TEXCOORD) {
    62. 

  62. 		node_shader_write_frag(kong, "output[0] = float4(tex_coord.xy, 0.0, 1.0);");
    63. 

  63. 	}
    64. 

  64. 	else if (context_raw.bake_type == bake_type_t.MATERIALID) {
    65. 

  65. 		node_shader_add_texture(kong, "texpaint_nor_undo", "_texpaint_nor_undo");
    66. 

  66. 		node_shader_write_frag(kong, "var sample_matid: float = sample_lod(texpaint_nor_undo, sampler_linear, tex_coord, 0.0).a + 1.0 / 255.0;");
    67. 

  67. 		node_shader_write_frag(kong, "var matid_r: float = frac(sin(dot(float2(sample_matid, sample_matid * 20.0), float2(12.9898, 78.233))) * 43758.5453);");
    68. 

  68. 		node_shader_write_frag(kong, "var matid_g: float = frac(sin(dot(float2(sample_matid * 20.0, sample_matid), float2(12.9898, 78.233))) * 43758.5453);");
    69. 

  69. 		node_shader_write_frag(kong, "var matid_b: float = frac(sin(dot(float2(sample_matid, sample_matid * 40.0), float2(12.9898, 78.233))) * 43758.5453);");
    70. 

  70. 		node_shader_write_frag(kong, "output[0] = float4(matid_r, matid_g, matid_b, 1.0);");
    71. 

  71. 	}
    72. 

  72. 	else if (context_raw.bake_type == bake_type_t.OBJECTID) {
    73. 

  73. 		node_shader_add_constant(kong, "object_id: float", "_object_id");
    74. 

  74. 		node_shader_write_frag(kong, "var obid: float = constants.object_id + 1.0 / 255.0;");
    75. 

  75. 		node_shader_write_frag(kong, "var id_r: float = frac(sin(dot(float2(obid, obid * 20.0), float2(12.9898, 78.233))) * 43758.5453);");
    76. 

  76. 		node_shader_write_frag(kong, "var id_g: float = frac(sin(dot(float2(obid * 20.0, obid), float2(12.9898, 78.233))) * 43758.5453);");
    77. 

  77. 		node_shader_write_frag(kong, "var id_b: float = frac(sin(dot(float2(obid, obid * 40.0), float2(12.9898, 78.233))) * 43758.5453);");
    78. 

  78. 		node_shader_write_frag(kong, "output[0] = float4(id_r, id_g, id_b, 1.0);");
    79. 

  79. 	}
    80. 

  80. 	else if (context_raw.bake_type == bake_type_t.VERTEX_COLOR) {
    81. 

  81. 		if (con.allow_vcols) {
    82. 

  82. 			node_shader_context_add_elem(con, "col", "short4norm");
    83. 

  83. 			node_shader_write_frag(kong, "output[0] = float4(vcolor.r, vcolor.g, vcolor.b, 1.0);");
    84. 

  84. 		}
    85. 

  85. 		else {
    86. 

  86. 			node_shader_write_frag(kong, "output[0] = float4(1.0, 1.0, 1.0, 1.0);");
    87. 

  87. 		}
    88. 

  88. 	}
    89. 

  89. }
    90. 

  90. 

  91. function make_bake_position_normal(kong: node_shader_t) {
    92. 

  92. 	node_shader_add_out(kong, "position: float3");
    93. 

  93. 	node_shader_add_out(kong, "normal: float3");
    94. 

  94. 	node_shader_add_constant(kong, "W: float4x4", "_world_matrix");
    95. 

  95. 	node_shader_write_vert(kong, "output.position = (constants.W * float4(input.pos.xyz, 1.0)).xyz;");
    96. 

  96. 	node_shader_write_vert(kong, "output.normal = float3(input.nor.xy, input.pos.w);");
    97. 

  97. 	node_shader_write_vert(kong, "var tpos: float2 = float2(input.tex.x * 2.0 - 1.0, (1.0 - input.tex.y) * 2.0 - 1.0);");
    98. 

  98. 	node_shader_write_vert(kong, "output.pos = float4(tpos, 0.0, 1.0);");
    99. 

  99. 	kong.frag_out = "float4[2]";
    100. 

  100. 	node_shader_write_frag(kong, "output[0] = float4(input.position, 1.0);");
    101. 

  101. 	node_shader_write_frag(kong, "output[1] = float4(input.normal, 1.0);");
    102. 

  102. }
    103. 

  103. 

  104. function make_bake_set_color_writes(con_paint: node_shader_context_t) {
    105. 

  105. 	// Bake into base color, disable other slots
    106. 

  106. 	con_paint.data.color_writes_red[1] = false;
    107. 

  107. 	con_paint.data.color_writes_green[1] = false;
    108. 

  108. 	con_paint.data.color_writes_blue[1] = false;
    109. 

  109. 	con_paint.data.color_writes_alpha[1] = false;
    110. 

  110. 	con_paint.data.color_writes_red[2] = false;
    111. 

  111. 	con_paint.data.color_writes_green[2] = false;
    112. 

  112. 	con_paint.data.color_writes_blue[2] = false;
    113. 

  113. 	con_paint.data.color_writes_alpha[2] = false;
    114. 

  114. }
    115. 

  115. 

  116. function make_bake_axis_string(i: i32): string {
    117. 

  117. 	return i == bake_axis_t.X  ? "float3(1,0,0)"  :
    118. 

  118. 		   i == bake_axis_t.Y  ? "float3(0,1,0)"  :
    119. 

  119. 		   i == bake_axis_t.Z  ? "float3(0,0,1)"  :
    120. 

  120. 		   i == bake_axis_t.MX ? "float3(-1,0,0)" :
    121. 

  121. 		   i == bake_axis_t.MY ? "float3(0,-1,0)" :
    122. 

  122. 								 "float3(0,0,-1)";
    123. 

  123. }
    124.