net.sourceforge.plantuml.klimt.creole.Fission Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of plantuml-mit Show documentation
PlantUML is a component that allows to quickly write diagrams from text.
There is a newer version: 1.2025.0
// THIS FILE HAS BEEN GENERATED BY A PREPROCESSOR.
package net.sourceforge.plantuml.klimt.creole;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Deque;
import java.util.List;
import java.util.Objects;

import net.sourceforge.plantuml.klimt.LineBreakStrategy;
import net.sourceforge.plantuml.klimt.creole.atom.AbstractAtom;
import net.sourceforge.plantuml.klimt.creole.atom.Atom;
import net.sourceforge.plantuml.klimt.drawing.UGraphic;
import net.sourceforge.plantuml.klimt.font.StringBounder;
import net.sourceforge.plantuml.klimt.geom.XDimension2D;

public class Fission {

	private final Stripe stripe;
	private final LineBreakStrategy maxWidth;

	public Fission(Stripe stripe, LineBreakStrategy maxWidth) {
		this.stripe = stripe;
		this.maxWidth = Objects.requireNonNull(maxWidth);
	}

	public List getSplitted(StringBounder stringBounder) {
		final double valueMaxWidth = Math.abs(maxWidth.getMaxWidth());
		if (valueMaxWidth == 0)
			return Arrays.asList(stripe);

		final List result = new ArrayList<>();
		StripeSimpleInternal line = new StripeSimpleInternal(false, stringBounder, stripe.getLHeader());
		result.add(line);

		final Deque all = new ArrayDeque<>();
		for (Atom atom : noHeader())
			for (Neutron n : atom.getNeutrons())
				all.addLast(n);
		if (all.peekLast().getType() != NeutronType.ZWSP_SEPARATOR)
			all.addLast(Neutron.zwspSeparator());

		while (all.size() > 0) {
			final Neutron current = all.removeFirst();
			if (current.getType() == NeutronType.ZWSP_SEPARATOR && line.getWidth() > valueMaxWidth) {
				all.addFirst(current);
				final List removed = line.slightyShorten();
				for (int i = removed.size() - 1; i >= 0; i--)
					all.addFirst(removed.get(i));

				line = new StripeSimpleInternal(true, stringBounder, blank(stripe.getLHeader()));
				result.add(line);
			} else
				line.addNeutron(current);
		}

		for (Stripe l : result)
			((StripeSimpleInternal) l).removeFinalSpaces();

		while (result.size() > 1 && ((StripeSimpleInternal) result.get(result.size() - 1)).isWhite())
			result.remove(result.size() - 1);

		return Collections.unmodifiableList(result);

	}

	private List noHeader() {
		final List atoms = stripe.getAtoms();
		if (stripe.getLHeader() == null)
			return atoms;

		return atoms.subList(1, atoms.size());
	}

	static class StripeSimpleInternal implements Stripe {

		private final boolean removeInitialSpaces;
		private final Atom header;
		private final List neutrons = new ArrayList<>();
		private final StringBounder stringBounder;
		private double width;

		private StripeSimpleInternal(boolean removeInitialSpaces, StringBounder stringBounder, Atom header) {
			this.removeInitialSpaces = removeInitialSpaces;
			this.stringBounder = stringBounder;
			this.header = header;
			if (header != null)
				width += header.calculateDimension(stringBounder).getWidth();
		}

		public double getWidth() {
			if (width == -1)
				throw new IllegalStateException();
//			double width = 0;
//			if (header != null)
//				width += header.calculateDimension(stringBounder).getWidth();
//			for (Neutron n : neutrons)
//				width += n.getWidth(stringBounder);
			return width;
		}

		@Override
		public String toString() {
			if (header != null)
				return header.toString() + " " + neutrons;
			return neutrons.toString();
		}

		public List slightyShorten() {
			if (neutrons.size() == 0)
				throw new IllegalStateException();

			final int lastZwsp = lastZwsp();
			if (lastZwsp == -1)
				return Collections.emptyList();

			this.width = -1;
			final List removed = new ArrayList(neutrons.subList(lastZwsp, neutrons.size()));
			while (neutrons.size() > lastZwsp)
				neutrons.remove(neutrons.size() - 1);

			return removed;

		}

		private boolean isWhite() {
			for (Neutron n : neutrons)
				if (n.getType() != NeutronType.ZWSP_SEPARATOR && n.getType() != NeutronType.SPACE)
					return false;
			return true;
		}

		private void removeFinalSpaces() {
			while (neutrons.size() > 0 && neutrons.get(0).getType() == NeutronType.ZWSP_SEPARATOR)
				neutrons.remove(0);
			while (neutrons.size() > 1
					&& (last().getType() == NeutronType.SPACE || last().getType() == NeutronType.ZWSP_SEPARATOR))
				neutrons.remove(neutrons.size() - 1);
		}

		private Neutron last() {
			return neutrons.get(neutrons.size() - 1);
		}

		private int lastZwsp() {
			for (int i = neutrons.size() - 1; i >= 0; i--)
				if (neutrons.get(i).getType() == NeutronType.ZWSP_SEPARATOR)
					return i;
			return -1;
		}

		public List getAtoms() {
			final List result = new ArrayList<>();
			if (header != null)
				result.add(header);

			for (Neutron n : neutrons)
				if (n.getType() != NeutronType.ZWSP_SEPARATOR) {
					if (removeInitialSpaces && result.size() == 0 && n.getType() == NeutronType.SPACE)
						continue;
					result.add(n.asAtom());
				}
			return Collections.unmodifiableList(result);
		}

		private void addNeutron(Neutron neutron) {
			if (width == -1)
				throw new IllegalStateException();

			if (neutron.getType() == NeutronType.ZWSP_SEPARATOR && this.neutrons.size() == 0)
				return;

			if (neutron.getType() == NeutronType.ZWSP_SEPARATOR && this.neutrons.size() > 0
					&& last().getType() == NeutronType.ZWSP_SEPARATOR)
				return;

			if (removeInitialSpaces && this.neutrons.size() == 0 && neutron.getType() == NeutronType.SPACE)
				return;

			this.neutrons.add(neutron);
			this.width += neutron.getWidth(stringBounder);
		}

		public Atom getLHeader() {
			return null;
		}

	}

	private static Atom blank(final Atom header) {
		if (header == null)
			return null;

		return new AbstractAtom() {

			public XDimension2D calculateDimension(StringBounder stringBounder) {
				return header.calculateDimension(stringBounder);
			}

			public double getStartingAltitude(StringBounder stringBounder) {
				return header.getStartingAltitude(stringBounder);
			}

			public void drawU(UGraphic ug) {
			}

		};
	}

}