Examples with Concept ai.grakn.concept.Concept used on opensource projects

Example 66 with Concept

use of ai.grakn.concept.Concept in project grakn by graknlabs.

the class AtomicTest method testUnification_RelationWithMetaRolesAndIds.

@Test
public void testUnification_RelationWithMetaRolesAndIds() {
    EmbeddedGraknTx<?> graph = unificationTestSet.tx();
    Concept instance = graph.graql().<GetQuery>parse("match $x isa subRoleEntity; get;").execute().iterator().next().get(var("x"));
    String relation = "{(role: $x, role: $y) isa binary; $y id '" + instance.getId().getValue() + "';}";
    String relation2 = "{(role: $z, role: $v) isa binary; $z id '" + instance.getId().getValue() + "';}";
    String relation3 = "{(role: $z, role: $v) isa binary; $v id '" + instance.getId().getValue() + "';}";
    exactUnification(relation, relation2, true, true, graph);
    exactUnification(relation, relation3, true, true, graph);
    exactUnification(relation2, relation3, true, true, graph);
}

Example 67 with Concept

use of ai.grakn.concept.Concept in project grakn by graknlabs.

the class TinkerComputeQueryRunner method run.

public TinkerComputeJob<List<List<Concept>>> run(PathsQuery query) {
    return runCompute(query, tinkerComputeQuery -> {
        ConceptId sourceId = query.from();
        ConceptId destinationId = query.to();
        if (!tinkerComputeQuery.verticesExistInSubgraph(sourceId, destinationId)) {
            throw GraqlQueryException.instanceDoesNotExist();
        }
        if (sourceId.equals(destinationId)) {
            return Collections.singletonList(Collections.singletonList(tx.getConcept(sourceId)));
        }
        ComputerResult result;
        Set<LabelId> subLabelIds = convertLabelsToIds(tinkerComputeQuery.subLabels());
        try {
            result = tinkerComputeQuery.compute(new ShortestPathVertexProgram(sourceId, destinationId), null, subLabelIds);
        } catch (NoResultException e) {
            return Collections.emptyList();
        }
        Multimap<Concept, Concept> predecessorMapFromSource = tinkerComputeQuery.getPredecessorMap(result);
        List<List<Concept>> allPaths = tinkerComputeQuery.getAllPaths(predecessorMapFromSource, sourceId);
        if (tinkerComputeQuery.isAttributeIncluded()) {
            // this can be slow
            return tinkerComputeQuery.getExtendedPaths(allPaths);
        }
        LOG.info("Number of paths: " + allPaths.size());
        return allPaths;
    });
}

Example 68 with Concept

use of ai.grakn.concept.Concept in project grakn by graknlabs.

the class RelationshipAtom method inferRoles.

/**
 * attempt to infer role types of this relation and return a fresh relationship with inferred role types
 * @return either this if nothing/no roles can be inferred or fresh relation with inferred role types
 */
