#pragma once

#include <QObject>

#include <QPainter>

#include <QQmlEngine>

#include <QTemporaryDir>

enum PrimitiveType { Polygon, Circle };

struct Primitive {

  PrimitiveType type;

  QPen pen;

  QBrush brush;

  QPointF origin;

  QList<QPointF> points;                    // polygon

  double radius1{}, radius2{}, rotation{};  // ellipse

  explicit Primitive(QSizeF size, const QList<QColor>& palette);

  double cost() const;

};

struct Solution {

  QList<Primitive> primitives;

  double fitness;

  QSizeF size;

  explicit Solution(QSizeF size, const QList<QColor>& palette);

  double cost() const;

};

class Tracer : public QObject {

  Q_OBJECT

  QML_ELEMENT

  Q_PROPERTY(QList<QColor> palette READ palette WRITE setPalette NOTIFY

                 paletteChanged FINAL)

  Q_PROPERTY(

      double bestSolution READ bestSolution NOTIFY bestSolutionChanged FINAL)

  Q_PROPERTY(QStringList solutions READ solutions NOTIFY solutionsChanged FINAL)

 public:

  explicit Tracer(QObject *parent = nullptr);

  QList<QColor> palette() const;

  void setPalette(const QList<QColor>& newPalette);

  double bestSolution() const;

  Q_INVOKABLE void start(const QString& fileName);

  Q_INVOKABLE void step();

  QStringList solutions() const;

 Q_SIGNALS:

  void paletteChanged();

  void bestSolutionChanged();

  void solutionsChanged();

 private:

  QList<QColor> palette_;

  double bestSolution_;

  QStringList solutions_;

  QList<Solution> generation_;

  QTemporaryDir tempDir_;

  QString newFileName();

};