/*
        AUTOMATICALLY GENERATED BY jTemp FROM
        /Users/jsh2/Work/openimaj/target/checkout/machine-learning/clustering/src/main/jtemp/org/openimaj/ml/clustering/#T#CentroidsResult.jtemp
*/
/**
 * Copyright (c) 2011, The University of Southampton and the individual contributors.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 *   *  Redistributions of source code must retain the above copyright notice,
 *      this list of conditions and the following disclaimer.
 *
 *   *  Redistributions in binary form must reproduce the above copyright notice,
 *      this list of conditions and the following disclaimer in the documentation
 *      and/or other materials provided with the distribution.
 *
 *   *  Neither the name of the University of Southampton nor the names of its
 *      contributors may be used to endorse or promote products derived from this
 *      software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
   
package org.openimaj.ml.clustering;

import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.Scanner;

import org.openimaj.ml.clustering.CentroidsProvider;
import org.openimaj.ml.clustering.Clusters;
import org.openimaj.ml.clustering.SpatialClusters;
import org.openimaj.ml.clustering.assignment.HardAssigner;
import org.openimaj.ml.clustering.assignment.hard.ExactShortAssigner;
import org.openimaj.util.pair.IntFloatPair;

/**
 * The result of a {@link SpatialClusterer} that just produces a flat set of centroids.
 * 
 * @author Jonathon Hare (jsh2@ecs.soton.ac.uk)
 */
public class ShortCentroidsResult implements SpatialClusters<short[]>, CentroidsProvider<short[]> {
        final static String HEADER = Clusters.CLUSTER_HEADER + "Short".charAt(0) + "Cen";
        
        /** The centroids of the clusters */
        public short[][] centroids;
                
        @Override
        public boolean equals(Object obj){
                if(!(obj instanceof ShortCentroidsResult)) 
                        return false;
                        
                ShortCentroidsResult other = (ShortCentroidsResult)obj;
                for (int i = 0; i < this.centroids.length; i++) {
                        if (!Arrays.equals(this.centroids[i], other.centroids[i]))
                                return false;
                }
                return true;
        }

        @Override
        public String asciiHeader() {
                return "ASCII"+HEADER ;
        }

        @Override
        public byte[] binaryHeader() {
                return HEADER.getBytes();
        }

        @Override
        public void readASCII(Scanner br) throws IOException {
                // Read Header
                final int K = Integer.parseInt(br.nextLine().trim());
                final int M = Integer.parseInt(br.nextLine().trim());
                
                centroids = new short[K][M];
                for (int k=0; k<K; k++) {
                        String [] parts = br.nextLine().split(",");
                        
                        for (int d=0; d<M; d++) {
                                centroids[k][d] = Short.parseShort(parts[d]);
                        }
                }
        }

        @Override
        public void readBinary(DataInput in) throws IOException {
                final int K = in.readInt();
                final int M = in.readInt();
                                
                centroids = new short[K][M];
                
                for (int k=0; k<K; k++) {
                        for (int d=0; d<M; d++) {
                                centroids[k][d] = in.readShort();
                        }
                }
        }

        @Override
        public void writeASCII(PrintWriter writer) throws IOException {
                writer.println(centroids.length);
                writer.println(centroids[0].length);
                
                for (int k=0; k<centroids.length; k++) {
                        for (int d=0; d<centroids[0].length; d++) {
                                writer.print(centroids[k][d] + ",");
                        }
                        writer.println();
                }
        }

        @Override
        public void writeBinary(DataOutput out) throws IOException {
                out.writeInt(centroids.length);
                out.writeInt(centroids[0].length);
                                
                for (int k=0; k<centroids.length; k++) {
                        for (int d=0; d<centroids[0].length; d++) {
                                out.writeShort(centroids[k][d]);
                        }
                }
        }
        
        @Override
        public String toString() {
                String str = "";
                str += "ShortCentroidsResult" + "\n";
                str += "No. of Clusters: "  + centroids.length + "\n";
                str += "No. of Dimensions: "  + centroids[0].length + "\n";
                return str;
        }
        
        @Override
        public short[][] getCentroids() {
                return this.centroids;
        }
        
        @Override
        public HardAssigner<short[], float[], IntFloatPair> defaultHardAssigner() {
                return new ExactShortAssigner(this);
        }
        
        @Override
        public int numDimensions() {
                return centroids[0].length;
        }

        @Override
        public int numClusters() {
                return centroids.length;
        }
}