Project

General

Profile

Bug #7067 » h777_test_new_model.py

h777 with additional 'Y' command - Tony Fuller, 10/22/2019 06:34 PM

 
# -*- coding: utf-8 -*-
# Copyright 2013 Andrew Morgan <ziltro@ziltro.com>
#
# 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 time
import os
import struct
import unittest
import logging

from chirp import chirp_common, directory, memmap
from chirp import bitwise, errors, util
from chirp.settings import RadioSetting, RadioSettingGroup, \
RadioSettingValueInteger, RadioSettingValueList, \
RadioSettingValueBoolean, RadioSettings

LOG = logging.getLogger(__name__)

MEM_FORMAT = """
#seekto 0x0010;
struct {
lbcd rxfreq[4];
lbcd txfreq[4];
lbcd rxtone[2];
lbcd txtone[2];
u8 unknown3:1,
unknown2:1,
unknown1:1,
skip:1,
highpower:1,
narrow:1,
beatshift:1,
bcl:1;
u8 unknown4[3];
} memory[16];
#seekto 0x02B0;
struct {
u8 voiceprompt;
u8 voicelanguage;
u8 scan;
u8 vox;
u8 voxlevel;
u8 voxinhibitonrx;
u8 lowvolinhibittx;
u8 highvolinhibittx;
u8 alarm;
u8 fmradio;
} settings;
#seekto 0x03C0;
struct {
u8 unused:6,
batterysaver:1,
beep:1;
u8 squelchlevel;
u8 sidekeyfunction;
u8 timeouttimer;
u8 unused2[3];
u8 unused3:7,
scanmode:1;
} settings2;
"""

CMD_ACK = "\x06"
BLOCK_SIZE = 0x08
UPLOAD_BLOCKS = [range(0x0000, 0x0110, 8),
range(0x02b0, 0x02c0, 8),
range(0x0380, 0x03e0, 8)]

# TODO: Is it 1 watt?
H777_POWER_LEVELS = [chirp_common.PowerLevel("Low", watts=1.00),
chirp_common.PowerLevel("High", watts=5.00)]
VOICE_LIST = ["English", "Chinese"]
TIMEOUTTIMER_LIST = ["Off", "30 seconds", "60 seconds", "90 seconds",
"120 seconds", "150 seconds", "180 seconds",
"210 seconds", "240 seconds", "270 seconds",
"300 seconds"]
SCANMODE_LIST = ["Carrier", "Time"]

SETTING_LISTS = {
"voice": VOICE_LIST,
}


def _h777_enter_programming_mode(radio):
serial = radio.pipe

try:
serial.write("\x02")
time.sleep(0.1)
serial.write("PROGRAM")
ack = serial.read(1)
except:
raise errors.RadioError("Error communicating with radio")

if not ack:
raise errors.RadioError("No response from radio")
elif ack != CMD_ACK:
raise errors.RadioError("Radio refused to enter programming mode")

try:
serial.write("\x02")
ident = serial.read(8)
except:
raise errors.RadioError("Error communicating with radio")

if not ident.startswith("P3107"):
LOG.debug(util.hexprint(ident))
raise errors.RadioError("Radio returned unknown identification string")

try:
serial.write(CMD_ACK)
ack = serial.read(1)
except:
raise errors.RadioError("Error communicating with radio")

if ack != CMD_ACK:
raise errors.RadioError("Radio refused to enter programming mode")


def _h777_exit_programming_mode(radio):
serial = radio.pipe
try:
serial.write("E")
except:
raise errors.RadioError("Radio refused to exit programming mode")


def _h777_read_block(radio, block_addr, block_size):
serial = radio.pipe

cmd = struct.pack(">cHb", 'R', block_addr, BLOCK_SIZE)
expectedresponse = "W" + cmd[1:]
LOG.debug("Reading block %04x..." % (block_addr))

