/* SPDX-FileCopyrightText: 2001-2002 NaN Holding BV. All rights reserved.
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

/** \file
 * \ingroup imbuf
 */

#include <cerrno>
#include <cstdlib>

#include "BLI_path_utils.hh" /* For assertions. */

#include "IMB_colormanagement.hh"
#include "IMB_filetype.hh"
#include "IMB_imbuf.hh"
#include "IMB_imbuf_types.hh"

#include "CLG_log.h"

namespace blender {

static CLG_LogRef LOG = {"image.write"};

static void ensure_byte_buffer(ImBuf *ibuf, const ImFileType *type)
{
  /* If writing byte image from float buffer, create a byte buffer for writing.
   *
   * For color managed image writing, IMB_colormanagement_imbuf_for_write should
   * have already created this byte buffer. This is a basic fallback for other
   * cases where we do not have a specific desired output colorspace. */
  if (!(type->flag & IM_FTYPE_FLOAT)) {
    if (ibuf->byte_data() == nullptr && ibuf->float_data()) {
      ibuf->byte_buffer.colorspace = colormanage_colorspace_get_roled(COLOR_ROLE_DEFAULT_BYTE);
      IMB_byte_from_float(ibuf);
    }
  }
}

bool IMB_save_image(ImBuf *ibuf, const char *filepath, ImBufFlags flags)
{
  errno = 0;

  BLI_assert(!BLI_path_is_rel(filepath));

  if (ibuf == nullptr) {
    return false;
  }
  ibuf->flags = flags;

  const ImFileType *type = IMB_file_type_from_ibuf(ibuf);
  if (type == nullptr || type->save == nullptr) {
    CLOG_ERROR(&LOG, "Couldn't save image to \"%s\"", filepath);
    return false;
  }

  BLI_assert((type->capability_write & eImFileTypeCapability::File) !=
             eImFileTypeCapability::Zero);

  ensure_byte_buffer(ibuf, type);
  return type->save(ibuf, filepath, flags);
}

Vector<uint8_t> IMB_save_image_to_buffer(ImBuf *ibuf, ImBufFlags flags)
{
  if (ibuf == nullptr) {
    return {};
  }
  ibuf->flags = flags;

  const ImFileType *type = IMB_file_type_from_ibuf(ibuf);
  if (type == nullptr || type->save_buffer == nullptr) {
    CLOG_ERROR(&LOG, "Couldn't save image to buffer");
    return {};
  }

  BLI_assert((type->capability_write & eImFileTypeCapability::Memory) !=
             eImFileTypeCapability::Zero);

  ensure_byte_buffer(ibuf, type);
  return type->save_buffer(ibuf, flags);
}

}  // namespace blender