001/*
002 * Copyright © 2025 CUI-OpenSource-Software (info@cuioss.de)
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *     http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package de.cuioss.test.generator.internal.net.java.quickcheck.generator.support;
017
018import de.cuioss.test.generator.internal.net.java.quickcheck.Generator;
019import de.cuioss.test.generator.internal.net.java.quickcheck.StatefulGenerator;
020
021import java.util.HashSet;
022import java.util.Set;
023
024public class SetGenerator<T> implements Generator<Set<T>> {
025
026    public static final int MAX_SIZE = ListGenerator.MAX_SIZE;
027    public static final int MAX_TRIES = VetoableGenerator.DEFAULT_MAX_TRIES;
028    private final ListGenerator<T> lists;
029    private final StatefulGenerator<T> content;
030
031    public SetGenerator(Generator<? extends T> content) {
032        this(content, MAX_TRIES);
033    }
034
035    public SetGenerator(Generator<? extends T> content, int tries) {
036        this(content, new IntegerGenerator(0, MAX_SIZE), tries);
037    }
038
039    public SetGenerator(Generator<? extends T> content, Generator<Integer> size, int tries) {
040        this.content = new UniqueValuesGenerator<>(content, tries);
041        this.lists = new ListGenerator<>(this.content, size);
042    }
043
044    @Override
045    public Set<T> next() {
046        content.reset();
047        return new HashSet<>(lists.next());
048    }
049}