[libsigrokdecode.git] / decoders / mxc6225xu / pd.py

##
## This file is part of the libsigrokdecode project.
##
## Copyright (C) 2012 Uwe Hermann <uwe@hermann-uwe.de>
##
## 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 2 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.
##
## You should have received a copy of the GNU General Public License
## along with this program; if not, see <http://www.gnu.org/licenses/>.
##

import sigrokdecode as srd

# Definitions of various bits in MXC6225XU registers.
status = {
    # SH[1:0]
    'sh': {
        0b00: 'none',
        0b01: 'shake left',
        0b10: 'shake right',
        0b11: 'undefined',
    },
    # ORI[1:0] and OR[1:0] (same format)
    'ori': {
        0b00: 'vertical in upright orientation',
        0b01: 'rotated 90 degrees clockwise',
        0b10: 'vertical in inverted orientation',
        0b11: 'rotated 90 degrees counterclockwise',
    },
    # SHTH[1:0]
    'shth': {
        0b00: '0.5g',
        0b01: '1.0g',
        0b10: '1.5g',
        0b11: '2.0g',
    },
    # SHC[1:0]
    'shc': {
        0b00: '16',
        0b01: '32',
        0b10: '64',
        0b11: '128',
    },
    # ORC[1:0]
    'orc': {
        0b00: '16',
        0b01: '32',
        0b10: '64',
        0b11: '128',
    },
}

