meshgrid.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <utility>
 
#include "NumCpp/Core/Internal/StaticAsserts.hpp"
#include "NumCpp/Functions/arange.hpp"
#include "NumCpp/NdArray.hpp"
 
namespace nc
{
    //============================================================================
    // Method Description:
    template<typename dtype>
    std::pair<NdArray<dtype>, NdArray<dtype>> meshgrid(const NdArray<dtype>& inICoords, const NdArray<dtype>& inJCoords)
    {
        STATIC_ASSERT_ARITHMETIC(dtype);
 
        const uint32 numRows      = inJCoords.size();
        const uint32 numCols      = inICoords.size();
        auto         returnArrayI = NdArray<dtype>(numRows, numCols);
        auto         returnArrayJ = NdArray<dtype>(numRows, numCols);
 
        // first the I array
        for (uint32 row = 0; row < numRows; ++row)
        {
            for (uint32 col = 0; col < numCols; ++col)
            {
                returnArrayI(row, col) = inICoords[col];
            }
        }
 
        // then the I array
        for (uint32 col = 0; col < numCols; ++col)
        {
            for (uint32 row = 0; row < numRows; ++row)
            {
                returnArrayJ(row, col) = inJCoords[row];
            }
        }
 
        return std::make_pair(returnArrayI, returnArrayJ);
    }
 
    //============================================================================
    // Method Description:
    template<typename dtype>
    std::pair<NdArray<dtype>, NdArray<dtype>> meshgrid(const Slice& inSlice1, const Slice& inSlice2)
    {
        return meshgrid(arange<dtype>(inSlice1), arange<dtype>(inSlice2));
    }
 
} // namespace nc