try:
serial.write(cmd)
response = serial.read(4 + BLOCK_SIZE)
if response[:4] != expectedresponse:
raise Exception("Error reading block %04x." % (block_addr))

block_data = response[4:]

serial.write(CMD_ACK)
ack = serial.read(1)
except:
raise errors.RadioError("Failed to read block at %04x" % block_addr)

if ack != CMD_ACK:
raise Exception("No ACK reading block %04x." % (block_addr))

return block_data


def _h777_write_block(radio, block_addr, block_size):
serial = radio.pipe

if 0x02B0 <= block_addr <= 0x02C0:
cmd = struct.pack(">cHb", 'Y', block_addr, BLOCK_SIZE)
else:
cmd = struct.pack(">cHb", 'W', block_addr, BLOCK_SIZE)
data = radio.get_mmap()[block_addr:block_addr + 8]

LOG.debug("Writing Data:")
LOG.debug(util.hexprint(cmd + data))

try:
serial.write(cmd + data)
if serial.read(1) != CMD_ACK:
raise Exception("No ACK")
except:
raise errors.RadioError("Failed to send block "
"to radio at %04x" % block_addr)


def do_download(radio):
LOG.debug("download")
_h777_enter_programming_mode(radio)

data = ""

status = chirp_common.Status()
status.msg = "Cloning from radio"

status.cur = 0
status.max = radio._memsize

for addr in range(0, radio._memsize, BLOCK_SIZE):
status.cur = addr + BLOCK_SIZE
radio.status_fn(status)

block = _h777_read_block(radio, addr, BLOCK_SIZE)
data += block

LOG.debug("Address: %04x" % addr)
LOG.debug(util.hexprint(block))

_h777_exit_programming_mode(radio)

return memmap.MemoryMap(data)


def do_upload(radio):
status = chirp_common.Status()
status.msg = "Uploading to radio"

_h777_enter_programming_mode(radio)

status.cur = 0
status.max = radio._memsize

for start_addr, end_addr in radio._ranges:
for addr in range(start_addr, end_addr, BLOCK_SIZE):
status.cur = addr + BLOCK_SIZE
radio.status_fn(status)
_h777_write_block(radio, addr, BLOCK_SIZE)

_h777_exit_programming_mode(radio)


class ArcshellAR5(chirp_common.Alias):
VENDOR = 'Arcshell'
MODEL = 'AR-5'


class ArcshellAR6(chirp_common.Alias):
VENDOR = 'Arcshell'
MODEL = 'AR-6'


class GV8SAlias(chirp_common.Alias):
VENDOR = 'Greaval'
MODEL = 'GV-8S'


class GV9SAlias(chirp_common.Alias):
VENDOR = 'Greaval'
MODEL = 'GV-9S'


class A8SAlias(chirp_common.Alias):
VENDOR = 'Ansoko'
MODEL = 'A-8S'


class TenwayTW325Alias(chirp_common.Alias):
VENDOR = 'Tenway'
MODEL = 'TW-325'


@directory.register
class H777Radio(chirp_common.CloneModeRadio):
"""HST H-777"""
# VENDOR = "Heng Shun Tong (恒顺通)"
# MODEL = "H-777"
VENDOR = "Baofeng"
MODEL = "BF-888"
BAUD_RATE = 9600

ALIASES = [ArcshellAR5, ArcshellAR6, GV8SAlias, GV9SAlias, A8SAlias,
TenwayTW325Alias]
SIDEKEYFUNCTION_LIST = ["Off", "Monitor", "Transmit Power", "Alarm"]

# This code currently requires that ranges start at 0x0000
# and are continious. In the original program 0x0388 and 0x03C8
# are only written (all bytes 0xFF), not read.
# _ranges = [
# (0x0000, 0x0110),
# (0x02B0, 0x02C0),
# (0x0380, 0x03E0)
# ]
# Memory starts looping at 0x1000... But not every 0x1000.

