// SPDX-License-Identifier: MIT
// Image view mode.
// Copyright (C) 2020 Artem Senichev <artemsen@gmail.com>

#include "viewer.hpp"

#include "application.hpp"
#include "imageloader.hpp"
#include "log.hpp"
#include "render.hpp"
#include "resources.hpp"
#include "text.hpp"

#include <cmath>
#include <format>
#include <utility>

/** Max scale factor. */
constexpr double MAX_SCALE = 100.0;

Viewer& Viewer::self()
{
    static Viewer singleton;
    return singleton;
}

Viewer::Viewer()
{
    // default settings

    auto_center = true;
    imagelist_loop = true;
    scale = 1.0;

    default_scale = Scale::Optimal;
    default_pos = Position::Center;

    window_bkg = argb_t(argb_t::max, 0, 0, 0);

    tr_chessboard = true;
    tr_cbsize = 20;
    tr_cbcolor[0] = { argb_t::max, 0x33, 0x33, 0x33 };
    tr_cbcolor[1] = { argb_t::max, 0x4c, 0x4c, 0x4c };
    tr_bgcolor = { argb_t::max, 0, 0, 0 };

    animation = true;

    text_scheme[static_cast<size_t>(Text::TopLeft)] = {
        "File:\t{name}",
        "Format:\t{format}",
        "File size:\t{sizehr}",
        "File time:\t{time}",
        "EXIF date:\t{meta.Exif.Photo.DateTimeOriginal}",
        "EXIF camera:\t{meta.Exif.Image.Model}"
    };
    text_scheme[static_cast<size_t>(Text::TopRight)] = {
        "Image:\t{list.index} of {list.total}",
        "Frame:\t{frame.index} of {frame.total}",
        "Size:\t{frame.width}x{frame.height}"
    };
    text_scheme[static_cast<size_t>(Text::BottomLeft)] = { "Scale: {scale}" };

    // default key bindings: general management
    bind_input(InputKeyboard { XKB_KEY_Escape, KEYMOD_NONE }, []() {
        Application::self().exit(0);
    });
    bind_input(InputKeyboard { XKB_KEY_Return, KEYMOD_NONE }, []() {
        Application::self().set_mode(Application::Mode::Gallery);
    });
    bind_input(InputKeyboard { XKB_KEY_s, KEYMOD_NONE }, []() {
        Application::self().set_mode(Application::Mode::Slideshow);
    });
    bind_input(InputKeyboard { XKB_KEY_Insert, KEYMOD_NONE }, [this]() {
        const ImageEntryPtr entry = current_entry();
        entry->mark = !entry->mark;
        Application::redraw();
    });
    bind_input(InputKeyboard { XKB_KEY_f, KEYMOD_NONE }, []() {
        Ui* ui = Application::get_ui();
        ui->set_fullscreen(!ui->get_fullscreen());
    });
    bind_input(InputKeyboard { XKB_KEY_a, KEYMOD_NONE }, []() {
        bool& antialiasing = Render::self().antialiasing;
        antialiasing = !antialiasing;
        Application::redraw();
    });
    // image transform
    bind_input(InputKeyboard { XKB_KEY_bracketleft, KEYMOD_NONE }, [this]() {
        rotate(270);
    });
    bind_input(InputKeyboard { XKB_KEY_bracketright, KEYMOD_NONE }, [this]() {
        rotate(90);
    });
    bind_input(InputKeyboard { XKB_KEY_m, KEYMOD_NONE }, [this]() {
        flip_vertical();
    });
    bind_input(InputKeyboard { XKB_KEY_m, KEYMOD_SHIFT }, [this]() {
        flip_horizontal();
    });
    // text layer
    bind_input(InputKeyboard { XKB_KEY_t, KEYMOD_NONE }, []() {
        Text& text = Text::self();
        if (text.is_visible()) {
            text.hide();
        } else {
            text.show();
        }
    });
    // next/prev image
    bind_input(InputKeyboard { XKB_KEY_Next, KEYMOD_NONE }, [this]() {
        open(ImageList::Dir::Next);
    });
    bind_input(InputKeyboard { XKB_KEY_Prior, KEYMOD_NONE }, [this]() {
        open(ImageList::Dir::Prev);
    });
    // next/prev frame
    bind_input(InputKeyboard { XKB_KEY_Next, KEYMOD_SHIFT }, [this]() {
        enable_animation(false);
        next_frame();
    });
    bind_input(InputKeyboard { XKB_KEY_Prior, KEYMOD_SHIFT }, [this]() {
        enable_animation(false);
        prev_frame();
    });
    // scale
    bind_input(InputKeyboard { XKB_KEY_equal, KEYMOD_NONE }, [this]() {
        set_scale(scale + scale / 10.0);
    });
    bind_input(InputKeyboard { XKB_KEY_plus, KEYMOD_SHIFT }, [this]() {
        set_scale(scale + scale / 10.0);
    });
    bind_input(InputKeyboard { XKB_KEY_minus, KEYMOD_NONE }, [this]() {
        set_scale(scale - scale / 10.0);
    });
    bind_input(InputKeyboard { XKB_KEY_BackSpace, KEYMOD_NONE }, [this]() {
        reset();
    });
    // image position
    bind_input(InputKeyboard { XKB_KEY_Left, KEYMOD_NONE }, [this]() {
        set_position(get_position() +
                     Point { static_cast<ssize_t>(window_size.width / 10), 0 });
    });
    bind_input(InputKeyboard { XKB_KEY_Right, KEYMOD_NONE }, [this]() {
        set_position(
            get_position() +
            Point { -static_cast<ssize_t>(window_size.width / 10), 0 });
    });
    bind_input(InputKeyboard { XKB_KEY_Up, KEYMOD_NONE }, [this]() {
        set_position(
            get_position() +
            Point { 0, static_cast<ssize_t>(window_size.height / 10) });
    });
    bind_input(InputKeyboard { XKB_KEY_Down, KEYMOD_NONE }, [this]() {
        set_position(
            get_position() +
            Point { 0, -static_cast<ssize_t>(window_size.height / 10) });
    });
    // mouse
    bind_input(InputMouse { InputMouse::SCROLL_UP, KEYMOD_NONE }, [this]() {
        set_position(get_position() + Point { 0, -20 });
    });
    bind_input(InputMouse { InputMouse::SCROLL_DOWN, KEYMOD_NONE }, [this]() {
        set_position(get_position() + Point { 0, 20 });
    });
    bind_input(InputMouse { InputMouse::SCROLL_LEFT, KEYMOD_NONE }, [this]() {
        set_position(get_position() + Point { -20, 0 });
    });
    bind_input(InputMouse { InputMouse::SCROLL_RIGHT, KEYMOD_NONE }, [this]() {
        set_position(get_position() + Point { 20, 0 });
    });
    bind_input(InputMouse { InputMouse::SCROLL_UP, KEYMOD_CTRL }, [this]() {
        set_scale(scale + scale / 10.0,
                  Application::self().get_ui()->get_mouse());
    });
    bind_input(InputMouse { InputMouse::SCROLL_DOWN, KEYMOD_CTRL }, [this]() {
        set_scale(scale - scale / 10.0,
                  Application::self().get_ui()->get_mouse());
    });

    bind_image_drag(InputMouse { InputMouse::BUTTON_LEFT, KEYMOD_NONE });
}

