chuYinKouXian/4dwebBanner/shaders/include/dynamic_grass.glslv

#ifndef DYNAMIC_GRASS_GLSLV
#define DYNAMIC_GRASS_GLSLV

// #import u_scale_threshold

/*==============================================================================
                                    VARS
==============================================================================*/
#var PRECISION highp
#var DYNAMIC_GRASS 0
#var BILLBOARD 0
#var GRASS_TEXTURE_SIZE 1024.0

/*============================================================================*/

#if DYNAMIC_GRASS

#include <math.glslv>
#include <to_world.glslv>

// U along X, V along -Z (0..1)
vec2 pos_to_uv(vec3 position, float dim, vec2 base_point)
{
    vec2 pos_uv = vec2((position.x - base_point.x) / dim,
            (position.y - base_point.y) / dim);
    return fract(pos_uv);
}

vec2 uv_to_pos(vec2 uv, float dim, vec2 base_point)
{
    return vec2(uv.x * dim, uv.y * dim) + base_point;
}

// compose infinity vertex
vertex infinity_vertex()
{
    return vertex(vec3(-10000.0), vec3(0.0), vec3(0.0), vec3(0.0), vec3(0.0),
            vec3(0.0), vec3(0.0));
}

#if GLSL3
vec4 texture2D_linear(sampler2D texture_sampler, vec2 tex_coord) {
    float grass_texel_size = 1.0 / GRASS_TEXTURE_SIZE;
    vec4 x_0_y_0 = GLSL_TEXTURE(texture_sampler, tex_coord);
    vec4 x_1_y_0 = GLSL_TEXTURE(texture_sampler, tex_coord + vec2(grass_texel_size, 0.0));
    vec4 x_0_y_1 = GLSL_TEXTURE(texture_sampler, tex_coord + vec2(0.0, grass_texel_size));
    vec4 x_1_y_1 = GLSL_TEXTURE(texture_sampler, tex_coord + vec2(grass_texel_size, grass_texel_size));

    float a = fract(tex_coord.x * GRASS_TEXTURE_SIZE);

    vec4 x_int_y_0 = mix(x_0_y_0, x_1_y_0, a);
    vec4 x_int_y_1 = mix(x_0_y_1, x_1_y_1, a);

    float b = fract(tex_coord.y * GRASS_TEXTURE_SIZE);
    return mix(x_int_y_0, x_int_y_1, b);
}
#endif

// Translate grass vertex from local to world space using grass maps
vertex grass_vertex(vec3 position, vec3 tangent, vec3 shade_tan, vec3 binormal,
        vec3 normal, vec3 center, PRECISION sampler2D grass_map_depth,
        sampler2D grass_map_color, vec3 grass_map_dim, float grass_size,
        vec3 camera_eye, vec4 camera_quat, mat3 view_tsr)
{

    // get camera view angles
    vec3 cam_view = qrot(camera_quat, vec3(0.0, 0.0, -1.0));
    float sin_alpha = -cam_view.x;
    float cos_alpha = cam_view.y;

    // get world position of base point ([0.0, 0.0] (left lower) on UV)
    vec2 base_point =vec2(camera_eye.x - grass_size * (1.0 + sin_alpha) / 2.0,
            camera_eye.y - grass_size * (1.0 - cos_alpha) / 2.0);

    vec2 cen_uv = pos_to_uv(center, grass_size, base_point);

    // rounding grass plane
    float grass_edge_length = grass_size / 2.0;
    vec2 pos_from_map_center = abs(uv_to_pos(cen_uv - vec2(0.5), grass_size, vec2(0.0)));
    float max_radius = grass_edge_length * (1.0 + sqrt(2.0)) / 2.0;

    if (length(pos_from_map_center) > max_radius)
        return infinity_vertex();

    // for map sampling scale uv coords according to BATCH/MAP size ratio
    float map_size_mult = grass_size / grass_map_dim.z;
    vec2 cen_uv_scaled = (cen_uv - vec2(0.5)) * map_size_mult + vec2(0.5);
    vec3 grass_map_trans = cam_view * (grass_size / grass_map_dim.z - 1.0) / 2.0;
    cen_uv_scaled.x += grass_map_trans.x;
    cen_uv_scaled.y += grass_map_trans.y;

    // remove short grass
    vec4 scale_color = GLSL_TEXTURE(grass_map_color, cen_uv_scaled);
    float scale = scale_color.r;

    if (scale < u_scale_threshold)
        return infinity_vertex();

    // (DELTA = (HIGH - LOW)) > 0
    float height_delta = grass_map_dim.y - grass_map_dim.x;
    // in world space: HIGH - MAP * DELTA

#if GLSL3
    float height = grass_map_dim.y -
            (texture2D_linear(grass_map_depth, cen_uv_scaled)).r * height_delta;
#else
    float height = grass_map_dim.y -
            (GLSL_TEXTURE(grass_map_depth, cen_uv_scaled)).r * height_delta;
#endif

    // calculate new center and position
    vec2 pos_cen_delta = position.xy - center.xy;
    center.xy = uv_to_pos(cen_uv, grass_size, base_point);
    center.z = height;

    position.xy = center.xy + pos_cen_delta;
    // scale only grass height for now
    position.z *= scale;
    position.z += height;

    vec3 color = scale_color.gba;
# if BILLBOARD
    // NOTE: position in local space: position - center
    position -= center;
    mat3 bill_tsr = billboard_tsr(camera_eye, center, view_tsr);
    vertex world = to_world(position, center, tangent, shade_tan, binormal,
            normal, bill_tsr);
    world.center = center;
    world.color = color;
    return tbn_norm(world);
# else
    return tbn_norm(vertex(position, center, tangent, shade_tan, binormal, normal,
            color));
# endif
}

#endif

#endif