private RelationshipAtom inferRoles(Answer sub) {
    // return if all roles known and non-meta
    List<Role> explicitRoles = getExplicitRoles().collect(Collectors.toList());
    Map<Var, Type> varTypeMap = getParentQuery().getVarTypeMap(sub);
    boolean allRolesMeta = explicitRoles.stream().allMatch(role -> Schema.MetaSchema.isMetaLabel(role.getLabel()));
    boolean roleRecomputationViable = allRolesMeta && (!sub.isEmpty() || !Sets.intersection(varTypeMap.keySet(), getRolePlayers()).isEmpty());
    if (explicitRoles.size() == getRelationPlayers().size() && !roleRecomputationViable)
        return this;
    GraknTx graph = getParentQuery().tx();
    Role metaRole = graph.admin().getMetaRole();
    List<RelationPlayer> allocatedRelationPlayers = new ArrayList<>();
    RelationshipType relType = getSchemaConcept() != null ? getSchemaConcept().asRelationshipType() : null;
    // explicit role types from castings
    List<RelationPlayer> inferredRelationPlayers = new ArrayList<>();
    getRelationPlayers().forEach(rp -> {
        Var varName = rp.getRolePlayer().var();
        VarPatternAdmin rolePattern = rp.getRole().orElse(null);
        if (rolePattern != null) {
            Label roleLabel = rolePattern.getTypeLabel().orElse(null);
            // allocate if variable role or if label non meta
            if (roleLabel == null || !Schema.MetaSchema.isMetaLabel(roleLabel)) {
                inferredRelationPlayers.add(RelationPlayer.of(rolePattern, varName.admin()));
                allocatedRelationPlayers.add(rp);
            }
        }
    });
    // remaining roles
    // role types can repeat so no matter what has been allocated still the full spectrum of possibilities is present
    // TODO make restrictions based on cardinality constraints
    Set<Role> possibleRoles = relType != null ? relType.relates().collect(toSet()) : inferPossibleTypes(sub).stream().filter(Concept::isRelationshipType).map(Concept::asRelationshipType).flatMap(RelationshipType::relates).collect(toSet());
    // possible role types for each casting based on its type
    Map<RelationPlayer, Set<Role>> mappings = new HashMap<>();
    getRelationPlayers().stream().filter(rp -> !allocatedRelationPlayers.contains(rp)).forEach(rp -> {
        Var varName = rp.getRolePlayer().var();
        Type type = varTypeMap.get(varName);
        mappings.put(rp, top(type != null ? compatibleRoles(type, possibleRoles) : possibleRoles));
    });
    // allocate all unambiguous mappings
    mappings.entrySet().stream().filter(entry -> entry.getValue().size() == 1).forEach(entry -> {
        RelationPlayer rp = entry.getKey();
        Var varName = rp.getRolePlayer().var();
        Role role = Iterables.getOnlyElement(entry.getValue());
        VarPatternAdmin rolePattern = Graql.var().label(role.getLabel()).admin();
        inferredRelationPlayers.add(RelationPlayer.of(rolePattern, varName.admin()));
        allocatedRelationPlayers.add(rp);
    });
    // fill in unallocated roles with metarole
    getRelationPlayers().stream().filter(rp -> !allocatedRelationPlayers.contains(rp)).forEach(rp -> {
        Var varName = rp.getRolePlayer().var();
        VarPatternAdmin rolePattern = rp.getRole().orElse(null);
        rolePattern = rolePattern != null ? rolePattern.var().label(metaRole.getLabel()).admin() : Graql.var().label(metaRole.getLabel()).admin();
        inferredRelationPlayers.add(RelationPlayer.of(rolePattern, varName.admin()));
    });
    VarPattern relationPattern = relationPattern(getVarName(), inferredRelationPlayers);
    VarPatternAdmin newPattern = (isDirect() ? relationPattern.directIsa(getPredicateVariable()) : relationPattern.isa(getPredicateVariable())).admin();
    return create(newPattern, getPredicateVariable(), getTypeId(), getParentQuery());
}

Example 69 with Concept

use of ai.grakn.concept.Concept in project grakn by graknlabs.

the class ConceptBuilder method tryGetConcept.

@Nullable
private Concept tryGetConcept() {
    Concept concept = null;
    if (has(ID)) {
        concept = executor.tx().getConcept(use(ID));
        if (has(LABEL)) {
            concept.asSchemaConcept().setLabel(use(LABEL));
        }
    } else if (has(LABEL)) {
        concept = executor.tx().getSchemaConcept(use(LABEL));
    }
    if (concept != null) {
        // The super can be changed on an existing concept
        if (has(SUPER_CONCEPT)) {
            SchemaConcept superConcept = use(SUPER_CONCEPT);
            setSuper(concept.asSchemaConcept(), superConcept);
        }
        validate(concept);
    }
    return concept;
}

Example 70 with Concept

use of ai.grakn.concept.Concept in project grakn by graknlabs.

the class ConceptBuilder method tryPutConcept.

private Concept tryPutConcept() {
    usedParams.clear();
    Concept concept;
    if (has(IS_ROLE)) {
        use(IS_ROLE);
        Label label = use(LABEL);
        Role role = executor.tx().putRole(label);
        if (has(SUPER_CONCEPT)) {
            setSuper(role, use(SUPER_CONCEPT));
        }
        concept = role;
    } else if (has(IS_RULE)) {
        use(IS_RULE);
        Label label = use(LABEL);
        Pattern when = use(WHEN);
        Pattern then = use(THEN);
        Rule rule = executor.tx().putRule(label, when, then);
        if (has(SUPER_CONCEPT)) {
            setSuper(rule, use(SUPER_CONCEPT));
        }
        concept = rule;
    } else if (has(SUPER_CONCEPT)) {
        concept = putSchemaConcept();
    } else if (has(TYPE)) {
        concept = putInstance();
    } else {
        throw GraqlQueryException.insertUndefinedVariable(executor.printableRepresentation(var));
    }
    // Check for any unexpected parameters
    preProvidedParams.forEach((param, value) -> {
        if (!usedParams.contains(param)) {
            throw GraqlQueryException.insertUnexpectedProperty(param.name(), value, concept);
        }
    });
    return concept;
}