SE3-TTSDeck/resources/TTSDeck.js

/*
 * TTSDeck.js
 *
 * Creates a deck image and corresponding "Saved Object" JSON for use
 * in Tabletop Simulator
 */

useLibrary('project');
useLibrary('imageutils');
useLibrary('threads');
useLibrary('uilayout');
useLibrary('uicontrols');
importClass(arkham.project.CopiesList);

const TTSJson = require('./TTSJson.js');

const TTS_CARDS_PER_IMAGE = 69;
const TTS_MAX_ROWS = 7;


const getName = () => 'TTSDeck';
const getDescription = () => 'Generates a TTS deck image and JSON file';
const getVersion = () => 1.0;
const getPluginType = () => arkham.plugins.Plugin.INJECTED;

function unload() {
  unregisterAll();
}

// Creates a test button during development that calls unload() to clean up.
testProjectScript();

// TODO: allow setting a default copy count
// Hack to override the default return value of 1
function copyCount(copies_list, name) {
  const entries = copies_list.getListEntries().map(x => String(x));
  if (entries.indexOf(String(name)) == -1) {
    return 1;
  } else {
    return copies_list.getCopyCount(name);
  }
}

function getImageFile(parent, format, page_num) {
  return new File(parent.file, parent.getName() + '_' + page_num + '.' + format);
};


function Card(member) {
  this.member = member;
  this.component = ResourceKit.getGameComponentFromFile(member.file);

  this.makeImageUncached = function makeImageUncached(resolution, back) {
    println("Generating image for card ", this.member);
    const sheets = this.component.createDefaultSheets();
    const card_image = sheets[back ? 1 : 0].paint(arkham.sheet.RenderTarget.EXPORT, resolution);

    return card_image;
  };

  // export front face, or retrive it from a cached file
  // TODO: handle two-sided cards
  this.makeImage = function makeImage(format, resolution) {
    const cache_dir = new File(this.member.parent.file, '.ttsdeck_cache');
    const cached_file = new File(cache_dir, this.member.file.name + '.' + format);

    if (cached_file.exists() && cached_file.lastModified() > this.member.file.lastModified()) {
      println("Got cached image for card", this.member);
      return ImageUtils.read(cached_file);
    } else {
      const card_image = this.makeImageUncached(resolution);

      cache_dir.mkdir();
      ImageUtils.write(card_image, cached_file, format, -1, false, resolution);

      return card_image;
    }
  };
}


function TTSDeckPage(busy_props, image_format, image_resolution, page_num, page_cards, copies_list) {
  this.rows = Math.min(Math.ceil(Math.sqrt(page_cards.length)), TTS_MAX_ROWS);
  this.columns = Math.ceil(page_cards.length / this.rows);
  this.deck_image = null;
  let deck_graphics;

  this.card_jsons = [];
  for (let row = 0; row < this.rows; row++) {
    for (let col = 0; col < this.columns && row * this.columns + col < page_cards.length; col++) {
      if (busy_props.cancelled) return;
      let index = row * this.columns + col;
      let card = page_cards[index];
      busy_props.status = "Processing Card " + card.member;
      busy_props.currentProgress = (page_num - 1) * TTS_CARDS_PER_IMAGE + index;

      try {
        let copies = copyCount(copies_list, card.baseName);

        for (let ii = 0; ii < copies; ii++) {
          this.card_jsons.push(TTSJson.makeCardJSON(page_num * 100 + index, card.component.getName()));
        }

        let card_image = card.makeImage(image_format, image_resolution);

        if (!this.deck_image) {
          this.deck_image = ImageUtils.create(
            card_image.width * this.columns, card_image.height * this.rows, false);
          deck_graphics = this.deck_image.createGraphics();
        }

        deck_graphics.drawImage(card_image, col * card_image.width, row * card_image.height, null);
      } catch (ex) {
        Thread.invokeLater(() => alert('Error while processing ' + card + ': ' + ex, true));
      }
    }
    println("End of Row ", row);
  }

  this.face_url = String(getImageFile(page_cards[0].member.parent, image_format, page_num).toPath().toUri());
  this.back_url = String(getImageFile(page_cards[0].member.parent, image_format, "back").toPath().toUri());
}

