ConsistencyChecksImageProvider.java

package com.proalpha.pds.checks;
 
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.util.List;
 
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.swt.graphics.Image;
 
import com.proalpha.pds.jaxb.DsDSConsistencyChecks.TtDSCheck;
import com.proalpha.pds.jaxb.DsDSConsistencyChecks.TtDSCheck.TtDSSubCheck;
import com.proalpha.pds.jaxb.DsDSConsistencyChecks.TtDSCheck.TtDSSubCheck.TtDSLogEntryRepos;
import com.proalpha.pds.jaxb.DsDSConsistencyChecks.TtDSCheck.TtDSSubCheck.TtDSLogEntrySrc;
 
class ConsistencyChecksImageProvider {
 
    private enum BugPriority {
        GREY(0, "/images/bug_grey.png"), RED(1, "/images/bug_red.png"), YELLOW(2, "/images/bug_yellow.png"),
        GREEN(3, "/images/bug_green.png");
 
        public static BugPriority getBugPriority(Integer value) {
            BugPriority[] priorities = values();
 
            for (int i = 0; i < priorities.length; i++) {
                if (priorities[i].getValue().equals(value)) {
                    return priorities[i];
                }
            }
 
            return GREY;
        }
 
        private Integer value;
        private String imagePath;
 
        private BugPriority(Integer value, String imagePath) {
            this.value = value;
            this.imagePath = imagePath;
        }
 
        public String getImagePath() {
            return imagePath;
        }
 
        public Integer getValue() {
            return value;
        }
    }
 
    public Image getDefaultImage() {
        return getImageByPriority(BugPriority.YELLOW);
    }
 
    public Image getImage(Object element) {
 
        Integer priority = 0;
 
        if (element instanceof TtDSSubCheck) {
            priority = getPriority((TtDSSubCheck) element);
        }
        if (element instanceof TtDSCheck) {
            priority = getPriority((TtDSCheck) element);
        }
 
        return getImageByPriority(BugPriority.getBugPriority(priority));
 
    }
 
    private Image getImageByPriority(BugPriority priority) {
        Image image = null;
        try {
            image = ImageDescriptor.createFromURL(getClass().getResource(priority.getImagePath()).toURI().toURL())
                    .createImage();
        } catch (MalformedURLException | URISyntaxException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return image;
    }
 
    private Integer getPriority(TtDSCheck element) {
        Integer priority = -1;
 
        for (TtDSSubCheck subCheck : element.getTtDSSubCheck()) {
 
            Integer tempPriority = getPriority(subCheck);
 
            if (priority == -1) {
                priority = tempPriority;
            } else if (tempPriority < priority) {
                priority = tempPriority;
            }
 
        }
 
        return priority;
    }
 
    private Integer getPriority(TtDSSubCheck subCheck) {
 
        Integer priority = 0;
        List<TtDSLogEntrySrc> entriesSrc = subCheck.getTtDSLogEntrySrc();
 
        if (entriesSrc.size() > 0) {
            priority = entriesSrc.get(0).getLogEntryPriority();
        } else {
            List<TtDSLogEntryRepos> entriesRepos = subCheck.getTtDSLogEntryRepos();
            if (entriesRepos.size() > 0) {
                priority = entriesRepos.get(0).getLogEntryPriority();
            }
        }
 
        return priority;
    }
 
}