complex_vari.hpp

Go to the documentation of this file.

#ifndef XPED_COMPLEX_VARI_HPP_
#define XPED_COMPLEX_VARI_HPP_
 
#include "stan/math/rev/core/vari.hpp"
 
namespace stan::math {
 
template <>
class vari_value<std::complex<double>, void> : public vari_base
{
public:
    using value_type = std::complex<double>;
    const value_type val_;
    value_type adj_{0.0, 0.0};
 
    template <typename S, require_convertible_t<S&, std::complex<double>>* = nullptr>
    vari_value(S x) noexcept
        : val_(x)
    { // NOLINT
        ChainableStack::instance_->var_stack_.push_back(this);
    }
 
    template <typename S, require_convertible_t<S&, std::complex<double>>* = nullptr>
    vari_value(S x, bool stacked) noexcept
        : val_(x)
    {
        if(stacked) {
            ChainableStack::instance_->var_stack_.push_back(this);
        } else {
            ChainableStack::instance_->var_nochain_stack_.push_back(this);
        }
    }
 
    inline const auto& val() const { return val_; }
 
    inline auto& adj() const { return adj_; }
 
    inline auto& adj() { return adj_; }
 
    inline void chain() {}
 
    inline void init_dependent() noexcept { adj_ = {1.0, 0.0}; }
 
    inline void set_zero_adjoint() noexcept final { adj_ = {0.0, 0.0}; }
 
    friend std::ostream& operator<<(std::ostream& os, const vari_value<value_type>* v) { return os << v->val_ << ":" << v->adj_; }
 
private:
    template <typename, typename>
    friend class var_value;
};
 
} // namespace stan::math
 
#endif