| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010 |
- //
- // This module builds upon Cycles nodes work licensed as
- // Copyright 2011-2013 Blender Foundation
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- //
- let parser_material_con: node_shader_context_t;
- let parser_material_vert: node_shader_t;
- let parser_material_frag: node_shader_t;
- let parser_material_curshader: node_shader_t;
- let parser_material_matcon: material_context_t;
- let parser_material_parsed: string[];
- let parser_material_parents: zui_node_t[];
- let parser_material_canvases: zui_node_canvas_t[];
- let parser_material_nodes: zui_node_t[];
- let parser_material_links: zui_node_link_t[];
- let parser_material_cotangent_frame_written: bool;
- let parser_material_tex_coord: string = "texCoord";
- let parser_material_eps: f32 = 0.000001;
- let parser_material_custom_nodes: map_t<any, any> = map_create();
- let parser_material_parse_surface: bool = true;
- let parser_material_parse_opacity: bool = true;
- let parser_material_parse_height: bool = false;
- let parser_material_parse_height_as_channel: bool = false;
- let parser_material_parse_emission: bool = false;
- let parser_material_parse_subsurface: bool = false;
- let parser_material_parsing_basecolor: bool = false;
- let parser_material_triplanar: bool = false; // Sample using texCoord/1/2 & texCoordBlend
- let parser_material_sample_keep_aspect: bool = false; // Adjust uvs to preserve texture aspect ratio
- let parser_material_sample_uv_scale: string = "1.0";
- let parser_material_transform_color_space: bool = true;
- let parser_material_blur_passthrough: bool = false;
- let parser_material_warp_passthrough: bool = false;
- let parser_material_bake_passthrough: bool = false;
- let parser_material_bake_passthrough_strength: string = "0.0";
- let parser_material_bake_passthrough_radius: string = "0.0";
- let parser_material_bake_passthrough_offset: string = "0.0";
- let parser_material_start_group: zui_node_canvas_t = null;
- let parser_material_start_parents: zui_node_t[] = null;
- let parser_material_start_node: zui_node_t = null;
- let parser_material_arm_export_tangents: bool = true;
- let parser_material_out_normaltan: string; // Raw tangent space normal parsed from normal map
- let parser_material_script_links: map_t<string, string> = null;
- let parser_material_parsed_map: map_t<string, string> = map_create();
- let parser_material_texture_map: map_t<string, string> = map_create();
- function parser_material_get_node(id: i32): zui_node_t {
- for (let i: i32 = 0; i < parser_material_nodes.length; ++i) {
- let n: zui_node_t = parser_material_nodes[i];
- if (n.id == id) {
- return n;
- }
- }
- return null;
- }
- function parser_material_get_link(id: i32): zui_node_link_t {
- for (let i: i32 = 0; i < parser_material_links.length; ++i) {
- let l: zui_node_link_t = parser_material_links[i];
- if (l.id == id) {
- return l;
- }
- }
- return null;
- }
- function parser_material_get_input_link(inp: zui_node_socket_t): zui_node_link_t {
- for (let i: i32 = 0; i < parser_material_links.length; ++i) {
- let l: zui_node_link_t = parser_material_links[i];
- if (l.to_id == inp.node_id) {
- let node: zui_node_t = parser_material_get_node(inp.node_id);
- if (node.inputs.length <= l.to_socket) {
- return null;
- }
- if (node.inputs[l.to_socket] == inp) {
- return l;
- }
- }
- }
- return null;
- }
- function parser_material_get_output_links(out: zui_node_socket_t): zui_node_link_t[] {
- let ls: zui_node_link_t[] = null;
- for (let i: i32 = 0; i < parser_material_links.length; ++i) {
- let l: zui_node_link_t = parser_material_links[i];
- if (l.from_id == out.node_id) {
- let node: zui_node_t = parser_material_get_node(out.node_id);
- if (node.outputs.length <= l.from_socket) {
- continue;
- }
- if (node.outputs[l.from_socket] == out) {
- if (ls == null) {
- ls = [];
- }
- array_push(ls, l);
- }
- }
- }
- return ls;
- }
- function parser_material_init() {
- parser_material_parsed = [];
- parser_material_parents = [];
- parser_material_cotangent_frame_written = false;
- parser_material_out_normaltan = "vec3(0.5, 0.5, 1.0)";
- parser_material_script_links = null;
- parser_material_parsing_basecolor = false;
- }
- function parser_material_parse(canvas: zui_node_canvas_t, _con: node_shader_context_t, _vert: node_shader_t, _frag: node_shader_t, _matcon: material_context_t): shader_out_t {
- parser_material_init();
- parser_material_canvases = [canvas];
- parser_material_nodes = canvas.nodes;
- parser_material_links = canvas.links;
- parser_material_con = _con;
- parser_material_vert = _vert;
- parser_material_frag = _frag;
- parser_material_curshader = parser_material_frag;
- parser_material_matcon = _matcon;
- if (parser_material_start_group != null) {
- parser_material_push_group(parser_material_start_group);
- parser_material_parents = parser_material_start_parents;
- }
- if (parser_material_start_node != null) {
- let link: zui_node_link_t = { id: 99999, from_id: parser_material_start_node.id, from_socket: 0, to_id: -1, to_socket: -1 };
- parser_material_write_result(link);
- return {
- out_basecol: "vec3(0.0, 0.0, 0.0)",
- out_roughness: "0.0",
- out_metallic: "0.0",
- out_occlusion: "1.0",
- out_opacity: "1.0",
- out_height: "0.0",
- out_emission: "0.0",
- out_subsurface: "0.0"
- }
- }
- let output_node: zui_node_t = parser_material_node_by_type(parser_material_nodes, "OUTPUT_MATERIAL");
- if (output_node != null) {
- return parser_material_parse_output(output_node);
- }
- output_node = parser_material_node_by_type(parser_material_nodes, "OUTPUT_MATERIAL_PBR");
- if (output_node != null) {
- return parser_material_parse_output_pbr(output_node);
- }
- return null;
- }
- function parser_material_finalize(con: node_shader_context_t) {
- let vert: node_shader_t = con.vert;
- let frag: node_shader_t = con.frag;
- if (frag.dotnv) {
- frag.vvec = true;
- frag.n = true;
- }
- if (frag.vvec) {
- vert.wposition = true;
- }
- if (frag.bposition) {
- if (parser_material_triplanar) {
- node_shader_write_attrib(frag, "vec3 bposition = vec3(\
- texCoord1.x * texCoordBlend.y + texCoord2.x * texCoordBlend.z,\
- texCoord.x * texCoordBlend.x + texCoord2.y * texCoordBlend.z,\
- texCoord.y * texCoordBlend.x + texCoord1.y * texCoordBlend.y);");
- }
- else if (frag.ndcpos) {
- node_shader_add_out(vert, "vec3 bposition");
- node_shader_write(vert, "bposition = (ndc.xyz / ndc.w);");
- }
- else {
- node_shader_add_out(vert, "vec3 bposition");
- node_shader_add_uniform(vert, "vec3 dim", "_dim");
- node_shader_add_uniform(vert, "vec3 hdim", "_half_dim");
- node_shader_write_attrib(vert, "bposition = (pos.xyz + hdim) / dim;");
- }
- }
- if (frag.wposition) {
- node_shader_add_uniform(vert, "mat4 W", "_world_matrix");
- node_shader_add_out(vert, "vec3 wposition");
- node_shader_write_attrib(vert, "wposition = vec4(mul(vec4(pos.xyz, 1.0), W)).xyz;");
- }
- else if (vert.wposition) {
- node_shader_add_uniform(vert, "mat4 W", "_world_matrix");
- node_shader_write_attrib(vert, "vec3 wposition = vec4(mul(vec4(pos.xyz, 1.0), W)).xyz;");
- }
- if (frag.vposition) {
- node_shader_add_uniform(vert, "mat4 WV", "_world_view_matrix");
- node_shader_add_out(vert, "vec3 vposition");
- node_shader_write_attrib(vert, "vposition = vec4(mul(vec4(pos.xyz, 1.0), WV)).xyz;");
- }
- if (frag.mposition) {
- node_shader_add_out(vert, "vec3 mposition");
- if (frag.ndcpos) {
- node_shader_write(vert, "mposition = (ndc.xyz / ndc.w);");
- }
- else {
- node_shader_write_attrib(vert, "mposition = pos.xyz;");
- }
- }
- if (frag.wtangent) {
- // NodeShaderadd_elem(con, "tang", "short4norm");
- // add_uniform(vert, "mat3 N", "_normal_matrix");
- node_shader_add_out(vert, "vec3 wtangent");
- // write_attrib(vert, "wtangent = normalize(mul(tang.xyz, N));");
- node_shader_write_attrib(vert, "wtangent = vec3(0.0, 0.0, 0.0);");
- }
- if (frag.vvec_cam) {
- node_shader_add_uniform(vert, "mat4 WV", "_world_view_matrix");
- node_shader_add_out(vert, "vec3 eyeDirCam");
- node_shader_write_attrib(vert, "eyeDirCam = vec4(mul(vec4(pos.xyz, 1.0), WV)).xyz; eyeDirCam.z *= -1.0;");
- node_shader_write_attrib(frag, "vec3 vVecCam = normalize(eyeDirCam);");
- }
- if (frag.vvec) {
- node_shader_add_uniform(vert, "vec3 eye", "_camera_pos");
- node_shader_add_out(vert, "vec3 eyeDir");
- node_shader_write_attrib(vert, "eyeDir = eye - wposition;");
- node_shader_write_attrib(frag, "vec3 vVec = normalize(eyeDir);");
- }
- if (frag.n) {
- node_shader_add_uniform(vert, "mat3 N", "_normal_matrix");
- node_shader_add_out(vert, "vec3 wnormal");
- node_shader_write_attrib(vert, "wnormal = mul(vec3(nor.xy, pos.w), N);");
- node_shader_write_attrib(frag, "vec3 n = normalize(wnormal);");
- }
- else if (vert.n) {
- node_shader_add_uniform(vert, "mat3 N", "_normal_matrix");
- node_shader_write_attrib(vert, "vec3 wnormal = normalize(mul(vec3(nor.xy, pos.w), N));");
- }
- if (frag.nattr) {
- node_shader_add_out(vert, "vec3 nAttr");
- node_shader_write_attrib(vert, "nAttr = vec3(nor.xy, pos.w);");
- }
- if (frag.dotnv) {
- node_shader_write_attrib(frag, "float dotNV = max(dot(n, vVec), 0.0);");
- }
- if (frag.wvpposition) {
- node_shader_add_out(vert, "vec4 wvpposition");
- node_shader_write_end(vert, "wvpposition = gl_Position;");
- }
- if (node_shader_context_is_elem(con, "col")) {
- node_shader_add_out(vert, "vec3 vcolor");
- node_shader_write_attrib(vert, "vcolor = col.rgb;");
- }
- }
- function parser_material_parse_output(node: zui_node_t): shader_out_t {
- if (parser_material_parse_surface || parser_material_parse_opacity) {
- return parser_material_parse_shader_input(node.inputs[0]);
- }
- return null;
- // Parse volume, displacement..
- }
- function parser_material_parse_output_pbr(node: zui_node_t): shader_out_t {
- if (parser_material_parse_surface || parser_material_parse_opacity) {
- return parser_material_parse_shader(node, null);
- }
- return null;
- // Parse volume, displacement..
- }
- function parser_material_get_group(name: string): zui_node_canvas_t {
- for (let i: i32 = 0; i < project_material_groups.length; ++i) {
- let g: node_group_t = project_material_groups[i];
- if (g.canvas.name == name) {
- return g.canvas;
- }
- }
- return null;
- }
- function parser_material_push_group(g: zui_node_canvas_t) {
- array_push(parser_material_canvases, g);
- parser_material_nodes = g.nodes;
- parser_material_links = g.links;
- }
- function parser_material_pop_group() {
- parser_material_canvases.pop();
- let g: zui_node_canvas_t = parser_material_canvases[parser_material_canvases.length - 1];
- parser_material_nodes = g.nodes;
- parser_material_links = g.links;
- }
- function parser_material_parse_group(node: zui_node_t, socket: zui_node_socket_t): string {
- array_push(parser_material_parents, node); // Entering group
- parser_material_push_group(parser_material_get_group(node.name));
- let output_node: zui_node_t = parser_material_node_by_type(parser_material_nodes, "GROUP_OUTPUT");
- if (output_node == null) {
- return null;
- }
- let index: i32 = parser_material_socket_index(node, socket);
- let inp: zui_node_socket_t = output_node.inputs[index];
- let out_group: string = parser_material_parse_input(inp);
- parser_material_parents.pop();
- parser_material_pop_group();
- return out_group;
- }
- function parser_material_parse_group_input(node: zui_node_t, socket: zui_node_socket_t): string {
- let parent: zui_node_t = parser_material_parents.pop(); // Leaving group
- parser_material_pop_group();
- let index: i32 = parser_material_socket_index(node, socket);
- let inp: zui_node_socket_t = parent.inputs[index];
- let res: string = parser_material_parse_input(inp);
- array_push(parser_material_parents, parent); // Return to group
- parser_material_push_group(parser_material_get_group(parent.name));
- return res;
- }
- function parser_material_parse_input(inp: zui_node_socket_t): string {
- if (inp.type == "RGB") {
- return parser_material_parse_vector_input(inp);
- }
- else if (inp.type == "RGBA") {
- return parser_material_parse_vector_input(inp);
- }
- else if (inp.type == "VECTOR") {
- return parser_material_parse_vector_input(inp);
- }
- else if (inp.type == "VALUE") {
- return parser_material_parse_value_input(inp);
- }
- return null;
- }
- function parser_material_parse_shader_input(inp: zui_node_socket_t): shader_out_t {
- let l: zui_node_link_t = parser_material_get_input_link(inp);
- let from_node: zui_node_t = l != null ? parser_material_get_node(l.from_id) : null;
- if (from_node != null) {
- if (from_node.type == "REROUTE") {
- return parser_material_parse_shader_input(from_node.inputs[0]);
- }
- return parser_material_parse_shader(from_node, from_node.outputs[l.from_socket]);
- }
- else {
- return {
- out_basecol: "vec3(0.8, 0.8, 0.8)",
- out_roughness: "0.0",
- out_metallic: "0.0",
- out_occlusion: "1.0",
- out_opacity: "1.0",
- out_height: "0.0",
- out_emission: "0.0",
- out_subsurface: "0.0"
- };
- }
- }
- function parser_material_parse_shader(node: zui_node_t, socket: zui_node_socket_t): shader_out_t {
- let sout: shader_out_t = {
- out_basecol: "vec3(0.8, 0.8, 0.8)",
- out_roughness: "0.0",
- out_metallic: "0.0",
- out_occlusion: "1.0",
- out_opacity: "1.0",
- out_height: "0.0",
- out_emission: "0.0",
- out_subsurface: "0.0"
- }
- if (node.type == "OUTPUT_MATERIAL_PBR") {
- if (parser_material_parse_surface) {
- // Normal - parsed first to retrieve uv coords
- parse_normal_map_color_input(node.inputs[5]);
- // Base color
- parser_material_parsing_basecolor = true;
- sout.out_basecol = parser_material_parse_vector_input(node.inputs[0]);
- parser_material_parsing_basecolor = false;
- // Occlusion
- sout.out_occlusion = parser_material_parse_value_input(node.inputs[2]);
- // Roughness
- sout.out_roughness = parser_material_parse_value_input(node.inputs[3]);
- // Metallic
- sout.out_metallic = parser_material_parse_value_input(node.inputs[4]);
- // Emission
- if (parser_material_parse_emission) {
- sout.out_emission = parser_material_parse_value_input(node.inputs[6]);
- }
- // Subsurface
- if (parser_material_parse_subsurface) {
- sout.out_subsurface = parser_material_parse_value_input(node.inputs[8]);
- }
- }
- if (parser_material_parse_opacity) {
- sout.out_opacity = parser_material_parse_value_input(node.inputs[1]);
- }
- // Displacement / Height
- if (node.inputs.length > 7 && parser_material_parse_height) {
- if (!parser_material_parse_height_as_channel) {
- parser_material_curshader = parser_material_vert;
- }
- sout.out_height = parser_material_parse_value_input(node.inputs[7]);
- if (!parser_material_parse_height_as_channel) {
- parser_material_curshader = parser_material_frag;
- }
- }
- }
- return sout;
- }
- function parser_material_parse_vector_input(inp: zui_node_socket_t): string {
- let l: zui_node_link_t = parser_material_get_input_link(inp);
- let from_node: zui_node_t = l != null ? parser_material_get_node(l.from_id) : null;
- if (from_node != null) {
- if (from_node.type == "REROUTE") {
- return parser_material_parse_vector_input(from_node.inputs[0]);
- }
- let res_var: string = parser_material_write_result(l);
- let st: string = from_node.outputs[l.from_socket].type;
- if (st == "RGB" || st == "RGBA" || st == "VECTOR") {
- return res_var;
- }
- else {// VALUE
- return parser_material_to_vec3(res_var);
- }
- }
- else {
- if (inp.type == "VALUE") { // Unlinked reroute
- return parser_material_vec3(f32_array_create_xyz(0.0, 0.0, 0.0));
- }
- else {
- return parser_material_vec3(inp.default_value);
- }
- }
- }
- function parser_material_parse_vector(node: zui_node_t, socket: zui_node_socket_t): string {
- if (node.type == "GROUP") {
- return parser_material_parse_group(node, socket);
- }
- else if (node.type == "GROUP_INPUT") {
- return parser_material_parse_group_input(node, socket);
- }
- else if (node.type == "ATTRIBUTE") {
- if (socket == node.outputs[0]) { // Color
- if (parser_material_curshader.context.allow_vcols) {
- node_shader_context_add_elem(parser_material_curshader.context, "col", "short4norm"); // Vcols only for now
- return "vcolor";
- }
- else {
- return("vec3(0.0, 0.0, 0.0)");
- }
- }
- else { // Vector
- node_shader_context_add_elem(parser_material_curshader.context, "tex", "short2norm"); // UVMaps only for now
- return "vec3(texCoord.x, texCoord.y, 0.0)";
- }
- }
- else if (node.type == "VERTEX_COLOR") {
- if (parser_material_curshader.context.allow_vcols) {
- node_shader_context_add_elem(parser_material_curshader.context, "col", "short4norm");
- return "vcolor";
- }
- else {
- return("vec3(0.0, 0.0, 0.0)");
- }
- }
- else if (node.type == "RGB") {
- return parser_material_vec3(socket.default_value);
- }
- else if (node.type == "TEX_BRICK") {
- node_shader_add_function(parser_material_curshader, str_tex_brick);
- let co: string = parser_material_get_coord(node);
- let col1: string = parser_material_parse_vector_input(node.inputs[1]);
- let col2: string = parser_material_parse_vector_input(node.inputs[2]);
- let col3: string = parser_material_parse_vector_input(node.inputs[3]);
- let scale: string = parser_material_parse_value_input(node.inputs[4]);
- let res: string = "tex_brick(" + co + " * " + scale + ", " + col1 + ", " + col2 + ", " + col3 + ")";
- return res;
- }
- else if (node.type == "TEX_CHECKER") {
- node_shader_add_function(parser_material_curshader, str_tex_checker);
- let co: string = parser_material_get_coord(node);
- let col1: string = parser_material_parse_vector_input(node.inputs[1]);
- let col2: string = parser_material_parse_vector_input(node.inputs[2]);
- let scale: string = parser_material_parse_value_input(node.inputs[3]);
- let res: string = "tex_checker(" + co + ", " + col1 + ", " + col2 + ", " + scale + ")";
- return res;
- }
- else if (node.type == "TEX_GRADIENT") {
- let co: string = parser_material_get_coord(node);
- let but: zui_node_button_t = node.buttons[0]; //gradient_type;
- let grad: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- grad = string_replace_all(grad, " ", "_");
- let f: string = parser_material_get_gradient(grad, co);
- let res: string = parser_material_to_vec3("clamp(" + f + ", 0.0, 1.0)");
- return res;
- }
- else if (node.type == "TEX_IMAGE") {
- // Already fetched
- if (array_index_of(parser_material_parsed, parser_material_res_var_name(node, node.outputs[1])) >= 0) { // TODO: node.outputs[0]
- let varname: string = parser_material_store_var_name(node);
- return varname + ".rgb";
- }
- let tex_name: string = parser_material_node_name(node);
- let tex: bind_tex_t = parser_material_make_texture(node, tex_name);
- if (tex != null) {
- let color_space: i32 = node.buttons[1].default_value[0];
- let texstore: string = parser_material_texture_store(node, tex, tex_name, color_space);
- return texstore + ".rgb";
- }
- else {
- let tex_store: string = parser_material_store_var_name(node); // Pink color for missing texture
- node_shader_write(parser_material_curshader, "vec4 " + tex_store + " = vec4(1.0, 0.0, 1.0, 1.0);");
- return tex_store + ".rgb";
- }
- }
- else if (node.type == "TEX_MAGIC") {
- node_shader_add_function(parser_material_curshader, str_tex_magic);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let res: string = "tex_magic(" + co + " * " + scale + " * 4.0)";
- return res;
- }
- else if (node.type == "TEX_NOISE") {
- node_shader_add_function(parser_material_curshader, str_tex_noise);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let res: string = "vec3(tex_noise(" + co + " * " + scale + "), tex_noise(" + co + " * " + scale + " + 0.33), tex_noise(" + co + " * " + scale + " + 0.66))";
- return res;
- }
- else if (node.type == "TEX_VORONOI") {
- node_shader_add_function(parser_material_curshader, str_tex_voronoi);
- node_shader_add_uniform(parser_material_curshader, "sampler2D snoise256", "$noise256.k");
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let but: zui_node_button_t = node.buttons[0]; //coloring;
- let coloring: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- coloring = string_replace_all(coloring, " ", "_");
- let res: string = "";
- if (coloring == "INTENSITY") {
- res = parser_material_to_vec3("tex_voronoi(" + co + " * " + scale + ", texturePass(snoise256)).a");
- }
- else { // Cells
- res = "tex_voronoi(" + co + " * " + scale + ", texturePass(snoise256)).rgb";
- }
- return res;
- }
- else if (node.type == "TEX_WAVE") {
- node_shader_add_function(parser_material_curshader, str_tex_wave);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let res: string = parser_material_to_vec3("tex_wave_f(" + co + " * " + scale + ")");
- return res;
- }
- else if (node.type == "BRIGHTCONTRAST") {
- let out_col: string = parser_material_parse_vector_input(node.inputs[0]);
- let bright: string = parser_material_parse_value_input(node.inputs[1]);
- let contr: string = parser_material_parse_value_input(node.inputs[2]);
- node_shader_add_function(parser_material_curshader, str_brightcontrast);
- return "brightcontrast(" + out_col + ", " + bright + ", " + contr + ")";
- }
- else if (node.type == "GAMMA") {
- let out_col: string = parser_material_parse_vector_input(node.inputs[0]);
- let gamma: string = parser_material_parse_value_input(node.inputs[1]);
- return "pow(" + out_col + ", " + parser_material_to_vec3(gamma) + ")";
- }
- else if (node.type == "DIRECT_WARP") {
- if (parser_material_warp_passthrough) {
- return parser_material_parse_vector_input(node.inputs[0]);
- }
- let angle: string = parser_material_parse_value_input(node.inputs[1], true);
- let mask: string = parser_material_parse_value_input(node.inputs[2], true);
- let tex_name: string = "texwarp_" + parser_material_node_name(node);
- node_shader_add_uniform(parser_material_curshader, "sampler2D " + tex_name, "_" + tex_name);
- let store: string = parser_material_store_var_name(node);
- node_shader_write(parser_material_curshader, "float " + store + "_rad = " + angle + " * (" + math_pi() + " / 180);");
- node_shader_write(parser_material_curshader, "float " + store + "_x = cos(" + store + "_rad);");
- node_shader_write(parser_material_curshader, "float " + store + "_y = sin(" + store + "_rad);");
- return "texture(" + tex_name + ", texCoord + vec2(" + store + "_x, " + store + "_y) * " + mask + ").rgb;";
- }
- else if (node.type == "BLUR") {
- if (parser_material_blur_passthrough) {
- return parser_material_parse_vector_input(node.inputs[0]);
- }
- let strength: string = parser_material_parse_value_input(node.inputs[1]);
- if (strength == "0.0") {
- return "vec3(0.0, 0.0, 0.0)";
- }
- let steps: string = "int(" + strength + " * 10 + 1)";
- let tex_name: string = "texblur_" + parser_material_node_name(node);
- node_shader_add_uniform(parser_material_curshader, "sampler2D " + tex_name, "_" + tex_name);
- let store: string = parser_material_store_var_name(node);
- node_shader_write(parser_material_curshader, "vec3 " + store + "_res = vec3(0.0, 0.0, 0.0);");
- node_shader_write(parser_material_curshader, "for (int i = -" + steps + "; i <= " + steps + "; ++i) {");
- node_shader_write(parser_material_curshader, "for (int j = -" + steps + "; j <= " + steps + "; ++j) {");
- node_shader_write(parser_material_curshader, store + "_res += texture(" + tex_name + ", texCoord + vec2(i, j) / vec2(textureSize(" + tex_name + ", 0))).rgb;");
- node_shader_write(parser_material_curshader, "}");
- node_shader_write(parser_material_curshader, "}");
- node_shader_write(parser_material_curshader, store + "_res /= (" + steps + " * 2 + 1) * (" + steps + " * 2 + 1);");
- return store + "_res";
- }
- else if (node.type == "HUE_SAT") {
- node_shader_add_function(parser_material_curshader, str_hue_sat);
- let hue: string = parser_material_parse_value_input(node.inputs[0]);
- let sat: string = parser_material_parse_value_input(node.inputs[1]);
- let val: string = parser_material_parse_value_input(node.inputs[2]);
- let fac: string = parser_material_parse_value_input(node.inputs[3]);
- let col: string = parser_material_parse_vector_input(node.inputs[4]);
- return "hue_sat(" + col + ", vec4(" + hue + "-0.5, " + sat + ", " + val + ", 1.0-" + fac + "))";
- }
- else if (node.type == "INVERT") {
- let fac: string = parser_material_parse_value_input(node.inputs[0]);
- let out_col: string = parser_material_parse_vector_input(node.inputs[1]);
- return "mix(" + out_col + ", vec3(1.0, 1.0, 1.0) - (" + out_col + "), " + fac + ")";
- }
- else if (node.type == "MIX_RGB") {
- let fac: string = parser_material_parse_value_input(node.inputs[0]);
- let fac_var: string = parser_material_node_name(node) + "_fac";
- node_shader_write(parser_material_curshader, "float " + fac_var + " = " + fac + ";");
- let col1: string = parser_material_parse_vector_input(node.inputs[1]);
- let col2: string = parser_material_parse_vector_input(node.inputs[2]);
- let but: zui_node_button_t = node.buttons[0]; // blend_type
- let blend: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- blend = string_replace_all(blend, " ", "_");
- let use_clamp: bool = node.buttons[1].default_value[0] > 0;
- let out_col: string = "";
- if (blend == "MIX") {
- out_col = "mix(" + col1 + ", " + col2 + ", " + fac_var + ")";
- }
- else if (blend == "DARKEN") {
- out_col = "min(" + col1 + ", " + col2 + " * " + fac_var + ")";
- }
- else if (blend == "MULTIPLY") {
- out_col = "mix(" + col1 + ", " + col1 + " * " + col2 + ", " + fac_var + ")";
- }
- else if (blend == "BURN") {
- out_col = "mix(" + col1 + ", vec3(1.0, 1.0, 1.0) - (vec3(1.0, 1.0, 1.0) - " + col1 + ") / " + col2 + ", " + fac_var + ")";
- }
- else if (blend == "LIGHTEN") {
- out_col = "max(" + col1 + ", " + col2 + " * " + fac_var + ")";
- }
- else if (blend == "SCREEN") {
- out_col = "(vec3(1.0, 1.0, 1.0) - (" + parser_material_to_vec3("1.0 - " + fac_var + "") + " + " + fac_var + " * (vec3(1.0, 1.0, 1.0) - " + col2 + ")) * (vec3(1.0, 1.0, 1.0) - " + col1 + "))";
- }
- else if (blend == "DODGE") {
- out_col = "mix(" + col1 + ", " + col1 + " / (vec3(1.0, 1.0, 1.0) - " + col2 + "), " + fac_var + ")";
- }
- else if (blend == "ADD") {
- out_col = "mix(" + col1 + ", " + col1 + " + " + col2 + ", " + fac_var + ")";
- }
- else if (blend == "OVERLAY") {
- out_col = "mix(" + col1 + ", vec3( \
- " + col1 + ".r < 0.5 ? 2.0 * " + col1 + ".r * " + col2 + ".r : 1.0 - 2.0 * (1.0 - " + col1 + ".r) * (1.0 - " + col2 + ".r), \
- " + col1 + ".g < 0.5 ? 2.0 * " + col1 + ".g * " + col2 + ".g : 1.0 - 2.0 * (1.0 - " + col1 + ".g) * (1.0 - " + col2 + ".g), \
- " + col1 + ".b < 0.5 ? 2.0 * " + col1 + ".b * " + col2 + ".b : 1.0 - 2.0 * (1.0 - " + col1 + ".b) * (1.0 - " + col2 + ".b) \
- ), " + fac_var + ")";
- }
- else if (blend == "SOFT_LIGHT") {
- out_col = "((1.0 - " + fac_var + ") * " + col1 + " + " + fac_var + " * ((vec3(1.0, 1.0, 1.0) - " + col1 + ") * " + col2 + " * " + col1 + " + " + col1 + " * (vec3(1.0, 1.0, 1.0) - (vec3(1.0, 1.0, 1.0) - " + col2 + ") * (vec3(1.0, 1.0, 1.0) - " + col1 + "))))";
- }
- else if (blend == "LINEAR_LIGHT") {
- out_col = "(" + col1 + " + " + fac_var + " * (vec3(2.0, 2.0, 2.0) * (" + col2 + " - vec3(0.5, 0.5, 0.5))))";
- }
- else if (blend == "DIFFERENCE") {
- out_col = "mix(" + col1 + ", abs(" + col1 + " - " + col2 + "), " + fac_var + ")";
- }
- else if (blend == "SUBTRACT") {
- out_col = "mix(" + col1 + ", " + col1 + " - " + col2 + ", " + fac_var + ")";
- }
- else if (blend == "DIVIDE") {
- let eps: f32 = 0.000001;
- col2 = "max(" + col2 + ", vec3(" + eps + ", " + eps + ", " + eps + "))";
- out_col = "(" + parser_material_to_vec3("(1.0 - " + fac_var + ") * " + col1 + " + " + fac_var + " * " + col1 + " / " + col2) + ")";
- }
- else if (blend == "HUE") {
- node_shader_add_function(parser_material_curshader, str_hue_sat);
- out_col = "mix(" + col1 + ", hsv_to_rgb(vec3(rgb_to_hsv(" + col2 + ").r, rgb_to_hsv(" + col1 + ").g, rgb_to_hsv(" + col1 + ").b)), " + fac_var + ")";
- }
- else if (blend == "SATURATION") {
- node_shader_add_function(parser_material_curshader, str_hue_sat);
- out_col = "mix(" + col1 + ", hsv_to_rgb(vec3(rgb_to_hsv(" + col1 + ").r, rgb_to_hsv(" + col2 + ").g, rgb_to_hsv(" + col1 + ").b)), " + fac_var + ")";
- }
- else if (blend == "COLOR") {
- node_shader_add_function(parser_material_curshader, str_hue_sat);
- out_col = "mix(" + col1 + ", hsv_to_rgb(vec3(rgb_to_hsv(" + col2 + ").r, rgb_to_hsv(" + col2 + ").g, rgb_to_hsv(" + col1 + ").b)), " + fac_var + ")";
- }
- else if (blend == "VALUE") {
- node_shader_add_function(parser_material_curshader, str_hue_sat);
- out_col = "mix(" + col1 + ", hsv_to_rgb(vec3(rgb_to_hsv(" + col1 + ").r, rgb_to_hsv(" + col1 + ").g, rgb_to_hsv(" + col2 + ").b)), " + fac_var + ")";
- }
- if (use_clamp) {
- return "clamp(" + out_col + ", vec3(0.0, 0.0, 0.0), vec3(1.0, 1.0, 1.0))";
- }
- else {
- return out_col;
- }
- }
- else if (node.type == "QUANTIZE") {
- let strength: string = parser_material_parse_value_input(node.inputs[0]);
- let col: string = parser_material_parse_vector_input(node.inputs[1]);
- return "(floor(100.0 * " + strength + " * " + col + ") / (100.0 * " + strength + "))";
- }
- else if (node.type == "VALTORGB") { // ColorRamp
- let fac: string = parser_material_parse_value_input(node.inputs[0]);
- let interp: string = node.buttons[0].data[0] == 0 ? "LINEAR" : "CONSTANT";
- let elems: f32[][] = null; //node.buttons[0].default_value;
- if (elems.length == 1) {
- // return parser_material_vec3(elems[0]);
- }
- // Write cols array
- let cols_var: string = parser_material_node_name(node) + "_cols";
- node_shader_write(parser_material_curshader, "vec3 " + cols_var + "[" + elems.length + "];"); // TODO: Make const
- for (let i: i32 = 0; i < elems.length; ++i) {
- // node_shader_write(parser_material_curshader, cols_var + "[" + i + "] = " + parser_material_vec3(elems[i]) + ";");
- }
- // Get index
- let fac_var: string = parser_material_node_name(node) + "_fac";
- node_shader_write(parser_material_curshader, "float " + fac_var + " = " + fac + ";");
- let index: string = "0";
- for (let i: i32 = 1; i < elems.length; ++i) {
- index += " + (" + fac_var + " > " + elems[i][4] + " ? 1 : 0)";
- }
- // Write index
- let index_var: string = parser_material_node_name(node) + "_i";
- node_shader_write(parser_material_curshader, "int " + index_var + " = " + index + ";");
- if (interp == "CONSTANT") {
- return "" + cols_var + "[" + index_var + "]";
- }
- else { // Linear
- // Write facs array
- let facs_var: string = parser_material_node_name(node) + "_facs";
- node_shader_write(parser_material_curshader, "float " + facs_var + "[" + elems + "length}];"); // TODO: Make const
- for (let i: i32 = 0; i < elems.length; ++i) {
- node_shader_write(parser_material_curshader, facs_var + "[" + i + "] = " + elems[i][4] + ";");
- }
- // Mix color
- // float f = (pos - start) * (1.0 / (finish - start))
- return "mix(" + cols_var + "[" + index_var + "], " + cols_var + "[" + index_var + " + 1], (" + fac_var + " - " + facs_var + "[" + index_var + "]) * (1.0 / (" + facs_var + "[" + index_var + " + 1] - " + facs_var + "[" + index_var + "]) ))";
- }
- }
- else if (node.type == "CURVE_VEC") {
- let fac: string = parser_material_parse_value_input(node.inputs[0]);
- let vec: string = parser_material_parse_vector_input(node.inputs[1]);
- let curves: any = node.buttons[0].default_value;
- let name: string = parser_material_node_name(node);
- let vc0: string = parser_material_vector_curve(name + "0", vec + ".x", curves[0]);
- let vc1: string = parser_material_vector_curve(name + "1", vec + ".y", curves[1]);
- let vc2: string = parser_material_vector_curve(name + "2", vec + ".z", curves[2]);
- // mapping.curves[0].points[0].handle_type // bezier curve
- return "(vec3(" + vc0 + ", " + vc1 + ", " + vc2 + ") * " + fac + ")";
- }
- else if (node.type == "CURVE_RGB") { // RGB Curves
- let fac: string = parser_material_parse_value_input(node.inputs[0]);
- let vec: string = parser_material_parse_vector_input(node.inputs[1]);
- let curves: any = node.buttons[0].default_value;
- let name: string = parser_material_node_name(node);
- // mapping.curves[0].points[0].handle_type
- let vc0: string = parser_material_vector_curve(name + "0", vec + ".x", curves[0]);
- let vc1: string = parser_material_vector_curve(name + "1", vec + ".y", curves[1]);
- let vc2: string = parser_material_vector_curve(name + "2", vec + ".z", curves[2]);
- let vc3a: string = parser_material_vector_curve(name + "3a", vec + ".x", curves[3]);
- let vc3b: string = parser_material_vector_curve(name + "3b", vec + ".y", curves[3]);
- let vc3c: string = parser_material_vector_curve(name + "3c", vec + ".z", curves[3]);
- return "(sqrt(vec3(" + vc0 + ", " + vc1 + ", " + vc2 + ") * vec3(" + vc3a + ", " + vc3b + ", " + vc3c + ")) * " + fac + ")";
- }
- else if (node.type == "COMBHSV") {
- node_shader_add_function(parser_material_curshader, str_hue_sat);
- let h: string = parser_material_parse_value_input(node.inputs[0]);
- let s: string = parser_material_parse_value_input(node.inputs[1]);
- let v: string = parser_material_parse_value_input(node.inputs[2]);
- return "hsv_to_rgb(vec3(" + h + ", " + s + ", " + v + "))";
- }
- else if (node.type == "COMBRGB") {
- let r: string = parser_material_parse_value_input(node.inputs[0]);
- let g: string = parser_material_parse_value_input(node.inputs[1]);
- let b: string = parser_material_parse_value_input(node.inputs[2]);
- return "vec3(" + r + ", " + g + ", " + b + ")";
- }
- else if (node.type == "WAVELENGTH") {
- node_shader_add_function(parser_material_curshader, str_wavelength_to_rgb);
- let wl: string = parser_material_parse_value_input(node.inputs[0]);
- node_shader_add_function(parser_material_curshader, str_wavelength_to_rgb);
- return "wavelength_to_rgb((" + wl + " - 450.0) / 150.0)";
- }
- else if (node.type == "CAMERA") {
- parser_material_curshader.vvec_cam = true;
- return "vVecCam";
- }
- else if (node.type == "LAYER") {
- let l: any = node.buttons[0].default_value;
- if (socket == node.outputs[0]) { // Base
- node_shader_add_uniform(parser_material_curshader, "sampler2D texpaint" + l, "_texpaint" + l);
- return "texture(texpaint" + l + ", texCoord).rgb";
- }
- else if (socket == node.outputs[5]) { // Normal
- node_shader_add_uniform(parser_material_curshader, "sampler2D texpaint_nor" + l, "_texpaint_nor" + l);
- return "texture(texpaint_nor" + l + ", texCoord).rgb";
- }
- }
- else if (node.type == "MATERIAL") {
- let result: string = "vec3(0.0, 0.0, 0.0)";
- let mi: any = node.buttons[0].default_value;
- if (mi >= project_materials.length) {
- return result;
- }
- let m: slot_material_t = project_materials[mi];
- let _nodes: zui_node_t[] = parser_material_nodes;
- let _links: zui_node_link_t[] = parser_material_links;
- parser_material_nodes = m.canvas.nodes;
- parser_material_links = m.canvas.links;
- array_push(parser_material_parents, node);
- let output_node: zui_node_t = parser_material_node_by_type(parser_material_nodes, "OUTPUT_MATERIAL_PBR");
- if (socket == node.outputs[0]) { // Base
- result = parser_material_parse_vector_input(output_node.inputs[0]);
- }
- else if (socket == node.outputs[5]) { // Normal
- result = parser_material_parse_vector_input(output_node.inputs[5]);
- }
- parser_material_nodes = _nodes;
- parser_material_links = _links;
- parser_material_parents.pop();
- return result;
- }
- else if (node.type == "PICKER") {
- if (socket == node.outputs[0]) { // Base
- node_shader_add_uniform(parser_material_curshader, "vec3 pickerBase", "_pickerBase");
- return "pickerBase";
- }
- else if (socket == node.outputs[5]) { // Normal
- node_shader_add_uniform(parser_material_curshader, "vec3 pickerNormal", "_pickerNormal");
- return "pickerNormal";
- }
- }
- else if (node.type == "NEW_GEOMETRY") {
- if (socket == node.outputs[0]) { // Position
- parser_material_curshader.wposition = true;
- return "wposition";
- }
- else if (socket == node.outputs[1]) { // Normal
- parser_material_curshader.n = true;
- return "n";
- }
- else if (socket == node.outputs[2]) { // Tangent
- parser_material_curshader.wtangent = true;
- return "wtangent";
- }
- else if (socket == node.outputs[3]) { // True Normal
- parser_material_curshader.n = true;
- return "n";
- }
- else if (socket == node.outputs[4]) { // Incoming
- parser_material_curshader.vvec = true;
- return "vVec";
- }
- else if (socket == node.outputs[5]) { // Parametric
- parser_material_curshader.mposition = true;
- return "mposition";
- }
- }
- else if (node.type == "OBJECT_INFO") {
- if (socket == node.outputs[0]) { // Location
- parser_material_curshader.wposition = true;
- return "wposition";
- }
- else if (socket == node.outputs[1]) { // Color
- return "vec3(0.0, 0.0, 0.0)";
- }
- }
- // else if (node.type == "PARTICLE_INFO") {
- // if (socket == node.outputs[3]) { // Location
- // return "vec3(0.0, 0.0, 0.0)";
- // }
- // else if (socket == node.outputs[5]) { // Velocity
- // return "vec3(0.0, 0.0, 0.0)";
- // }
- // else if (socket == node.outputs[6]) { // Angular Velocity
- // return "vec3(0.0, 0.0, 0.0)";
- // }
- // }
- else if (node.type == "TANGENT") {
- parser_material_curshader.wtangent = true;
- return "wtangent";
- }
- else if (node.type == "TEX_COORD") {
- if (socket == node.outputs[0]) { // Generated - bounds
- parser_material_curshader.bposition = true;
- return "bposition";
- }
- else if (socket == node.outputs[1]) { // Normal
- parser_material_curshader.n = true;
- return "n";
- }
- else if (socket == node.outputs[2]) {// UV
- node_shader_context_add_elem(parser_material_curshader.context, "tex", "short2norm");
- return "vec3(texCoord.x, texCoord.y, 0.0)";
- }
- else if (socket == node.outputs[3]) { // Object
- parser_material_curshader.mposition = true;
- return "mposition";
- }
- else if (socket == node.outputs[4]) { // Camera
- parser_material_curshader.vposition = true;
- return "vposition";
- }
- else if (socket == node.outputs[5]) { // Window
- parser_material_curshader.wvpposition = true;
- return "wvpposition.xyz";
- }
- else if (socket == node.outputs[6]) { // Reflection
- return "vec3(0.0, 0.0, 0.0)";
- }
- }
- else if (node.type == "UVMAP") {
- node_shader_context_add_elem(parser_material_curshader.context, "tex", "short2norm");
- return "vec3(texCoord.x, texCoord.y, 0.0)";
- }
- else if (node.type == "BUMP") {
- let strength: string = parser_material_parse_value_input(node.inputs[0]);
- // let distance: string = parse_value_input(node.inputs[1]);
- let height: string = parser_material_parse_value_input(node.inputs[2]);
- let nor: string = parser_material_parse_vector_input(node.inputs[3]);
- let sample_bump_res: string = parser_material_store_var_name(node) + "_bump";
- node_shader_write(parser_material_curshader, "float " + sample_bump_res + "_x = dFdx(float(" + height + ")) * (" + strength + ") * 16.0;");
- node_shader_write(parser_material_curshader, "float " + sample_bump_res + "_y = dFdy(float(" + height + ")) * (" + strength + ") * 16.0;");
- return "(normalize(vec3(" + sample_bump_res + "_x, " + sample_bump_res + "_y, 1.0) + " + nor + ") * vec3(0.5, 0.5, 0.5) + vec3(0.5, 0.5, 0.5))";
- }
- else if (node.type == "MAPPING") {
- let out: string = parser_material_parse_vector_input(node.inputs[0]);
- let node_translation: string = parser_material_parse_vector_input(node.inputs[1]);
- let node_rotation: string = parser_material_parse_vector_input(node.inputs[2]);
- let node_scale: string = parser_material_parse_vector_input(node.inputs[3]);
- if (node_scale != "vec3(1, 1, 1)") {
- out = "(" + out + " * " + node_scale + ")";
- }
- if (node_rotation != "vec3(0, 0, 0)") {
- // ZYX rotation, Z axis for now..
- let a: string = node_rotation + ".z * (3.1415926535 / 180)";
- // x * cos(theta) - y * sin(theta)
- // x * sin(theta) + y * cos(theta)
- out = "vec3(" + out + ".x * cos(" + a + ") - " + out + ".y * sin(" + a + "), " + out + ".x * sin(" + a + ") + " + out + ".y * cos(" + a + "), 0.0)";
- }
- // if node.rotation[1] != 0.0:
- // a = node.rotation[1]
- // out = "vec3({0}.x * {1} - {0}.z * {2}, {0}.x * {2} + {0}.z * {1}, 0.0)".format(out, math_cos(a), math_sin(a))
- // if node.rotation[0] != 0.0:
- // a = node.rotation[0]
- // out = "vec3({0}.y * {1} - {0}.z * {2}, {0}.y * {2} + {0}.z * {1}, 0.0)".format(out, math_cos(a), math_sin(a))
- if (node_translation != "vec3(0, 0, 0)") {
- out = "(" + out + " + " + node_translation + ")";
- }
- // if node.use_min:
- // out = "max({0}, vec3({1}, {2}, {3}))".format(out, node.min[0], node.min[1])
- // if node.use_max:
- // out = "min({0}, vec3({1}, {2}, {3}))".format(out, node.max[0], node.max[1])
- return out;
- }
- else if (node.type == "NORMAL") {
- if (socket == node.outputs[0]) {
- return parser_material_vec3(node.outputs[0].default_value);
- }
- else if (socket == node.outputs[1]) {
- let nor: string = parser_material_parse_vector_input(node.inputs[0]);
- let norout: string = parser_material_vec3(node.outputs[0].default_value);
- return parser_material_to_vec3("dot(" + norout + ", " + nor + ")");
- }
- }
- else if (node.type == "NORMAL_MAP") {
- let strength: string = parser_material_parse_value_input(node.inputs[0]);
- let norm: string = parser_material_parse_vector_input(node.inputs[1]);
- let store: string = parser_material_store_var_name(node);
- node_shader_write(parser_material_curshader, "vec3 " + store + "_texn = " + norm + " * 2.0 - 1.0;");
- node_shader_write(parser_material_curshader, "" + store + "_texn.xy = " + strength + " * " + store + "_texn.xy;");
- node_shader_write(parser_material_curshader, "" + store + "_texn = normalize(" + store + "_texn);");
- return "(0.5 * " + store + "_texn + 0.5)";
- }
- else if (node.type == "MIX_NORMAL_MAP") {
- let nm1: string = parser_material_parse_vector_input(node.inputs[0]);
- let nm2: string = parser_material_parse_vector_input(node.inputs[1]);
- let but: zui_node_button_t = node.buttons[0];
- let blend: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0])); // blend_type
- blend = string_replace_all(blend, " ", "_");
- let store: string = parser_material_store_var_name(node);
- // The blending algorithms are based on the paper "Blending in Detail" by Colin Barré-Brisebois and Stephen Hill 2012
- // https://blog.selfshadow.com/publications/blending-in-detail/
- if (blend == "PARTIAL_DERIVATIVE") { //partial derivate blending
- node_shader_write(parser_material_curshader, "vec3 " + store + "_n1 = " + nm1 + " * 2.0 - 1.0;");
- node_shader_write(parser_material_curshader, "vec3 " + store + "_n2 = " + nm2 + " * 2.0 - 1.0;");
- return "0.5 * normalize(vec3(" + store + "_n1.xy * " + store + "_n2.z + " + store + "_n2.xy * " + store + "_n1.z, " + store + "_n1.z * " + store + "_n2.z)) + 0.5;";
- }
- else if (blend == "WHITEOUT") { //whiteout blending
- node_shader_write(parser_material_curshader, "vec3 " + store + "_n1 = " + nm1 + " * 2.0 - 1.0;");
- node_shader_write(parser_material_curshader, "vec3 " + store + "_n2 = " + nm2 + " * 2.0 - 1.0;");
- return "0.5 * normalize(vec3(" + store + "_n1.xy + " + store + "_n2.xy, " + store + "_n1.z * " + store + "_n2.z)) + 0.5;";
- }
- else if (blend == "REORIENTED") { //reoriented normal mapping
- node_shader_write(parser_material_curshader, "vec3 " + store + "_n1 = " + nm1 + " * 2.0 - vec3(1.0, 1.0, 0.0);");
- node_shader_write(parser_material_curshader, "vec3 " + store + "_n2 = " + nm2 + " * vec3(-2.0, -2.0, 2.0) - vec3(-1.0, -1.0, 1.0);");
- return "0.5 * normalize(" + store + "_n1 * dot(" + store + "_n1, " + store + "_n2) - " + store + "_n2 * " + store + "_n1.z) + 0.5";
- }
- }
- else if (node.type == "VECT_TRANSFORM") {
- // type = node.vector_type
- // conv_from = node.convert_from
- // conv_to = node.convert_to
- // // Pass throuh
- // return parse_vector_input(node.inputs[0])
- }
- else if (node.type == "COMBXYZ") {
- let x: string = parser_material_parse_value_input(node.inputs[0]);
- let y: string = parser_material_parse_value_input(node.inputs[1]);
- let z: string = parser_material_parse_value_input(node.inputs[2]);
- return "vec3(" + x + ", " + y + ", " + z + ")";
- }
- else if (node.type == "VECT_MATH") {
- let vec1: string = parser_material_parse_vector_input(node.inputs[0]);
- let vec2: string = parser_material_parse_vector_input(node.inputs[1]);
- let but: zui_node_button_t = node.buttons[0]; //operation;
- let op: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- op = string_replace_all(op, " ", "_");
- if (op == "ADD") {
- return "(" + vec1 + " + " + vec2 + ")";
- }
- else if (op == "SUBTRACT") {
- return "(" + vec1 + " - " + vec2 + ")";
- }
- else if (op == "AVERAGE") {
- return "((" + vec1 + " + " + vec2 + ") / 2.0)";
- }
- else if (op == "DOT_PRODUCT") {
- return parser_material_to_vec3("dot(" + vec1 + ", " + vec2 + ")");
- }
- else if (op == "LENGTH") {
- return parser_material_to_vec3("length(" + vec1 + ")");
- }
- else if (op == "DISTANCE") {
- return parser_material_to_vec3("distance(" + vec1 + ", " + vec2 + ")");
- }
- else if (op == "CROSS_PRODUCT") {
- return "cross(" + vec1 + ", " + vec2 + ")";
- }
- else if (op == "NORMALIZE") {
- return "normalize(" + vec1 + ")";
- }
- else if (op == "MULTIPLY") {
- return "(" + vec1 + " * " + vec2 + ")";
- }
- else if (op == "DIVIDE") {
- return "vec3(" + vec1 + ".x / (" + vec2 + ".x == 0 ? 0.000001 : " + vec2 + ".x), " + vec1 + ".y / (" + vec2 + ".y == 0 ? 0.000001 : " + vec2 + ".y), " + vec1 + ".z / (" + vec2 + ".z == 0 ? 0.000001 : " + vec2 + ".z))";
- }
- else if (op == "PROJECT") {
- return "(dot(" + vec1 + ", " + vec2 + ") / dot(" + vec2 + ", " + vec2 + ") * " + vec2 + ")";
- }
- else if (op == "REFLECT") {
- return "reflect(" + vec1 + ", normalize(" + vec2 + "))";
- }
- else if (op == "SCALE") {
- return "(" + vec2 + ".x * " + vec1 + ")";
- }
- else if (op == "ABSOLUTE") {
- return "abs(" + vec1 + ")";
- }
- else if (op == "MINIMUM") {
- return "min(" + vec1 + ", " + vec2 + ")";
- }
- else if (op == "MAXIMUM") {
- return "max(" + vec1 + ", " + vec2 + ")";
- }
- else if (op == "FLOOR") {
- return "floor(" + vec1 + ")";
- }
- else if (op == "CEIL") {
- return "ceil(" + vec1 + ")";
- }
- else if (op == "FRACTION") {
- return "fract(" + vec1 + ")";
- }
- else if (op == "MODULO") {
- return "mod(" + vec1 + ", " + vec2 + ")";
- }
- else if(op == "SNAP") {
- return "(floor(" + vec1 + " / " + vec2 + ") * " + vec2 + ")";
- }
- else if (op == "SINE") {
- return "sin(" + vec1 + ")";
- }
- else if (op == "COSINE") {
- return "cos(" + vec1 + ")";
- }
- else if (op == "TANGENT") {
- return "tan(" + vec1 + ")";
- }
- }
- else if (node.type == "Displacement") {
- let height: string = parser_material_parse_value_input(node.inputs[0]);
- return parser_material_to_vec3(height);
- }
- else if (map_get(parser_material_custom_nodes, node.type) != null) {
- return map_get(parser_material_custom_nodes, node.type)(node, socket);
- }
- return "vec3(0.0, 0.0, 0.0)";
- }
- function parse_normal_map_color_input(inp: zui_node_socket_t) {
- parser_material_frag.write_normal++;
- parser_material_out_normaltan = parser_material_parse_vector_input(inp);
- if (!parser_material_arm_export_tangents) {
- node_shader_write(parser_material_frag, "vec3 texn = (" + parser_material_out_normaltan + ") * 2.0 - 1.0;");
- node_shader_write(parser_material_frag, "texn.y = -texn.y;");
- if (!parser_material_cotangent_frame_written) {
- parser_material_cotangent_frame_written = true;
- node_shader_add_function(parser_material_frag, str_cotangent_frame);
- }
- parser_material_frag.n = true;
- ///if (krom_direct3d11 || krom_direct3d12 || krom_metal || krom_vulkan)
- node_shader_write(parser_material_frag, "mat3 TBN = cotangentFrame(n, vVec, texCoord);");
- ///else
- node_shader_write(parser_material_frag, "mat3 TBN = cotangentFrame(n, -vVec, texCoord);");
- ///end
- node_shader_write(parser_material_frag, "n = mul(normalize(texn), TBN);");
- }
- parser_material_frag.write_normal--;
- }
- function parser_material_parse_value_input(inp: zui_node_socket_t, vector_as_grayscale: bool = false) : string {
- let l: zui_node_link_t = parser_material_get_input_link(inp);
- let from_node: zui_node_t = l != null ? parser_material_get_node(l.from_id) : null;
- if (from_node != null) {
- if (from_node.type == "REROUTE") {
- return parser_material_parse_value_input(from_node.inputs[0]);
- }
- let res_var: string = parser_material_write_result(l);
- let st: string = from_node.outputs[l.from_socket].type;
- if (st == "RGB" || st == "RGBA" || st == "VECTOR") {
- if (vector_as_grayscale) {
- return "dot(" + res_var + ".rbg, vec3(0.299, 0.587, 0.114))";
- }
- else {
- return res_var + ".x";
- }
- }
- else { // VALUE
- return res_var;
- }
- }
- else {
- return parser_material_vec1(inp.default_value[0]);
- }
- }
- function parser_material_parse_value(node: zui_node_t, socket: zui_node_socket_t): string {
- if (node.type == "GROUP") {
- return parser_material_parse_group(node, socket);
- }
- else if (node.type == "GROUP_INPUT") {
- return parser_material_parse_group_input(node, socket);
- }
- else if (node.type == "ATTRIBUTE") {
- node_shader_add_uniform(parser_material_curshader, "float time", "_time");
- return "time";
- }
- else if (node.type == "VERTEX_COLOR") {
- return "1.0";
- }
- else if (node.type == "WIREFRAME") {
- node_shader_add_uniform(parser_material_curshader, "sampler2D texuvmap", "_texuvmap");
- // let use_pixel_size: bool = node.buttons[0].default_value == "true";
- // let pixel_size: f32 = parse_value_input(node.inputs[0]);
- return "textureLod(texuvmap, texCoord, 0.0).r";
- }
- else if (node.type == "CAMERA") {
- if (socket == node.outputs[1]) { // View Z Depth
- node_shader_add_uniform(parser_material_curshader, "vec2 cameraProj", "_camera_plane_proj");
- ///if (krom_direct3d11 || krom_direct3d12 || krom_metal || krom_vulkan)
- parser_material_curshader.wvpposition = true;
- return "(cameraProj.y / ((wvpposition.z / wvpposition.w) - cameraProj.x))";
- ///else
- return "(cameraProj.y / (gl_FragCoord.z - cameraProj.x))";
- ///end
- }
- else { // View Distance
- node_shader_add_uniform(parser_material_curshader, "vec3 eye", "_camera_pos");
- parser_material_curshader.wposition = true;
- return "distance(eye, wposition)";
- }
- }
- else if (node.type == "LAYER") {
- let l: any = node.buttons[0].default_value;
- if (socket == node.outputs[1]) { // Opac
- node_shader_add_uniform(parser_material_curshader, "sampler2D texpaint" + l, "_texpaint" + l);
- return "texture(texpaint" + l + ", texCoord).a";
- }
- else if (socket == node.outputs[2]) { // Occ
- node_shader_add_uniform(parser_material_curshader, "sampler2D texpaint_pack" + l, "_texpaint_pack" + l);
- return "texture(texpaint_pack" + l + ", texCoord).r";
- }
- else if (socket == node.outputs[3]) { // Rough
- node_shader_add_uniform(parser_material_curshader, "sampler2D texpaint_pack" + l, "_texpaint_pack" + l);
- return "texture(texpaint_pack" + l + ", texCoord).g";
- }
- else if (socket == node.outputs[4]) { // Metal
- node_shader_add_uniform(parser_material_curshader, "sampler2D texpaint_pack" + l, "_texpaint_pack" + l);
- return "texture(texpaint_pack" + l + ", texCoord).b";
- }
- else if (socket == node.outputs[7]) { // Height
- node_shader_add_uniform(parser_material_curshader, "sampler2D texpaint_pack" + l, "_texpaint_pack" + l);
- return "texture(texpaint_pack" + l + ", texCoord).a";
- }
- }
- else if (node.type == "LAYER_MASK") {
- if (socket == node.outputs[0]) {
- let l: any = node.buttons[0].default_value;
- node_shader_add_uniform(parser_material_curshader, "sampler2D texpaint" + l, "_texpaint" + l);
- return "texture(texpaint" + l + ", texCoord).r";
- }
- }
- else if (node.type == "MATERIAL") {
- let result: string = "0.0";
- let mi: any = node.buttons[0].default_value;
- if (mi >= project_materials.length) return result;
- let m: slot_material_t = project_materials[mi];
- let _nodes: zui_node_t[] = parser_material_nodes;
- let _links: zui_node_link_t[] = parser_material_links;
- parser_material_nodes = m.canvas.nodes;
- parser_material_links = m.canvas.links;
- array_push(parser_material_parents, node);
- let output_node: zui_node_t = parser_material_node_by_type(parser_material_nodes, "OUTPUT_MATERIAL_PBR");
- if (socket == node.outputs[1]) { // Opac
- result = parser_material_parse_value_input(output_node.inputs[1]);
- }
- else if (socket == node.outputs[2]) { // Occ
- result = parser_material_parse_value_input(output_node.inputs[2]);
- }
- else if (socket == node.outputs[3]) { // Rough
- result = parser_material_parse_value_input(output_node.inputs[3]);
- }
- else if (socket == node.outputs[4]) { // Metal
- result = parser_material_parse_value_input(output_node.inputs[4]);
- }
- else if (socket == node.outputs[7]) { // Height
- result = parser_material_parse_value_input(output_node.inputs[7]);
- }
- parser_material_nodes = _nodes;
- parser_material_links = _links;
- parser_material_parents.pop();
- return result;
- }
- else if (node.type == "PICKER") {
- if (socket == node.outputs[1]) {
- node_shader_add_uniform(parser_material_curshader, "float pickerOpacity", "_pickerOpacity");
- return "pickerOpacity";
- }
- else if (socket == node.outputs[2]) {
- node_shader_add_uniform(parser_material_curshader, "float pickerOcclusion", "_pickerOcclusion");
- return "pickerOcclusion";
- }
- else if (socket == node.outputs[3]) {
- node_shader_add_uniform(parser_material_curshader, "float pickerRoughness", "_pickerRoughness");
- return "pickerRoughness";
- }
- else if (socket == node.outputs[4]) {
- node_shader_add_uniform(parser_material_curshader, "float pickerMetallic", "_pickerMetallic");
- return "pickerMetallic";
- }
- else if (socket == node.outputs[7]) {
- node_shader_add_uniform(parser_material_curshader, "float pickerHeight", "_pickerHeight");
- return "pickerHeight";
- }
- }
- else if (node.type == "FRESNEL") {
- let ior: string = parser_material_parse_value_input(node.inputs[0]);
- parser_material_curshader.dotnv = true;
- return "pow(1.0 - dotNV, 7.25 / " + ior + ")";
- }
- else if (node.type == "NEW_GEOMETRY") {
- if (socket == node.outputs[6]) { // Backfacing
- ///if (krom_direct3d11 || krom_direct3d12 || krom_metal || krom_vulkan)
- return "0.0"; // SV_IsFrontFace
- ///else
- return "(1.0 - float(gl_FrontFacing))";
- ///end
- }
- else if (socket == node.outputs[7]) { // Pointiness
- let strength: f32 = 1.0;
- let radius: f32 = 1.0;
- let offset: f32 = 0.0;
- let store: string = parser_material_store_var_name(node);
- parser_material_curshader.n = true;
- node_shader_write(parser_material_curshader, "vec3 " + store + "_dx = dFdx(n);");
- node_shader_write(parser_material_curshader, "vec3 " + store + "_dy = dFdy(n);");
- node_shader_write(parser_material_curshader, "float " + store + "_curvature = max(dot(" + store + "_dx, " + store + "_dx), dot(" + store + "_dy, " + store + "_dy));");
- node_shader_write(parser_material_curshader, store + "_curvature = clamp(pow(" + store + "_curvature, (1.0 / " + radius + ") * 0.25) * " + strength + " * 2.0 + " + offset + " / 10.0, 0.0, 1.0);");
- return store + "_curvature";
- }
- else if (socket == node.outputs[8]) { // Random Per Island
- return "0.0";
- }
- }
- else if (node.type == "HAIR_INFO") {
- return "0.5";
- }
- else if (node.type == "LAYER_WEIGHT") {
- let blend: string = parser_material_parse_value_input(node.inputs[0]);
- if (socket == node.outputs[0]) { // Fresnel
- parser_material_curshader.dotnv = true;
- return "clamp(pow(1.0 - dotNV, (1.0 - " + blend + ") * 10.0), 0.0, 1.0)";
- }
- else if (socket == node.outputs[1]) { // Facing
- parser_material_curshader.dotnv = true;
- return "((1.0 - dotNV) * " + blend + ")";
- }
- }
- else if (node.type == "OBJECT_INFO") {
- if (socket == node.outputs[1]) { // Object Index
- node_shader_add_uniform(parser_material_curshader, "float objectInfoIndex", "_object_info_index");
- return "objectInfoIndex";
- }
- else if (socket == node.outputs[2]) { // Material Index
- node_shader_add_uniform(parser_material_curshader, "float objectInfoMaterialIndex", "_object_info_material_index");
- return "objectInfoMaterialIndex";
- }
- else if (socket == node.outputs[3]) { // Random
- node_shader_add_uniform(parser_material_curshader, "float objectInfoRandom", "_object_info_random");
- return "objectInfoRandom";
- }
- }
- else if (node.type == "VALUE") {
- return parser_material_vec1(node.outputs[0].default_value[0]);
- }
- else if (node.type == "TEX_BRICK") {
- node_shader_add_function(parser_material_curshader, str_tex_brick);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[4]);
- let res: string = "tex_brick_f(" + co + " * " + scale + ")";
- return res;
- }
- else if (node.type == "TEX_CHECKER") {
- node_shader_add_function(parser_material_curshader, str_tex_checker);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[3]);
- let res: string = "tex_checker_f(" + co + ", " + scale + ")";
- return res;
- }
- else if (node.type == "TEX_GRADIENT") {
- let co: string = parser_material_get_coord(node);
- let but: zui_node_button_t = node.buttons[0]; //gradient_type;
- let grad: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- grad = string_replace_all(grad, " ", "_");
- let f: string = parser_material_get_gradient(grad, co);
- let res: string = "(clamp(" + f + ", 0.0, 1.0))";
- return res;
- }
- else if (node.type == "TEX_IMAGE") {
- // Already fetched
- if (array_index_of(parser_material_parsed, parser_material_res_var_name(node, node.outputs[0])) >= 0) { // TODO: node.outputs[1]
- let varname: string = parser_material_store_var_name(node);
- return varname + ".a";
- }
- let tex_name: string = parser_material_node_name(node);
- let tex: bind_tex_t = parser_material_make_texture(node, tex_name);
- if (tex != null) {
- let color_space: i32 = node.buttons[1].default_value[0];
- let texstore: string = parser_material_texture_store(node, tex, tex_name, color_space);
- return texstore + ".a";
- }
- }
- else if (node.type == "TEX_MAGIC") {
- node_shader_add_function(parser_material_curshader, str_tex_magic);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let res: string = "tex_magic_f(" + co + " * " + scale + " * 4.0)";
- return res;
- }
- else if (node.type == "TEX_MUSGRAVE") {
- node_shader_add_function(parser_material_curshader, str_tex_musgrave);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let res: string = "tex_musgrave_f(" + co + " * " + scale + " * 0.5)";
- return res;
- }
- else if (node.type == "TEX_NOISE") {
- node_shader_add_function(parser_material_curshader, str_tex_noise);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let res: string = "tex_noise(" + co + " * " + scale + ")";
- return res;
- }
- else if (node.type == "TEX_VORONOI") {
- node_shader_add_function(parser_material_curshader, str_tex_voronoi);
- node_shader_add_uniform(parser_material_curshader, "sampler2D snoise256", "$noise256.k");
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let but: zui_node_button_t = node.buttons[0]; // coloring
- let coloring: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- coloring = string_replace_all(coloring, " ", "_");
- let res: string = "";
- if (coloring == "INTENSITY") {
- res = "tex_voronoi(" + co + " * " + scale + ", texturePass(snoise256)).a";
- }
- else { // Cells
- res = "tex_voronoi(" + co + " * " + scale + ", texturePass(snoise256)).r";
- }
- return res;
- }
- else if (node.type == "TEX_WAVE") {
- node_shader_add_function(parser_material_curshader, str_tex_wave);
- let co: string = parser_material_get_coord(node);
- let scale: string = parser_material_parse_value_input(node.inputs[1]);
- let res: string = "tex_wave_f(" + co + " * " + scale + ")";
- return res;
- }
- else if (node.type == "BAKE_CURVATURE") {
- if (parser_material_bake_passthrough) {
- parser_material_bake_passthrough_strength = parser_material_parse_value_input(node.inputs[0]);
- parser_material_bake_passthrough_radius = parser_material_parse_value_input(node.inputs[1]);
- parser_material_bake_passthrough_offset = parser_material_parse_value_input(node.inputs[2]);
- return "0.0";
- }
- let tex_name: string = "texbake_" + parser_material_node_name(node);
- node_shader_add_uniform(parser_material_curshader, "sampler2D " + tex_name, "_" + tex_name);
- let store: string = parser_material_store_var_name(node);
- node_shader_write(parser_material_curshader, "float " + store + "_res = texture(" + tex_name + ", texCoord).r;");
- return store + "_res";
- }
- else if (node.type == "NORMAL") {
- let nor: string = parser_material_parse_vector_input(node.inputs[0]);
- let norout: string = parser_material_vec3(node.outputs[0].default_value);
- return "dot(" + norout + ", " + nor + ")";
- }
- else if (node.type == "COLMASK") {
- let inputColor: string = parser_material_parse_vector_input(node.inputs[0]);
- let maskColor: string = parser_material_parse_vector_input(node.inputs[1]);
- let radius: string = parser_material_parse_value_input(node.inputs[2]);
- let fuzziness: string = parser_material_parse_value_input(node.inputs[3]);
- return "clamp(1.0 - (distance(" + inputColor + ", " + maskColor + ") - " + radius + ") / max(" + fuzziness + ", " + parser_material_eps + "), 0.0, 1.0)";
- }
- else if (node.type == "MATH") {
- let val1: string = parser_material_parse_value_input(node.inputs[0]);
- let val2: string = parser_material_parse_value_input(node.inputs[1]);
- let but: zui_node_button_t = node.buttons[0]; // operation
- let op: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- op = string_replace_all(op, " ", "_");
- let use_clamp: bool = node.buttons[1].default_value[0] > 0;
- let out_val: string = "";
- if (op == "ADD") {
- out_val = "(" + val1 + " + " + val2 + ")";
- }
- else if (op == "SUBTRACT") {
- out_val = "(" + val1 + " - " + val2 + ")";
- }
- else if (op == "MULTIPLY") {
- out_val = "(" + val1 + " * " + val2 + ")";
- }
- else if (op == "DIVIDE") {
- val2 = "(" + val2 + " == 0.0 ? " + parser_material_eps + " : " + val2 + ")";
- out_val = "(" + val1 + " / " + val2 + ")";
- }
- else if (op == "POWER") {
- out_val = "pow(" + val1 + ", " + val2 + ")";
- }
- else if (op == "LOGARITHM") {
- out_val = "log(" + val1 + ")";
- }
- else if (op == "SQUARE_ROOT") {
- out_val = "sqrt(" + val1 + ")";
- }
- else if(op == "INVERSE_SQUARE_ROOT") {
- out_val = "inversesqrt(" + val1 + ")";
- }
- else if (op == "EXPONENT") {
- out_val = "exp(" + val1 + ")";
- }
- else if (op == "ABSOLUTE") {
- out_val = "abs(" + val1 + ")";
- }
- else if (op == "MINIMUM") {
- out_val = "min(" + val1 + ", " + val2 + ")";
- }
- else if (op == "MAXIMUM") {
- out_val = "max(" + val1 + ", " + val2 + ")";
- }
- else if (op == "LESS_THAN") {
- out_val = "float(" + val1 + " < " + val2 + ")";
- }
- else if (op == "GREATER_THAN") {
- out_val = "float(" + val1 + " > " + val2 + ")";
- }
- else if (op == "SIGN") {
- out_val = "sign(" + val1 + ")";
- }
- else if (op == "ROUND") {
- out_val = "floor(" + val1 + " + 0.5)";
- }
- else if (op == "FLOOR") {
- out_val = "floor(" + val1 + ")";
- }
- else if (op == "CEIL") {
- out_val = "ceil(" + val1 + ")";
- }
- else if(op == "SNAP") {
- out_val = "(floor(" + val1 + " / " + val2 + ") * " + val2 + ")";
- }
- else if (op == "TRUNCATE") {
- out_val = "trunc(" + val1 + ")";
- }
- else if (op == "FRACTION") {
- out_val = "fract(" + val1 + ")";
- }
- else if (op == "MODULO") {
- out_val = "mod(" + val1 + ", " + val2 + ")";
- }
- else if (op == "PING-PONG") {
- out_val = "((" + val2 + " != 0.0) ? abs(fract((" + val1 + " - " + val2 + ") / (" + val2 + " * 2.0)) * " + val2 + " * 2.0 - " + val2 + ") : 0.0)";
- }
- else if (op == "SINE") {
- out_val = "sin(" + val1 + ")";
- }
- else if (op == "COSINE") {
- out_val = "cos(" + val1 + ")";
- }
- else if (op == "TANGENT") {
- out_val = "tan(" + val1 + ")";
- }
- else if (op == "ARCSINE") {
- out_val = "asin(" + val1 + ")";
- }
- else if (op == "ARCCOSINE") {
- out_val = "acos(" + val1 + ")";
- }
- else if (op == "ARCTANGENT") {
- out_val = "atan(" + val1 + ")";
- }
- else if (op == "ARCTAN2") {
- out_val = "atan2(" + val1 + ", " + val2 + ")";
- }
- else if (op == "HYPERBOLIC_SINE") {
- out_val = "sinh(" + val1 + ")";
- }
- else if (op == "HYPERBOLIC_COSINE") {
- out_val = "cosh(" + val1 + ")";
- }
- else if (op == "HYPERBOLIC_TANGENT") {
- out_val = "tanh(" + val1 + ")";
- }
- else if (op == "TO_RADIANS") {
- out_val = "radians(" + val1 + ")";
- }
- else if (op == "TO_DEGREES") {
- out_val = "degrees(" + val1 + ")";
- }
- if (use_clamp) {
- return "clamp(" + out_val + ", 0.0, 1.0)";
- }
- else {
- return out_val;
- }
- }
- else if (node.type == "SCRIPT_CPU") {
- if (parser_material_script_links == null) {
- parser_material_script_links = map_create();
- }
- let script: any = node.buttons[0].default_value;
- let link: string = parser_material_node_name(node);
- map_set(parser_material_script_links, link, script);
- node_shader_add_uniform(parser_material_curshader, "float " + link, "_" + link);
- return link;
- }
- else if (node.type == "SHADER_GPU") {
- let shader: any = node.buttons[0].default_value;
- return shader == "" ? "0.0" : shader;
- }
- else if (node.type == "RGBTOBW") {
- let col: string = parser_material_parse_vector_input(node.inputs[0]);
- return "(((" + col + ".r * 0.3 + " + col + ".g * 0.59 + " + col + ".b * 0.11) / 3.0) * 2.5)";
- }
- else if (node.type == "SEPHSV") {
- node_shader_add_function(parser_material_curshader, str_hue_sat);
- let col: string = parser_material_parse_vector_input(node.inputs[0]);
- if (socket == node.outputs[0]) {
- return "rgb_to_hsv(" + col + ").r";
- }
- else if (socket == node.outputs[1]) {
- return "rgb_to_hsv(" + col + ").g";
- }
- else if (socket == node.outputs[2]) {
- return "rgb_to_hsv(" + col + ").b";
- }
- }
- else if (node.type == "SEPRGB") {
- let col: string = parser_material_parse_vector_input(node.inputs[0]);
- if (socket == node.outputs[0]) {
- return col + ".r";
- }
- else if (socket == node.outputs[1]) {
- return col + ".g";
- }
- else if (socket == node.outputs[2]) {
- return col + ".b";
- }
- }
- else if (node.type == "SEPXYZ") {
- let vec: string = parser_material_parse_vector_input(node.inputs[0]);
- if (socket == node.outputs[0]) {
- return vec + ".x";
- }
- else if (socket == node.outputs[1]) {
- return vec + ".y";
- }
- else if (socket == node.outputs[2]) {
- return vec + ".z";
- }
- }
- else if (node.type == "VECT_MATH") {
- let vec1: string = parser_material_parse_vector_input(node.inputs[0]);
- let vec2: string = parser_material_parse_vector_input(node.inputs[1]);
- let but: zui_node_button_t = node.buttons[0]; //operation;
- let op: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- op = string_replace_all(op, " ", "_");
- if (op == "DOT_PRODUCT") {
- return "dot(" + vec1 + ", " + vec2 + ")";
- }
- else if (op == "LENGTH") {
- return "length(" + vec1 + ")";
- }
- else if (op == "DISTANCE") {
- return "distance(" + vec1 + ", " + vec2 + ")";
- }
- else {
- return "0.0";
- }
- }
- else if (node.type == "CLAMP") {
- let val: string = parser_material_parse_value_input(node.inputs[0]);
- let min: string = parser_material_parse_value_input(node.inputs[1]);
- let max: string = parser_material_parse_value_input(node.inputs[2]);
- let but: zui_node_button_t = node.buttons[0]; //operation;
- let op: string = to_upper_case(u8_array_string_at(but.data, but.default_value[0]));
- op = string_replace_all(op, " ", "_");
- if (op == "MIN_MAX") {
- return "(clamp(" + val + ", " + min + ", " + max + "))";
- }
- else if (op == "RANGE") {
- return "(clamp(" + val + ", min(" + min + ", " + max + "), max(" + min + ", " + max + ")))";
- }
- }
- else if (node.type == "MAPRANGE") {
- let val: string = parser_material_parse_value_input(node.inputs[0]);
- let fmin: string = parser_material_parse_value_input(node.inputs[1]);
- let fmax: string = parser_material_parse_value_input(node.inputs[2]);
- let tmin: string = parser_material_parse_value_input(node.inputs[3]);
- let tmax: string = parser_material_parse_value_input(node.inputs[4]);
- let use_clamp: bool = node.buttons[0].default_value[0] > 0;
- let a: string = "((" + tmin + " - " + tmax + ") / (" + fmin + " - " + fmax + "))";
- let out_val: string = "(" + a + " * " + val + " + " + tmin + " - " + a + " * " + fmin + ")";
- if (use_clamp) {
- return "(clamp(" + out_val + ", " + tmin + ", " + tmax + "))";
- }
- else {
- return out_val;
- }
- }
- else if (map_get(parser_material_custom_nodes, node.type) != null) {
- return map_get(parser_material_custom_nodes, node.type)(node, socket);
- }
- return "0.0";
- }
- function parser_material_get_coord(node: zui_node_t): string {
- if (parser_material_get_input_link(node.inputs[0]) != null) {
- return parser_material_parse_vector_input(node.inputs[0]);
- }
- else {
- parser_material_curshader.bposition = true;
- return "bposition";
- }
- }
- function parser_material_get_gradient(grad: string, co: string): string {
- if (grad == "LINEAR") {
- return co + ".x";
- }
- else if (grad == "QUADRATIC") {
- return "0.0";
- }
- else if (grad == "EASING") {
- return "0.0";
- }
- else if (grad == "DIAGONAL") {
- return "(" + co + ".x + " + co + ".y) * 0.5";
- }
- else if (grad == "RADIAL") {
- return "atan2(" + co + ".x, " + co + ".y) / (3.141592 * 2.0) + 0.5";
- }
- else if (grad == "QUADRATIC_SPHERE") {
- return "0.0";
- }
- else { // "SPHERICAL"
- return "max(1.0 - sqrt(" + co + ".x * " + co + ".x + " + co + ".y * " + co + ".y + " + co + ".z * " + co + ".z), 0.0)";
- }
- }
- function parser_material_vector_curve(name: string, fac: string, points: f32_array_t[]): string {
- // Write Ys array
- let ys_var: string = name + "_ys";
- let num: i32 = points.length;
- node_shader_write(parser_material_curshader, "float " + ys_var + "[" + num + "];"); // TODO: Make const
- for (let i: i32 = 0; i < num; ++i) {
- node_shader_write(parser_material_curshader, ys_var + "[" + i + "] = " + points[i][1] + ";");
- }
- // Get index
- let fac_var: string = name + "_fac";
- node_shader_write(parser_material_curshader, "float " + fac_var + " = " + fac + ";");
- let index: string = "0";
- for (let i: i32 = 1; i < num; ++i) {
- index += " + (" + fac_var + " > " + points[i][0] + " ? 1 : 0)";
- }
- // Write index
- let index_var: string = name + "_i";
- node_shader_write(parser_material_curshader, "int " + index_var + " = " + index + ";");
- // Linear
- // Write Xs array
- let facs_var: string = name + "_xs";
- node_shader_write(parser_material_curshader, "float " + facs_var + "[" + num + "];"); // TODO: Make const
- for (let i: i32 = 0; i < num; ++i) {
- node_shader_write(parser_material_curshader, "" + facs_var + "[" + i + "] = " + points[i][0] + ";");
- }
- // Map vector
- return "mix(" + ys_var + "[" + index_var + "], " + ys_var + "[" + index_var + " + 1], (" + fac_var + " - " + facs_var + "[" + index_var + "]) * (1.0 / (" + facs_var + "[" + index_var + " + 1] - " + facs_var + "[" + index_var + "])))";
- }
- function parser_material_res_var_name(node: zui_node_t, socket: zui_node_socket_t): string {
- return parser_material_node_name(node) + "_" + parser_material_safesrc(socket.name) + "_res";
- }
- function parser_material_write_result(l: zui_node_link_t): string {
- let from_node: zui_node_t = parser_material_get_node(l.from_id);
- let from_socket: zui_node_socket_t = from_node.outputs[l.from_socket];
- let res_var: string = parser_material_res_var_name(from_node, from_socket);
- let st: string = from_socket.type;
- if (array_index_of(parser_material_parsed, res_var) < 0) {
- array_push(parser_material_parsed, res_var);
- if (st == "RGB" || st == "RGBA" || st == "VECTOR") {
- let res: string = parser_material_parse_vector(from_node, from_socket);
- if (res == null) {
- return null;
- }
- map_set(parser_material_parsed_map, res_var, res);
- node_shader_write(parser_material_curshader, "vec3 " + res_var + " = " + res + ";");
- }
- else if (st == "VALUE") {
- let res: string = parser_material_parse_value(from_node, from_socket);
- if (res == null) {
- return null;
- }
- map_set(parser_material_parsed_map, res_var, res);
- node_shader_write(parser_material_curshader, "float " + res_var + " = " + res + ";");
- }
- }
- return res_var;
- }
- function parser_material_store_var_name(node: zui_node_t): string {
- return parser_material_node_name(node) + "_store";
- }
- function parser_material_texture_store(node: zui_node_t, tex: bind_tex_t, tex_name: string, color_space: i32): string {
- array_push(parser_material_matcon.bind_textures, tex);
- node_shader_context_add_elem(parser_material_curshader.context, "tex", "short2norm");
- node_shader_add_uniform(parser_material_curshader, "sampler2D " + tex_name);
- let uv_name: string = "";
- if (parser_material_get_input_link(node.inputs[0]) != null) {
- uv_name = parser_material_parse_vector_input(node.inputs[0]);
- }
- else {
- uv_name = parser_material_tex_coord;
- }
- let tex_store: string = parser_material_store_var_name(node);
- if (parser_material_sample_keep_aspect) {
- node_shader_write(parser_material_curshader, "vec2 " + tex_store + "_size = vec2(textureSize(" + tex_name + ", 0));");
- node_shader_write(parser_material_curshader, "float " + tex_store + "_ax = " + tex_store + "_size.x / " + tex_store + "_size.y;");
- node_shader_write(parser_material_curshader, "float " + tex_store + "_ay = " + tex_store + "_size.y / " + tex_store + "_size.x;");
- node_shader_write(parser_material_curshader, "vec2 " + tex_store + "_uv = ((" + uv_name + ".xy / " + parser_material_sample_uv_scale + " - vec2(0.5, 0.5)) * vec2(max(" + tex_store + "_ay, 1.0), max(" + tex_store + "_ax, 1.0))) + vec2(0.5, 0.5);");
- node_shader_write(parser_material_curshader, "if (" + tex_store + "_uv.x < 0.0 || " + tex_store + "_uv.y < 0.0 || " + tex_store + "_uv.x > 1.0 || " + tex_store + "_uv.y > 1.0) discard;");
- node_shader_write(parser_material_curshader, "" + tex_store + "_uv *= " + parser_material_sample_uv_scale + ";");
- uv_name = tex_store + "_uv";
- }
- if (parser_material_triplanar) {
- node_shader_write(parser_material_curshader, "vec4 " + tex_store + " = vec4(0.0, 0.0, 0.0, 0.0);");
- node_shader_write(parser_material_curshader, "if (texCoordBlend.x > 0) " + tex_store + " += texture(" + tex_name + ", " + uv_name + ".xy) * texCoordBlend.x;");
- node_shader_write(parser_material_curshader, "if (texCoordBlend.y > 0) " + tex_store + " += texture(" + tex_name + ", " + uv_name + "1.xy) * texCoordBlend.y;");
- node_shader_write(parser_material_curshader, "if (texCoordBlend.z > 0) " + tex_store + " += texture(" + tex_name + ", " + uv_name + "2.xy) * texCoordBlend.z;");
- }
- else {
- if (parser_material_curshader == parser_material_frag) {
- map_set(parser_material_texture_map, tex_store, "texture(" + tex_name + ", " + uv_name + ".xy)");
- node_shader_write(parser_material_curshader, "vec4 " + tex_store + " = texture(" + tex_name + ", " + uv_name + ".xy);");
- }
- else {
- map_set(parser_material_texture_map, tex_store, "textureLod(" + tex_name + ", " + uv_name + ".xy, 0.0)");
- node_shader_write(parser_material_curshader, "vec4 " + tex_store + " = textureLod(" + tex_name + ", " + uv_name + ".xy, 0.0);");
- }
- if (!ends_with(tex.file, ".jpg")) { // Pre-mult alpha
- node_shader_write(parser_material_curshader, tex_store + ".rgb *= " + tex_store + ".a;");
- }
- }
- if (parser_material_transform_color_space) {
- // Base color socket auto-converts from sRGB to linear
- if (color_space == color_space_t.LINEAR && parser_material_parsing_basecolor) { // Linear to sRGB
- node_shader_write(parser_material_curshader, tex_store + ".rgb = pow(" + tex_store + ".rgb, vec3(2.2, 2.2, 2.2));");
- }
- else if (color_space == color_space_t.SRGB && !parser_material_parsing_basecolor) { // sRGB to linear
- node_shader_write(parser_material_curshader, tex_store + ".rgb = pow(" + tex_store + ".rgb, vec3(1.0 / 2.2, 1.0 / 2.2, 1.0 / 2.2));");
- }
- else if (color_space == color_space_t.DIRECTX_NORMAL_MAP) { // DirectX normal map to OpenGL normal map
- node_shader_write(parser_material_curshader, tex_store + ".y = 1.0 - " + tex_store + ".y;");
- }
- }
- return tex_store;
- }
- function parser_material_vec1(v: f32): string {
- ///if krom_android
- return "float(" + v + ")";
- ///else
- return v + "";
- ///end
- }
- function parser_material_vec3(v: f32_array_t): string {
- ///if krom_android
- return "vec3(float(" + v[0] + "), float(" + v[1] + "), float(" + v[2] + "))";
- ///else
- return "vec3(" + v[0] + ", " + v[1] + ", " + v[2] + ")";
- ///end
- }
- function parser_material_to_vec3(s: string): string {
- ///if (krom_direct3d11 || krom_direct3d12)
- return "(" + s + ").xxx";
- ///else
- return "vec3(" + s + ")";
- ///end
- }
- function parser_material_node_by_type(nodes: zui_node_t[], ntype: string): zui_node_t {
- for (let i: i32 = 0; i < nodes.length; ++i) {
- let n: zui_node_t = nodes[i];
- if (n.type == ntype) {
- return n;
- }
- }
- return null;
- }
- function parser_material_socket_index(node: zui_node_t, socket: zui_node_socket_t): i32 {
- for (let i: i32 = 0; i < node.outputs.length; ++i) {
- if (node.outputs[i] == socket) {
- return i;
- }
- }
- return -1;
- }
- function parser_material_node_name(node: zui_node_t, _parents: zui_node_t[] = null): string {
- if (_parents == null) {
- _parents = parser_material_parents;
- }
- let s: string = node.name;
- for (let i: i32 = 0; i < _parents.length; ++i) {
- let p: zui_node_t = _parents[i];
- s = p.name + p.id + "_" + s;
- }
- s = parser_material_safesrc(s) + node.id;
- return s;
- }
- function parser_material_safesrc(s: string): string {
- for (let i: i32 = 0; i < s.length; ++i) {
- let code: i32 = char_code_at(s, i);
- let letter: bool = (code >= 65 && code <= 90) || (code >= 97 && code <= 122);
- let digit: bool = code >= 48 && code <= 57;
- if (!letter && !digit) {
- s = string_replace_all(s, char_at(s, i), "_");
- }
- if (i == 0 && digit) {
- s = "_" + s;
- }
- }
- ///if krom_opengl
- while (string_index_of(s, "__") >= 0) {
- s = string_replace_all(s, "__", "_");
- }
- ///end
- return s;
- }
- function parser_material_enum_data(s: string): string {
- for (let i: i32 = 0; i < project_assets.length; ++i) {
- let a: asset_t = project_assets[i];
- if (a.name == s) {
- return a.file;
- }
- }
- return "";
- }
- function parser_material_make_texture(image_node: zui_node_t, tex_name: string, matname: string = null): bind_tex_t {
- let filepath: string = parser_material_enum_data(base_enum_texts(image_node.type)[image_node.buttons[0].default_value[0]]);
- if (filepath == "" || string_index_of(filepath, ".") == -1) {
- return null;
- }
- let tex: bind_tex_t = {
- name: tex_name,
- file: filepath
- };
- if (context_raw.texture_filter) {
- tex.min_filter = "anisotropic";
- tex.mag_filter = "linear";
- tex.mipmap_filter = "linear";
- tex.generate_mipmaps = true;
- }
- else {
- tex.min_filter = "point";
- tex.mag_filter = "point";
- tex.mipmap_filter = "no";
- }
- tex.u_addressing = "repeat";
- tex.v_addressing = "repeat";
- return tex;
- }
- function parser_material_is_pow(num: i32): bool {
- return ((num & (num - 1)) == 0) && num != 0;
- }
- function parser_material_asset_path(s: string): string {
- return s;
- }
- function parser_material_extract_filename(s: string): string {
- let ar: string[] = string_split(s, ".");
- return ar[ar.length - 2] + "." + ar[ar.length - 1];
- }
- function parser_material_safestr(s: string): string {
- return s;
- }
- function u8_array_string_at(a: u8_array_t, i: i32): string {
- let s: string = u8_array_to_string(a);
- let ss: string[] = string_split(s, "\0");
- return ss[i];
- }
- type shader_out_t = {
- out_basecol?: string;
- out_roughness?: string;
- out_metallic?: string;
- out_occlusion?: string;
- out_opacity?: string;
- out_height?: string;
- out_emission?: string;
- out_subsurface?: string;
- };
- enum color_space_t {
- AUTO, // sRGB for base color, otherwise linear
- LINEAR,
- SRGB,
- DIRECTX_NORMAL_MAP,
- }
- ///if is_lab
- type slot_material_t = any;
- ///end
|