141 lines
3.1 KiB
C++
141 lines
3.1 KiB
C++
// -*-C++-*-
|
|
|
|
#version 120
|
|
|
|
varying vec3 vertex;
|
|
varying vec3 viewDir;
|
|
|
|
uniform float osg_SimulationTime;
|
|
uniform float thrust_collimation;
|
|
uniform float flame_radius_fraction;
|
|
uniform float thrust_density;
|
|
uniform float base_flame_density;
|
|
uniform float shock_frequency;
|
|
uniform float noise_strength;
|
|
uniform float noise_scale;
|
|
uniform float deflection_coeff;
|
|
|
|
uniform float flame_color_low_r;
|
|
uniform float flame_color_low_g;
|
|
uniform float flame_color_low_b;
|
|
|
|
uniform float flame_color_high_r;
|
|
uniform float flame_color_high_g;
|
|
uniform float flame_color_high_b;
|
|
|
|
uniform float base_flame_r;
|
|
uniform float base_flame_g;
|
|
uniform float base_flame_b;
|
|
|
|
uniform int use_shocks;
|
|
uniform int use_noise;
|
|
|
|
float Noise2D(in vec2 coord, in float wavelength);
|
|
|
|
const int n_steps = 15;
|
|
|
|
float spherical_smoothstep (in vec3 pos)
|
|
{
|
|
|
|
float l = length(vec3 (pos.x/2.0, pos.y,pos.z) );
|
|
|
|
return 10.0 * thrust_density * base_flame_density * (1.0 - smoothstep(0.5* flame_radius_fraction, flame_radius_fraction, l));
|
|
|
|
}
|
|
|
|
|
|
|
|
float thrust_flame (in vec3 pos)
|
|
{
|
|
|
|
|
|
float noise = 0.0;
|
|
|
|
pos.z +=8.0 * deflection_coeff;
|
|
|
|
float d_rad = length(pos.yz - vec2 (0.0, deflection_coeff * pos.x * pos.x));
|
|
//float longFade = smoothstep(0.0, 5.0, pos.x) ;
|
|
float longFade = pos.x/5.0;
|
|
|
|
float density = 1.0 - longFade;
|
|
float radius = flame_radius_fraction + thrust_collimation * 1.0 * pow((pos.x+0.1),0.5);
|
|
|
|
if (d_rad > radius) {return 0.0;}
|
|
|
|
|
|
if (use_noise ==1)
|
|
{
|
|
noise = Noise2D(vec2(pos.x - osg_SimulationTime * 30.0 , d_rad), noise_scale);
|
|
}
|
|
|
|
density *= (1.0 - smoothstep(0.125, radius, d_rad)) * (1.0 - noise_strength + noise_strength* noise);
|
|
|
|
if (use_shocks == 1)
|
|
{
|
|
float shock = sin(pos.x * 10.0 * shock_frequency);
|
|
density += shock * shock * shock * shock * (1.0 - longFade) * (1.0 - smoothstep(0.25*flame_radius_fraction, 0.5*flame_radius_fraction, d_rad)) * (1.0 - smoothstep(0.0, 1.0, thrust_collimation));
|
|
}
|
|
|
|
|
|
return 10.0 * thrust_density * density / (radius/0.2);
|
|
}
|
|
|
|
|
|
|
|
void main()
|
|
{
|
|
|
|
vec3 vDir = normalize(viewDir);
|
|
|
|
float x_E, y_E, z_E;
|
|
|
|
if (vDir.x > 0.0) {x_E = 5.0;} else {x_E = 0.0;}
|
|
if (vDir.y > 0.0) {y_E = 1.0;} else {y_E = -1.0;}
|
|
if (vDir.z > 0.0) {z_E = 1.0;} else {z_E = -1.0;}
|
|
|
|
float t_x = (x_E - vertex.x) / vDir.x;
|
|
float t_y = (y_E - vertex.y) / vDir.y;
|
|
float t_z = (z_E - vertex.z) / vDir.z;
|
|
|
|
float t_min = min(t_x, t_y);
|
|
t_min = min(t_min, t_z);
|
|
|
|
|
|
float dt = t_min / float(n_steps);
|
|
|
|
vec3 step = viewDir * dt;
|
|
vec3 pos = vertex;
|
|
|
|
float density1 = 0.0;
|
|
float density2 = 0.0;
|
|
|
|
|
|
|
|
for (int i = 0; i < n_steps; i++)
|
|
{
|
|
pos = pos + step;
|
|
density1 += spherical_smoothstep(pos) * dt;
|
|
density2 += thrust_flame(pos) * dt;
|
|
}
|
|
|
|
|
|
|
|
|
|
float density = density1 + density2;
|
|
//density = clamp(density,0.0,1.0);
|
|
density = 1.0 - exp(-density);
|
|
|
|
density1 = 1.0 - exp(-density1);
|
|
density2 = 1.0 - exp(-density2);
|
|
|
|
|
|
vec3 flame_color_low = vec3 (flame_color_low_r, flame_color_low_g, flame_color_low_b);
|
|
vec3 flame_color_high = vec3 (flame_color_high_r, flame_color_high_g, flame_color_high_b);
|
|
|
|
vec3 color = mix(flame_color_low, flame_color_high, density2);
|
|
color = mix(color, vec3(0.8, 1.0, 1.0), density1);
|
|
|
|
vec4 finalColor = vec4 (color.rgb, density);
|
|
|
|
gl_FragColor = finalColor;
|
|
}
|