bool Viewer::open(const ImageList::Dir dir)
{
    return switch_image(dir, nullptr);
}

bool Viewer::open(const ImageEntryPtr& entry)
{
    ImagePtr new_image = nullptr;

    if (image && image->entry == entry) {
        // remove entry from cache in reloading mode
        const std::scoped_lock lock(image_pool.mutex);
        image_pool.history.get(entry);
        image_pool.preload.get(entry);
    } else {
        // get file from history/preload cache
        const std::scoped_lock lock(image_pool.mutex);
        new_image = image_pool.preload.get(entry);
        if (new_image) {
            Log::verbose("Got image {} from preloading cache",
                         entry->path.filename().string());
        } else {
            new_image = image_pool.history.get(entry);
            if (new_image) {
                Log::verbose("Got image {} from history cache",
                             entry->path.filename().string());
            }
        }
    }

    if (!new_image) {
        new_image = ImageLoader::self().load(entry);
        if (!new_image) {
            return false; // failed to load
        }
    }

    set_image(new_image);
    return true;
}

bool Viewer::reload()
{
    if (!open(image->entry) && !open(ImageList::Dir::Next) &&
        !open(ImageList::Dir::Prev)) {
        Log::info("No more images to view, exit");
        Application::self().exit(0);
        return false;
    }
    return true;
}

