opengl_getbitmap.rb

require 'bundler/setup'
require 'opengl'
require 'glut'
include Gl, Glut

class BitMap
  def initialize(width, height, dpi = 96)
    @width = width
    @height = height
    @line_size = width * 3 + (4 - (width * 3) % 4) % 4
    @buf_size = @line_size * height
    @buf = ("\000" * @buf_size).encode('BINARY')
    @bit_count = 24
    @compression = 0  # 圧縮無し
    @size_image = 0
    @x_pix_per_meter = (39.375 * dpi).round
    @y_pix_per_meter = (39.375 * dpi).round
    @clr_used = 0
    @cir_important = 0
  end
  
  attr_accessor :buf
  attr_reader :width, :height

  # BMPファイルを出力する
  def write(filename)
    file_size = 14 + 40 + @buf_size
    data_offset = 14 + 40

    open(filename, "wb") do |f|
      f.print 'BM'
      f.print [file_size, 0, data_offset].pack("l*")
      f.print [40, @width, @height].pack("l*")
      f.print [1, @bit_count].pack("S*")
      f.print [@compression, @size_image,
               @x_pix_per_meter, @y_pix_per_meter,
               @clr_used, @cir_important].pack("l*")
      f.print @buf
    end
  end
  
  def self.gl_capture(fname)
    x = glutGet(GLUT_WINDOW_WIDTH)
    y = glutGet(GLUT_WINDOW_HEIGHT)
    
    bitmap = BitMap.new(x, y)
    glPixelStorei(GL_PACK_ALIGNMENT, 1)
    imgdata = glReadPixels(0, 0, x, y, GL_BGR, GL_UNSIGNED_BYTE).unpack("H*")[0]
    
    data = ""
    y.times do |j|
      (x * 3).times {|i| data += imgdata[6 * x * j + i * 2, 2].to_i(16).chr}
      data += "\x00" * ((4 - 3 * x % 4) % 4)
    end
    
    bitmap.buf = data
    bitmap.write(fname)
  end
end


if __FILE__ ==  $0
  def init
    glClearColor(0, 0, 255, 1)
  end
  
  display = proc do 
    vertex = [-0.9, 0.9, 0.9, 0.9, 0, -0.9]
    glClear(GL_COLOR_BUFFER_BIT)
    glEnableClientState(GL_VERTEX_ARRAY)
    glVertexPointer(2, GL_FLOAT, 0, vertex)
    glLineWidth(3)             #線の太さ
    glColor3f(0, 1, 0)
    glDrawArrays(GL_LINE_LOOP, 0, 3)
    BitMap.gl_capture("sample.bmp")
    glFlush
  end

  Width = Height = 101
  glutInit
  glutInitWindowSize(Width, Height)
  glutInitWindowPosition(200, 100)
  glutCreateWindow("Triangle")
  init
  glutDisplayFunc(display)
  glutMainLoop
end