portier

pitfall/pitfall/core.rkt

@@ -1,4 +1,5 @@
 #lang racket/base
+(require racket/match racket/port)
 (provide (all-defined-out))
 
 ;; structs
@@ -15,7 +16,6 @@
 
 (define current-pdf-version (make-parameter 1.3))
 (define current-auto-first-page (make-parameter #t))
-(define current-doc-offset (make-parameter 'doc-offset-not-initialized))
 
 ;; helpers
 
@@ -25,4 +25,13 @@
   (let ([x (if round? (/ (round (* x 1e6)) 1e6) x)])
     (number->string (if (integer? x)
                         (inexact->exact x)
-                        x))))
+                        x))))
+
+(define (to-bytes x)
+  (match x
+    [(? bytes?) x]
+    [(? input-port?) (port->bytes x)]
+    [_ (string->bytes/latin-1 (string-append x "\n"))]))
+
+(define (write-bytes-out x)
+  (void (write-bytes (to-bytes x))))

pitfall/pitfall/document.rkt

@@ -28,7 +28,7 @@
                                      (yield refid)
                                      (loop (add1 refid))))]
            [@root (ref (mhasheq 'Type "Catalog"
-                                'Pages (ref (mhasheq 'Type "Pages"))))] ; top object
+                                'Pages (ref (mhasheq 'Type "Pages"))))]
            [(@x x) 0]
            [(@y y) 0]
            [@info (mhasheq 'Producer "PITFALL"
@@ -37,7 +37,9 @@
                                                             0
                                                             (current-seconds)) #f))])  ; Initialize the metadata
 
-    ;; Initialize mixins
+    
+    
+    ;; initialize mixins
     (send this initColor)
     (send this initVector)
     (inherit-field _ctm)
@@ -49,22 +51,19 @@
     ;; initialize params
     (current-compress-streams? (hash-ref @options 'compress #t))
     (current-auto-first-page (hash-ref @options 'autoFirstPage #t))
-    (current-doc-offset 0)
     (when (current-auto-first-page) (add-page))
 
+    ;; copy options
+    (for ([(key val) (in-hash (hash-ref @options 'info (hasheq)))]) 
+         (hash-set! @info key val))
+
     (define/public (page) (first @pages))
 
     (define/public (ref [payload (mhasheq)])
-      (define refid (@ref-gen))
+      (define new-ref (make-object PDFReference this (@ref-gen) payload))
-      (define new-ref (make-object PDFReference this refid payload))
       (set! @refs (cons new-ref @refs))
       new-ref)
 
-    (define/public (write x)
-      (define bstr (if (bytes? x) x (string->bytes/latin-1 (string-append x "\n"))))
-      (write-bytes bstr)
-      (current-doc-offset (file-position (current-output-port))))
-      
     (define/public (add-page [options-arg @options])
       ;; create a page object
       (define page-parent (send @root get-key 'Pages))
@@ -84,10 +83,11 @@
       this)
 
     (define/public (end)
-      (write (format "%PDF-~a\n%ÿÿÿÿ" (current-pdf-version)))
+      (write-bytes-out (format "%PDF-~a" (current-pdf-version)))
+      (write-bytes-out "%ÿÿÿÿ")
 
-      (for ([p (in-list @pages)])
+      (for ([page (in-list @pages)])
-           (send p end))
+           (send page end))
 
       (define doc-info (ref))
       (for ([(key val) (in-hash @info)])
@@ -104,34 +104,20 @@
       
       (send @root end)
       
-      (define xref-offset (current-doc-offset))
+      (define xref-offset (file-position (current-output-port)))
-      (match-define (list this-idxs this-offsets)
+      (write-bytes-out "xref")
-        (match (reverse @refs)
+      (write-bytes-out (format "0 ~a" (add1 (length @refs))))
-          [(list refs ...) (list (map (λ (ref) (get-field id ref)) refs)
+      (write-bytes-out "0000000000 65535 f ")
-                                 (map (λ (ref) (get-field offset ref)) refs))]))
+      (for ([ref (in-list (reverse @refs))])
-      (write "xref")
+           (write-bytes-out
-      (write (format "0 ~a" (add1 (length this-offsets))))
+            (string-append (~r (get-field offset ref) #:min-width 10 #:pad-string "0") " 00000 n ")))
-      (write "0000000000 65535 f ")
+      (write-bytes-out "trailer")
-      (let ([missing-offsets (for/list ([offset (in-list this-offsets)]
+      (write-bytes-out (convert (mhasheq 'Size (add1 (length @refs))
-                                        [idx (in-list this-idxs)]
+                                         'Root @root
-                                        #:unless (number? offset))
+                                         'Info doc-info)))
-                                       idx)])
+      (write-bytes-out "startxref")
-        (unless (empty? missing-offsets)
+      (write-bytes-out (numberizer xref-offset))
-          (raise-argument-error 'document:end "numerical offsets" missing-offsets)))
+      (write-bytes-out "%%EOF"))))
-      (for ([offset (in-list this-offsets)]
-            [idx (in-list this-idxs)])
-           (write (string-append (~r offset #:min-width 10 #:pad-string "0") " 00000 n ")))
-    
-      (write "trailer")
-      (write (convert (mhasheq 'Size (add1 (length this-offsets))
-                               'Root @root
-                               'Info doc-info)))
-      (write "startxref")
-      (write (numberizer xref-offset))
-      (write "%%EOF"))
-
-    (for ([(key val) (in-hash (hash-ref @options 'info (hasheq)))]) 
-         (hash-set! @info key val))))
 
 (module+ test
   (define d (new PDFDocument)))

pitfall/pitfall/jpeg-structy.rkt

@@ -61,7 +61,7 @@
     ;; min and max values from the default, we invert the colors. See
     ;; section 4.8.4 of the spec. 
     (when (equal? (· this colorSpace) "DeviceCMYK")
-      (send obj set-key! 'Decode '(1.0 0.0 1.0 0.0 1.0 0.0 1.0 0.0)))
+      (send (· this obj) set-key! 'Decode '(1.0 0.0 1.0 0.0 1.0 0.0 1.0 0.0)))
 
     (port-position (· this data) 0)
     (send (· this obj) end (· this data))))

pitfall/pitfall/reference.rkt

@@ -14,14 +14,10 @@
                 [(@id id)]
                 [(@payload payload) (make-hasheq)])
     (field [(@offset offset) #f]
-           [@portal (open-output-bytes)])
+           [@port (open-output-bytes)])
 
-    (define/public (write x [op @portal])
+    (define/public (write x)
-      (define bstr (match x
+      (write-bytes (to-bytes x) @port))
-                     [(? bytes?) x]
-                     [(? input-port?) (port->bytes x)]
-                     [_ (string->bytes/latin-1 (format "~a\n" x))]))
-      (write-bytes bstr op))
 
     (define/public (get-key key)
       (hash-ref @payload key))
@@ -33,12 +29,12 @@
       (hash-update! @payload key updater))
 
     (define/public (end)
-      (set! @offset (current-doc-offset))
+      (set! @offset (file-position (current-output-port)))
   
-      (send @doc write (format "~a 0 obj" @id))
+      (write-bytes-out (format "~a 0 obj" @id))
         
       (define bstr
-        (let ([bstr (get-output-bytes @portal)])
+        (let ([bstr (get-output-bytes @port)])
           (cond
             [(zero? (bytes-length bstr)) #false]
             [(and (current-compress-streams?) (not (hash-ref @payload 'Filter #f)))
@@ -48,12 +44,12 @@
         
       (when bstr
         (hash-set! @payload 'Length (bytes-length bstr)))
-      (send @doc write (convert @payload))
+      (write-bytes-out (convert @payload))
         
       (when bstr
-        (send @doc write (bytes-append #"stream\n" bstr #"\nendstream")))
+        (write-bytes-out (bytes-append #"stream\n" bstr #"\nendstream")))
       
-      (send @doc write "\nendobj"))
+      (write-bytes-out "\nendobj"))
 
     (define/public (to-string)
       (format "~a 0 R" @id))))