# 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/>.

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 = b'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
