Move startup stuff to start()

[picsort] / picsorter.js

var picinfo = {};
var exposure = 20;
var zoom = "sm/";
var input_index = -1;

function save_picinfo() {
  localStorage.picsorter_picinfo = JSON.stringify(picinfo);
}

function endsWith(str, suffix) {
  return str.indexOf(suffix, str.length - suffix.length) !== -1;
}

function stripFromEnd(str, suffix) {
  if (endsWith(str, suffix)) {
    return str.substr(0, str.length - suffix.length);
  }
  return str;
}

function setpic() {
  var display_filename;
  if (endsWith(files[input_index], ".NEF")) {
    if ("exposure" in picinfo[files[input_index]]) {
      exposure = picinfo[files[input_index]].exposure;
    } else {
      picinfo[files[input_index]].exposure = exposure;
      save_picinfo();
    }
    display_filename = zoom + exposure + "/" + stripFromEnd(files[input_index], ".NEF") + ".jpeg";
  } else {
    display_filename = zoom + files[input_index];
  }
  var $pic = $("#pic");
  $pic.on("load", function() {
    $pic.removeClass("rot90 rot180 rot270");
    if ("rotate" in picinfo[files[input_index]]) {
      $pic.addClass("rot" + picinfo[files[input_index]].rotate);
    }
  });
  $pic.attr("src", display_filename);
}

function say(message) {
  $("#message").text(message).removeClass("fade");
  setTimeout(function() { $("#message").addClass("fade"); }, 100);
}
function announce() {
  var msg = input_index + " " + (picinfo[files[input_index]].name || "");
  if ("deleted" in picinfo[files[input_index]]) {
    msg += "(" + picinfo[files[input_index]].deleted+ ")";
  }
  say(msg);
}

function toggle_zoom() {
  if (zoom) {
    zoom = "";
  } else {
    zoom = "sm/";
  }
  setpic();
}

function move_by_filter(direction, filter) {
  // Keep moving in direction until filter is satisfied
  var new_index = input_index;
  while (true) {
    var next = new_index + direction;
    if (next < 0) {
      say("At beginning");
      return;
    }
    if (next >= files.length) {
      say("At end");
      return;
    }
    new_index = next;
    if (filter(new_index)) {
      break;
    }
  }
  input_index = new_index;
  announce();
  setpic();
}
function move(direction) {
  move_by_filter(direction, function() { return true; });
}
function move_to_begenning() {
  move_by_filter(-1, function(i) { return i == 0; });
}
function move_to_end() {
  move_by_filter(1, function(i) { return i == files.length - 1; });
}
function move_to_nondeleted(direction) {
  move_by_filter(direction, function(i) {
    return !("deleted" in picinfo[files[i]]); });
}
function move_to_unnamed(direction) {
  move_by_filter(direction, function(i) {
    return !("deleted" in picinfo[files[i]]) &&
           !("name" in picinfo[files[i]]); });
}

function mark_deleted(method) {
  picinfo[files[input_index]].deleted = method;
  save_picinfo();
  say(method);
}

function mark_not_deleted() {
  delete picinfo[files[input_index]].deleted;
  save_picinfo();
  say("Undeleted");
}

function say_exposure() {
  if (!endsWith(files[input_index], ".NEF")) {
    say("Exposure adjustment not available");
    return;
  }
  var display_exposure = (exposure / 4) - 3.5;
  say((display_exposure >= 0 ? "+" : "") + display_exposure);
}

function change_exposure(amount) {
  if (!endsWith(files[input_index], ".NEF")) {
    say("Exposure adjustment not available");
    return;
  }
  exposure += amount;
  picinfo[files[input_index]].exposure = exposure;
  save_picinfo();
  setpic();
  say_exposure();
}

function rotate() {
  var rotation = picinfo[files[input_index]].rotate || 0;
  rotation = (rotation + 90) % 360;
  if (rotation > 1e-5) {
    picinfo[files[input_index]].rotate = rotation;
  } else {
    delete picinfo[files[input_index]].rotate;
  }
  save_picinfo();
  setpic(); 
}


function set_name(name) {
  if (name) {
    picinfo[files[input_index]].name = name;
    save_picinfo();
    say("Named " + name);
    last_name = name;
  }
}

