#ifndef LFORTRAN_LLVM_ARR_UTILS_H
#define LFORTRAN_LLVM_ARR_UTILS_H

#include <map>
#include <memory>
#include <string>
#include <unordered_map>

#include <llvm/IR/Type.h>
#include <llvm/IR/Value.h>
#include <llvm/IR/DerivedTypes.h>
#include <llvm/IR/IRBuilder.h>

#include <libasr/alloc.h>
#include <libasr/asr.h>
#include <libasr/codegen/llvm_utils.h>

namespace LCompilers {

    // Forward declared
    class ASRToLLVMVisitor;

    namespace LLVMArrUtils {

        llvm::Value* lfortran_malloc(llvm::LLVMContext &context, llvm::Module &module,
                llvm::IRBuilder<> &builder, llvm::Value* arg_size);
        llvm::Value* lfortran_realloc(llvm::LLVMContext &context, llvm::Module &module,
                llvm::IRBuilder<> &builder, llvm::Value* ptr, llvm::Value* arg_size);

        /*
        * This function checks whether the
        * dimensions are available at compile time.
        * Returns true if all the dimensions reduce
        * to constant integers and false otherwise.
        */
        bool compile_time_dimensions_t(
        ASR::dimension_t* m_dims, int n_dims);

        /*
        * This function checks if the given
        * an variable is an array and all the
        * dimensions are available at compile time.
        */
        bool is_explicit_shape(ASR::Variable_t* v);

        /*
        * Available descriptors are listed
        * under this enum.
        */
        enum DESCR_TYPE
        {
            _SimpleCMODescriptor
        };

        /*
        * Abstract class which defines the interface
        * to be followed by any subclass intending
        * to implement a specific array descriptor.
        */
        class Descriptor {

            public:

                virtual ~Descriptor() {}

                /*
                * Factory method which creates
                * new descriptors and returns a
                * pointer to it. It accepts one of
                * the members DESCR_TYPE enum
                * to create a new descriptor.
                */
                static
                std::unique_ptr<Descriptor>
                get_descriptor(
                    llvm::LLVMContext& context,
                    llvm::IRBuilder<>* builder,
                    LLVMUtils* llvm_utils,
                    DESCR_TYPE descr_type,
                    CompilerOptions& co_);

                /*
                * Checks whether the given ASR::ttype_t* is an
                * array and follows the same structure as
                * the current descriptor.
                */
                virtual
                bool is_array(ASR::ttype_t* asr_type) = 0;

                /*
                * Converts a given array llvm::Value*
                * into an argument of the specified type.
                */
                virtual
                llvm::Value* convert_to_argument(llvm::Value* tmp,
                    ASR::ttype_t* asr_arg_type, llvm::Type* arg_type,
                    bool data_only=false) = 0;

                /*
                * Returns the type of the argument to be
                * used in LLVM functions for passing an array
                * following the current descriptor structure.
                */
                virtual
                llvm::Type* get_argument_type(llvm::Type* type,
                    std::uint32_t, std::string,
                    std::unordered_map
                    <std::uint32_t, std::unordered_map
                                    <std::string, llvm::Type*>>&
                                    arr_arg_type_cache) = 0;

                /*
                * Creates an array llvm::Type* following
                * the same structure as the current descriptor.
                * Uses element type, kind, rank and dimensions
                * to create the array llvm::Type*.
                */
                virtual
                llvm::Type* get_array_type(
                    ASR::expr_t* expr,
                    ASR::ttype_t* m_type_,
                    llvm::Type* el_type,
                    bool get_pointer=false) = 0;

                /*
                * Fills the elements of the input array descriptor
                * for arrays on stack memory.
                */
                virtual
                void fill_array_details(
                    llvm::Type* arr_ty, llvm::Value* arr, llvm::Type* llvm_data_type, int n_dims,
                    std::vector<std::pair<llvm::Value*, llvm::Value*>>& llvm_dims,
                    llvm::Module* module, bool reserve_data_memory=true) = 0;