class Decoder(srd.Decoder):
    api_version = 2
    id = 'mxc6225xu'
    name = 'MXC6225XU'
    longname = 'MEMSIC MXC6225XU'
    desc = 'Digital Thermal Orientation Sensor (DTOS) protocol.'
    license = 'gplv2+'
    inputs = ['i2c']
    outputs = ['mxc6225xu']
    annotations = (
        ('text', 'Human-readable text'),
    )

    def __init__(self):
        self.state = 'IDLE'

    def start(self):
        self.out_ann = self.register(srd.OUTPUT_ANN)

    def putx(self, data):
        self.put(self.ss, self.es, self.out_ann, data)

    def handle_reg_0x00(self, b):
        # XOUT: 8-bit x-axis acceleration output.
        # Data is in 2's complement, values range from -128 to 127.
        self.putx([0, ['XOUT: %d' % b]])

    def handle_reg_0x01(self, b):
        # YOUT: 8-bit y-axis acceleration output.
        # Data is in 2's complement, values range from -128 to 127.
        self.putx([0, ['YOUT: %d' % b]])

    def handle_reg_0x02(self, b):
        # STATUS: Orientation and shake status.

        # Bits[7:7]: INT
        int_val = (b >> 7) & 1
        s = 'unchanged and no' if (int_val == 0) else 'changed or'
        ann = 'INT = %d: Orientation %s shake event occurred\n' % (int_val, s)

        # Bits[6:5]: SH[1:0]
        sh = (((b >> 6) & 1) << 1) | ((b >> 5) & 1)
        ann += 'SH[1:0] = %s: Shake event: %s\n' % \
               (bin(sh)[2:], status['sh'][sh])

        # Bits[4:4]: TILT
        tilt = (b >> 4) & 1
        s = '' if (tilt == 0) else 'not '
        ann += 'TILT = %d: Orientation measurement is %svalid\n' % (tilt, s)

        # Bits[3:2]: ORI[1:0]
        ori = (((b >> 3) & 1) << 1) | ((b >> 2) & 1)
        ann += 'ORI[1:0] = %s: %s\n' % (bin(ori)[2:], status['ori'][ori])

        # Bits[1:0]: OR[1:0]
        or_val = (((b >> 1) & 1) << 1) | ((b >> 0) & 1)
        ann += 'OR[1:0] = %s: %s\n' % (bin(or_val)[2:], status['ori'][or_val])

        # ann += 'b = %s\n' % (bin(b))

        self.putx([0, [ann]])

    def handle_reg_0x03(self, b):
        # DETECTION: Powerdown, orientation and shake detection parameters.
        # Note: This is a write-only register.

        # Bits[7:7]: PD
        pd = (b >> 7) & 1
        s = 'Do not power down' if (pd == 0) else 'Power down'
        ann = 'PD = %d: %s the device (into a low-power state)\n' % (pd, s)

        # Bits[6:6]: SHM
        shm = (b >> 6) & 1
        ann = 'SHM = %d: Set shake mode to %d\n' % (shm, shm)

        # Bits[5:4]: SHTH[1:0]
        shth = (((b >> 5) & 1) << 1) | ((b >> 4) & 1)
        ann += 'SHTH[1:0] = %s: Set shake threshold to %s\n' \
               % (bin(shth)[2:], status['shth'][shth])

        # Bits[3:2]: SHC[1:0]
        shc = (((b >> 3) & 1) << 1) | ((b >> 2) & 1)
        ann += 'SHC[1:0] = %s: Set shake count to %s readings\n' \
               % (bin(shc)[2:], status['shc'][shc])

        # Bits[1:0]: ORC[1:0]
        orc = (((b >> 1) & 1) << 1) | ((b >> 0) & 1)
        ann += 'ORC[1:0] = %s: Set orientation count to %s readings\n' \
               % (bin(orc)[2:], status['orc'][orc])

        self.putx([0, [ann]])

    # TODO: Fixup, this is copy-pasted from another PD.
    # TODO: Handle/check the ACKs/NACKs.
    def decode(self, ss, es, data):
        cmd, databyte = data

        # Store the start/end samples of this I²C packet.
        self.ss, self.es = ss, es

        # State machine.
        if self.state == 'IDLE':
            # Wait for an I²C START condition.
            if cmd != 'START':
                return
            self.state = 'GET SLAVE ADDR'
        elif self.state == 'GET SLAVE ADDR':
            # Wait for an address write operation.
            # TODO: We should only handle packets to the slave(?)
            if cmd != 'ADDRESS WRITE':
                return
            self.state = 'GET REG ADDR'
        elif self.state == 'GET REG ADDR':
            # Wait for a data write (master selects the slave register).
            if cmd != 'DATA WRITE':
                return
            self.reg = databyte
            self.state = 'WRITE REGS'
        elif self.state == 'WRITE REGS':
            # If we see a Repeated Start here, it's a multi-byte read.
            if cmd == 'START REPEAT':
                self.state = 'READ REGS'
                return
            # Otherwise: Get data bytes until a STOP condition occurs.
            if cmd == 'DATA WRITE':
                handle_reg = getattr(self, 'handle_reg_0x%02x' % self.reg)
                handle_reg(databyte)
                self.reg += 1
                # TODO: Check for NACK!
            elif cmd == 'STOP':
                # TODO
                self.state = 'IDLE'
            else:
                pass # TODO
        elif self.state == 'READ REGS':
            # Wait for an address read operation.
            # TODO: We should only handle packets to the slave(?)
            if cmd == 'ADDRESS READ':
                self.state = 'READ REGS2'
                return
            else:
                pass # TODO
        elif self.state == 'READ REGS2':
            if cmd == 'DATA READ':
                handle_reg = getattr(self, 'handle_reg_0x%02x' % self.reg)
                handle_reg(databyte)
                self.reg += 1
                # TODO: Check for NACK!
            elif cmd == 'STOP':
                # TODO
                self.state = 'IDLE'
            else:
                pass # TODO?
Commit	Line	Data
d997c01a	1	##
50bd5d25	2	## This file is part of the libsigrokdecode project.
d997c01a UH	3	##
	4	## Copyright (C) 2012 Uwe Hermann <uwe@hermann-uwe.de>
	5	##
	6	## This program is free software; you can redistribute it and/or modify
	7	## it under the terms of the GNU General Public License as published by
	8	## the Free Software Foundation; either version 2 of the License, or
	9	## (at your option) any later version.
	10	##
	11	## This program is distributed in the hope that it will be useful,
	12	## but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	14	## GNU General Public License for more details.
	15	##
	16	## You should have received a copy of the GNU General Public License
