/*
 * Copyright 2000, Georges-Edouard Berenger. All rights reserved.
 * Copyright 2022, Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 */
#include "TeamBarMenuItem.h"

#include "Colors.h"
#include "ProcessController.h"
#include "ThreadBarMenu.h"
#include "ThreadBarMenuItem.h"
#include "Utilities.h"

#include <Bitmap.h>
#include <ControlLook.h>


TeamBarMenuItem::TeamBarMenuItem(BMenu* menu, BMessage* kill_team, team_id team,
	BBitmap* icon, bool deleteIcon)
	:
	IconMenuItem(icon, menu, true, deleteIcon),
	fTeamID(team)
{
	SetMessage(kill_team);
	Init();
}


void
TeamBarMenuItem::Init()
{
	if (get_team_usage_info(fTeamID, B_TEAM_USAGE_SELF, &fTeamUsageInfo) != B_OK)
		fTeamUsageInfo.kernel_time = fTeamUsageInfo.user_time = 0;

	if (fTeamID == B_SYSTEM_TEAM) {
		thread_info thinfos;
		bigtime_t idle = 0;
		for (unsigned int t = 1; t <= gCPUcount; t++) {
			if (get_thread_info(t, &thinfos) == B_OK)
				idle += thinfos.kernel_time + thinfos.user_time;
		}
		fTeamUsageInfo.kernel_time += fTeamUsageInfo.user_time;
		fTeamUsageInfo.user_time = idle;
	}

	fLastTime = system_time();
	fKernel = -1;
	fGrenze1 = -1;
	fGrenze2 = -1;
}


TeamBarMenuItem::~TeamBarMenuItem()
{
}


void
TeamBarMenuItem::DrawContent()
{
	BPoint loc;

	DrawIcon();
	if (fKernel < 0)
		BarUpdate();
	else
		DrawBar(true);

	loc = ContentLocation();
	loc.x += ceilf(be_control_look->DefaultLabelSpacing() * 3.3f);
	Menu()->MovePenTo(loc);
	BMenuItem::DrawContent();
}


void
TeamBarMenuItem::DrawBar(bool force)
{
	const bool selected = IsSelected();
	BRect frame = Frame();
	BMenu* menu = Menu();
	rgb_color highColor = menu->HighColor();

	BFont font;
	menu->GetFont(&font);
	frame = bar_rect(frame, &font);

	if (fKernel < 0)
		return;

	if (fGrenze1 < 0)
		force = true;

	if (force) {
		if (selected)
			menu->SetHighColor(gFrameColorSelected);
		else
			menu->SetHighColor(gFrameColor);

		menu->StrokeRect(frame);
	}

	frame.InsetBy(1, 1);
	BRect r = frame;
	float grenze1 = frame.left + (frame.right - frame.left)
		* fKernel / gCPUcount;
	float grenze2 = frame.left + (frame.right - frame.left)
		* (fKernel + fUser) / gCPUcount;

	if (grenze1 > frame.right)
		grenze1 = frame.right;

	if (grenze2 > frame.right)
		grenze2 = frame.right;

	r.right = grenze1;
	if (!force)
		r.left = fGrenze1;

	if (r.left < r.right) {
		if (selected)
			menu->SetHighColor(gKernelColorSelected);
		else
			menu->SetHighColor(gKernelColor);

		menu->FillRect(r);
	}

	r.left = grenze1;
	r.right = grenze2;

	if (!force) {
		if (fGrenze2 > r.left && r.left >= fGrenze1)
			r.left = fGrenze2;

		if (fGrenze1 < r.right && r.right <= fGrenze2)
			r.right = fGrenze1;
	}

	if (r.left < r.right) {
		if (selected) {
			menu->SetHighColor(fTeamID == B_SYSTEM_TEAM
				? gIdleColorSelected
				: gUserColorSelected);
		} else {
			menu->SetHighColor(fTeamID == B_SYSTEM_TEAM
				? gIdleColor
				: gUserColor);
		}

		menu->FillRect(r);
	}

	r.left = grenze2;
	r.right = frame.right;

	if (!force)
		r.right = fGrenze2;

	if (r.left < r.right) {
		if (selected)
			menu->SetHighColor(gWhiteSelected);
		else
			menu->SetHighColor(kWhite);

		menu->FillRect(r);
	}

	menu->SetHighColor(highColor);
	fGrenze1 = grenze1;
	fGrenze2 = grenze2;
}


void
TeamBarMenuItem::GetContentSize(float* width, float* height)
{
	IconMenuItem::GetContentSize(width, height);
	if (width != NULL)
		*width += 40 + kBarWidth;
}


void
TeamBarMenuItem::BarUpdate()
{
	team_usage_info usage;
	if (get_team_usage_info(fTeamID, B_TEAM_USAGE_SELF, &usage) == B_OK) {
		bigtime_t now = system_time();
		bigtime_t idle = 0;
		if (fTeamID == B_SYSTEM_TEAM) {
			thread_info thinfos;
			for (unsigned int t = 1; t <= gCPUcount; t++) {
				if (get_thread_info(t, &thinfos) == B_OK)
					idle += thinfos.kernel_time + thinfos.user_time;
			}
			usage.kernel_time += usage.user_time;
			usage.user_time = idle;
			idle -= fTeamUsageInfo.user_time;
		}

		fKernel = double(usage.kernel_time - fTeamUsageInfo.kernel_time - idle)
			/ double(now - fLastTime);

		fUser = double(usage.user_time - fTeamUsageInfo.user_time)
			/ double(now - fLastTime);

		if (fKernel < 0)
			fKernel = 0;

		fLastTime = now;
		fTeamUsageInfo = usage;
		DrawBar(false);
	} else
		fKernel = -1;
}


void
TeamBarMenuItem::Reset(char* name, team_id team, BBitmap* icon, bool deleteIcon)
{
	IconMenuItem::Reset(icon, deleteIcon);

	SetLabel(name);
	fTeamID = team;
	Init();

	Message()->ReplaceInt32("team", team);
	((ThreadBarMenu*)Submenu())->Reset(team);
	BarUpdate();
}