                virtual
                void fill_array_details(ASR::expr_t* src_expr, ASR::expr_t* dest_expr,
                    llvm::Value* source, llvm::Value* destination,
                    ASR::ttype_t* source_array_type, ASR::ttype_t* dest_array_type, llvm::Module* module, bool ignore_data) = 0;

                /*
                * Fills the elements of the input array descriptor
                * for allocatable arrays.
                */
                virtual
                void fill_malloc_array_details(
                    llvm::Value* arr, llvm::Type *arr_type, llvm::Type* llvm_data_type, ASR::ttype_t* asr_type, int n_dims,
                    std::vector<std::pair<llvm::Value*, llvm::Value*>>& llvm_dims, llvm::Value* string_len,
                    ASR::symbol_t* variable_declaration, llvm::Module* module, 
                    ASR::symbol_t* allocated_subclass=nullptr, bool realloc=false,
                    ASR::ttype_t* alloc_type=nullptr) = 0;

                virtual
                void fill_dimension_descriptor(llvm::Type* type,llvm::Value* arr, int n_dims) = 0;

                virtual
                void reset_array_details(
                    llvm::Type* type, llvm::Value* arr, llvm::Type* source_arr_type, llvm::Value* source_arr, int n_dims) = 0;

                virtual
                void reset_array_details(
                    llvm::Type* type, llvm::Value* arr, llvm::Value* source_arr,
                    llvm::Value** lbs, llvm::Value** lengths,
                    int n_dims) = 0;

                virtual
                void fill_descriptor_for_array_section(
                    llvm::Value* value_desc, llvm::Type* value_el_type, ASR::ttype_t* value_type,
                    llvm::Type* value_desc_type,
                    llvm::Value* target, ASR::ttype_t* target_type, ASR::expr_t* /*target_expr*/,
                    llvm::Type* target_desc_type,
                    llvm::Value** lbs, llvm::Value** ubs,
                    llvm::Value** ds, llvm::Value** non_sliced_indices,
                    int value_rank, int target_rank, LocationManager& lm) = 0;

                virtual
                void fill_descriptor_for_array_section_data_only(
                    llvm::Value* value_desc, llvm::Type* value_el_type, ASR::ttype_t* value_type,
                    llvm::Value* target, ASR::ttype_t* target_type, ASR::expr_t* /*target_expr*/,
                    llvm::Type* target_desc_type,
                    llvm::Value** lbs, llvm::Value** ubs,
                    llvm::Value** ds, llvm::Value** non_sliced_indices,
                    llvm::Value** llvm_diminfo, int value_rank, int target_rank, LocationManager& lm) = 0;

                /*
                * Allocates a descriptor on the heap, zero-initializes it,
                * sets up the dimension descriptor array pointer and rank.
                * Returns a pointer to the descriptor.
                */
                virtual
                llvm::Value* allocate_descriptor_on_heap(llvm::Type* array_desc_type) = 0;

                virtual
                llvm::Value* create_descriptor_alloca(llvm::Type* array_desc_type,
                    const std::string& name = "arr_desc") = 0;

                /*
                * Returns the llvm::Type* associated with the
                * dimension descriptor used by the current class.
                */
                virtual
                llvm::Type* get_dimension_descriptor_type(bool get_pointer=false) = 0;

                /*
                * Returns the llvm::Type* used for array indices (i32 or i64).
                */
                virtual
                llvm::Type* get_index_type() const = 0;

                /*
                * Returns pointer to data in the input
                * array descriptor according to the rules
                * implemented by current class.
                */
                virtual
                llvm::Value* get_pointer_to_data(llvm::Type* type, llvm::Value* arr) = 0;

                /*
                * Returns pointer to data in the input
                * array descriptor according to the rules
                * implemented by current class.
                * Uses ASR type to get the corresponding LLVM type
                */
                virtual
                llvm::Value* get_pointer_to_data(ASR::expr_t* arr_expr, ASR::ttype_t* arr_type, llvm::Value* arr, llvm::Module* module) = 0;