_ranges = [
(0x0000, 0x0110),
(0x0380, 0x03E0),
(0x02B0, 0x02C0),
]
_memsize = 0x03E0
_has_fm = True
_has_sidekey = True

def get_features(self):
rf = chirp_common.RadioFeatures()
rf.has_settings = True
rf.valid_modes = ["NFM", "FM"] # 12.5 KHz, 25 kHz.
rf.valid_skips = ["", "S"]
rf.valid_tmodes = ["", "Tone", "TSQL", "DTCS", "Cross"]
rf.valid_duplexes = ["", "-", "+", "split", "off"]
rf.can_odd_split = True
rf.has_rx_dtcs = True
rf.has_ctone = True
rf.has_cross = True
rf.valid_cross_modes = [
"Tone->Tone",
"DTCS->",
"->DTCS",
"Tone->DTCS",
"DTCS->Tone",
"->Tone",
"DTCS->DTCS"]
rf.has_tuning_step = False
rf.has_bank = False
rf.has_name = False
rf.memory_bounds = (1, 16)
rf.valid_bands = [(400000000, 470000000)]
rf.valid_power_levels = H777_POWER_LEVELS
rf.valid_tuning_steps = [2.5, 5.0, 6.25, 10.0, 12.5, 15.0, 20.0, 25.0,
50.0, 100.0]

return rf

def process_mmap(self):
self._memobj = bitwise.parse(MEM_FORMAT, self._mmap)

def sync_in(self):
self._mmap = do_download(self)
self.process_mmap()

def sync_out(self):
do_upload(self)

def get_raw_memory(self, number):
return repr(self._memobj.memory[number - 1])

def _decode_tone(self, val):
val = int(val)
if val == 16665:
return '', None, None
elif val >= 12000:
return 'DTCS', val - 12000, 'R'
elif val >= 8000:
return 'DTCS', val - 8000, 'N'
else:
return 'Tone', val / 10.0, None

def _encode_tone(self, memval, mode, value, pol):
if mode == '':
memval[0].set_raw(0xFF)
memval[1].set_raw(0xFF)
elif mode == 'Tone':
memval.set_value(int(value * 10))
elif mode == 'DTCS':
flag = 0x80 if pol == 'N' else 0xC0
memval.set_value(value)
memval[1].set_bits(flag)
else:
raise Exception("Internal error: invalid mode `%s'" % mode)

def get_memory(self, number):
_mem = self._memobj.memory[number - 1]

mem = chirp_common.Memory()

mem.number = number
mem.freq = int(_mem.rxfreq) * 10

# We'll consider any blank (i.e. 0MHz frequency) to be empty
if mem.freq == 0:
mem.empty = True
return mem

if _mem.rxfreq.get_raw() == "\xFF\xFF\xFF\xFF":
mem.freq = 0
mem.empty = True
return mem

if _mem.txfreq.get_raw() == "\xFF\xFF\xFF\xFF":
mem.duplex = "off"
mem.offset = 0
elif int(_mem.rxfreq) == int(_mem.txfreq):
mem.duplex = ""
mem.offset = 0
else:
mem.duplex = int(_mem.rxfreq) > int(_mem.txfreq) and "-" or "+"
mem.offset = abs(int(_mem.rxfreq) - int(_mem.txfreq)) * 10

mem.mode = not _mem.narrow and "FM" or "NFM"
mem.power = H777_POWER_LEVELS[_mem.highpower]

mem.skip = _mem.skip and "S" or ""

txtone = self._decode_tone(_mem.txtone)
rxtone = self._decode_tone(_mem.rxtone)
chirp_common.split_tone_decode(mem, txtone, rxtone)

mem.extra = RadioSettingGroup("Extra", "extra")
rs = RadioSetting("bcl", "Busy Channel Lockout",
RadioSettingValueBoolean(not _mem.bcl))
mem.extra.append(rs)
rs = RadioSetting("beatshift", "Beat Shift(scramble)",
RadioSettingValueBoolean(not _mem.beatshift))
mem.extra.append(rs)

