#include "tgl.h"

// Qt includes

#include <qpixmap></qpixmap>

#include <qimage></qimage>

#include <qopenglwidget></qopenglwidget>

#include <qdesktopwidget></qdesktopwidget>

#include <qapplication></qapplication>

#include <qscreen></qscreen>

#include "toonzqt/pickrgbutils.h"

//*************************************************************************************

//    Local namespace

//*************************************************************************************

namespace {

QRgb meanColor(const QImage &img, const QRect &rect) {

  if (rect.width() == 1 && rect.height() == 1)

    return img.pixel(rect.left(), rect.top());

  int r, g, b, m;

  int x, y, x1 = rect.right(), y1 = rect.bottom(),

            count = rect.width() * rect.height();

  r = g = b = m = 0;

  for (y = rect.top(); y <= y1; ++y)

    for (x = rect.left(); x <= x1; ++x) {

      QRgb color = img.pixel(x, y);

      r += (color & 0xff);

      g += ((color >> 8) & 0xff);

      b += ((color >> 16) & 0xff);

      m += ((color >> 24) & 0xff);

    }

  r = r / count;

  g = g / count;

  b = b / count;

  m = m / count;

  return r | (g << 8) | (b << 16) | (m << 24);

}

}  // namespace

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

QRgb pickRGB(QWidget *widget, const QRect &rect) {

  QImage img(widget->grab(rect).toImage());

  return meanColor(img, img.rect());

}

//------------------------------------------------------------------------------

QRgb pickScreenRGB(const QRect &rect) {

  QWidget *widget = QApplication::desktop();

#ifdef MACOSX

  //   #Bugzilla 6514, possibly related to #QTBUG 23516

  // Screen grabbing on a secondary screen seems to be broken on MAC

  // (see what happens using PixelTool), when no piece of the *primary*

  // screen is involved in the grab. As it seems to be very Qt-based,

  // the workaround is to trivially grab the smallest rect including the

  // requested one and a part of the primary screen.

  const QRect &screen0Geom = QApplication::desktop()->screenGeometry(0);

  int left   = std::min(rect.left(), screen0Geom.right());

  int top    = std::min(rect.top(), screen0Geom.bottom());

  int right  = std::max(rect.right(), screen0Geom.left());

  int bottom = std::max(rect.bottom(), screen0Geom.right());

  QRect theRect(QPoint(left, top), QPoint(right, bottom));

#else

  const QRect &theRect = rect;

#endif

#if (QT_VERSION >= QT_VERSION_CHECK(5, 14, 0))

  QImage img(widget->screen()

                 ->grabWindow(widget->winId(), theRect.x(), theRect.y(),

                              theRect.width(), theRect.height())

                 .toImage());

#else

  QImage img(QPixmap::grabWindow(widget->winId(), theRect.x(), theRect.y(),

                                 theRect.width(), theRect.height())

                 .toImage());

#endif

  return meanColor(

      img, QRect(rect.left() - theRect.left(), rect.top() - theRect.top(),

                 rect.width(), rect.height()));

}

//------------------------------------------------------------------------------

QRgb pickRGB(QOpenGLWidget *widget, const QRect &rect) {

  widget->makeCurrent();

  glReadBuffer(GL_FRONT);

  QImage img(rect.size(), QImage::Format_ARGB32);

  glReadPixels(rect.x(), widget->height() - rect.y(), rect.width(),

               rect.height(), GL_BGRA_EXT, GL_UNSIGNED_BYTE, img.bits());

  widget->doneCurrent();

  return meanColor(img, img.rect());

}