size_t Viewer::next_frame()
{
    size_t index = frame_index + 1;
    if (index >= image->frames.size()) {
        index = 0;
    }
    set_frame(index);
    return index;
}

size_t Viewer::prev_frame()
{
    const size_t index =
        frame_index ? frame_index - 1 : image->frames.size() - 1;
    set_frame(index);
    return index;
}

void Viewer::export_frame(const std::filesystem::path& path) const
{
    const Pixmap& pm = image->frames[frame_index].pm;
    if (!pm.save(path)) {
        Text::self().set_status("Failed to export image");
    }
}

void Viewer::set_scale(const Scale sc)
{
    const Pixmap& pm = image->frames[frame_index].pm;
    const double ratio_w = static_cast<double>(window_size.width) / pm.width();
    const double ratio_h =
        static_cast<double>(window_size.height) / pm.height();

    double abs_sc = 1.0;
    switch (sc) {
        case Scale::Keep:
        case Scale::Optimal:
            abs_sc = std::min(ratio_w, ratio_h);
            if (abs_sc > 1.0) {
                abs_sc = 1.0;
            }
            break;
        case Scale::FitWindow:
            abs_sc = std::min(ratio_w, ratio_h);
            break;
        case Scale::FitWidth:
            abs_sc = ratio_w;
            break;
        case Scale::FitHeight:
            abs_sc = ratio_h;
            break;
        case Scale::FillWindow:
            abs_sc = std::max(ratio_w, ratio_h);
            break;
        case Scale::RealSize:
            abs_sc = 1.0; // 100 %
            break;
    }

    set_scale(abs_sc);
}

void Viewer::set_scale(const double sc, const Point& preserve)
{
    double new_scale = sc;

    // check scale limits
    const Pixmap& pm = image->frames[frame_index].pm;
    const Size scaled = static_cast<Size>(pm) * new_scale;
    if (!scaled) {
        return;
    }
    if (sc > MAX_SCALE) {
        new_scale = MAX_SCALE;
    }

    // save base point of the image
    const Point bpt(preserve ? preserve.x : window_size.width / 2,
                    preserve ? preserve.y : window_size.height / 2);
    const double pr_x = (static_cast<double>(bpt.x) - position.x) / scale;
    const double pr_y = (static_cast<double>(bpt.y) - position.y) / scale;

    // set new scale
    scale = new_scale;

    // restore center
    position.x = bpt.x - pr_x * scale;
    position.y = bpt.y - pr_y * scale;

    update_text(TextUpdate::Scale);

    fixup_position();
}

void Viewer::reset()
{
    if (std::holds_alternative<Scale>(default_scale)) {
        set_scale(std::get<Scale>(default_scale));
    } else {
        set_scale(std::get<double>(default_scale));
    }
    set_position(default_pos);
}

