Function BitmapFromArray

src/pycolmap/sensor/bitmap.cc:39–68

}

Bitmap BitmapFromArray(py::array_t<uint8_t, py::array::c_style> array,
                       bool linear_colorspace) {
  int channels = 1;
  if (array.ndim() == 3) {
    channels = array.shape(2);
  } else if (array.ndim() != 2) {
    throw std::runtime_error("Input array must have 2 or 3 dimensions!");
  }

  const int width = array.shape(1);
  const int height = array.shape(0);
  if (width == 0 || height == 0) {
    throw std::runtime_error(
        "Input array must have positive width and height!");
  }

  if (channels != 1 && channels != 3) {
    throw std::runtime_error("Input array must have 1 or 3 channels!");
  }

  const bool as_rgb = channels != 1;
  Bitmap output(width,
                height,
                /*as_rgb=*/as_rgb,
                /*linear_colorspace=*/linear_colorspace);

  const uint8_t* input_row_ptr = static_cast<uint8_t*>(array.request().ptr);
  std::memcpy(output.RowMajorData().data(), input_row_ptr, output.NumBytes());
  return output;
}

void BindBitmap(pybind11::module& m) {
  using BitmapRescaleFilter = Bitmap::RescaleFilter;