adjust the code that adds the "Save Image..." right-click menu item t…

…o work with picts too
racket · Jul 17, 2024 · 648f922 · 648f922
1 parent d735688
commit 648f922
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/drracket/drracket/private/pict-snip.rkt b/drracket/drracket/private/pict-snip.rkt
@@ -54,13 +54,18 @@
     (inherit set-snipclass)
     (set-snipclass snip-class)
 
+    (define/public (get-bitmap)
+      (define bm (make-bitmap (inexact->exact (ceiling w))
+                              (inexact->exact (ceiling h))))
+      (define dc (send bm make-dc))
+      (draw dc 0 0 0 0 w h 0 0 #f)
+      (send dc set-bitmap #f)
+      bm)
+
     (define/public (convert r d)
       (case r
         [(png-bytes)
-         (define bm (make-bitmap (inexact->exact (ceiling w))
-                                 (inexact->exact (ceiling h))))
-         (define dc (send bm make-dc))
-         (draw dc 0 0 0 0 w h 0 0 #f)
+         (define bm (get-bitmap))
          (define b (open-output-bytes))
          (send bm save-file b 'png)
          (get-output-bytes b)]

diff --git a/drracket/drracket/private/unit.rkt b/drracket/drracket/private/unit.rkt
@@ -30,6 +30,7 @@
          "parse-logger-args.rkt"
          drracket/get-module-path
          "named-undefined.rkt"
+         (prefix-in pict-snip: "pict-snip.rkt")
          (prefix-in drracket:arrow: "../arrow.rkt")
          (prefix-in icons: images/compile-time)
          mred
@@ -165,7 +166,8 @@
                  (let ([snip (send text find-snip pos 'after-or-none)])
                    (when (or (is-a? snip image-snip%)
                              (is-a? snip image-core:image%)
-                             (is-a? snip cache-image-snip%))
+                             (is-a? snip cache-image-snip%)
+                             (is-a? snip pict-snip:pict-snip%))
                      (add-sep)
                      (define (save-image-callback _1 _2)
                        (define fn
@@ -178,7 +180,8 @@
                            [kind
                             (cond
                               [(or (is-a? snip image-snip%)
-                                   (is-a? snip cache-image-snip%))
+                                   (is-a? snip cache-image-snip%)
+                                   (is-a? snip pict-snip:pict-snip%))
                                (send (send snip get-bitmap) save-file fn kind)]
                               [else
                                (image-core:save-image-as-bitmap snip fn kind)])]