optimize the scratch? method

asciidoctor · Jun 9, 2015 · b27e700 · b27e700
1 parent b4a0688
commit b27e700
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/asciidoctor-pdf/prawn_ext/extensions.rb b/lib/asciidoctor-pdf/prawn_ext/extensions.rb
@@ -519,10 +519,10 @@ def get_scratch_document
     end
   end
 
-  def is_scratch?
-    !!state.store.info.data[:Scratch]
+  def scratch?
+    (@_label ||= (state.store.info.data[:Scratch] ? :scratch : :primary)) == :scratch
   end
-  alias :scratch? :is_scratch?
+  alias :is_scratch? :scratch?
 
   # TODO document me
   def dry_run &block