Как прочитать информацию об изображении из строки python байтов с помощью Boost :: GIL? - PullRequest
Новая
       29

Как прочитать информацию об изображении из строки python байтов с помощью Boost :: GIL?

1 голос
/ 07 мая 2020

Я пишу функцию для чтения ширины и высоты изображения из заголовка файла 

int read_image_dimensions() {
    namespace bg = boost::gil;

    std::string filepath = "image.png";
    std::ifstream byte_stream(filepath, std::ios::binary);

    int width = 0, height = 0;
    bg::image_read_info<bg::png_tag> png_tag_info;

    png_tag_info = bg::read_image_info(byte_stream, bg::image_read_settings<bg::png_tag>())._info;
    width = png_tag_info._width; height = png_tag_info._height;

    std::cout << width << "x" << height << std::endl;
    return 0;
}

И я понятия не имею, как читать информацию об изображении из python байтов данных изображения, взятых из Blender API , то же самое, что и в файле изображения. 

using namespace boost::python;

int read_image_dimensions(object &image) {
    object image_packed_file = extract<object>(image.attr("packed_file"));
    object packed_data = extract<object>(image_packed_file.attr("data"));
    size_t packed_size = extract<size_t>(image_packed_file.attr("size"));

    // ... 
}

1 Ответ

1 голос
/ 08 мая 2020

Итак, игнорируя материал python, давайте просто предположим, что вы каким-то образом удержали байты: 

std::vector<char> pydata;

Затем вы можете просто создать поток из этого. Есть несколько способов добиться этого, но позвольте мне использовать Boost Iostreams, чтобы добиться большей эффективности: 

io::stream<io::array_source> byte_stream(pydata.data(), pydata.size());

Вот и все. Остальное - это то, что у вас было.

Live Demo

Это демонстрирует, что мое изображение в профиле имеет размер 200x200 пикселей: 

#include <boost/gil/extension/io/png.hpp>
#include <boost/iostreams/device/array.hpp>
#include <boost/iostreams/stream.hpp>
#include <iostream>

namespace bg = boost::gil;
namespace io = boost::iostreams;

extern std::vector<char> pydata;

int main() {
    io::stream<io::array_source> byte_stream(pydata.data(), pydata.size());

    auto info = bg::read_image_info(
            byte_stream,
            bg::image_read_settings<bg::png_tag>())
        ._info;

    std::cout << "Dimensions: " << info._width << "x" << info._height << "\n";
}

static char const polar_bear[] {
  "\x89\x50\x4e\x47\x0d\x0a\x1a\x0a\x00\x00\x00\x0d"
  "\x49\x48\x44\x52\x00\x00\x00\xc8\x00\x00\x00\xc8"
  "\x08\x06\x00\x00\x00\xad\x58\xae\x9e\x00\x00\x20"
  "\x00\x49\x44\x41\x54\x78\x9c\x94\xbd\x59\xb3\x2d"
  "\xcb\x71\x1e\xf6\x65\x56\x75\xf7\x5a\x7b\xef\x73"
  "\xee\x39\x77\xc0\x9d\x31\x03\xc4\x44\xc0\x00\x88"
  // 5623 lines snipped...
  "\x45\x4e\x44\xae\x42\x60\x82"
};

std::vector<char> pydata { std::begin(polar_bear), std::end(polar_bear) };

Печать 

Dimensions: 200x200
...