6e6999ce57
Originally I simply thought that passing file paths is quite OK, but as Linus pointed to, it turned out that passing file paths to ensure some files are able to be decoded is awkward because it does not work with images being served over HTTP. Therefore, ideally we should just use the MIME type as an optional argument to ensure that we can always fallback to use that in case sniffing for the correct image type has failed so we can still detect files like with the TGA format, which has no magic bytes.
105 lines
2.8 KiB
C++
105 lines
2.8 KiB
C++
/*
|
|
* Copyright (c) 2018-2021, Andreas Kling <kling@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/LexicalPath.h>
|
|
#include <LibGfx/BMPLoader.h>
|
|
#include <LibGfx/DDSLoader.h>
|
|
#include <LibGfx/GIFLoader.h>
|
|
#include <LibGfx/ICOLoader.h>
|
|
#include <LibGfx/ImageDecoder.h>
|
|
#include <LibGfx/JPGLoader.h>
|
|
#include <LibGfx/PBMLoader.h>
|
|
#include <LibGfx/PGMLoader.h>
|
|
#include <LibGfx/PNGLoader.h>
|
|
#include <LibGfx/PPMLoader.h>
|
|
#include <LibGfx/QOILoader.h>
|
|
#include <LibGfx/TGALoader.h>
|
|
|
|
namespace Gfx {
|
|
|
|
static OwnPtr<ImageDecoderPlugin> probe_and_sniff_for_appropriate_plugin(ReadonlyBytes bytes)
|
|
{
|
|
auto* data = bytes.data();
|
|
auto size = bytes.size();
|
|
OwnPtr<ImageDecoderPlugin> plugin;
|
|
|
|
plugin = make<PNGImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<GIFImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<BMPImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<PBMImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<PGMImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<PPMImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<ICOImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<JPGImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<DDSImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
plugin = make<QOIImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
|
|
return {};
|
|
}
|
|
|
|
static OwnPtr<ImageDecoderPlugin> probe_and_sniff_for_appropriate_plugin_with_known_mime_type(StringView mime_type, ReadonlyBytes bytes)
|
|
{
|
|
auto* data = bytes.data();
|
|
auto size = bytes.size();
|
|
OwnPtr<ImageDecoderPlugin> plugin;
|
|
if (mime_type == "image/x-targa"sv) {
|
|
plugin = make<TGAImageDecoderPlugin>(data, size);
|
|
if (plugin->sniff())
|
|
return plugin;
|
|
}
|
|
return {};
|
|
}
|
|
|
|
RefPtr<ImageDecoder> ImageDecoder::try_create_for_raw_bytes(ReadonlyBytes bytes, Optional<DeprecatedString> mime_type)
|
|
{
|
|
OwnPtr<ImageDecoderPlugin> plugin = probe_and_sniff_for_appropriate_plugin(bytes);
|
|
if (!plugin) {
|
|
if (mime_type.has_value()) {
|
|
plugin = probe_and_sniff_for_appropriate_plugin_with_known_mime_type(mime_type.value(), bytes);
|
|
if (!plugin)
|
|
return {};
|
|
} else {
|
|
return {};
|
|
}
|
|
}
|
|
return adopt_ref_if_nonnull(new (nothrow) ImageDecoder(plugin.release_nonnull()));
|
|
}
|
|
|
|
ImageDecoder::ImageDecoder(NonnullOwnPtr<ImageDecoderPlugin> plugin)
|
|
: m_plugin(move(plugin))
|
|
{
|
|
}
|
|
|
|
}
|