button: Add SetImage

roblillack · roblillack · commit 0512f302184a · 2024-05-27T12:30:11.000+02:00
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;
+}

Original file line number	Diff line number	Diff line change
`@@ -108,3 +108,9 @@ int Button_State(ButtonPtr btnPtr) {`
`108`	`108`	`return 3;`
`109`	`109`	`}`
`110`	`110`	`}`
	`111`	`+`
	`112`	`+void Button_SetImage(ButtonPtr ptr, ImagePtr imagePtr) {`
	`113`	`+ NSImage theImage = (NSImage )imagePtr;`
	`114`	`+ NSButton control = (NSButton )ptr;`
	`115`	`+ [control setImage:theImage];`
	`116`	`+}`