DispatcherKM.cs

Go to the documentation of this file.

using UnityEngine;
 
namespace ClusteringAlgorithms
{
    public class DispatcherKM : ASimpleDispatcer
    {
        public DispatcherKM(
            ComputeShader computeShader,
            int numIterations,
            bool doRandomizeEmptyClusters,
            bool useFullResTexRef,
            ClusteringRTsAndBuffers clusteringRTsAndBuffers
        )
            : base(
                computeShader: computeShader,
                numIterations: numIterations,
                doRandomizeEmptyClusters: doRandomizeEmptyClusters,
                useFullResTexRef: useFullResTexRef,
                clusteringRTsAndBuffers: clusteringRTsAndBuffers
            ) { }
 
        public override string name => "KM";
 
        public override bool doesReadback => false;
 
        /// <summary>
        /// Each iteration first attributes pixels to clusters,
        /// then updates cluster centers.
        /// In order to use the resulting cluster centers,
        /// one final cluster attribution is required!
        /// </summary>
        public override void RunClustering(ClusteringTextures clusteringTextures)
        {
            for (int i = 0; i < this.numIterations; i++)
            {
                this.KMiteration(clusteringTextures, rejectOld: false);
            }
        }
 
        /// <summary>
        /// First attributes pixels to clusters.
        /// Then updates cluster centers.
        /// In order to use the resulting cluster centers,
        /// one final cluster attribution is required!
        /// </summary>
        protected void KMiteration(ClusteringTextures textures, bool rejectOld)
        {
            this.AttributeClustersKM(textures);
            this.UpdateClusterCenters(textures, rejectOld);
        }
 
        public override void SingleIteration(ClusteringTextures textures)
        {
            this.computeShader.SetBool(
                "do_random_sample_empty_clusters",
                this.doRandomizeEmptyClusters
            );
            this.computeShader.SetInt("num_clusters", this.clusteringRTsAndBuffers.numClusters);
 
            this.KMiteration(textures, rejectOld: false);
        }
    }
}