io.gitlab.chaver.mining.patterns.util.CpSkyTransactionGetter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of data-mining Show documentation

Data mining with Choco solver

There is a newer version: 1.0.1

/*
 * This file is part of io.gitlab.chaver:data-mining (https://gitlab.com/chaver/data-mining)
 *
 * Copyright (c) 2022, IMT Atlantique
 *
 * Licensed under the MIT license.
 *
 * See LICENSE file in the project root for full license information.
 */
package io.gitlab.chaver.mining.patterns.util;

import lombok.AllArgsConstructor;
import org.chocosolver.solver.variables.BoolVar;

import java.util.stream.IntStream;

@AllArgsConstructor
public class CpSkyTransactionGetter implements TransactionGetter {

    private BoolVar[] transactions;

    @Override
    public int[] getTransactions() {
        return IntStream.range(0, transactions.length).filter(i -> transactions[i].getValue() == 1).toArray();
    }
}