Method getitem_mask

src/python/PyImath/PyImathFixedVArray.cpp:617–660

template <class T>
FixedArray<int>
FixedVArray<T>::SizeHelper::getitem_mask (const FixedArray<int>& mask) const
{
    int len = mask.len();
    if (len != _a.len())
    {
        throw std::invalid_argument("Dimensions of mask do not match array");
    }
    
    int count = 0;
    for (Py_ssize_t i = 0; i < mask.len(); ++i)
    {
        if (mask[i]) count += 1;
    }

    FixedArray<int> f(count);
    
    if (_a._indices)
    {
        size_t index = 0;
        for (Py_ssize_t i = 0; i < mask.len(); ++i)
        {
            if (mask[i])
            {
                f.direct_index(index) = _a._ptr[_a.raw_ptr_index(i)*_a._stride].size();
                index += 1;
            }
        }
    }
    else
    {
        size_t index = 0;
        for (Py_ssize_t i = 0; i < mask.len(); ++i)
        {
            if (mask[i])
            {
                f.direct_index(index) = _a._ptr[i*_a._stride].size();
                index += 1;
            }
        }
    }

    return f;
}


template <class T>