org.stathissideris.ascii2image.graphics.CompositeDiagramShape 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
Show all versions of plantuml-mit Show documentation
PlantUML is a component that allows to quickly write diagrams from text.
// THIS FILE HAS BEEN GENERATED BY A PREPROCESSOR.
package org.stathissideris.ascii2image.graphics;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import org.stathissideris.ascii2image.core.DebugUtils;
import org.stathissideris.ascii2image.text.CellSet;
import org.stathissideris.ascii2image.text.TextGrid;
/**
*
* @author Efstathios Sideris
*/
public class CompositeDiagramShape extends DiagramComponent {
private static final boolean DEBUG = false;
private ArrayList shapes = new ArrayList();
public static void main(String[] args) {
}
public static DiagramComponent createFromBoundaryCells(
final TextGrid grid,
final CellSet boundaryCells,
final int cellWidth,
final int cellHeight) {
return createOpenFromBoundaryCells(
grid,
boundaryCells,
cellWidth, cellHeight,
false);
}
public static DiagramComponent createOpenFromBoundaryCells(
final TextGrid grid,
final CellSet boundaryCells,
final int cellWidth,
final int cellHeight,
boolean allRound) {
if(boundaryCells.getType(grid) != CellSet.TYPE_OPEN) throw new IllegalArgumentException("This shape is closed and cannot be handled by this method");
if(boundaryCells.size() == 0) return null;
CompositeDiagramShape compositeShape = new CompositeDiagramShape();
TextGrid workGrid = new TextGrid(grid.getWidth(), grid.getHeight());
grid.copyCellsTo(boundaryCells, workGrid);
if(DEBUG) {
System.out.println("Making composite shape from grid:");
workGrid.printDebug();
}
CellSet visitedCells = new CellSet();
List shapes = new ArrayList(100);
for(TextGrid.Cell cell : boundaryCells) {
if(workGrid.isLinesEnd(cell)) {
CellSet nextCells = workGrid.followCell(cell);
shapes.addAll(growEdgesFromCell(workGrid, cellWidth, cellHeight, allRound, nextCells.getFirst(), cell, visitedCells));
break;
}
}
//dashed shapes should "infect" the rest of the shapes
boolean dashedShapeExists = false;
for(DiagramShape shape : shapes)
if(shape.isStrokeDashed())
dashedShapeExists = true;
for(DiagramShape shape : shapes) {
if(dashedShapeExists) shape.setStrokeDashed(true);
compositeShape.addToShapes(shape);
}
return compositeShape;
}
private static List growEdgesFromCell(
TextGrid workGrid,
final int cellWidth,
final int cellHeight,
boolean allRound,
TextGrid.Cell cell,
TextGrid.Cell previousCell,
CellSet visitedCells) {
List result = new ArrayList(50);
visitedCells.add(previousCell);
DiagramShape shape = new DiagramShape();
shape.addToPoints(makePointForCell(previousCell, workGrid, cellWidth, cellHeight, allRound));
if(DEBUG) System.out.println("point at "+previousCell+" (call from line: "+DebugUtils.getLineNumber()+")");
if(workGrid.cellContainsDashedLineChar(previousCell)) shape.setStrokeDashed(true);
boolean finished = false;
int nb = 0;
while(!finished) {
// https://github.com/plantuml/plantuml/issues/1036
nb++;
if (nb > 1000)
return result;
visitedCells.add(cell);
if(workGrid.isPointCell(cell)) {
if(DEBUG) System.out.println("point at "+cell+" (call from line: "+DebugUtils.getLineNumber()+")");
shape.addToPoints(makePointForCell(cell, workGrid, cellWidth, cellHeight, allRound));
}
if(workGrid.cellContainsDashedLineChar(cell)) shape.setStrokeDashed(true);
if(workGrid.isLinesEnd(cell)){
finished = true;
if(DEBUG) System.out.println("finished shape");
}
CellSet nextCells = workGrid.followCell(cell, previousCell);
if(nextCells.size() == 1) {
previousCell = cell;
cell = (TextGrid.Cell) nextCells.getFirst();
} else if(nextCells.size() > 1) {//3- or 4- way intersection
finished = true;
if(DEBUG) System.out.println("finished shape");
for(TextGrid.Cell nextCell : nextCells)
result.addAll(growEdgesFromCell(workGrid, cellWidth, cellHeight, allRound, nextCell, cell, visitedCells));
}
}
result.add(shape);
return result;
}
/**
* Returns a new diagram component with the lines of
* this CompositeDiagramShape connected. It can a composite
* or simple shape
*
* @return
*/
public DiagramComponent connectLines(){
CompositeDiagramShape result = new CompositeDiagramShape();
//find all lines
ArrayList lines = new ArrayList();
Iterator it = shapes.iterator();
while(it.hasNext()){
DiagramShape shape = (DiagramShape) it.next();
if(shape.getPoints().size() == 2){
lines.add(shape);
}
}
it = lines.iterator();
while(it.hasNext()){
DiagramShape line1 = (DiagramShape) it.next();
Iterator it2 = lines.iterator();
while(it2.hasNext()){
DiagramShape line2 = (DiagramShape) it.next();
ShapePoint commonPoint = null;
ShapePoint line1UncommonPoint = null;
ShapePoint line2UncommonPoint = null;
if(line1.getPoint(0).equals(line2.getPoint(0))){
commonPoint = line1.getPoint(0);
line1UncommonPoint = line1.getPoint(1);
line2UncommonPoint = line2.getPoint(1);
}
if(line1.getPoint(0).equals(line2.getPoint(1))){
commonPoint = line1.getPoint(0);
line1UncommonPoint = line1.getPoint(1);
line2UncommonPoint = line2.getPoint(0);
}
if(line1.getPoint(1).equals(line2.getPoint(0))){
commonPoint = line1.getPoint(1);
line1UncommonPoint = line1.getPoint(0);
line2UncommonPoint = line2.getPoint(1);
}
if(line1.getPoint(1).equals(line2.getPoint(1))){
commonPoint = line1.getPoint(1);
line1UncommonPoint = line1.getPoint(0);
line2UncommonPoint = line2.getPoint(0);
}
if(commonPoint != null){
}
}
}
return result;
}
public void connectEndsToAnchors(TextGrid grid, Diagram diagram){
Iterator it = shapes.iterator();
while (it.hasNext()) {
DiagramShape shape = (DiagramShape) it.next();
if(!shape.isClosed()){
shape.connectEndsToAnchors(grid, diagram);
}
}
}
private static DiagramShape makeLine(TextGrid grid, TextGrid.Cell start, TextGrid.Cell end, int cellWidth, int cellHeight){
DiagramShape line = new DiagramShape();
if(grid.isHorizontalLine(start)){
if(start.isWestOf(end)){
line.addToPoints(new ShapePoint(
Diagram.getCellMinX(start, cellWidth),
Diagram.getCellMidY(start, cellHeight)));
} else {
line.addToPoints(new ShapePoint(
Diagram.getCellMaxX(start, cellWidth),
Diagram.getCellMidY(start, cellHeight)));
}
} else if(grid.isVerticalLine(start)){
if(start.isNorthOf(end)){
line.addToPoints(new ShapePoint(
Diagram.getCellMidX(start, cellWidth),
Diagram.getCellMinY(start, cellHeight)));
} else {
line.addToPoints(new ShapePoint(
Diagram.getCellMidX(start, cellWidth),
Diagram.getCellMaxY(start, cellHeight)));
}
} else { //corner
if(DEBUG) System.out.println("Corner");
int type = (grid.isRoundCorner(start))?ShapePoint.TYPE_ROUND:ShapePoint.TYPE_NORMAL;
line.addToPoints(new ShapePoint(
Diagram.getCellMidX(start, cellWidth),
Diagram.getCellMidY(start, cellHeight),
type));
}
if(grid.isHorizontalLine(end)){
if(start.isWestOf(start)){
line.addToPoints(new ShapePoint(
Diagram.getCellMinX(end, cellWidth),
Diagram.getCellMidY(end, cellHeight)));
} else {
line.addToPoints(new ShapePoint(
Diagram.getCellMaxX(end, cellWidth),
Diagram.getCellMidY(end, cellHeight)));
}
} else if(grid.isVerticalLine(end)){
if(start.isNorthOf(start)){
line.addToPoints(new ShapePoint(
Diagram.getCellMidX(end, cellWidth),
Diagram.getCellMinY(end, cellHeight)));
} else {
line.addToPoints(new ShapePoint(
Diagram.getCellMidX(end, cellWidth),
Diagram.getCellMaxY(end, cellHeight)));
}
} else { //corner
int type = (grid.isRoundCorner(end))?ShapePoint.TYPE_ROUND:ShapePoint.TYPE_NORMAL;
if(DEBUG) System.out.println("Corner");
line.addToPoints(new ShapePoint(
Diagram.getCellMidX(end, cellWidth),
Diagram.getCellMidY(end, cellHeight),
type));
}
return line;
}
public void addToShapes(DiagramShape shape){
shapes.add(shape);
}
private Iterator getShapesIterator(){
return shapes.iterator();
}
public void scale(float factor){
Iterator it = getShapesIterator();
while(it.hasNext()){
DiagramShape shape = (DiagramShape) it.next();
shape.scale(factor);
}
}
/**
* @return
*/
public ArrayList getShapes() {
return shapes;
}
}