#!/usr/bin/python3
# -*- coding: utf-8 -*-
#******************************************************************************
# ZYNTHIAN PROJECT: Zynthian Control Device Driver
#
# Zynthian Control Device Driver for "Novation Launchkey MK4"
#
# Copyright (C) 2015-2023 Fernando Moyano <jofemodo@zynthian.org>
#                         Brian Walton <brian@riban.co.uk>
#
#******************************************************************************
#
# 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 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.
#
# For a full copy of the GNU General Public License see the LICENSE.txt file.
#
#******************************************************************************

import logging
from time import sleep

# Zynthian specific modules
#from zyngui.zynthian_ctrldev_manager import zynthian_ctrldev_zynpad
#from zyngine.ctrldev.zynthian_ctrldev_base import zynthian_ctrldev_zynpad
#from zyngine.zynthian_ctrldev_manager import zynthian_ctrldev_zynpad
from zyngine.ctrldev.zynthian_ctrldev_base import zynthian_ctrldev_zynpad
from zyncoder.zyncore import lib_zyncore
from zynlibs.zynseq import zynseq

# ------------------------------------------------------------------------------------------------------------------
# Novation Launchkey MK4
# ------------------------------------------------------------------------------------------------------------------

class zynthian_ctrldev_launchkey_mk4(zynthian_ctrldev_zynpad):

    dev_ids = ["Launchkey MK4 37 DAW In", "Launchkey MK4 37 2"]
	driver_name = "Launchkey Mk4"
	driver_description = "Interface Novation Launchkey Mk4 with zynpad"

	PAD_COLOURS = [71, 104, 76, 51, 104, 41, 64, 12, 11, 71, 4, 67, 42, 9, 105, 15]
	STARTING_COLOUR = 123
	STOPPING_COLOUR = 120
	
	def __init__(self, state_manager, idev_in, idev_out=None):
		self.shift = False
		super().__init__(state_manager, idev_in, idev_out)
		self.sys_ex_header = (0xF0, 0x00, 0x20, 0x29, 0x02, 0x14)

	def send_sysex(self, data):
		if self.idev_out is not None:
			msg = self.sys_ex_header + bytes.fromhex(data) + (0xF7,)
			lib_zyncore.dev_send_midi_event(self.idev_out, msg, len(msg))
			sleep(0.05)

	def init(self):
		# Enable DAW mode on launchkey
		lib_zyncore.dev_send_note_on(self.idev_out, 15, 12, 127)
		self.cols = 8
		self.rows = 2
		super().init()

	def end(self):
		super().end()
		# Disable DAW mode on launchkey
		lib_zyncore.dev_send_note_on(self.idev_out, 15, 12, 0)

	# Zynpad Scene LED feedback
	def refresh_zynpad_bank(self):
		# This function is not implemented in the Launchkey MK4 driver, so we'll leave it as a pass.
		pass


	# Zynpad Pad LED feedback
	def update_pad(self, pad, state, mode):
		if self.idev_out is None or self.zynseq.bank is None:
			return

		col, row = self.zynseq.get_xy_from_pad(pad)
		if row > 1:
			return
		note = 96 + row * 16 + col
		
		try:
			group = self.zynseq.libseq.getGroup(self.zynseq.bank, pad)
			if mode == 0 or group > 16:
				chan = 0
				vel = 0
			elif state == zynseq.SEQ_STOPPED:
				chan = 0
				vel = self.PAD_COLOURS[group]
			elif state == zynseq.SEQ_PLAYING:
				chan = 2
				vel = self.PAD_COLOURS[group]
			elif state in [zynseq.SEQ_STOPPING, zynseq.SEQ_STOPPINGSYNC]:
				chan = 1
				vel = self.STOPPING_COLOUR
			elif state == zynseq.SEQ_STARTING:
				chan = 1
				vel = self.STARTING_COLOUR
			else:
				chan = 0
				vel = 0
		except Exception as e:
			chan = 0
			vel = 0
			# logging.warning(e)

		lib_zyncore.dev_send_note_on(self.idev_out, chan, note, vel)

	def midi_event(self, ev):
		# logging.debug("Launchkey MK4 MIDI handler => {}".format(ev))
		evtype = (ev[0] >> 4) & 0x0F
		# Note ON => launch/stop sequence
		if evtype == 0x9:
			note = ev[1] & 0x7F
			vel = ev[2] & 0x7F
			if vel > 0:
				try:
					col = (note - 96) % 16
					row = (note - 96) // 16
					pad = row * self.zynseq.col_in_bank + col
					if pad < self.zynseq.seq_in_bank:
						self.zynseq.libseq.togglePlayState(self.zynseq.bank, pad)
				except:
					pass
			return True
		# CC => scene change
		elif evtype == 0xB:
			ccnum = ev[1] & 0x7F
			ccval = ev[2] & 0x7F
			if ccnum == 0x6C:
				# SHIFT
				self.shift = ccval != 0
			elif ccnum == 0 or ccval == 0:
				return True
			elif (self.shift and 20 < ccnum < 29) or (20 < ccnum < 25):
				chain = self.chain_manager.get_chain_by_position(
					ccnum - 21, midi=False)
				if chain and chain.mixer_chan is not None and chain.mixer_chan < 17:
					self.zynmixer.set_level(chain.mixer_chan, ccval / 127.0)
			elif 24 < ccnum < 29:
				self.state_manager.send_cuia("ZYNPOT_ABS", [ccnum - 25, ccval/127])
			elif ccnum == 0x66:
				# TRACK RIGHT
				self.state_manager.send_cuia("ARROW_RIGHT")
			elif ccnum == 0x67:
				# TRACK LEFT
				self.state_manager.send_cuia("ARROW_LEFT")
			elif ccnum == 0x68:
				# UP
				self.state_manager.send_cuia("ARROW_UP")
			elif ccnum == 0x69:
				# DOWN
				self.state_manager.send_cuia("ARROW_DOWN")
			elif ccnum == 0x73:
				# PLAY
				if self.shift:
					self.state_manager.send_cuia("TOGGLE_MIDI_PLAY")
				else:
					self.state_manager.send_cuia("TOGGLE_PLAY")
			elif ccnum == 0x75:
				# RECORD
				if self.shift:
					self.state_manager.send_cuia("TOGGLE_MIDI_RECORD")
				else:
					self.state_manager.send_cuia("TOGGLE_RECORD")
			return True
		elif evtype == 0xC:
			val1 = ev[1] & 0x7F
			self.zynseq.select_bank(val1 + 1)
		return True

# ---