// ==========================================================================
//                 SeqAn - The Library for Sequence Analysis
// ==========================================================================
// Copyright (c) 2006-2024, Knut Reinert, FU Berlin
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
//     * Redistributions of source code must retain the above copyright
//       notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above copyright
//       notice, this list of conditions and the following disclaimer in the
//       documentation and/or other materials provided with the distribution.
//     * Neither the name of Knut Reinert or the FU Berlin nor the names of
//       its contributors may be used to endorse or promote products derived
//       from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL KNUT REINERT OR THE FU BERLIN BE LIABLE
// FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
// CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
// LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
// OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
// DAMAGE.
//
// ==========================================================================
// Author: David Weese <david.weese@fu-berlin.de>
// ==========================================================================

#ifndef SEQAN_MODIFIER_MODIFIER_FUNCTORS_H_
#define SEQAN_MODIFIER_MODIFIER_FUNCTORS_H_

#include <cctype>

// TODO(holtgrew): Make the structs here into classes.

namespace seqan2
{

// ==========================================================================
// Forwards
// ==========================================================================

// ==========================================================================
// Classes, Enums, Typedefs
// ==========================================================================

// --------------------------------------------------------------------------
// Class FunctorUpcase
// --------------------------------------------------------------------------

/*!
 * @class FunctorUpcase
 * @headerfile <seqan/modifier.h>
 * @brief Functor that returns the upper case character for a given character.
 *
 * @signature template <typename TInType[, typename TResult]>
 *            struct FunctorUpcase;
 *
 * @tparam TInType The parameter/input type.
 * @tparam TResult The return/result type, defaults to TInType.
 *
 *
 * @fn FunctorUpcase::operator()
 * @brief Function call operator.
 * @signature TResult FunctorUpcase::operator()(in);
 *
 * @param[in] in The value to convert (<tt>in</tt>).
 *
 * @return TResult The converted value (<tt>TResult</tt>).
 */

template <typename InType, typename Result = InType>
struct FunctorUpcase : public std::function<Result(InType)>
{
    inline Result operator()(InType x) const
    {
        return toUpperValue(x);
    }
};

// --------------------------------------------------------------------------
// Class FunctorLowcase
// --------------------------------------------------------------------------

/*!
 * @class FunctorLowcase
 * @headerfile <seqan/modifier.h>
 * @brief Functor that returns the lower case character for a given character.
 *
 * @signature template <typename TInType[, typename TResult]>
 *            struct FunctorLowcase;
 *
 * @tparam TInType The parameter/input type.
 * @tparam TResult The return/result type, defaults to TInType.
 *
 *
 * @fn FunctorLowcase::operator()
 * @brief Function call operator.
 * @signature TResult FunctorLowcase::operator()(in);
 *
 * @param[in] in The value to convert (<tt>in</tt>).
 *
 * @return TResult The converted value (<tt>TResult</tt>).
 */

template <typename InType, typename Result = InType>
struct FunctorLowcase : public std::function<Result(InType)>
{
    inline Result operator()(InType x) const
    {
        return tolower(x);
    }
};

// --------------------------------------------------------------------------
// Class FunctorConvert
// --------------------------------------------------------------------------

/*!
 * @class FunctorConvert
 * @headerfile <seqan/modifier.h>
 * @brief Functor that converts between two types.
 *
 * @signature template <typename TInType, typename TOutType>
 *            struct FunctorConvert;
 *
 * @tparam TInType  The parameter/input type.
 * @tparam TOutType The return/result type, defaults to TInType.
 *
 *
 * @fn FunctorConvert::operator()
 * @brief Function call operator.
 * @signature TOutType FunctorLowcase::operator()(in);
 *
 * @param[in] in The value to convert (<tt>in</tt>).
 *
 * @return TOutType The converted value (<tt>TOutType</tt>).
 */

template <typename InType, typename OutType>
struct FunctorConvert : public std::function<OutType(InType)>
{
    inline OutType operator()(InType x) const
    {
        return x;
    }
};

// --------------------------------------------------------------------------
// Helper Structs with Translation Tables
// --------------------------------------------------------------------------

// Manual forward for the complementing of characters.
template <typename TValue> struct FunctorComplement;


template <typename T = void>
struct TranslateTableDna5ToDna5Complement_
{
    static char const VALUE[5];
};

template <typename T>
char const TranslateTableDna5ToDna5Complement_<T>::VALUE[5] = {'T', 'G', 'C', 'A', 'N'};

template <typename T = void>
struct TranslateTableRna5ToRna5Complement_
{
    static char const VALUE[5];
};

template <typename T>
char const TranslateTableRna5ToRna5Complement_<T>::VALUE[5] = {'U', 'G', 'C', 'A', 'N'};

template <typename T = void>
struct TranslateTableIupacToIupacComplement_
{
    static signed char const VALUE[16];
};

template <typename T>
signed char const TranslateTableIupacToIupacComplement_<T>::VALUE[16] = {0, 8, 4, 12, 2, 10, 6, 14, 1, 9, 5, 13, 3, 11, 7, 15};

// --------------------------------------------------------------------------
// Class FunctorComplement
// --------------------------------------------------------------------------

/*!
 * @class FunctorComplement
 * @headerfile <seqan/modifier.h>
 * @brief Functor that returns the complement nucleotide for a given nucleotide.
 *
 * @signature template <typename TValue>
 *            struct FunctorComplement.
 *
 * @tparam TValue The value type.
 *
 * @section Remarks
 *
 * If TValue is char, then the characters are complemented as Dna5.
 *
 *
 * @fn FunctorComplement::operator()
 * @brief Function call operator.
 * @signature TValue FunctorComplement::operator()(in);
 *
 * @param[in] in The value to convert (<tt>in</tt>).
 *
 * @return TValue The converted value (<tt>TValue</tt>).
 */

template <>
struct FunctorComplement<char> : public std::function<Dna5(Dna5)>
{
    inline Dna5 operator()(Dna5 x) const
    {
        return TranslateTableDna5ToDna5Complement_<>::VALUE[x.value];
    }
};

template <>
struct FunctorComplement<Dna> : public std::function<Dna(Dna)>
{
    inline Dna operator()(Dna x) const
    {
        return TranslateTableDna5ToDna5Complement_<>::VALUE[x.value];
    }
};

template <>
struct FunctorComplement<Dna5> : public std::function<Dna5(Dna5)>
{
    inline Dna5 operator()(Dna5 x) const
    {
        return TranslateTableDna5ToDna5Complement_<>::VALUE[x.value];
    }
};

template <>
struct FunctorComplement<Rna> : public std::function<Rna(Rna)>
{
    inline Rna operator()(Rna x) const
    {
        return TranslateTableRna5ToRna5Complement_<>::VALUE[x.value];
    }
};

template <>
struct FunctorComplement<Rna5> : public std::function<Rna5(Rna5)>
{
    inline Dna5 operator()(Rna5 x) const
    {
        return TranslateTableRna5ToRna5Complement_<>::VALUE[x.value];
    }
};

template <>
struct FunctorComplement<DnaQ> : public std::function<DnaQ(DnaQ)>
{
    inline DnaQ operator()(DnaQ x) const
    {
        int qual = getQualityValue(x);
        x = TranslateTableDna5ToDna5Complement_<>::VALUE[ordValue((Dna)x)];
        assignQualityValue(x, qual);
        return x;
    }
};

template <>
struct FunctorComplement<Dna5Q> : public std::function<Dna5Q(Dna5Q)>
{
    inline Dna5Q operator()(Dna5Q x) const {
        int qual = getQualityValue(x);
        x = TranslateTableDna5ToDna5Complement_<>::VALUE[ordValue((Dna5)x)];
        assignQualityValue(x, qual);
        return x;
    }
};

template <>
struct FunctorComplement<Iupac> : public std::function<Iupac(Iupac)>
{
    inline Iupac operator()(Iupac x) const
    {
        return TranslateTableIupacToIupacComplement_<>::VALUE[x.value];
    }
};


}  // namespace seqan2

#endif  // SEQAN_MODIFIER_MODIFIER_FUNCTORS_H_
