button: Add SetImage

roblillack · May 27, 2024 · 0512f30 · 0512f30
1 parent 0aa6fa8
commit 0512f30
Show file tree

Hide file tree

Showing 5 changed files with 46 additions and 0 deletions.
diff --git a/button.go b/button.go
@@ -3,9 +3,12 @@ package gocoa
 // #cgo CFLAGS: -x objective-c
 // #cgo LDFLAGS: -framework Cocoa
 // #import "button.h"
+// #import "image.h"
+// #include <stdlib.h>
 import "C"
 import (
 	"fmt"
+	"image"
 	"unsafe"
 )
 
@@ -147,3 +150,10 @@ func (btn *Button) OnClick(fn func()) {
 func (btn *Button) Remove() {
 	C.Button_Remove(btn.buttonPtr)
 }
+
+func (btn *Button) SetImage(img *image.RGBA) {
+	bytes := C.CBytes(img.Pix)
+	nsImage := C.Image_NewWithRGBA(C.int(img.Bounds().Dx()), C.int(img.Bounds().Dy()), (*C.uchar)(bytes))
+	C.Button_SetImage(btn.buttonPtr, nsImage)
+	C.free(bytes)
+}
diff --git a/button.h b/button.h
@@ -1,3 +1,4 @@
+#import "image.h"
 #import <Cocoa/Cocoa.h>
 
 // typedef void (*callback)(void);
@@ -25,3 +26,4 @@ void Button_SetBorderColor(ButtonPtr btnPtr, int r, int g, int b, int a);
 void Button_SetBorderWidth(ButtonPtr btnPtr, int borderWidth);
 void Button_SetState(ButtonPtr btnPtr, int state);
 int Button_State(ButtonPtr btnPtr);
+void Button_SetImage(ButtonPtr btnPtr, ImagePtr imagePtr);
diff --git a/button.m b/button.m
@@ -108,3 +108,9 @@ int Button_State(ButtonPtr btnPtr) {
         return 3;
     }
 }
+
+void Button_SetImage(ButtonPtr ptr, ImagePtr imagePtr) {
+  NSImage *theImage = (NSImage *)imagePtr;
+  NSButton *control = (NSButton *)ptr;
+  [control setImage:theImage];
+}
diff --git a/image.h b/image.h
@@ -0,0 +1,4 @@
+#import <Cocoa/Cocoa.h>
+
+typedef void *ImagePtr;
+ImagePtr Image_NewWithRGBA(int w, int h, unsigned char *rgba);
diff --git a/image.m b/image.m
@@ -0,0 +1,24 @@
+#import "image.h"
+#include "_cgo_export.h"
+
+ImagePtr Image_NewWithRGBA(int w, int h, unsigned char *rgba) {
+  NSBitmapImageRep *bitmap = [[NSBitmapImageRep alloc]
+      initWithBitmapDataPlanes:NULL
+                    pixelsWide:w
+                    pixelsHigh:h
+                 bitsPerSample:8
+               samplesPerPixel:4
+                      hasAlpha:YES
+                      isPlanar:NO
+                colorSpaceName:NSCalibratedRGBColorSpace
+                   bytesPerRow:w * 4
+                  bitsPerPixel:32];
+
+  unsigned char *bitmapData = [bitmap bitmapData];
+  memcpy(bitmapData, rgba, w * h * 4);
+
+  NSImage *theImage = [[NSImage alloc] init];
+  [theImage addRepresentation:bitmap];
+
+  return (ImagePtr)theImage;
+}