add mini numbits

Author: Iro96

numbits/CMakeLists.txt

@@ -0,0 +1,77 @@
+cmake_minimum_required(VERSION 3.15)
+project(NumBits VERSION 1.0.0 LANGUAGES CXX)
+
+set(CMAKE_CXX_STANDARD 17)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+set(CMAKE_CXX_EXTENSIONS OFF)
+
+# Build options
+option(NUMBITS_BUILD_TESTS "Build NumBits tests" ON)
+option(NUMBITS_BUILD_EXAMPLES "Build NumBits examples" ON)
+
+# Include directories
+include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include)
+
+# Source files
+set(NUMBITS_SOURCES
+    src/array.cpp
+    src/operations.cpp
+    src/math_functions.cpp
+    src/linear_algebra.cpp
+    src/broadcasting.cpp
+    src/array_manipulation.cpp
+    src/indexing.cpp
+)
+
+set(NUMBITS_HEADERS
+    include/numbits/array.hpp
+    include/numbits/operations.hpp
+    include/numbits/math_functions.hpp
+    include/numbits/linear_algebra.hpp
+    include/numbits/broadcasting.hpp
+    include/numbits/array_manipulation.hpp
+    include/numbits/indexing.hpp
+    include/numbits/io.hpp
+    include/numbits/types.hpp
+    include/numbits/utils.hpp
+    include/numbits/numbits.hpp
+)
+
+# Create library
+add_library(numbits STATIC ${NUMBITS_SOURCES} ${NUMBITS_HEADERS})
+
+target_include_directories(numbits PUBLIC
+    $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
+    $<INSTALL_INTERFACE:include>
+)
+
+# Build examples
+if(NUMBITS_BUILD_EXAMPLES)
+    add_subdirectory(examples)
+endif()
+
+# Build tests
+if(NUMBITS_BUILD_TESTS)
+    enable_testing()
+    add_subdirectory(tests)
+endif()
+
+# Installation
+install(TARGETS numbits
+    EXPORT NumBitsTargets
+    LIBRARY DESTINATION lib
+    ARCHIVE DESTINATION lib
+    RUNTIME DESTINATION bin
+)
+
+install(DIRECTORY include/numbits
+    DESTINATION include
+    FILES_MATCHING PATTERN "*.hpp"
+)
+
+install(EXPORT NumBitsTargets
+    FILE NumBitsTargets.cmake
+    NAMESPACE NumBits::
+    DESTINATION lib/cmake/NumBits
+)
+

numbits/examples/CMakeLists.txt

@@ -0,0 +1,17 @@
+cmake_minimum_required(VERSION 3.15)
+
+# Example 1: Basic array operations
+add_executable(example_basic example_basic.cpp)
+target_link_libraries(example_basic numbits)
+
+# Example 2: Linear algebra
+add_executable(example_linear_algebra example_linear_algebra.cpp)
+target_link_libraries(example_linear_algebra numbits)
+
+# Example 3: Math functions
+add_executable(example_math example_math.cpp)
+target_link_libraries(example_math numbits)
+
+# Example 4: Broadcasting
+add_executable(example_broadcasting example_broadcasting.cpp)
+target_link_libraries(example_broadcasting numbits)

numbits/examples/example_basic.cpp

@@ -0,0 +1,69 @@
+#include <iostream>
+#include "numbits/ndarray.hpp"
+#include "numbits/operations.hpp"
+
+using namespace numbits;
+
+int main() {
+    std::cout << "=== NumBits Basic Example ===\n\n";
+    
+    // Create ndarrays
+    ndarray<float> a({2, 3}, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f});
+    ndarray<float> b({2, 3}, {7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f});
+    
+    std::cout << "n-D array a:\n";
+    a.print();
+    std::cout << "\n\n";
+    
+    std::cout << "n-D array b:\n";
+    b.print();
+    std::cout << "\n\n";
+    
+    // Basic operations
+    auto c = a + b;
+    std::cout << "a + b:\n";
+    c.print();
+    std::cout << "\n\n";
+    
+    auto d = a * 2.0f;
+    std::cout << "a * 2:\n";
+    d.print();
+    std::cout << "\n\n";
+    
+    // Scalar operations
+    auto e = a + 10.0f;
+    std::cout << "a + 10:\n";
+    e.print();
+    std::cout << "\n\n";
+    
+    // Create zeros and ones
+    auto zeros = ndarray<float>::zeros({3, 4});
+    std::cout << "Zeros (3x4):\n";
+    zeros.print();
+    std::cout << "\n\n";
+    
+    auto ones = ndarray<float>::ones({2, 2});
+    std::cout << "Ones (2x2):\n";
+    ones.print();
+    std::cout << "\n\n";
+    
+    // Reshape
+    auto flattened = a.flatten();
+    std::cout << "Flattened a:\n";
+    flattened.print();
+    std::cout << "\n\n";
+    
+    auto reshaped = flattened.reshape({3, 2});
+    std::cout << "Reshaped to (3x2):\n";
+    reshaped.print();
+    std::cout << "\n\n";
+    
+    // Sum and mean
+    std::cout << "Sum of a: " << sum(a) << "\n";
+    std::cout << "Mean of a: " << mean(a) << "\n";
+    std::cout << "Min of a: " << min(a) << "\n";
+    std::cout << "Max of a: " << max(a) << "\n";
+    
+    return 0;
+}
+