function makeTTSDeck(busy_props, image_format, image_resolution, cards, copies_list) {
  const pages = [];

  busy_props.title = "Processing Cards";
  busy_props.maximumProgress = cards.length;

  for (let page_num = 0; page_num * TTS_CARDS_PER_IMAGE < cards.length; page_num++) {
    let page_cards = cards.slice(page_num * TTS_CARDS_PER_IMAGE, (page_num + 1) * TTS_CARDS_PER_IMAGE);
    printf("Making page %d, with %d cards:\n", page_num + 1, page_cards.length);
    pages.push(new TTSDeckPage(busy_props, image_format, image_resolution, page_num + 1, page_cards, copies_list));
    if (busy_props.cancelled) return [,];
  }

  const deck_json = TTSJson.makeDeckJSON(pages);

  return [deck_json, pages.map(page => page.deck_image)];
}

function settingsDialog(deck_task) {
  const task_settings = deck_task.getSettings();

  const image_format_field = comboBox([
    ImageUtils.FORMAT_JPEG,
    ImageUtils.FORMAT_PNG
  ]);
  image_format_field.setSelectedItem(task_settings.get("tts_image_format", "jpg"));
  const resolution_field = textField(task_settings.get("tts_image_resolution", "200"), 15);

  const clear_cache_button = button("Clear Cache", undefined, function (e) {
    const cache_dir = new File(deck_task.file, '.ttsdeck_cache');
    cache_dir.listFiles().forEach((file) => file.delete());
  });

  const panel = new Grid();
  panel.place(
    "Image Format", "",
    image_format_field, "grow,span",
    "Resolution", "",
    resolution_field, "grow,span",
    clear_cache_button, "grow,span"
  );
  const close_button = panel.createDialog('TTS Export').showDialog();
  return [close_button, image_format_field.getSelectedItem(), Number(resolution_field.text)];
}

function run() {
  const ttsDeckAction = JavaAdapter(TaskAction, {
    getLabel: () => 'Generate TTS Deck',
    getActionName: () => 'ttsdeck',
    // Applies to Deck Tasks
    appliesTo: function appliesTo(project, task, member) {
        if (member != null || task == null) {
            return false;
        }
        const type = task.settings.get(Task.KEY_TYPE);
        if (NewTaskType.DECK_TYPE.equals(type)) {
            return true;
        }
        return false;
    },
    perform: function perform(project, task, member) {
      let deck_task = ProjectUtilities.simplify(project, task, member);
      const [close_button, image_format, image_resolution] = settingsDialog(deck_task);

      // User canceled the dialog or closed it without pressing ok
      if (close_button != 1) {
        return;
      }
      // persist settings
      const task_settings = deck_task.getSettings();
      task_settings.set("tts_image_format", image_format);
      task_settings.set("tts_image_resolution", image_resolution);
      deck_task.writeTaskSettings();

      Eons.setWaitCursor(true);
      try {
        Thread.busyWindow(
          (busy_props) => this.performImpl(busy_props, image_format, image_resolution, deck_task),
          'Setting up...',
          true);
      } catch (ex) {
        Error.handleUncaught(ex);
      } finally {
        Eons.setWaitCursor(false);
      }
    },
    performImpl: function performImpl(busy_props, image_format, image_resolution, member) {
      let copies_list;
      try {
        copies_list = new CopiesList(member);
      } catch (ex) {
        copies_list = new CopiesList();
        alert("unable to read copies list, using card count of 2 for all files", true);
      }

      const children = member.getChildren();
      const cards = children
        .map(child => {
          if (ProjectUtilities.matchExtension(child, 'eon')) {
            let card = new Card(child);
            if (card.component.isDeckLayoutSupported()) {
              return card;
            }
          }
          return undefined;
        })
        .filter(card => card !== undefined);

      const [deck_json, deck_images] = makeTTSDeck(busy_props, image_format, image_resolution, cards, copies_list);
      if (busy_props.cancelled) return;
      const saved_object = TTSJson.makeSavedObjectJSON([deck_json], member.getName());

      busy_props.status = "";
      busy_props.maximumProgress = -1;
      busy_props.title = "Writing JSON";
      const json_file = new File(member.file, member.getName() + '.json');
      ProjectUtilities.writeTextFile(json_file, JSON.stringify(saved_object, null, 4));

      busy_props.title = "Writing Images";
      busy_props.maximumProgress = deck_images.length;
      deck_images.forEach((deck_image, index) => {
        busy_props.currentProgress = index;
        const image_file = getImageFile(member, image_format, index + 1);
        ImageUtils.write(deck_image, image_file, image_format, -1, false, image_resolution);
      });

      let back_image = cards[0].makeImageUncached(image_resolution, true);
      const back_image_file = getImageFile(member, image_format, "back");
      ImageUtils.write(back_image, back_image_file, image_format, -1, false, image_resolution);

      member.synchronize();
    }
  });

  ActionRegistry.register(ttsDeckAction, Actions.PRIORITY_IMPORT_EXPORT);
}