// SPDX-FileCopyrightText: 2006-2024 Knut Reinert & Freie Universität Berlin
// SPDX-FileCopyrightText: 2016-2024 Knut Reinert & MPI für molekulare Genetik
// SPDX-License-Identifier: BSD-3-Clause

/*!\file
 * \brief Provides seqan3::structure_file_output_format and auxiliary classes.
 * \author Jörg Winkler <j.winkler AT fu-berlin.de>
 */

#pragma once

#include <fstream>
#include <set>
#include <string>
#include <vector>

#include <seqan3/alphabet/nucleotide/rna5.hpp>
#include <seqan3/alphabet/structure/structured_rna.hpp>
#include <seqan3/alphabet/structure/wuss.hpp>
#include <seqan3/io/structure_file/output_options.hpp>
#include <seqan3/utility/type_list/type_list.hpp>

namespace seqan3::detail
{

/*!\brief Internal class used to expose the actual format interface to write structure records into the file.
 * \ingroup io_structure_file
 *
 * \tparam format_type The type of the format to be exposed.
 *
 * \details
 *
 * Exposes the protected member function `write_structure_record` from the given `format_type`, such that the file can
 * call the proper function for the selected format.
 */
template <typename format_type>
struct structure_file_output_format_exposer : public format_type
{
public:
    // Can't use `using format_type::write_structure_record` as it produces a hard failure in the format concept check
    // for types that do not model the format concept, i.e. don't offer the proper write_structure_record interface.
    //!\brief Forwards to the seqan3::structure_file_output_format::write_structure_record interface.
    template <typename... ts>
    void write_structure_record(ts &&... args)
    {
        format_type::write_structure_record(std::forward<ts>(args)...);
    }
};

} // namespace seqan3::detail

