ble/ble_darwin.go

379 lines
7.9 KiB
Go
Raw Normal View History

2019-10-23 18:28:19 -04:00
package ble
import "C"
import (
"encoding/binary"
"fmt"
"runtime"
"sync"
"time"
"unsafe"
2019-10-23 18:28:19 -04:00
"git.wow.st/gmp/ble/ns"
)
type BLE struct {
events chan interface{}
state ns.CBManagerState
peripherals Peripherals
hr int
ready, wantScan bool
2019-10-23 18:28:19 -04:00
sync.Mutex
cd *ns.CBDelegate
cm *ns.CBCentralManager
2019-10-23 18:28:19 -04:00
}
var cdLookup map[unsafe.Pointer] *BLE
var pdLookup map[unsafe.Pointer] *BLE
2019-10-23 18:28:19 -04:00
type State string
type Peripheral struct {
Name string
RSSI int
Identifier string
2019-10-23 18:28:19 -04:00
p *ns.CBPeripheral
}
func peripheralName(p *ns.CBPeripheral) string {
var ret string
nsname := p.Name()
if nsname.Ptr() != nil {
ret = nsname.String()
}
return ret
}
func newPeripheral(x *ns.CBPeripheral) Peripheral {
return Peripheral{
Name: peripheralName(x),
Identifier: x.Identifier().UUIDString().String(),
p: x,
}
2019-10-23 18:28:19 -04:00
}
type PeripheralListItem struct {
p Peripheral
seen time.Time
}
type Peripherals struct {
items []PeripheralListItem
sync.Mutex
}
func (ps *Peripherals) Add(x Peripheral) bool {
ps.Lock()
defer ps.Unlock()
found := false
for n,item := range ps.items {
if item.p.Identifier == x.Identifier {
2019-10-23 18:28:19 -04:00
item.p = x
item.seen = time.Now()
ps.items[n] = item
found = true
break
}
}
if found {
return false
}
//take ownership of this Objective-C object
x.p.Retain()
x.p.GC()
item := PeripheralListItem {p: x, seen: time.Now()}
ps.items = append(ps.items, item)
return true
}
type UpdateStateEvent struct {
State string
}
type DiscoverEvent struct {
Peripheral Peripheral
}
type ConnectEvent struct {
Peripheral Peripheral
}
func (b *BLE) Events() chan interface{} {
return b.events
}
func (b *BLE) HR() int {
b.Lock()
defer b.Unlock()
return b.hr
}
func stringState(x ns.CBManagerState) string {
switch x {
case (ns.CBManagerState)(ns.CBManagerStateResetting):
return "resetting"
case (ns.CBManagerState)(ns.CBManagerStateUnsupported):
return "unsupported"
case (ns.CBManagerState)(ns.CBManagerStateUnauthorized):
return "unauthorized"
case (ns.CBManagerState)(ns.CBManagerStatePoweredOff):
return "powered off"
case (ns.CBManagerState)(ns.CBManagerStatePoweredOn):
return "powered on"
case (ns.CBManagerState)(ns.CBManagerStateUnknown):
return "unknown"
default:
return "no state"
}
}
func (b *BLE) State() string {
b.Lock()
defer b.Unlock()
return stringState(b.state)
2019-10-23 18:28:19 -04:00
}
func (b *BLE) setState(x ns.CBManagerState) {
b.Lock()
defer b.Unlock()
b.state = x
if b.ready && b.wantScan {
2019-10-23 18:28:19 -04:00
go func() {
fmt.Printf("Go: Scanning\n")
b.cm.ScanForPeripheralsWithServices(nil, nil)
2019-10-23 18:28:19 -04:00
}()
}
}
func (b *BLE) Scan() {
b.Lock()
if b.state != (ns.CBManagerState)(ns.CBManagerStatePoweredOn) {
b.wantScan = true
b.Unlock()
} else {
b.Unlock()
fmt.Printf("Go: Scanning\n")
b.cm.ScanForPeripheralsWithServices(nil, nil)
2019-10-23 18:28:19 -04:00
}
}
func (b *BLE) StopScan() {
b.Lock()
if !b.ready {
b.Unlock()
return
}
b.Unlock()
2019-10-23 18:28:19 -04:00
fmt.Printf("Go: stopping scan\n")
b.cm.StopScan()
2019-10-23 18:28:19 -04:00
}
func (b *BLE) Connect(p Peripheral) {
b.Lock()
if !b.ready {
b.Unlock()
return
}
b.Unlock()
if p.p == nil {
ps := b.cm.RetrievePeripheralsWithIdentifiers(ns.NSArrayWithObjects(ns.NSUUIDAlloc().InitWithUUIDString(ns.NSStringWithGoString(p.Identifier))))
if (int)(ps.Count()) > 0 {
p.p = ps.ObjectAtIndex(0).CBPeripheral()
} else {
return
}
}
b.cm.ConnectPeripheral(p.p, nil)
fmt.Printf("cm.ConnectPeripheral() returned\n")
2019-10-23 18:28:19 -04:00
}
func updateState(c *ns.CBCentralManager) {
ble := cdLookup[c.Ptr()]
2019-10-23 18:28:19 -04:00
fmt.Printf("Go: did update state\n")
st := c.CBManager.State()
if st == (ns.CBManagerState)(ns.CBManagerStatePoweredOn) {
ble.ready = true
} else {
ble.ready = false
}
2019-10-23 18:28:19 -04:00
ble.setState(st)
ble.events <- UpdateStateEvent{State: stringState(st)}
}
func discoverPeripheral(c *ns.CBCentralManager, p *ns.CBPeripheral, d *ns.NSDictionary, rssi *ns.NSNumber) {
ble := cdLookup[c.Ptr()]
peripheral := newPeripheral(p)
peripheral.RSSI = (int)(rssi.IntValue())
2019-10-23 18:28:19 -04:00
if peripheral.Name == "" {
return
}
if ok := ble.peripherals.Add(peripheral); ok {
pdLookup[p.Ptr()] = ble
2019-10-23 18:28:19 -04:00
ble.events <- DiscoverEvent{Peripheral: peripheral}
}
}
func connectPeripheral(c *ns.CBCentralManager, p *ns.CBPeripheral) {
fmt.Printf("Did connect peripheral\n")
b := cdLookup[c.Ptr()]
2019-10-23 18:28:19 -04:00
// set ourselves up as a peripheral delegate
p.SetDelegate(b.cd)
2019-10-23 18:28:19 -04:00
uuidstring := p.Identifier().UUIDString().String()
peripheral := newPeripheral(p)
2019-10-23 18:28:19 -04:00
found := false
b.peripherals.Lock()
for _, item := range b.peripherals.items {
if item.p.Identifier == uuidstring {
2019-10-23 18:28:19 -04:00
peripheral = item.p
found = true
break
}
}
b.peripherals.Unlock()
2019-10-23 18:28:19 -04:00
if !found {
peripheral.Name = peripheralName(p)
if ok := b.peripherals.Add(peripheral); ok {
pdLookup[p.Ptr()] = b
}
2019-10-23 18:28:19 -04:00
}
b.events <- ConnectEvent{peripheral}
fmt.Printf("Go: connectPeripheral returning\n")
2019-10-23 18:28:19 -04:00
}
func DiscoverServices(x Peripheral) {
fmt.Printf("Discovering services on %s\n", x.Name)
p := x.p
// discover all services on this device
p.DiscoverServices(nil)
}
2019-10-23 18:28:19 -04:00
func discoverServices(p *ns.CBPeripheral, e *ns.NSError) {
fmt.Printf("Did discover services\n")
p.Services().ObjectEnumerator().ForIn(func(o *ns.Id) bool {
serv := o.CBService()
uuid := serv.UUID()
switch {
case uuid.IsEqualTo(hrm_uuid):
fmt.Printf("--heart rate monitor service\n")
p.DiscoverCharacteristics(nil, serv)
case uuid.IsEqualTo(info_uuid):
fmt.Printf("--device information service\n")
p.DiscoverCharacteristics(nil, serv)
default:
fmt.Printf("--unknown service\n")
}
return true
})
fmt.Printf("Go: discoverServices returning\n")
}
func hr(d *ns.NSData) int {
if l := int(d.Length()); l < 4 {
return 0
}
x := C.GoBytes(d.Bytes(), 4)
flags := x[0]
if flags&0x80 != 0 { // uint16 format
return int(binary.BigEndian.Uint16(x[1:2]))
} else {
return int(x[1])
}
}
func discoverCharacteristics(p *ns.CBPeripheral, s *ns.CBService, e *ns.NSError) {
fmt.Printf("Did discover characteristics\n")
uuid := s.UUID()
fmt.Printf("----%s\n", uuid.UUIDString())
if uuid.IsEqualTo(hrm_uuid) {
s.Characteristics().ObjectEnumerator().ForIn(func(o *ns.Id) bool {
chr := o.CBCharacteristic()
chuuid := chr.UUID()
fmt.Printf("------%s\n", chuuid.UUIDString())
if chuuid.IsEqualTo(hrv_uuid) {
p.SetNotifyValue(1, chr)
v := chr.Value()
fmt.Println(hr(v))
}
return true
})
}
fmt.Printf("Go: discoverCharacteristics returning\n")
}
func updateValue(p *ns.CBPeripheral, chr *ns.CBCharacteristic, e *ns.NSError) {
//ble := pdLookup[p]
ble := gble
2019-10-23 18:28:19 -04:00
if chr.UUID().IsEqualTo(hrv_uuid) {
v := chr.Value()
ble.Lock()
ble.hr = hr(v)
ble.Unlock()
fmt.Printf("Heart rate: %d\n", ble.hr)
}
}
var (
hrm_uuid *ns.CBUUID
hrv_uuid *ns.CBUUID
info_uuid *ns.CBUUID
gble *BLE
2019-10-23 18:28:19 -04:00
)
func NewBLE() *BLE {
ps := Peripherals{items: make([]PeripheralListItem,0)}
ble := &BLE{events: make(chan interface{}), peripherals: ps}
gble = ble
2019-10-23 18:28:19 -04:00
queue := ns.DispatchQueueCreate(ns.CharWithGoString("go_hrm_queue"), nil)
cd := ns.CBDelegateAlloc()
2019-10-23 18:28:19 -04:00
cd.CentralManagerDidUpdateStateCallback(updateState)
cd.CentralManagerDidDiscoverPeripheralCallback(discoverPeripheral)
cd.CentralManagerDidConnectPeripheralCallback(connectPeripheral)
cd.PeripheralDidDiscoverServicesCallback(discoverServices)
cd.PeripheralDidDiscoverCharacteristicsForServiceCallback(discoverCharacteristics)
cd.PeripheralDidUpdateValueForCharacteristicCallback(updateValue)
hrm_uuid = ns.CBUUIDWithGoString("180D")
hrv_uuid = ns.CBUUIDWithGoString("2A37")
info_uuid = ns.CBUUIDWithGoString("180A")
ble.cd = cd
if cdLookup == nil {
cdLookup = make(map[unsafe.Pointer]*BLE,0)
}
if pdLookup == nil {
pdLookup = make(map[unsafe.Pointer]*BLE,0)
}
2019-10-23 18:28:19 -04:00
// We defined our own queue because this won't work on the main queue.
ble.cm = ns.CBCentralManagerAlloc().InitWithDelegateQueue(cd, queue)
cdLookup[ble.cm.Ptr()] = ble
2019-10-23 18:28:19 -04:00
// For debugging purposes, run GC every second to make sure things are
// not over-released.
go func() {
for {
runtime.GC()
time.Sleep(time.Second * 30)
}
}()
return ble
}