#ifndef FUNCTIONVALUE_H
#define FUNCTIONVALUE_H

#include <math.h>
#include "JSystem/JGadget/std-vector.h"
#include "JSystem/JGadget/define.h"
#include "JSystem/JGadget/search.h"

namespace JStudio {

typedef f64 TValue;

typedef f64 (*ExtrapolateParameter)(f64, f64);

class TFunctionValue;
class TFunctionValueAttributeSet;

class TFunctionValueAttribute_refer;
class TFunctionValueAttribute_range;
class TFunctionValueAttribute_interpolate;

class TFunctionValue {
public:
    enum TEProgress { PROG_INIT };
    enum TEAdjust { ADJ_INIT, ADJ_UNK1, ADJ_UNK2, ADJ_UNK3, ADJ_UNK4 };
    enum TEOutside { OUT_INIT };
    enum TEInterpolate {};

    TFunctionValue();
    virtual ~TFunctionValue() = 0;

    virtual u32 getType() const = 0;
    virtual TFunctionValueAttributeSet getAttributeSet() = 0;
    virtual void initialize() = 0;
    virtual void prepare() = 0;
    virtual TValue getValue(f64 arg1) = 0;

    static ExtrapolateParameter toFunction_outside(int);

    static ExtrapolateParameter toFunction(TFunctionValue::TEOutside outside) {
        return toFunction_outside(outside);
    }
};

class TFunctionValueAttributeSet_const {
public:
    TFunctionValueAttributeSet_const(TFunctionValueAttribute_refer* refer,
                                     TFunctionValueAttribute_range* range,
                                     TFunctionValueAttribute_interpolate* interp)
        : refer_(refer), range_(range), interp_(interp) {}

    TFunctionValueAttribute_refer* refer_get() const { return refer_; }
    TFunctionValueAttribute_range* range_get() const { return range_; }
    TFunctionValueAttribute_interpolate* interpolate_get() const { return interp_; }

private:
    /* 0x00 */ TFunctionValueAttribute_refer* refer_;
    /* 0x04 */ TFunctionValueAttribute_range* range_;
    /* 0x08 */ TFunctionValueAttribute_interpolate* interp_;
};

class TFunctionValueAttributeSet : public TFunctionValueAttributeSet_const {
public:
    TFunctionValueAttributeSet(TFunctionValueAttribute_refer* refer,
                               TFunctionValueAttribute_range* range,
                               TFunctionValueAttribute_interpolate* interp)
        : TFunctionValueAttributeSet_const(refer, range, interp) {}

    TFunctionValueAttribute_refer* refer_get() const {
        return TFunctionValueAttributeSet_const::refer_get();
    }
    TFunctionValueAttribute_range* range_get() const {
        return TFunctionValueAttributeSet_const::range_get();
    }
    TFunctionValueAttribute_interpolate* interpolate_get() const {
        return TFunctionValueAttributeSet_const::interpolate_get();
    }
};

class TFunctionValueAttribute_refer : public JGadget::TVector_pointer<TFunctionValue*> {
public:
    TFunctionValueAttribute_refer() : JGadget::TVector_pointer<TFunctionValue*>(JGadget::TAllocator<void*>()) {}
    ~TFunctionValueAttribute_refer() {}

    void refer_initialize();

    const TFunctionValueAttribute_refer* refer_getContainer() const { return this; }
    JGadget::TVector_pointer<TFunctionValue*>& refer_referContainer() { return *this; }
    bool refer_isReferring(const TFunctionValue* p) const { return false; }  // todo
};

class TFunctionValueAttribute_range {
public:
    TFunctionValueAttribute_range();

    void range_initialize();
    void range_prepare();
    void range_set(f64, f64);
    f64 range_getParameter(f64, f64, f64) const;

