one.empty3.feature.kmeans.MakeDataset Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of empty3-library-generic Show documentation

3D rendering engine. Plus modeling. Expected glsl textures 3d and 2d rendering

There is a newer version: 2023.5

Show newest version

/*
 * Copyright (c) 2022-2023. Manuel Daniel Dahmen
 *
 *
 *    Copyright 2012-2023 Manuel Daniel Dahmen
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

package one.empty3.feature.kmeans;

import one.empty3.feature.PixM;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.HashMap;

/*
 line : l, c, r, g, b
 */
public class MakeDataset {
    public MakeDataset(File image,
                       File outputCsv, int res) {
        try {
            BufferedImage img = ImageIO.read(image);
            PixM pix;
            if (res > 0)

                pix = PixM.getPixM(img, res);
            else
                pix = new PixM(img);

            PrintWriter pw = new PrintWriter(outputCsv);
            for (int l = 0; l < pix.getLines(); l++)
                for (int c = 0; c < pix.getColumns(); c++) {
                    //if (pix.luminance(c, l) > 0.1) { // ADDED
                    pix.setCompNo(0);
                    double r = pix.get(c, l);

                    pix.setCompNo(1);
                    double g = pix.get(c, l);

                    pix.setCompNo(2);
                    double b = pix.get(c, l);

                    pw.println("" + c + " " + l + " " +
                            r + " " + g + " " + b);
                    //}
                }
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

    public static void writeFeature
            (HashMap data,
             File csvout) {
        try {

            PrintWriter pw = new PrintWriter(csvout);

            for (int i = 0; i < data.size(); i++) {
                for (int j = 0; j < data.get(i).length; j++) {
                    pw.print(data.get(i)[j] + " ");
                }
                pw.println();
            }
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

}