tahoma2d/toonz/sources/include/toonzqt/spectrumfield.h

#pragma once

#ifndef SPECTRUMFIELD_H
#define SPECTRUMFIELD_H

#include <QWidget>
#include "toonzqt/colorfield.h"
#include "tpixel.h"
#include "tspectrum.h"

#undef DVAPI
#undef DVVAR
#ifdef TOONZQT_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

// forward declaration

//=============================================================================

namespace DVGui {

//=============================================================================
// SpectrumBar
//-----------------------------------------------------------------------------

class DVAPI SpectrumBar : public QWidget {
  Q_OBJECT

  int m_x0;
  int m_currentKeyIndex;

  QPixmap m_chessBg;

  TSpectrum m_spectrum;

public:
  SpectrumBar(QWidget *parent = 0, TPixel32 color = TPixel32(0, 0, 0, 255));

  ~SpectrumBar();

  int getCurrentKeyIndex() { return m_currentKeyIndex; }
  void setCurrentKeyIndex(int index);

  int getCurrentPos();
  TPixel32 getCurrentColor();

  TSpectrum &getSpectrum() { return m_spectrum; }
  void setSpectrum(TSpectrum &spectrum) {
    m_spectrum = spectrum;
    /*-- Undoの場合、Spectrumの差し替えによってIndexがあふれてしまうことがある
     * --*/
    if (m_currentKeyIndex >= m_spectrum.getKeyCount())
      setCurrentKeyIndex(getMaxPosKeyIndex());
    update();
  }

public slots:
  void setCurrentPos(int pos, bool isDragging);
  void setCurrentColor(const TPixel32 &color);
  void addKeyAt(int pos);

signals:
  void currentPosChanged(bool isDragging);
  void currentKeyChanged();
  void currentKeyAdded(int);
  void currentKeyRemoved(int);

protected:
  double posToSpectrumValue(int pos);
  int spectrumValueToPos(double spectrumValue);

  void paintEvent(QPaintEvent *e) override;
  void mousePressEvent(QMouseEvent *e) override;
  void mouseMoveEvent(QMouseEvent *e) override;

  void mouseReleaseEvent(QMouseEvent *e) override;

  int getMaxPosKeyIndex();
  int getMinPosKeyIndex();
  int getNearPosKeyIndex(int pos);
};

//=============================================================================
// SpectrumField
//-----------------------------------------------------------------------------

class DVAPI SpectrumField : public QWidget {
  Q_OBJECT

  int m_margin;
  int m_spacing;

  ColorField *m_colorField;
  SpectrumBar *m_spectrumbar;

public:
  SpectrumField(QWidget *parent = 0, TPixel32 color = TPixel32(0, 0, 0, 255));

  ~SpectrumField();

  TSpectrum &getSpectrum() { return m_spectrumbar->getSpectrum(); }
  void setSpectrum(TSpectrum &spectrum) {
    m_spectrumbar->setSpectrum(spectrum);
    m_colorField->setColor(m_spectrumbar->getCurrentColor());
  }

  int getCurrentKeyIndex() { return m_spectrumbar->getCurrentKeyIndex(); }
  void setCurrentKeyIndex(int index) {
    m_spectrumbar->setCurrentKeyIndex(index);
    m_colorField->setColor(m_spectrumbar->getCurrentColor());
  }

protected slots:
  void onCurrentPosChanged(bool isDragging);
  void onCurrentKeyChanged();
  void onColorChanged(const TPixel32 &color, bool isDragging);

protected:
  void paintEvent(QPaintEvent *e) override;

signals:
  void keyColorChanged(bool);
  void keyPositionChanged(bool);
  void keyAdded(int);
  void keyRemoved(int);
};

//-----------------------------------------------------------------------------
}  // namespace DVGui
//-----------------------------------------------------------------------------

#endif  // SPECTRUMFIELD_H