io.gitlab.chaver.mining.patterns.util.MeasureListConverter 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 io.gitlab.chaver.chocotools.problem.SetUpException;
import io.gitlab.chaver.mining.patterns.measure.Measure;
import picocli.CommandLine.ITypeConverter;

import java.util.LinkedList;
import java.util.List;

import static io.gitlab.chaver.mining.patterns.measure.MeasureFactory.*;

public class MeasureListConverter implements ITypeConverter> {

    @Override
    public List convert(String arg) throws Exception {
        String[] argSplit = arg.split("");
        List measures = new LinkedList<>();
        int i = 0;
        while (i < argSplit.length) {
            if (argSplit[i].equals("f")) measures.add(freq());
            else if (argSplit[i].equals("a")) measures.add(area());
            else if (argSplit[i].equals("g")) measures.add(growthRate());
            else if (argSplit[i].equals("c")) measures.add(allConf());
            else if (argSplit[i].equals("l")) measures.add(length());
            else if (argSplit[i].equals("F")) measures.add(maxFreq());
            else if (argSplit[i].equals("n")) measures.add(mean(Integer.parseInt(argSplit[++i])));
            else if (argSplit[i].equals("m")) measures.add(min(Integer.parseInt(argSplit[++i])));
            else if (argSplit[i].equals("M")) measures.add(max(Integer.parseInt(argSplit[++i])));
            else throw new SetUpException("This measure doesn't exist : " + argSplit[++i]);
            i++;
        }
        return measures;
    }
}