Function af_save_image_native

src/api/c/imageio2.cpp:410–540

Save an image to disk.

// Save an image to disk.
af_err af_save_image_native(const char* filename, const af_array in) {
    try {
        ARG_ASSERT(0, filename != NULL);

        FreeImage_Module& _ = getFreeImagePlugin();

        // set your own FreeImage error handler
        _.FreeImage_SetOutputMessage(FreeImageErrorHandler);

        // try to guess the file format from the file extension
        FREE_IMAGE_FORMAT fif = _.FreeImage_GetFileType(filename, 0);
        if (fif == FIF_UNKNOWN) {
            fif = _.FreeImage_GetFIFFromFilename(filename);
        }

        if (fif == FIF_UNKNOWN) {
            AF_ERROR("FreeImage Error: Unknown Filetype", AF_ERR_NOT_SUPPORTED);
        }

        const ArrayInfo& info = getInfo(in);
        // check image color type
        auto channels = static_cast<FI_CHANNELS>(info.dims()[2]);
        DIM_ASSERT(1, channels <= 4);
        DIM_ASSERT(1, channels != 2);

        // sizes
        uint fi_w = info.dims()[1];
        uint fi_h = info.dims()[0];

        af_dtype type = info.getType();

        // FI assumes [0-255] for u8
        // FI assumes [0-65k] for u16
        // FI assumes [0-1]   for f32
        int fi_bpp = 0;
        switch (type) {
            case u8: fi_bpp = channels * 8; break;
            case u16: fi_bpp = channels * 16; break;
            case f32: fi_bpp = channels * 32; break;
            default: TYPE_ERROR(1, type);
        }

        FREE_IMAGE_TYPE fit_type = getFIT(channels, type);

        // create the result image storage using FreeImage
        bitmap_ptr pResultBitmap = make_bitmap_ptr(nullptr);
        switch (type) {
            case u8:
            case u16:
            case f32:
                pResultBitmap.reset(_.FreeImage_AllocateT(fit_type, fi_w, fi_h,
                                                          fi_bpp, 0, 0, 0));
                break;
            default: TYPE_ERROR(1, type);
        }

        if (pResultBitmap == NULL) {
            AF_ERROR("FreeImage Error: Error creating image or file",