    TFunctionValue::TEProgress range_getProgress() const {
        return (TFunctionValue::TEProgress)mProgress;
    }
    void range_setProgress(TFunctionValue::TEProgress progress) { mProgress = progress; }
    TFunctionValue::TEAdjust range_getAdjust() const { return (TFunctionValue::TEAdjust)mAdjust; }
    void range_setAdjust(TFunctionValue::TEAdjust adjust) { mAdjust = adjust; }
    void range_setOutside(TFunctionValue::TEOutside outside) { range_setOutside(outside, outside); }
    void range_setOutside(TFunctionValue::TEOutside begin, TFunctionValue::TEOutside end) {
        range_setOutside_begin(begin);
        range_setOutside_end(end);
    }
    void range_setOutside_begin(TFunctionValue::TEOutside begin) { mBegin = begin; }
    void range_setOutside_end(TFunctionValue::TEOutside end) { mEnd = end; }
    f64 range_getParameter_outside(f64 arg1) const {
        f64 result = arg1;
        result -= fBegin_;
        if (result < 0.0) {
            result = TFunctionValue::toFunction(mBegin)(result, fDifference_);
        } else if (result >= fDifference_) {
            result = TFunctionValue::toFunction(mEnd)(result, fDifference_);
        }
        result += fBegin_;
        return result;
    }
    f64 range_getParameter_progress(f64 arg1) const { return _20 + _28 * (arg1 - _20); }
    f64 range_getBegin() const { return fBegin_;}
    f64 range_getEnd() const { return fEnd_;}
    f64 range_getDifference() const { return fDifference_; }

private:
    /* 0x00 */ f64 fBegin_;
    /* 0x08 */ f64 fEnd_;
    /* 0x10 */ f64 fDifference_;
    /* 0x18 */ s8 mProgress;
    /* 0x19 */ s8 mAdjust;
    /* 0x1A */ s8 _1a[2];
    /* 0x1C */ u32 _1c;
    /* 0x20 */ f64 _20;
    /* 0x28 */ f64 _28;
    /* 0x30 */ TFunctionValue::TEOutside mBegin;
    /* 0x34 */ TFunctionValue::TEOutside mEnd;
};

class TFunctionValueAttribute_interpolate {
public:
    TFunctionValueAttribute_interpolate() : interpolate_(0) {}

    void interpolate_initialize() { interpolate_ = 0; }
    void interpolate_prepare() {}
    u32 interpolate_get() const { return interpolate_; }
    void interpolate_set(TFunctionValue::TEInterpolate interpolate) { interpolate_ = interpolate; }

private:
    /* 0x0 */ u32 interpolate_;
};

class TFunctionValue_constant : public TFunctionValue {
public:
    TFunctionValue_constant();

    virtual u32 getType() const;
    virtual TFunctionValueAttributeSet getAttributeSet();
    virtual void initialize();
    virtual void prepare();
    virtual f64 getValue(f64);

    void data_set(f64 value) { fValue_ = value; }

private:
    f64 fValue_;
};

class TFunctionValue_composite : public TFunctionValue, public TFunctionValueAttribute_refer {
public:
    struct TData {
        TData(void* data) : u32data((u32)data) {}
        TData(const void* data) : rawData(data) {}
        TData(u32 data) : u32data(data) {}
        TData(f32 data) : f32data(data) {}

        inline void operator=(const TData& rhs) { f32data = rhs.f32data; }
        u32 get_unsignedInteger() const { return u32data; }
        u32 get_outside() const { return u32data; }
        f64 get_value() const { return f32data; }

        union {
            const void* rawData;
            u32 u32data;
            f64 f32data;
        };
    };
    typedef f64 (*UnkFunc)(f64, const TFunctionValueAttribute_refer*,
                           const TFunctionValue_composite::TData*);
    typedef f64 (*CompositeFunc)(const JGadget::TVector_pointer<TFunctionValue*>&,
                                 const TFunctionValue_composite::TData&, f64);

    TFunctionValue_composite();

    virtual u32 getType() const;
    virtual TFunctionValueAttributeSet getAttributeSet();
    virtual void initialize();
    virtual void prepare();
    virtual f64 getValue(f64);
    static f64 composite_raw(TVector_pointer<TFunctionValue*> const&, TData const&, f64);
    static f64 composite_index(TVector_pointer<TFunctionValue*> const&, TData const&, f64);
    static f64 composite_parameter(TVector_pointer<TFunctionValue*> const&,
                                                  TData const&, f64);
    static f64 composite_add(TVector_pointer<JStudio::TFunctionValue*> const&,
                                            TData const&, f64);
    static f64 composite_subtract(TVector_pointer<TFunctionValue*> const&, TData const&,
                                           f64);
    static f64 composite_multiply(TVector_pointer<TFunctionValue*> const&, TData const&,
                                           f64);
    static f64 composite_divide(TVector_pointer<TFunctionValue*> const&, TData const&,
                                         f64);

