D-Net

package eu.dnetlib.msro.worker;

import java.awt.Image;

import java.io.IOException;

import javax.imageio.ImageIO;

import javax.xml.ws.soap.MTOM;

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.soap.ManagerWorkerService;

@MTOM

public class ManagerWorkerServiceImpl extends AbstractBaseService implements ManagerWorkerService {

	@Autowired

	private LocalWorkflowRegistry registry;

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

	@Override

	public Image getProcessImage(final String procId) {

		final GraphProcess proc = registry.findProcess(procId);

		try {

			if (proc != null) {

				try {

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

					return imageMapCreator.getImage();

				} catch (Throwable e) {

					return ImageIO.read(getClass().getResourceAsStream("/eu/dnetlib/msro/worker/ui/wf_problem.png"));

				}

			} else {

				return ImageIO.read(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;

		}

	}

	@Override

	public String getProcessImageMap(final String procId) {

		final GraphProcess proc = registry.findProcess(procId);

		if (proc != null) {

			try {

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

				return imageMapCreator.getMapContents();

			} catch (Throwable e) {}

		}

		return "";

	}

}