bundles/org.simantics.issues.common/src/org/simantics/issues/common/ListModelIssuesBySeverity.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2013 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     Semantum Oy - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.issues.common;\r
  13 \r
  14 import gnu.trove.map.hash.THashMap;\r
  15 import gnu.trove.set.hash.THashSet;\r
  16 \r
  17 import java.util.ArrayList;\r
  18 import java.util.Collection;\r
  19 import java.util.Collections;\r
  20 import java.util.List;\r
  21 import java.util.Map;\r
  22 \r
  23 import org.simantics.Simantics;\r
  24 import org.simantics.db.ReadGraph;\r
  25 import org.simantics.db.Resource;\r
  26 import org.simantics.db.common.request.ObjectsWithType;\r
  27 import org.simantics.db.common.request.QuaternaryRead;\r
  28 import org.simantics.db.exception.DatabaseException;\r
  29 import org.simantics.db.layer0.adapter.Instances;\r
  30 import org.simantics.issues.Severity;\r
  31 import org.simantics.issues.common.preferences.IssuePrefs;\r
  32 import org.simantics.issues.ontology.IssueResource;\r
  33 import org.simantics.layer0.Layer0;\r
  34 \r
  35 /**\r
  36  * @author Tuukka Lehtonen\r
  37  * @see CountModelIssuesBySeverity\r
  38  */\r
  39 public class ListModelIssuesBySeverity extends QuaternaryRead<Resource, Boolean, Boolean, Severity, Map<Severity, List<Resource>>> {\r
  40 \r
  41     public ListModelIssuesBySeverity(Resource model, boolean onlyUnresolved, boolean respectVisibility, Severity minSeverity) {\r
  42         super(model, onlyUnresolved, respectVisibility, minSeverity);\r
  43     }\r
  44 \r
  45     @SuppressWarnings("unchecked")\r
  46     @Override\r
  47     public Map<Severity, List<Resource>> perform(ReadGraph graph) throws DatabaseException {\r
  48         Layer0 L0 = Layer0.getInstance(graph);\r
  49         IssueResource ISSUE = IssueResource.getInstance(graph);\r
  50 \r
  51         Severity[] keys = Severity.values();\r
  52         @SuppressWarnings("rawtypes")\r
  53         List[] values = new List[keys.length];\r
  54 \r
  55         Resource project = Simantics.getProjectResource();\r
  56         boolean showHidden = false;\r
  57         boolean showNormal = true;\r
  58         boolean showUser = true;\r
  59         if (parameter3 && project != null) {\r
  60             showHidden = IssuePrefs.showHiddenIssues(graph, project);\r
  61             showNormal = IssuePrefs.showNormalIssues(graph, project);\r
  62             showUser = IssuePrefs.showUserIssues(graph, project);\r
  63         }\r
  64 \r
  65         Instances issueIndex = graph.getPossibleAdapter(ISSUE.Issue, Instances.class);\r
  66         Collection<Resource> modelIssues = graph.syncRequest(new ObjectsWithType(parameter, L0.ConsistsOf, ISSUE.Issue));\r
  67         Collection<Resource> indexedIssues = issueIndex != null ? issueIndex.find(graph, parameter) : Collections.<Resource>emptyList();\r
  68         Collection<Resource> issues = !indexedIssues.isEmpty() ? new THashSet<Resource>(modelIssues.size() + indexedIssues.size()) : modelIssues;\r
  69         if (!indexedIssues.isEmpty()) {\r
  70             issues.addAll(modelIssues);\r
  71             issues.addAll(indexedIssues);\r
  72         }\r
  73 \r
  74         final boolean ignoreResolved = parameter2;\r
  75         for (Resource issue : issues) {\r
  76             if (ignoreResolved && graph.hasStatement(issue, ISSUE.Resolved))\r
  77                 continue;\r
  78             boolean hidden = graph.hasStatement(issue, ISSUE.Hidden);\r
  79             boolean user = graph.hasStatement(issue, ISSUE.UserIssue);\r
  80             boolean normal = !hidden && !user;\r
  81             if (!showHidden && hidden)\r
  82                 continue;\r
  83             if (!showUser && user)\r
  84                 continue;\r
  85             if (!showNormal && normal)\r
  86                 continue;\r
  87 \r
  88             Severity s = IssueUtils.toSeverity(ISSUE, graph.getPossibleObject(issue, ISSUE.Issue_HasSeverity));\r
  89             if (null != s && s.compareTo(parameter4) <= 0) {\r
  90                 int index = s.ordinal();\r
  91                 @SuppressWarnings("rawtypes")\r
  92                 List list = values[index];\r
  93                 if (list == null)\r
  94                     list = values[index] = new ArrayList<Resource>();\r
  95                 list.add(issue);\r
  96             }\r
  97         }\r
  98 \r
  99         THashMap<Severity, List<Resource>> result = new THashMap<Severity, List<Resource>>(values.length);\r
 100         for (int i = 0; i < values.length; ++i)\r
 101             if (values[i] != null)\r
 102                 result.put(keys[i], values[i]);\r
 103         return result;\r
 104     }\r
 105 \r
 106 }\r