1 /*
2 * (c) Copyright 2006-2020 by rapiddweller GmbH & Volker Bergmann. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, is permitted under the terms of the
6 * GNU General Public License.
7 *
8 * For redistributing this software or a derivative work under a license other
9 * than the GPL-compatible Free Software License as defined by the Free
10 * Software Foundation or approved by OSI, you must first obtain a commercial
11 * license to this software product from rapiddweller GmbH & Volker Bergmann.
12 *
13 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
14 * WITHOUT A WARRANTY OF ANY KIND. ALL EXPRESS OR IMPLIED CONDITIONS,
15 * REPRESENTATIONS AND WARRANTIES, INCLUDING ANY IMPLIED WARRANTY OF
16 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE
17 * HEREBY EXCLUDED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
18 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
19 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
20 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
21 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
22 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
23 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
24 * POSSIBILITY OF SUCH DAMAGE.
25 */
26
27 package com.rapiddweller.benerator.composite;
28
29 import com.rapiddweller.benerator.Generator;
30 import com.rapiddweller.benerator.NonNullGenerator;
31 import com.rapiddweller.benerator.wrapper.ProductWrapper;
32 import com.rapiddweller.benerator.wrapper.SingleSourceArrayGenerator;
33
34 /**
35 * Creates a stochastic number of instances of a type. The number of elements is determined by the values
36 * minCount, maxCount, countDistribution.
37 * If the number of items is not one, an array of respective size is returned,
38 * otherwise a single object.<br/><br/>
39 * Created: 06.03.2008 15:43:54
40 *
41 * @param <S> the type parameter
42 * @author Volker Bergmann
43 * @since 0.5.0
44 */
45 public class SimplifyingSingleSourceArrayGenerator<S> extends SingleSourceArrayGenerator<S, Object> {
46
47 /**
48 * Instantiates a new Simplifying single source array generator.
49 *
50 * @param source the source
51 * @param countGenerator the count generator
52 */
53 public SimplifyingSingleSourceArrayGenerator(Generator<S> source, NonNullGenerator<Integer> countGenerator) {
54 super(source, source.getGeneratedType(), countGenerator);
55 }
56
57 @Override
58 public ProductWrapper<Object> generate(ProductWrapper<Object> wrapper) {
59 Object[] array = (Object[]) super.generate();
60 if (array == null) {
61 return null;
62 }
63 if (array.length == 1) {
64 return wrapper.wrap(array[0]);
65 } else {
66 return wrapper.wrap(array);
67 }
68 }
69
70 }