|
| 1 | +use gtk::prelude::*; |
| 2 | +use gtk::subclass::prelude::*; |
| 3 | +use gtk::{glib, gsk}; |
| 4 | + |
| 5 | +const SNOW_SHADER: &[u8] = r#" |
| 6 | +uniform float u_time; |
| 7 | +uniform vec4 u_color; |
| 8 | +
|
| 9 | +float random(vec2 co) { |
| 10 | + return fract(sin(dot(co.xy,vec2(12.9898,78.233))) * 43758.5453); |
| 11 | +} |
| 12 | +
|
| 13 | +float map(vec2 p) { |
| 14 | + float w = 0.05; |
| 15 | + p.y -= u_time * 0.05; |
| 16 | + vec2 idx = floor(p / w * 0.5 + 0.5); |
| 17 | + p.x += cos(u_time * random(idx)) * 0.01; |
| 18 | + float r = sin(random(idx)) * 0.005 + 0.001; |
| 19 | + p = mod(p + w, w * 2.) - w; |
| 20 | + float d = length(p - 0.03 * vec2(cos(idx.y), cos(idx.x))) - r; |
| 21 | + float color = d < 0.0 ? 0.5 : 0.0; |
| 22 | + color *= (sin(u_time * random(idx)) + 1.0) / 2.0; |
| 23 | + return color; |
| 24 | +} |
| 25 | +
|
| 26 | +void mainImage(out vec4 fragColor, |
| 27 | + in vec2 fragCoord, |
| 28 | + in vec2 resolution, |
| 29 | + in vec2 uv) { |
| 30 | +
|
| 31 | + float sum = 0; |
| 32 | + // anti-aliasing |
| 33 | + for (float i = -0.5; i < 1.5; i += 0.2) { |
| 34 | + for (float j = -0.5; j < 1.5; j += 0.2) { |
| 35 | + vec2 uv2 = (fragCoord.xy + vec2(i, j)) / resolution.yy * 0.2; |
| 36 | + sum += map(uv2); |
| 37 | + } |
| 38 | + } |
| 39 | + sum /= 25; |
| 40 | +
|
| 41 | + fragColor = u_color * sum; |
| 42 | +} |
| 43 | +"# |
| 44 | +.as_bytes(); |
| 45 | + |
| 46 | +mod imp { |
| 47 | + use super::*; |
| 48 | + use gtk::graphene; |
| 49 | + use std::cell::{Cell, RefCell}; |
| 50 | + |
| 51 | + #[derive(Default)] |
| 52 | + pub struct Snow { |
| 53 | + pub(super) snow_shader: RefCell<Option<gsk::GLShader>>, |
| 54 | + pub(super) time_start: Cell<i64>, |
| 55 | + } |
| 56 | + |
| 57 | + #[glib::object_subclass] |
| 58 | + impl ObjectSubclass for Snow { |
| 59 | + const NAME: &'static str = "ComponentsSnow"; |
| 60 | + type Type = super::Snow; |
| 61 | + type ParentType = gtk::Widget; |
| 62 | + } |
| 63 | + |
| 64 | + impl ObjectImpl for Snow { |
| 65 | + fn constructed(&self) { |
| 66 | + if super::Snow::correct_date() { |
| 67 | + self.obj().add_tick_callback(|widget, _clock| { |
| 68 | + widget.queue_draw(); |
| 69 | + glib::Continue(true) |
| 70 | + }); |
| 71 | + } else { |
| 72 | + self.obj().set_visible(false); |
| 73 | + } |
| 74 | + } |
| 75 | + } |
| 76 | + |
| 77 | + impl WidgetImpl for Snow { |
| 78 | + fn snapshot(&self, snapshot: >k::Snapshot) { |
| 79 | + let widget = self.obj(); |
| 80 | + widget.ensure_shader(); |
| 81 | + |
| 82 | + if let Some(shader) = &*self.snow_shader.borrow() { |
| 83 | + let frame_time = widget.frame_clock().unwrap().frame_time(); |
| 84 | + let time = (frame_time - self.time_start.get()) as f32 / 1000000.0; |
| 85 | + |
| 86 | + let color = self.obj().style_context().color(); |
| 87 | + let color = |
| 88 | + graphene::Vec4::new(color.red(), color.green(), color.blue(), color.alpha()); |
| 89 | + let args_builder = gsk::ShaderArgsBuilder::new(shader, None); |
| 90 | + |
| 91 | + args_builder.set_float(0, time); |
| 92 | + args_builder.set_vec4(1, &color); |
| 93 | + |
| 94 | + let wigth = widget.width() as f32; |
| 95 | + let height = widget.height() as f32; |
| 96 | + let bounds = graphene::Rect::new(0.0, 0.0, wigth, height); |
| 97 | + |
| 98 | + snapshot.push_gl_shader(shader, &bounds, &args_builder.to_args()); |
| 99 | + snapshot.pop(); |
| 100 | + } |
| 101 | + } |
| 102 | + } |
| 103 | +} |
| 104 | + |
| 105 | +glib::wrapper! { |
| 106 | + pub struct Snow(ObjectSubclass<imp::Snow>) |
| 107 | + @extends gtk::Widget; |
| 108 | +} |
| 109 | + |
| 110 | +impl Snow { |
| 111 | + fn correct_date() -> bool { |
| 112 | + match glib::DateTime::now_local() { |
| 113 | + Ok(date_time) => { |
| 114 | + let month = date_time.month(); |
| 115 | + let day = date_time.day_of_month(); |
| 116 | + match month { |
| 117 | + 12 => day >= 24, |
| 118 | + 1 => day <= 5, |
| 119 | + _ => false, |
| 120 | + } |
| 121 | + } |
| 122 | + Err(_) => false, |
| 123 | + } |
| 124 | + } |
| 125 | + |
| 126 | + fn ensure_shader(&self) { |
| 127 | + let imp = self.imp(); |
| 128 | + if imp.snow_shader.borrow().is_none() { |
| 129 | + let renderer = self.native().unwrap().renderer(); |
| 130 | + |
| 131 | + let shader = { |
| 132 | + let bytes = glib::Bytes::from_static(SNOW_SHADER); |
| 133 | + let shader = gsk::GLShader::from_bytes(&bytes); |
| 134 | + match shader.compile(&renderer) { |
| 135 | + Err(e) => { |
| 136 | + // That shader isn't important |
| 137 | + log::error!("Couldn't compile snow shader: {}", e); |
| 138 | + self.set_visible(false); |
| 139 | + None |
| 140 | + } |
| 141 | + Ok(_) => Some(shader), |
| 142 | + } |
| 143 | + }; |
| 144 | + |
| 145 | + imp.snow_shader.replace(shader); |
| 146 | + imp.time_start.set(self.frame_clock().unwrap().frame_time()); |
| 147 | + } |
| 148 | + } |
| 149 | +} |
0 commit comments