CHIRP

# Copyright 2017 Dan Smith <dsmith@danplanet.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 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.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""Rugged RH5R V2 radio management module"""

import struct

import logging

from chirp import chirp_common, bitwise, errors, directory, memmap, util

from chirp.settings import RadioSetting, RadioSettingGroup, \

    RadioSettingValueInteger, RadioSettingValueList, \

    RadioSettingValueBoolean, RadioSettingValueString, \

    RadioSettings

LOG = logging.getLogger(__name__)

def _identify(radio):

    try:

        radio.pipe.write("PGM2015")

        ack = radio.pipe.read(2)

        if ack != "\x06\x30":

            raise errors.RadioError("Radio did not ACK first command: %r" %

                                    ack)

    except:

        LOG.exception('')

        raise errors.RadioError("Unable to communicate with the radio")

def _download(radio):

    _identify(radio)

    data = []

    for i in range(0, 0x2000, 0x40):

        msg = struct.pack('>cHb', 'R', i, 0x40)

        radio.pipe.write(msg)

        block = radio.pipe.read(0x40 + 4)

        if len(block) != (0x40 + 4):

            raise errors.RadioError("Radio sent a short block (%02x/%02x)" % (

                len(block), 0x44))

        data += block[4:]

        if radio.status_fn:

            status = chirp_common.Status()

            status.cur = i

            status.max = 0x2000

            status.msg = "Cloning from radio"

            radio.status_fn(status)

    radio.pipe.write("E")

    data += 'PGM2015'

    return memmap.MemoryMap(data)

def _upload(radio):

    _identify(radio)

    for i in range(0, 0x2000, 0x40):

        msg = struct.pack('>cHb', 'W', i, 0x40)

        msg += radio._mmap[i:(i + 0x40)]

        radio.pipe.write(msg)

        ack = radio.pipe.read(1)

        if ack != '\x06':

            raise errors.RadioError('Radio did not ACK block %i (0x%04x)' % (

                i, i))

        if radio.status_fn:

            status = chirp_common.Status()

            status.cur = i

            status.max = 0x2000

            status.msg = "Cloning from radio"

            radio.status_fn(status)

    radio.pipe.write("E")

MEM_FORMAT = """

struct memory {

  bbcd rx_freq[4];

  bbcd tx_freq[4];

  lbcd rx_tone[2];

  lbcd tx_tone[2];

  u8 unknown10:5,

     highpower:1,

     unknown11:2;

  u8 unknown20:4,

     narrow:1,

     unknown21:3;

  u8 unknown31:1,

     scanadd:1,

     unknown32:6;

  u8 unknown4;

};

struct name {

  char name[7];

};

#seekto 0x0010;

struct memory channels[128];

#seekto 0x08C0;

struct name names[128];

#seekto 0x2020;

struct memory vfo1;

struct memory vfo2;

"""

POWER_LEVELS = [chirp_common.PowerLevel('Low', watts=1),

                chirp_common.PowerLevel('High', watts=5)]

