class Analyzer

# File lib/analyzer.rb, line 8
def initialize(*test_files, bootstrap: nil, lib: nil, n: nil)
  @bootstrap = bootstrap
  @lib = lib
  @n = n || [1]
  @test_files = test_files
  
  @keys = test_files.map { |fn| File.basename(fn, '.*') }
  depth = -1
  while @keys.uniq.length < @keys.length
    @keys = test_files.map { |fn| ( File.dirname(fn).split(File::SEPARATOR)[depth..-1] + [File.basename(fn, '.*')] ).join(File::SEPARATOR) }
    depth -= 1
  end
end

# File lib/analyzer.rb, line 30
def bootstrap
  if @bootstrap
    system("ruby", "-r", "#{File.expand_path(@lib)}", File.expand_path(@bootstrap))
    if $?.exitstatus > 0
      exit 1
    end
  end
end

# File lib/analyzer.rb, line 54
def calculate_results_for(type, src, n=1)
  lib = @lib ? File.read(File.expand_path(@lib)) : ''
  setup = "N=#{n}\n"
  
  if src.index("\n__SETUP__\n")
    src = src.split("\n__SETUP__\n")
    setup << src[0]
    src = src[1]
  end
  
  analyzer_dir = File.expand_path("../", __FILE__)
  template = File.read(File.expand_path("../templates/#{type}.rb.erb", __FILE__))
  _script = ''
  ERB.new(template, nil, nil, "_script").result(binding)
  results = `ruby <<'TESTSCRIPT'\n#{_script}\nTESTSCRIPT`

  if $? == 0
    JSON.parse(results)
  else
    puts results
    exit
  end
end

# File lib/analyzer.rb, line 100
def escape(value)
  value.gsub('_', '\\\\_')
end

# File lib/analyzer.rb, line 22
def key(value)
  @keys[@test_files.index(value)]
end

# File lib/analyzer.rb, line 82
def output_gc_results(results, output)
  last_row = results.shift
  first_row = last_row
  results.each do |row|
    output.puts [row[0], row[0] - first_row[0], row[1], row[2], row[1] - last_row[1], row[2] - last_row[2], row[3], row[4], row[3] - last_row[3], row[4] - last_row[4]].join(' ')
    last_row = row
  end
end

# File lib/analyzer.rb, line 78
def output_ips_results(results, output)
  output.write(results.join("\n"))
end

# File lib/analyzer.rb, line 91
def output_rss_results(results, output)
  last_row = results.shift
  first_row = last_row
  results.each do |row|
    output.puts [row[0], row[0] - first_row[0], row[1], row[1] - last_row[1]].join(' ')
    last_row = row
  end
end

# File lib/analyzer.rb, line 104
def plot(output)
  Dir.mktmpdir do |dir|
    write_results_for('ips', dir)
    write_results_for('gc', dir)
    write_results_for('rss', dir)
    
    template = File.read(File.expand_path('../templates/gnuplot.sh.erb', __FILE__))
    _script = ''
    data_dir = dir
    ERB.new(template, nil, nil, "_script").result(binding)
    `gnuplot <<-GNUPLOT\n#{_script}\nGNUPLOT`
  end
end

# File lib/analyzer.rb, line 26
def tests
  @test_files.map{ |f| key(f) }
end

# File lib/analyzer.rb, line 39
def write_results_for(type, dir)
  @n.each do |n|
    @test_files.each do |src|
      if key(src).index(File::SEPARATOR)
        FileUtils.mkdir_p(File.join(dir, type, n.to_s, File.dirname(key(src))))
      else
        FileUtils.mkdir_p(File.join(dir, type, n.to_s))
      end
      File.open(File.join(dir, type, n.to_s, key(src)), 'w') do |file|
        send(:"output_#{type}_results", calculate_results_for(type, File.read(src), n), file)
      end
    end
  end
end