void Viewer::set_position(const Position pos)
{
    const Pixmap& pm = image->frames[frame_index].pm;
    const Size scaled = static_cast<Size>(pm) * scale;

    switch (pos) {
        case Position::Center:
            position.x = window_size.width / 2 - scaled.width / 2;
            position.y = window_size.height / 2 - scaled.height / 2;
            break;
        case Position::TopCenter:
            position.x = window_size.width / 2 - scaled.width / 2;
            position.y = 0;
            break;
        case Position::BottomCenter:
            position.x = window_size.width / 2 - scaled.width / 2;
            position.y = window_size.height - scaled.height;
            break;
        case Position::LeftCenter:
            position.x = 0;
            position.y = window_size.height / 2 - scaled.height / 2;
            break;
        case Position::RightCenter:
            position.x = window_size.width - scaled.width;
            position.y = window_size.height / 2 - scaled.height / 2;
            break;
        case Position::TopLeft:
            position.x = 0;
            position.y = 0;
            break;
        case Position::TopRight:
            position.x = window_size.width - scaled.width;
            position.y = 0;
            break;
        case Position::BottomLeft:
            position.x = 0;
            position.y = window_size.height - scaled.height;
            break;
        case Position::BottomRight:
            position.x = window_size.width - scaled.width;
            position.y = window_size.height - scaled.height;
            break;
    }

    Application::redraw();
}

void Viewer::set_position(const Point& pos)
{
    position = pos;
    fixup_position();
}

void Viewer::flip_vertical()
{
    image->flip_vertical();
    Application::redraw();
}

void Viewer::flip_horizontal()
{
    image->flip_horizontal();
    Application::redraw();
}

void Viewer::rotate(const size_t angle)
{
    assert(angle == 90 || angle == 180 || angle == 270);

    image->rotate(angle);

    const Pixmap& pm = image->frames[frame_index].pm;
    const ssize_t diff =
        static_cast<ssize_t>(pm.width()) - static_cast<ssize_t>(pm.height());
    const ssize_t shift = (scale * diff) / 2;
    position.x -= shift;
    position.y += shift;

    update_text(TextUpdate::Frame);

    fixup_position();
}

void Viewer::enable_animation(const bool enable)
{
    const size_t duration = image->frames[frame_index].duration;
    animation = enable && image->frames.size() > 1 && duration;
    animation_timer.reset(enable ? duration : 0, 0);
}

void Viewer::set_window_background(const argb_t& color)
{
    window_bkg = color;
    if (is_active()) {
        Application::redraw();
    }
}

void Viewer::set_window_background(const Background mode)
{
    window_bkg = mode;
    if (is_active()) {
        Application::redraw();
    }
}

void Viewer::set_image_background(const argb_t& color)
{
    tr_chessboard = false;
    tr_bgcolor = color;
    if (is_active()) {
        Application::redraw();
    }
}

void Viewer::set_image_chessboard(const size_t size, const argb_t& color1,
                                  const argb_t& color2)
{
    tr_chessboard = true;
    tr_cbsize = size;
    tr_cbcolor[0] = color1;
    tr_cbcolor[1] = color2;
    if (is_active()) {
        Application::redraw();
    }
}

void Viewer::set_preload_limit(const size_t size)
{
    image_pool.preload.capacity = size;
}

void Viewer::set_history_limit(const size_t size)
{
    image_pool.history.capacity = size;
}

void Viewer::bind_image_drag(const InputMouse& input)
{
    drag = input;
    bind_input(drag, []() {
        Application::get_ui()->set_cursor(Ui::CursorShape::Drag);
    });
}

void Viewer::initialize()
{
    Application::self().add_fdpoll(animation_timer, [this]() {
        next_frame();
        enable_animation(true);
    });
}

