opengl_lib.rb in mittsu-0.1.4

- old
+ new
@@ -11,41 +11,109 @@
       fail "Unsupported platform."
     end
   end
 
   class Linux
+    def initialize(loader = Linux)
+      @loader = loader
+    end
+
     def path
-      # http://www.pilotlogic.com/sitejoom/index.php/wiki?id=398<F37>
-      # 32              64
-      # /usr/lib        /usr/lib64       redhat, mandriva
-      # /usr/lib32      /usr/lib64       arch, gento
-      # /usr/lib        /usr/lib64       slackware
-      # /usr/lib/i386.. /usr/libx86_64/  debian
-      libs = Dir.glob("/usr/lib*/**/libGL.so")
-      if libs.size == 0
-        puts "no libGL.so"
-        exit 1
-      end
-      case kernel_module_in_use
-      when /nvidia/
-        return File.dirname(libs.grep(/nvidia/)[0])
-      end
-      # Get the same architecture that the runnning ruby
-      if 1.size == 8 # 64 bits
-        File.dirname(libs.grep(/64/)[0])
-      else # 32 bits
-        File.dirname(libs[0])
-      end
+      return nil if file_path.nil?
+      File.dirname file_path
     end
 
     def file
-      'libGL.so'
+      return nil if file_path.nil?
+      File.basename file_path
     end
 
-    private
+    class << self
       def kernel_module_in_use
         lspci_line = `lspci -nnk | grep -i vga -A3 | grep 'in use'`
-        return /in use:\s*(\S+)/ =~ lspci_line && $1
+        /in use:\s*(\S+)/ =~ lspci_line && $1
+      rescue
+        ''
+      end
+
+      def libgl_paths
+        Dir.glob('/usr/lib*/**/libGL.so*')
+      rescue
+        []
+      end
+
+      def sixtyfour_bits?
+        1.size == 8
+      end
+
+      def ldconfig
+        `ldconfig -p | grep 'libGL\\.so'`.lines
+      rescue
+        []
+      end
+    end
+
+    private
+      def file_path
+        @_file_path ||= begin
+          ldconfig_lib || driver_specific_lib || sixtyfour_bit_lib || fallback_lib || give_up
+        end.tap do |result|
+          print_debug_info(result) if Mittsu::DEBUG
+        end
+      end
+
+      def ldconfig_lib
+        return nil if ldconfig.empty?
+        @_debug = { source: 'ldconfig', info: ldconfig.inspect } if Mittsu::DEBUG
+        ldconfig_for_arch = ldconfig.reject { |lib| @loader.sixtyfour_bits? ^ ldconfig_64?(lib) }
+        ldconfig_for_arch.first.match(/=> (\/.*)$/)[1]
+      end
+
+      def ldconfig_64?(lib)
+        lib =~ /\([^\)]*64[^\)]*\) =>/
+      end
+
+      def driver_specific_lib
+        return nil if libs.empty?
+        @_debug = { source: 'driver', info: kernel_module } if Mittsu::DEBUG
+        libs.select { |lib| lib =~ /nvidia/ }.first if kernel_module =~ /nvidia/
+      end
+
+      def sixtyfour_bit_lib
+        return nil if libs.empty?
+        sixtyfour_bit_libs = libs.select { |lib| lib =~ /64/ }
+        @_debug = { source: '64', info: sixtyfour_bit_libs.inspect } if Mittsu::DEBUG
+        sixtyfour_bit_libs.first if @loader.sixtyfour_bits?
+      end
+
+      def fallback_lib
+        return nil if libs.empty?
+        @_debug = { source: 'fallback', info: libs.inspect } if Mittsu::DEBUG
+        libs.first
+      end
+
+      def give_up
+        @_debug = { source: 'none', info: nil } if Mittsu::DEBUG
+        nil
+      end
+
+      def kernel_module
+        @_kernel_module ||= @loader.kernel_module_in_use
+      end
+
+      def libs
+        @_libs ||= @loader.libgl_paths.sort_by(&:length)
+      end
+
+      def ldconfig
+        @_ldconfig ||= @loader.ldconfig
+      end
+
+      def print_debug_info(result)
+        puts "Loading lib path: #{result.inspect}"
+        puts "Source: #{@_debug[:source]}"
+        puts "Info: #{@_debug[:info]}"
+        puts "64-bit? #{@loader.sixtyfour_bits?}"
       end
   end
 
   # TODO
   class Windows