one.empty3.apps.morph.DataModel Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of empty3-library-3d Show documentation
Show all versions of empty3-library-3d Show documentation
3D rendering engine. Plus modeling. Expected glsl textures 3d and 2d rendering
/*
* 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.apps.morph;
import one.empty3.feature.app.replace.javax.imageio.ImageIO;
import one.empty3.library.Point3D;
import one.empty3.library.StructureMatrix;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.*;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.zip.*;
public class DataModel {
protected File file;
private MorphUI morphUI;
private BufferedImage[] bis = new BufferedImage[2];
private File[] imagesFiles = new File[2];
private StructureMatrix[] grids = new StructureMatrix[4];
private Properties[] properties = new Properties[2];
public DataModel(MorphUI mui) {
this.morphUI = mui;
}
public void saveFile(ZipOutputStream zipOut, FileOutputStream fos,
File tmpFile, String filenameInZip) throws IOException {
FileInputStream fis = new FileInputStream(tmpFile);
ZipEntry zipEntry = new ZipEntry(filenameInZip);
zipOut.putNextEntry(zipEntry);
byte[] bytes = new byte[1024];
int length;
while ((length = fis.read(bytes)) >= 0) {
zipOut.write(bytes, 0, length);
}
fis.close();
}
public File writeTextTmp() throws IOException {
return File.createTempFile("tmp", "txt");
}
public void save(File file) {
this.file = file;
save();
}
public void save() {
if (file == null) {
return;
}
try {
if (morphUI != null && morphUI.getImageControls1() != null && morphUI.getImageControls2() != null) {
File outputZip = file;
FileOutputStream fos = new FileOutputStream(outputZip);
ZipOutputStream zipOut = new ZipOutputStream(fos);
File tmp = writeTextTmp();
Properties properties1 = new Properties();
Properties properties2 = new Properties();
Properties[] properties3 = {properties1, properties2};
ImageControls[] controls = new ImageControls[]{morphUI.getImageControls1(), morphUI.getImageControls2()};
for (int i = 0; i < controls.length; i++) {
ImageControls imageControls = controls[i];
Properties properties = properties3[i];
properties.put("xGrid", "" + imageControls.getXgrid());
properties.put("yGrid", "" + imageControls.getYgrid());
properties.put("resX", "" + imageControls.getResX());
properties.put("resX", "" + imageControls.getResY());
properties.put("noDeformation", imageControls.getPointView().getCheckBoxNoDeformation().isSelected() ? "1" : "0");
properties.put("morphing", imageControls.getPointView().getCheckBoxMorphing().isSelected() ? "1" : "0");
properties.put("uv", imageControls.getPointView().getCheckBoxUv().isSelected() ? "1" : "0");
}
properties3[0].store(new FileOutputStream(tmp), "ImageControls1 values");
saveFile(zipOut, fos, tmp, "params1.properties");
tmp = writeTextTmp();
properties3[1].store(new FileOutputStream(tmp), "ImageControls2 values");
saveFile(zipOut, fos, tmp, "params2.properties");
tmp = writeTextTmp();
saveObjectArray2d(tmp, morphUI.getImageControls1().getGrid());
saveFile(zipOut, fos, tmp, "gridXY1.txt");
tmp = writeTextTmp();
saveObjectArray2d(tmp, morphUI.getImageControls2().getGrid());
saveFile(zipOut, fos, tmp, "gridXY2.txt");
tmp = writeTextTmp();
saveObjectArray2d(tmp, morphUI.getImageControls1().getGridUv());
saveFile(zipOut, fos, tmp, "gridUV1.txt");
tmp = writeTextTmp();
saveObjectArray2d(tmp, morphUI.getImageControls2().getGridUv());
saveFile(zipOut, fos, tmp, "gridUV2.txt");
tmp = writeTextTmp();
ImageIO.write(morphUI.getImageControls1().getImage(), "jpg", tmp);
saveFile(zipOut, fos, tmp, "image1.jpg");
ImageIO.write(morphUI.getImageControls2().getImage(), "jpg", tmp);
saveFile(zipOut, fos, tmp, "image2.jpg");
zipOut.close();
fos.close();
}
} catch (Exception ex) {}
}
private void saveObjectArray2d(File tmp, int xgrid) {
}
private void saveObjectArray2d(File tmp, StructureMatrix grid) {
try {
PrintWriter printWriter = new PrintWriter(tmp);
printWriter.println(grid.getData2d().size());
for (List point3DS : grid.getData2d()) {
printWriter.println(point3DS.size());
for (Point3D point3D : point3DS) {
for (int i = 0; i < 3; i++) {
printWriter.println(point3D.get(i));
}
}
}
printWriter.close();
} catch (FileNotFoundException e) {
throw new RuntimeException(e);
}
}
public DataModel load(File dataFile) throws IOException, ClassNotFoundException {
ZipFile zipIn = null;
ZipInputStream zipInputStream;
try {
zipIn = new ZipFile(dataFile);
zipInputStream = new ZipInputStream(new FileInputStream(dataFile));
} catch (ZipException ex) {
ex.printStackTrace();
return null;
}
file = dataFile;
Iterator entries = zipIn.entries().asIterator();
while (entries.hasNext()) {
ZipEntry zipEntry = entries.next();
String name = zipEntry.getName();
BufferedInputStream inputStream = new BufferedInputStream(zipIn.getInputStream(zipEntry));
int i = 0;
byte[] data = new byte[(int) zipEntry.getSize()];
while (i < zipEntry.getSize()) {
int read = inputStream.read(data, i, data.length);
i += read;
}
if (name.endsWith(".jpg")) {
File tmpFile = File.createTempFile("tmp_open_file_" + UUID.randomUUID(), ".jpg");
FileOutputStream fileOutputStream2 = new FileOutputStream(tmpFile);
fileOutputStream2.write(data);
fileOutputStream2.close();
BufferedImage read = ImageIO.read(tmpFile);
if (read == null) {
Logger.getAnonymousLogger().log(Level.SEVERE, "Fichier image endommagé ou non reconnu" +
" ou erreur lors de la lecture du ZIP");
System.exit(-1);
}
switch (name) {
case "image1.jpg":
imagesFiles[0] = tmpFile;
bis[0] = read;
break;
case "image2.jpg":
imagesFiles[1] = tmpFile;
bis[1] = read;
break;
default:
throw new IllegalStateException("Unexpected value: " + name);
}
} else {
StructureMatrix structureMatrix = new StructureMatrix(2, Point3D.class);
if (name.toLowerCase().startsWith("grid")) {
loadObjectString(loadZipEntry(zipIn, zipEntry), structureMatrix);
}
switch (name) {
case "gridXY1.txt":
grids[0] = (structureMatrix);
break;
case "gridXY2.txt":
grids[1] = (structureMatrix);
break;
case "gridUV1.txt":
grids[2] = (structureMatrix);
break;
case "gridUV2.txt":
grids[3] = (structureMatrix);
break;
case "params1.properties":
case "params2.properties":
int c = 0;
if (name.equals("params2.properties"))
c = 1;
String s = loadZipEntry(zipIn, zipEntry);
properties[c] = new Properties();
properties[c].load(new StringReader(s));
break;
}
}
}
morphUI.chooseFile1(imagesFiles[0]);
morphUI.chooseFile2(imagesFiles[1]);
morphUI.getImageControls1().setGrid(grids[0]);
morphUI.getImageControls2().setGrid(grids[1]);
morphUI.getImageControls1().setGridUv(grids[2]);
morphUI.getImageControls2().setGridUv(grids[3]);
for (int i = 0; i < grids.length; i++) {
System.out.println(grids[i].getData2d().toString());
}
for (int i = 0; i < imagesFiles.length; i++) {
System.out.println(imagesFiles[i].getName());
}
ImageControls[] controls = new ImageControls[]{morphUI.getImageControls1(), morphUI.getImageControls2()};
for (int c = 0; c < controls.length; c++) {
ImageControls imageControls = controls[c];
if (properties[c] != null) {
if (properties[c].get("xGrid") != null)
imageControls.setXgrid(Integer.parseInt((String) properties[c].get("xGrid")));
if (properties[c].get("yGrid") != null)
imageControls.setYgrid(Integer.parseInt((String) properties[c].get("yGrid")));
if (properties[c].get("resX") != null)
imageControls.setResX(Integer.parseInt((String) properties[c].get("resX")));
if (properties[c].get("resY") != null)
imageControls.setResY(Integer.parseInt((String) properties[c].get("resY")));
if (properties[c].get("noDeformation") != null)
imageControls.getPointView().getCheckBoxNoDeformation().setSelected(
Integer.parseInt((String) properties[c].get("noDeformation")) == 1);
if (properties[c].get("uv") != null)
imageControls.getPointView().getCheckBoxUv().setSelected(
Integer.parseInt((String) properties[c].get("uv")) == 1);
if (properties[c].get("morphing") != null)
imageControls.getPointView().getCheckBoxMorphing().setSelected(
Integer.parseInt((String) properties[c].get("morphing")) == 1);
}
}
return this;
}
private String loadZipEntry(ZipFile zipIn, ZipEntry zipEntry) {
try {
BufferedInputStream inputStream = new BufferedInputStream(zipIn.getInputStream(zipEntry));
int i = 0;
byte[] data = new byte[(int) zipEntry.getSize()];
while (i < zipEntry.getSize()) {
int read = inputStream.read(data, i, data.length);
i += read;
}
String s = new String(data, "UTF-8");
return s;
} catch (IOException e) {
throw new RuntimeException(e);
}
}
private void loadObjectString(String txt, StructureMatrix structureMatrix) {
String[] split = txt.split("\n");
int j = 0;
int length = 0;
int i = 1;
int errors = 0;
while (i < split.length) {
try {
Logger.getAnonymousLogger().log(Level.INFO, split[i]);
length = Integer.parseInt(split[i].trim());
} catch (NumberFormatException ex) {
errors++;
}
i++;
j = 0;
structureMatrix.getData2d().add(new ArrayList<>());
while (j < length && i < split.length) {
double d1 = Double.parseDouble(split[i].trim());
i++;
double d2 = Double.parseDouble(split[i].trim());
i++;
double d3 = Double.parseDouble(split[i].trim());
i++;
structureMatrix.getData2d().get(structureMatrix.getData2d().size() - 1).add(new Point3D(d1, d2, d3));
j++;
}
}
System.out.println(errors);
}
public File getFile() {
return file;
}
public void setFile(File file) {
this.file = file;
}
}