/* file ArenaAlloc.hpp
 *
 * author: Roland Conybeare, Jul 2025
 */

#pragma once

#include "IAlloc.hpp"
#include "ObjectStatistics.hpp"

namespace xo {
    namespace gc {
        /** @class ArenaAlloc
         *  @brief Bump allocator with fixed capacity
         *
         *  @text
         *
         *    <-----allocated----> <--------free------->
         *    XXXXXXXXXXXXXXXXXXXX______________________
         *    ^                   ^                     ^
         *    lo                  free                  hi
         *                                              limit
         *  @endtext
         **/
        class ArenaAlloc : public IAlloc {
        public:
            ArenaAlloc(const ArenaAlloc &) = delete;
            ArenaAlloc(ArenaAlloc &&) = delete;
            ~ArenaAlloc();

            /** Create allocator with capacity @p z,
             *  Reserve memory addresses for @p z bytes,
             *  but don't commit them until needed
             **/
            static up<ArenaAlloc> make(const std::string & name,
                                       std::size_t z,
                                       bool debug_flag);

            /** size of virtual address range reserved for this allocator **/
            std::size_t reserved() const { return this->size(); }

            std::size_t page_size() const { return page_z_; }
            std::byte * free_ptr() const { return free_ptr_; }
            void       set_free_ptr(std::byte * x);

            /** if address @p x is allocated from this arena,
             *  return true along with offset relative to base address @ref lo_
             *  otherwise return false with 0
             **/
            std::pair<bool, std::size_t> location_of(const void * x) const;

            /** allocated span **/
            std::pair<const std::byte *, const std::byte *> allocated_span() const {
                return std::make_pair(lo_, free_ptr_);
            }

            /** Reset to empty state; provision at least @p need_z bytes of (committed) space **/
            void reset(std::size_t need_z);

            void capture_object_statistics(capture_phase phase,
                                           ObjectStatistics * p_dest) const;

            /** expand available (i.e. committed) space to size at least @p z
             *  In practice will round up to a multiple of @ref page_z_
             **/
            bool expand(std::size_t z);

            // inherited from IAlloc...

            virtual const std::string & name() const final override;
            virtual std::size_t size() const final override;
            virtual std::size_t committed() const final override;
            virtual std::size_t available() const final override;
            virtual std::size_t allocated() const final override;
            virtual bool contains(const void * x) const final override;
            virtual bool is_before_checkpoint(const void * x) const final override;
            virtual std::size_t before_checkpoint() const final override;
            virtual std::size_t after_checkpoint() const final override;
            virtual bool debug_flag() const final override;

            virtual void clear() final override;
            virtual void checkpoint() final override;
            virtual std::byte * alloc(std::size_t z) final override;

            ArenaAlloc & operator=(const ArenaAlloc &) = delete;
            ArenaAlloc & operator=(ArenaAlloc &&) = delete;

        private:
            ArenaAlloc(const std::string & name,
                       std::size_t z, bool debug_flag);

        private:
            /**
             *  Invariants:
             *  - @ref free_ always a multiple of word size (assumed to be sizeof(void*))
             **/

            /** optional instance name, for diagnostics **/
            std::string name_;

            /** size of a VM page **/
            std::size_t page_z_;

            /** allocator owns memory in range [@ref lo_, @ref hi_) **/
            std::byte * lo_ = nullptr;
            /** prefix of this size is actually committed.
             *  Remainder uses uncommitted virtual address space
             **/
            std::size_t committed_z_ = 0;
            /** checkpoint (for GC support); divides objects into
             *  older (addresses below checkpoint)
             *  and younger (addresses above checkpoint)
             **/
            std::byte * checkpoint_;
            /** free pointer. memory in range [@ref free_, @ref limit_) available **/
            std::byte * free_ptr_ = nullptr;
            /** soft limit: end of committed virtual memory **/
            std::byte * limit_ = nullptr;
            /** hard limit: end of reserved virtual memory **/
            std::byte * hi_ = nullptr;
            /** true to enable detailed debug logging **/
            bool debug_flag_ = false;
        };

    } /*namespace gc*/
} /*namespace xo*/


/* end ArenaAlloc.hpp */