/*
 * Copyright 2011, Haiku Inc. All rights reserved.
 * This file may be used under the terms of the MIT License.
 *
 * Authors:
 *		Jérôme Duval
 *		Chế Vũ Gia Hy
 */


#include "Chunk.h"


//#define TRACE_BTRFS
#ifdef TRACE_BTRFS
#	define TRACE(x...) dprintf("\33[34mbtrfs:\33[0m " x)
#else
#	define TRACE(x...) ;
#endif
#	define FATAL(x...) dprintf("\33[34mbtrfs:\33[0m " x)


Chunk::Chunk(btrfs_chunk* chunk, fsblock_t offset)
	:
	fChunk(NULL),
	fInitStatus(B_OK)
{
	fChunkOffset = offset;
	fChunk = (btrfs_chunk*)malloc(sizeof(btrfs_chunk)
		+ chunk->StripeCount() * sizeof(btrfs_stripe));
	if (fChunk == NULL) {
		fInitStatus = B_NO_MEMORY;
		return;
	}

	memcpy(fChunk, chunk, sizeof(btrfs_chunk)
		+ chunk->StripeCount() * sizeof(btrfs_stripe));

	TRACE("chunk[0] length %" B_PRIu64 " owner %" B_PRIu64 " stripe_length %"
		B_PRIu64 " type %" B_PRIu64 " stripe_count %u sub_stripes %u "
		"sector_size %" B_PRIu32 "\n", chunk->Length(), chunk->Owner(),
		chunk->StripeLength(), chunk->Type(), chunk->StripeCount(),
		chunk->SubStripes(), chunk->SectorSize());
	for (int32 i = 0; i < chunk->StripeCount(); i++) {
		TRACE("chunk.stripe[%" B_PRId32 "].physical %" B_PRId64 " deviceid %"
			B_PRId64 "\n", i, chunk->stripes[i].Offset(),
			chunk->stripes[i].DeviceID());
	}
}


Chunk::~Chunk()
{
	free(fChunk);
}


uint32
Chunk::Size() const
{
	return sizeof(btrfs_chunk)
		+ fChunk->StripeCount() * sizeof(btrfs_stripe);
}


status_t
Chunk::FindBlock(off_t logical, off_t& physical)
{
	if (fChunk == NULL)
		return B_NO_INIT;

	if (logical < (off_t)fChunkOffset
		|| logical > (off_t)(fChunkOffset + fChunk->Length()))
		return B_BAD_VALUE;

	// only one stripe
	physical = logical + fChunk->stripes[0].Offset() - fChunkOffset;
	return B_OK;
}