                /*
                * Returns offset in the input
                * array descriptor according to the rules
                * implemented by current class).
                */
                virtual
                llvm::Value* get_offset(llvm::Type* type, llvm::Value* dim_des, bool load=true) = 0;

                /*
                * Returns lower bound in the input
                * dimension descriptor according to the rules
                * implemented by current class).
                */
                virtual
                llvm::Value* get_lower_bound(llvm::Value* dim_des, bool load=true) = 0;

                /*
                * Returns upper bound in the input
                * dimension descriptor according to the rules
                * implemented by current class.
                */
                virtual
                llvm::Value* get_upper_bound(llvm::Value* dim_des) = 0;

                /*
                * Returns stride in the input
                * dimension descriptor according to the rules
                * implemented by current class.
                */
                virtual
                llvm::Value* get_stride(llvm::Value* dim_des, bool load=true) = 0;

                /*
                * Returns dimension size in the input
                * dimension descriptor according to the rules
                * implemented by current class.
                */
                virtual
                llvm::Value* get_dimension_size(llvm::Value* dim_des_arr,
                    llvm::Value* dim, bool load=true) = 0;

                virtual
                llvm::Value* get_dimension_size(llvm::Value* dim_des_arr,
                    bool load=true) = 0;

                virtual
                llvm::Value* get_rank(llvm::Type* type, llvm::Value* arr, bool get_pointer=false) = 0;

                virtual
                void set_rank(llvm::Type* type, llvm::Value* arr, llvm::Value* rank) = 0;

                /*
                * Returns pointer to dimension descriptor array
                * in the input array descriptor according to the rules
                * implemented by current class.
                */
                virtual
                llvm::Value* get_pointer_to_dimension_descriptor_array(llvm::Type *type, llvm::Value* arr, bool load=true) = 0;


                /*
                * Returns pointer to the dimension descriptor
                * in the input dimension descriptor array according
                * to the rules implemented by current class.
                */
                virtual
                llvm::Value* get_pointer_to_dimension_descriptor(llvm::Value* dim_des_arr,
                    llvm::Value* dim) = 0;

                /*
                * Returns the indexed element
                * in the input dimension descriptor array according
                * to the rules implemented by current class.
                */
                virtual
                llvm::Value* get_single_element(llvm::Type *type, llvm::Value* array,
                    std::vector<llvm::Value*>& m_args, int n_args,
                    ASR::ttype_t* asr_type, ASR::expr_t* expr, LocationManager& lm,
                    ASR::symbol_t* variable_type_decl,
                    bool data_only=false, bool is_fixed_size=false,
                    llvm::Value** llvm_diminfo=nullptr,
                    bool polymorphic=false, llvm::Type* polymorphic_type=nullptr,
                    bool is_unbounded_pointer_to_data = false, bool check_for_bounds = false, std::string array_name = "", std::string infile = "") = 0;

                virtual
                llvm::Value* get_is_allocated_flag(llvm::Value* array, ASR::expr_t* array_exp) = 0;

                virtual
                void reset_is_allocated_flag(llvm::Type* tmp_typ, llvm::Value* array, llvm::Type* llvm_data_type) = 0;


                virtual
                llvm::Value* reshape(llvm::Type* arr_type, llvm::Value* array, llvm::Type* llvm_data_type,
                                     llvm::Type* shape_type, llvm::Value* shape, ASR::ttype_t* asr_shape_type,
                                     llvm::Module* module, ASR::expr_t* array_expr = nullptr,
                                     ASR::ttype_t* asr_data_type = nullptr,
                                     llvm::Type* result_desc_type = nullptr,
                                     llvm::Value* order = nullptr,
                                     ASR::expr_t* order_expr = nullptr) = 0;

                virtual
                void copy_array(llvm::Type* src_ty, llvm::Value* src, llvm::Type* dest_ty, llvm::Value* dest,
                                llvm::Module* module, ASR::expr_t* array_expr, ASR::ttype_t* asr_data_type,
                                bool reserve_memory) = 0;