namespace seqan3
{

/*!\interface seqan3::structure_file_output_format <>
 * \brief The generic concept for structure file out formats.
 * \ingroup io_structure_file
 *
 * \details
 *
 * The details of this concept are only relevant to developers who wish to implement their own format.
 * The requirements for this concept are given as related functions and type traits.
 * Types that satisfy this concept are shown as "implementing this interface".
 *
 * \remark For a complete overview, take a look at \ref io_structure_file
 */
//!\cond
template <typename t>
concept structure_file_output_format = requires (detail::structure_file_output_format_exposer<t> & v,
                                                 std::ofstream & f,
                                                 structure_file_output_options & options,
                                                 rna5_vector & seq,
                                                 std::string & id,
                                                 std::vector<std::set<std::pair<double, size_t>>> & bpp,
                                                 std::vector<wuss51> & structure,
                                                 std::vector<structured_rna<rna5, wuss51>> & structured_seq,
                                                 double energy,
                                                 double react,
                                                 double react_err,
                                                 std::string & comment,
                                                 size_t offset) {
    t::file_extensions;

    {
        v.write_structure_record(f, options, seq, id, bpp, structure, energy, react, react_err, comment, offset)
    } -> std::same_as<void>;

    {
        v.write_structure_record(f,
                                 options,
                                 seq,
                                 id,
                                 bpp,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore)
    } -> std::same_as<void>;

    {
        v.write_structure_record(f,
                                 options,
                                 structured_seq,
                                 id,
                                 std::ignore,
                                 structured_seq,
                                 energy,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore)
    } -> std::same_as<void>;

    {
        v.write_structure_record(f,
                                 options,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore,
                                 std::ignore)
    } -> std::same_as<void>;
    // the last is required to be compile time valid, but should always throw at run-time.
};
//!\endcond

// Workaround for https://github.com/doxygen/doxygen/issues/9379
#if SEQAN3_DOXYGEN_ONLY(1) 0
template <typename t>
class structure_file_output_format
{};
#endif

/*!\name Requirements for seqan3::structure_file_output_format
 * \brief You can expect these **members** on all types that implement seqan3::structure_file_output_format.
 * \memberof seqan3::structure_file_output_format
 * \{
 */

/*!\fn void write_structure_record(stream_type & stream,
 *                                 structure_file_output_options const & options,
 *                                 seq_type && seq,
 *                                 id_type && id,
 *                                 bpp_type && bpp,
 *                                 structure_type && structure,
 *                                 energy_type && energy,
 *                                 react_type && react,
 *                                 react_type && react_err,
 *                                 comment_type && comment,
 *                                 offset_type && offset)
 * \brief Write the given fields to the specified stream.
 * \tparam stream_type      Output stream, must satisfy seqan3::output_stream_over with `char`.
 * \tparam seq_type         Type of the seqan3::field::seq output; must satisfy std::ranges::output_range
 * over a seqan3::alphabet.
 * \tparam id_type          Type of the seqan3::field::id output; must satisfy std::ranges::output_range
 * over a seqan3::alphabet.
 * \tparam bpp_type         Type of the seqan3::field::bpp output; must satisfy std::ranges::output_range
 * over a set of pair of types satisfying std::is_floating_point and std::numeric_limits::is_integer, respectively.
 * \tparam structure_type   Type of the seqan3::field::structure output; must satisfy std::ranges::output_range
 * over a seqan3::rna_structure_alphabet.
 * \tparam energy_type      Type of the seqan3::field::energy output; must satisfy std::is_floating_point.
 * \tparam react_type       Type of the seqan3::field::react and seqan3::field::react_err output;
 * must satisfy std::is_floating_point.
 * \tparam comment_type     Type of the seqan3::field::comment output; must satisfy std::ranges::output_range
 * over a seqan3::alphabet.
 * \tparam offset_type      Type of the seqan3::field::offset output; must satisfy std::numeric_limits::is_integer.
 * \param[in,out] stream    The output stream to write into.
 * \param[in]     options   File specific options passed to the format.
 * \param[in]     seq       The data for seqan3::field::seq output, i.e. the "sequence".
 * \param[in]     id        The data for seqan3::field::id output, e.g. the header line.
 * \param[in]     bpp       The data for seqan3::field::bpp output.
 * \param[in]     structure The data for seqan3::field::structure output.
 * \param[in]     energy    The data for seqan3::field::energy output.
 * \param[in]     react     The data for seqan3::field::react output.
 * \param[in]     react_err The data for seqan3::field::react_err output.
 * \param[in]     comment   The data for seqan3::field::comment output.
 * \param[in]     offset    The data for seqan3::field::offset output.
 *
 * \details
 *
 * ### Additional requirements
 *
 *   * The format must also accept std::ignore as parameter for any of the fields, however it shall throw an exception
 * if one of the fields required for writing the format is marked as such. [this shall be checked inside the function]
 *   * The format does not handle seqan3::field::structured_seq, instead seqan3::structure_file_output splits it into
 * two views and passes it to the format as if they were separate.
 */
/*!\var static inline std::vector<std::string> seqan3::structure_file_output_format::file_extensions
 * \brief The format type is required to provide a vector of all supported file extensions.
 */

//!\}

} // namespace seqan3

namespace seqan3::detail
{

/*!\brief Auxiliary value metafuncton that checks whether a type is a seqan3::type_list and all types meet
 * seqan3::structure_file_output_format [default is false].
 * \ingroup core
 * \see seqan3::type_list_specialisationOfstructure_file_output_formats
 */
template <typename t>
constexpr bool is_type_list_of_structure_file_output_formats_v = false;

/*!\brief Auxiliary value metafuncton that checks whether a type is a seqan3::type_list and all types meet
 * seqan3::structure_file_output_format [overload].
 * \ingroup core
 * \see seqan3::type_list_specialisationOfstructure_file_output_formats
 */
template <typename... ts>
constexpr bool is_type_list_of_structure_file_output_formats_v<type_list<ts...>> =
    (structure_file_output_format<ts> && ...);

/*!\brief Auxiliary concept that checks whether a type is a seqan3::type_list and all types meet
 *        seqan3::structure_file_output_format.
 * \ingroup core
 * \see seqan3::is_type_list_of_structure_file_formats_v
 */
template <typename t>
concept type_list_of_structure_file_output_formats = is_type_list_of_structure_file_output_formats_v<t>;
} // namespace seqan3::detail
