All Downloads are FREE. Search and download functionalities are using the official Maven repository.

one.empty3.feature.Lines5colors Maven / Gradle / Ivy

There is a newer version: 2024.5.10
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;

import one.empty3.io.ProcessFile;
import one.empty3.library.ColorTexture;
import one.empty3.library.LineSegment;
import one.empty3.library.Point3D;
import one.empty3.library.core.lighting.Colors;
import one.empty3.library.core.nurbs.CourbeParametriquePolynomialeBezier;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Random;
import java.util.function.Consumer;

public class Lines5colors extends ProcessFile {


    ArrayList listTmpCurve = new ArrayList<>();
    ArrayList listTmpX = new ArrayList<>();
    ArrayList listTmpY = new ArrayList<>();
    ArrayList listTmpZ = new ArrayList<>();
    private PixM pixM;
    private double pz;
    private double py;
    private double px;
    private double distMax;
    private Random random = new Random();

    public Lines5colors() {
    }

    public List relierPoints(List> points, Point3D p0) {
        List list = new ArrayList<>();

        List p = points.get(0);

        for (int i = 0; i < p.size(); i++) {
            Point3D proche = proche(p0, p);
            if (proche == null)
                return list;
            else {
                p.remove(proche);
                list.add(proche);
            }
        }

        return list;
    }

    private Point3D proche(Point3D point3D, List p) {
        double dist = distMax;
        Point3D pRes = null;
        for (Point3D p2 : p) {
            if (Point3D.distance(point3D, p2) < dist && p2 != point3D && !p2.equals(point3D)) {
                dist = Point3D.distance(point3D, p2);
                pRes = p2;
            }
        }
        return pRes;
    }

    public double r() {
        return (random.doubles().iterator().nextDouble() + 1.) / 2;
    }