                virtual
                void copy_array_move_allocation(llvm::Type* src_ty, llvm::Value* src, llvm::Type* dest_ty, llvm::Value* dest,
                                llvm::Module* module, ASR::expr_t* array_exp, ASR::ttype_t* asr_data_type,
                                bool reset_lower_bound) = 0;

                virtual
                void copy_array_data_only(llvm::Value* src, llvm::Value* dest,
                                          llvm::Module* module, llvm::Type* llvm_data_type,
                                          ASR::ttype_t* arr_type,
                                          llvm::Value* num_elements) = 0;

                virtual
                llvm::Value* get_array_size(llvm::Type* type, llvm::Value* array, llvm::Value* dim,
                                            int output_kind, int dim_kind=4) = 0;

                /*
                * Creates a contiguous copy of data from a DescriptorArray.
                * Helpful when having strided access to an array.
                * Returns a pointer to the newly allocated contiguous buffer.
                * Allocation is done on heap using lfortran_malloc,
                * So, please make sure to free the memory after use.
                *
                * Note: Similar logic exists in ASR passes (pass/print_arr.cpp & pass/array_op.cpp).
                * But, the ASR pass for print statements was disabled in #6351.
                * Enabling it opened other issues, hence this separate LLVM implementation
                * to handle strided arrays for writing array into strings (used in print statements), 
                * by storing in a contiguous buffer first.
                */
                virtual
                llvm::Value* create_contiguous_copy_from_descriptor(
                    llvm::Type* source_llvm_type, llvm::Value* source_desc,
                    llvm::Type* elem_type, int rank, llvm::Module* module) = 0;

                /*
                * Copies contiguous data into a potentially-strided descriptor array.
                * The inverse of create_contiguous_copy_from_descriptor.
                * Uses element-wise copy respecting the target descriptor's strides.
                */
                virtual
                void copy_contiguous_data_to_descriptor(
                    llvm::Value* source_data,
                    llvm::Type* dest_llvm_type, llvm::Value* dest_desc,
                    llvm::Type* elem_type, int rank, llvm::Module* module) = 0;

                // CFI interop: convert internal descriptor to CFI layout
                virtual
                llvm::StructType* get_cfi_type(llvm::Type* el_type, int n_dims) = 0;

                virtual
                llvm::Value* internal_to_cfi(
                    llvm::Type* internal_type, llvm::Value* internal_desc,
                    llvm::Type* el_type, int n_dims, uint64_t elem_size,
                    int8_t type_code, int cfi_attr) = 0;

                virtual
                llvm::Value* cfi_to_internal(
                    llvm::Type* internal_type,
                    llvm::Type* el_type, llvm::Value* cfi_desc,
                    int n_dims, uint64_t elem_size) = 0;

                virtual
                void push_descriptor_array_args(
                    ASR::expr_t* val_expr, ASR::ttype_t* expr_type_full,
                    ASR::ttype_t* val_type, llvm::Value* var_ptr,
                    llvm::Module* module, std::vector<llvm::Value*>& args) = 0;

        };

        class SimpleCMODescriptor: public Descriptor {

            public:

                // CFI_dim_t sub-field indices (standard order)
                static constexpr int DIM_LOWER_BOUND = 0;
                static constexpr int DIM_EXTENT      = 1;
                static constexpr int DIM_STRIDE      = 2;  // sm (stride multiplier)

            private:

                llvm::LLVMContext& context;
                LLVMUtils* llvm_utils;
                llvm::IRBuilder<>* builder;

                // CFI_cdesc_t-compatible field indices
                static constexpr int FIELD_BASE_ADDR   = 0;  // element_type*
                static constexpr int FIELD_ELEM_LEN    = 1;  // i64
                static constexpr int FIELD_VERSION     = 2;  // i32
                static constexpr int FIELD_RANK        = 3;  // i8
                static constexpr int FIELD_TYPE        = 4;  // i8
                static constexpr int FIELD_ATTRIBUTE   = 5;  // i8
                static constexpr int FIELD_EXTRA       = 6;  // i8 (reserved)
                static constexpr int FIELD_OFFSET      = 7;  // i64
                static constexpr int FIELD_DIMS        = 8;  // [rank x {i64,i64,i64}]

