path: root/main.js

let canvas;
let video;
let ctx;
let settings = {
  "brightness": {},
  "saturate": {},
  "contrast": {},
  "hue-rotate": {filter: value => value + "deg"},
  "grayscale": {},
  "sepia": {},
  "invert": {},
  "blur": { filter: value => value * canvas.width / 100 + 'px' }
};
const img = new Image();
let lensflare_active = false;
let cursor = {
  x: 0,
  y: 0
};

// wait for site to be parsed so element can be found
document.addEventListener("DOMContentLoaded", function() {
  canvas = document.getElementById("myCanvas");
  ctx = canvas.getContext("2d");
  video = document.getElementById("video");
  document.getElementById("back").addEventListener("click", () => document.body.className = "import-active");

  // bind listeners
  document.getElementById("viewport").addEventListener("drop", drop_handler);
  document.getElementById("viewport").addEventListener("dragover", event => event.preventDefault());
  document.getElementById("take-picture").addEventListener("click", use_camera);
  document.getElementById("cheese").addEventListener("click", take_picture);
  document.getElementById("upload-image").addEventListener("change", upload_image)

  document.getElementById("save").addEventListener("click", save_image)
  document.getElementById("share").addEventListener("click", share_image)

  document.getElementById("lensflare").addEventListener("click", function(event) {
    lensflare_active = !lensflare_active;
    if (lensflare_active) {
      event.target.classList.add("is-primary");
    } else {
      event.target.classList.remove("is-primary");
    }
    draw(true);
  });

  document.getElementById("viewport").addEventListener("mousemove", function(event) {
    if (event.buttons === 1 && lensflare_active) {
      cursor.x = (event.clientX - canvas.offsetLeft) / canvas.clientWidth - 0.5;
      cursor.y = (event.clientY - canvas.offsetTop) / canvas.clientWidth - 0.5;
      console.log(cursor);
      draw(true);
    }
  });

  window.addEventListener("resize", () => draw(true))

  video.addEventListener("canplay", function() {
    const width = 320;
    let height = video.videoHeight / (video.videoWidth / width);

    // Firefox currently has a bug where the height can't be read from
    // the video, so we will make assumptions if this happens.

      if (isNaN(height)) {
        height = width / (4 / 3);
      }
    video.width = width;
    video.height = height;
  });

  for (let setting in settings) {
    // make an array out of an iterable
    const elements = [...document.getElementsByClassName(setting)];
    settings[setting].elements = elements;
    // if filter is not definded, use identity function
    settings[setting].filter ||= value => value;
    for (let element of elements) {
      element.addEventListener("input", settings_apply);
    }

    const reset_elements = document.getElementsByClassName(setting + "-reset");
    for (let reset_element of reset_elements) {
      reset_element.addEventListener("click", () => reset_all(setting));
    }
  }
});

function reset_all(setting) {
  console.log("reseting " + setting);
  for (let element of settings[setting].elements) {
    element.value = element.defaultValue;
  }
  draw(true);
}

function use_camera() {
  navigator.mediaDevices
    .getUserMedia({ video: true, audio: false })
    .then((stream) => {
      video.srcObject = stream;
      video.play();
      document.body.className = "camera-active";
    })
    .catch((err) => {
      console.error(`An error occurred: ${err}`);
    });
}

function take_picture() {
  canvas.width = video.width;
  canvas.height = video.height;
  ctx.drawImage(video, 0, 0, canvas.width, canvas.height);
  img.src = canvas.toDataURL("image/png");
  img.onload = () => draw(true);

  document.body.className = "editor-active";
}

function upload_image() {
  document.body.className = "editor-active";

  console.log(this.files[0]);

  img.src = URL.createObjectURL(this.files[0]);
  img.onload = () => draw(true);
}

// https://developer.mozilla.org/en-US/docs/Web/API/HTML_Drag_and_Drop_API/File_drag_and_drop
function drop_handler(ev) {
  ev.preventDefault();
  let file;

  if (ev.dataTransfer.items) {
    const item = ev.dataTransfer.items[0];
    if (item.kind !== "file") return;
    file = item.getAsFile();
  } else {
    file = ev.dataTransfer.files[0];
  }
  console.log(file);

  document.body.className = "editor-active";
  img.src = URL.createObjectURL(file);
  img.onload = () => draw(true);
}

function save_image(event) {
  event.preventDefault();
  draw(false);

  const dataUrl = canvas.toDataURL("image/png");
  // downloading only works with links but not window.open
  const link = document.createElement('a');
  link.href = dataUrl;
  link.download = 'imagine.png';
  link.click();
}

function share_image(event) {
  event.preventDefault();
  if (!navigator.share) {
    console.log("navigator.share does not exist");
    return;
  }

  canvas.toBlob(async (blob) => {
    if (!blob) return;
    const file = new File([blob], 'imagine.png', {type: 'image/png'});

    try {
      await navigator.share({files: [file]});
    } catch (error) {
      console.log('Error sharing:', error);
    }
  }, 'image/png');
}

