#include "stdinc.h"
#include "dat.h"
#include "fns.h"

int	count[VtMaxLumpSize][VtMaxType];

enum
{
	ClumpChunks	= 32*1024
};

static
readArenaInfo(Arena *arena)
{
	ClumpInfo *ci, *cis;
	u32int clump;
	int i, n, ok;

	if(arena->clumps)
		fprint(2, "reading directory for arena=%s with %d entries\n", arena->name, arena->clumps);

	cis = MKN(ClumpInfo, ClumpChunks);
	ok = 1;
	for(clump = 0; clump < arena->clumps; clump += n){
		n = ClumpChunks;

		if(n > arena->clumps - clump)
			n = arena->clumps - clump;

		if(readClumpInfos(arena, clump, cis, n) != n){
			setErr(EOk, "arena directory read failed: %R");
			ok = 0;
			break;
		}

		for(i = 0; i < n; i++){
			ci = &cis[i];
			if(ci->type >= VtMaxType || ci->uncsize >= VtMaxLumpSize) {
				fprint(2, "bad clump: %d: type = %d: size = %d\n", clump+i, ci->type, ci->uncsize);
				continue;
			}
if(ci->uncsize == 422)
print("%s: %d: %V\n", arena->name, clump+i, ci->score);
			count[ci->uncsize][ci->type]++;
		}
	}
	free(cis);
	if(!ok)
		return TWID32;
	return clump;
}

static void
clumpStats(Index *ix)
{
	int ok;
	ulong clumps, n;
	int i, j, t;

	ok = 1;
	clumps = 0;
	for(i = 0; i < ix->narenas; i++){
		n = readArenaInfo(ix->arenas[i]);
		if(n == TWID32){
			ok = 0;
			break;
		}
		clumps += n;
	}

	if(!ok)
		return;

	print("clumps = %ld\n", clumps);
	for(i=0; i<VtMaxLumpSize; i++) {
		t = 0;
		for(j=0; j<VtMaxType; j++)
			t += count[i][j];
		if(t == 0)
			continue;
		print("%d\t%d", i, t);
		for(j=0; j<VtMaxType; j++)
			print("\t%d", count[i][j]);
		print("\n");
	}
}


void
usage(void)
{
	fprint(2, "usage: clumpstats [-B blockcachesize] config\n");
	exits(0);
}

int
main(int argc, char *argv[])
{
	u32int bcmem;

	vtAttach();

	bcmem = 0;

	ARGBEGIN{
	case 'B':
		bcmem = unittoull(ARGF());
		break;
	default:
		usage();
		break;
	}ARGEND

	readonly = 1;

	if(argc != 1)
		usage();

	if(!initVenti(argv[0]))
		fatal("can't init venti: %R");

	if(bcmem < maxBlockSize * (mainIndex->narenas + mainIndex->nsects * 4 + 16))
		bcmem = maxBlockSize * (mainIndex->narenas + mainIndex->nsects * 4 + 16);
	fprint(2, "initialize %d bytes of disk block cache\n", bcmem);
	initDCache(bcmem);

	clumpStats(mainIndex);
	
	exits(0);
	return 0;	/* shut up stupid compiler */
}