    void data_set(CompositeFunc fn, const TData& dat) {
        pfn_ = (UnkFunc)fn;
        data_setData(dat);
    }
    const TData* data_getData() const { return &data; }
    void data_setData(const TData& dat) { data = dat; }

// private:
    UnkFunc pfn_;
    TData data;
};

class TFunctionValue_transition : public TFunctionValue,
                                  public TFunctionValueAttribute_range,
                                  public TFunctionValueAttribute_interpolate {
public:
    TFunctionValue_transition();

    virtual u32 getType() const;
    virtual TFunctionValueAttributeSet getAttributeSet();
    virtual void initialize();
    virtual void prepare();
    virtual f64 getValue(f64);

    void data_set(f64 a1, f64 a2) {
        _48 = a1;
        _50 = a2;
    }

    f64 data_getDifference() const { return _50 - _48; }

private:
    /* 0x48 */ f64 _48;
    /* 0x50 */ f64 _50;
};

class TFunctionValue_list : public TFunctionValue,
                            public TFunctionValueAttribute_range,
                            public TFunctionValueAttribute_interpolate {
public:
    struct TIndexData_ {
        f64 _0;
        f64 _8;
        u32 _10;
    };
    typedef f64 (*update_INTERPOLATE)(const TFunctionValue_list&, const TIndexData_&);

    TFunctionValue_list();

    virtual u32 getType() const;
    virtual TFunctionValueAttributeSet getAttributeSet();
    virtual void initialize();
    virtual void prepare();
    virtual f64 getValue(f64);

    void data_set(const f32* pf, u32 u) {
        ASSERT((pf != NULL) || (u == 0));
        _44 = pf;
        uData_ = u;
    }

    void data_setInterval(f64 f) {
        // ASSERT(f > TValue(0));
        _50 = f;
    }

    static f64
    update_INTERPOLATE_NONE_(JStudio::TFunctionValue_list const&,
                             JStudio::TFunctionValue_list::TIndexData_ const&);
    static f64
    update_INTERPOLATE_LINEAR_(JStudio::TFunctionValue_list const&,
                               JStudio::TFunctionValue_list::TIndexData_ const&);
    static f64
    update_INTERPOLATE_PLATEAU_(JStudio::TFunctionValue_list const&,
                                JStudio::TFunctionValue_list::TIndexData_ const&);
    static f64
    update_INTERPOLATE_BSPLINE_dataMore3_(JStudio::TFunctionValue_list const&,
                                          JStudio::TFunctionValue_list::TIndexData_ const&);

private:
    /* 0x44 */ const f32* _44;
    /* 0x48 */ u32 uData_;
    /* 0x50 */ f64 _50;
    /* 0x58 */ update_INTERPOLATE pfnUpdate_;
};

class TFunctionValue_list_parameter : public TFunctionValue,
                                      public TFunctionValueAttribute_range,
                                      public TFunctionValueAttribute_interpolate {
public:
    struct TIterator_data_
        : public JGadget::TIterator<
            std::random_access_iterator_tag,
            const f32,
            ptrdiff_t,
            const f32*,
            const f32&
        >
    {
        TIterator_data_(const TFunctionValue_list_parameter& rParent, const f32* value) {
#if DEBUG
            pOwn_ = &rParent;
#endif
            pf_ = value;
        }

        const f32* get() const { return pf_; }
        void set(const f32* value) { pf_ = value; }

        friend bool operator==(const TIterator_data_& r1, const TIterator_data_& r2) {
#if DEBUG
            if (!(r1.pOwn_==r2.pOwn_)) {
                JGadget_outMessage msg(JGadget_outMessage::warning, __FILE__, 124);
                msg << "r1.pOwn_==r2.pOwn_";
            }
#endif
            return r1.pf_ == r2.pf_;
        }

        f32 operator*() {
            // this guard is required - removing it breaks float regalloc in std::upper_bound
            #if DEBUG
            JUT_ASSERT(947, pf_!=NULL);
            #endif
            return *pf_;
        }

        TIterator_data_& operator+=(s32 n) {
            pf_ += suData_size * n;
            return *this;
        }
        TIterator_data_& operator-=(s32 n) {
            pf_ -= suData_size * n;
            return *this;
        }
        TIterator_data_& operator++() {
            pf_ += suData_size;
            return *this;
        }
        TIterator_data_& operator--() {
            pf_ -= suData_size;
            return *this;
        }

        friend s32 operator-(const TIterator_data_& r1, const TIterator_data_& r2) {
#if DEBUG
            if (!(r1.pOwn_==r2.pOwn_)) {
                JGadget_outMessage msg(JGadget_outMessage::warning, __FILE__, 124);
                msg << "r1.pOwn_==r2.pOwn_";
            }
#endif
            return (r1.pf_ - r2.pf_) / suData_size;
        }

#if DEBUG
        /* 0x00 */ const TFunctionValue_list_parameter* pOwn_;
#endif
        /* 0x00 */ const f32* pf_;
    };
    typedef f64 (*update_INTERPOLATE)(const TFunctionValue_list_parameter&, f64);

    TFunctionValue_list_parameter();

    virtual u32 getType() const;
    virtual TFunctionValueAttributeSet getAttributeSet();
    void data_set(f32 const*, u32);
    virtual void initialize();
    virtual void prepare();
    virtual f64 getValue(f64);

    static f64
    update_INTERPOLATE_NONE_(JStudio::TFunctionValue_list_parameter const&, f64);
    static f64
    update_INTERPOLATE_LINEAR_(JStudio::TFunctionValue_list_parameter const&, f64);
    static f64
    update_INTERPOLATE_PLATEAU_(JStudio::TFunctionValue_list_parameter const&, f64);
    static f64
    update_INTERPOLATE_BSPLINE_dataMore3_(JStudio::TFunctionValue_list_parameter const&, f64);

    static const u32 suData_size = 2;

    f64 data_getValue_back() const { 
        return pfData_[(uData_ - 1) * suData_size];
    }
    f64 data_getValue_front() const { return pfData_[0]; }

private:
    /* 0x44 */ const f32* pfData_;
    /* 0x48 */ u32 uData_;
    /* 0x4c */ TIterator_data_ dat1;
    /* 0x50 */ TIterator_data_ dat2;
    /* 0x54 */ TIterator_data_ dat3;
    /* 0x58 */ update_INTERPOLATE pfnUpdate_;
};

class TFunctionValue_hermite : public TFunctionValue, public TFunctionValueAttribute_range {
public:
    struct TIterator_data_
        : public JGadget::TIterator<
            std::random_access_iterator_tag,
            const f32,
            ptrdiff_t,
            const f32*,
            const f32&
        >
    {
        TIterator_data_(const TFunctionValue_hermite& rParent, const f32* value) {
#if DEBUG
            pOwn_ = &rParent;
#endif
            pf_ = value;
            uSize_ = rParent.data_getSize();
        }

        const f32* get() const { return pf_; }
        void set(const f32* value, u32 size) {
            pf_ = value;
            uSize_ = size;
        }

        friend bool operator==(const TIterator_data_& r1, const TIterator_data_& r2) {
#if DEBUG
            if (!(r1.pOwn_==r2.pOwn_)) {
                JGadget_outMessage msg(JGadget_outMessage::warning, __FILE__, 124);
                msg << "r1.pOwn_==r2.pOwn_";
            }
#endif
            return r1.pf_ == r2.pf_;
        }

        f32 operator*() {
#if DEBUG
            JUT_ASSERT(1098, pf_!=NULL);
#endif
            return *pf_;
        }

        TIterator_data_& operator+=(s32 n) {
            pf_ += uSize_ * n;
            return *this;
        }
        TIterator_data_& operator-=(s32 n) {
            pf_ -= uSize_ * n;
            return *this;
        }
        TIterator_data_& operator++() {
            pf_ += uSize_;
            return *this;
        }
        TIterator_data_& operator--() {
            pf_ -= uSize_;
            return *this;
        }

        friend s32 operator-(const TIterator_data_& r1, const TIterator_data_& r2) {
#if DEBUG
            if (!(r1.pOwn_==r2.pOwn_)) {
                JGadget_outMessage msg(JGadget_outMessage::warning, __FILE__, 124);
                msg << "r1.pOwn_==r2.pOwn_";
            }
            if (!(r1.uSize_==r2.uSize_)) {
                JGadget_outMessage msg(JGadget_outMessage::warning, __FILE__, 124);
                msg << "r1.uSize_==r2.uSize_";
            }
            JUT_ASSERT(0, r1.uSize_>0);
#endif
            return (r1.pf_ - r2.pf_) / r1.uSize_;
        }

#if DEBUG
        /* 0x00 */ const TFunctionValue_hermite* pOwn_;
        /* 0x04 */ const f32* pf_;
        /* 0x08 */ u32 uSize_;
#else
        /* 0x00 */ const f32* pf_;
        /* 0x04 */ u32 uSize_;
#endif
    };

