#include "spkstretchlayout.h"

SpkStretchLayout::SpkStretchLayout(QWidget *parent) : QLayout(parent)
{
}

SpkStretchLayout::~SpkStretchLayout()
{
  QLayoutItem *item;
  while((item = takeAt(0)))
    delete item;
}

void SpkStretchLayout::addItem(QLayoutItem *item)
{
  mItems.append(item);
}

QSize SpkStretchLayout::sizeHint() const
{
  if(mItems.isEmpty())
    return { 300, 300 };
  auto w = geometry().width() - spacing();
  auto it = mItems.first();
  int countPerLine = w / (it->minimumSize().width() + spacing());
  int lines = ceil((double)mItems.size() / countPerLine);
  auto h = static_cast<int>(it->minimumSize().height() * lines + spacing() * lines);
  return { w, h };
}

QSize SpkStretchLayout::minimumSize() const
{
  // It works this way, but I honestly have no idea WHY IT WORKS
  auto r = sizeHint();
  r.setWidth(300);
  return r;
}

int SpkStretchLayout::count() const
{
  return mItems.size();
}

QLayoutItem *SpkStretchLayout::itemAt(int i) const
{
  return mItems.value(i);
}

QLayoutItem *SpkStretchLayout::takeAt(int i)
{
  return i >=0 && i < mItems.size() ? mItems.takeAt(i) : nullptr;
}

void SpkStretchLayout::setGeometry(const QRect &rect)
{
  QLayout::setGeometry(rect);
  if(mItems.isEmpty())
    return;

  int spc = spacing(), w = rect.width() - spc;
  QSize size;
  auto itm = mItems.first();

  // All items are considered the same, so we only calculate with the first item.
  // Figure out how many at most can we squeeze into one line
  int countPerLine = w / (itm->minimumSize().width() + spacing());

  if(countPerLine < 1)
    countPerLine = 1;

  if(countPerLine >= mItems.size()) // All items fit in one line
    size = itm->minimumSize();
  // Won't fit in one line.
  else // Stretch items.
    size = QSize {(w / countPerLine - spc), itm->maximumSize().height() };

  auto origin = geometry().topLeft();

  QLayoutItem *o;
  for(int i = 0; i < mItems.size(); i++)
  {
    o = mItems.at(i);
    QRect geo;
    geo.setSize(size);
    geo.moveTo((i % countPerLine) * (size.width() + spc) + spc + origin.x(),
               (i / countPerLine) * (size.height() + spacing()) + spc + origin.y());
    o->setGeometry(geo);
  }
}