abaddon/imgmanager.cpp

#include "imgmanager.hpp"
#include "util.hpp"

ImageManager::ImageManager() {
    m_cb_dispatcher.connect(sigc::mem_fun(*this, &ImageManager::RunCallbacks));
}

Cache &ImageManager::GetCache() {
    return m_cache;
}

Glib::RefPtr<Gdk::Pixbuf> ImageManager::ReadFileToPixbuf(std::string path) {
    const auto &data = ReadWholeFile(path);
    if (data.size() == 0) return Glib::RefPtr<Gdk::Pixbuf>(nullptr);
    auto loader = Gdk::PixbufLoader::create();
    loader->signal_size_prepared().connect([&loader](int w, int h) {
        int cw, ch;
        GetImageDimensions(w, h, cw, ch); // what could go wrong
        loader->set_size(cw, ch);
    });
    loader->write(static_cast<const guint8 *>(data.data()), data.size());
    loader->close();
    auto buf = loader->get_pixbuf();

    return buf;
}

void ImageManager::LoadFromURL(std::string url, callback_type cb) {
    sigc::signal<void(Glib::RefPtr<Gdk::Pixbuf>)> signal;
    signal.connect(cb);
    m_cache.GetFileFromURL(url, [this, url, signal](std::string path) {
        try {
            auto buf = ReadFileToPixbuf(path);
            m_cb_mutex.lock();
            m_cb_queue.push([signal, buf]() { signal.emit(buf); });
            m_cb_dispatcher.emit();
            m_cb_mutex.unlock();
        } catch (std::exception &e) {
            fprintf(stderr, "err loading pixbuf from %s: %s\n", path.c_str(), e.what());
        }
    });
}

void ImageManager::RunCallbacks() {
    m_cb_mutex.lock();
    m_cb_queue.front()();
    m_cb_queue.pop();
    m_cb_mutex.unlock();
}

Glib::RefPtr<Gdk::Pixbuf> ImageManager::GetFromURLIfCached(std::string url) {
    std::string path = m_cache.GetPathIfCached(url);
    if (path != "")
        return ReadFileToPixbuf(path);

    return Glib::RefPtr<Gdk::Pixbuf>(nullptr);
}

Glib::RefPtr<Gdk::Pixbuf> ImageManager::GetPlaceholder(int size) {
    std::string name = "/placeholder" + std::to_string(size);
    if (m_pixs.find(name) != m_pixs.end())
        return m_pixs.at(name);

    try {
        auto buf = Gdk::Pixbuf::create_from_file("res/decamarks.png", size, size);
        m_pixs[name] = buf;
        return buf;
    } catch (std::exception &e) {
        fprintf(stderr, "error loading placeholder\n");
        return Glib::RefPtr<Gdk::Pixbuf>(nullptr);
    }
}