/************************************************************************
 *
 * Copyright (C) 2014-2024 IRCAD France
 * Copyright (C) 2014-2019 IHU Strasbourg
 *
 * This file is part of Sight.
 *
 * Sight is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Sight is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with Sight. If not, see <https://www.gnu.org/licenses/>.
 *
 ***********************************************************************/

#include "data/calibration_info.hpp"

#include <core/com/signal.hpp>
#include <core/com/signal.hxx>
#include <core/com/signals.hpp>

#include <data/registry/macros.hpp>

#include <algorithm>
#include <iterator>

namespace sight::data
{

SIGHT_REGISTER_DATA(sight::data::calibration_info);

const core::com::signals::key_t calibration_info::ADDED_RECORD_SIG   = "added_record";
const core::com::signals::key_t calibration_info::REMOVED_RECORD_SIG = "removed_record";
const core::com::signals::key_t calibration_info::RESET_RECORD_SIG   = "reset_record";
const core::com::signals::key_t calibration_info::GET_RECORD_SIG     = "get_record";

using image_container_t            = std::list<image::sptr>;
using point_list_container_t       = std::list<point_list::sptr>;
using const_image_container_t      = std::list<image::csptr>;
using const_point_list_container_t = std::list<point_list::csptr>;

//------------------------------------------------------------------------------

calibration_info::calibration_info() :
    m_sig_added_record(std::make_shared<added_record_signal_t>()),
    m_sig_removed_record(std::make_shared<removed_record_signal_t>()),
    m_sig_reset_record(std::make_shared<reset_record_signal_t>()),
    m_sig_get_record(std::make_shared<get_record_signal_t>())
{
    core::com::has_signals::m_signals(ADDED_RECORD_SIG, m_sig_added_record);
    core::com::has_signals::m_signals(REMOVED_RECORD_SIG, m_sig_removed_record);
    core::com::has_signals::m_signals(RESET_RECORD_SIG, m_sig_reset_record);
    core::com::has_signals::m_signals(GET_RECORD_SIG, m_sig_get_record);
}

//------------------------------------------------------------------------------

void calibration_info::shallow_copy(const object::csptr& _source)
{
    const auto& other = std::dynamic_pointer_cast<const calibration_info>(_source);

    SIGHT_THROW_EXCEPTION_IF(
        exception(
            "Unable to copy " + (_source ? _source->get_classname() : std::string("<NULL>"))
            + " to " + get_classname()
        ),
        !bool(other)
    );

    m_image_container        = other->m_image_container;
    m_point_list_container_t = other->m_point_list_container_t;

    base_class_t::shallow_copy(other);
}

//------------------------------------------------------------------------------

void calibration_info::deep_copy(const object::csptr& _source, const std::unique_ptr<deep_copy_cache_t>& _cache)
{
    const auto& other = std::dynamic_pointer_cast<const calibration_info>(_source);

    SIGHT_THROW_EXCEPTION_IF(
        exception(
            "Unable to copy " + (_source ? _source->get_classname() : std::string("<NULL>"))
            + " to " + get_classname()
        ),
        !bool(other)
    );

    this->reset_records();
    SIGHT_ASSERT(
        "Lists have not the same size",
        other->m_point_list_container_t.size() == other->m_image_container.size()
    );

    auto img_iter = other->m_image_container.begin();
    for(const data::point_list::sptr& pl : other->m_point_list_container_t)
    {
        data::image::sptr other_img     = data::object::copy(*img_iter, _cache);
        data::point_list::sptr other_pl = data::object::copy(pl, _cache);
        this->add_record(other_img, other_pl);

        ++img_iter;
    }

    base_class_t::deep_copy(other, _cache);
}

//------------------------------------------------------------------------------

void calibration_info::add_record(const data::image::sptr& _img, const data::point_list::sptr& _pl)
{
    m_image_container.push_back(_img);
    m_point_list_container_t.push_back(_pl);
}

//------------------------------------------------------------------------------

void calibration_info::remove_record(std::size_t _idx)
{
    SIGHT_ASSERT("index out of bound ", _idx < m_point_list_container_t.size());

    auto pl_it  = m_point_list_container_t.begin();
    auto img_it = m_image_container.begin();

    std::advance(pl_it, static_cast<point_list_container_t::iterator::difference_type>(_idx));
    std::advance(img_it, static_cast<image_container_t::iterator::difference_type>(_idx));

    m_point_list_container_t.erase(pl_it);
    m_image_container.erase(img_it);
}

//------------------------------------------------------------------------------

void calibration_info::reset_records()
{
    m_point_list_container_t.clear();
    m_image_container.clear();
}

//------------------------------------------------------------------------------

image_container_t calibration_info::get_image_container()
{
    return m_image_container;
}

//------------------------------------------------------------------------------

const_image_container_t calibration_info::get_image_container() const
{
    // Transform to csptr
    const_image_container_t const_image_list;
    for(const auto& image : m_image_container)
    {
        const_image_list.push_back(image);
    }

    return const_image_list;
}

//------------------------------------------------------------------------------

point_list_container_t calibration_info::get_point_list_container()
{
    return m_point_list_container_t;
}

//------------------------------------------------------------------------------

const_point_list_container_t calibration_info::get_point_list_container() const
{
    // Transform to csptr
    const_point_list_container_t const_point_list;
    for(const auto& pl : m_point_list_container_t)
    {
        const_point_list.push_back(pl);
    }

    return const_point_list;
}

//------------------------------------------------------------------------------

data::point_list::csptr calibration_info::get_point_list(const data::image::csptr& _img) const
{
    data::point_list::sptr pl;

    SIGHT_ASSERT("Lists have not the same size", m_image_container.size() == m_point_list_container_t.size());
    std::size_t dist = 0;
    image_container_t::const_iterator it;
    for(it = m_image_container.begin() ; it != m_image_container.end() && *(it) != _img ; ++it, ++dist)
    {
    }

    auto pl_it = m_point_list_container_t.begin();
    std::advance(pl_it, static_cast<point_list_container_t::const_iterator::difference_type>(dist));

    if(it != m_image_container.end())
    {
        pl = *(pl_it);
    }

    return pl;
}

//------------------------------------------------------------------------------

data::image::csptr calibration_info::get_image(const data::point_list::csptr& _pl) const
{
    data::image::sptr img;

    SIGHT_ASSERT("Lists have not the same size", m_image_container.size() == m_point_list_container_t.size());

    std::size_t dist = 0;
    point_list_container_t::const_iterator it;
    for(it = m_point_list_container_t.begin() ; it != m_point_list_container_t.end() && *(it) != _pl ; ++it, ++dist)
    {
    }

    auto img_it = m_image_container.begin();
    std::advance(img_it, static_cast<image_container_t::const_iterator::difference_type>(dist));

    if(it != m_point_list_container_t.end())
    {
        img = *(img_it);
    }

    return img;
}

//------------------------------------------------------------------------------

data::image::sptr calibration_info::get_image(std::size_t _idx)
{
    SIGHT_ASSERT("index out of bound ", _idx < m_image_container.size());

    auto img_it = m_image_container.begin();

    std::advance(img_it, static_cast<image_container_t::const_iterator::difference_type>(_idx));

    return *(img_it);
}

//------------------------------------------------------------------------------

data::image::csptr calibration_info::get_image(std::size_t _idx) const
{
    SIGHT_ASSERT("index out of bound ", _idx < m_image_container.size());

    auto img_it = m_image_container.begin();

    std::advance(img_it, static_cast<image_container_t::const_iterator::difference_type>(_idx));

    return *(img_it);
}

//------------------------------------------------------------------------------

bool calibration_info::operator==(const calibration_info& _other) const noexcept
{
    if(!core::is_equal(m_image_container, _other.m_image_container)
       || !core::is_equal(m_point_list_container_t, _other.m_point_list_container_t))
    {
        return false;
    }

    // Super class last
    return base_class_t::operator==(_other);
}

//------------------------------------------------------------------------------

bool calibration_info::operator!=(const calibration_info& _other) const noexcept
{
    return !(*this == _other);
}

//------------------------------------------------------------------------------

} //namespace sight::data