/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _BCACHEFS_COMPRESS_H
#define _BCACHEFS_COMPRESS_H

#include "extents_types.h"

static const unsigned __bch2_compression_opt_to_type[] = {
#define x(t, n) [BCH_COMPRESSION_OPT_##t] = BCH_COMPRESSION_TYPE_##t,
	BCH_COMPRESSION_OPTS()
#undef x
};

struct bch_compression_opt {
	u8		type:4,
			level:4;
};

static inline struct bch_compression_opt __bch2_compression_decode(unsigned v)
{
	return (struct bch_compression_opt) {
		.type	= v & 15,
		.level	= v >> 4,
	};
}

static inline bool bch2_compression_opt_valid(unsigned v)
{
	struct bch_compression_opt opt = __bch2_compression_decode(v);

	return opt.type < ARRAY_SIZE(__bch2_compression_opt_to_type) && !(!opt.type && opt.level);
}

static inline struct bch_compression_opt bch2_compression_decode(unsigned v)
{
	return bch2_compression_opt_valid(v)
		? __bch2_compression_decode(v)
		: (struct bch_compression_opt) { 0 };
}

static inline unsigned bch2_compression_encode(struct bch_compression_opt opt)
{
	return opt.type|(opt.level << 4);
}

static inline enum bch_compression_type bch2_compression_opt_to_type(unsigned v)
{
	return __bch2_compression_opt_to_type[bch2_compression_decode(v).type];
}

static inline void bch2_prt_compression_type(struct printbuf *out, enum bch_compression_type type)
{
	if (type < BCH_COMPRESSION_TYPE_NR)
		prt_str(out, __bch2_compression_types[type]);
	else
		prt_printf(out, "(invalid compression type %u)", type);
}

int bch2_bio_uncompress_inplace(struct bch_fs *, struct bio *,
				struct bch_extent_crc_unpacked *);
int bch2_bio_uncompress(struct bch_fs *, struct bio *, struct bio *,
		       struct bvec_iter, struct bch_extent_crc_unpacked);
unsigned bch2_bio_compress(struct bch_fs *, struct bio *, size_t *,
			   struct bio *, size_t *, unsigned);

int bch2_check_set_has_compressed_data(struct bch_fs *, unsigned);
void bch2_fs_compress_exit(struct bch_fs *);
int bch2_fs_compress_init(struct bch_fs *);

void bch2_compression_opt_to_text(struct printbuf *, u64);

int bch2_opt_compression_parse(struct bch_fs *, const char *, u64 *, struct printbuf *);
void bch2_opt_compression_to_text(struct printbuf *, struct bch_fs *, struct bch_sb *, u64);
int bch2_opt_compression_validate(u64, struct printbuf *);

#define bch2_opt_compression (struct bch_opt_fn) {		\
	.parse		= bch2_opt_compression_parse,		\
	.to_text	= bch2_opt_compression_to_text,		\
	.validate	= bch2_opt_compression_validate,	\
}

#endif /* _BCACHEFS_COMPRESS_H */