                llvm::Type* index_type;  // always i64 for descriptor indices
                llvm::StructType* dim_des;

                std::map<std::string, std::pair<llvm::StructType*, llvm::Type*>> tkr2array;
                std::map<std::pair<llvm::Type*, int>, llvm::StructType*> rank_array_cache;

                CompilerOptions& co;

                llvm::Value* cmo_convertor_single_element(
                    llvm::Type* type, llvm::Value* arr, std::vector<llvm::Value*>& m_args,
                    int n_args, bool check_for_bounds, LocationManager& lm, std::string array_name = "", std::string infile = "", Location loc = {0, 0});

                llvm::Value* cmo_convertor_single_element_data_only(
                    llvm::Value** llvm_diminfo, std::vector<llvm::Value*>& m_args,
                    int n_args, bool check_for_bounds, LocationManager& lm, bool is_unbounded_pointer_to_data = false, std::string array_name = "", std::string infile = "", Location loc = {0, 0});

            public:

                SimpleCMODescriptor(llvm::LLVMContext& _context,
                    llvm::IRBuilder<>* _builder,
                    LLVMUtils* _llvm_utils, CompilerOptions& co_,
                    llvm::Type* _index_type = nullptr);

                // Get the index type used by this descriptor (i32 or i64)
                llvm::Type* get_index_type() const { return index_type; }

                virtual
                bool is_array(ASR::ttype_t* asr_type);

                virtual
                llvm::Value* convert_to_argument(llvm::Value* tmp,
                    ASR::ttype_t* asr_arg_type, llvm::Type* arg_type,
                    bool data_only=false);

                virtual
                llvm::Type* get_argument_type(llvm::Type* type,
                    std::uint32_t m_h, std::string arg_name,
                    std::unordered_map
                    <std::uint32_t, std::unordered_map
                                    <std::string, llvm::Type*>>&
                        arr_arg_type_cache);

                virtual
                llvm::Type* get_array_type(
                    ASR::expr_t* expr,
                    ASR::ttype_t* m_type_,
                    llvm::Type* el_type,
                    bool get_pointer=false);

                virtual
                void fill_array_details(
                    llvm::Type* arr_ty, llvm::Value* arr, llvm::Type* llvm_data_type, int n_dims,
                    std::vector<std::pair<llvm::Value*, llvm::Value*>>& llvm_dims,
                    llvm::Module* module, bool reserve_data_memory=true);

                virtual
                void fill_array_details(ASR::expr_t* src_expr, ASR::expr_t* dest_expr,
                    llvm::Value* source, llvm::Value* destination,
                    ASR::ttype_t* source_array_type, ASR::ttype_t* dest_array_type, llvm::Module* module, bool ignore_data);

                virtual
                void fill_malloc_array_details(
                    llvm::Value* arr, llvm::Type *arr_type, llvm::Type* llvm_data_type, ASR::ttype_t* asr_type, int n_dims,
                    std::vector<std::pair<llvm::Value*, llvm::Value*>>& llvm_dims, llvm::Value* string_len,
                    ASR::symbol_t* variable_declaration, llvm::Module* module, 
                    ASR::symbol_t* allocated_subclass=nullptr, bool realloc=false,
                    ASR::ttype_t* alloc_type=nullptr);

                virtual
                void fill_dimension_descriptor(llvm::Type* type, llvm::Value* arr, int n_dims);

                virtual
                void reset_array_details(
                    llvm::Type* type, llvm::Value* arr, llvm::Type* source_arr_type, llvm::Value* source_arr, int n_dims);

                virtual
                void reset_array_details(
                    llvm::Type* type, llvm::Value* arr, llvm::Value* source_arr,
                    llvm::Value** lbs, llvm::Value** lengths,
                    int n_dims);