void Viewer::activate(const ImageEntryPtr& entry, const Size& wnd)
{
    AppMode::activate(entry, wnd);

    window_size = wnd;

    if (image && image->entry == entry) {
        set_image(image);
    } else if (!switch_image(ImageList::Dir::Next, entry) &&
               !switch_image(ImageList::Dir::Prev, entry)) {
        Log::info("No more images to view, exit");
        Application::self().exit(0);
    }
}

void Viewer::deactivate()
{
    preloader_stop();

    // restore cursor and content type
    Ui* ui = Application::self().get_ui();
    ui->set_ctype(Ui::ContentType::Static);
    ui->set_cursor(Ui::CursorShape::Default);
}

ImageEntryPtr Viewer::current_entry()
{
    return image ? image->entry : nullptr;
}

void Viewer::window_resize(const Size& wnd)
{
    window_size = wnd;
    fixup_position();
}

void Viewer::window_redraw(Pixmap& wnd)
{
    const Pixmap& pm = image->frames[frame_index].pm;
    const Rectangle imgr = { position, static_cast<Size>(pm) * scale };

    // clear image background
    if (pm.format() == Pixmap::ARGB) {
        if (tr_chessboard) {
            wnd.grid(imgr, tr_cbsize, tr_cbcolor[0], tr_cbcolor[1]);
        } else {
            wnd.fill(imgr, tr_bgcolor);
        }
    }

    // put image on window surface
    image->draw(frame_index, wnd, scale, position.x, position.y);

    // fill window background
    if (std::holds_alternative<argb_t>(window_bkg)) {
        Render::self().fill_inverse(wnd, imgr, std::get<argb_t>(window_bkg));
    } else {
        switch (std::get<Background>(window_bkg)) {
            case Background::Mirror:
                Render::self().mirror_background(wnd, imgr);
                break;
            case Background::Extend:
                Render::self().extend_background(wnd, imgr);
                break;
            case Background::Auto:
                if (imgr.width > imgr.height) {
                    Render::self().mirror_background(wnd, imgr);
                } else {
                    Render::self().extend_background(wnd, imgr);
                }
                break;
        }
    }

    // mark icon
    if (image->entry->mark) {
        const ssize_t margin = 10;
        const ssize_t x = static_cast<ssize_t>(wnd.width()) -
            static_cast<ssize_t>(Resource::mark.width()) - margin;
        const ssize_t y = static_cast<ssize_t>(wnd.height()) -
            static_cast<ssize_t>(Resource::mark.height()) - margin;
        wnd.mask(Resource::mark, { x, y }, mark_color);
    }
}

void Viewer::handle_mmove(const InputMouse& input, const Point&,
                          const Point& delta)
{
    if (drag && drag == input) {
        set_position(position + delta);
    }
}

void Viewer::handle_pinch(const double scale_delta)
{
    set_scale(scale + scale_delta * pinch_factor);
}

void Viewer::handle_imagelist(const ImageListEvent event,
                              const ImageEntryPtr& entry)
{
    if (event == ImageListEvent::Modify || event == ImageListEvent::Remove) {
        // remove entry from cache
        const std::scoped_lock lock(image_pool.mutex);
        image_pool.history.get(entry);
        image_pool.preload.get(entry);
    }

    switch (event) {
        case ImageListEvent::Create:
            preloader_start();
            break;
        case ImageListEvent::Modify:
            if (entry == image->entry) {
                reload();
            }
            break;
        case ImageListEvent::Remove:
            if (entry == image->entry && !open(ImageList::Dir::Next) &&
                !open(ImageList::Dir::Prev)) {
                Log::info("No more images to view, exit");
                Application::self().exit(0);
            }
            break;
    }
}