numbits/examples/example_broadcasting.cpp

@@ -0,0 +1,72 @@
+#include <iostream>
+#include "numbits/ndarray.hpp"
+#include "numbits/operations.hpp"
+#include "numbits/broadcasting.hpp"
+
+using namespace numbits;
+
+int main() {
+    std::cout << "=== NumBits Broadcasting Example ===\n\n";
+    
+    // Create ndarrays of different shapes
+    ndarray<float> a({2, 3}, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f});
+    ndarray<float> b({3}, {10.0f, 20.0f, 30.0f});
+    
+    std::cout << "n-D array a (2x3):\n";
+    a.print();
+    std::cout << "\n\n";
+    
+    std::cout << "n-D array b (3):\n";
+    b.print();
+    std::cout << "\n\n";
+    
+    // Broadcasting: a + b
+    auto c = a + b;
+    std::cout << "a + b (broadcasted):\n";
+    c.print();
+    std::cout << "\n\n";
+    
+    // Scalar broadcasting
+    ndarray<float> scalar_arr({1}, {5.0f});
+    auto d = a + scalar_arr;
+    std::cout << "a + scalar (5):\n";
+    d.print();
+    std::cout << "\n\n";
+    
+    // Different broadcasting scenario
+    ndarray<float> row({1, 4}, {1.0f, 2.0f, 3.0f, 4.0f});
+    ndarray<float> col({4, 1}, {1.0f, 10.0f, 100.0f, 1000.0f});
+    
+    std::cout << "Row vector (1x4):\n";
+    row.print();
+    std::cout << "\n\n";
+    
+    std::cout << "Column vector (4x1):\n";
+    col.print();
+    std::cout << "\n\n";
+    
+    auto broadcasted = row + col;
+    std::cout << "Row + Column (broadcasted to 4x4):\n";
+    broadcasted.print();
+    std::cout << "\n\n";
+    
+    // Element-wise multiplication with broadcasting
+    ndarray<float> matrix({3, 3}, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f});
+    ndarray<float> vec({3}, {2.0f, 3.0f, 4.0f});
+    
+    std::cout << "Matrix (3x3):\n";
+    matrix.print();
+    std::cout << "\n\n";
+    
+    std::cout << "Vector (3):\n";
+    vec.print();
+    std::cout << "\n\n";
+    
+    auto multiplied = matrix * vec;
+    std::cout << "Matrix * Vector (broadcasted):\n";
+    multiplied.print();
+    std::cout << "\n";
+    
+    return 0;
+}
+

numbits/examples/example_linear_algebra.cpp