                virtual
                void fill_descriptor_for_array_section(
                    llvm::Value* value_desc, llvm::Type* value_el_type, ASR::ttype_t* value_type,
                    llvm::Type* value_desc_type,
                    llvm::Value* target, ASR::ttype_t* target_type, ASR::expr_t* /*target_expr*/,
                    llvm::Type* target_desc_type,
                    llvm::Value** lbs, llvm::Value** ubs,
                    llvm::Value** ds, llvm::Value** non_sliced_indices,
                    int value_rank, int target_rank, LocationManager& lm);

                virtual
                void fill_descriptor_for_array_section_data_only(
                    llvm::Value* value_desc, llvm::Type* value_el_type, ASR::ttype_t* value_type,
                    llvm::Value* target, ASR::ttype_t* target_type, ASR::expr_t* /*target_expr*/,
                    llvm::Type* target_desc_type,
                    llvm::Value** lbs, llvm::Value** ubs,
                    llvm::Value** ds, llvm::Value** non_sliced_indices,
                    llvm::Value** llvm_diminfo, int value_rank, int target_rank, LocationManager& lm);

                virtual
                llvm::Value* allocate_descriptor_on_heap(llvm::Type* array_desc_type);

                virtual
                llvm::Value* create_descriptor_alloca(llvm::Type* array_desc_type,
                    const std::string& name = "arr_desc");

                llvm::Type* get_array_type_for_rank(llvm::Type* el_type, int n_dims);

                virtual
                llvm::Type* get_dimension_descriptor_type(bool get_pointer=false);

                virtual
                llvm::Value* get_pointer_to_data(llvm::Type* type, llvm::Value* arr);

                /*
                 * Return pointer to data in array descriptor,
                 * Used arr_type to get the corresponding llvm::Type (LLVM 17+).
                */
                virtual
                llvm::Value* get_pointer_to_data(ASR::expr_t* arr_expr, ASR::ttype_t* arr_type, llvm::Value* arr, llvm::Module* module);


                virtual
                llvm::Value* get_rank(llvm::Type* type, llvm::Value* arr, bool get_pointer=false);

                virtual
                void set_rank(llvm::Type* type, llvm::Value* arr, llvm::Value* rank);

                virtual
                llvm::Value* get_offset(llvm::Type* type, llvm::Value* dim_des, bool load=true);

                virtual
                llvm::Value* get_lower_bound(llvm::Value* dim_des, bool load=true);

                virtual
                llvm::Value* get_upper_bound(llvm::Value* dim_des);

                virtual
                llvm::Value* get_dimension_size(llvm::Value* dim_des_arr,
                    llvm::Value* dim, bool load=true);

                virtual
                llvm::Value* get_dimension_size(llvm::Value* dim_des_arr,
                    bool load=true);

                virtual
                llvm::Value* get_pointer_to_dimension_descriptor_array(llvm::Type* type, llvm::Value* arr, bool load=true);

                virtual
                llvm::Value* get_pointer_to_dimension_descriptor(llvm::Value* dim_des_arr,
                    llvm::Value* dim);

                virtual
                llvm::Value* get_stride(llvm::Value* dim_des, bool load=true);

                virtual
                llvm::Value* get_single_element(llvm::Type *type, llvm::Value* array,
                    std::vector<llvm::Value*>& m_args, int n_args,
                    ASR::ttype_t* asr_type, ASR::expr_t* expr, LocationManager& lm,
                    ASR::symbol_t* variable_type_decl,
                    bool data_only=false, bool is_fixed_size=false,
                    llvm::Value** llvm_diminfo=nullptr,
                    bool polymorphic=false, llvm::Type* polymorphic_type=nullptr,
                    bool is_unbounded_pointer_to_data = false, bool check_for_bounds = false, std::string array_name = "", std::string infile = "");

                virtual
                llvm::Value* get_is_allocated_flag(llvm::Value* array, ASR::expr_t* array_exp);

                virtual
                void reset_is_allocated_flag(llvm::Type* typ_tmp, llvm::Value* array, llvm::Type* llvm_data_type);