    @Override
    public boolean process(File in, File out) {
        listTmpCurve = new ArrayList<>();
        listTmpX = new ArrayList<>();
        listTmpY = new ArrayList<>();
        listTmpZ = new ArrayList<>();
        try {
            pixM = null;
            pixM = new PixM(ImageIO.read(in));
            ArrayList> lists = new ArrayList<>();
            lists.add(new ArrayList<>());
            PixM o = new PixM(pixM.getColumns(), pixM.getLines());

            double valueDiff = 0.1;


            for (double levels : Arrays.asList(1.0, 0.8, 0.6, 0.4, 0.3, 0.2/*,0.1,0.0*/)) {
                int[][] p = new int[pixM.getColumns()][pixM.getLines()];//!!

                pz = 0.0;
                py = 0.0;
                px = 0.0;
                distMax = (pixM.getColumns() + pixM.getLines()) / 2;//???
                random = new Random();
                listTmpCurve = new ArrayList();
                listTmpX = new ArrayList();
                listTmpY = new ArrayList();
                listTmpZ = new ArrayList();


                for (int x = 0; x < pixM.getColumns(); x++)
                    for (int y = 0; y < pixM.getLines(); y++)
                        p[x][y] = 0;

                for (int i = 0; i < pixM.getColumns(); i++) {
                    for (int j = 0; j < pixM.getLines(); j++) {


                        int x = i;
                        int y = j;
                        if (!isInBound(new Point3D((double) x, (double) y, 0.0)))
                            continue;
                        double valueAvg = pixM.luminance(x, y);

                        if (p[x][y] == 0) {
                            listTmpCurve.add(new Point3D((double) x, (double) y, valueAvg));
                        } else {
                            continue;
                        }

                        int cont = 1;

                        while (valueAvg >= levels - valueDiff && valueAvg <= levels + valueDiff && cont == 1 && p[x][y] == 0) {//2nd condition

                            p[x][y] = 1;


                            neighborhood((int) (double) x, (int) (double) y, valueAvg, valueDiff, levels);
//
                            while (listTmpX.size() > 0) {
                                getTmp(0);
                                x = (int) px;
                                y = (int) py;
                                removeTmp(0);
                                if (!isInBound(new Point3D(px, py, 0.0)))
                                    break;

                                if (p[x][y] == 0) {
                                    listTmpCurve.add(new Point3D((double) x, (double) y, levels));

                                    cont = 1;

                                    valueAvg = pixM.luminance(x, y);

                                } else cont = 0;
                            }

                        }
                        /*
                        for (List ps : lists)
                            for (int k = 0; k < ps.size(); k++) {
                                Point3D p0 = ps.get(k);
                                for (int c = 0; c < listTmpCurve.size(); c++)
                                    if (listTmpCurve.get(c).equals(p0) &&
                                            lists.get(0).contains(p0)) {
                                        Point3D p1 = listTmpCurve.get(c);
                                        if (isInBound(p1))
                                            p[(int) (double) p1.getX()]
                                                    [(int) (double) p1.getY()] = 1;
                                        //listTmpCurve.remove(c);
                                        lists.get(0).remove(p0);
                                    }
                            }*/
                        if (listTmpCurve.size() == 1)
                            lists.get(0).add(listTmpCurve.get(0));
                        else if (listTmpCurve.size() > 1 && !lists.contains(listTmpCurve)) {
                            lists.add(listTmpCurve);
                            //listTmpCurve = new ArrayList<>();//!!
                        }
                    }
                }
            }


            List> lists2 = new ArrayList<>();
            int index3 = 0;
            while (index3 < lists.size() && lists.get(index3).size() == 0)
                index3++;
            if (index3 < lists.size() && lists.get(index3).size() > 0) {

                List point3DS = relierPoints(lists, lists.get(index3).get(0));
                int index = 0;
                do {
                    if (point3DS != null) {
                        index++;
                        lists2.add(point3DS);
                    }
                    while (index3 < lists.size() && lists.get(index3).size() == 0)
                        index3++;
                    if (index3 < lists.size()) {
                        point3DS = relierPoints(lists, lists.get(index3).get(index));
                    }
                } while (index3 < lists.size() && point3DS != null && point3DS.size() > 0 && index < lists.get(0).size() - 1);
            }

            List lines = new ArrayList<>();
            List> list3 = new ArrayList<>();

            for (List listP : lists2) {
                int i = 0;
                int a = 0;
                int b = listP.size() - 1;
                list3.add(new ArrayList<>());
                if (listP.size() > 0) {
                    boolean passed = false;
                    list3.get(list3.size() - 1).add(listP.get(0));
                    for (Point3D point3D : listP) {
                        Double distNormal = 1.0;//0.9??
                        if (isInBound(point3D)) {
                            int j = 0;
                            for (j = 1; j < listP.size(); j++) {
                                if (Point3D.distance(point3D, listP.get(j)) >= distNormal) {
                                    distNormal = Point3D.distance(point3D, listP.get(j));
                                    list3.get(list3.size() - 1).add(listP.get(j));
                                }  //j--;break;


                            }
                            if (j == listP.size()) {
                                break;
                            }
                            a = j - 1;
                            int k;
                            for (k = 0; k >= -listP.size(); k--) {
                                if (Point3D.distance(point3D, listP.get((listP.size() + k) % listP.size())) >= distNormal) {
                                    distNormal = Point3D.distance(point3D, listP.get((listP.size() + k) % listP.size()));
                                    list3.get(list3.size() - 1).add(listP.get((listP.size() + k) % listP.size()));
                                }
                            }
                            if (k == -listP.size()) {
                                break;
                            }
                            b = (k + 1 + listP.size()) % listP.size();

                            passed = true;
                        }

                    }
                    if (list3.get(list3.size() - 1).size() >= 2 && passed) {
                        Point3D p1 = list3.get(list3.size() - 1).get(a);
                        Point3D p2 = list3.get(list3.size() - 1).get(b);
                        list3.remove(list3.size() - 1);
                        list3.add(new ArrayList<>());
                        list3.get(list3.size() - 1).add(p1);
                        list3.get(list3.size() - 1).add(p2);

                    }
                }

            }

            List coefficients = new ArrayList<>();


            lists2.forEach(p3s -> {
                Color r = new Color((float) r(), (float) r(), (float) r());
                double xA;
                double yA;
                p3s.forEach(point3D -> {
                    if (isInBound(point3D)) {
                        o.setValues((int) (double) (point3D.getX()), (int) (double) (point3D.getY()), r.getRed() / 255., r.getGreen() / 255., r.getBlue() / 255.);
                    }


                });
            });

            PixM img3 = new PixM(pixM.getColumns(), pixM.getLines());
            list3.forEach(p3s -> {
                Color r = new Color((float) r(), (float) r(), (float) r());
                double xA;
                double yA;
                /*p3s.forEach(point3D -> {
                    if (isInBound(point3D)) {
                        img3.setValues((int) (double) (point3D.getX()), (int) (double) (point3D.getY()), r.getRed() / 255., r.getGreen() / 255., r.getBlue() / 255.);
                    }


                });*/
            });

            list3.forEach(p3s -> {
                Color r = new Color((float) r(), (float) r(), (float) r());
                if (p3s.size() > 2) {
                    for (int j = 0; j < p3s.size() - 1; j++) {
                        Point3D p1 = p3s.get(j);
                        Point3D p2 = p3s.get(j + 1);
                        double length = p1.moins(p2).norme();
                        for (double i = 0; i < 1.0; i += 1. / length) {
                            Point3D point3D = p1.plus(p2.moins(p1).mult(i));
                            img3.setValues((int) (double) (point3D.getX()), (int) (double) (point3D.getY()),
                                    p1.getZ(), p1.getZ(), p1.getZ()
                                    /*        r.getRed() / 255., r.getGreen() / 255., r.getBlue() / 255.*/
                            );
                        }
                    }
                }
            });
            List> points = new ArrayList<>();
            CourbeParametriquePolynomialeBezier[] courbeParametriquePolynomialeBeziers = new CourbeParametriquePolynomialeBezier[lists2.size()];

            int[] i = new int[]{0};
            lists2.forEach(p3s -> {
                Color r = new Color((float) r(), (float) r(), (float) r());
                final Point3D[][] extremes = {new Point3D[2], new Point3D[2]};
                final Double[] distMaxMinP1 = {2.5, 1000.0};
                List pointsCurrent = new ArrayList<>();
                points.add(pointsCurrent);

                p3s.forEach(point3D1 -> {
                    Point3D p1 = point3D1;
                    p3s.forEach(point3D2 -> {
                        Point3D p2 = point3D2;
                        if (Point3D.distance(p1, p2) >= distMaxMinP1[0] && isInBound(p1) && isInBound(p2)) {
                            extremes[0][0] = p1;
                            extremes[0][1] = p2;
                            distMaxMinP1[0] = Point3D.distance(p1, p2);
                        }
                        if (Point3D.distance(p1, p2) <= distMaxMinP1[1] && isInBound(p1) && isInBound(p2)) {
                            extremes[1][0] = p1;
                            extremes[1][1] = p2;
                            distMaxMinP1[1] = Point3D.distance(p1, p2);
                            pointsCurrent.add(p2);
                        }
                    });
                });
                if (extremes[0][0] != null && extremes[0][1] != null && isInBound(extremes[0][0]) && isInBound(extremes[0][1])) {
                    lines.add(new LineSegment(extremes[0][0], extremes[0][1], new ColorTexture(r)));
                }
                // Calculer le segment AB qui approxime un maximum de points dans l'ensemble
                /*b = (yB − yA)/(xB − xA); a = (yA − bxA). :?yA = yMoyen, xA = xMoyen*/

                // Recouper les lignes par db min max


                /*CourbeParametriquePolynomialeBezier parametricCurve = new CourbeParametriquePolynomialeBezier();
                p3s.forEach(point3D -> parametricCurve.getCoefficients().getData1d().add(point3D));
                courbeParametriquePolynomialeBeziers[i[0]++] = parametricCurve;
*/

                if (pointsCurrent.size() > 2)
                    lines.add(new LineSegment(extremes[1][0], extremes[1][1], new ColorTexture(r)));

            });

            BufferedImage bLines = new BufferedImage(o.getColumns(), o.getLines(), BufferedImage.TYPE_INT_RGB);
            Graphics g = bLines.getGraphics();
            for (LineSegment line : lines) {
                g.setColor(Colors.random());
                if (line.getLength() >= 2) {
                    /*for (double c = 0.0; c <= 1.0; c += 1 / line.getLength()) {
                        Point3D pDraw = line.getOrigine().plus(
                                line.getOrigine().plus(line.getExtremite().moins(line.getOrigine().mult(c))));
                        int x = (int) (double) pDraw.getX();
                        int y = (int) (double) pDraw.getY();
                        if (isInBound(pDraw))
                            bLines.setRGB(x, y, line.texture().getColorAt(0, 0));
                    }*/

                    Point3D pDraw1 = line.getOrigine().plus(
                            line.getOrigine().plus(line.getExtremite().moins(line.getOrigine().mult(0.0))));
                    Point3D pDraw2 = line.getOrigine().plus(
                            line.getOrigine().plus(line.getExtremite().moins(line.getOrigine().mult(1.0))));
                    int x1 = (int) (double) pDraw1.getX();
                    int y1 = (int) (double) pDraw1.getY();
                    int x2 = (int) (double) pDraw2.getX();
                    int y2 = (int) (double) pDraw2.getY();
                    if (isInBound(pDraw1) && isInBound(pDraw2)) {
                        g.drawLine(x1, y1, x2, y2);
                    }
                }
            }

//            BufferedImage linesImg2 = new BufferedImage(o.getColumns(), o.getLines(), BufferedImage.TYPE_INT_RGB);
//
//
//            coefficients.forEach(doubles -> {
//                double a = doubles[0];
//                double b = doubles[1];
//                // y = ax+b
//                Point p1 = new Point(0, (int) b);
//                Point p2 = new Point((int) (-b / a), 0);
//                Graphics g2 = linesImg2.getGraphics();
//                g2.setColor(Colors.random());
//                g2.drawLine((int) p1.getX(), (int) p1.getY(), (int) p2.getX(), (int) p2.getY());
//            });
            points.forEach(new Consumer>() {
                @Override
                public void accept(List point3DS) {

                }
            });
            ImageIO.write(o.normalize(0.0, 1.0).getImage(), "jpg", new File(out.getAbsolutePath() + "-dotted.jpg"));
            //ImageIO.write(bLines, "jpg",
            //        new File(out.getAbsolutePath() + "-blines.jpg"));
//            ImageIO.write(linesImg2, "jpg",
//                    new File(out.getAbsolutePath() + "-lines-yAxB.jpg"));
            ImageIO.write(img3.normalize(0.0, 1.0).getImage(), "jpg",
                    new File(out.getAbsolutePath()));
            return true;
        } catch (
                IOException e) {
            e.printStackTrace();
            return false;
        }

    }

