bundles/org.simantics.layer0.utils/src/org/simantics/layer0/utils/binaryPredicates/UnionPredicate.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 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  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.layer0.utils.binaryPredicates;\r
  13 \r
  14 import java.util.Arrays;\r
  15 import java.util.Collection;\r
  16 import java.util.Collections;\r
  17 import java.util.HashSet;\r
  18 import java.util.Iterator;\r
  19 import java.util.Set;\r
  20 \r
  21 import org.simantics.db.Resource;\r
  22 import org.simantics.db.ReadGraph;\r
  23 import org.simantics.db.WriteGraph;\r
  24 import org.simantics.db.exception.DatabaseException;\r
  25 import org.simantics.utils.datastructures.Pair;\r
  26 \r
  27 public class UnionPredicate extends BinaryPredicate {\r
  28 \r
  29         Collection<IBinaryPredicate> predicates;\r
  30 \r
  31         public UnionPredicate(Collection<IBinaryPredicate> predicates) {\r
  32                 this.predicates = predicates;\r
  33         }\r
  34         \r
  35         public UnionPredicate(IBinaryPredicate ... predicates) {\r
  36                 this(Arrays.asList(predicates));\r
  37         }\r
  38 \r
  39         @Override\r
  40         public Collection<Resource> getObjects(ReadGraph g, Resource subject) throws DatabaseException {\r
  41                 Iterator<IBinaryPredicate> it = predicates.iterator();\r
  42                 \r
  43                 while(it.hasNext()) {\r
  44                         Collection<Resource> result = it.next().getObjects(g, subject);\r
  45                         if(!result.isEmpty()) {\r
  46                                 while(it.hasNext()) {\r
  47                                         Collection<Resource> temp = it.next().getObjects(g, subject);\r
  48                                         if(!temp.isEmpty()) {\r
  49                                                 Set<Resource> merged = new HashSet<Resource>(result.size() + temp.size());\r
  50                                                 merged.addAll(result);\r
  51                                                 merged.addAll(temp);\r
  52                                                 while(it.hasNext())\r
  53                                                         merged.addAll(it.next().getObjects(g, subject));\r
  54                                                 return merged;\r
  55                                         }\r
  56                                 }\r
  57                                 return result;\r
  58                         }\r
  59                 }\r
  60                 return Collections.emptyList();\r
  61         }\r
  62 \r
  63         @Override\r
  64         public Collection<Pair<Resource, Resource>> getStatements(ReadGraph g) throws DatabaseException {\r
  65                 Iterator<IBinaryPredicate> it = predicates.iterator();\r
  66                 \r
  67                 while(it.hasNext()) {\r
  68                         Collection<Pair<Resource, Resource>> result = it.next().getStatements(g);\r
  69                         if(!result.isEmpty()) {\r
  70                                 while(it.hasNext()) {\r
  71                                         Collection<Pair<Resource, Resource>> temp = it.next().getStatements(g);\r
  72                                         if(!temp.isEmpty()) {\r
  73                                                 Set<Pair<Resource, Resource>> merged = new HashSet<Pair<Resource, Resource>>(result.size() + temp.size());\r
  74                                                 merged.addAll(result);\r
  75                                                 merged.addAll(temp);\r
  76                                                 while(it.hasNext())\r
  77                                                         merged.addAll(it.next().getStatements(g));\r
  78                                                 return merged;\r
  79                                         }\r
  80                                 }\r
  81                                 return result;\r
  82                         }\r
  83                 }\r
  84                 return Collections.emptyList();\r
  85         }\r
  86 \r
  87         @Override\r
  88         public Collection<Resource> getSubjects(ReadGraph g, Resource object) throws DatabaseException {\r
  89                 Iterator<IBinaryPredicate> it = predicates.iterator();\r
  90                 \r
  91                 while(it.hasNext()) {\r
  92                         Collection<Resource> result = it.next().getSubjects(g, object);\r
  93                         if(!result.isEmpty()) {\r
  94                                 while(it.hasNext()) {\r
  95                                         Collection<Resource> temp = it.next().getSubjects(g, object);\r
  96                                         if(!temp.isEmpty()) {\r
  97                                                 Set<Resource> merged = new HashSet<Resource>(result.size() + temp.size());\r
  98                                                 merged.addAll(result);\r
  99                                                 merged.addAll(temp);\r
 100                                                 while(it.hasNext())\r
 101                                                         merged.addAll(it.next().getSubjects(g, object));\r
 102                                                 return merged;\r
 103                                         }\r
 104                                 }\r
 105                                 return result;\r
 106                         }\r
 107                 }\r
 108                 return Collections.emptyList();\r
 109         }\r
 110 \r
 111         @Override\r
 112         public boolean has(ReadGraph g, Resource subject, Resource object) throws DatabaseException {\r
 113                 for(IBinaryPredicate pred : predicates)\r
 114                         if(pred.has(g, subject, object))\r
 115                                 return true;\r
 116                 return false;\r
 117         }\r
 118 \r
 119         @Override\r
 120         public boolean supportsGetObjects() {\r
 121                 for(IBinaryPredicate pred : predicates)\r
 122                         if(!pred.supportsGetObjects())\r
 123                                 return false;\r
 124                 return true;\r
 125         }\r
 126 \r
 127         @Override\r
 128         public boolean supportsGetStatements() {\r
 129                 for(IBinaryPredicate pred : predicates)\r
 130                         if(!pred.supportsGetStatements())\r
 131                                 return false;\r
 132                 return true;\r
 133         }\r
 134 \r
 135         @Override\r
 136         public boolean supportsGetSubjects() {\r
 137                 for(IBinaryPredicate pred : predicates)\r
 138                         if(!pred.supportsGetSubjects())\r
 139                                 return false;\r
 140                 return true;\r
 141         }\r
 142 \r
 143         @Override\r
 144         public void add(WriteGraph g, Resource subject, Resource object) {\r
 145                 throw new UnsupportedOperationException();\r
 146         }\r
 147 \r
 148         @Override\r
 149         public void remove(WriteGraph g, Resource subject, Resource object) {\r
 150                 throw new UnsupportedOperationException();\r
 151         }\r
 152 \r
 153         @Override\r
 154         public boolean supportsAdditions() {\r
 155                 return false;\r
 156         }\r
 157 \r
 158         @Override\r
 159         public boolean supportsRemovals() {\r
 160                 return false;\r
 161         }\r
 162 \r
 163         @Override\r
 164         public int hashCode() {\r
 165                 final int prime = 31;\r
 166                 int result = 1;\r
 167                 result = prime * result\r
 168                                 + ((predicates == null) ? 0 : predicates.hashCode());\r
 169                 return result;\r
 170         }\r
 171 \r
 172         @Override\r
 173         public boolean equals(Object obj) {\r
 174                 if (this == obj)\r
 175                         return true;\r
 176                 if (obj == null)\r
 177                         return false;\r
 178                 if (getClass() != obj.getClass())\r
 179                         return false;\r
 180                 UnionPredicate other = (UnionPredicate) obj;\r
 181                 if (predicates == null) {\r
 182                         if (other.predicates != null)\r
 183                                 return false;\r
 184                 } else if (!predicates.equals(other.predicates))\r
 185                         return false;\r
 186                 return true;\r
 187         }\r
 188 \r
 189 }\r