/*
 * 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
 *
 */


#include <Atom/ImageProcessing/PixelFormats.h>
#include <Atom/ImageProcessing/ImageObject.h>
#include <Converters/PixelOperation.h>
#include <Processing/PixelFormatInfo.h>

#include <Converters/Histogram.h>

///////////////////////////////////////////////////////////////////////////////////

namespace ImageProcessingAtom
{
    float GetLuminance(const float& r, const float& g, const float& b)
    {
        return (r * 0.30f + g * 0.59f + b * 0.11f);
    }

    bool ComputeLuminanceHistogram(IImageObjectPtr imageObject, Histogram<256>& histogram)
    {
        EPixelFormat pixelFormat = imageObject->GetPixelFormat();
        if (!(CPixelFormats::GetInstance().IsPixelFormatUncompressed(pixelFormat)))
        {
            AZ_Assert(false, "%s function only works with uncompressed pixel format", __FUNCTION__);
            return false;
        }

        //create pixel operation function
        IPixelOperationPtr pixelOp = CreatePixelOperation(pixelFormat);

        //setup histogram bin
        static const size_t binCount = 256;
        Histogram<binCount>::Bins bins;
        Histogram<binCount>::clearBins(bins);

        //get count of bytes per pixel
        AZ::u32 pixelBytes = CPixelFormats::GetInstance().GetPixelFormatInfo(pixelFormat)->bitsPerBlock / 8;

        const  AZ::u32 mipCount = imageObject->GetMipCount();
        float color[4];
        for (uint32 mip = 0; mip < mipCount; ++mip)
        {
            AZ::u8* pixelBuf;
            AZ::u32 pitch;
            imageObject->GetImagePointer(mip, pixelBuf, pitch);
            const uint32 pixelCount = imageObject->GetPixelCount(mip);

            for (uint32 i = 0; i < pixelCount; ++i, pixelBuf += pixelBytes)
            {
                pixelOp->GetRGBA(pixelBuf, color[0], color[1], color[2], color[3]);

                const float luminance = AZ::GetClamp(GetLuminance(color[0], color[1], color[2]), 0.0f, 1.0f);
                const float f = luminance * binCount;
                if (f <= 0)
                {
                    ++bins[0];
                }
                else
                {
                    const int bin = int(f);
                    ++bins[(bin < binCount) ? bin : binCount - 1];
                }
            }
        }

        histogram.set(bins);
        return true;
    }
} // end namespace ImageProcessingAtom
