D-Net

package eu.dnetlib.msro.worker;

import java.awt.image.BufferedImage;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import javax.imageio.ImageIO;

import org.apache.commons.codec.binary.Base64;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.springframework.beans.factory.annotation.Autowired;

import com.googlecode.sarasvati.GraphProcess;

import com.googlecode.sarasvati.visual.ProcessImageMapCreator;

import eu.dnetlib.common.services.AbstractBaseService;

import eu.dnetlib.msro.worker.ui.ProcessToImageMapHelper;

import eu.dnetlib.rmi.object.manager.ProcessImageDesc;

import eu.dnetlib.rmi.soap.ManagerWorkerService;

public class ManagerWorkerServiceImpl extends AbstractBaseService implements ManagerWorkerService {

	@Autowired

	private LocalWorkflowRegistry registry;

	private static final Log log = LogFactory.getLog(ManagerWorkerServiceImpl.class);

	@Override

	public ProcessImageDesc getProcessImageDesc(final String procId) {

		final GraphProcess proc = registry.findProcess(procId);

		byte[] imgBytes;

		String mapContent = "";

		try {

			if (proc != null) {

				final ByteArrayOutputStream bos = new ByteArrayOutputStream();

				try {

					final ProcessImageMapCreator imageMapCreator = new ProcessImageMapCreator(proc, new ProcessToImageMapHelper(procId));

					final BufferedImage image = imageMapCreator.getImage();

					ImageIO.write(image, "png", bos);

					imgBytes = bos.toByteArray();

					mapContent = imageMapCreator.getMapContents();

				} catch (Throwable e) {

					imgBytes = IOUtils.toByteArray(getClass().getResourceAsStream("/eu/dnetlib/msro/worker/ui/wf_problem.png"));

				} finally {

					bos.close();

				}

			} else {

				imgBytes = IOUtils.toByteArray(getClass().getResourceAsStream("/eu/dnetlib/msro/worker/ui/wf_not_found.png"));

			}

		} catch (IOException e) {

			log.error("Error generating wf image for process: " + procId, e);

			return null;

		}

		return new ProcessImageDesc("image/png", new String(Base64.encodeBase64(imgBytes)), mapContent);

	}

}