CHIRP

# Copyright 2015 Eric Dropps <kc1ckh@kc1ckh.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/>.

from builtins import bytes

import logging

from chirp.drivers import id31

from chirp import directory, bitwise

LOG = logging.getLogger(__name__)

MEM_FORMAT = """

struct {

  u24 freq;

  u16 offset;

  u16 rtone:6,

      ctone:6,

      unknown2:1,

      mode:3;

  u8 dtcs;

  u8 tune_step:4,

     dsql_mode:2,

     unknown5:2;

  u8 unknown4;

  u8 tmode:4,

     duplex:2,

     dtcs_polarity:2;

  char name[16];

  u8 digcode;

  u8 urcall[7];

  u8 rpt1call[7];

  u8 rpt2call[7];

} memory[500];

#seekto 0x6A40;

u8 used_flags[70];

#seekto 0x6A86;

u8 skip_flags[69];

#seekto 0x6ACB;

u8 pskp_flags[69];

#seekto 0x6B40;

struct {

  u8 unknown:3,

     bank:5;

  u8 index;

} banks[500];

#seekto 0x6FD0;

struct {

  char name[16];

} bank_names[26];

#seekto 0xA8C0;

struct {

  u24 freq;

  u16 offset;

  u8 unknown1[4];

  u8 call[7];

  char name[16];

  char subname[8];

  u8 unknown3[10];

} repeaters[750];

#seekto 0x1384E;

struct {

  u8 call[7];

} rptcall[750];

#seekto 0x14FBE;

struct {

 char name[16];

} rptgroup_names[30];

#seekto 0x1519E;

struct {

  char call[8];

  char tag[4];

} mycall[6];

#seekto 0x151E6;

struct {

  char call[8];

} urcall[200];

#seekto 0x15826;

struct {

  char name[16];

} urcallname[200];

"""

@directory.register

class ID51PLUSRadio(id31.ID31Radio):

    """Icom ID-51 Plus/50th Anniversary"""

    MODEL = "ID-51 Plus"

    _memsize = 0x1FB40

    _model = "\x33\x90\x00\x02"

    _endframe = "Icom Inc\x2E\x44\x41"

    _bank_class = id31.ID31Bank

    _ranges = [(0x00000, 0x1FB40, 32)]

    MODES = {0: "FM", 1: "NFM", 3: "AM", 5: "DV"}

    @classmethod

    def match_model(cls, filedata, filename):

        """Given contents of a stored file (@filedata), return True if

        this radio driver handles the represented model"""

        # The default check for ICOM is just to check memory size

        # Since the ID-51 and ID-51 Plus/Anniversary have exactly

        # the same memory size, we need to do a more detailed check.

        if len(filedata) == cls._memsize:

            snip = bytes(filedata[0x1AF40:0x1AF60])

            if snip != bytes(b'\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF'

                             b'\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF'

                             b'\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF'

                             b'\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF'):

                return True

        return False

    def _get_bank(self, loc):

        _bank = self._memobj.banks[loc]

        LOG.debug("Bank Value for location %s is %s" % (loc, _bank.bank))

        if _bank.bank == 0x1F:

            return None

        else:

            return _bank.bank

    def _set_bank(self, loc, bank):

        _bank = self._memobj.banks[loc]

        if bank is None:

            _bank.bank = 0x1F

        else:

            _bank.bank = bank

    def get_features(self):

        rf = super(ID51PLUSRadio, self).get_features()

        rf.valid_bands = [(108000000, 174000000), (380000000, 479000000)]

        return rf

    def get_repeater_call_list(self):

        calls = []

        # Unlike previous DStar radios, there is not a separate repeater

        # callsign list. It's only the DV Memory banks.

        for repeater in self._memobj.repeaters:

            call = id31._decode_call(repeater.call)

            if call == "CALLSIGN":

                call = ""

            calls.append(call.rstrip())

        return calls

    def process_mmap(self):

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

@directory.register

class ID51PLUS2Radio(ID51PLUSRadio):

    MODEL = 'ID-51 Plus2'

    _model = b'\x33\x90\x00\x03'

    _endframe = 'Icom Inc.DA'

    _raw_frames = True

    _highbit_flip = True

    _icf_data = {

        'MapRev': 1,

        'EtcData': 0x400001,

    }

    @classmethod

    def match_model(cls, filedata, filename):

        # This model is always matched with metadata

        return False