4539e9ca	17	## along with this program; if not, see <http://www.gnu.org/licenses/>.
d997c01a UH	18	##
d997c01a UH	19
d997c01a UH	20	import sigrokdecode as srd
d997c01a UH	21
17160de7	22	# Definitions of various bits in MXC6225XU registers.
d997c01a UH	23	status = {
	24	# SH[1:0]
	25	'sh': {
17160de7	26	0b00: 'none',
d997c01a UH	27	0b01: 'shake left',
	28	0b10: 'shake right',
	29	0b11: 'undefined',
	30	},
	31	# ORI[1:0] and OR[1:0] (same format)
	32	'ori': {
	33	0b00: 'vertical in upright orientation',
	34	0b01: 'rotated 90 degrees clockwise',
	35	0b10: 'vertical in inverted orientation',
	36	0b11: 'rotated 90 degrees counterclockwise',
	37	},
17160de7 UH	38	# SHTH[1:0]
	39	'shth': {
	40	0b00: '0.5g',
	41	0b01: '1.0g',
	42	0b10: '1.5g',
	43	0b11: '2.0g',
	44	},
	45	# SHC[1:0]
	46	'shc': {
	47	0b00: '16',
	48	0b01: '32',
	49	0b10: '64',
	50	0b11: '128',
	51	},
	52	# ORC[1:0]
	53	'orc': {
	54	0b00: '16',
	55	0b01: '32',
	56	0b10: '64',
	57	0b11: '128',
	58	},
d997c01a UH	59	}
	60
	61	class Decoder(srd.Decoder):
12851357	62	api_version = 2
d997c01a UH	63	id = 'mxc6225xu'
	64	name = 'MXC6225XU'
	65	longname = 'MEMSIC MXC6225XU'
a465436e	66	desc = 'Digital Thermal Orientation Sensor (DTOS) protocol.'
d997c01a UH	67	license = 'gplv2+'
	68	inputs = ['i2c']
	69	outputs = ['mxc6225xu']
da9bcbd9 BV	70	annotations = (
	71	('text', 'Human-readable text'),
	72	)
d997c01a	73
92b7b49f	74	def __init__(self):
d997c01a UH	75	self.state = 'IDLE'
d997c01a UH	76
8915b346	77	def start(self):
be465111	78	self.out_ann = self.register(srd.OUTPUT_ANN)
d997c01a	79
d997c01a UH	80	def putx(self, data):
	81	self.put(self.ss, self.es, self.out_ann, data)
	82
	83	def handle_reg_0x00(self, b):
	84	# XOUT: 8-bit x-axis acceleration output.
	85	# Data is in 2's complement, values range from -128 to 127.
17160de7	86	self.putx([0, ['XOUT: %d' % b]])
d997c01a UH	87
	88	def handle_reg_0x01(self, b):
	89	# YOUT: 8-bit y-axis acceleration output.
	90	# Data is in 2's complement, values range from -128 to 127.
17160de7	91	self.putx([0, ['YOUT: %d' % b]])
d997c01a UH	92
	93	def handle_reg_0x02(self, b):
	94	# STATUS: Orientation and shake status.
	95
17160de7	96	# Bits[7:7]: INT
d997c01a UH	97	int_val = (b >> 7) & 1
d997c01a UH	98	s = 'unchanged and no' if (int_val == 0) else 'changed or'
868fd207	99	ann = 'INT = %d: Orientation %s shake event occurred\n' % (int_val, s)
d997c01a UH	100
	101	# Bits[6:5]: SH[1:0]
	102	sh = (((b >> 6) & 1) << 1) \| ((b >> 5) & 1)
17160de7 UH	103	ann += 'SH[1:0] = %s: Shake event: %s\n' % \
17160de7 UH	104	(bin(sh)[2:], status['sh'][sh])
d997c01a UH	105
d997c01a UH	106	# Bits[4:4]: TILT
17160de7	107	tilt = (b >> 4) & 1
d997c01a UH	108	s = '' if (tilt == 0) else 'not '
	109	ann += 'TILT = %d: Orientation measurement is %svalid\n' % (tilt, s)
	110
	111	# Bits[3:2]: ORI[1:0]
	112	ori = (((b >> 3) & 1) << 1) \| ((b >> 2) & 1)
	113	ann += 'ORI[1:0] = %s: %s\n' % (bin(ori)[2:], status['ori'][ori])
	114
	115	# Bits[1:0]: OR[1:0]
	116	or_val = (((b >> 1) & 1) << 1) \| ((b >> 0) & 1)
	117	ann += 'OR[1:0] = %s: %s\n' % (bin(or_val)[2:], status['ori'][or_val])
	118
	119	# ann += 'b = %s\n' % (bin(b))
	120
	121	self.putx([0, [ann]])
	122
	123	def handle_reg_0x03(self, b):
	124	# DETECTION: Powerdown, orientation and shake detection parameters.
	125	# Note: This is a write-only register.
	126
