org.fuzzydb.random.RandomMultiEnum Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of org.fuzzydb.util Show documentation

Contains classes not specific to fuzzydb implementation which could be used in any implementation of fuzzy matching, or as general utility classes such as those in the geo package.

The newest version!

/******************************************************************************
 * Copyright (c) 2005-2008 Whirlwind Match Limited. All rights reserved.
 *
 * This is open source software; you can use, redistribute and/or modify
 * it under the terms of the Open Software Licence v 3.0 as published by the 
 * Open Source Initiative.
 *
 * You should have received a copy of the Open Software Licence along with this
 * application. if not, contact the Open Source Initiative (www.opensource.org)
 *****************************************************************************/
package org.fuzzydb.random;

import java.util.TreeSet;

import org.fuzzydb.dto.attributes.MultiEnumAttribute;
import org.fuzzydb.dto.attributes.OptionsSource;
import org.fuzzydb.util.MTRandom;


public class RandomMultiEnum extends AbstractRandomGenerator {

    OptionsSource options;

    public RandomMultiEnum(OptionsSource enumdef) {
        this.options = enumdef;
    }

    public RandomMultiEnum(OptionsSource optionsSource, float nullProportion) {
    	super(nullProportion);
        this.options = optionsSource;
    }

    @Override
	protected MultiEnumAttribute randomResult(String attrName) {
        int numvals = MTRandom.getInstance().nextInt(options.size() - 1) + 1;

        TreeSet values = new TreeSet();
        while (numvals > values.size()) {
            short randIndex = (short) MTRandom.getInstance().nextInt(options.size());
            values.add(options.findAsString(randIndex));
        }

        String[] result = new String[numvals];
        result = values.toArray(result); // uses supplied array
        return new MultiEnumAttribute(attrName, options.getName(), result);
    }
}