01: /*
02: * (c) Copyright 2007 by Volker Bergmann. All rights reserved.
03: *
04: * Redistribution and use in source and binary forms, with or without
05: * modification, is permitted under the terms of the
06: * GNU General Public License.
07: *
08: * For redistributing this software or a derivative work under a license other
09: * 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 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 org.databene.benerator;
28:
29: import org.databene.commons.TreeModel;
30:
31: import java.util.List;
32: import java.util.ArrayList;
33:
34: /**
35: * Generates random paths of a tree model.
36: * The property 'rootIncluded' tells if the tree's root node should be included in the generated path.<br/>
37: * <br/>
38: * Created: 30.07.2007 18:48:23
39: */
40: public class TreePathGenerator extends LightweightGenerator<List> {
41:
42: private TreeModel model;
43:
44: private boolean rootIncluded;
45:
46: public TreePathGenerator() {
47: this (null);
48: }
49:
50: public TreePathGenerator(TreeModel model) {
51: super (List.class);
52: this .model = model;
53: }
54:
55: // properties ------------------------------------------------------------------------------------------------------
56:
57: public boolean isRootIncluded() {
58: return rootIncluded;
59: }
60:
61: public void setRootIncluded(boolean rootIncluded) {
62: this .rootIncluded = rootIncluded;
63: }
64:
65: // Generator interface ---------------------------------------------------------------------------------------------
66:
67: public void validate() {
68: if (model == null)
69: throw new InvalidGeneratorSetupException("model", "is null");
70: }
71:
72: public List generate() {
73: List path = new ArrayList();
74: Object node = model.getRoot();
75: if (rootIncluded)
76: path.add(node);
77: while (model.getChildCount(node) > 0) {
78: node = randomChild(node);
79: path.add(node);
80: }
81: return path;
82: }
83:
84: private Object randomChild(Object node) {
85: int childCount = model.getChildCount(node);
86: int index = SimpleRandom.randomInt(0, childCount - 1);
87: return model.getChild(node, index);
88: }
89:
90: // java.lang.Object overrides --------------------------------------------------------------------------------------
91:
92: public String toString() {
93: return getClass().getSimpleName() + '[' + model + ']';
94: }
95: }
|