| 109 | } |
| 110 | |
| 111 | std::vector<py::ssize_t> ComputeNumpyShape(const Image& img) |
| 112 | { |
| 113 | std::vector<py::ssize_t> shape; |
| 114 | |
| 115 | for (unsigned int i = 0; i < img.GetDimension(); ++i) |
| 116 | shape.push_back(img.GetDimension(i)); |
| 117 | |
| 118 | if (img.GetPixelType().GetNumberOfComponents() > 1) |
| 119 | shape.push_back(img.GetPixelType().GetNumberOfComponents()); |
| 120 | |
| 121 | std::reverse(shape.begin(), shape.end()); |
| 122 | |
| 123 | return shape; |
| 124 | } |
| 125 | |
| 126 | py::array BuildNumpyArray(const Image& img, const void* data, bool writeable, py::capsule capsule) |
| 127 | { |
no test coverage detected