CHIRP

# Copyright 2016 Jim Unroe <rock.unroe@gmail.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 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];                 // confirmed

  lbcd txfreq[4];                 // confirmed

  ul16 rx_tone;                   // confirmed

  ul16 tx_tone;                   // confirmed

  u8 unknown1;

  u8 unknown3:2,

     highpower:1, // Power Level  // confirmed

     wide:1,      // Bandwidth    // confirmed

     unknown4:4;

  u8 unknown5[2];

} memory[16];

#seekto 0x012F;

struct {

  u8 voice;       // Voice Annunciation

  u8 tot;         // Time-out Timer                   

  u8 unknown1[3];

  u8 squelch;     // Squelch Level                    

  u8 save;        // Battery Saver                    

  u8 beep;        // Beep                             

  u8 unknown2[2];

  u8 vox;         // VOX                              

  u8 voxgain;     // VOX Gain                         

  u8 voxdelay;    // VOX Delay                        

  u8 unknown3[2];

  u8 pf2key;      // PF2 Key                          

} settings;

#seekto 0x017E;

u8 skipflags[2];  // SCAN_ADD

"""

CMD_ACK = "\x06"

RECV_BLOCK_SIZE = 0x40

SEND_BLOCK_SIZE = 0x10

RT22_POWER_LEVELS = [chirp_common.PowerLevel("Low",  watts=2.00),

                     chirp_common.PowerLevel("High", watts=5.00)]

RT22_DTCS = sorted(chirp_common.DTCS_CODES + [645])

PF2KEY_LIST = ["Scan", "Local Alarm", "Remote Alarm"]

TIMEOUTTIMER_LIST = [""] + ["%s seconds" % x for x in range(15, 615, 15)]

VOICE_LIST = ["Off", "Chinese", "English"]

VOX_LIST = ["OFF"] + ["%s" % x for x in range(1, 17)]

VOXDELAY_LIST = ["0.5", "1.0", "1.5", "2.0", "2.5", "3.0"]

SETTING_LISTS = {

    "pf2key": PF2KEY_LIST,

    "tot": TIMEOUTTIMER_LIST,

    "voice": VOICE_LIST,

    "vox": VOX_LIST,

    "voxdelay": VOXDELAY_LIST,

    }

def _rt22_enter_programming_mode(radio):

    serial = radio.pipe

    magic = "PROGRGS"

    exito = False

    for i in range(0, 5):

        for j in range(0, len(magic)):

            time.sleep(0.005)

            serial.write(magic[j])

        ack = serial.read(1)

        try:

            if ack == CMD_ACK:

                exito = True

                break

        except:

            LOG.debug("Attempt #%s, failed, trying again" % i)

            pass

    # check if we had EXITO

    if exito is False:

        msg = "The radio did not accept program mode after five tries.\n"

        msg += "Check you interface cable and power cycle your radio."

        raise errors.RadioError(msg)

    try:

        serial.write("\x02")

        ident = serial.read(8)

    except:

        raise errors.RadioError("Error communicating with radio")

    if not ident.startswith("P32073"):

        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")

    try:

        serial.write("\x07")

        ack = serial.read(1)

    except:

        raise errors.RadioError("Error communicating with radio")

    if ack != "\x4E":

        raise errors.RadioError("Radio refused to enter programming mode")

def _rt22_exit_programming_mode(radio):

    serial = radio.pipe

    try:

        serial.write("E")

    except:

        raise errors.RadioError("Radio refused to exit programming mode")

def _rt22_read_block(radio, block_addr, block_size):

    serial = radio.pipe

    cmd = struct.pack(">cHb", 'R', block_addr, RECV_BLOCK_SIZE)

    expectedresponse = "W" + cmd[1:]

    LOG.debug("Reading block %04x..." % (block_addr))

    try:

        for j in range(0, len(cmd)):

            time.sleep(0.005)

            serial.write(cmd[j])

        response = serial.read(4 + RECV_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 _rt22_write_block(radio, block_addr, block_size):

    serial = radio.pipe

    cmd = struct.pack(">cHb", 'W', block_addr, SEND_BLOCK_SIZE)

    data = radio.get_mmap()[block_addr:block_addr + SEND_BLOCK_SIZE]

    LOG.debug("Writing Data:")

    LOG.debug(util.hexprint(cmd + data))

    try:

        for j in range(0, len(cmd)):

            time.sleep(0.01)

            serial.write(cmd[j])

        for j in range(0, len(data)):

            time.sleep(0.01)

            serial.write(data[j])

        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")

    _rt22_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, RECV_BLOCK_SIZE):

        status.cur = addr + RECV_BLOCK_SIZE

        radio.status_fn(status)

        block = _rt22_read_block(radio, addr, RECV_BLOCK_SIZE)

        data += block

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

        LOG.debug(util.hexprint(block))

    data += radio.MODEL.ljust(8)

    _rt22_exit_programming_mode(radio)

    return memmap.MemoryMap(data)

def do_upload(radio):

    status = chirp_common.Status()

    status.msg = "Uploading to radio"

    _rt22_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, SEND_BLOCK_SIZE):

            status.cur = addr + SEND_BLOCK_SIZE

            radio.status_fn(status)

            _rt22_write_block(radio, addr, SEND_BLOCK_SIZE)

    _rt22_exit_programming_mode(radio)

def model_match(cls, data):

    """Match the opened/downloaded image to the correct version"""

    if len(data) == 0x0348:

        rid = data[0x0340:0x0348]

        return rid.startswith(cls.MODEL)

    else:

        return False

@directory.register

class RT22Radio(chirp_common.CloneModeRadio):

    """Retevis RT22"""

    VENDOR = "Retevis"

    MODEL = "RT22"

    BAUD_RATE = 9600

    _ranges = [

               (0x0000, 0x0340),

              ]

    _memsize = 0x0340

    def get_features(self):

        rf = chirp_common.RadioFeatures()

        rf.has_settings = True

        rf.has_bank = False

        rf.has_ctone = True

        rf.has_cross = True

        rf.has_rx_dtcs = True

        rf.has_tuning_step = False

        rf.can_odd_split = True

        rf.has_name = False

        rf.valid_skips = ["", "S"]

        rf.valid_tmodes = ["", "Tone", "TSQL", "DTCS", "Cross"]

        rf.valid_cross_modes = ["Tone->Tone", "Tone->DTCS", "DTCS->Tone",

                                "->Tone", "->DTCS", "DTCS->", "DTCS->DTCS"]

        rf.valid_power_levels = RT22_POWER_LEVELS

        rf.valid_duplexes = ["", "-", "+", "split", "off"]

        rf.valid_modes = ["NFM", "FM"]  # 12.5 KHz, 25 kHz.

        rf.memory_bounds = (1, 16)

        rf.valid_bands = [(400000000, 520000000)]

        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 _get_tone(self, _mem, mem):

        def _get_dcs(val):

            code = int("%03o" % (val & 0x07FF))

            pol = (val & 0x8000) and "R" or "N"

            return code, pol

        if _mem.tx_tone != 0xFFFF and _mem.tx_tone > 0x2800:

            tcode, tpol = _get_dcs(_mem.tx_tone)

            mem.dtcs = tcode

            txmode = "DTCS"

        elif _mem.tx_tone != 0xFFFF:

            mem.rtone = _mem.tx_tone / 10.0

            txmode = "Tone"

        else:

            txmode = ""

        if _mem.rx_tone != 0xFFFF and _mem.rx_tone > 0x2800:

            rcode, rpol = _get_dcs(_mem.rx_tone)

            mem.rx_dtcs = rcode

            rxmode = "DTCS"

        elif _mem.rx_tone != 0xFFFF:

            mem.ctone = _mem.rx_tone / 10.0

            rxmode = "Tone"

        else:

            rxmode = ""

        if txmode == "Tone" and not rxmode:

            mem.tmode = "Tone"

        elif txmode == rxmode and txmode == "Tone" and mem.rtone == mem.ctone:

            mem.tmode = "TSQL"

        elif txmode == rxmode and txmode == "DTCS" and mem.dtcs == mem.rx_dtcs:

            mem.tmode = "DTCS"

        elif rxmode or txmode:

            mem.tmode = "Cross"

            mem.cross_mode = "%s->%s" % (txmode, rxmode)

        if mem.tmode == "DTCS":

            mem.dtcs_polarity = "%s%s" % (tpol, rpol)

        LOG.debug("Got TX %s (%i) RX %s (%i)" %

                  (txmode, _mem.tx_tone, rxmode, _mem.rx_tone))

    def get_memory(self, number):

        bitpos = (1 << ((number - 1) % 8))

        bytepos = ((number - 1) / 8)

        LOG.debug("bitpos %s" % bitpos)

        LOG.debug("bytepos %s" % bytepos)

        _mem = self._memobj.memory[number - 1]

        _skp = self._memobj.skipflags[bytepos]

        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 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 = _mem.wide and "FM" or "NFM"

        self._get_tone(_mem, mem)

        mem.power = RT22_POWER_LEVELS[_mem.highpower]

        mem.skip = "" if (_skp & bitpos) else "S"

        LOG.debug("mem.skip %s" % mem.skip)

        return mem

    def _set_tone(self, mem, _mem):

        def _set_dcs(code, pol):

            val = int("%i" % code, 8) + 0x2800

            if pol == "R":

                val += 0x8000

            return val

        if mem.tmode == "Cross":

            tx_mode, rx_mode = mem.cross_mode.split("->")

        elif mem.tmode == "Tone":

            tx_mode = mem.tmode

            rx_mode = None

        else:

            tx_mode = rx_mode = mem.tmode

        if tx_mode == "DTCS":

            _mem.tx_tone = mem.tmode != "DTCS" and \

                           _set_dcs(mem.dtcs, mem.dtcs_polarity[0]) or \

                           _set_dcs(mem.rx_dtcs, mem.dtcs_polarity[0])

        elif tx_mode:

            _mem.tx_tone = tx_mode == "Tone" and \

                int(mem.rtone * 10) or int(mem.ctone * 10)

        else:

            _mem.tx_tone = 0xFFFF

        if rx_mode == "DTCS":

            _mem.rx_tone = _set_dcs(mem.rx_dtcs, mem.dtcs_polarity[1])

        elif rx_mode:

            _mem.rx_tone = int(mem.ctone * 10)

        else:

            _mem.rx_tone = 0xFFFF

        LOG.debug("Set TX %s (%i) RX %s (%i)" %

                  (tx_mode, _mem.tx_tone, rx_mode, _mem.rx_tone))

    def set_memory(self, mem):

        bitpos = (1 << ((mem.number - 1) % 8))

        bytepos = ((mem.number - 1) / 8)

        LOG.debug("bitpos %s" % bitpos)

        LOG.debug("bytepos %s" % bytepos)

        _mem = self._memobj.memory[mem.number - 1]

        _skp = self._memobj.skipflags[bytepos]

        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

        _mem.wide = mem.mode == "FM"

        self._set_tone(mem, _mem)

        _mem.highpower = mem.power == RT22_POWER_LEVELS[1]

        if mem.skip != "S":

            _skp |= bitpos

        else:

            _skp &= ~bitpos

        LOG.debug("_skp %s" % _skp)

    def get_settings(self):

        _settings = self._memobj.settings

        basic = RadioSettingGroup("basic", "Basic Settings")

        top = RadioSettings(basic)

        rs = RadioSetting("squelch", "Squelch Level",

                          RadioSettingValueInteger(0, 9, _settings.squelch))

        basic.append(rs)

        rs = RadioSetting("tot", "Time-out timer",

                          RadioSettingValueList(

                              TIMEOUTTIMER_LIST,

                              TIMEOUTTIMER_LIST[_settings.tot]))

        basic.append(rs)

        rs = RadioSetting("voice", "Voice Prompts",

                          RadioSettingValueList(

                              VOICE_LIST, VOICE_LIST[_settings.voice]))

        basic.append(rs)

        rs = RadioSetting("pf2key", "PF2 Key",

                          RadioSettingValueList(

                              PF2KEY_LIST, PF2KEY_LIST[_settings.pf2key]))

        basic.append(rs)

        rs = RadioSetting("vox", "Vox",

                          RadioSettingValueBoolean(_settings.vox))

        basic.append(rs)

        rs = RadioSetting("voxgain", "VOX Level",

                          RadioSettingValueList(

                              VOX_LIST, VOX_LIST[_settings.voxgain]))

        basic.append(rs)

        rs = RadioSetting("voxdelay", "VOX Delay Time",

                          RadioSettingValueList(

                              VOXDELAY_LIST,

                              VOXDELAY_LIST[_settings.voxdelay]))

        basic.append(rs)

        rs = RadioSetting("save", "Battery Save",

                          RadioSettingValueBoolean(_settings.save))

        basic.append(rs)

        rs = RadioSetting("beep", "Beep",

                          RadioSettingValueBoolean(_settings.beep))

        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()

                    LOG.debug("Setting %s = %s" % (setting, element.value))

                    setattr(obj, setting, element.value)

                except Exception, e:

                    LOG.debug(element.get_name())

                    raise

    @classmethod

    def match_model(cls, filedata, filename):

        match_size = False

        match_model = False

        # testing the file data size

        if len(filedata) in [0x0340, 0x0348]:

            match_size = True

        print "Match Size"

        print match_size

        # testing the firmware model fingerprint

        match_model = model_match(cls, filedata)

        print "Match Model"

        print match_model

        if match_size and match_model:

            return True

        else:

            return False

@directory.register

class KDC1(RT22Radio):

    """WLN KD-C1"""

    VENDOR = "WLN"

    MODEL = "KD-C1"

@directory.register

class ZTX6(RT22Radio):

    """Zastone ZT-X6"""

    VENDOR = "Zastone"

    MODEL = "ZT-X6"