FractalTree/FractalTreeImage.h at master · Indidev/FractalTree · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
#ifndef FRACTALTREEPIXMAP_H
#define FRACTALTREEPIXMAP_H

#include <QImage>
#include <QPainter>
#include <QPaintEvent>
#include <math.h>
#include <QPoint>
#include <QVector2D>
#include <QList>
#include <QPen>
#include <QColor>
#include <QPaintEngine>
#include <time.h>

using namespace std;

class FractalTreeImage : public QImage
{
public:
    explicit FractalTreeImage(int width = 100, int height = 100, int numBranches = 2, int recursionDepth = 1, int rootWidth = 4,
                              float baseSize = 1.0, float branchStretch = 1.0, float leafSize = 0.1, float lowerJitter = 1.0, float upperJitter = 1.0,
                              unsigned int seed = 0, QColor treeColor = Qt::black, QColor leafColor = QColor(0, 198, 0, 200));

    explicit FractalTreeImage(int width = 100, int height = 100, int numBranches = 2, int recursionDepth = 1, int rootWidth = 4,
                              float baseSize = 1.0, float branchStretch = 1.0, float leafSize = 0.1, float lowerJitter = 1.0, float upperJitter = 1.0,
                              unsigned int seed = 0, QColor treeColor = Qt::black, QList<QColor> leafColors = QList<QColor>());
    unsigned int getSeed();

protected:
    struct Endpoint {
        QPoint point;
        QVector2D vec;
    };

    double branches;
    int maxDepth;
    int rootWidth;
    unsigned int seed;
    float baseSize;
    float leafSize;
    float branchStretch;
    float lowerJitter;
    float upperJitter;
    QList<QColor> leafColors;
    QColor treeColor;

    void drawTree();

    void drawLine(QPoint start, QPoint end, QPainter &painter, float thickness = 1);
    void drawLeaf(Endpoint point, int length, QPainter &painter);
    QPoint add(QPoint p, QVector2D v);
    QVector2D rotate(QVector2D vec, double angle);


    double drand();
};

#endif // FRACTALTREEPIXMAP_H