                virtual
                llvm::Value* reshape(llvm::Type* arr_type, llvm::Value* array, llvm::Type* llvm_data_type,
                                     llvm::Type* shape_type, llvm::Value* shape, ASR::ttype_t* asr_shape_type,
                                     llvm::Module* module, ASR::expr_t* array_expr = nullptr,
                                     ASR::ttype_t* asr_data_type = nullptr,
                                     llvm::Type* result_desc_type = nullptr,
                                     llvm::Value* order = nullptr,
                                     ASR::expr_t* order_expr = nullptr);

                virtual
                void copy_array(llvm::Type* src_ty, llvm::Value* src, llvm::Type* dest_ty, llvm::Value* dest,
                                llvm::Module* module, ASR::expr_t* array_expr, ASR::ttype_t* asr_data_type,
                                bool reserve_memory);

                virtual
                void copy_array_move_allocation(llvm::Type* src_ty, llvm::Value* src, llvm::Type* dest_ty, llvm::Value* dest,
                                llvm::Module* module, ASR::expr_t* array_exp, ASR::ttype_t* asr_data_type,
                                bool reset_lower_bound);

                virtual
                void copy_array_data_only(llvm::Value* src, llvm::Value* dest,
                                          llvm::Module* module, llvm::Type* llvm_data_type,
                                          ASR::ttype_t* arr_type,
                                          llvm::Value* num_elements);

                virtual
                llvm::Value* get_array_size(llvm::Type* type, llvm::Value* array, llvm::Value* dim,
                                            int output_kind, int dim_kind=4);

                virtual
                llvm::Value* create_contiguous_copy_from_descriptor(
                    llvm::Type* source_llvm_type, llvm::Value* source_desc,
                    llvm::Type* elem_type, int rank, llvm::Module* module);

                virtual
                void copy_contiguous_data_to_descriptor(
                    llvm::Value* source_data,
                    llvm::Type* dest_llvm_type, llvm::Value* dest_desc,
                    llvm::Type* elem_type, int rank, llvm::Module* module);

                // CFI field indices (C-interop layout, no offset field)
                static constexpr int CFI_FIELD_BASE_ADDR   = 0;
                static constexpr int CFI_FIELD_ELEM_LEN    = 1;
                static constexpr int CFI_FIELD_VERSION     = 2;
                static constexpr int CFI_FIELD_RANK        = 3;
                static constexpr int CFI_FIELD_TYPE        = 4;
                static constexpr int CFI_FIELD_ATTRIBUTE   = 5;
                static constexpr int CFI_FIELD_EXTRA       = 6;
                static constexpr int CFI_FIELD_DIMS        = 7;

                // Get or create the CFI struct type (no offset field)
                llvm::StructType* get_cfi_type(llvm::Type* el_type, int n_dims);

                // Convert an internal descriptor to a CFI descriptor:
                // folds offset into base_addr, converts strides to bytes.
                llvm::Value* internal_to_cfi(
                    llvm::Type* internal_type, llvm::Value* internal_desc,
                    llvm::Type* el_type, int n_dims, uint64_t elem_size,
                    int8_t type_code, int cfi_attr);

                // Convert a CFI descriptor to an internal descriptor:
                // sets offset=0, converts byte strides to element strides.
                // internal_type: the cached internal struct type to use for the alloca.
                llvm::Value* cfi_to_internal(
                    llvm::Type* internal_type,
                    llvm::Type* el_type, llvm::Value* cfi_desc,
                    int n_dims, uint64_t elem_size);
                
                /*
                 * Extracts descriptor fields from a DescriptorArray and appends
                 * { is_descriptor_array=1, type_code, data_ptr, n_elems, stride }
                 * to `args`.  Used by formatted read codegen for
                 * descriptor-unwrapping logic.
                 */
                void push_descriptor_array_args(
                    ASR::expr_t* val_expr, ASR::ttype_t* expr_type_full,
                    ASR::ttype_t* val_type, llvm::Value* var_ptr,
                    llvm::Module* module, std::vector<llvm::Value*>& args);

        };

    } // LLVMArrUtils

} // namespace LCompilers

#endif // LFORTRAN_LLVM_ARR_UTILS_H
