Add support for classification of non-fiction books.

[quanlib.git] / cover.rb

diff --git a/cover.rb b/cover.rb
index b88f3406267c9963822e46b3eae3c3e3a14081fa..9b974847d087a800518bdccd983f02bc63e4f59b 100644 (file)
--- a/cover.rb
+++ b/cover.rb
@@ -1,11 +1,19 @@
 
 class Cover
   def initialize(inputStream, path, mimeType)
-    @data = inputStream.read
+    if nil != inputStream
+      @data = inputStream.read
+    else
+      @data = nil
+    end
     @path = path
     @mimeType = mimeType
   end
 
+  def path 
+    @path
+  end
+
   def inspect
     info = []
     if nil != @data
@@ -22,12 +30,17 @@ class Cover
     return '(Cover:' + info.join(',') + ')'
   end
 
+  def read_image(filename)
+    open(filename, 'rb') do |fd|
+      @data = fd.read()
+    end
+  end
+
   def to_s
     return inspect
   end
 
-  def writeImage(outputDir, baseName)
-    filename = baseName + getExt()
+  def write_image(outputDir, filename)
     open(outputDir + '/' + filename, 'wb') do |fd|
       fd.write(@data)
     end