function settings_apply(event) {
  const changed_setting = event.target.id;
  const new_value = event.target.value;

  // update all inputs for that setting (mobile and desktop) to the new value
  for (let element of settings[changed_setting].elements) {
    if (element == event.target) continue;
    element.value = new_value;
  }

  draw(true);
}

// https://www.shadertoy.com/view/ldSXWK
function lensflare(pos_x, pos_y) {
  const imgdata = ctx.getImageData(0, 0, canvas.width, canvas.height);
  const pixel_count = imgdata.data.length / 4;
  const aspect_ratio = canvas.width / canvas.height;
  for (let i = 0; i < pixel_count; i++) {
    const x = i % canvas.width;
    const y = i / canvas.width;
    const u = (x / canvas.width - .5) * aspect_ratio;
    const v = y / canvas.height - .5;

    const intensity = 1.5 * 255;
    const uv_len = u * u + v * v;
    const uvd_x = u * uv_len;
    const uvd_y = v * uv_len;
    const uvd_len = uvd_x * uvd_x + uvd_y * uvd_y;

    const uvd_pos_x = uvd_x + pos_x;
    const uvd_pos_y = uvd_y + pos_y;
    let temp = uvd_pos_x * uvd_pos_x + uvd_pos_y * uvd_pos_y
    const f2 = Math.max(1.0 / (1.0 + 32.0 * temp * .64), .0) * 0.1;
    const f22 = Math.max(1.0 / (1.0 + 32.0 * temp * .72), .0) * 0.08;
    const f23 = Math.max(1.0 / (1.0 + 32.0 * temp * .81), .0) * 0.06;

    let uvx_x = u * (1 + 0.5) + uvd_x * 0.5;
    let uvx_y = v * (1 + 0.5) + uvd_y * 0.5;
    let uvx_pos_x = uvx_x + pos_x;
    let uvx_pos_y = uvx_y + pos_y;

    temp = uvx_pos_x * uvx_pos_x + uvx_pos_y * uvx_pos_y
    const f4 = Math.max(0.01 - temp * .16, .0) * 6.0;
    const f42 = Math.max(0.01 - temp * .20, .0) * 5.0;
    const f43 = Math.max(0.01 - temp * .25, .0) * 3.0;

    uvx_x = u * (1 + 0.4) + uvd_x * 0.4;
    uvx_y = v * (1 + 0.4) + uvd_y * 0.4;
    uvx_pos_x = uvx_x + pos_x;
    uvx_pos_y = uvx_y + pos_y;

    temp = uvx_pos_x * uvx_pos_x + uvx_pos_y * uvx_pos_y
    const f5 = Math.max(0.01 - temp * .04, .0) * 2.0;
    const f52 = Math.max(0.01 - temp * .16, .0) * 2.0;
    const f53 = Math.max(0.01 - temp * .36, .0) * 2.0;

    uvx_x = u * (1 + 0.5) + uvd_x * 0.5;
    uvx_y = v * (1 + 0.5) + uvd_y * 0.5;
    uvx_pos_x = uvx_x - pos_x;
    uvx_pos_y = uvx_y - pos_y;

    temp = uvx_pos_x * uvx_pos_x + uvx_pos_y * uvx_pos_y
    const f6 = Math.max(0.01 - temp * .9, .0) * 6.0;
    const f62 = Math.max(0.01 - temp * .1, .0) * 3.0;
    const f63 = Math.max(0.01 - temp * .12, .0) * 5.0;

    imgdata.data[4 * i + 0] += 1.2 * ((f2 + f4 + f5 + f6) * 1.3 - uvd_len * .05) * intensity;
    imgdata.data[4 * i + 1] += 1.5 * ((f22 + f42 + f52 + f62) * 1.3 - uvd_len * .05) * intensity;
    imgdata.data[4 * i + 2] += 1.3 * ((f23 + f43 + f53 + f63) * 1.3 - uvd_len * .05) * intensity;
    imgdata.data[4 * i + 3] = 255;
  }
  ctx.putImageData(imgdata, 0, 0);
}

function draw(viewport_scale) {
  const filter = Object.entries(settings)
    .map(([setting, { elements, filter }]) => `${setting}(${filter(elements[0].value)})`)
    .join(" ")

  // set the resolution to the original and then scale down to the viewport to
  // only calculate the filter on pixels the user can see. 
  canvas.height = img.naturalHeight;
  canvas.width = img.naturalWidth;
  if (viewport_scale) {
    canvas.height = canvas.clientHeight;
    canvas.width = canvas.clientWidth;
  }
  ctx.filter = filter
  console.log(filter);
  ctx.drawImage(img, 0, 0, canvas.width, canvas.height);
  ctx.filter = '';

  if (lensflare_active) {
    lensflare(cursor.x, cursor.y);
  }
}