displacement.lisp

(in-package #:org.shirakumo.fraf.kandria)

(define-shader-entity displacer (renderable)
  ((texture :initarg :texture :reader texture)
   (strength :initform 1f0 :initarg :strength :accessor strength)))

(defmethod object-renderable-p ((displacer displacer) (pass shader-pass)) NIL)

(defmethod bind-textures ((displacer displacer))
  (gl:active-texture :texture0)
  (gl:bind-texture :texture-2d (gl-name (texture displacer))))

(defmethod stage :after ((displacer displacer) (area staging-area))
  (stage (texture displacer) area))

(defmethod render ((displacer displacer) (program shader-program))
  (setf (uniform program "model_matrix") (model-matrix))
  (setf (uniform program "effect_strength") (strength displacer))
  (render (// 'kandria '16x) program))

(define-class-shader (displacer :vertex-shader)
  "layout (location = TRIAL_V_LOCATION) in vec3 position;
layout (location = TRIAL_V_UV) in vec2 in_uv;
out vec2 uv;

uniform mat4 model_matrix;
uniform mat4 view_matrix;
uniform mat4 projection_matrix;

void main(){
  maybe_call_next_method();
  gl_Position = projection_matrix * view_matrix * model_matrix * vec4(position, 1.0f);
  uv = in_uv;
}")

(define-class-shader (displacer :fragment-shader)
  "uniform sampler2D texture_image;
uniform float effect_strength = 1.0;
in vec2 uv;
out vec4 color;

void main(){
  maybe_call_next_method();
  vec2 strength = texture(texture_image, uv).rg*2-1;
  color = vec4(strength*effect_strength, 0, 1);
}")

(define-shader-entity shockwave (displacer facing-entity located-entity listener)
  ((texture :initform (// 'kandria 'shockwave))
   (clock :initform 0f0 :accessor clock)
   (lifetime :initform 0.3 :initarg :lifetime :accessor lifetime)
   (initial-strength :initform 1f0 :initarg :strength :accessor initial-strength)))

(defmethod handle ((ev tick) (displacer shockwave))
  (incf (clock displacer) (dt ev))
  (setf (strength displacer)
        (* (initial-strength displacer)
           (- 1 (/ (clock displacer) (lifetime displacer)))))
  (when (< (lifetime displacer) (clock displacer))
    (leave displacer T)))

(defmethod apply-transforms progn ((displacer shockwave))
  (let* ((s (- 11.0 (min 11.0 (/ 1.0 (+ 0.01 (expt (clock displacer) 1.001)))))))
    (scale-by s s 1)))

(define-shader-entity heatwave (displacer sized-entity resizable listener ephemeral creatable)
  ((texture :initform (// 'kandria 'heatwave))
   (strength :initform 0.02)
   (offset :initform 0.0 :accessor offset))
  (:inhibit-shaders (displacer :fragment-shader)))

(defmethod handle ((ev tick) (heatwave heatwave))
  (incf (offset heatwave) (* (dt ev) -0.2)))

(defmethod handle ((ev change-time) (heatwave heatwave))
  ;; Scale strength based on time of day (8 day hours)
  (setf (strength heatwave)
        (if (setting :gameplay :visual-safe-mode)
            0.0
            (* (float (max 0 (/ (- 4 (abs (- (hour +world+) 12))) 4.0)) 0f0)
               0.0075))))

(defmethod render :before ((heatwave heatwave) (program shader-program))
  (setf (uniform program "offset") (offset heatwave)))

(defmethod apply-transforms progn ((heatwave heatwave))
  (scale-by (/ (vx (bsize heatwave)) 8) (/ (vy (bsize heatwave)) 8) 1))

(define-class-shader (heatwave :fragment-shader)
  "uniform sampler2D texture_image;
uniform float effect_strength = 1.0;
uniform float offset = 0.0;
in vec2 uv;
out vec4 color;

void main(){
  maybe_call_next_method();
  vec2 strength = texture(texture_image, vec2(uv.x, uv.y*2+offset)).rg*2-1;
  color = vec4(strength*effect_strength*(1-uv.y), 0, 1);
}")

(define-shader-entity scanline (displacer transformed)
  ((texture :initform (// 'kandria 'scanline))
   (strength :initform 1.0)))

(defmethod apply-transforms progn ((displacer scanline))
  (translate (vxy_ (location (camera +world+))))
  (scale-by 40 25 1))

(define-shader-pass displacement-render-pass (per-object-pass)
  ((displacement-map :port-type output :attachment :color-attachment0
                     :texspec (:internal-format :rg16f))
   (name :initform 'displacement-render-pass)))

(defmethod stage :before ((pass displacement-render-pass) (area staging-area))
  (stage (make-instance 'shockwave) pass)
  (stage (// 'kandria '16x) area)
  (stage (// 'kandria 'scanline) area)
  (stage (// 'kandria 'shockwave) area)
  (stage (// 'kandria 'dashwave) area))

(defmethod object-renderable-p ((renderable renderable) (pass displacement-render-pass)) NIL)
(defmethod object-renderable-p ((displacer displacer) (pass displacement-render-pass)) T)
(defmethod object-renderable-p ((displacer displacer) (pass shader-pass)) NIL)

(defmethod prepare-pass-program :after ((pass displacement-render-pass) (program shader-program))
  (setf (uniform program "view_matrix") (view-matrix))
  (setf (uniform program "projection_matrix") (projection-matrix)))

(defmethod render ((pass displacement-render-pass) (program shader-program))
  (gl:clear-color 0 0 0 1)
  (gl:clear :color-buffer)
  (gl:blend-func :one :one)
  (call-next-method)
  (gl:blend-func :src-alpha :one-minus-src-alpha))

(defmethod trial:scene ((pass displacement-render-pass)) pass)

(define-shader-pass displacement-pass (simple-post-effect-pass)
  ((name :initform 'displacement-pass)
   (displacement-map :port-type input)))

(define-class-shader (displacement-pass :fragment-shader)
  "uniform sampler2D previous_pass;
uniform sampler2D displacement_map;
in vec2 uv;
out vec4 color;

void main(){
  maybe_call_next_method();
  vec2 displacement = texture(displacement_map, uv).rg;
  vec3 previous = texture(previous_pass, uv+displacement).rgb;
  color = vec4(previous, 1);
}")