static_mutable_compute_cell.hpp

/*
 * live_cells_cpp
 * Copyright (C) 2024  Alexander Gutev <alex.gutev@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License.  You
 * may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied.  See the License for the specific language governing
 * permissions and limitations under the License.
 */
 
#ifndef LIVE_CELLS_STATIC_MUTABLE_COMPUTE_CELL_HPP
#define LIVE_CELLS_STATIC_MUTABLE_COMPUTE_CELL_HPP
 
#include <functional>
#include <memory>
#include <concepts>
#include <type_traits>
 
#include "keys.hpp"
#include "mutable_compute_cell_state.hpp"
#include "changes_only_state.hpp"
 
namespace live_cells {
 
    template <std::invocable F, typename R>
    class static_mutable_compute_cell_state : public mutable_compute_cell_state<std::invoke_result_t<F>> {
    protected:
        typedef std::invoke_result_t<F> value_type;
 
        typedef mutable_compute_cell_state<value_type> parent;
 
    public:
 
        template <typename T, typename U>
        static_mutable_compute_cell_state(key_ref key, T&& compute, U&& reverse, const std::unordered_set<cell> &arguments) :
            parent(key, arguments),
            compute_fn(std::forward<T>(compute)),
            reverse_fn(std::forward<U>(reverse)) {}
 
    protected:
        value_type compute() override {
            return compute_fn();
        }
 
        void reverse_compute(value_type value) override {
            reverse_fn(value);
        }
 
        void init() override {
            parent::init();
 
            for (auto arg : this->arguments) {
                arg.add_observer(this->observer_ptr());
            }
        }
 
    private:
        const F compute_fn;
 
        const R reverse_fn;
 
    };
 
 
    template <std::invocable F, typename R>
    class static_mutable_compute_changes_only_cell_state :
        public changes_only_cell_state<static_mutable_compute_cell_state<F,R>> {
 
        typedef changes_only_cell_state<static_mutable_compute_cell_state<F,R>> parent;
 
    public:
        using parent::parent;
    };
 
    template <std::invocable F, typename R, typename State = static_mutable_compute_cell_state<F,R>>
    class static_mutable_compute_cell : public stateful_cell<State> {
 
        typedef stateful_cell<State> parent;
 
    public:
        typedef std::invoke_result_t<F> value_type;
 
        template <typename T, typename U, typename... Args>
        static_mutable_compute_cell(key_ref k, T&& compute, U&& reverse, Args&&... args) :
            parent(k,
                   std::forward<T>(compute),
                   std::forward<U>(reverse),
                   std::unordered_set<cell>({cell(args)...})) {}
 
        template <typename T, typename U, typename... Args>
        static_mutable_compute_cell(T&& compute, U&& reverse, Args&&... args) :
            parent(key_ref::create<unique_key>(), std::forward<T>(compute), std::forward<U>(reverse), std::unordered_set<cell>({cell(args)...})) {}
 
        value_type value() const {
            return this->state->value();
        }
 
        void value(value_type value) const {
            this->state->value(value);
        }
 
        value_type operator=(const value_type &value) {
            this->value(value);
            return value;
        }
 
        value_type operator=(const value_type &value) const {
            this->value(value);
            return value;
        }
 
        value_type operator()() const {
            argument_tracker::global().track_argument(*this);
            return value();
        }
    };
 
    template <std::invocable C, typename R, Cell... As>
    auto make_mutable_compute_cell(C compute, R reverse, As... args) {
        return static_mutable_compute_cell<C,R>(compute, reverse, args...);
    }
 
    template <std::invocable C, typename R, Cell... As>
    auto make_mutable_compute_cell(changes_only_option option, C compute, R reverse, As... args) {
        typedef static_mutable_compute_changes_only_cell_state<C,R> Base;
 
        return static_mutable_compute_cell<C,R,Base>(compute, reverse, args...);
    }
 
    template <std::invocable C, typename R, Cell... As>
    auto make_mutable_compute_cell(key_ref key, C compute, R reverse, As... args) {
        return static_mutable_compute_cell<C,R>(key, compute, reverse, args...);
    }
 
    template <std::invocable C, typename R, Cell... As>
    auto make_mutable_compute_cell(changes_only_option option, key_ref key, C compute, R reverse, As... args) {
        typedef static_mutable_compute_changes_only_cell_state<C,R> Base;
 
        return static_mutable_compute_cell<C,R,Base>(key, compute, reverse, args...);
    }
 
}  // live_cells
 
#endif /* LIVE_CELLS_STATIC_MUTABLE_COMPUTE_CELL_HPP */