1 // Copyright 2008, Google Inc. All rights reserved.
2 //
3 // Redistribution and use in source and binary forms, with or without
4 // modification, are permitted provided that the following conditions are met:
5 //
6 // 1. Redistributions of source code must retain the above copyright notice,
7 // this list of conditions and the following disclaimer.
8 // 2. Redistributions in binary form must reproduce the above copyright notice,
9 // this list of conditions and the following disclaimer in the documentation
10 // and/or other materials provided with the distribution.
11 // 3. Neither the name of Google Inc. nor the names of its contributors may be
12 // used to endorse or promote products derived from this software without
13 // specific prior written permission.
14 //
15 // THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
16 // WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
17 // MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
18 // EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
19 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
21 // OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
22 // WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
23 // OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
24 // ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25
26 // This file contains the implementation of the GetRootFeature() and
27 // VisitFeatureHierarchy() functions.
28
29 #include "kml/engine/feature_visitor.h"
30 #include "kml/dom.h"
31
32 using kmldom::ContainerPtr;
33 using kmldom::ElementPtr;
34 using kmldom::FeaturePtr;
35 using kmldom::KmlPtr;
36
37 namespace kmlengine {
38
GetRootFeature(const ElementPtr & root)39 const FeaturePtr GetRootFeature(const ElementPtr& root) {
40 const KmlPtr kml = kmldom::AsKml(root);
41 if (kml && kml->has_feature()) {
42 return kml->get_feature();
43 }
44 return kmldom::AsFeature(root);
45 }
46
VisitFeatureHierarchy(const FeaturePtr & feature,FeatureVisitor & feature_visitor)47 void VisitFeatureHierarchy(const FeaturePtr& feature,
48 FeatureVisitor& feature_visitor) {
49 if (!feature) {
50 return;
51 }
52 feature_visitor.VisitFeature(feature);
53 if (const ContainerPtr container = kmldom::AsContainer(feature)) {
54 for (size_t i = 0; i < container->get_feature_array_size(); ++i) {
55 VisitFeatureHierarchy(container->get_feature_array_at(i),
56 feature_visitor);
57 }
58 }
59 }
60
61 } // end namespace kmlengine
62