ObjectDiagram.java

gehe zur Dokumentation dieser Datei

package de.lathanda.eos.gui.objectchart;
 
import java.awt.Color;
import java.awt.Dimension;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.util.LinkedList;
 
import javax.swing.JPanel;
 
import de.lathanda.eos.gui.diagram.Unit;
import de.lathanda.eos.vm.MemoryEntry;
import de.lathanda.eos.baseparser.AbstractProgram;
import de.lathanda.eos.baseparser.SystemType;
import de.lathanda.eos.baseparser.Type;
import de.lathanda.eos.config.Language;
import de.lathanda.eos.gui.diagram.Drawing;
 
public class ObjectDiagram extends JPanel {
    private static final long serialVersionUID = -8489967668090334257L;
    private LinkedList<Unit> units;
    private final Drawing d;
    private static final float SPACE = 5;
    public ObjectDiagram() {
        super();
        setBackground(Color.WHITE);
        units = new LinkedList<>();
        d = new Drawing();
    }
 
    @Override
    protected void paintComponent(Graphics g) {
        super.paintComponent(g);
        render((Graphics2D) g);
    }
 
    public BufferedImage export(float dpi) {
        if (units.isEmpty()) {
            return new BufferedImage(1, 1, BufferedImage.TYPE_INT_RGB);
        }
        Drawing drawing = new Drawing(dpi);
        Dimension dim = layout(drawing);
        BufferedImage image = new BufferedImage(dim.width, dim.height, BufferedImage.TYPE_INT_RGB);
        Graphics2D g = image.createGraphics();
        g.setBackground(Color.WHITE);
        g.clearRect(0, 0, dim.width, dim.height);
        drawing.init(g);
        render(drawing);
        layout(d);
        return image;
    }
 
    public void render(Graphics2D g) {
        d.init(g);
        render(d);
    }
 
    private Dimension layout(Drawing d) {
        units.forEach(p -> p.layout(d));
        float h = 0;
        float w = 0;
        for (Unit u : units) {
            u.setOffsetY(h);
            h = h + u.getHeight() + SPACE;
            if (w < u.getWidth()) {
                w = u.getWidth();
            }
        }
        h -= SPACE;
        return new Dimension(d.convertmm2pixel(w + SPACE * 2), d.convertmm2pixel(h + SPACE * 2));
    }
 
    private void render(Drawing d) {
        d.pushTransform();
        d.translate(SPACE, SPACE);
        d.setDrawWidth(0.5f);
        d.setColor(Color.BLACK);
        units.stream().forEachOrdered(p -> p.draw(d));
        d.popTransform();
    }
 
    public void setData(LinkedList<MemoryEntry> data, AbstractProgram program) {
        units.clear();
        for (MemoryEntry v : data) {
            Type type = SystemType.getInstanceByID(v.type);
            if (type != null) {
                units.add(Toolkit.create(v.name, Language.def.getClassLabel(type), v.data, type));              
            } else {
                type = program.getTypeByName(v.type);
                units.add(Toolkit.create(v.name, v.type, v.data, type));
            }
        }
        setPreferredSize(layout(d));
        revalidate();
        repaint();
    }
 
}