return mem

def set_memory(self, mem):
# Get a low-level memory object mapped to the image
_mem = self._memobj.memory[mem.number - 1]

if mem.empty:
_mem.set_raw("\xFF" * (_mem.size() / 8))
return

_mem.rxfreq = mem.freq / 10

if mem.duplex == "off":
for i in range(0, 4):
_mem.txfreq[i].set_raw("\xFF")
elif mem.duplex == "split":
_mem.txfreq = mem.offset / 10
elif mem.duplex == "+":
_mem.txfreq = (mem.freq + mem.offset) / 10
elif mem.duplex == "-":
_mem.txfreq = (mem.freq - mem.offset) / 10
else:
_mem.txfreq = mem.freq / 10

txtone, rxtone = chirp_common.split_tone_encode(mem)
self._encode_tone(_mem.txtone, *txtone)
self._encode_tone(_mem.rxtone, *rxtone)

_mem.narrow = 'N' in mem.mode
_mem.highpower = mem.power == H777_POWER_LEVELS[1]
_mem.skip = mem.skip == "S"

for setting in mem.extra:
# NOTE: Only two settings right now, both are inverted
setattr(_mem, setting.get_name(), not int(setting.value))

# When set to one, official programming software (BF-480) shows always
# "WFM", even if we choose "NFM". Therefore, for compatibility
# purposes, we will set these to zero.
_mem.unknown1 = 0
_mem.unknown2 = 0
_mem.unknown3 = 0

def get_settings(self):
_settings = self._memobj.settings
basic = RadioSettingGroup("basic", "Basic Settings")
top = RadioSettings(basic)

# TODO: Check that all these settings actually do what they
# say they do.

rs = RadioSetting("voiceprompt", "Voice prompt",
RadioSettingValueBoolean(_settings.voiceprompt))
basic.append(rs)

rs = RadioSetting("voicelanguage", "Voice language",
RadioSettingValueList(
VOICE_LIST,
VOICE_LIST[_settings.voicelanguage]))
basic.append(rs)

rs = RadioSetting("scan", "Scan",
RadioSettingValueBoolean(_settings.scan))
basic.append(rs)

rs = RadioSetting("settings2.scanmode", "Scan mode",
RadioSettingValueList(
SCANMODE_LIST,
SCANMODE_LIST[self._memobj.settings2.scanmode]))
basic.append(rs)

rs = RadioSetting("vox", "VOX",
RadioSettingValueBoolean(_settings.vox))
basic.append(rs)

rs = RadioSetting("voxlevel", "VOX level",
RadioSettingValueInteger(
1, 5, _settings.voxlevel + 1))
basic.append(rs)

rs = RadioSetting("voxinhibitonrx", "Inhibit VOX on receive",
RadioSettingValueBoolean(_settings.voxinhibitonrx))
basic.append(rs)

rs = RadioSetting("lowvolinhibittx", "Low voltage inhibit transmit",
RadioSettingValueBoolean(_settings.lowvolinhibittx))
basic.append(rs)

rs = RadioSetting("highvolinhibittx", "High voltage inhibit transmit",
RadioSettingValueBoolean(_settings.highvolinhibittx))
basic.append(rs)

rs = RadioSetting("alarm", "Alarm",
RadioSettingValueBoolean(_settings.alarm))
basic.append(rs)

# TODO: This should probably be called “FM Broadcast Band Radio”
# or something. I'm not sure if the model actually has one though.
if self._has_fm:
rs = RadioSetting("fmradio", "FM function",
RadioSettingValueBoolean(_settings.fmradio))
basic.append(rs)

rs = RadioSetting("settings2.beep", "Beep",
RadioSettingValueBoolean(
self._memobj.settings2.beep))
basic.append(rs)

