Examples with CDateSet com.bakdata.conquery.models.common.CDateSet used on opensource projects

Example 16 with CDateSet

use of com.bakdata.conquery.models.common.CDateSet in project conquery by bakdata.

the class RelativeFormQueryPlan method execute.

@Override
public Optional<MultilineEntityResult> execute(QueryExecutionContext ctx, Entity entity) {
    // Don't set the query date aggregator here because the subqueries should set their aggregator independently
    Optional<? extends EntityResult> preResult = query.execute(ctx, entity);
    if (preResult.isEmpty()) {
        return Optional.empty();
    }
    int size = calculateCompleteLength();
    EntityResult contained = preResult.get();
    // Gather all validity dates from prerequisite
    CDateSet dateSet = query.getValidityDateAggregator().map(Aggregator::createAggregationResult).orElseGet(CDateSet::create);
    final OptionalInt sampled = indexSelector.sample(dateSet);
    // dateset is empty or sampling failed.
    if (sampled.isEmpty()) {
        log.warn("Sampled empty result for Entity[{}]: `{}({})`", contained.getEntityId(), indexSelector, dateSet);
        List<Object[]> results = new ArrayList<>();
        results.add(new Object[size]);
        return Optional.of(ResultModifier.modify(new MultilineEntityResult(entity.getId(), results), ResultModifier.existAggValuesSetterFor(getAggregators(), OptionalInt.of(getFirstAggregatorPosition()))));
    }
    int sample = sampled.getAsInt();
    List<DateContext> contexts = DateContext.generateRelativeContexts(sample, indexPlacement, timeCountBefore, timeCountAfter, timeUnit, resolutionsAndAlignmentMap);
    // create feature and outcome plans
    featureSubquery = createSubQuery(featurePlan, contexts, FeatureGroup.FEATURE);
    outcomeSubquery = createSubQuery(outcomePlan, contexts, FeatureGroup.OUTCOME);
    // determine result length and check against aggregators in query
    int featureLength = featureSubquery.columnCount();
    int outcomeLength = outcomeSubquery.columnCount();
    featureSubquery.init(ctx, entity);
    outcomeSubquery.init(ctx, entity);
    Optional<MultilineEntityResult> featureResult = featureSubquery.execute(ctx, entity);
    Optional<MultilineEntityResult> outcomeResult = outcomeSubquery.execute(ctx, entity);
    // determine result length and check against aggregators in query
    assertResultWidth(featureResult.get(), featureLength);
    assertResultWidth(outcomeResult.get(), outcomeLength);
    List<Object[]> featureResultValues = featureResult.get().getValues();
    List<Object[]> outcomeResultValues = outcomeResult.get().getValues();
    int resultStartIndex = 0;
    List<Object[]> values = new ArrayList<>();
    if (hasCompleteDateContexts(contexts)) {
        // merge a line for the complete daterange, when two dateContext were generated
        // that don't target the same feature group,
        // which would be a mistake by the generation
        // Since the DateContexts are primarily ordered by their coarseness and COMPLETE
        // is the most coarse resolution it must be at the first
        // to indexes of the list.
        Object[] mergedFull = new Object[size];
        if (featurePlan.getAggregatorSize() > 0) {
            setFeatureValues(mergedFull, featureResultValues.get(resultStartIndex));
        }
        if (outcomePlan.getAggregatorSize() > 0) {
            setOutcomeValues(mergedFull, outcomeResultValues.get(resultStartIndex));
        }
        values.add(mergedFull);
        resultStartIndex++;
    }
    // append all other lines directly
    for (int i = resultStartIndex; i < featureResultValues.size(); i++) {
        Object[] result = new Object[size];
        setFeatureValues(result, featureResultValues.get(i));
        values.add(result);
    }
    for (int i = resultStartIndex; i < outcomeResultValues.size(); i++) {
        Object[] result = new Object[size];
        setOutcomeValues(result, outcomeResultValues.get(i));
        values.add(result);
    }
    return Optional.of(new MultilineEntityResult(entity.getId(), values));
}

Example 17 with CDateSet

use of com.bakdata.conquery.models.common.CDateSet in project conquery by bakdata.

the class CDateSetTest method testMaskedAddAtMostMaskClosed.

@Test
public void testMaskedAddAtMostMaskClosed() {
    CDateSet set = CDateSet.create();
    CDateSet mask = CDateSet.create(CDateRange.of(-10, 10));
    set.maskedAdd(CDateRange.atMost(5), mask);
    assertThat(set.asRanges()).containsExactly(CDateRange.of(-10, 5));
}

Example 18 with CDateSet

use of com.bakdata.conquery.models.common.CDateSet in project conquery by bakdata.

the class CDateSetTest method testMaskedTEST.

@Test
public void testMaskedTEST() {
    CDateSet set = CDateSet.create();
    CDateSet mask = CDateSet.create(CDateRange.of(-10, 10));
    set.maskedAdd(CDateRange.of(-5, 5), mask);
    assertThat(set.asRanges()).containsExactly(CDateRange.of(-5, 5));
}

Example 19 with CDateSet

use of com.bakdata.conquery.models.common.CDateSet in project conquery by bakdata.

the class CDateSetTest method testAddMerging.

@ParameterizedTest(name = "{0}")
@MethodSource("arguments")
public void testAddMerging(String expected, CDateRange[] ranges) {
    CDateSet set = CDateSet.create();
    for (CDateRange range : ranges) {
        set.add(range);
    }
    assertThat(set).hasToString(expected);
}

Example 20 with CDateSet

use of com.bakdata.conquery.models.common.CDateSet in project conquery by bakdata.

the class CDateSetTest method testRetain.

@Test
public void testRetain() {
    CDateSet set = CDateSet.create();
    set.add(CDateRange.of(LocalDate.of(2000, 1, 1), LocalDate.of(2000, 12, 31)));
    CDateSet retain = CDateSet.create();
    retain.add(CDateRange.of(LocalDate.of(2000, 6, 1), LocalDate.of(2000, 6, 20)));
    retain.add(CDateRange.atLeast(LocalDate.of(2000, 12, 1)));
    set.retainAll(retain);
    assertThat(set).hasToString("{2000-06-01/2000-06-20, 2000-12-01/2000-12-31}");
}