Matrix.h

Go to the documentation of this file.

 
/*******************************************************************************
 * This file is part of the Zivid 3D Camera API
 *
 * Copyright 2015-2022 (C) Zivid AS
 * All rights reserved.
 *
 * Zivid Software License, v1.0
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * 3. Neither the name of Zivid AS nor the names of its contributors may be used
 * to endorse or promote products derived from this software without specific
 * prior written permission.
 *
 * 4. This software, with or without modification, must not be used with any
 * other 3D camera than from Zivid AS.
 *
 * 5. Any software provided in binary form under this license must not be
 * reverse engineered, decompiled, modified and/or disassembled.
 *
 * THIS SOFTWARE IS PROVIDED BY ZIVID AS "AS IS" AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY, NONINFRINGEMENT, AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL ZIVID AS OR CONTRIBUTORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Contact: Zivid Customer Success Team <customersuccess@zivid.com>
 * Info:    http://www.zivid.com
 ******************************************************************************/
 
#pragma once
 
#include <array>
#include <exception>
#include <functional>
#include <ostream>
#include <sstream>
#include <string>
 
namespace Zivid
{
    template<typename T, size_t rowCount, size_t colCount>
    class Matrix
    {
        static_assert(rowCount > 0 && colCount > 0, "rowCount and colCount must be > 0");
 
    private:
        using Storage = std::array<T, rowCount * colCount>;
 
    public:
        using ValueType = T;
 
        using Iterator = typename Storage::iterator;
 
        using ConstIterator = typename Storage::const_iterator;
 
        static constexpr size_t rows{ rowCount };
 
        static constexpr size_t cols{ colCount };
 
        Matrix() = default;
 
        explicit Matrix(const std::array<T, colCount * rowCount> &arrArr)
            : m_mat{ arrArr }
        {}
 
        template<typename Iterator>
        Matrix(Iterator beginIt, Iterator endIt)
        {
            if(std::distance(beginIt, endIt) != rows * cols)
            {
                throw std::out_of_range("Input size does not match matrix size. Expected " + std::to_string(rows * cols)
                                        + " items, got " + std::to_string(std::distance(beginIt, endIt)) + " items.");
            }
 
            std::copy(beginIt, endIt, m_mat.begin());
        }
 
        explicit Matrix(std::initializer_list<T> values)
            : Matrix{ values.begin(), values.end() }
        {}
 
        explicit Matrix(std::initializer_list<std::initializer_list<T>> values)
        {
            if(values.size() != rows)
            {
                throw std::out_of_range("Input size does not match matrix size. Expected " + std::to_string(rows)
                                        + "rows, got " + std::to_string(values.size()) + " rows");
            }
 
            auto iterator = begin();
            for(const auto &row : values)
            {
                if(row.size() != cols)
                {
                    throw std::out_of_range("Input size does not match matrix size. Expected all rows to be of size "
                                            + std::to_string(cols) + ", got row with " + std::to_string(row.size())
                                            + " values");
                }
                for(const auto &value : row)
                {
                    *(iterator++) = value;
                }
            }
        }
 
        Iterator begin()
        {
            return m_mat.begin();
        }
 
        Iterator end()
        {
            return m_mat.end();
        }
 
        ConstIterator cbegin() const
        {
            return m_mat.cbegin();
        }
 
        ConstIterator cend() const
        {
            return m_mat.cend();
        }
 
        T &at(size_t row, size_t col)
        {
            throwIfOutOfBounds(row, col);
            return m_mat[row * cols + col];
        }
 
        const T &at(size_t row, size_t col) const
        {
            throwIfOutOfBounds(row, col);
            return m_mat[row * cols + col];
        }
 
        T &operator()(size_t row, size_t col)
        {
            return m_mat[row * cols + col];
        }
 
        const T &operator()(size_t row, size_t col) const
        {
            return m_mat[row * cols + col];
        }
 
        T *data()
        {
            return m_mat.data();
        }
 
        const T *data() const
        {
            return m_mat.data();
        }
 
        std::string toString() const
        {
            std::stringstream ss;
            ss << "[ ";
            for(size_t row = 0; row < rowCount; row++)
            {
                ss << (row == 0 ? "[" : "\n  [");
                for(size_t col = 0; col < colCount; col++)
                {
                    const auto value{ m_mat[row * colCount + col] };
                    ss << (value >= 0 ? " " : "") << std::to_string(value);
                    ss << (col == colCount - 1 ? "" : ", ");
                }
                ss << (row == rowCount - 1 ? "]" : "], ");
            }
            ss << " ]";
            return ss.str();
        }
 
    private:
        static void throwIfOutOfBounds(size_t row, size_t col)
        {
            if(row >= rows)
            {
                throw std::out_of_range("Trying to access row with index " + std::to_string(row)
                                        + ", but allowed range is [0, " + std::to_string(rows - 1) + "]");
            }
            if(col >= cols)
            {
                throw std::out_of_range("Trying to access column with index " + std::to_string(col)
                                        + ", but allowed range is [0, " + std::to_string(cols - 1) + "]");
            }
        }
 
        Storage m_mat{};
    };
 
    template<typename T, size_t rowCount, size_t colCount>
    std::ostream &operator<<(std::ostream &stream, const Matrix<T, rowCount, colCount> &matrix)
    {
        return stream << matrix.toString();
    }
 
    using Matrix3x3 = Matrix<float, 3, 3>;
 
    using Matrix3x3d = Matrix<double, 3, 3>;
 
    using Matrix4x4 = Matrix<float, 4, 4>;
 
    using Matrix4x4d = Matrix<double, 4, 4>;
} // namespace Zivid