activiti当前任务高亮（解决乱码问题）_JAVA_编程开发_程序员俱乐部

class="java" name="code">package com.winse.activiti.util;
 
import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Paint;
import java.awt.Stroke;
import java.awt.geom.RoundRectangle2D;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
 
import javax.imageio.ImageIO;
 
import org.activiti.engine.RepositoryService;
import org.activiti.engine.RuntimeService;
import org.activiti.engine.impl.RepositoryServiceImpl;
import org.activiti.engine.impl.persistence.entity.ProcessDefinitionEntity;
import org.activiti.engine.impl.pvm.process.ActivityImpl;
import org.activiti.engine.runtime.ProcessInstance;
import org.apache.commons.io.FilenameUtils;
 
 
/**
 * @see org.activiti.engine.impl.bpmn.diagram.ProcessDiagramGenerator
 * @see org.activiti.engine.impl.bpmn.diagram.ProcessDiagramCanvas
 * 
 * @author winse
 */
public class ProcessDefinitionGeneratorEx {
  
  private RepositoryService repositoryService;
  private RuntimeService runtimeService;
 
	public ProcessDefinitionGeneratorEx() {
	}
 
	public ProcessDefinitionGeneratorEx(RepositoryService repositoryService, RuntimeService runtimeService) {
		this.repositoryService = repositoryService;
		this.runtimeService = runtimeService;
	}
 
	public RepositoryService getRepositoryService() {
		return repositoryService;
	}
 
	public RuntimeService getRuntimeService() {
		return runtimeService;
	}
 
	private static Color HIGHLIGHT_COLOR = Color.RED;
	private static Stroke THICK_TASK_BORDER_STROKE = new BasicStroke(3.0f);
 
	public InputStream generateDiagramWithHighLight(String processInstanceId) throws IOException {
		ProcessInstance processInstance = runtimeService.createProcessInstanceQuery()
				.processInstanceId(processInstanceId).singleResult();
		String processDefinitionId = processInstance.getProcessDefinitionId();
 
		ProcessDefinitionEntity definition = (ProcessDefinitionEntity) ((RepositoryServiceImpl) repositoryService)
				.getDeployedProcessDefinition(processDefinitionId);
		String diagramResourceName = definition.getDiagramResourceName();
		String deploymentId = definition.getDeploymentId();
		InputStream originDiagram = repositoryService.getResourceAsStream(deploymentId, diagramResourceName);
		BufferedImage image = ImageIO.read(originDiagram);
 
		List<String> activeActivityIds = ActivityUtil.getActiveActivityIds(runtimeService, processInstanceId);
		List<ActivityImpl> definitionActivities = ActivityUtil.getFlatAllActivities(definition);
		for (ActivityImpl activity : definitionActivities) {
			if (activeActivityIds.contains(activity.getId())) {
				decorate(image, //
						activity.getX(), activity.getY(), activity.getWidth(), activity.getHeight());
			}
		}
 
		ByteArrayOutputStream out = new ByteArrayOutputStream();
		String formatName = getDiagramExtension(diagramResourceName);
		ImageIO.write(image, formatName, out);
		return new ByteArrayInputStream(out.toByteArray());
 
	}
	
	/** 
	@Test
	public void testDecorete() throws FileNotFoundException, IOException {
		String diagramResourceName = "C:/Users/Winseliu/Desktop/03.jpg";
		BufferedImage image = ImageIO.read(new FileInputStream(diagramResourceName));
		decorate(image, 100, 100, 100, 100);
		FileOutputStream out = new FileOutputStream("C:/Users/Winseliu/Desktop/03-hh.jpg");
		try {
			ImageIO.write(image, getDiagramExtension(diagramResourceName), out);
		} finally {
			IoUtil.closeSilently(out);
		}
	}
	//*/
 
	private static void decorate(BufferedImage image, int x, int y, int width, int height) {
		Graphics2D g = image.createGraphics();
		try {
			drawHighLight(x, y, width, height, g);
		} finally {
			g.dispose();
		}
	}
 
	private static String getDiagramExtension(String diagramResourceName) {
//		return diagramResourceName.substring(diagramResourceName.lastIndexOf(".") + 1);
		return FilenameUtils.getExtension(diagramResourceName);
	}
 
	private static void drawHighLight(int x, int y, int width, int height, Graphics2D g) {
		Paint originalPaint = g.getPaint();
		Stroke originalStroke = g.getStroke();
 
		g.setPaint(HIGHLIGHT_COLOR);
		g.setStroke(THICK_TASK_BORDER_STROKE);
 
		RoundRectangle2D rect = new RoundRectangle2D.Double(x, y, width, height, 20, 20);
		g.draw(rect);
 
		g.setPaint(originalPaint);
		g.setStroke(originalStroke);
	}
 
}

package com.winse.activiti.util;
 
import java.util.ArrayList;
import java.util.List;
 
import org.activiti.engine.RuntimeService;
import org.activiti.engine.impl.persistence.entity.ProcessDefinitionEntity;
import org.activiti.engine.impl.pvm.process.ActivityImpl;
 
/**
 * 
 * @author winse
 * 
 */
public class ActivityUtil {
 
  public static List<String> getActiveActivityIds(RuntimeService runtimeService, String processInstanceId) {
		// 会通过id -> parent_id -> parent_id -> ... 循环找出所有的执行中的子流程
		return runtimeService.getActiveActivityIds(processInstanceId);
	}
 
	public static List<ActivityImpl> getFlatAllActivities(ProcessDefinitionEntity processDefinition) {
		List<ActivityImpl> result = new ArrayList<ActivityImpl>();
		flattenActivities(result, processDefinition.getActivities());
		return result;
	}
 
	private static void flattenActivities(List<ActivityImpl> container, List<ActivityImpl> ancestors) {
		if (ancestors.size() > 0) {
			for (ActivityImpl activity : ancestors) {
				flattenActivities(container, activity.getActivities());
			}
			container.addAll(ancestors);
		}
	}
 
}