FigureFactory.java

package husacct.graphics.domain.figures;


import java.awt.Color;
import java.util.HashSet;

import org.apache.log4j.Logger;

import husacct.common.dto.AbstractDTO;
import husacct.common.dto.DependencyDTO;
import husacct.common.dto.ModuleDTO;
import husacct.common.dto.ProjectDTO;
import husacct.common.dto.RuleDTO;
import husacct.common.dto.SoftwareUnitDTO;
import husacct.common.dto.UmlLinkDTO;
import husacct.common.dto.ViolationDTO;
import husacct.graphics.domain.decorators.ViolationsDecorator;

public final class FigureFactory {
	protected Logger	logger				= Logger.getLogger(FigureFactory.class);
	private final String PROJECT_TYPE		= "Project";

	public RelationFigure createRelationFigure_Dependency(DependencyDTO[] dependencyDTOs) {
		if (dependencyDTOs.length <= 0) throw new RuntimeException("No dependencies received. Cannot create a dependency figure.");
		return new RelationFigure("Dependency from " + dependencyDTOs[0].from + " to " + dependencyDTOs[0].to, RelationType.DEPENDENCY, Integer.toString(dependencyDTOs.length));
	}
	public RelationFigure createRelationFigure_UmlLink(UmlLinkDTO[] UmlLinkDTO) {
		if (UmlLinkDTO.length <= 0) throw new RuntimeException("No dependencies received. Cannot create a dependency figure.");

		RelationFigure UmlLinkFigure = new RelationFigure("Dependency from " + UmlLinkDTO[0].from + " to " + UmlLinkDTO[0].to, RelationType.fromString(UmlLinkDTO[0].type), Integer.toString(UmlLinkDTO.length));
		UmlLinkFigure.setComposite(UmlLinkDTO[0].isComposite);
		UmlLinkFigure.setMultiplicity();
		
		return UmlLinkFigure;
	}
	
	public RelationFigure createRelationFigure_DependencyWithViolations(DependencyDTO[] dependencyDTOs, ViolationDTO[] violationDTOs) {
		RelationFigure violatedRelationFigure = new RelationFigure("Violated dependency from " + violationDTOs[0].fromClasspath
						+ " to " + violationDTOs[0].toClasspath, RelationType.VIOLATION, violationDTOs.length + "/" + dependencyDTOs.length);
		violatedRelationFigure.addDecorator(createViolationsDecorator());
		return violatedRelationFigure;
	}

	public RelationFigure createRelationFigure_Rule(RuleDTO[] ruleDTOs) {
		if (ruleDTOs.length <= 0) throw new RuntimeException("No rules received. Cannot create a rule figure");
		return new RelationFigure("Rule from " + ruleDTOs[0].moduleFrom.logicalPath + " to " + ruleDTOs[0].moduleTo.logicalPath, RelationType.RULELINK, Integer.toString(ruleDTOs.length));
	}

	public RelationFigure createRelationFigure_RuleViolation(ViolationDTO[] violationDTOs) {
		HashSet<String> uniqueRuleViolations = new HashSet<>();
		for (ViolationDTO violationDTO : violationDTOs) {
			uniqueRuleViolations.add(violationDTO.ruleTypeKey);
		}
		RelationFigure violatedRelationFigure = new RelationFigure("Violated rule from " + violationDTOs[0].fromClasspath
						+ " to " + violationDTOs[0].toClasspath, RelationType.VIOLATION, ""+uniqueRuleViolations.size() + " (" +violationDTOs.length + ")");
		violatedRelationFigure.addDecorator(createViolationsDecorator());
		return violatedRelationFigure;
	}

	
	// May return null!
	public ModuleFigure createModuleFigure(AbstractDTO dto) {
		String type;
		String name;
		String uniqueName;
		if (dto != null) {
			if (dto instanceof ModuleDTO) {
				type = ((ModuleDTO) dto).type;
				name = ((ModuleDTO) dto).name;
				uniqueName = ((ModuleDTO) dto).logicalPath;
			} else if (dto instanceof SoftwareUnitDTO) {
				type = ((SoftwareUnitDTO) dto).type;
				name = ((SoftwareUnitDTO) dto).name;
				uniqueName = ((SoftwareUnitDTO) dto).uniqueName;
			} else if (dto instanceof ProjectDTO) {
				type = PROJECT_TYPE;
				name = ((ProjectDTO) dto).name;
				uniqueName = name;
			} else {
				logger.error("Unimplemented dto type!");
				return null;
			}
		} else {
			logger.error("Null value for DTO!");
			return null;
		}
		return new ModuleFigure(name, uniqueName, type);
	}
	
	public ParentFigure createParentFigure(String parentUniqueName, String type) {
		return new ParentFigure(parentUniqueName, type);
	}
	
	public ViolationsDecorator createViolationsDecorator() {
		return new ViolationsDecorator(Color.RED);
	}
}