bool Viewer::switch_image(const ImageList::Dir dir, const ImageEntryPtr& entry)
{
    assert(image || entry);

    ImageList& il = ImageList::self();

    ImageEntryPtr next_entry = entry ? entry : il.get(image->entry, dir);

    const bool forward =
        dir != ImageList::Dir::Prev && dir != ImageList::Dir::PrevParent;

    while (true) {
        if (!next_entry && imagelist_loop) {
            // reshuffle random on new loop
            if (il.get_order() == ImageList::Order::Random) {
                il.set_order(ImageList::Order::Random);
            }

            next_entry =
                il.get(nullptr,
                       forward ? ImageList::Dir::First : ImageList::Dir::Last);

            if (image && image->entry == next_entry &&
                il.get_order() == ImageList::Order::Random) {
                next_entry = il.get(next_entry, dir);
            }
        }
        if (!next_entry) {
            break; // no more entries
        }
        if (image && image->entry == next_entry) {
            break; // loop complete
        }

        if (open(next_entry)) {
            return true;
        }

        next_entry = il.remove(next_entry, forward);
    }

    return false;
}

void Viewer::set_image(const ImagePtr& img)
{
    if (image && image != img) {
        // put current image to history
        const std::scoped_lock lock(image_pool.mutex);
        image_pool.history.put(image);
    }

    image = img;
    frame_index = 0;
    const Pixmap& pm = image->frames[0].pm;

    const bool is_animation =
        image->frames.size() > 1 && image->frames[0].duration;

    // set content type
    Ui* ui = Application::self().get_ui();
    ui->set_ctype(is_animation ? Ui::ContentType::Animation
                               : Ui::ContentType::Static);

    if (std::holds_alternative<Scale>(default_scale) &&
        std::get<Scale>(default_scale) == Scale::Keep && previmg) {
        // handle "keep scale" mode
        const ssize_t diff_w = static_cast<ssize_t>(previmg.width) -
            static_cast<ssize_t>(pm.width());
        const ssize_t diff_h = static_cast<ssize_t>(previmg.height) -
            static_cast<ssize_t>(pm.height());
        position.x += std::floor(scale * diff_w) / 2.0;
        position.y += std::floor(scale * diff_h) / 2.0;
        fixup_position();
    } else {
        if (std::holds_alternative<Scale>(default_scale)) {
            set_scale(std::get<Scale>(default_scale));
        } else {
            set_scale(std::get<double>(default_scale));
        }

        set_position(default_pos);
    }

    previmg = pm;

    enable_animation(is_animation);

    preloader_start();

    switch_current();
    update_text(TextUpdate::All);
}

void Viewer::set_frame(const size_t index)
{
    assert(index < image->frames.size());

    frame_index = index;
    update_text(TextUpdate::Frame);
    Application::redraw();
}

void Viewer::update_text(const TextUpdate what) const
{
    Text& text = Text::self();

    if (what == TextUpdate::All) {
        text.reset(image);
    }

    if (what == TextUpdate::All || what == TextUpdate::Frame) {
        const Pixmap& pm = image->frames[frame_index].pm;
        text.set_field(Text::FIELD_FRAME_INDEX,
                       std::to_string(frame_index + 1));
        text.set_field(Text::FIELD_FRAME_WIDTH, std::to_string(pm.width()));
        text.set_field(Text::FIELD_FRAME_HEIGHT, std::to_string(pm.height()));
    }

    if (what == TextUpdate::All || what == TextUpdate::Scale) {
        std::string tscale;
        if (scale >= 0.1) {
            tscale = std::format("{}%", static_cast<size_t>(scale * 100));
        } else {
            tscale = std::format("{:.02}%", scale * 100);
        }
        text.set_field(Text::FIELD_SCALE, tscale);
    }

    text.update();
}

