You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
o3de/Gems/Atom/Asset/ImageProcessingAtom/Code/Source/Converters/Histogram.h

81 lines
2.0 KiB
C++

/*
* Copyright (c) Contributors to the Open 3D Engine Project.
* For complete copyright and license terms please see the LICENSE at the root of this distribution.
*
* SPDX-License-Identifier: Apache-2.0 OR MIT
*
*/
#pragma once
namespace ImageProcessingAtom
{
template <size_t BIN_COUNT>
class Histogram
{
public:
typedef AZ::u64 Bins[BIN_COUNT];
public:
Histogram()
{
}
static void clearBins(Bins& bins)
{
memset(&bins, 0, sizeof(bins));
}
void set(const Bins& bins)
{
m_bins[0] = bins[0];
m_binsCumulative[0] = bins[0];
double sum = 0.0f;
for (size_t i = 1; i < BIN_COUNT; ++i)
{
m_bins[i] = bins[i];
m_binsCumulative[i] = m_binsCumulative[i - 1] + bins[i];
sum += i * double(bins[i]);
}
const AZ::u64 totalCount = getTotalSampleCount();
m_meanBin = (totalCount <= 0) ? 0.0f : float(sum / totalCount);
}
AZ::u64 getTotalSampleCount() const
{
return m_binsCumulative[BIN_COUNT - 1];
}
float getPercentage(size_t minBin, size_t maxBin) const
{
const AZ::u64 totalCount = getTotalSampleCount();
if ((totalCount <= 0) || (minBin > maxBin) || (maxBin < 0) || (minBin >= BIN_COUNT))
{
return 0.0f;
}
minBin = AZ::GetMax(minBin, size_t(0));
maxBin = AZ::GetMin(maxBin, BIN_COUNT - 1);
const AZ::u64 count = m_binsCumulative[maxBin] - ((minBin <= 0) ? 0 : m_binsCumulative[minBin - 1]);
return float((double(count) * 100.0) / double(totalCount));
}
float getMeanBin() const
{
return m_meanBin;
}
private:
Bins m_bins;
Bins m_binsCumulative;
float m_meanBin;
};
bool ComputeLuminanceHistogram(IImageObjectPtr imageObject, Histogram<256>& histogram);
}