rs = RadioSetting("settings2.batterysaver", "Battery saver",
RadioSettingValueBoolean(
self._memobj.settings2.batterysaver))
basic.append(rs)

rs = RadioSetting("settings2.squelchlevel", "Squelch level",
RadioSettingValueInteger(
0, 9, self._memobj.settings2.squelchlevel))
basic.append(rs)

if self._has_sidekey:
rs = RadioSetting("settings2.sidekeyfunction", "Side key function",
RadioSettingValueList(
self.SIDEKEYFUNCTION_LIST,
self.SIDEKEYFUNCTION_LIST[
self._memobj.settings2.sidekeyfunction]))
basic.append(rs)

rs = RadioSetting("settings2.timeouttimer", "Timeout timer",
RadioSettingValueList(
TIMEOUTTIMER_LIST,
TIMEOUTTIMER_LIST[
self._memobj.settings2.timeouttimer]))
basic.append(rs)

return top

def set_settings(self, settings):
for element in settings:
if not isinstance(element, RadioSetting):
self.set_settings(element)
continue
else:
try:
if "." in element.get_name():
bits = element.get_name().split(".")
obj = self._memobj
for bit in bits[:-1]:
obj = getattr(obj, bit)
setting = bits[-1]
else:
obj = self._memobj.settings
setting = element.get_name()

if element.has_apply_callback():
LOG.debug("Using apply callback")
element.run_apply_callback()
elif setting == "voxlevel":
setattr(obj, setting, int(element.value) - 1)
else:
LOG.debug("Setting %s = %s" % (setting, element.value))
setattr(obj, setting, element.value)
except Exception, e:
LOG.debug(element.get_name())
raise


class H777TestCase(unittest.TestCase):

def setUp(self):
self.driver = H777Radio(None)
self.testdata = bitwise.parse("lbcd foo[2];",
memmap.MemoryMap("\x00\x00"))

def test_decode_tone_dtcs_normal(self):
mode, value, pol = self.driver._decode_tone(8023)
self.assertEqual('DTCS', mode)
self.assertEqual(23, value)
self.assertEqual('N', pol)

def test_decode_tone_dtcs_rev(self):
mode, value, pol = self.driver._decode_tone(12023)
self.assertEqual('DTCS', mode)
self.assertEqual(23, value)
self.assertEqual('R', pol)

def test_decode_tone_tone(self):
mode, value, pol = self.driver._decode_tone(885)
self.assertEqual('Tone', mode)
self.assertEqual(88.5, value)
self.assertEqual(None, pol)

def test_decode_tone_none(self):
mode, value, pol = self.driver._decode_tone(16665)
self.assertEqual('', mode)
self.assertEqual(None, value)
self.assertEqual(None, pol)

def test_encode_tone_dtcs_normal(self):
self.driver._encode_tone(self.testdata.foo, 'DTCS', 23, 'N')
self.assertEqual(8023, int(self.testdata.foo))

def test_encode_tone_dtcs_rev(self):
self.driver._encode_tone(self.testdata.foo, 'DTCS', 23, 'R')
self.assertEqual(12023, int(self.testdata.foo))

def test_encode_tone(self):
self.driver._encode_tone(self.testdata.foo, 'Tone', 88.5, 'N')
self.assertEqual(885, int(self.testdata.foo))

def test_encode_tone_none(self):
self.driver._encode_tone(self.testdata.foo, '', 67.0, 'N')
self.assertEqual(16665, int(self.testdata.foo))


@directory.register
class ROGA2SRadio(H777Radio):
VENDOR = "Radioddity"
MODEL = "GA-2S"
_has_fm = False
SIDEKEYFUNCTION_LIST = ["Off", "Monitor", "Unused", "Alarm"]

@classmethod
def match_model(cls, filedata, filename):
# This model is only ever matched via metadata
return False
    (1-1/1)