function set_name_from_form() {
  var name_input = $("#name").hide().get(0);
  set_name(name_input.value);
  name_input.value = "";
}

function shell_escape(x) {
  return x.replace(/'/g, "'\\''");
}

function show_commands() {
  var commands = [];
  $.each(files, function(i, f) {
    var escaped_filename = "'" + shell_escape(f) + "'";
    if (picinfo[f].deleted == "deleted") {
      commands.push("shred -u " + escaped_filename);
    } else if (picinfo[f].deleted == "extra") {
      commands.push("mv -vi " + escaped_filename + " \"$EXTRADIR\"");
    } else if (picinfo[f].deleted == "blurry") {
      commands.push("mv -vi " + escaped_filename + " \"$BLURRYDIR\"");
    } else if ("name" in picinfo[f] && picinfo[f].name.length > 0) {
      var command = ["pic-mv"];
      if ("exposure" in picinfo[f]) {
        command.push("-e " + picinfo[f].exposure);
      }
      if ("rotate" in picinfo[f]) {
        command.push("-r " + picinfo[f].rotate);
      }
      command.push(escaped_filename);
      command.push("'" + shell_escape(picinfo[f].name) + "'");
      commands.push(command.join(" "));
    }
  });
  commands.push("");
  $("#shell_out").text(commands.join("\n")).show();
}

function clean_picinfo() {
  files_index = {};
  $.each(files, function(i, f) {
    files_index[f] = true;
  });
  for (f in picinfo) {
    if (!files_index[f]) {
      delete picinfo[f];
    }
  }
}

function undelete_all() {
  for (f in picinfo) {
  	delete picinfo[f].deleted;
  }
  save_picinfo();
  say("Undeleted everything");
}

function start() {
  if ("picsorter_picinfo" in localStorage) {
    picinfo = JSON.parse(localStorage.picsorter_picinfo);
  }
  $.each(files, function(i, f) {
    if (!(f in picinfo)) {
      picinfo[f] = {};
    }
  });
  save_picinfo();

  $(function() {
    $("#name").hide().on("keyup", function(e) { e.which == 13 && set_name_from_form(); });
    move_to_nondeleted(1);
  });

  Mousetrap.bind('z', toggle_zoom);
  Mousetrap.bind('Z', function() { $("#pic").toggleClass("fit_view"); return false; });
  Mousetrap.bind(['n', 'l'], function() { move_to_nondeleted(1); return false; });
  Mousetrap.bind(['p', 'h'], function() { move_to_nondeleted(-1); return false; });
  Mousetrap.bind(['N', 'L'], function() { move_to_unnamed(1); return false; });
  Mousetrap.bind(['P', 'H'], function() { move_to_unnamed(-1); return false; });
  Mousetrap.bind(['m n', 'm l'], function() { move(1); return false; });
  Mousetrap.bind(['m n', 'm h'], function() { move(-1); return false; });
  Mousetrap.bind(['b', 'k'], function() { change_exposure(1); return false; });
  Mousetrap.bind(['d', 'j'], function() { change_exposure(-1); return false; });
  Mousetrap.bind('0', move_to_begenning);
  Mousetrap.bind('$', move_to_end);
  Mousetrap.bind('x', function(){ mark_deleted("deleted"); return false; });
  Mousetrap.bind('X', mark_not_deleted);
  Mousetrap.bind('e', function(){ mark_deleted("extra"); return false; });
  Mousetrap.bind('E', mark_not_deleted);
  Mousetrap.bind('g', function(){ mark_deleted("blurry"); return false; });
  Mousetrap.bind('G', mark_not_deleted);
  Mousetrap.bind('r', rotate);
  Mousetrap.bind('i', announce);
  Mousetrap.bind('f', function() { say(files[input_index]); return false; });
  Mousetrap.bind('B', say_exposure);
  Mousetrap.bind('c', function() { $("#name").show().focus(); return false; });
  Mousetrap.bind('C', function() { if (last_name) { set_name(last_name); } return false; });
  Mousetrap.bind('%', function() { say((100 * input_index / files.length).toFixed(2) + "%"); return false; });
  Mousetrap.bind('!', show_commands);
  Mousetrap.bind('esc', function() { $("#name").hide(); $("#shell_out").hide(); return false; });
}