@@ -0,0 +1,67 @@
+#include <iostream>
+#include "numbits/ndarray.hpp"
+#include "numbits/linear_algebra.hpp"
+
+using namespace numbits;
+
+int main() {
+    std::cout << "=== NumBits Linear Algebra Example ===\n\n";
+    
+    // Create matrices
+    ndarray<float> A({2, 3}, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f});
+    ndarray<float> B({3, 2}, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f});
+    
+    std::cout << "Matrix A (2x3):\n";
+    A.print();
+    std::cout << "\n\n";
+    
+    std::cout << "Matrix B (3x2):\n";
+    B.print();
+    std::cout << "\n\n";
+    
+    // Matrix multiplication
+    auto C = matmul(A, B);
+    std::cout << "A @ B (matrix multiplication):\n";
+    C.print();
+    std::cout << "\n\n";
+    
+    // Transpose
+    auto At = transpose(A);
+    std::cout << "A transpose:\n";
+    At.print();
+    std::cout << "\n\n";
+    
+    // Determinant
+    ndarray<float> square({2, 2}, {1.0f, 2.0f, 3.0f, 4.0f});
+    std::cout << "Square matrix:\n";
+    square.print();
+    std::cout << "\n";
+    std::cout << "Determinant: " << determinant(square) << "\n\n";
+    
+    // Inverse
+    ndarray<float> invertible({2, 2}, {4.0f, 7.0f, 2.0f, 6.0f});
+    std::cout << "Invertible matrix:\n";
+    invertible.print();
+    std::cout << "\n";
+    
+    auto inv = inverse(invertible);
+    std::cout << "Inverse:\n";
+    inv.print();
+    std::cout << "\n\n";
+    
+    // Verify: A @ A^(-1) should be identity
+    auto identity_check = matmul(invertible, inv);
+    std::cout << "A @ A^(-1) (should be identity):\n";
+    identity_check.print();
+    std::cout << "\n\n";
+    
+    // Trace
+    ndarray<float> matrix({3, 3}, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f});
+    std::cout << "Matrix:\n";
+    matrix.print();
+    std::cout << "\n";
+    std::cout << "Trace: " << trace(matrix) << "\n";
+    
+    return 0;
+}
+

numbits/examples/example_math.cpp

@@ -0,0 +1,84 @@
+#include <iostream>
+#include <cmath>
+#include "numbits/ndarray.hpp"
+#include "numbits/math_functions.hpp"
+
+using namespace numbits;
+
+int main() {
+    std::cout << "=== NumBits Math Functions Example ===\n\n";
+    
+    // Create ndarray
+    ndarray<float> arr({2, 3}, {0.0f, M_PI/4.0f, M_PI/2.0f, M_PI, 2.0f*M_PI, 1.0f});
+    
+    std::cout << "Original n-D array:\n";
+    arr.print();
+    std::cout << "\n\n";
+    
+    // Trigonometric functions
+    auto sin_arr = sin(arr);
+    std::cout << "sin(arr):\n";
+    sin_arr.print();
+    std::cout << "\n\n";
+    
+    auto cos_arr = cos(arr);
+    std::cout << "cos(arr):\n";
+    cos_arr.print();
+    std::cout << "\n\n";
+    
+    // Exponential and logarithmic
+    ndarray<float> exp_arr({1, 4}, {0.0f, 1.0f, 2.0f, -1.0f});
+    std::cout << "n-D array for exp/log:\n";
+    exp_arr.print();
+    std::cout << "\n";
+    
+    auto exp_result = exp(exp_arr);
+    std::cout << "exp(arr):\n";
+    exp_result.print();
+    std::cout << "\n";
+    
+    auto log_result = log(exp_result);
+    std::cout << "log(exp(arr)) (should equal original):\n";
+    log_result.print();
+    std::cout << "\n\n";
+    
+    // Power and sqrt
+    ndarray<float> power_arr({2, 2}, {4.0f, 9.0f, 16.0f, 25.0f});
+    std::cout << "n-D array:\n";
+    power_arr.print();
+    std::cout << "\n";
+    
+    auto sqrt_result = sqrt(power_arr);
+    std::cout << "sqrt(arr):\n";
+    sqrt_result.print();
+    std::cout << "\n";
+    
+    auto pow_result = pow(power_arr, 0.5f);
+    std::cout << "pow(arr, 0.5) (should equal sqrt):\n";
+    pow_result.print();
+    std::cout << "\n\n";
+    
+    // Rounding functions
+    ndarray<float> round_arr({1, 5}, {1.4f, 1.6f, -1.4f, -1.6f, 2.5f});
+    std::cout << "n-D array:\n";
+    round_arr.print();
+    std::cout << "\n";
+    
+    auto floor_result = floor(round_arr);
+    std::cout << "floor(arr):\n";
+    floor_result.print();
+    std::cout << "\n";
+    
+    auto ceil_result = ceil(round_arr);
+    std::cout << "ceil(arr):\n";
+    ceil_result.print();
+    std::cout << "\n";
+    
+    auto round_result = round(round_arr);
+    std::cout << "round(arr):\n";
+    round_result.print();
+    std::cout << "\n";
+    
+    return 0;
+}
+