Add .blend export

src/main/java/com/replaymod/render/blend/BlendState.java

@@ -0,0 +1,170 @@
+package com.replaymod.render.blend;
+
+import com.replaymod.render.blend.data.DScene;
+import com.replaymod.render.blend.data.Serializer;
+import com.replaymod.render.blend.exporters.ChunkExporter;
+import com.replaymod.render.blend.exporters.EntityExporter;
+import com.replaymod.render.blend.exporters.ItemExporter;
+import com.replaymod.render.blend.exporters.ParticlesExporter;
+import com.replaymod.render.blend.exporters.RenderState;
+import com.replaymod.render.blend.exporters.TileEntityExporter;
+import net.minecraft.crash.CrashReport;
+import net.minecraft.crash.CrashReportCategory;
+import net.minecraft.util.ReportedException;
+import org.apache.commons.io.output.NullOutputStream;
+import org.blender.utils.BlenderFactory;
+import org.cakelab.blender.io.BlenderFile;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.PrintStream;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.NoSuchElementException;
+
+import static com.replaymod.render.ReplayModRender.LOGGER;
+
+// Note:
+// - Chunk exporter currently assumes VBO are enabled and in use
+public class BlendState implements Exporter {
+    private static BlendState currentState;
+
+    public static void setState(BlendState state) {
+        currentState = state;
+    }
+
+    public static BlendState getState() {
+        return currentState;
+    }
+
+    private final List<Exporter> exporters = new ArrayList<>();
+    private final BlenderFile blenderFile;
+    private final BlenderFactory factory;
+    private final DScene scene = new DScene();
+    private final BlendMaterials materials = new BlendMaterials();
+
+    public BlendState(File file) throws IOException {
+        this.blenderFile = BlenderFactory.newBlenderFile(file);
+        this.factory = new BlenderFactory(blenderFile);
+
+        RenderState renderState = new RenderState(this);
+        register(renderState);
+        register(new ChunkExporter());
+        register(new EntityExporter(renderState));
+        register(new TileEntityExporter(renderState));
+        register(new ParticlesExporter(renderState));
+        register(new ItemExporter(renderState));
+    }
+
+    public void register(Exporter exporter) {
+        exporters.add(exporter);
+    }
+
+    public <T extends Exporter> T get(Class<T> clazz) {
+        for (Exporter exporter : exporters) {
+            if (clazz.isInstance(exporter)) {
+                return clazz.cast(exporter);
+            }
+        }
+        throw new NoSuchElementException("No exporter of type " + clazz);
+    }
+
+    @Override
+    public void setup() {
+        for (Exporter exporter : exporters) {
+            try {
+                exporter.setup();
+            } catch (IOException e) {
+                CrashReport report = CrashReport.makeCrashReport(e, "Setup of blend exporter");
+                CrashReportCategory category = report.makeCategory("Exporter");
+                category.addCrashSectionCallable("Exporter", exporter::toString);
+                throw new ReportedException(report);
+            }
+        }
+    }
+
+    @Override
+    public void tearDown() {
+        for (Exporter exporter : exporters) {
+            try {
+                exporter.tearDown();
+            } catch (IOException e) {
+                CrashReport report = CrashReport.makeCrashReport(e, "Tear down of blend exporter");
+                CrashReportCategory category = report.makeCategory("Exporter");
+                category.addCrashSectionCallable("Exporter", exporter::toString);
+                throw new ReportedException(report);
+            }
+        }
+
+        try {
+            Serializer serializer = new Serializer();
+            this.scene.serialize(serializer);
+            // TODO pre-configure render settings serializer.writeMajor(new Object(), new DId(BlockCodes.ID_REND))
+
+            //noinspection UseOfSystemOutOrSystemErr
+            PrintStream sysout = System.out;
+            try {
+                System.setOut(new PrintStream(new NullOutputStream()));
+                blenderFile.write();
+            } finally {
+                System.setOut(sysout);
+            }
+        } catch (IOException e) {
+            LOGGER.error("Exception writing blend file: ", e);
+        } finally {
+            try {
+                blenderFile.close();
+            } catch (IOException e) {
+                LOGGER.error("Exception closing blend file: ", e);
+            }
+        }
+    }
+
+    @Override
+    public void preFrame(int frame) {
+        for (Exporter exporter : exporters) {
+            try {
+            exporter.preFrame(frame);
+            } catch (IOException e) {
+                CrashReport report = CrashReport.makeCrashReport(e, "Pre frame of blend exporter");
+                CrashReportCategory category = report.makeCategory("Exporter");
+                category.addCrashSectionCallable("Exporter", exporter::toString);
+                category.addCrashSection("Frame", frame);
+                throw new ReportedException(report);
+            }
+        }
+    }
+
+    @Override
+    public void postFrame(int frame) {
+        for (Exporter exporter : exporters) {
+            try {
+                exporter.postFrame(frame);
+            } catch (IOException e) {
+                CrashReport report = CrashReport.makeCrashReport(e, "Post frame of blend exporter");
+                CrashReportCategory category = report.makeCategory("Exporter");
+                category.addCrashSectionCallable("Exporter", exporter::toString);
+                category.addCrashSection("Frame", frame);
+                throw new ReportedException(report);
+            }
+        }
+
+        scene.endFrame = frame;
+    }
+
+    public BlenderFile getBlenderFile() {
+        return blenderFile;
+    }
+
+    public BlenderFactory getFactory() {
+        return factory;
+    }
+
+    public DScene getScene() {
+        return scene;
+    }
+
+    public BlendMaterials getMaterials() {
+        return materials;
+    }
+}