haiku.git

/*
 * Copyright 2013-2014, Stephan Aßmus <superstippi@gmx.de>.
 * Copyright 2013, Rene Gollent <rene@gollent.com>.
 * Copyright 2016-2025, Andrew Lindesay <apl@lindesay.co.nz>.
 * All rights reserved. Distributed under the terms of the MIT License.
 */


#include "UserRatingSummary.h"

#include "HaikuDepotConstants.h"
#include "Logger.h"


UserRatingSummary::UserRatingSummary()
	:
	fAverageRating(RATING_MISSING),
	fRatingCount(0),
	fRatingCountNoStar(0)
{
	for (int i = 0; i <= 5; i++)
		fRatingCountByStar[i] = 0;
}


UserRatingSummary::UserRatingSummary(const UserRatingSummary& other)
	:
	fAverageRating(other.AverageRating()),
	fRatingCount(other.RatingCount()),
	fRatingCountNoStar(other.RatingCountByStar(RATING_MISSING_STAR))
{
	for (int i = 0; i < 6; i++)
		fRatingCountByStar[i] = other.RatingCountByStar(i);
}


float
UserRatingSummary::AverageRating() const
{
	return fAverageRating;
}


int
UserRatingSummary::RatingCount() const
{
	return fRatingCount;
}


void
UserRatingSummary::SetAverageRating(float value)
{
	fAverageRating = value;
}


void
UserRatingSummary::SetRatingCount(int value)
{
	fRatingCount = value;
}


int
UserRatingSummary::RatingCountByStar(int star) const
{
	if (star == RATING_MISSING_STAR)
		return fRatingCountNoStar;

	if (star >= 0 && star <= 5)
		return fRatingCountByStar[star];

	return 0;
}


void
UserRatingSummary::SetRatingByStar(int star, int ratingCount)
{
	if (star == RATING_MISSING_STAR)
		fRatingCountNoStar = ratingCount;

	if (ratingCount < 0) {
		HDERROR("bad rating count %" B_PRId32, ratingCount);
		return;
	}

	if (star < RATING_MISSING_STAR || star > 5) {
		HDERROR("bad star %" B_PRId32, star);
		return;
	}

	fRatingCountByStar[star] = ratingCount;
}


bool
UserRatingSummary::operator==(const UserRatingSummary& other) const
{
	if (fAverageRating != other.AverageRating() || fRatingCount != other.RatingCount()
		|| fRatingCountNoStar != other.RatingCountByStar(-1)) {
		return false;
	}

	for (int i = 0; i < 6; i++) {
		if (fRatingCountByStar[i] != other.RatingCountByStar(i))
			return false;
	}

	return true;
}


bool
UserRatingSummary::operator!=(const UserRatingSummary& other) const
{
	return !(*this == other);
}


// #pragma mark - UserRatingSummaryBuilder


UserRatingSummaryBuilder::UserRatingSummaryBuilder()
	:
	fAverageRating(RATING_MISSING),
	fRatingCount(0),
	fRatingCountNoStar(0)
{
	for (int i = 0; i < 6; i++)
		fRatingCountByStar[i] = 0;
}


UserRatingSummaryBuilder::UserRatingSummaryBuilder(const UserRatingSummaryRef& other)
	:
	fSource(),
	fAverageRating(RATING_MISSING),
	fRatingCount(0),
	fRatingCountNoStar(0)
{
	for (int i = 0; i < 6; i++)
		fRatingCountByStar[i] = other->RatingCountByStar(i);
}


UserRatingSummaryBuilder::~UserRatingSummaryBuilder()
{
}


UserRatingSummaryRef
UserRatingSummaryBuilder::BuildRef()
{
	if (fSource.IsSet())
		return fSource;

	UserRatingSummary* userRatingSummary = new UserRatingSummary();
	userRatingSummary->SetAverageRating(fAverageRating);
	userRatingSummary->SetRatingCount(fRatingCount);
	userRatingSummary->SetRatingByStar(RATING_MISSING_STAR, fRatingCountNoStar);

	for (int star = 0; star <= 5; star++)
		userRatingSummary->SetRatingByStar(star, fRatingCountByStar[star]);

	return UserRatingSummaryRef(userRatingSummary, true);
}


UserRatingSummaryBuilder&
UserRatingSummaryBuilder::WithAverageRating(float value)
{
	if (!fSource.IsSet() || fSource->AverageRating() != value) {
		_InitFromSource();
		fAverageRating = value;
	}
	return *this;
}


UserRatingSummaryBuilder&
UserRatingSummaryBuilder::WithRatingCount(int value)
{
	if (!fSource.IsSet() || fSource->RatingCount() != value) {
		_InitFromSource();
		fRatingCount = value;
	}
	return *this;
}


UserRatingSummaryBuilder&
UserRatingSummaryBuilder::AddRatingByStar(int star, int ratingCount)
{
	if (star < RATING_MISSING_STAR || star > 5) {
		HDERROR("bad star %" B_PRId32 " in builder", star);
		return *this;
	}

	if (!fSource.IsSet() || fSource->RatingCountByStar(star) != ratingCount) {
		_InitFromSource();

		if (star == RATING_MISSING_STAR)
			fRatingCountNoStar = ratingCount;
		else
			fRatingCountByStar[star] = ratingCount;
	}

	return *this;
}


void
UserRatingSummaryBuilder::_InitFromSource()
{
	if (fSource.IsSet()) {
		_Init(fSource.Get());
		fSource.Unset();
	}
}


void
UserRatingSummaryBuilder::_Init(const UserRatingSummary* value)
{
	fAverageRating = value->AverageRating();
	fRatingCount = value->RatingCount();
	fRatingCountNoStar = value->RatingCountByStar(RATING_MISSING_STAR);

	for (int i = 0; i <= 5; i++)
		fRatingCountByStar[i] = value->RatingCountByStar(i);
}