golf.rb

#!/usr/bin/ruby -w

class Golf
  def self.run args
    args << "01" if args.empty?
    report args.join ","
  end

  def self.report dir_or_file
    if File.file? dir_or_file then
      report_file dir_or_file
    else
      Dir.chdir ".." until File.file? "Rakefile"

      report_dir dir_or_file
    end
  end

  def self.report_dir dir
    golfers = {}

    glob = "{#{dir}}/**/*.{hdl,asm}"

    users = Dir["*"].find_all { |f| File.directory? f } - %w(projects tools)
    users.each do |user|
      golf = Golf.new user
      golf.find_files glob
      golf.scan
      golfers[user] = golf
    end

    # if the two agree on the gates, it must be true. or something.
    gates = golfers["thagomizer"].gates & golfers["itscaleb"].gates

    gates = %w(thagomizer itscaleb).map { |u| golfers[u].gates }.inject(&:&)

    golfers.each do |_,user|
      gates.each do |gate|
        user.sum += user.scores[gate]
      end
    end

    users.reject! { |u|
      s = golfers[u].scores
      gates.any? { |k| s[k].nil? or s[k].zero? }
    }

    sorted_users = users.sort_by { |u| golfers[u].sum }

    fmt = "%%-10s: %s" % sorted_users.map { |u| "%#{[u.size, 8].max}X" }.join(" | ")

    data = ["gate\\user", *sorted_users]
    puts fmt.gsub("X", "s") % data
    data = ["total"] + sorted_users.map { |user| golfers[user].sum }
    puts fmt.gsub("X", "d") % data
    puts

    best = golfers[sorted_users.first]
    gates = best.path.keys.sort_by { |g| best.gate[g] }
    fmt += "%s"
    gates.each do |gate|
      scores = sorted_users.map { |user| golfers[user].gate[gate] || 0 }
      extra = " =" if scores.uniq.size == 1
      data = [gate] + scores + [extra]
      puts fmt.gsub("X", "d") % data
    end
  end

  def self.report_file path
    until File.file? "Rakefile"
      path = "#{File.basename Dir.pwd}/#{path}"
      Dir.chdir ".."
    end

    gate = File.basename path, File.extname(path)

    user = path.split("/").first

    glob = "**/*.{hdl,asm}"

    golf = Golf.new user
    golf.find_files glob

    golf.process gate

    max1 = golf.count.values.map { |n| Math.log10(n).ceil }.max
    max2 = golf.count.keys.map { |n| Math.log10(golf.gate[n]).ceil }.max

    golf.count.sort_by { |g,c| c * golf.gate[g] }.each do |g, c|
      n = golf.gate[g]
      puts "%-15s = %#{max1}d * %#{max2}d = %#{max2}d" % [g, c, n, c * n]
    end
  end

  attr_accessor :gate
  attr_accessor :name
  attr_accessor :path
  attr_accessor :scores
  attr_accessor :sum
  attr_accessor :count

  def initialize name = ""
    self.name = name
    self.path = {}
    self.gate = {}
    self.scores = {}
    self.count = Hash.new 0
    self.sum = 0
  end

  def gates
    scores.keys.sort
  end

  def find_files glob
    Dir["#{self.name}/#{glob}"].each do |path|
      gate = File.basename path, File.extname(path)

      self.path[gate] = path
    end
  end

  def scan
    self.path.keys.each do |gate|
      score = process(gate)
      self.scores[gate] = score
    end
  end

  def process gate
    self.count[gate] += 1
    return self.gate[gate] if self.gate[gate]
    return self.process_asm gate if path[gate] =~ /asm$/

    score = 0

    path = self.path[gate] || "#{self.name}/01/#{gate}.hdl"

    file = File.read(path) rescue ""
    file.gsub!(/\/\/.*/, "")
    file.gsub!(/\/\*.*?\*\//m, "")

    file.scan(/(\w+)\s*\(/).flatten.each do |sub|
      case sub
      when "Nand" then
        score += 1
      when "DFF" then
        score += 1
      else
        score += process sub
      end
    end

    self.gate[gate] = score
  end

  def process_asm name
    path = self.path[name]

    score = 0

    file = File.read(path) rescue ""
    file.gsub!(/(\/\/|\*).*/, "")

    file.each_line do |line|
      case line.strip!
      when "" then
        # do nothing
      when /^@\w/ then
        score += 1
      when /^[ADM]+\s*=/ then
        score += 1
      when /^[ADM0];/ then
        score += 1
      when /^\(\w/ then
        score += 1
      else
        raise "Unparsed: #{line.inspect}"
      end
    end

    self.gate[name] = score
  end
end

Golf.run ARGV if $0 == __FILE__