aboutsummaryrefslogtreecommitdiff
path: root/AT91SAM7S256/armdebug/nxt-python-fantom/nxt/fantomglue.py
blob: ac125ed876d1655da8919a63f1ef86aff27432a4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# fantom.py module -- Glue code from NXT_Python to Fantom, allowing
# NXT_Python to run on Mac without modification.  Supports subset of
# PyBluez/bluetooth.py used by NXT_Python.
#
# Copyright (C) 2011  Tat-Chee Wan
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

import pyfantom

RFCOMM=11           # lightblue const
FANTOM_BT = RFCOMM  # For compatibilty with lightblue
FANTOM_USB = 0

# Bluetooth Iterator
def discover_devices(lookup_names=False):  # parameter is ignored
    pairs = []
    for d in pyfantom.NXTIterator(True):
        # h: host, n: name
        h = d.get_resource_string()
        nxt = d.get_nxt()
        device_info = nxt.get_device_info()
        n = device_info.name
        del nxt
        pairs.append((h, n))
    return pairs

class BluetoothSocket:

    def __init__(self, proto = FANTOM_BT, _sock=None):
        # We instantiate a NXT object only when we connect if none supplied
        self._sock = _sock
        self._proto = proto

    def connect(self, addrport):
        addr, port = addrport
        if self._sock is None:
            # Port is ignored
            self._sock = pyfantom.NXT(addr)
    
    def send(self, data):
        return self._sock.write( data )

    def recv(self, numbytes):
        return self._sock.read( numbytes )
    
    def close(self):
        if self._sock is not None:
            del self._sock
    
class BluetoothError(IOError):
    pass    

def _check_brick(arg, value):
    return arg is None or arg == value


def find_devices(lookup_names=False):  # parameter is ignored
    devicelist = []
    for d in pyfantom.NXTIterator(False):
        #name = d.get_name()
        #print "NXT name: ", name
        nxt = d.get_nxt()
        devicelist.append(nxt)
    return devicelist

# FIXME: The semantics of find_devices is different from discover_devices
#
#        # h: host, n: name
#        hostmatched = None
#        namematched = None
#        h = d.get_resource_string()
#        nxt = d.get_nxt()
#        device_info = nxt.get_device_info()
#        n = device_info.name
#        if _check_brick(host, h) and _check_brick(name, n):
#            yield nxt
#        else:
#            del nxt             # Does not match criteria

class USBSocket:

    def __init__(self, device=None):
        # We instantiate a NXT object only when we connect if none supplied
        #self.device = device
        self._sock = device
        self.debug = False

    def device_name(self):
        devinfo = self._sock.deviceinfo()
        return devinfo.name

    def connect(self):
        if self._sock is None:
            # Port is ignored
            self._sock = pyfantom.NXT(addr)
    
    def send(self, data):
        return self._sock.write( data )

    def recv(self, numbytes):
        return self._sock.read( numbytes )
    
    def close(self):
        if self._sock is not None:
            del self._sock