#pragma once

#include <limits>
#include <algorithm>
#include <cmath>

#include "hostdevice.h"

namespace ctc_helper {

    static const float threshold = 1e-1;

    template<typename T>
    HOSTDEVICE
    T neg_inf() { return -T(INFINITY); }

    inline int div_up(int x, int y) {
        return (x + y - 1) / y;
    }

    template<typename Arg, typename Res = Arg>
    struct maximum {
        HOSTDEVICE
        Res operator()(const Arg &x, const Arg &y) const {
            return x < y ? y : x;
        }
    };

    template<typename Arg, typename Res = Arg>
    struct minimum {
        HOSTDEVICE
        Res operator()(const Arg &x, const Arg &y) const {
            return x < y ? x : y;
        }
    };

    template<typename Arg, typename Res = Arg>
    struct add {
        HOSTDEVICE
        Res operator()(const Arg &x, const Arg &y) const {
            return x + y;
        }
    };

    template<typename Arg, typename Res = Arg>
    struct identity {
        HOSTDEVICE Res operator()(const Arg &x) const {
            return Res(x);
        }
    };

    template<typename Arg, typename Res = Arg>
    struct negate {
        HOSTDEVICE Res operator()(const Arg &x) const {
            return Res(-x);
        }
    };

    template<typename Arg, typename Res = Arg>
    struct exponential {
        HOSTDEVICE Res operator()(const Arg &x) const { return std::exp(x); }
    };

    template<typename Arg1, typename Arg2 = Arg1, typename Res=Arg1>
    struct log_plus {
        typedef Res result_type;
        HOSTDEVICE
        Res operator()(const Arg1 &p1, const Arg2 &p2) {
            if (p1 == neg_inf<Arg1>())
                return p2;
            if (p2 == neg_inf<Arg2>())
                return p1;
            Res result = log1p(exp(-fabs(p1 - p2))) + maximum<Res>()(p1, p2);
            return result;
        }
    };

//template<typename Arg1, typename Arg2 = Arg1, typename Res=Arg1>
//struct log_plus {
//    HOSTDEVICE
//    Res operator()(const Arg1& p1, const Arg2& p2) {
//        Res p12_max = maximum<Res>()(p1, p2);
//        Res p12_min = minimum<Res>()(p1, p2);
//        Res p12_diff = p12_min-p12_max;
//        Res NEGATIVE_CUTOFF_VAL = -(Res)100000;
//
//        Res result = p12_diff <= NEGATIVE_CUTOFF_VAL ? maximum<Res>()(p12_max, NEGATIVE_CUTOFF_VAL)
//                                        : maximum<Res>()(p12_max + log(exp(p12_diff) + 1), NEGATIVE_CUTOFF_VAL);
//
//
//        return result;
//    }
//};

}