Rakefile

#!/usr/bin/env rake
require "bundler/gem_tasks"
require_relative "lib/ocarina.rb"
require 'powerbar'
require 'rake/testtask'

Rake::TestTask.new do |t|
  t.libs << 'test'
end

desc "Run tests"
task :default => :test

namespace :ocarina do
  include Ocarina::Util

  desc "generates and persists bitmaps for reference/noise images"
  task :bitmaps do |t, args|

    # train/eval does not actually need bitmaps on disk- we provide this so that
    # you can examine the generated bitmap images for debugging purposes.
    #
    # images are saved to data/images/{reference,noise}

    generator = Ocarina::CharacterGenerator.new(config)
    generator.persist_tiles
  end

  desc "builds and trains the network"
  task :train do |t, args|
    network = Ocarina::Network.new(config)
    generator = Ocarina::CharacterGenerator.new(network.config)

    training_iterations = 350
    pbar = PowerBar.new
    pbar.settings.tty.finite.template.barchar = '#'
    pbar.settings.tty.finite.template.padchar = '-'

    training_iterations.times do |i|
      generator.reference_image_hash.each_pair do |char, tile|
        network.train tile, char
        pbar.show(msg: "current error: #{'%.10f' % network.current_error}", done: i + 1, total: training_iterations)
      end
    end

    puts "\nfinal training error: #{network.current_error}\n"

    # run with: ruby -I"lib:test" test/network_test.rb -n test_network

    file = "#{Ocarina::DATA_DIR}/train.bin"
    network.save_network_to_file file

    puts "trained network saved to: #{file}"
  end

  desc "runs the training images back through the network for evaluation"
  task :eval do |t, args|
    file = "#{Ocarina::DATA_DIR}/train.bin"
    network = Ocarina::Network.load_network_from_file file
    generator = Ocarina::CharacterGenerator.new(network.config)

    puts "##### testing against reference images #####"
    stats = Ocarina::ErrorStats.new(network.config)

    generator.reference_image_hash.each_pair do |char, tile|
      result = network.recognize tile
      stats.check_error char.ord, result
    end
    stats.report

    puts
    puts "##### testing against noise images #####"

    stats = Ocarina::ErrorStats.new(network.config)

    generator.noise_image_hash.each_pair do |char, tile|
      result = network.recognize tile
      stats.check_error char.ord, result
    end
    stats.report
  end

  desc "builds and trains network using Letterpress board tiles"
  task :letterpress do |t, args|
    network = Ocarina::Network.new(config)
    generator = Ocarina::LetterpressCharacterGenerator.new(network.config)

    training_iterations = 1200
    pbar = PowerBar.new
    pbar.settings.tty.finite.template.barchar = '#'
    pbar.settings.tty.finite.template.padchar = '-'

    training_iterations.times do |i|
      generator.reference_image_hash.each_pair do |char, tile|
        network.train tile, char
        pbar.show(msg: "current error: #{'%.10f' % network.current_error}", done: i + 1, total: training_iterations)
      end
    end

    puts "\nfinal training error: #{network.current_error}\n"

    network.save_network_to_file "#{Ocarina::DATA_DIR}/letterpress-train.bin"

    puts "trained network saved to: #{file}"

    puts "##### testing against reference images #####"
    stats = Ocarina::ErrorStats.new(network.config)

    generator.reference_image_hash.each_pair do |char, tile|
      result = network.recognize tile
      stats.check_error char.ord, result
    end
    stats.report

  end

  desc "deciphers letters from a letterpress game board"
  task :gameboard, [:board_file]  do |t, args|
    network = Ocarina::Network.load_network_from_file "#{Ocarina::DATA_DIR}/letterpress-train.bin"

    puts "reading letterpress board: #{args.board_file}..."

    board = Magick::Image.read(args.board_file).first

    cropper = Ocarina::LetterpressCropper.new(network.config)
    chars = cropper.decipher_board(network, board)
    result = ""

    chars.each do |row|
      row.each do |char|
        result << "  #{char}"
      end

      result << "\n"
    end

    puts "result: \n\n#{result}"
  end

  def config
    # need 8 bits to represent 0..255 in binary
    @config ||= Ocarina::Config.new("ABCDEFGHIJKLMNOPQRSTUVWXYZ", 8, 16, 16)
  end


end