/*
 * Copyright (C) EdgeTX
 *
 * Based on code named
 *   opentx - https://github.com/opentx/opentx
 *   th9x - http://code.google.com/p/th9x
 *   er9x - http://code.google.com/p/er9x
 *   gruvin9x - http://code.google.com/p/gruvin9x
 *
 * License GPLv2: http://www.gnu.org/licenses/gpl-2.0.html
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * 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.
 */

#include "autobitmappedcombobox.h"

AutoBitMappedComboBox::AutoBitMappedComboBox(QWidget * parent):
  QComboBox(parent),
  AutoWidget(),
  m_field(nullptr),
  m_next(0),
  m_hasModel(false),
  m_rawSource(nullptr),
  m_rawSwitch(nullptr)
{
  setBits();
  connect(this, QOverload<int>::of(&QComboBox::currentIndexChanged), this, &AutoBitMappedComboBox::onCurrentIndexChanged);
}

AutoBitMappedComboBox::~AutoBitMappedComboBox()
{
}

void AutoBitMappedComboBox::clear()
{
  if (m_hasModel)
    return;

  setLock(true);
  QComboBox::clear();
  m_next = 0;
  setLock(false);
}

void AutoBitMappedComboBox::insertItems(int index, const QStringList & items)
{
  if (m_hasModel)
    return;

  foreach(QString item, items) {
    addItem(item);
  }
}

void AutoBitMappedComboBox::addItem(const QString & item)
{
  if (m_hasModel)
    return;

  addItem(item, m_next++);
}

void AutoBitMappedComboBox::addItem(const QString & item, int value)
{
  if (m_hasModel)
    return;

  setLock(true);
  QComboBox::addItem(item, value);
  setLock(false);
  updateValue();
}

void AutoBitMappedComboBox::setField(unsigned int & field, GenericPanel * panel)
{
  m_field = (int *)&field;
  m_rawSource = nullptr;
  m_rawSwitch = nullptr;
  setFieldInit(panel);
}

void AutoBitMappedComboBox::setField(int & field, GenericPanel * panel)
{
  m_field = &field;
  m_rawSource = nullptr;
  m_rawSwitch = nullptr;
  setFieldInit(panel);
}

void AutoBitMappedComboBox::setField(RawSource & field, GenericPanel * panel)
{
  m_rawSource = &field;
  m_rawSwitch = nullptr;
  m_field = nullptr;
  setFieldInit(panel);
}

void AutoBitMappedComboBox::setField(RawSwitch & field, GenericPanel * panel)
{
  m_rawSwitch = &field;
  m_rawSource = nullptr;
  m_field = nullptr;
  setFieldInit(panel);
}

void AutoBitMappedComboBox::setFieldInit(GenericPanel * panel)
{
  setPanel(panel);
  updateValue();
}

void AutoBitMappedComboBox::setBits(const unsigned int numBits, const unsigned int offsetBits,
                                    const unsigned int index, const unsigned int indexBits)
{
  m_bits = numBits;
  m_offsetBits = offsetBits;
  m_index = index;
  m_indexBits = indexBits;

  if (m_offsetBits + m_bits > m_indexBits)
    m_indexBits = m_offsetBits + m_bits;
}

void AutoBitMappedComboBox::setModel(QAbstractItemModel * model)
{
  setLock(true);
  QComboBox::setModel(model);
  setLock(false);
  m_hasModel = true;
  updateValue();
}

void AutoBitMappedComboBox::setAutoIndexes()
{
  if (m_hasModel)
    return;

  for (int i = 0; i < count(); ++i)
    setItemData(i, i);
  updateValue();
}

void AutoBitMappedComboBox::updateValue()
{
  if (!m_field && !m_rawSource && !m_rawSwitch)
    return;

  setLock(true);

  if (m_field) {
    int val = (*m_field >> shiftbits()) & bitmask();
    setCurrentIndex(findData(val));
  }
  else if (m_rawSource)
    setCurrentIndex(findData(m_rawSource->toValue()));
  else if (m_rawSwitch)
    setCurrentIndex(findData(m_rawSwitch->toValue()));

  setLock(false);
}

void AutoBitMappedComboBox::onCurrentIndexChanged(int index)
{
  if (lock() || index < 0)
    return;

  bool ok;
  const int val = itemData(index).toInt(&ok);
  if (!ok)
    return;

  if (m_field && *m_field != val) {
    unsigned int fieldmask = (bitmask() << shiftbits());
    *m_field = (*m_field & ~fieldmask) | (val << shiftbits());
  }
  else if (m_rawSource && m_rawSource->toValue() != val) {
    *m_rawSource = RawSource(val);
  }
  else if (m_rawSwitch && m_rawSwitch->toValue() != val) {
    *m_rawSwitch = RawSwitch(val);
  }
  else
    return;

  emit currentDataChanged(val);
  dataChanged();
}

unsigned int AutoBitMappedComboBox::shiftbits()
{
  return m_indexBits * m_index + m_offsetBits;
}

unsigned int AutoBitMappedComboBox::bitmask()
{
  int mask = -1;
  mask = ~(mask << m_bits);
  return (unsigned int)mask;
}