# Copyright 2012 Tom Hayward <tom@tomh.us>
#
# 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 re
from chirp import chirp_common, directory, util, errors
from chirp.drivers import kenwood_live
from chirp.drivers.kenwood_live import NOCACHE, KenwoodLiveRadio, command, iserr

TS2000_SSB_STEPS = [1.0, 2.5, 5.0, 10.0]
TS2000_FM_STEPS = [5.0, 6.25, 10.0, 12.5, 15.0, 20.0, 25.0, 30.0, 50.0, 100.0]
TS2000_DUPLEX = dict(kenwood_live.DUPLEX)
TS2000_DUPLEX[3] = "="
TS2000_DUPLEX[4] = "split"
TS2000_MODES = ["?", "LSB", "USB", "CW", "FM", "AM",
                "FSK", "CR-R", "?", "FSK-R"]
TS2000_TMODES = ["", "Tone", "TSQL", "DTCS"]
TS2000_TONES = list(chirp_common.OLD_TONES)
TS2000_TONES.remove(69.3)

@directory.register
class TS2000Radio(KenwoodLiveRadio):
    """Kenwood TS-2000"""
    MODEL = "TS-2000"

    _upper = 289
    _kenwood_split = True
    _kenwood_valid_tones = list(TS2000_TONES)

    def get_features(self):
        rf = chirp_common.RadioFeatures()
        rf.has_dtcs_polarity = False
        rf.has_bank = False
        rf.can_odd_split = True
        rf.valid_modes = ["LSB", "USB", "CW", "FM", "AM"]
        rf.valid_tmodes = list(TS2000_TMODES)
        rf.valid_tuning_steps = list(TS2000_SSB_STEPS + TS2000_FM_STEPS)
        rf.valid_bands = [(1000, 1300000000)]
        rf.valid_skips = ["", "S"]
        rf.valid_duplexes = TS2000_DUPLEX.values()

        # TS-2000 uses ";" as a message separator even though it seems to
        # allow you to to use all printable ASCII characters at the manual
        # controls.  The radio doesn't send the name after the ";" if you
        # input one from the manual controls.
        rf.valid_characters = chirp_common.CHARSET_ASCII.replace(';', '')
        rf.valid_name_length = 7    # 7 character channel names
        rf.memory_bounds = (0, self._upper)
        return rf

    def _cmd_set_memory(self, number, spec):
        return "MW0%03i%s" % (number, spec)

    def _cmd_set_split(self, number, spec):
        return "MW1%03i%s" % (number, spec)

    def _cmd_get_memory(self, number):
        return "MR0%03i" % number

    def _cmd_get_split(self, number):
        return "MR1%03i" % number

    def _cmd_recall_memory(self, number):
        return "MC%03i" % (number)

    def _cmd_cur_memory(self, number):
        return "MC"

    def _cmd_get_split(self, number):
        return "MR1%03i" % number

    def _cmd_set_split(self, number, spec):
        return "MW1%03i%s" % (number, spec)

    def _cmd_erase_memory(self, number):
        # write a memory channel that's effectively zeroed except the channel number
        return "MW%04i%035i" % (number, 0)

    def erase_memory(self, number):
        if number not in self._memcache:
            return

        resp = command(self.pipe, *self._cmd_erase_memory(number))
        if iserr(resp):
            raise errors.RadioError("Radio refused delete of %i" % number)
        del self._memcache[number]

    def get_memory(self, number):
        if number < 0 or number > self._upper:
            raise errors.InvalidMemoryLocation( \
                "Number must be between 0 and %i" % self._upper)
        if number in self._memcache and not NOCACHE:
            return self._memcache[number]

        result = command(self.pipe, *self._cmd_get_memory(number))
        if result == "N":
            mem = chirp_common.Memory()
            mem.number = number
            mem.empty = True
            self._memcache[mem.number] = mem
            return mem

        spec = result[2:]

        mem = self._parse_mem_spec(result)
        self._memcache[mem.number] = mem

        # check for split frequency operation
        if mem.duplex == "" and self._kenwood_split:
            result = command(self.pipe, *self._cmd_get_split(number))
            self._parse_split_spec(mem, result)

        return mem
    
    def _parse_mem_spec(self, spec):
        mem = chirp_common.Memory()

        # pad string so indexes match Kenwood docs
        spec = " " + spec
        
        # use the same variable names as the Kenwood docs
        _p1  = spec[3]
        _p2  = spec[4]
        _p3  = spec[5:7]
        _p4  = spec[7:18]
        _p5  = spec[18]
        _p6  = spec[19]
        _p7  = spec[20]
        _p8  = spec[21:23]
        _p9  = spec[23:25]
        _p10 = spec[25:28]
        _p11 = spec[28]
        _p12 = spec[29]
        _p13 = spec[30:39]
        _p14 = spec[39:41]
        _p15 = spec[41]
        _p16 = spec[42:49]

        mem.number = int(_p2 + _p3) # concat bank num and chan num
        mem.freq = int(_p4)
        mem.mode = TS2000_MODES[int(_p5)]
        mem.skip = ["", "S"][int(_p6)]
        mem.tmode = TS2000_TMODES[int(_p7)]
        mem.rtone = self._kenwood_valid_tones[int(_p8) - 1] # PL is 1 indexed
        mem.ctone = self._kenwood_valid_tones[int(_p9) - 1] # CTCSS is 1 indexed
        mem.dtcs = chirp_common.DTCS_CODES[int(_p10)]       # DCS is 0 indexed
        mem.duplex = TS2000_DUPLEX[int(_p12)]
        mem.offset = int(_p13) # 9-digit
        if mem.mode in ["AM", "FM"]:
            mem.tuning_step = TS2000_FM_STEPS[int(_p14)]
        else:
            mem.tuning_step = TS2000_SSB_STEPS[int(_p14)]
        mem.name = _p16

        return mem
    
    def _parse_split_spec(self, mem, spec):

        # pad string so indexes match Kenwood docs
        spec = " " + spec
        
        # use the same variable names as the Kenwood docs
        split_freq = int(spec[7:18])
        if mem.freq != split_freq:
            mem.duplex = "split"
            mem.offset = split_freq

        return mem

    def set_memory(self, memory):
        if memory.number < 0 or memory.number > self._upper:
            raise errors.InvalidMemoryLocation( \
                "Number must be between 0 and %i" % self._upper)

        spec = self._make_mem_spec(memory)
        spec = "".join(spec)
        r1 = command(self.pipe, *self._cmd_set_memory(memory.number, spec))
        if not iserr(r1):
            memory.name = memory.name.rstrip()
            self._memcache[memory.number] = memory

            # if we're tuned to the channel, reload it
            r1 = command(self.pipe, *self._cmd_cur_memory(memory.number))
            if not iserr(r1):
                pattern = re.compile("MC([0-9]{3})")
                match = pattern.search(r1)
                if match is not None:
                    cur_mem = int(match.group(1))
                    if cur_mem == memory.number:
                        cur_mem = command(self.pipe, *self._cmd_recall_memory(memory.number))
        else:
            raise errors.InvalidDataError("Radio refused %i" % memory.number)

        # FIXME
        if memory.duplex == "split" and self._kenwood_split: 
            spec = "".join(self._make_split_spec(memory))
            result = command(self.pipe, *self._cmd_set_split(memory.number,
                                                             spec))
            if iserr(result):
                raise errors.InvalidDataError("Radio refused %i" % \
                                                  memory.number)
    
    def _make_mem_spec(self, mem):
        if mem.duplex in " +-":
            duplex = util.get_dict_rev(TS2000_DUPLEX, mem.duplex)
            offset = mem.offset
        elif mem.duplex == "split":
            duplex = 0
            offset = 0
        else:
            print "Bug: unsupported duplex `%s'" % mem.duplex
        if mem.mode in ["AM", "FM"]:
            step = TS2000_FM_STEPS.index(mem.tuning_step)
        else:
            step = TS2000_SSB_STEPS.index(mem.tuning_step)

        # TS-2000 won't accept channels with tone mode off if they have
        # tone values
        if mem.tmode == "":
            rtone = 0
            ctone = 0
            dtcs = 0
        else:
            rtone = (self._kenwood_valid_tones.index(mem.rtone) + 1)    # PL is 1 indexed
            ctone = (self._kenwood_valid_tones.index(mem.ctone) + 1)    # CTCSS is 1 indexed
            dtcs = (chirp_common.DTCS_CODES.index(mem.dtcs))            # DCS is 0 indexed

        spec = ( \
            "%011i" % mem.freq,
            "%i" % (TS2000_MODES.index(mem.mode)),
            "%i" % (mem.skip == "S"),
            "%i" % TS2000_TMODES.index(mem.tmode),
            "%02i" % (rtone),
            "%02i" % (ctone),
            "%03i" % (dtcs),
            "0", # REVERSE status
            "%i" % duplex,
            "%09i" % offset,
            "%02i" % step,
            "0", # Memory Group number (0-9)
            "%s" % mem.name,
        )

        return spec
    
    def _make_split_spec(self, mem):
        if mem.duplex in " +-":
            duplex = util.get_dict_rev(TS2000_DUPLEX, mem.duplex)
            offset = mem.offset
        elif mem.duplex == "split":
            duplex = 0
            offset = 0
        else:
            print "Bug: unsupported duplex `%s'" % mem.duplex
        if mem.mode in ["AM", "FM"]:
            step = TS2000_FM_STEPS.index(mem.tuning_step)
        else:
            step = TS2000_SSB_STEPS.index(mem.tuning_step)

        # TS-2000 won't accept channels with tone mode off if they have
        # tone values
        if mem.tmode == "":
            rtone = 0
            ctone = 0
            dtcs = 0
        else:
            rtone = (self._kenwood_valid_tones.index(mem.rtone) + 1)    # PL is 1 indexed
            ctone = (self._kenwood_valid_tones.index(mem.ctone) + 1)    # CTCSS is 1 indexed
            dtcs = (chirp_common.DTCS_CODES.index(mem.dtcs))            # DCS is 0 indexed

        spec = ( \
            "%011i" % mem.offset,
            "%i" % (TS2000_MODES.index(mem.mode)),
            "%i" % (mem.skip == "S"),
            "%i" % TS2000_TMODES.index(mem.tmode),
            "%02i" % (rtone),
            "%02i" % (ctone),
            "%03i" % (dtcs),
            "0", # REVERSE status
            "%i" % duplex,
            "%09i" % 0,
            "%02i" % step,
            "0", # Memory Group number (0-9)
            "%s" % mem.name,
        )

        return spec