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.domain.address;
28
29 import com.rapiddweller.benerator.NonNullGenerator;
30 import com.rapiddweller.benerator.WeightedGenerator;
31 import com.rapiddweller.benerator.dataset.AbstractDatasetGenerator;
32 import com.rapiddweller.benerator.dataset.Dataset;
33 import com.rapiddweller.benerator.distribution.FeatureWeight;
34 import com.rapiddweller.benerator.distribution.IndividualWeight;
35 import com.rapiddweller.benerator.sample.IndividualWeightSampleGenerator;
36 import com.rapiddweller.benerator.util.GeneratorUtil;
37
38 /**
39 * Generates {@link City} objects.<br/>
40 * <br/>
41 * Created: 14.10.2007 21:24:25
42 *
43 * @author Volker Bergmann
44 */
45 public class CityGenerator extends AbstractDatasetGenerator<City>
46 implements NonNullGenerator<City> {
47
48 private static final String REGION = "/com/rapiddweller/dataset/region";
49
50 /**
51 * Instantiates a new City generator.
52 *
53 * @param dataset the dataset
54 */
55 public CityGenerator(String dataset) {
56 super(City.class, REGION, dataset, true);
57 }
58
59 @Override
60 protected boolean isAtomic(Dataset dataset) {
61 Country country = Country.getInstance(dataset.getName(), false);
62 return (country != null);
63 }
64
65 @SuppressWarnings({"rawtypes", "unchecked"})
66 @Override
67 protected WeightedGenerator<City> createGeneratorForAtomicDataset(
68 Dataset dataset) {
69 IndividualWeightSampleGenerator<City> generator =
70 new IndividualWeightSampleGenerator<City>(
71 City.class,
72 (IndividualWeight) new FeatureWeight("population"));
73 Country country = Country.getInstance(dataset.getName());
74 country.checkCities();
75 for (State state : country.getStates()) {
76 for (City city : state.getCities()) {
77 generator.addValue(city);
78 }
79 }
80 return (generator.getVariety() > 0 ? generator : null);
81 }
82
83 @Override
84 public City generate() {
85 return GeneratorUtil.generateNonNull(this);
86 }
87
88 }