const FeatureProfile*
FeatureListSource::createFeatureProfile()
{
const SpatialReference* srs = 0L;
osgEarth::Bounds bounds;
if ( !_features.empty() )
{
// Get the SRS of the first feature
srs = _features.front()->getSRS();
// Compute the extent of the features
for (FeatureList::iterator itr = _features.begin(); itr != _features.end(); ++itr)
{
Feature* feature = itr->get();
if (feature->getGeometry())
{
bounds.expandBy( feature->getGeometry()->getBounds() );
}
}
}
// return the new profile, or a default extent if the profile could not be computed.
if ( srs && bounds.isValid() )
return new FeatureProfile( GeoExtent(srs, bounds) );
else
return new FeatureProfile( _defaultExtent );
}
osg::Vec3dArray*
FeatureMaskLayer::getOrCreateMaskBoundary(float heightScale,
const SpatialReference* srs,
ProgressCallback* progress)
{
if (!_featureSource.valid())
return 0L;
if (!_boundary.valid())
{
Threading::ScopedMutexLock lock(_boundaryMutex);
if (!_boundary.valid())
{
osg::ref_ptr<FeatureCursor> cursor = _featureSource->createFeatureCursor();
if (cursor.valid() && cursor->hasMore())
{
Feature* f = cursor->nextFeature();
if (f && f->getGeometry())
{
f->transform(srs);
_boundary = f->getGeometry()->createVec3dArray();
}
}
}
}
return _boundary.get();
}
FilterContext
ScaleFilter::push( FeatureList& input, FilterContext& cx )
{
for( FeatureList::iterator i = input.begin(); i != input.end(); ++i )
{
Feature* input = i->get();
if ( input && input->getGeometry() )
{
Bounds envelope = input->getGeometry()->getBounds();
// now scale and shift everything
GeometryIterator scale_iter( input->getGeometry() );
while( scale_iter.hasMore() )
{
Geometry* geom = scale_iter.next();
for( osg::Vec3dArray::iterator v = geom->begin(); v != geom->end(); v++ )
{
double xr = (v->x() - envelope.xMin()) / envelope.width();
v->x() += (xr - 0.5) * _scale;
double yr = (v->y() - envelope.yMin()) / envelope.height();
v->y() += (yr - 0.5) * _scale;
}
}
}
}
return cx;
}
Config
FeatureNode::getConfig() const
{
Config conf("feature");
if ( !_features.empty() )
{
// Write out a single feature for now.
Feature* feature = _features.begin()->get();
conf.set("name", getName());
Config geomConf("geometry");
geomConf.value() = GeometryUtils::geometryToWKT( feature->getGeometry() );
conf.add(geomConf);
std::string srs = feature->getSRS() ? feature->getSRS()->getHorizInitString() : "";
if ( !srs.empty() ) conf.set("srs", srs);
std::string vsrs = feature->getSRS() ? feature->getSRS()->getVertInitString() : "";
if ( !vsrs.empty() ) conf.set("vdatum", vsrs);
if ( feature->geoInterp().isSet() )
conf.set("geointerp", feature->geoInterp() == GEOINTERP_GREAT_CIRCLE? "greatcircle" : "rhumbline");
}
if (!_style.empty() )
{
conf.set( "style", _style );
}
return conf;
}
bool
AddPointHandler::addPoint( float x, float y, osgViewer::View* view )
{
osg::Vec3d world;
MapNode* mapNode = _featureNode->getMapNode();
if ( mapNode->getTerrain()->getWorldCoordsUnderMouse( view, x, y, world ) )
{
// Get the map point from the world
GeoPoint mapPoint;
mapPoint.fromWorld( mapNode->getMapSRS(), world );
Feature* feature = _featureNode->getFeature();
if ( feature )
{
// Convert the map point to the feature's SRS
GeoPoint featurePoint = mapPoint.transform( feature->getSRS() );
feature->getGeometry()->push_back( featurePoint.vec3d() );
_featureNode->init();
return true;
}
}
return false;
}
osg::Vec3dArray* createBoundary( const SpatialReference* srs, ProgressCallback* progress )
{
if ( _failed )
return 0L;
if ( _features.valid() )
{
if ( _features->getFeatureProfile() )
{
osg::ref_ptr<FeatureCursor> cursor = _features->createFeatureCursor();
if ( cursor )
{
if ( cursor->hasMore() )
{
Feature* f = cursor->nextFeature();
if ( f && f->getGeometry() )
{
// Init a filter to tranform feature in desired SRS
if (!srs->isEquivalentTo(_features->getFeatureProfile()->getSRS())) {
FilterContext cx;
cx.profile() = new FeatureProfile(_features->getFeatureProfile()->getExtent());
//cx.isGeocentric() = _features->getFeatureProfile()->getSRS()->isGeographic();
TransformFilter xform( srs );
FeatureList featureList;
featureList.push_back(f);
cx = xform.push(featureList, cx);
}
return f->getGeometry()->toVec3dArray();
}
}
}
}
else
{
OE_WARN << LC << "Failed to create boundary; feature source has no SRS" << std::endl;
_failed = true;
}
}
else
{
OE_WARN << LC << "Unable to create boundary; invalid feature source" << std::endl;
_failed = true;
}
return 0L;
}
void
AltitudeFilter::pushAndDontClamp( FeatureList& features, FilterContext& cx )
{
NumericExpression scaleExpr;
if ( _altitude.valid() && _altitude->verticalScale().isSet() )
scaleExpr = *_altitude->verticalScale();
NumericExpression offsetExpr;
if ( _altitude.valid() && _altitude->verticalOffset().isSet() )
offsetExpr = *_altitude->verticalOffset();
for( FeatureList::iterator i = features.begin(); i != features.end(); ++i )
{
Feature* feature = i->get();
// run a symbol script if present.
if ( _altitude.valid() && _altitude->script().isSet() )
{
StringExpression temp( _altitude->script().get() );
feature->eval( temp, &cx );
}
double minHAT = DBL_MAX;
double maxHAT = -DBL_MAX;
double scaleZ = 1.0;
if ( _altitude.valid() && _altitude->verticalScale().isSet() )
scaleZ = feature->eval( scaleExpr, &cx );
double offsetZ = 0.0;
if ( _altitude.valid() && _altitude->verticalOffset().isSet() )
offsetZ = feature->eval( offsetExpr, &cx );
GeometryIterator gi( feature->getGeometry() );
while( gi.hasMore() )
{
Geometry* geom = gi.next();
for( Geometry::iterator g = geom->begin(); g != geom->end(); ++g )
{
g->z() *= scaleZ;
g->z() += offsetZ;
if ( g->z() < minHAT )
minHAT = g->z();
if ( g->z() > maxHAT )
maxHAT = g->z();
}
}
if ( minHAT != DBL_MAX )
{
feature->set( "__min_hat", minHAT );
feature->set( "__max_hat", maxHAT );
}
}
}
void
FeatureEditor::init()
{
removeChildren( 0, getNumChildren() );
Feature* feature = _featureNode->getFeatures().front();
//Create a dragger for each point
for (unsigned int i = 0; i < feature->getGeometry()->size(); i++)
{
SphereDragger* dragger = new SphereDragger( _featureNode->getMapNode() );
dragger->setColor( _color );
dragger->setPickColor( _pickColor );
dragger->setSize( _size );
dragger->setPosition(GeoPoint(feature->getSRS(), (*feature->getGeometry())[i].x(), (*feature->getGeometry())[i].y()));
dragger->addPositionChangedCallback(new MoveFeatureDraggerCallback( _featureNode.get(), i) );
addChild(dragger);
}
}
FilterContext
BufferFilter::push( FeatureList& input, FilterContext& context )
{
if ( !isSupported() )
{
OE_WARN << "BufferFilter support not enabled - please compile osgEarth with GEOS" << std::endl;
return context;
}
//OE_NOTICE << "Buffer: input = " << input.size() << " features" << std::endl;
for( FeatureList::iterator i = input.begin(); i != input.end(); )
{
Feature* feature = i->get();
if ( !feature || !feature->getGeometry() )
continue;
osg::ref_ptr<Symbology::Geometry> output;
Symbology::BufferParameters params;
params._capStyle =
_capStyle == Stroke::LINECAP_ROUND ? Symbology::BufferParameters::CAP_ROUND :
_capStyle == Stroke::LINECAP_SQUARE ? Symbology::BufferParameters::CAP_SQUARE :
_capStyle == Stroke::LINECAP_FLAT ? Symbology::BufferParameters::CAP_FLAT :
Symbology::BufferParameters::CAP_SQUARE;
params._cornerSegs = _numQuadSegs;
if ( feature->getGeometry()->buffer( _distance.value(), output, params ) )
{
feature->setGeometry( output.get() );
++i;
}
else
{
i = input.erase( i );
OE_DEBUG << LC << "feature " << feature->getFID() << " yielded no geometry" << std::endl;
}
}
return context;
}
FilterContext
ConvertTypeFilter::push( FeatureList& input, FilterContext& context )
{
if ( !isSupported() )
{
OE_WARN << "ConvertTypeFilter support not enabled" << std::endl;
return context;
}
bool ok = true;
for( FeatureList::iterator i = input.begin(); i != input.end(); ++i )
{
Feature* input = i->get();
if ( input && input->getGeometry() && input->getGeometry()->getComponentType() != _toType )
{
input->setGeometry( input->getGeometry()->cloneAs(_toType) );
}
}
return context;
}
FilterContext
ClampFilter::push( FeatureList& features, const FilterContext& cx )
{
const Session* session = cx.getSession();
if ( !session ) {
OE_WARN << LC << "No session - session is required for elevation clamping" << std::endl;
return cx;
}
// the map against which we'll be doing elevation clamping
MapFrame mapf = session->createMapFrame( Map::ELEVATION_LAYERS );
const SpatialReference* mapSRS = mapf.getProfile()->getSRS();
const SpatialReference* featureSRS = cx.profile()->getSRS();
bool isGeocentric = session->getMapInfo().isGeocentric();
// establish an elevation query interface based on the features' SRS.
ElevationQuery eq( mapf );
for( FeatureList::iterator i = features.begin(); i != features.end(); ++i )
{
Feature* feature = i->get();
GeometryIterator gi( feature->getGeometry() );
while( gi.hasMore() )
{
Geometry* geom = gi.next();
if ( isGeocentric )
{
// convert to map coords:
cx.toWorld( geom );
mapSRS->transformFromECEF( geom );
// populate the elevations:
eq.getElevations( geom, mapSRS );
// convert back to geocentric:
mapSRS->transformToECEF( geom );
cx.toLocal( geom );
}
else
{
// clamps the entire array to the highest available resolution.
eq.getElevations( geom, featureSRS );
}
}
}
return cx;
}
FilterContext
ScatterFilter::push(FeatureList& features, FilterContext& context )
{
if ( !isSupported() ) {
OE_WARN << LC << "support for this filter is not enabled" << std::endl;
return context;
}
// seed the random number generator so the randomness is the same each time
_prng = Random( _randomSeed, Random::METHOD_FAST );
for( FeatureList::iterator i = features.begin(); i != features.end(); ++i )
{
Feature* f = i->get();
Geometry* geom = f->getGeometry();
if ( !geom )
continue;
const SpatialReference* geomSRS = context.profile()->getSRS();
osg::ref_ptr< PointSet > points = new PointSet();
if ( geom->getComponentType() == Geometry::TYPE_POLYGON )
{
polyScatter( geom, geomSRS, context, points.get() );
}
else if (
geom->getComponentType() == Geometry::TYPE_LINESTRING ||
geom->getComponentType() == Geometry::TYPE_RING )
{
lineScatter( geom, geomSRS, context, points.get() );
}
else {
points = static_cast< PointSet*>(geom->cloneAs(Geometry::TYPE_POINTSET));
}
// replace the source geometry with the scattered points.
f->setGeometry( points.get() );
}
return context;
}
FilterContext
CentroidFilter::push(FeatureList& features, FilterContext& context )
{
for( FeatureList::iterator i = features.begin(); i != features.end(); ++i )
{
Feature* f = i->get();
Geometry* geom = f->getGeometry();
if ( !geom )
continue;
PointSet* newGeom = new PointSet();
newGeom->push_back( geom->getBounds().center() );
f->setGeometry( newGeom );
}
return context;
}
v8::Handle<v8::Value>
JSFeature::PropertyCallback(v8::Local<v8::String> name, const v8::AccessorInfo& info)
{
Feature* feature = V8Util::UnwrapObject<Feature>(info.Holder());
v8::String::Utf8Value utf8_value(name);
std::string prop(*utf8_value);
if (!feature || prop.empty())
return v8::Handle<v8::Value>();
if (prop == "fid")
return v8::Uint32::New(feature->getFID());
else if (prop == "attrs" || prop == "attributes")
return V8Util::WrapObject(feature, GetAttributesObjectTemplate());
else if (prop == "geometry")
return JSSymbologyGeometry::WrapGeometry(feature->getGeometry());
//return GetFeatureAttr(prop, feature);
return v8::Handle<v8::Value>();
}
void apply( osg::Geode& geode )
{
// save the geode's drawables..
osg::Geode::DrawableList old_drawables = geode.getDrawableList();
//OE_DEBUG << "ClusterVisitor geode " << &geode << " featureNode=" << _featureNode << " drawables=" << old_drawables.size() << std::endl;
// ..and clear out the drawables list.
geode.removeDrawables( 0, geode.getNumDrawables() );
// foreach each drawable that was originally in the geode...
for( osg::Geode::DrawableList::iterator i = old_drawables.begin(); i != old_drawables.end(); i++ )
{
osg::Geometry* originalDrawable = dynamic_cast<osg::Geometry*>( i->get() );
if ( !originalDrawable )
continue;
// go through the list of input features...
for( FeatureList::const_iterator j = _features.begin(); j != _features.end(); j++ )
{
Feature* feature = j->get();
osg::Matrixd scaleMatrix;
if ( _symbol->scale().isSet() )
{
double scale = feature->eval( _scaleExpr, &_cx );
scaleMatrix.makeScale( scale, scale, scale );
}
osg::Matrixd rotationMatrix;
if ( _modelSymbol && _modelSymbol->heading().isSet() )
{
float heading = feature->eval( _headingExpr, &_cx );
rotationMatrix.makeRotate( osg::Quat(osg::DegreesToRadians(heading), osg::Vec3(0,0,1)) );
}
GeometryIterator gi( feature->getGeometry(), false );
while( gi.hasMore() )
{
Geometry* geom = gi.next();
// if necessary, transform the points to the target SRS:
if ( !_makeECEF && !_targetSRS->isEquivalentTo(_srs) )
{
_srs->transform( geom->asVector(), _targetSRS );
}
for( Geometry::const_iterator k = geom->begin(); k != geom->end(); ++k )
{
osg::Vec3d point = *k;
osg::Matrixd mat;
if ( _makeECEF )
{
osg::Matrixd rotation;
ECEF::transformAndGetRotationMatrix( point, _srs, point, _targetSRS, rotation );
mat = rotationMatrix * rotation * scaleMatrix * osg::Matrixd::translate(point) * _f2n->world2local();
}
else
{
mat = rotationMatrix * scaleMatrix * osg::Matrixd::translate(point) * _f2n->world2local();
}
// clone the source drawable once for each input feature.
osg::ref_ptr<osg::Geometry> newDrawable = osg::clone(
originalDrawable,
osg::CopyOp::DEEP_COPY_ARRAYS | osg::CopyOp::DEEP_COPY_PRIMITIVES );
osg::Vec3Array* verts = dynamic_cast<osg::Vec3Array*>( newDrawable->getVertexArray() );
if ( verts )
{
for( osg::Vec3Array::iterator v = verts->begin(); v != verts->end(); ++v )
{
(*v).set( (*v) * mat );
}
// add the new cloned, translated drawable back to the geode.
geode.addDrawable( newDrawable.get() );
if ( _cx.featureIndex() )
_cx.featureIndex()->tagPrimitiveSets( newDrawable.get(), feature );
}
}
}
}
}
geode.dirtyBound();
MeshConsolidator::run( geode );
osg::NodeVisitor::apply( geode );
}
bool
BuildGeometryFilter::process( FeatureList& features, const FilterContext& context )
{
bool makeECEF = false;
const SpatialReference* featureSRS = 0L;
const SpatialReference* mapSRS = 0L;
if ( context.isGeoreferenced() )
{
makeECEF = context.getSession()->getMapInfo().isGeocentric();
featureSRS = context.extent()->getSRS();
mapSRS = context.getSession()->getMapInfo().getProfile()->getSRS();
}
for( FeatureList::iterator f = features.begin(); f != features.end(); ++f )
{
Feature* input = f->get();
GeometryIterator parts( input->getGeometry(), false );
while( parts.hasMore() )
{
Geometry* part = parts.next();
// skip empty geometry
if ( part->size() == 0 )
continue;
const Style& myStyle = input->style().isSet() ? *input->style() : _style;
bool setLinePropsHere = input->style().isSet(); // otherwise it will be set globally, we assume
float width = 1.0f;
bool hasPolyOutline = false;
const PointSymbol* pointSymbol = myStyle.get<PointSymbol>();
const LineSymbol* lineSymbol = myStyle.get<LineSymbol>();
const PolygonSymbol* polySymbol = myStyle.get<PolygonSymbol>();
// resolve the geometry type from the component type and the symbology:
Geometry::Type renderType = Geometry::TYPE_UNKNOWN;
// First priority is a matching part type and symbol:
if ( polySymbol != 0L && part->getType() == Geometry::TYPE_POLYGON )
{
renderType = Geometry::TYPE_POLYGON;
}
else if ( lineSymbol != 0L && part->isLinear() )
{
renderType = part->getType();
}
else if ( pointSymbol != 0L && part->getType() == Geometry::TYPE_POINTSET )
{
renderType = Geometry::TYPE_POINTSET;
}
// Second priority is the symbol:
else if ( polySymbol != 0L )
{
renderType = Geometry::TYPE_POLYGON;
}
else if ( lineSymbol != 0L )
{
if ( part->getType() == Geometry::TYPE_POLYGON )
renderType = Geometry::TYPE_RING;
else
renderType = Geometry::TYPE_LINESTRING;
}
else if ( pointSymbol != 0L )
{
renderType = Geometry::TYPE_POINTSET;
}
// No symbol? just use the geometry type.
else
{
renderType = part->getType();
}
// validate the geometry:
if ( renderType == Geometry::TYPE_POLYGON && part->size() < 3 )
continue;
else if ( (renderType == Geometry::TYPE_LINESTRING || renderType == Geometry::TYPE_RING) && part->size() < 2 )
continue;
// resolve the color:
osg::Vec4f primaryColor =
polySymbol ? osg::Vec4f(polySymbol->fill()->color()) :
lineSymbol ? osg::Vec4f(lineSymbol->stroke()->color()) :
pointSymbol ? osg::Vec4f(pointSymbol->fill()->color()) :
osg::Vec4f(1,1,1,1);
osg::Geometry* osgGeom = new osg::Geometry();
osgGeom->setUseVertexBufferObjects( _useVertexBufferObjects.value() );
if ( _featureNameExpr.isSet() )
{
const std::string& name = input->eval( _featureNameExpr.mutable_value(), &context );
osgGeom->setName( name );
}
// build the geometry:
osg::Vec3Array* allPoints = 0L;
if ( renderType == Geometry::TYPE_POLYGON )
{
buildPolygon(part, featureSRS, mapSRS, makeECEF, true, osgGeom);
allPoints = static_cast<osg::Vec3Array*>( osgGeom->getVertexArray() );
}
else
{
// line or point geometry
GLenum primMode =
renderType == Geometry::TYPE_LINESTRING ? GL_LINE_STRIP :
renderType == Geometry::TYPE_RING ? GL_LINE_LOOP :
GL_POINTS;
allPoints = new osg::Vec3Array();
transformAndLocalize( part->asVector(), featureSRS, allPoints, mapSRS, _world2local, makeECEF );
osgGeom->addPrimitiveSet( new osg::DrawArrays( primMode, 0, part->size() ) );
osgGeom->setVertexArray( allPoints );
applyLineAndPointSymbology( osgGeom->getOrCreateStateSet(), lineSymbol, pointSymbol );
if ( primMode == GL_POINTS && allPoints->size() == 1 )
{
const osg::Vec3d& center = (*allPoints)[0];
osgGeom->setInitialBound( osg::BoundingBox(center-osg::Vec3(.5,.5,.5), center+osg::Vec3(.5,.5,.5)) );
}
}
if (allPoints->getVertexBufferObject())
allPoints->getVertexBufferObject()->setUsage(GL_STATIC_DRAW_ARB);
// subdivide the mesh if necessary to conform to an ECEF globe:
if ( makeECEF && renderType != Geometry::TYPE_POINTSET )
{
// check for explicit tessellation disable:
const LineSymbol* line = _style.get<LineSymbol>();
bool disableTess = line && line->tessellation().isSetTo(0);
if ( makeECEF && !disableTess )
{
double threshold = osg::DegreesToRadians( *_maxAngle_deg );
OE_DEBUG << "Running mesh subdivider with threshold " << *_maxAngle_deg << std::endl;
MeshSubdivider ms( _world2local, _local2world );
//ms.setMaxElementsPerEBO( INT_MAX );
if ( input->geoInterp().isSet() )
ms.run( *osgGeom, threshold, *input->geoInterp() );
else
ms.run( *osgGeom, threshold, *_geoInterp );
}
}
// assign the primary color:
#if USE_SINGLE_COLOR
osg::Vec4Array* colors = new osg::Vec4Array( 1 );
(*colors)[0] = primaryColor;
osgGeom->setColorBinding( osg::Geometry::BIND_OVERALL );
#else
osg::Vec4Array* colors = new osg::Vec4Array( osgGeom->getVertexArray()->getNumElements() ); //allPoints->size() );
for(unsigned c=0; c<colors->size(); ++c)
(*colors)[c] = primaryColor;
osgGeom->setColorBinding( osg::Geometry::BIND_PER_VERTEX );
#endif
osgGeom->setColorArray( colors );
_geode->addDrawable( osgGeom );
// record the geometry's primitive set(s) in the index:
if ( context.featureIndex() )
context.featureIndex()->tagPrimitiveSets( osgGeom, input );
// build secondary geometry, if necessary (polygon outlines)
if ( renderType == Geometry::TYPE_POLYGON && lineSymbol )
{
// polygon offset on the poly so the outline doesn't z-fight
osgGeom->getOrCreateStateSet()->setAttributeAndModes( new osg::PolygonOffset(1,1), 1 );
osg::Geometry* outline = new osg::Geometry();
outline->setUseVertexBufferObjects( _useVertexBufferObjects.value() );
buildPolygon(part, featureSRS, mapSRS, makeECEF, false, outline);
if ( outline->getVertexArray()->getVertexBufferObject() )
outline->getVertexArray()->getVertexBufferObject()->setUsage(GL_STATIC_DRAW_ARB);
osg::Vec4f outlineColor = lineSymbol->stroke()->color();
osg::Vec4Array* outlineColors = new osg::Vec4Array();
#if USE_SINGLE_COLOR
outlineColors->reserve(1);
outlineColors->push_back( outlineColor );
outline->setColorBinding( osg::Geometry::BIND_OVERALL );
#else
unsigned pcount = part->getTotalPointCount();
outlineColors->reserve( pcount );
for( unsigned c=0; c < pcount; ++c )
outlineColors->push_back( outlineColor );
outline->setColorBinding( osg::Geometry::BIND_PER_VERTEX );
#endif
outline->setColorArray(outlineColors);
// check for explicit tessellation disable:
bool disableTess = lineSymbol && lineSymbol->tessellation().isSetTo(0);
// subdivide if necessary.
if ( makeECEF && !disableTess )
{
double threshold = osg::DegreesToRadians( *_maxAngle_deg );
OE_DEBUG << "Running mesh subdivider for outlines with threshold " << *_maxAngle_deg << std::endl;
MeshSubdivider ms( _world2local, _local2world );
if ( input->geoInterp().isSet() )
ms.run( *outline, threshold, *input->geoInterp() );
else
ms.run( *outline, threshold, *_geoInterp );
}
applyLineAndPointSymbology( outline->getOrCreateStateSet(), lineSymbol, 0L );
// make normals before adding an outline
osgUtil::SmoothingVisitor sv;
_geode->accept( sv );
_geode->addDrawable( outline );
//_featureNode->addDrawable( outline, input->getFID() );
// Mark each primitive set with its feature ID.
if ( context.featureIndex() )
context.featureIndex()->tagPrimitiveSets( outline, input );
}
}
}
return true;
}
void
FeatureNode::build()
{
if ( !_clampCallback.valid() )
_clampCallback = new ClampCallback(this);
_attachPoint = 0L;
// if there is existing geometry, kill it
this->removeChildren( 0, this->getNumChildren() );
if ( !getMapNode() )
return;
if ( _features.empty() )
return;
const Style &style = getStyle();
// compilation options.
GeometryCompilerOptions options = _options;
// figure out what kind of altitude manipulation we need to perform.
AnnotationUtils::AltitudePolicy ap;
AnnotationUtils::getAltitudePolicy( style, ap );
// If we're doing auto-clamping on the CPU, shut off compiler map clamping
// clamping since it would be redundant.
if ( ap.sceneClamping )
{
options.ignoreAltitudeSymbol() = true;
}
_clamperData.clear();
osg::Node* node = _compiled.get();
if (_needsRebuild || !_compiled.valid() )
{
// Clone the Features before rendering as the GeometryCompiler and it's filters can change the coordinates
// of the geometry when performing localization or converting to geocentric.
_extent = GeoExtent::INVALID;
FeatureList clone;
for(FeatureList::iterator itr = _features.begin(); itr != _features.end(); ++itr)
{
Feature* feature = new Feature( *itr->get(), osg::CopyOp::DEEP_COPY_ALL);
GeoExtent featureExtent(feature->getSRS(), feature->getGeometry()->getBounds());
if (_extent.isInvalid())
{
_extent = featureExtent;
}
else
{
_extent.expandToInclude( featureExtent );
}
clone.push_back( feature );
}
// prep the compiler:
GeometryCompiler compiler( options );
Session* session = new Session( getMapNode()->getMap(), _styleSheet.get() );
FilterContext context( session, new FeatureProfile( _extent ), _extent, _index);
_compiled = compiler.compile( clone, style, context );
node = _compiled.get();
_needsRebuild = false;
// Compute the world bounds
osg::BoundingSphered bounds;
for( FeatureList::iterator itr = _features.begin(); itr != _features.end(); ++itr)
{
osg::BoundingSphered bs;
itr->get()->getWorldBound(getMapNode()->getMapSRS(), bs);
bounds.expandBy(bs);
}
// The polytope will ensure we only clamp to intersecting tiles:
Feature::getWorldBoundingPolytope(bounds, getMapNode()->getMapSRS(), _featurePolytope);
}
if ( node )
{
if ( AnnotationUtils::styleRequiresAlphaBlending( style ) &&
getStyle().get<ExtrusionSymbol>() )
{
node = AnnotationUtils::installTwoPassAlpha( node );
}
_attachPoint = new osg::Group();
_attachPoint->addChild( node );
// Draped (projected) geometry
if ( ap.draping )
{
DrapeableNode* d = new DrapeableNode();
d->addChild( _attachPoint );
this->addChild( d );
}
// GPU-clamped geometry
else if ( ap.gpuClamping )
{
ClampableNode* clampable = new ClampableNode();
clampable->addChild( _attachPoint );
this->addChild( clampable );
}
else
{
this->addChild( _attachPoint );
// set default lighting based on whether we are extruding:
setDefaultLighting( style.has<ExtrusionSymbol>() );
}
applyRenderSymbology(style);
if ( getMapNode()->getTerrain() )
{
if ( ap.sceneClamping )
{
// Need dynamic data variance since scene clamping will change the verts
SetDataVarianceVisitor sdv(osg::Object::DYNAMIC);
this->accept(sdv);
getMapNode()->getTerrain()->addTerrainCallback(_clampCallback.get());
clamp(getMapNode()->getTerrain()->getGraph(), getMapNode()->getTerrain());
}
else
{
getMapNode()->getTerrain()->removeTerrainCallback( _clampCallback.get() );
}
}
}
}
osg::HeightField* createHeightField( const TileKey& key,
ProgressCallback* progress)
{
if (key.getLevelOfDetail() > _maxDataLevel)
{
//OE_NOTICE << "Reached maximum data resolution key=" << key.getLevelOfDetail() << " max=" << _maxDataLevel << std::endl;
return NULL;
}
int tileSize = _options.tileSize().value();
//Allocate the heightfield
osg::ref_ptr<osg::HeightField> hf = new osg::HeightField;
hf->allocate(tileSize, tileSize);
for (unsigned int i = 0; i < hf->getHeightList().size(); ++i) hf->getHeightList()[i] = NO_DATA_VALUE;
if (intersects(key))
{
//Get the extents of the tile
double xmin, ymin, xmax, ymax;
key.getExtent().getBounds(xmin, ymin, xmax, ymax);
const SpatialReference* featureSRS = _features->getFeatureProfile()->getSRS();
GeoExtent extentInFeatureSRS = key.getExtent().transform( featureSRS );
const SpatialReference* keySRS = key.getProfile()->getSRS();
// populate feature list
// assemble a spatial query. It helps if your features have a spatial index.
Query query;
query.bounds() = extentInFeatureSRS.bounds();
FeatureList featureList;
osg::ref_ptr<FeatureCursor> cursor = _features->createFeatureCursor(query);
while ( cursor.valid() && cursor->hasMore() )
{
Feature* f = cursor->nextFeature();
if ( f && f->getGeometry() )
featureList.push_back(f);
}
// We now have a feature list in feature SRS.
bool transformRequired = !keySRS->isHorizEquivalentTo(featureSRS);
if (!featureList.empty())
{
// Iterate over the output heightfield and sample the data that was read into it.
double dx = (xmax - xmin) / (tileSize-1);
double dy = (ymax - ymin) / (tileSize-1);
for (int c = 0; c < tileSize; ++c)
{
double geoX = xmin + (dx * (double)c);
for (int r = 0; r < tileSize; ++r)
{
double geoY = ymin + (dy * (double)r);
float h = NO_DATA_VALUE;
for (FeatureList::iterator f = featureList.begin(); f != featureList.end(); ++f)
{
osgEarth::Symbology::Polygon* boundary = dynamic_cast<osgEarth::Symbology::Polygon*>((*f)->getGeometry());
if (!boundary)
{
OE_WARN << LC << "NOT A POLYGON" << std::endl;
}
else
{
GeoPoint geo(keySRS, geoX, geoY, 0.0, ALTMODE_ABSOLUTE);
if ( transformRequired )
geo = geo.transform(featureSRS);
if ( boundary->contains2D(geo.x(), geo.y()) )
{
h = (*f)->getDouble(_options.attr().value());
if ( keySRS->isGeographic() )
{
// for a round earth, must adjust the final elevation accounting for the
// curvature of the earth; so we have to adjust it in the feature boundary's
// local tangent plane.
Bounds bounds = boundary->getBounds();
GeoPoint anchor( featureSRS, bounds.center().x(), bounds.center().y(), h, ALTMODE_ABSOLUTE );
if ( transformRequired )
anchor = anchor.transform(keySRS);
// For transforming between ECEF and local tangent plane:
osg::Matrix localToWorld, worldToLocal;
anchor.createLocalToWorld(localToWorld);
worldToLocal.invert( localToWorld );
// Get the ECEF location of the anchor point:
osg::Vec3d ecef;
geo.toWorld( ecef );
// Move it into Local Tangent Plane coordinates:
osg::Vec3d local = ecef * worldToLocal;
// Reset the Z to zero, since the LTP is centered on the "h" elevation:
local.z() = 0.0;
// Back into ECEF:
ecef = local * localToWorld;
// And back into lat/long/alt:
geo.fromWorld( geo.getSRS(), ecef);
h = geo.z();
}
break;
}
}
}
hf->setHeight(c, r, h-0.1);
}
}
}
}
return hf.release();
}
/**
* Creates a complete set of positioned label nodes from a feature list.
*/
osg::Node* createNode(
const FeatureList& input,
const Style& style,
FilterContext& context )
{
if ( style.get<TextSymbol>() == 0L && style.get<IconSymbol>() == 0L )
return 0L;
// copy the style so we can (potentially) modify the text symbol.
Style styleCopy = style;
TextSymbol* text = styleCopy.get<TextSymbol>();
IconSymbol* icon = styleCopy.get<IconSymbol>();
osg::Group* group = new osg::Group();
StringExpression textContentExpr ( text ? *text->content() : StringExpression() );
NumericExpression textPriorityExpr( text ? *text->priority() : NumericExpression() );
NumericExpression textSizeExpr ( text ? *text->size() : NumericExpression() );
StringExpression iconUrlExpr ( icon ? *icon->url() : StringExpression() );
NumericExpression iconScaleExpr ( icon ? *icon->scale() : NumericExpression() );
NumericExpression iconHeadingExpr ( icon ? *icon->heading() : NumericExpression() );
for( FeatureList::const_iterator i = input.begin(); i != input.end(); ++i )
{
Feature* feature = i->get();
if ( !feature )
continue;
// run a symbol script if present.
if ( text && text->script().isSet() )
{
StringExpression temp( text->script().get() );
feature->eval( temp, &context );
}
// run a symbol script if present.
if ( icon && icon->script().isSet() )
{
StringExpression temp( icon->script().get() );
feature->eval( temp, &context );
}
const Geometry* geom = feature->getGeometry();
if ( !geom )
continue;
Style tempStyle = styleCopy;
// evaluate expressions into literals.
// TODO: Later we could replace this with a generate "expression evaluator" type
// that we could pass to PlaceNode in the DB options. -gw
if ( text )
{
if ( text->content().isSet() )
tempStyle.get<TextSymbol>()->content()->setLiteral( feature->eval( textContentExpr, &context ) );
if ( text->size().isSet() )
tempStyle.get<TextSymbol>()->size()->setLiteral( feature->eval(textSizeExpr, &context) );
}
if ( icon )
{
if ( icon->url().isSet() )
tempStyle.get<IconSymbol>()->url()->setLiteral( feature->eval(iconUrlExpr, &context) );
if ( icon->scale().isSet() )
tempStyle.get<IconSymbol>()->scale()->setLiteral( feature->eval(iconScaleExpr, &context) );
if ( icon->heading().isSet() )
tempStyle.get<IconSymbol>()->heading()->setLiteral( feature->eval(iconHeadingExpr, &context) );
}
osg::Node* node = makePlaceNode(
context,
feature,
tempStyle,
textPriorityExpr);
if ( node )
{
if ( context.featureIndex() )
{
context.featureIndex()->tagNode(node, feature);
}
group->addChild( node );
}
}
// Note to self: need to change this to support picking later. -gw
//VirtualProgram* vp = VirtualProgram::getOrCreate(group->getOrCreateStateSet());
//vp->setInheritShaders( false );
return group;
}
void
AltitudeFilter::pushAndClamp( FeatureList& features, FilterContext& cx )
{
const Session* session = cx.getSession();
// the map against which we'll be doing elevation clamping
//MapFrame mapf = session->createMapFrame( Map::ELEVATION_LAYERS );
MapFrame mapf = session->createMapFrame(
(Map::ModelParts)(Map::TERRAIN_LAYERS | Map::MODEL_LAYERS) );
const SpatialReference* mapSRS = mapf.getProfile()->getSRS();
osg::ref_ptr<const SpatialReference> featureSRS = cx.profile()->getSRS();
// establish an elevation query interface based on the features' SRS.
ElevationQuery eq( mapf );
// want a result even if it's low res
eq.setFallBackOnNoData( true );
NumericExpression scaleExpr;
if ( _altitude->verticalScale().isSet() )
scaleExpr = *_altitude->verticalScale();
NumericExpression offsetExpr;
if ( _altitude->verticalOffset().isSet() )
offsetExpr = *_altitude->verticalOffset();
// whether to record the min/max height-above-terrain values.
bool collectHATs =
_altitude->clamping() == AltitudeSymbol::CLAMP_RELATIVE_TO_TERRAIN ||
_altitude->clamping() == AltitudeSymbol::CLAMP_ABSOLUTE;
// whether to clamp every vertex (or just the centroid)
bool perVertex =
_altitude->binding() == AltitudeSymbol::BINDING_VERTEX;
// whether the SRS's have a compatible vertical datum.
bool vertEquiv =
featureSRS->isVertEquivalentTo( mapSRS );
for( FeatureList::iterator i = features.begin(); i != features.end(); ++i )
{
Feature* feature = i->get();
// run a symbol script if present.
if ( _altitude.valid() && _altitude->script().isSet() )
{
StringExpression temp( _altitude->script().get() );
feature->eval( temp, &cx );
}
double maxTerrainZ = -DBL_MAX;
double minTerrainZ = DBL_MAX;
double minHAT = DBL_MAX;
double maxHAT = -DBL_MAX;
double scaleZ = 1.0;
if ( _altitude.valid() && _altitude->verticalScale().isSet() )
scaleZ = feature->eval( scaleExpr, &cx );
double offsetZ = 0.0;
if ( _altitude.valid() && _altitude->verticalOffset().isSet() )
offsetZ = feature->eval( offsetExpr, &cx );
GeometryIterator gi( feature->getGeometry() );
while( gi.hasMore() )
{
Geometry* geom = gi.next();
// Absolute heights in Z. Only need to collect the HATs; the geometry
// remains unchanged.
if ( _altitude->clamping() == AltitudeSymbol::CLAMP_ABSOLUTE )
{
if ( perVertex )
{
std::vector<double> elevations;
elevations.reserve( geom->size() );
if ( eq.getElevations( geom->asVector(), featureSRS, elevations, _maxRes ) )
{
for( unsigned i=0; i<geom->size(); ++i )
{
osg::Vec3d& p = (*geom)[i];
p.z() *= scaleZ;
p.z() += offsetZ;
double z = p.z();
if ( !vertEquiv )
{
osg::Vec3d tempgeo;
if ( !featureSRS->transform(p, mapSRS->getGeographicSRS(), tempgeo) )
z = tempgeo.z();
}
double hat = z - elevations[i];
if ( hat > maxHAT )
maxHAT = hat;
if ( hat < minHAT )
minHAT = hat;
if ( elevations[i] > maxTerrainZ )
maxTerrainZ = elevations[i];
if ( elevations[i] < minTerrainZ )
minTerrainZ = elevations[i];
}
}
}
else // per centroid
{
osgEarth::Bounds bounds = geom->getBounds();
const osg::Vec2d& center = bounds.center2d();
GeoPoint centroid(featureSRS, center.x(), center.y());
double centroidElevation;
if ( eq.getElevation( centroid, centroidElevation, _maxRes ) )
{
for( unsigned i=0; i<geom->size(); ++i )
{
osg::Vec3d& p = (*geom)[i];
p.z() *= scaleZ;
p.z() += offsetZ;
double z = p.z();
if ( !vertEquiv )
{
osg::Vec3d tempgeo;
if ( !featureSRS->transform(p, mapSRS->getGeographicSRS(), tempgeo) )
z = tempgeo.z();
}
double hat = z - centroidElevation;
if ( hat > maxHAT )
maxHAT = hat;
if ( hat < minHAT )
minHAT = hat;
}
if ( centroidElevation > maxTerrainZ )
maxTerrainZ = centroidElevation;
if ( centroidElevation < minTerrainZ )
minTerrainZ = centroidElevation;
}
}
}
// Heights-above-ground in Z. Need to resolve this to an absolute number
// and record HATs along the way.
else if ( _altitude->clamping() == AltitudeSymbol::CLAMP_RELATIVE_TO_TERRAIN )
{
osg::ref_ptr<const SpatialReference> featureSRSwithMapVertDatum = !vertEquiv ?
SpatialReference::create(featureSRS->getHorizInitString(), mapSRS->getVertInitString()) : 0L;
if ( perVertex )
{
std::vector<double> elevations;
elevations.reserve( geom->size() );
if ( eq.getElevations( geom->asVector(), featureSRS, elevations, _maxRes ) )
{
for( unsigned i=0; i<geom->size(); ++i )
{
osg::Vec3d& p = (*geom)[i];
p.z() *= scaleZ;
p.z() += offsetZ;
double hat = p.z();
p.z() = elevations[i] + p.z();
// if necessary, convert the Z value (which is now in the map's SRS) back to
// the feature's SRS.
if ( !vertEquiv )
{
featureSRSwithMapVertDatum->transform(p, featureSRS, p);
}
if ( hat > maxHAT )
maxHAT = hat;
if ( hat < minHAT )
minHAT = hat;
if ( elevations[i] > maxTerrainZ )
maxTerrainZ = elevations[i];
if ( elevations[i] < minTerrainZ )
minTerrainZ = elevations[i];
}
}
}
else // per-centroid
{
osgEarth::Bounds bounds = geom->getBounds();
const osg::Vec2d& center = bounds.center2d();
GeoPoint centroid(featureSRS, center.x(), center.y());
double centroidElevation;
if ( eq.getElevation( centroid, centroidElevation, _maxRes ) )
{
for( unsigned i=0; i<geom->size(); ++i )
{
osg::Vec3d& p = (*geom)[i];
p.z() *= scaleZ;
p.z() += offsetZ;
double hat = p.z();
p.z() = centroidElevation + p.z();
// if necessary, convert the Z value (which is now in the map's SRS) back to
// the feature's SRS.
if ( !vertEquiv )
{
featureSRSwithMapVertDatum->transform(p, featureSRS, p);
}
if ( hat > maxHAT )
maxHAT = hat;
if ( hat < minHAT )
minHAT = hat;
}
if ( centroidElevation > maxTerrainZ )
maxTerrainZ = centroidElevation;
if ( centroidElevation < minTerrainZ )
minTerrainZ = centroidElevation;
}
}
}
// Clamp - replace the geometry's Z with the terrain height.
else // CLAMP_TO_TERRAIN
{
if ( perVertex )
{
eq.getElevations( geom->asVector(), featureSRS, true, _maxRes );
// if necessary, transform the Z values (which are now in the map SRS) back
// into the feature's SRS.
if ( !vertEquiv )
{
osg::ref_ptr<const SpatialReference> featureSRSwithMapVertDatum =
SpatialReference::create(featureSRS->getHorizInitString(), mapSRS->getVertInitString());
osg::Vec3d tempgeo;
for( unsigned i=0; i<geom->size(); ++i )
{
osg::Vec3d& p = (*geom)[i];
featureSRSwithMapVertDatum->transform(p, featureSRS, p);
}
}
}
else // per-centroid
{
osgEarth::Bounds bounds = geom->getBounds();
const osg::Vec2d& center = bounds.center2d();
GeoPoint centroid(featureSRS, center.x(), center.y());
double centroidElevation;
osg::ref_ptr<const SpatialReference> featureSRSWithMapVertDatum;
if ( !vertEquiv )
featureSRSWithMapVertDatum = SpatialReference::create(featureSRS->getHorizInitString(), mapSRS->getVertInitString());
if ( eq.getElevation( centroid, centroidElevation, _maxRes ) )
{
for( unsigned i=0; i<geom->size(); ++i )
{
osg::Vec3d& p = (*geom)[i];
p.z() = centroidElevation;
if ( !vertEquiv )
{
featureSRSWithMapVertDatum->transform(p, featureSRS, p);
}
}
}
}
}
if ( !collectHATs )
{
for( Geometry::iterator i = geom->begin(); i != geom->end(); ++i )
{
i->z() *= scaleZ;
i->z() += offsetZ;
}
}
}
if ( minHAT != DBL_MAX )
{
feature->set( "__min_hat", minHAT );
feature->set( "__max_hat", maxHAT );
}
if ( minTerrainZ != DBL_MAX )
{
feature->set( "__min_terrain_z", minTerrainZ );
feature->set( "__max_terrain_z", maxTerrainZ );
}
}
}
bool
SubstituteModelFilter::process(const FeatureList& features,
const MarkerSymbol* symbol,
Session* session,
osg::Group* attachPoint,
FilterContext& context )
{
bool makeECEF = context.getSession()->getMapInfo().isGeocentric();
// first, go through the features and build the model cache. Apply the model matrix' scale
// factor to any AutoTransforms directly (cloning them as necessary)
std::map< std::pair<URI, float>, osg::ref_ptr<osg::Node> > uniqueModels;
//std::map< Feature*, osg::ref_ptr<osg::Node> > featureModels;
StringExpression uriEx = *symbol->url();
NumericExpression scaleEx = *symbol->scale();
for( FeatureList::const_iterator f = features.begin(); f != features.end(); ++f )
{
Feature* input = f->get();
// evaluate the marker URI expression:
StringExpression uriEx = *symbol->url();
URI markerURI( input->eval(uriEx, &context), uriEx.uriContext() );
// find the corresponding marker in the cache
MarkerResource* marker = 0L;
MarkerCache::Record rec = _markerCache.get( markerURI );
if ( rec.valid() ) {
marker = rec.value();
}
else {
marker = new MarkerResource();
marker->uri() = markerURI;
_markerCache.insert( markerURI, marker );
}
// evalute the scale expression (if there is one)
float scale = 1.0f;
osg::Matrixd scaleMatrix;
if ( symbol->scale().isSet() )
{
scale = input->eval( scaleEx, &context );
if ( scale == 0.0 )
scale = 1.0;
scaleMatrix = osg::Matrix::scale( scale, scale, scale );
}
// how that we have a marker source, create a node for it
std::pair<URI,float> key( markerURI, scale );
osg::ref_ptr<osg::Node>& model = uniqueModels[key];
if ( !model.valid() )
{
model = context.resourceCache()->getMarkerNode( marker );
if ( scale != 1.0f && dynamic_cast<osg::AutoTransform*>( model.get() ) )
{
// clone the old AutoTransform, set the new scale, and copy over its children.
osg::AutoTransform* oldAT = dynamic_cast<osg::AutoTransform*>(model.get());
osg::AutoTransform* newAT = osg::clone( oldAT );
// make a scaler and put it between the new AutoTransform and its kids
osg::MatrixTransform* scaler = new osg::MatrixTransform(osg::Matrix::scale(scale,scale,scale));
for( unsigned i=0; i<newAT->getNumChildren(); ++i )
scaler->addChild( newAT->getChild(0) );
newAT->removeChildren(0, newAT->getNumChildren());
newAT->addChild( scaler );
model = newAT;
}
}
if ( model.valid() )
{
GeometryIterator gi( input->getGeometry(), false );
while( gi.hasMore() )
{
Geometry* geom = gi.next();
for( unsigned i=0; i<geom->size(); ++i )
{
osg::Matrixd mat;
osg::Vec3d point = (*geom)[i];
if ( makeECEF )
{
// the "rotation" element lets us re-orient the instance to ensure it's pointing up. We
// could take a shortcut and just use the current extent's local2world matrix for this,
// but if the tile is big enough the up vectors won't be quite right.
osg::Matrixd rotation;
ECEF::transformAndGetRotationMatrix( point, context.profile()->getSRS(), point, rotation );
mat = rotation * scaleMatrix * osg::Matrixd::translate( point ) * _world2local;
}
else
{
mat = scaleMatrix * osg::Matrixd::translate( point ) * _world2local;
}
osg::MatrixTransform* xform = new osg::MatrixTransform();
xform->setMatrix( mat );
xform->addChild( model.get() );
attachPoint->addChild( xform );
// name the feature if necessary
if ( !_featureNameExpr.empty() )
{
const std::string& name = input->eval( _featureNameExpr, &context);
if ( !name.empty() )
xform->setName( name );
}
}
}
}
}
return true;
}
bool
SubstituteModelFilter::process(const FeatureList& features,
const InstanceSymbol* symbol,
Session* session,
osg::Group* attachPoint,
FilterContext& context )
{
// Establish SRS information:
bool makeECEF = context.getSession()->getMapInfo().isGeocentric();
const SpatialReference* targetSRS = context.getSession()->getMapInfo().getSRS();
// first, go through the features and build the model cache. Apply the model matrix' scale
// factor to any AutoTransforms directly (cloning them as necessary)
std::map< std::pair<URI, float>, osg::ref_ptr<osg::Node> > uniqueModels;
// keep track of failed URIs so we don't waste time or warning messages on them
std::set< URI > missing;
StringExpression uriEx = *symbol->url();
NumericExpression scaleEx = *symbol->scale();
const ModelSymbol* modelSymbol = dynamic_cast<const ModelSymbol*>(symbol);
const IconSymbol* iconSymbol = dynamic_cast<const IconSymbol*> (symbol);
NumericExpression headingEx;
if ( modelSymbol )
headingEx = *modelSymbol->heading();
for( FeatureList::const_iterator f = features.begin(); f != features.end(); ++f )
{
Feature* input = f->get();
// evaluate the instance URI expression:
StringExpression uriEx = *symbol->url();
URI instanceURI( input->eval(uriEx, &context), uriEx.uriContext() );
// find the corresponding marker in the cache
osg::ref_ptr<InstanceResource> instance;
if ( !findResource(instanceURI, symbol, context, missing, instance) )
continue;
// evalute the scale expression (if there is one)
float scale = 1.0f;
osg::Matrixd scaleMatrix;
if ( symbol->scale().isSet() )
{
scale = input->eval( scaleEx, &context );
if ( scale == 0.0 )
scale = 1.0;
if ( scale != 1.0 )
_normalScalingRequired = true;
scaleMatrix = osg::Matrix::scale( scale, scale, scale );
}
osg::Matrixd rotationMatrix;
if ( modelSymbol && modelSymbol->heading().isSet() )
{
float heading = input->eval(headingEx, &context);
rotationMatrix.makeRotate( osg::Quat(osg::DegreesToRadians(heading), osg::Vec3(0,0,1)) );
}
// how that we have a marker source, create a node for it
std::pair<URI,float> key( instanceURI, scale );
// cache nodes per instance.
osg::ref_ptr<osg::Node>& model = uniqueModels[key];
if ( !model.valid() )
{
context.resourceCache()->getInstanceNode( instance.get(), model );
// if icon decluttering is off, install an AutoTransform.
if ( iconSymbol )
{
if ( iconSymbol->declutter() == true )
{
Decluttering::setEnabled( model->getOrCreateStateSet(), true );
}
else if ( dynamic_cast<osg::AutoTransform*>(model.get()) == 0L )
{
osg::AutoTransform* at = new osg::AutoTransform();
at->setAutoRotateMode( osg::AutoTransform::ROTATE_TO_SCREEN );
at->setAutoScaleToScreen( true );
at->addChild( model );
model = at;
}
}
}
if ( model.valid() )
{
GeometryIterator gi( input->getGeometry(), false );
while( gi.hasMore() )
{
Geometry* geom = gi.next();
// if necessary, transform the points to the target SRS:
if ( !makeECEF && !targetSRS->isEquivalentTo(context.profile()->getSRS()) )
{
context.profile()->getSRS()->transform( geom->asVector(), targetSRS );
}
for( unsigned i=0; i<geom->size(); ++i )
{
osg::Matrixd mat;
osg::Vec3d point = (*geom)[i];
if ( makeECEF )
{
// the "rotation" element lets us re-orient the instance to ensure it's pointing up. We
// could take a shortcut and just use the current extent's local2world matrix for this,
// but if the tile is big enough the up vectors won't be quite right.
osg::Matrixd rotation;
ECEF::transformAndGetRotationMatrix( point, context.profile()->getSRS(), point, targetSRS, rotation );
mat = rotationMatrix * rotation * scaleMatrix * osg::Matrixd::translate( point ) * _world2local;
}
else
{
mat = rotationMatrix * scaleMatrix * osg::Matrixd::translate( point ) * _world2local;
}
osg::MatrixTransform* xform = new osg::MatrixTransform();
xform->setMatrix( mat );
xform->setDataVariance( osg::Object::STATIC );
xform->addChild( model.get() );
attachPoint->addChild( xform );
if ( context.featureIndex() && !_useDrawInstanced )
{
context.featureIndex()->tagNode( xform, input );
}
// name the feature if necessary
if ( !_featureNameExpr.empty() )
{
const std::string& name = input->eval( _featureNameExpr, &context);
if ( !name.empty() )
xform->setName( name );
}
}
}
}
}
if ( iconSymbol )
{
// activate decluttering for icons if requested
if ( iconSymbol->declutter() == true )
{
Decluttering::setEnabled( attachPoint->getOrCreateStateSet(), true );
}
// activate horizon culling if we are in geocentric space
if ( context.getSession() && context.getSession()->getMapInfo().isGeocentric() )
{
HorizonCullingProgram::install( attachPoint->getOrCreateStateSet() );
}
}
// active DrawInstanced if required:
if ( _useDrawInstanced && Registry::capabilities().supportsDrawInstanced() )
{
DrawInstanced::convertGraphToUseDrawInstanced( attachPoint );
// install a shader program to render draw-instanced.
DrawInstanced::install( attachPoint->getOrCreateStateSet() );
}
return true;
}
bool
FeatureGridder::cullFeatureListToCell( int i, FeatureList& features ) const
{
bool success = true;
int inCount = features.size();
Bounds b;
if ( getCellBounds( i, b ) )
{
if ( _policy.cullingTechnique() == GriddingPolicy::CULL_BY_CENTROID )
{
for( FeatureList::iterator f_i = features.begin(); f_i != features.end(); )
{
bool keepFeature = false;
Feature* feature = f_i->get();
Symbology::Geometry* featureGeom = feature->getGeometry();
if ( featureGeom )
{
osg::Vec3d centroid = featureGeom->getBounds().center();
if ( b.contains( centroid.x(), centroid.y() ) )
{
keepFeature = true;
}
}
if ( keepFeature )
++f_i;
else
f_i = features.erase( f_i );
}
}
else // CULL_BY_CROPPING (requires GEOS)
{
#ifdef OSGEARTH_HAVE_GEOS
// create the intersection polygon:
osg::ref_ptr<Symbology::Polygon> poly = new Symbology::Polygon( 4 );
poly->push_back( osg::Vec3d( b.xMin(), b.yMin(), 0 ));
poly->push_back( osg::Vec3d( b.xMax(), b.yMin(), 0 ));
poly->push_back( osg::Vec3d( b.xMax(), b.yMax(), 0 ));
poly->push_back( osg::Vec3d( b.xMin(), b.yMax(), 0 ));
for( FeatureList::iterator f_i = features.begin(); f_i != features.end(); )
{
bool keepFeature = false;
Feature* feature = f_i->get();
Symbology::Geometry* featureGeom = feature->getGeometry();
if ( featureGeom )
{
osg::ref_ptr<Symbology::Geometry> croppedGeometry;
if ( featureGeom->crop( poly.get(), croppedGeometry ) )
{
feature->setGeometry( croppedGeometry.get() );
keepFeature = true;
}
}
if ( keepFeature )
++f_i;
else
f_i = features.erase( f_i );
}
#endif // OSGEARTH_HAVE_GEOS
}
}
OE_INFO << LC
<< "Grid cell " << i << ": bounds="
<< b.xMin() << "," << b.yMin() << " => " << b.xMax() << "," << b.yMax()
<< "; in=" << inCount << "; out=" << features.size()
<< std::endl;
return success;
}
bool
ExtrudeGeometryFilter::process( FeatureList& features, FilterContext& context )
{
// seed our random number generators
Random wallSkinPRNG( _wallSkinSymbol.valid()? *_wallSkinSymbol->randomSeed() : 0, Random::METHOD_FAST );
Random roofSkinPRNG( _roofSkinSymbol.valid()? *_roofSkinSymbol->randomSeed() : 0, Random::METHOD_FAST );
for( FeatureList::iterator f = features.begin(); f != features.end(); ++f )
{
Feature* input = f->get();
GeometryIterator iter( input->getGeometry(), false );
while( iter.hasMore() )
{
Geometry* part = iter.next();
osg::ref_ptr<osg::Geometry> walls = new osg::Geometry();
walls->setUseVertexBufferObjects( _useVertexBufferObjects.get() );
osg::ref_ptr<osg::Geometry> rooflines = 0L;
osg::ref_ptr<osg::Geometry> baselines = 0L;
osg::ref_ptr<osg::Geometry> outlines = 0L;
if ( part->getType() == Geometry::TYPE_POLYGON )
{
rooflines = new osg::Geometry();
rooflines->setUseVertexBufferObjects( _useVertexBufferObjects.get() );
// prep the shapes by making sure all polys are open:
static_cast<Polygon*>(part)->open();
}
// fire up the outline geometry if we have a line symbol.
if ( _outlineSymbol != 0L )
{
outlines = new osg::Geometry();
outlines->setUseVertexBufferObjects( _useVertexBufferObjects.get() );
}
// make a base cap if we're doing stencil volumes.
if ( _makeStencilVolume )
{
baselines = new osg::Geometry();
baselines->setUseVertexBufferObjects( _useVertexBufferObjects.get() );
}
// calculate the extrusion height:
float height;
if ( _heightCallback.valid() )
{
height = _heightCallback->operator()(input, context);
}
else if ( _heightExpr.isSet() )
{
height = input->eval( _heightExpr.mutable_value(), &context );
}
else
{
height = *_extrusionSymbol->height();
}
// calculate the height offset from the base:
float offset = 0.0;
if ( _heightOffsetExpr.isSet() )
{
offset = input->eval( _heightOffsetExpr.mutable_value(), &context );
}
osg::ref_ptr<osg::StateSet> wallStateSet;
osg::ref_ptr<osg::StateSet> roofStateSet;
// calculate the wall texturing:
SkinResource* wallSkin = 0L;
if ( _wallSkinSymbol.valid() )
{
if ( _wallResLib.valid() )
{
SkinSymbol querySymbol( *_wallSkinSymbol.get() );
querySymbol.objectHeight() = fabs(height) - offset;
wallSkin = _wallResLib->getSkin( &querySymbol, wallSkinPRNG, context.getDBOptions() );
}
else
{
//TODO: simple single texture?
}
}
// calculate the rooftop texture:
SkinResource* roofSkin = 0L;
if ( _roofSkinSymbol.valid() )
{
if ( _roofResLib.valid() )
{
SkinSymbol querySymbol( *_roofSkinSymbol.get() );
roofSkin = _roofResLib->getSkin( &querySymbol, roofSkinPRNG, context.getDBOptions() );
}
else
{
//TODO: simple single texture?
}
}
// calculate the colors:
osg::Vec4f wallColor(1,1,1,0), wallBaseColor(1,1,1,0), roofColor(1,1,1,0), outlineColor(1,1,1,1);
if ( _wallPolygonSymbol.valid() )
{
wallColor = _wallPolygonSymbol->fill()->color();
if ( _extrusionSymbol->wallGradientPercentage().isSet() )
{
wallBaseColor = Color(wallColor).brightness( 1.0 - *_extrusionSymbol->wallGradientPercentage() );
}
else
{
wallBaseColor = wallColor;
}
}
if ( _roofPolygonSymbol.valid() )
{
roofColor = _roofPolygonSymbol->fill()->color();
}
if ( _outlineSymbol.valid() )
{
outlineColor = _outlineSymbol->stroke()->color();
}
// Create the extruded geometry!
if (extrudeGeometry(
part, height, offset,
*_extrusionSymbol->flatten(),
walls.get(), rooflines.get(), baselines.get(), outlines.get(),
wallColor, wallBaseColor, roofColor, outlineColor,
wallSkin, roofSkin,
context ) )
{
if ( wallSkin )
{
context.resourceCache()->getStateSet( wallSkin, wallStateSet );
}
// generate per-vertex normals, altering the geometry as necessary to avoid
// smoothing around sharp corners
osgUtil::SmoothingVisitor::smooth(
*walls.get(),
osg::DegreesToRadians(_wallAngleThresh_deg) );
// tessellate and add the roofs if necessary:
if ( rooflines.valid() )
{
osgUtil::Tessellator tess;
tess.setTessellationType( osgUtil::Tessellator::TESS_TYPE_GEOMETRY );
tess.setWindingType( osgUtil::Tessellator::TESS_WINDING_ODD );
tess.retessellatePolygons( *(rooflines.get()) );
// generate default normals (no crease angle necessary; they are all pointing up)
// TODO do this manually; probably faster
if ( !_makeStencilVolume )
osgUtil::SmoothingVisitor::smooth( *rooflines.get() );
if ( roofSkin )
{
context.resourceCache()->getStateSet( roofSkin, roofStateSet );
}
}
if ( baselines.valid() )
{
osgUtil::Tessellator tess;
tess.setTessellationType( osgUtil::Tessellator::TESS_TYPE_GEOMETRY );
tess.setWindingType( osgUtil::Tessellator::TESS_WINDING_ODD );
tess.retessellatePolygons( *(baselines.get()) );
}
std::string name;
if ( !_featureNameExpr.empty() )
name = input->eval( _featureNameExpr, &context );
FeatureSourceIndex* index = context.featureIndex();
addDrawable( walls.get(), wallStateSet.get(), name, input, index );
if ( rooflines.valid() )
{
addDrawable( rooflines.get(), roofStateSet.get(), name, input, index );
}
if ( baselines.valid() )
{
addDrawable( baselines.get(), 0L, name, input, index );
}
if ( outlines.valid() )
{
addDrawable( outlines.get(), 0L, name, input, index );
}
}
}
}
return true;
}
Status initialize( const osgDB::Options* dbOptions )
{
Cache* cache = 0;
_dbOptions = Registry::instance()->cloneOrCreateOptions( dbOptions );
if ( _dbOptions.valid() )
{
// Set up a Custom caching bin for this TileSource
cache = Cache::get( _dbOptions.get() );
if ( cache )
{
Config optionsConf = _options.getConfig();
std::string binId = Stringify() << std::hex << hashString(optionsConf.toJSON());
_cacheBin = cache->addBin( binId );
if ( _cacheBin.valid() )
{
_cacheBin->apply( _dbOptions.get() );
}
}
}
if ( !_options.featureOptions().isSet() )
{
return Status::Error( Stringify() << LC << "Illegal: feature source is required" );
}
_features = FeatureSourceFactory::create( _options.featureOptions().value() );
if ( !_features.valid() )
{
return Status::Error( Stringify() << "Illegal: no valid feature source provided");
}
//if ( _features->getGeometryType() != osgEarth::Symbology::Geometry::TYPE_POLYGON )
//{
// Status::Error( Stringify() << "Illegal: only polygon features are currently supported");
// return false;
//}
_features->initialize( _dbOptions );
// populate feature list
osg::ref_ptr<FeatureCursor> cursor = _features->createFeatureCursor();
while ( cursor.valid() && cursor->hasMore() )
{
Feature* f = cursor->nextFeature();
if ( f && f->getGeometry() )
_featureList.push_back(f);
}
if (_features->getFeatureProfile())
{
if (getProfile() && !getProfile()->getSRS()->isEquivalentTo(_features->getFeatureProfile()->getSRS()))
OE_WARN << LC << "Specified profile does not match feature profile, ignoring specified profile." << std::endl;
_extents = _features->getFeatureProfile()->getExtent();
const Profile* profile = Profile::create(
_extents.getSRS(),
_extents.bounds().xMin(),
_extents.bounds().yMin(),
_extents.bounds().xMax(),
_extents.bounds().yMax());
setProfile( profile );
}
else if (getProfile())
{
_extents = getProfile()->getExtent();
}
else
{
return Status::Error( Stringify() << "Failed to establish a profile for " << this->getName() );
}
getDataExtents().push_back( DataExtent(_extents, 0, _maxDataLevel) );
return STATUS_OK;
}
void
FeatureNode::build()
{
// if there's a decoration, clear it out first.
this->clearDecoration();
_attachPoint = 0L;
// if there is existing geometry, kill it
this->removeChildren( 0, this->getNumChildren() );
if ( !getMapNode() )
return;
if ( _features.empty() )
return;
const Style &style = getStyle();
// compilation options.
GeometryCompilerOptions options = _options;
// figure out what kind of altitude manipulation we need to perform.
AnnotationUtils::AltitudePolicy ap;
AnnotationUtils::getAltitudePolicy( style, ap );
// If we're doing auto-clamping on the CPU, shut off compiler map clamping
// clamping since it would be redundant.
// TODO: I think this is OBE now that we have "scene" clamping technique..
if ( ap.sceneClamping )
{
options.ignoreAltitudeSymbol() = true;
}
osg::Node* node = _compiled.get();
if (_needsRebuild || !_compiled.valid() )
{
// Clone the Features before rendering as the GeometryCompiler and it's filters can change the coordinates
// of the geometry when performing localization or converting to geocentric.
_extent = GeoExtent::INVALID;
FeatureList clone;
for(FeatureList::iterator itr = _features.begin(); itr != _features.end(); ++itr)
{
Feature* feature = new Feature( *itr->get(), osg::CopyOp::DEEP_COPY_ALL);
GeoExtent featureExtent(feature->getSRS(), feature->getGeometry()->getBounds());
if (_extent.isInvalid())
{
_extent = featureExtent;
}
else
{
_extent.expandToInclude( featureExtent );
}
clone.push_back( feature );
}
// prep the compiler:
GeometryCompiler compiler( options );
Session* session = new Session( getMapNode()->getMap(), _styleSheet.get() );
FilterContext context( session, new FeatureProfile( _extent ), _extent );
_compiled = compiler.compile( clone, style, context );
node = _compiled.get();
_needsRebuild = false;
// Compute the world bounds
osg::BoundingSphered bounds;
for( FeatureList::iterator itr = _features.begin(); itr != _features.end(); ++itr)
{
osg::BoundingSphered bs;
itr->get()->getWorldBound(getMapNode()->getMapSRS(), bs);
bounds.expandBy(bs);
}
// The polytope will ensure we only clamp to intersecting tiles:
Feature::getWorldBoundingPolytope(bounds, getMapNode()->getMapSRS(), _featurePolytope);
}
if ( node )
{
if ( AnnotationUtils::styleRequiresAlphaBlending( style ) &&
getStyle().get<ExtrusionSymbol>() )
{
node = AnnotationUtils::installTwoPassAlpha( node );
}
//OE_NOTICE << GeometryUtils::geometryToGeoJSON( _feature->getGeometry() ) << std::endl;
_attachPoint = new osg::Group();
_attachPoint->addChild( node );
// Draped (projected) geometry
if ( ap.draping )
{
DrapeableNode* d = new DrapeableNode(); // getMapNode() );
d->addChild( _attachPoint );
this->addChild( d );
}
// GPU-clamped geometry
else if ( ap.gpuClamping )
{
ClampableNode* clampable = new ClampableNode( getMapNode() );
clampable->addChild( _attachPoint );
this->addChild( clampable );
const RenderSymbol* render = style.get<RenderSymbol>();
if ( render && render->depthOffset().isSet() )
{
clampable->setDepthOffsetOptions( *render->depthOffset() );
}
}
else
{
this->addChild( _attachPoint );
// CPU-clamped geometry?
if ( ap.sceneClamping )
{
// save for later when we need to reclamp the mesh on the CPU
_altitude = style.get<AltitudeSymbol>();
// activate the terrain callback:
setCPUAutoClamping( true );
// set default lighting based on whether we are extruding:
setLightingIfNotSet( style.has<ExtrusionSymbol>() );
// do an initial clamp to get started.
clampMesh( getMapNode()->getTerrain()->getGraph() );
}
applyRenderSymbology( style );
}
}
updateClusterCulling();
}
bool
ExtrudeGeometryFilter::process( FeatureList& features, FilterContext& context )
{
// seed our random number generators
Random wallSkinPRNG( _wallSkinSymbol.valid()? *_wallSkinSymbol->randomSeed() : 0, Random::METHOD_FAST );
Random roofSkinPRNG( _roofSkinSymbol.valid()? *_roofSkinSymbol->randomSeed() : 0, Random::METHOD_FAST );
for( FeatureList::iterator f = features.begin(); f != features.end(); ++f )
{
Feature* input = f->get();
GeometryIterator iter( input->getGeometry(), false );
while( iter.hasMore() )
{
Geometry* part = iter.next();
osg::ref_ptr<osg::Geometry> walls = new osg::Geometry();
walls->setUseVertexBufferObjects( _useVertexBufferObjects.get() );
osg::ref_ptr<osg::Geometry> rooflines = 0L;
osg::ref_ptr<osg::Geometry> baselines = 0L;
osg::ref_ptr<osg::Geometry> outlines = 0L;
if ( part->getType() == Geometry::TYPE_POLYGON )
{
rooflines = new osg::Geometry();
rooflines->setUseVertexBufferObjects( _useVertexBufferObjects.get() );
// prep the shapes by making sure all polys are open:
static_cast<Polygon*>(part)->open();
}
// fire up the outline geometry if we have a line symbol.
if ( _outlineSymbol != 0L )
{
outlines = new osg::Geometry();
outlines->setUseVertexBufferObjects( _useVertexBufferObjects.get() );
}
// make a base cap if we're doing stencil volumes.
if ( _makeStencilVolume )
{
baselines = new osg::Geometry();
baselines->setUseVertexBufferObjects( _useVertexBufferObjects.get() );
}
// calculate the extrusion height:
float height;
if ( _heightCallback.valid() )
{
height = _heightCallback->operator()(input, context);
}
else if ( _heightExpr.isSet() )
{
height = input->eval( _heightExpr.mutable_value(), &context );
}
else
{
height = *_extrusionSymbol->height();
}
// calculate the height offset from the base:
float offset = 0.0;
if ( _heightOffsetExpr.isSet() )
{
offset = input->eval( _heightOffsetExpr.mutable_value(), &context );
}
osg::ref_ptr<osg::StateSet> wallStateSet;
osg::ref_ptr<osg::StateSet> roofStateSet;
// calculate the wall texturing:
SkinResource* wallSkin = 0L;
if ( _wallSkinSymbol.valid() )
{
if ( _wallResLib.valid() )
{
SkinSymbol querySymbol( *_wallSkinSymbol.get() );
querySymbol.objectHeight() = fabs(height) - offset;
wallSkin = _wallResLib->getSkin( &querySymbol, wallSkinPRNG, context.getDBOptions() );
}
else
{
//TODO: simple single texture?
}
}
// calculate the rooftop texture:
SkinResource* roofSkin = 0L;
if ( _roofSkinSymbol.valid() )
{
if ( _roofResLib.valid() )
{
SkinSymbol querySymbol( *_roofSkinSymbol.get() );
roofSkin = _roofResLib->getSkin( &querySymbol, roofSkinPRNG, context.getDBOptions() );
}
else
{
//TODO: simple single texture?
}
}
// Build the data model for the structure.
Structure structure;
buildStructure(
part,
height,
offset,
_extrusionSymbol->flatten().get(),
wallSkin,
roofSkin,
structure,
context);
// Create the walls.
if ( walls.valid() )
{
osg::Vec4f wallColor(1,1,1,1), wallBaseColor(1,1,1,1);
if ( _wallPolygonSymbol.valid() )
{
wallColor = _wallPolygonSymbol->fill()->color();
}
if ( _extrusionSymbol->wallGradientPercentage().isSet() )
{
wallBaseColor = Color(wallColor).brightness( 1.0 - *_extrusionSymbol->wallGradientPercentage() );
}
else
{
wallBaseColor = wallColor;
}
buildWallGeometry(structure, walls.get(), wallColor, wallBaseColor, wallSkin);
if ( wallSkin )
{
// Get a stateset for the individual wall stateset
context.resourceCache()->getOrCreateStateSet( wallSkin, wallStateSet );
}
}
// tessellate and add the roofs if necessary:
if ( rooflines.valid() )
{
osg::Vec4f roofColor(1,1,1,1);
if ( _roofPolygonSymbol.valid() )
{
roofColor = _roofPolygonSymbol->fill()->color();
}
buildRoofGeometry(structure, rooflines.get(), roofColor, roofSkin);
if ( roofSkin )
{
// Get a stateset for the individual roof skin
context.resourceCache()->getOrCreateStateSet( roofSkin, roofStateSet );
}
}
if ( outlines.valid() )
{
osg::Vec4f outlineColor(1,1,1,1);
if ( _outlineSymbol.valid() )
{
outlineColor = _outlineSymbol->stroke()->color();
}
float minCreaseAngle = _outlineSymbol->creaseAngle().value();
buildOutlineGeometry(structure, outlines.get(), outlineColor, minCreaseAngle);
}
if ( baselines.valid() )
{
//TODO.
osgUtil::Tessellator tess;
tess.setTessellationType( osgUtil::Tessellator::TESS_TYPE_GEOMETRY );
tess.setWindingType( osgUtil::Tessellator::TESS_WINDING_ODD );
tess.retessellatePolygons( *(baselines.get()) );
}
// Set up for feature naming and feature indexing:
std::string name;
if ( !_featureNameExpr.empty() )
name = input->eval( _featureNameExpr, &context );
FeatureSourceIndex* index = context.featureIndex();
if ( walls.valid() )
{
addDrawable( walls.get(), wallStateSet.get(), name, input, index );
}
if ( rooflines.valid() )
{
addDrawable( rooflines.get(), roofStateSet.get(), name, input, index );
}
if ( baselines.valid() )
{
addDrawable( baselines.get(), 0L, name, input, index );
}
if ( outlines.valid() )
{
addDrawable( outlines.get(), 0L, name, input, index );
}
}
}
return true;
}
/**
* Creates a complete set of positioned label nodes from a feature list.
*/
osg::Node* createNode(
const FeatureList& input,
const Style& style,
const FilterContext& context )
{
const TextSymbol* text = style.get<TextSymbol>();
if ( !text )
return 0L;
osg::Group* group = new osg::Group();
Decluttering::setEnabled( group->getOrCreateStateSet(), true );
if ( text->priority().isSet() )
{
DeclutteringOptions dco = Decluttering::getOptions();
dco.sortByPriority() = text->priority().isSet();
Decluttering::setOptions( dco );
}
StringExpression contentExpr ( *text->content() );
NumericExpression priorityExpr( *text->priority() );
if ( text->removeDuplicateLabels() == true )
{
// in remove-duplicates mode, make a list of unique features, selecting
// the one with the largest area as the one we'll use for labeling.
typedef std::pair<double, osg::ref_ptr<const Feature> > Entry;
typedef std::map<std::string, Entry> EntryMap;
EntryMap used;
for( FeatureList::const_iterator i = input.begin(); i != input.end(); ++i )
{
Feature* feature = i->get();
if ( feature && feature->getGeometry() )
{
const std::string& value = feature->eval( contentExpr );
if ( !value.empty() )
{
double area = feature->getGeometry()->getBounds().area2d();
if ( used.find(value) == used.end() )
{
used[value] = Entry(area, feature);
}
else
{
Entry& biggest = used[value];
if ( area > biggest.first )
{
biggest.first = area;
biggest.second = feature;
}
}
}
}
}
for( EntryMap::iterator i = used.begin(); i != used.end(); ++i )
{
const std::string& value = i->first;
const Feature* feature = i->second.second.get();
group->addChild( makeLabelNode(context, feature, value, text, priorityExpr) );
}
}
else
{
for( FeatureList::const_iterator i = input.begin(); i != input.end(); ++i )
{
const Feature* feature = i->get();
if ( !feature )
continue;
const Geometry* geom = feature->getGeometry();
if ( !geom )
continue;
const std::string& value = feature->eval( contentExpr, &context );
if ( value.empty() )
continue;
group->addChild( makeLabelNode(context, feature, value, text, priorityExpr) );
}
}
#if 0 // good idea but needs work.
DepthOffsetGroup* dog = new DepthOffsetGroup();
dog->setMinimumOffset( 500.0 );
dog->addChild( group );
return dog;
#endif
return group;
}
osg::Node* BuildTextOperator::operator()(const FeatureList& features,
const TextSymbol* symbol,
const FilterContext& context)
{
if (!symbol) return 0;
std::set< std::string > labelNames;
bool removeDuplicateLabels = symbol->removeDuplicateLabels().isSet() ? symbol->removeDuplicateLabels().get() : false;
osg::Geode* result = new osg::Geode;
for (FeatureList::const_iterator itr = features.begin(); itr != features.end(); ++itr)
{
Feature* feature = itr->get();
if (!feature->getGeometry()) continue;
std::string text;
//If the feature is a TextAnnotation, just get the value from it
TextAnnotation* annotation = dynamic_cast<TextAnnotation*>(feature);
if (annotation)
{
text = annotation->text();
}
else if (symbol->content().isSet())
{
//Get the text from the specified content and referenced attributes
std::string content = symbol->content().value();
text = parseAttributes(feature, content, symbol->contentAttributeDelimiter().value());
}
else if (symbol->attribute().isSet())
{
//Get the text from the specified attribute
std::string attr = symbol->attribute().value();
text = feature->getAttr(attr);
}
if (text.empty()) continue;
//See if there is a duplicate name
if (removeDuplicateLabels && labelNames.find(text) != labelNames.end()) continue;
bool rotateToScreen = symbol->rotateToScreen().isSet() ? symbol->rotateToScreen().value() : false;
// find the centroid
osg::Vec3d position;
osg::Quat orientation;
GeometryIterator gi( feature->getGeometry() );
while( gi.hasMore() )
{
Geometry* geom = gi.next();
TextSymbol::LinePlacement linePlacement = symbol->linePlacement().isSet() ? symbol->linePlacement().get() : TextSymbol::LINEPLACEMENT_ALONG_LINE;
if (geom->getType() == Symbology::Geometry::TYPE_LINESTRING && linePlacement == TextSymbol::LINEPLACEMENT_ALONG_LINE)
{
//Compute the "middle" of the line string
LineString* lineString = static_cast<LineString*>(geom);
double length = lineString->getLength();
double center = length / 2.0;
osg::Vec3d start, end;
if (lineString->getSegment(center, start, end))
{
TextSymbol::LineOrientation lineOrientation = symbol->lineOrientation().isSet() ? symbol->lineOrientation().get() : TextSymbol::LINEORIENTATION_HORIZONTAL;
position = (end + start) / 2.0;
//We don't want to orient the text at all if we are rotating to the screen
if (!rotateToScreen && lineOrientation != TextSymbol::LINEORIENTATION_HORIZONTAL)
{
osg::Vec3d dir = (end-start);
dir.normalize();
if (lineOrientation == TextSymbol::LINEORIENTATION_PERPENDICULAR)
{
osg::Vec3d up(0,0,1);
const SpatialReference* srs = context.profile()->getSRS();
if (srs && context.isGeocentric() && srs->getEllipsoid())
{
osg::Vec3d w = context.toWorld( position );
up = srs->getEllipsoid()->computeLocalUpVector(w.x(), w.y(), w.z());
}
dir = up ^ dir;
}
orientation.makeRotate(osg::Vec3d(1,0,0), dir);
}
}
else
{
//Fall back on using the center
position = lineString->getBounds().center();
}
}
else
{
position = geom->getBounds().center();
}
}
osgText::Text* t = new osgText::Text();
t->setText( text );
std::string font = "fonts/arial.ttf";
if (symbol->font().isSet() && !symbol->font().get().empty())
{
font = symbol->font().value();
}
t->setFont( font );
t->setAutoRotateToScreen( rotateToScreen );
TextSymbol::SizeMode sizeMode = symbol->sizeMode().isSet() ? symbol->sizeMode().get() : TextSymbol::SIZEMODE_SCREEN;
if (sizeMode == TextSymbol::SIZEMODE_SCREEN) {
t->setCharacterSizeMode( osgText::TextBase::SCREEN_COORDS );
}
else if (sizeMode == TextSymbol::SIZEMODE_OBJECT) {
t->setCharacterSizeMode( osgText::TextBase::OBJECT_COORDS );
}
float size = symbol->size().isSet() ? symbol->size().get() : 32.0f;
t->setCharacterSize( size );
//t->setCharacterSizeMode( osgText::TextBase::OBJECT_COORDS_WITH_MAXIMUM_SCREEN_SIZE_CAPPED_BY_FONT_HEIGHT );
//t->setCharacterSize( 300000.0f );
t->setPosition( position );
t->setRotation( orientation);
t->setAlignment( osgText::TextBase::CENTER_CENTER );
t->getOrCreateStateSet()->setAttributeAndModes( new osg::Depth(osg::Depth::ALWAYS), osg::StateAttribute::ON );
t->getOrCreateStateSet()->setRenderBinDetails( 99999, "RenderBin" );
// apply styling as appropriate:
osg::Vec4f textColor = symbol->fill()->color();
osg::Vec4f haloColor = symbol->halo()->color();
t->setColor( textColor );
t->setBackdropColor( haloColor );
t->setBackdropType( osgText::Text::OUTLINE );
if ( context.isGeocentric() )
{
// install a cluster culler
t->setCullCallback( new CullPlaneCallback( position * context.inverseReferenceFrame() ) );
}
result->addDrawable( t );
if (removeDuplicateLabels) labelNames.insert(text);
}
return result;
}
osg::Group*
createLabels( Map* map )
{
osg::ref_ptr<osg::Group> labels = new osg::Group();
// first, open up the source shapefile
OGRFeatureOptions fo;
fo.url() = g_featureFile;
osg::ref_ptr<FeatureSource> features = FeatureSourceFactory::create( fo );
if ( !features.valid() )
{
OE_WARN << LC << "Unable to load features!" << std::endl;
return 0L;
}
features->initialize( "" );
const FeatureProfile* featureProfile = features->getFeatureProfile();
if ( !featureProfile || !featureProfile->getSRS() )
{
OE_WARN << LC << "Feature data has no spatial reference!" << std::endl;
return 0L;
}
osg::ref_ptr<FeatureCursor> cursor = features->createFeatureCursor();
if ( !cursor.valid() )
{
OE_WARN << LC << "Failed to query the feature source!" << std::endl;
return 0L;
}
//SceneControlBin* priorityBin = canvas->getSceneControls();
unsigned count = 0;
std::set<std::string> antiDupeSet;
while( cursor->hasMore() )
{
Feature* feature = cursor->nextFeature();
Geometry* geom = feature->getGeometry();
if ( !geom )
continue;
// we will display the country name:
std::string text = feature->getString( g_labelAttr );
if ( text.empty() )
continue;
// and use the population to prioritize labels:
float population = feature->getDouble(g_priorityAttr, 0.0);
// remove duplicate labels:
if ( g_removeDupes )
{
if ( antiDupeSet.find(text) != antiDupeSet.end() )
continue;
antiDupeSet.insert(text);
}
// calculate the world location of the label:
osg::Vec3d centerPoint = geom->getBounds().center();
osg::Vec3d mapPoint;
if ( !map->toMapPoint( centerPoint, featureProfile->getSRS(), mapPoint ) )
continue;
osg::Vec3d worldPoint;
if ( !map->mapPointToGeocentricPoint( mapPoint, worldPoint ) )
continue;
// create the label and place it:
osg::MatrixTransform* xform = new osg::MatrixTransform( osg::Matrix::translate(worldPoint) );
xform->setCullCallback( new CullNodeByNormal(worldPoint) );
xform->addChild( new ControlNode(new LabelControl(text)) );
labels->addChild( xform );
++count;
//OE_NOTICE << LC << "Added: " << text << std::endl;
}
OE_NOTICE << LC << "Found " << count << " features. " << std::endl;
return labels.release();
}
