image/bitmap.cpp at master · AnnaFrolova/image · 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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#include "bitmap.h"
#include <QImage>
#include <QRgb>
#include <QPixmap>
#include <vector>
#include "pixel.h"
#include <QDebug>
#include <iostream>

Bitmap::Bitmap()
{

}

// Загрузка изображения из файла.

Bitmap::Bitmap(QString filename, double scaleValue)
{
    this->pixels.clear();
    QImage image;
    // Загрузка изображения в QImage. Избавляет от кучи проблем с декодированием.
    image.load(filename);
    if(scaleValue != 1)
    {
        image = image.scaled(image.width() * scaleValue, image.height() * scaleValue, Qt::KeepAspectRatio);
    }
    // Сохраняем формат изображения. Понадобится при сохранении.
    this->imageFormat = image.format();
    for(int x = 0; x < image.width(); ++x)
    {
        for(int y = 0; y < image.height(); ++y)
        {
            QRgb value = image.pixel(x, y);
            int red = qRed(value);
            int green = qGreen(value);
            int blue = qBlue(value);
            this->pixels.push_back(Pixel(red, green, blue));
        }
    }
    this->height = image.height();
    this->width = image.width();
}

// Конструктор копирования
Bitmap::Bitmap(const Bitmap &other)
{
    pixels = other.pixels;
    width = other.width;
    height = other.height;
    imageFormat = other.imageFormat;
}

Bitmap::Bitmap(size_t width, size_t height, std::vector<Pixel> pixels, QImage::Format format)
{
    this->height = height;
    this->width = width;
    this->pixels = pixels;
    this->imageFormat = format;
}

// Экспорт в QPixmap --- формат, используемый QT для вывода изображений на экран, сохранения.

QImage Bitmap::toImage()
{
    QImage image = QImage(this->width, this->height, this->imageFormat);
    // Изображение должно быть проинициализировано хоть чем-то перед тем как с ним работать.
    image.fill(Qt::gray);
    // Перебор всех точек, затем установка значений в image.
    for(size_t x = 0; x < this->width; ++x)
    {
        for(size_t y = 0; y < this->height; ++y)
        {
            Pixel pixel = this->getPixel(x, y);
            image.setPixel(x, y, qRgba(pixel.red, pixel.green, pixel.blue, 1));
        }
    }
    return image;
}

// Доступ к точке по ее координатам.
Pixel Bitmap::getPixel(int x, int y)
{
    int position = x * this->height + y;
    Pixel pixel = this->pixels[position];
    return pixel;
}

// Устанавливает значение точки.
void Bitmap::setPixel(int x, int y, Pixel pixel)
{
    int position = x * this->height + y;
    pixel.validate();
    this->pixels[position] = pixel;
}
//Среднее значение пикселей изображения
Pixel Bitmap::averagePixel()
{
    long long totalRed = 0, totalGreen = 0, totalBlue = 0;

    for(auto selectedPixel = this->pixels.begin(); selectedPixel != this->pixels.end(); ++selectedPixel)
    {
        totalRed += selectedPixel->red;
        totalGreen += selectedPixel->green;
        totalBlue += selectedPixel->blue;
    }

    totalRed /= this->pixels.size();
    totalGreen /= this->pixels.size();
    totalBlue /= this->pixels.size();

    Pixel average = Pixel(totalRed, totalGreen, totalBlue);
    // Вызываем validate, но все должно быть ок и без него.
    average.validate();
    return average;
}