summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibGfx/ImageFormats/PPMLoader.cpp
blob: c8859350afbc54c7abca579a1d7262187eac2664 (plain)
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
/*
 * Copyright (c) 2020, Hüseyin ASLITÜRK <asliturk@hotmail.com>
 * Copyright (c) 2022, the SerenityOS developers.
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include "PPMLoader.h"
#include "PortableImageLoaderCommon.h"
#include <AK/Endian.h>
#include <AK/LexicalPath.h>
#include <AK/ScopeGuard.h>
#include <AK/StringBuilder.h>
#include <LibGfx/Streamer.h>
#include <string.h>

namespace Gfx {

bool read_image_data(PPMLoadingContext& context, Streamer& streamer)
{
    Vector<Gfx::Color> color_data;
    color_data.ensure_capacity(context.width * context.height);

    if (context.type == PPMLoadingContext::Type::ASCII) {
        while (true) {
            auto const red_or_error = read_number(streamer);
            if (red_or_error.is_error())
                break;

            if (!read_whitespace(context, streamer))
                break;

            auto const green_or_error = read_number(streamer);
            if (green_or_error.is_error())
                break;

            if (!read_whitespace(context, streamer))
                break;

            auto const blue_or_error = read_number(streamer);
            if (blue_or_error.is_error())
                break;

            if (!read_whitespace(context, streamer))
                break;

            Color color { (u8)red_or_error.value(), (u8)green_or_error.value(), (u8)blue_or_error.value() };
            if (context.format_details.max_val < 255)
                color = adjust_color(context.format_details.max_val, color);
            color_data.append(color);
        }
    } else if (context.type == PPMLoadingContext::Type::RAWBITS) {
        u8 pixel[3];
        while (streamer.read_bytes(pixel, 3)) {
            color_data.append({ pixel[0], pixel[1], pixel[2] });
        }
    }

    if (context.width * context.height != color_data.size())
        return false;

    if (!create_bitmap(context)) {
        return false;
    }

    set_pixels(context, color_data);

    context.state = PPMLoadingContext::State::Bitmap;
    return true;
}
}