void Viewer::fixup_position()
{
    const Pixmap& pm = image->frames[frame_index].pm;
    const Size scaled = static_cast<Size>(pm) * scale;

    if (!auto_center) {
        // don't let canvas to be far out of window
        if (position.x + static_cast<ssize_t>(scaled.width) < 0) {
            position.x = -static_cast<ssize_t>(scaled.width);
        }
        if (std::cmp_greater(position.x, window_size.width)) {
            position.x = window_size.width;
        }
        if (position.y + static_cast<ssize_t>(scaled.height) < 0) {
            position.y = -static_cast<ssize_t>(scaled.height);
        }
        if (std::cmp_greater(position.y, window_size.height)) {
            position.y = window_size.height;
        }
    } else {
        if (scaled.width <= window_size.width) {
            // entire image fits horizontally, center it
            position.x = window_size.width / 2 - scaled.width / 2;
        } else {
            // prevent going outside the window
            const ssize_t right = position.x + scaled.width;
            if (position.x > 0 && std::cmp_greater(right, window_size.width)) {
                position.x = 0;
            }
            if (position.x < 0 && std::cmp_less(right, window_size.width)) {
                position.x = window_size.width - scaled.width;
            }
        }

        if (scaled.height <= window_size.height) {
            // entire image fits vertically, center it
            position.y = window_size.height / 2 - scaled.height / 2;
        } else {
            // prevent going outside the window
            const ssize_t bottom = position.y + scaled.height;
            if (position.y > 0 &&
                std::cmp_greater(bottom, window_size.height)) {
                position.y = 0;
            }
            if (position.y < 0 && std::cmp_less(bottom, window_size.height)) {
                position.y = window_size.height - scaled.height;
            }
        }
    }

    Application::redraw();
}

void Viewer::preloader_start()
{
    if (image_pool.thread.joinable()) {
        if (image_pool.stop) {
            image_pool.thread.join();
        } else {
            return; // already in progress
        }
    }

    image_pool.thread = std::thread([this] {
        ImageList& il = ImageList::self();
        ImagePtr current_image = image;
        ImageEntryPtr last_entry = image->entry;
        size_t counter = 0;

        while (!image_pool.stop && counter < image_pool.preload.capacity) {
            if (current_image != image) {
                // current image has changed, restart preloading
                current_image = image;
                last_entry = image->entry;
                counter = 0;
            }

            ImageEntryPtr next_entry = il.get(last_entry, ImageList::Dir::Next);
            if (!next_entry && imagelist_loop) {
                next_entry = il.get(nullptr, ImageList::Dir::First);
            }
            if (!next_entry || next_entry == last_entry) {
                // no more images to preload
                const std::scoped_lock lock(image_pool.mutex);
                image_pool.preload.trim(counter);
                break;
            }

            ImagePtr next_image = nullptr;

            // get existing image form history/preload cache
            {
                const std::scoped_lock lock(image_pool.mutex);
                next_image = image_pool.preload.get(next_entry);
                if (!next_image) {
                    next_image = image_pool.history.get(next_entry);
                }
            }

            // load image
            if (!next_image) {
                next_image = ImageLoader::self().load(next_entry);
            }

            if (!next_image) {
                il.remove(next_entry);
            } else {
                Log::verbose("Put image {} to cache",
                             next_entry->path.filename().string());
                const std::scoped_lock lock(image_pool.mutex);
                image_pool.preload.put(next_image);
                last_entry = next_image->entry;
                ++counter;
            }
        }

        image_pool.stop = true;
    });
}

void Viewer::preloader_stop()
{
    if (image_pool.thread.joinable()) {
        image_pool.stop = true;
        image_pool.thread.join();
    }
}

void Viewer::Cache::trim(const size_t size)
{
    cache.resize(size);
}

void Viewer::Cache::put(const ImagePtr& image)
{
    if (capacity) {
        if (cache.size() == capacity) {
            cache.pop_back();
        }
        cache.push_front(image);
    }
}

ImagePtr Viewer::Cache::get(const ImageEntryPtr& entry)
{
    auto it = std::find_if(cache.begin(), cache.end(),
                           [&entry](const ImagePtr& image) {
                               return image->entry == entry;
                           });
    if (it == cache.end()) {
        return nullptr;
    }
    ImagePtr image = *it;
    cache.erase(it);
    return image;
}