    TFunctionValue_hermite();

    virtual u32 getType() const;
    virtual TFunctionValueAttributeSet getAttributeSet();
    void data_set(f32 const*, u32, u32);
    virtual void initialize();
    virtual void prepare();
    virtual f64 getValue(f64);

    u32 data_getSize() const { return uSize_; }
    f64 data_getValue_back() const { 
        return pfData_[(u_ - 1) * uSize_];
    }
    f64 data_getValue_front() const { return pfData_[0]; }

private:
    /* 0x40 */ const f32* pfData_;
    /* 0x44 */ u32 u_;
    /* 0x48 */ u32 uSize_;
    /* 0x4c */ TIterator_data_ dat1;
    /* 0x50 */ TIterator_data_ dat2;
    /* 0x54 */ TIterator_data_ dat3;
};

namespace functionvalue {

inline f64 extrapolateParameter_raw(f64 a1, f64 a2) {
    return a1;
}

inline f64 extrapolateParameter_repeat(f64 a1, f64 a2) {
    f64 t = fmod(a1, a2);

    if (t < 0.0)
        t += a2;

    return t;
}

f64 extrapolateParameter_turn(f64, f64);

inline f64 extrapolateParameter_clamp(f64 value, f64 max) {
    if (value <= 0.0)
        return 0.0;

    if (max <= value)
        return max;

    return value;
}

};  // namespace functionvalue

}  // namespace JStudio

#endif /* FUNCTIONVALUE_H */