class TYTTHUVF8_V2(chirp_common.CloneModeRadio):

    VENDOR = "TYT"

    MODEL = "TH-UVF8F"

    BAUD_RATE = 9600

    _FILEID = 'OEMOEM \XFF'

    def get_features(self):

        rf = chirp_common.RadioFeatures()

        rf.memory_bounds = (1, 128)

        rf.has_bank = False

        rf.has_ctone = True

        rf.valid_tuning_steps = [5, 6.25, 10, 12.5]

        rf.has_tuning_step = False

        rf.has_cross = True

        rf.has_rx_dtcs = True

        rf.has_settings = False

        rf.can_odd_split = False

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

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

        rf.valid_characters = chirp_common.CHARSET_UPPER_NUMERIC + "-"

        rf.valid_bands = [(136000000, 174000000),

                          (400000000, 480000000)]

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

        rf.valid_power_levels = POWER_LEVELS

        rf.valid_modes = ["FM", "NFM"]

        rf.valid_name_length = 7

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

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

        return rf

    def sync_in(self):

        self._mmap = _download(self)

        self.process_mmap()

    def sync_out(self):

        _upload(self)

    @classmethod

    def match_model(cls, filedata, filename):

        return (filedata.endswith("PGM2015") and

                filedata[0x840:0x848] == cls._FILEID)

    def process_mmap(self):

        print MEM_FORMAT

        self._memobj = bitwise.parse(MEM_FORMAT, self._mmap)

    def _is_txinh(self, _mem):

        raw_tx = ""

        for i in range(0, 4):

            raw_tx += _mem.tx_freq[i].get_raw()

        return raw_tx == "\x00\x00\x00\x00"

    def get_raw_memory(self, number):

        return (repr(self._memobj.channels[number - 1]) +

                repr(self._memobj.names[number - 1]))

    def _get_memobjs(self, number):

        return (self._memobj.channels[number - 1],

                self._memobj.names[number - 1])

    def _decode_tone(self, toneval):

        pol = "N"

        rawval = (toneval[1].get_bits(0xFF) << 8) | toneval[0].get_bits(0xFF)

        if toneval[0].get_bits(0xFF) == 0xFF:

            mode = ""

            val = 0

        elif toneval[1].get_bits(0xC0) == 0xC0:

            mode = "DTCS"

            val = int("%x" % (rawval & 0x3FFF))

            pol = "R"

        elif toneval[1].get_bits(0x80):

            mode = "DTCS"

            val = int("%x" % (rawval & 0x3FFF))

        else:

            mode = "Tone"

            val = int(toneval) / 10.0

        return mode, val, pol

    def _encode_tone(self, _toneval, mode, val, pol):

        toneval = 0

        if mode == "Tone":

            toneval = int("%i" % (val * 10), 16)

        elif mode == "DTCS":

            toneval = int("%i" % val, 16)

            toneval |= 0x8000

            if pol == "R":

                toneval |= 0x4000

        else:

            toneval = 0xFFFF

        _toneval[0].set_raw(toneval & 0xFF)

        _toneval[1].set_raw((toneval >> 8) & 0xFF)

    def get_memory(self, number):

        _mem, _name = self._get_memobjs(number)

        mem = chirp_common.Memory()

        if isinstance(number, str):

            mem.number = SPECIALS[number]

            mem.extd_number = number

        else:

            mem.number = number

        if _mem.get_raw().startswith("\xFF\xFF\xFF\xFF"):

            mem.empty = True

            return mem

        mem.freq = int(_mem.rx_freq) * 10

        offset = (int(_mem.tx_freq) - int(_mem.rx_freq)) * 10

        if self._is_txinh(_mem):

            mem.duplex = "off"

            mem.offset = 0

        elif not offset:

            mem.offset = 0

            mem.duplex = ''

        elif offset < 0:

            mem.offset = abs(offset)

            mem.duplex = '-'

        else:

            mem.offset = offset

            mem.duplex = '+'

        txmode, txval, txpol = self._decode_tone(_mem.tx_tone)

        rxmode, rxval, rxpol = self._decode_tone(_mem.rx_tone)

        chirp_common.split_tone_decode(mem,

                                       (txmode, txval, txpol),

                                       (rxmode, rxval, rxpol))

        mem.mode = 'NFM' if _mem.narrow else 'FM'

        mem.skip = '' if _mem.scanadd else 'S'

        mem.power = POWER_LEVELS[int(_mem.highpower)]

        mem.name = str(_name.name).rstrip('\xFF ')

        return mem

    def set_memory(self, mem):

        _mem, _name = self._get_memobjs(mem.number)

        if mem.empty:

            _mem.set_raw('\xFF' * 16)

            _name.set_raw('\xFF' * 7)

            return

        _mem.set_raw('\x00' * 16)

        _mem.rx_freq = mem.freq / 10

        if mem.duplex == "off":

            for i in range(0, 4):

                _mem.tx_freq[i].set_raw("\xFF")

        elif mem.duplex == '-':

            mult = -1

        elif not mem.duplex:

            mult = 0

        else:

            mult = 1

        _mem.tx_freq = (mem.freq + (mem.offset * mult)) / 10

        (txmode, txval, txpol), (rxmode, rxval, rxpol) = \

                chirp_common.split_tone_encode(mem)

        self._encode_tone(_mem.tx_tone, txmode, txval, txpol)

        self._encode_tone(_mem.rx_tone, rxmode, rxval, rxpol)

        _mem.narrow = mem.mode == 'NFM'

        _mem.scanadd = mem.skip != 'S'

        _mem.highpower = POWER_LEVELS.index(mem.power) if mem.power else 1

        _name.name = mem.name.rstrip(' ').ljust(7, '\xFF')

@directory.register

class RH5RV2(TYTTHUVF8_V2):

    VENDOR = "Rugged"

    MODEL = "RH5R-V2"

    _FILEID = 'RUGGED \xFF'