17160de7	127	# Bits[7:7]: PD
d997c01a	128	pd = (b >> 7) & 1
17160de7 UH	129	s = 'Do not power down' if (pd == 0) else 'Power down'
17160de7 UH	130	ann = 'PD = %d: %s the device (into a low-power state)\n' % (pd, s)
d997c01a	131
17160de7	132	# Bits[6:6]: SHM
d997c01a	133	shm = (b >> 6) & 1
17160de7 UH	134	ann = 'SHM = %d: Set shake mode to %d\n' % (shm, shm)
	135
	136	# Bits[5:4]: SHTH[1:0]
	137	shth = (((b >> 5) & 1) << 1) \| ((b >> 4) & 1)
	138	ann += 'SHTH[1:0] = %s: Set shake threshold to %s\n' \
	139	% (bin(shth)[2:], status['shth'][shth])
d997c01a	140
17160de7 UH	141	# Bits[3:2]: SHC[1:0]
	142	shc = (((b >> 3) & 1) << 1) \| ((b >> 2) & 1)
	143	ann += 'SHC[1:0] = %s: Set shake count to %s readings\n' \
	144	% (bin(shc)[2:], status['shc'][shc])
	145
	146	# Bits[1:0]: ORC[1:0]
	147	orc = (((b >> 1) & 1) << 1) \| ((b >> 0) & 1)
	148	ann += 'ORC[1:0] = %s: Set orientation count to %s readings\n' \
	149	% (bin(orc)[2:], status['orc'][orc])
	150
	151	self.putx([0, [ann]])
d997c01a UH	152
d997c01a UH	153	# TODO: Fixup, this is copy-pasted from another PD.
17160de7	154	# TODO: Handle/check the ACKs/NACKs.
d997c01a UH	155	def decode(self, ss, es, data):
	156	cmd, databyte = data
	157
31f1a296	158	# Store the start/end samples of this I²C packet.
d997c01a UH	159	self.ss, self.es = ss, es
	160
	161	# State machine.
	162	if self.state == 'IDLE':
31f1a296	163	# Wait for an I²C START condition.
d997c01a UH	164	if cmd != 'START':
	165	return
	166	self.state = 'GET SLAVE ADDR'
d997c01a UH	167	elif self.state == 'GET SLAVE ADDR':
d997c01a UH	168	# Wait for an address write operation.
17160de7	169	# TODO: We should only handle packets to the slave(?)
d997c01a UH	170	if cmd != 'ADDRESS WRITE':
	171	return
	172	self.state = 'GET REG ADDR'
	173	elif self.state == 'GET REG ADDR':
	174	# Wait for a data write (master selects the slave register).
	175	if cmd != 'DATA WRITE':
	176	return
	177	self.reg = databyte
17160de7 UH	178	self.state = 'WRITE REGS'
	179	elif self.state == 'WRITE REGS':
	180	# If we see a Repeated Start here, it's a multi-byte read.
d997c01a	181	if cmd == 'START REPEAT':
17160de7	182	self.state = 'READ REGS'
d997c01a UH	183	return
	184	# Otherwise: Get data bytes until a STOP condition occurs.
	185	if cmd == 'DATA WRITE':
	186	handle_reg = getattr(self, 'handle_reg_0x%02x' % self.reg)
	187	handle_reg(databyte)
	188	self.reg += 1
	189	# TODO: Check for NACK!
	190	elif cmd == 'STOP':
	191	# TODO
	192	self.state = 'IDLE'
	193	else:
	194	pass # TODO
17160de7	195	elif self.state == 'READ REGS':
d997c01a	196	# Wait for an address read operation.
17160de7	197	# TODO: We should only handle packets to the slave(?)
d997c01a	198	if cmd == 'ADDRESS READ':
17160de7	199	self.state = 'READ REGS2'
d997c01a UH	200	return
	201	else:
	202	pass # TODO
17160de7	203	elif self.state == 'READ REGS2':
d997c01a UH	204	if cmd == 'DATA READ':
	205	handle_reg = getattr(self, 'handle_reg_0x%02x' % self.reg)
	206	handle_reg(databyte)
	207	self.reg += 1
	208	# TODO: Check for NACK!
	209	elif cmd == 'STOP':
	210	# TODO
	211	self.state = 'IDLE'
	212	else:
	213	pass # TODO?