    private boolean isInBound(Point3D p1) {
        return p1.get(0) >= 0 && p1.get(0) < pixM.getColumns() && p1.get(1) >= 0 && p1.get(1) < pixM.getLines();
    }

    public void addTmp(double x, double y, double z) {
        listTmpX.add(x);
        listTmpY.add(y);
        listTmpZ.add(z);
    }

    public void removeTmp(int i) {
        listTmpX.remove(i);
        listTmpY.remove(i);
        listTmpZ.remove(i);
    }

    public void getTmp(int i) {
        px = listTmpX.get(i);
        py = listTmpY.get(i);
        pz = listTmpZ.get(i);
    }

    private void neighborhood(int i, int j, double valueAvg, double valueDiff, double valueMin) {
        listTmpX.clear();
        listTmpY.clear();
        listTmpZ.clear();
        listTmpCurve.clear();
        for (int x = 0; x < 3; x++) {
            for (int y = 0; y < 3; y++) {
                int x2 = i + (x - 1);
                int y2 = j + (y - 1);
                if (x2 != i && y2 != j) {
                    Point point = new Point(x2, y2);
                    px = point.getX();
                    py = point.getY();
                    pz = pixM.luminance((int) point.getX(), (int) point.getY());
                    if (pz >= valueAvg - valueDiff && pz <= valueAvg + valueDiff && pz > valueMin
                            && px >= 0.0 && px < pixM.getColumns() && py >= 0 && py < pixM.getLines()) {
                        addTmp(px, py, pz);
                        break;
                    }
                }
            }
        }
    }

    public double getDistMax() {
        return distMax;
    }

    public void setDistMax(double distMax) {
        this.distMax = distMax;
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy