2011-05-06 23:55:38 +00:00
/*
2012-02-03 03:38:30 +00:00
Open Asset Import Library ( assimp )
2011-05-06 23:55:38 +00:00
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
2012-02-03 03:38:30 +00:00
Copyright ( c ) 2006 - 2012 , assimp team
2011-05-06 23:55:38 +00:00
All rights reserved .
Redistribution and use of this software in source and binary forms ,
with or without modification , are permitted provided that the
following conditions are met :
* Redistributions of source code must retain the above
copyright notice , this list of conditions and the
following disclaimer .
* Redistributions in binary form must reproduce the above
copyright notice , this list of conditions and the
following disclaimer in the documentation and / or other
materials provided with the distribution .
2012-02-03 03:38:30 +00:00
* Neither the name of the assimp team , nor the names of its
2011-05-06 23:55:38 +00:00
contributors may be used to endorse or promote products
derived from this software without specific prior
2012-02-03 03:38:30 +00:00
written permission of the assimp team .
2011-05-06 23:55:38 +00:00
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
" AS IS " AND ANY EXPRESS OR IMPLIED WARRANTIES , INCLUDING , BUT NOT
LIMITED TO , THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
A PARTICULAR PURPOSE ARE DISCLAIMED . IN NO EVENT SHALL THE COPYRIGHT
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT , INDIRECT , INCIDENTAL ,
SPECIAL , EXEMPLARY , OR CONSEQUENTIAL DAMAGES ( INCLUDING , BUT NOT
LIMITED TO , PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ; LOSS OF USE ,
DATA , OR PROFITS ; OR BUSINESS INTERRUPTION ) HOWEVER CAUSED AND ON ANY
THEORY OF LIABILITY , WHETHER IN CONTRACT , STRICT LIABILITY , OR TORT
( INCLUDING NEGLIGENCE OR OTHERWISE ) ARISING IN ANY WAY OUT OF THE USE
OF THIS SOFTWARE , EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE .
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
*/
2011-06-14 15:58:22 +00:00
/** @file IFCLoad.cpp
2011-05-18 18:08:18 +00:00
* @ brief Implementation of the Industry Foundation Classes loader .
2011-05-06 23:55:38 +00:00
*/
# include "AssimpPCH.h"
# ifndef ASSIMP_BUILD_NO_IFC_IMPORTER
2011-05-13 00:52:50 +00:00
# include <iterator>
# include <boost/tuple/tuple.hpp>
2013-03-28 12:38:42 +00:00
# ifndef ASSIMP_BUILD_NO_COMPRESSED_IFC
# include ".. / contrib / unzip / unzip.h"
# endif
2011-05-13 00:52:50 +00:00
2011-05-06 23:55:38 +00:00
# include "IFCLoader.h"
# include "STEPFileReader.h"
2011-06-14 15:58:22 +00:00
# include "IFCUtil.h"
2011-05-13 00:52:50 +00:00
2011-05-06 23:55:38 +00:00
# include "StreamReader.h"
# include "MemoryIOWrapper.h"
2011-05-09 18:00:55 +00:00
2012-03-20 21:31:56 +00:00
namespace Assimp {
template < > const std : : string LogFunctions < IFCImporter > : : log_prefix = " IFC: " ;
}
2011-05-06 23:55:38 +00:00
using namespace Assimp ;
using namespace Assimp : : Formatter ;
2011-06-14 15:58:22 +00:00
using namespace Assimp : : IFC ;
2011-05-06 23:55:38 +00:00
/* DO NOT REMOVE this comment block. The genentitylist.sh script
2011-06-14 15:58:22 +00:00
* just looks for names adhering to the IfcSomething naming scheme
2011-05-06 23:55:38 +00:00
* and includes all matches in the whitelist for code - generation . Thus ,
* all entity classes that are only indirectly referenced need to be
* mentioned explicitly .
2011-06-14 15:58:22 +00:00
IfcRepresentationMap
IfcProductRepresentation
IfcUnitAssignment
IfcClosedShell
IfcDoor
2011-05-06 23:55:38 +00:00
*/
namespace {
// forward declarations
void SetUnits ( ConversionData & conv ) ;
void SetCoordinateSpace ( ConversionData & conv ) ;
void ProcessSpatialStructures ( ConversionData & conv ) ;
2011-06-14 15:58:22 +00:00
aiNode * ProcessSpatialStructure ( aiNode * parent , const IfcProduct & el , ConversionData & conv ) ;
void ProcessProductRepresentation ( const IfcProduct & el , aiNode * nd , ConversionData & conv ) ;
2011-05-06 23:55:38 +00:00
void MakeTreeRelative ( ConversionData & conv ) ;
2011-05-31 14:23:43 +00:00
void ConvertUnit ( const EXPRESS : : DataType & dt , ConversionData & conv ) ;
2011-05-06 23:55:38 +00:00
} // anon
2012-04-22 22:26:26 +00:00
static const aiImporterDesc desc = {
" Industry Foundation Classes (IFC) Importer " ,
" " ,
" " ,
" " ,
aiImporterFlags_SupportBinaryFlavour ,
0 ,
0 ,
0 ,
0 ,
2012-07-11 19:18:51 +00:00
" ifc ifczip "
2012-04-22 22:26:26 +00:00
} ;
2011-05-06 23:55:38 +00:00
// ------------------------------------------------------------------------------------------------
// Constructor to be privately used by Importer
IFCImporter : : IFCImporter ( )
{ }
// ------------------------------------------------------------------------------------------------
// Destructor, private as well
IFCImporter : : ~ IFCImporter ( )
{
}
// ------------------------------------------------------------------------------------------------
// Returns whether the class can handle the format of the given file.
bool IFCImporter : : CanRead ( const std : : string & pFile , IOSystem * pIOHandler , bool checkSig ) const
{
const std : : string & extension = GetExtension ( pFile ) ;
2012-07-11 19:18:51 +00:00
if ( extension = = " ifc " | | extension = = " ifczip " ) {
2011-05-06 23:55:38 +00:00
return true ;
}
else if ( ( ! extension . length ( ) | | checkSig ) & & pIOHandler ) {
// note: this is the common identification for STEP-encoded files, so
// it is only unambiguous as long as we don't support any further
// file formats with STEP as their encoding.
const char * tokens [ ] = { " ISO-10303-21 " } ;
return SearchFileHeaderForToken ( pIOHandler , pFile , tokens , 1 ) ;
}
return false ;
}
// ------------------------------------------------------------------------------------------------
// List all extensions handled by this loader
2012-04-22 22:26:26 +00:00
const aiImporterDesc * IFCImporter : : GetInfo ( ) const
2011-05-06 23:55:38 +00:00
{
2012-04-22 22:26:26 +00:00
return & desc ;
2011-05-06 23:55:38 +00:00
}
// ------------------------------------------------------------------------------------------------
// Setup configuration properties for the loader
void IFCImporter : : SetupProperties ( const Importer * pImp )
{
settings . skipSpaceRepresentations = pImp - > GetPropertyBool ( AI_CONFIG_IMPORT_IFC_SKIP_SPACE_REPRESENTATIONS , true ) ;
2011-05-09 18:39:13 +00:00
settings . skipCurveRepresentations = pImp - > GetPropertyBool ( AI_CONFIG_IMPORT_IFC_SKIP_CURVE_REPRESENTATIONS , true ) ;
2011-05-18 18:08:18 +00:00
settings . useCustomTriangulation = pImp - > GetPropertyBool ( AI_CONFIG_IMPORT_IFC_CUSTOM_TRIANGULATION , true ) ;
2011-06-14 15:58:22 +00:00
2012-07-12 00:20:28 +00:00
settings . conicSamplingAngle = 10.f ;
2011-06-14 15:58:22 +00:00
settings . skipAnnotations = true ;
2011-05-06 23:55:38 +00:00
}
// ------------------------------------------------------------------------------------------------
// Imports the given file into the given scene structure.
void IFCImporter : : InternReadFile ( const std : : string & pFile ,
aiScene * pScene , IOSystem * pIOHandler )
{
2011-05-07 23:41:17 +00:00
boost : : shared_ptr < IOStream > stream ( pIOHandler - > Open ( pFile ) ) ;
2011-05-06 23:55:38 +00:00
if ( ! stream ) {
ThrowException ( " Could not open file for reading " ) ;
}
2013-03-28 12:38:42 +00:00
2012-07-11 19:18:51 +00:00
// if this is a ifczip file, decompress its contents first
if ( GetExtension ( pFile ) = = " ifczip " ) {
2013-03-28 12:38:42 +00:00
# ifndef ASSIMP_BUILD_NO_COMPRESSED_IFC
2012-07-11 19:18:51 +00:00
unzFile zip = unzOpen ( pFile . c_str ( ) ) ;
if ( zip = = NULL ) {
ThrowException ( " Could not open ifczip file for reading, unzip failed " ) ;
}
// chop 'zip' postfix
std : : string fileName = pFile . substr ( 0 , pFile . length ( ) - 3 ) ;
std : : string : : size_type s = pFile . find_last_of ( ' \\ ' ) ;
if ( s = = std : : string : : npos ) {
s = pFile . find_last_of ( ' / ' ) ;
}
if ( s ! = std : : string : : npos ) {
fileName = fileName . substr ( s + 1 ) ;
}
// search file (same name as the IFCZIP except for the file extension) and place file pointer there
2013-06-24 22:22:47 +00:00
if ( UNZ_OK = = unzGoToFirstFile ( zip ) ) {
do {
//
// get file size, etc.
unz_file_info fileInfo ;
char filename [ 256 ] ;
unzGetCurrentFileInfo ( zip , & fileInfo , filename , sizeof ( filename ) , 0 , 0 , 0 , 0 ) ;
if ( GetExtension ( filename ) ! = " ifc " ) {
continue ;
}
uint8_t * buff = new uint8_t [ fileInfo . uncompressed_size ] ;
LogInfo ( " Decompressing IFCZIP file " ) ;
unzOpenCurrentFile ( zip ) ;
const int ret = unzReadCurrentFile ( zip , buff , fileInfo . uncompressed_size ) ;
size_t filesize = fileInfo . uncompressed_size ;
if ( ret < 0 | | size_t ( ret ) ! = filesize )
{
delete [ ] buff ;
ThrowException ( " Failed to decompress IFC ZIP file " ) ;
}
unzCloseCurrentFile ( zip ) ;
stream . reset ( new MemoryIOStream ( buff , fileInfo . uncompressed_size , true ) ) ;
break ;
if ( unzGoToNextFile ( zip ) = = UNZ_END_OF_LIST_OF_FILE ) {
ThrowException ( " Found no IFC file member in IFCZIP file (1) " ) ;
}
} while ( true ) ;
2012-07-11 19:18:51 +00:00
}
else {
2013-06-24 22:22:47 +00:00
ThrowException ( " Found no IFC file member in IFCZIP file (2) " ) ;
2012-07-11 19:18:51 +00:00
}
unzClose ( zip ) ;
2013-03-28 12:38:42 +00:00
# else
ThrowException ( " Could not open ifczip file for reading, assimp was built without ifczip support " ) ;
# endif
2012-07-11 19:18:51 +00:00
}
2011-05-06 23:55:38 +00:00
boost : : scoped_ptr < STEP : : DB > db ( STEP : : ReadFileHeader ( stream ) ) ;
2012-03-20 21:11:05 +00:00
const STEP : : HeaderInfo & head = static_cast < const STEP : : DB & > ( * db ) . GetHeader ( ) ;
2011-05-06 23:55:38 +00:00
if ( ! head . fileSchema . size ( ) | | head . fileSchema . substr ( 0 , 3 ) ! = " IFC " ) {
ThrowException ( " Unrecognized file schema: " + head . fileSchema ) ;
}
if ( ! DefaultLogger : : isNullLogger ( ) ) {
LogDebug ( " File schema is \' " + head . fileSchema + ' \' ' ) ;
if ( head . timestamp . length ( ) ) {
LogDebug ( " Timestamp \' " + head . timestamp + ' \' ' ) ;
}
if ( head . app . length ( ) ) {
LogDebug ( " Application/Exporter identline is \' " + head . app + ' \' ' ) ;
}
}
// obtain a copy of the machine-generated IFC scheme
EXPRESS : : ConversionSchema schema ;
2011-06-14 15:58:22 +00:00
GetSchema ( schema ) ;
2011-05-06 23:55:38 +00:00
2011-05-21 15:38:39 +00:00
// tell the reader which entity types to track with special care
static const char * const types_to_track [ ] = {
" ifcsite " , " ifcbuilding " , " ifcproject "
} ;
2011-05-30 15:38:48 +00:00
// tell the reader for which types we need to simulate STEPs reverse indices
static const char * const inverse_indices_to_track [ ] = {
2013-03-28 22:28:49 +00:00
" ifcrelcontainedinspatialstructure " , " ifcrelaggregates " , " ifcrelvoidselement " , " ifcreldefinesbyproperties " , " ifcpropertyset " , " ifcstyleditem "
2011-05-30 15:38:48 +00:00
} ;
2011-05-06 23:55:38 +00:00
// feed the IFC schema into the reader and pre-parse all lines
2011-05-30 15:38:48 +00:00
STEP : : ReadFile ( * db , schema , types_to_track , inverse_indices_to_track ) ;
2011-05-06 23:55:38 +00:00
const STEP : : LazyObject * proj = db - > GetObject ( " ifcproject " ) ;
if ( ! proj ) {
ThrowException ( " missing IfcProject entity " ) ;
}
2011-06-14 15:58:22 +00:00
ConversionData conv ( * db , proj - > To < IfcProject > ( ) , pScene , settings ) ;
2011-05-06 23:55:38 +00:00
SetUnits ( conv ) ;
SetCoordinateSpace ( conv ) ;
ProcessSpatialStructures ( conv ) ;
MakeTreeRelative ( conv ) ;
2011-05-13 00:52:50 +00:00
// NOTE - this is a stress test for the importer, but it works only
// in a build with no entities disabled. See
// scripts/IFCImporter/CPPGenerator.py
// for more information.
2011-05-06 23:55:38 +00:00
# ifdef ASSIMP_IFC_TEST
db - > EvaluateAll ( ) ;
# endif
// do final data copying
if ( conv . meshes . size ( ) ) {
pScene - > mNumMeshes = static_cast < unsigned int > ( conv . meshes . size ( ) ) ;
pScene - > mMeshes = new aiMesh * [ pScene - > mNumMeshes ] ( ) ;
std : : copy ( conv . meshes . begin ( ) , conv . meshes . end ( ) , pScene - > mMeshes ) ;
// needed to keep the d'tor from burning us
conv . meshes . clear ( ) ;
}
if ( conv . materials . size ( ) ) {
pScene - > mNumMaterials = static_cast < unsigned int > ( conv . materials . size ( ) ) ;
pScene - > mMaterials = new aiMaterial * [ pScene - > mNumMaterials ] ( ) ;
std : : copy ( conv . materials . begin ( ) , conv . materials . end ( ) , pScene - > mMaterials ) ;
// needed to keep the d'tor from burning us
conv . materials . clear ( ) ;
}
2011-05-08 20:08:28 +00:00
// apply world coordinate system (which includes the scaling to convert to meters and a -90 degrees rotation around x)
2011-05-07 13:14:56 +00:00
aiMatrix4x4 scale , rot ;
2012-02-02 02:09:49 +00:00
aiMatrix4x4 : : Scaling ( static_cast < aiVector3D > ( IfcVector3 ( conv . len_scale ) ) , scale ) ;
2011-05-07 13:14:56 +00:00
aiMatrix4x4 : : RotationX ( - AI_MATH_HALF_PI_F , rot ) ;
pScene - > mRootNode - > mTransformation = rot * scale * conv . wcs * pScene - > mRootNode - > mTransformation ;
2011-05-06 23:55:38 +00:00
// this must be last because objects are evaluated lazily as we process them
if ( ! DefaultLogger : : isNullLogger ( ) ) {
LogDebug ( ( Formatter : : format ( ) , " STEP: evaluated " , db - > GetEvaluatedObjectCount ( ) , " object records " ) ) ;
}
}
namespace {
// ------------------------------------------------------------------------------------------------
2011-06-14 15:58:22 +00:00
void ConvertUnit ( const IfcNamedUnit & unit , ConversionData & conv )
2011-05-06 23:55:38 +00:00
{
2011-06-14 15:58:22 +00:00
if ( const IfcSIUnit * const si = unit . ToPtr < IfcSIUnit > ( ) ) {
2011-05-06 23:55:38 +00:00
2011-05-10 12:30:13 +00:00
if ( si - > UnitType = = " LENGTHUNIT " ) {
conv . len_scale = si - > Prefix ? ConvertSIPrefix ( si - > Prefix ) : 1.f ;
IFCImporter : : LogDebug ( " got units used for lengths " ) ;
}
if ( si - > UnitType = = " PLANEANGLEUNIT " ) {
if ( si - > Name ! = " RADIAN " ) {
IFCImporter : : LogWarn ( " expected base unit for angles to be radian " ) ;
2011-05-06 23:55:38 +00:00
}
}
2011-05-10 12:30:13 +00:00
}
2011-06-14 15:58:22 +00:00
else if ( const IfcConversionBasedUnit * const convu = unit . ToPtr < IfcConversionBasedUnit > ( ) ) {
2011-05-10 12:30:13 +00:00
if ( convu - > UnitType = = " PLANEANGLEUNIT " ) {
try {
conv . angle_scale = convu - > ConversionFactor - > ValueComponent - > To < EXPRESS : : REAL > ( ) ;
2011-05-31 14:23:43 +00:00
ConvertUnit ( * convu - > ConversionFactor - > UnitComponent , conv ) ;
2011-05-10 12:30:13 +00:00
IFCImporter : : LogDebug ( " got units used for angles " ) ;
}
catch ( std : : bad_cast & ) {
IFCImporter : : LogError ( " skipping unknown IfcConversionBasedUnit.ValueComponent entry - expected REAL " ) ;
}
2011-05-06 23:55:38 +00:00
}
}
2011-05-10 12:30:13 +00:00
}
// ------------------------------------------------------------------------------------------------
2011-05-31 14:23:43 +00:00
void ConvertUnit ( const EXPRESS : : DataType & dt , ConversionData & conv )
2011-05-10 12:30:13 +00:00
{
try {
2011-06-14 15:58:22 +00:00
const EXPRESS : : ENTITY & e = dt . To < ENTITY > ( ) ;
2011-05-10 12:30:13 +00:00
2011-06-14 15:58:22 +00:00
const IfcNamedUnit & unit = e . ResolveSelect < IfcNamedUnit > ( conv . db ) ;
2011-05-10 12:30:13 +00:00
if ( unit . UnitType ! = " LENGTHUNIT " & & unit . UnitType ! = " PLANEANGLEUNIT " ) {
return ;
}
ConvertUnit ( unit , conv ) ;
}
catch ( std : : bad_cast & ) {
// not entity, somehow
IFCImporter : : LogError ( " skipping unknown IfcUnit entry - expected entity " ) ;
}
}
2011-05-06 23:55:38 +00:00
2011-05-10 12:30:13 +00:00
// ------------------------------------------------------------------------------------------------
void SetUnits ( ConversionData & conv )
{
// see if we can determine the coordinate space used to express.
for ( size_t i = 0 ; i < conv . proj . UnitsInContext - > Units . size ( ) ; + + i ) {
2011-05-31 14:23:43 +00:00
ConvertUnit ( * conv . proj . UnitsInContext - > Units [ i ] , conv ) ;
2011-05-10 12:30:13 +00:00
}
2011-05-06 23:55:38 +00:00
}
// ------------------------------------------------------------------------------------------------
2011-06-14 15:58:22 +00:00
void SetCoordinateSpace ( ConversionData & conv )
2011-05-06 23:55:38 +00:00
{
2011-06-14 15:58:22 +00:00
const IfcRepresentationContext * fav = NULL ;
BOOST_FOREACH ( const IfcRepresentationContext & v , conv . proj . RepresentationContexts ) {
fav = & v ;
// Model should be the most suitable type of context, hence ignore the others
if ( v . ContextType & & v . ContextType . Get ( ) = = " Model " ) {
break ;
2011-05-06 23:55:38 +00:00
}
}
2011-06-14 15:58:22 +00:00
if ( fav ) {
if ( const IfcGeometricRepresentationContext * const geo = fav - > ToPtr < IfcGeometricRepresentationContext > ( ) ) {
ConvertAxisPlacement ( conv . wcs , * geo - > WorldCoordinateSystem , conv ) ;
IFCImporter : : LogDebug ( " got world coordinate system " ) ;
}
2011-05-06 23:55:38 +00:00
}
}
2011-06-14 15:58:22 +00:00
2011-05-06 23:55:38 +00:00
// ------------------------------------------------------------------------------------------------
2011-06-14 15:58:22 +00:00
void ResolveObjectPlacement ( aiMatrix4x4 & m , const IfcObjectPlacement & place , ConversionData & conv )
2011-05-06 23:55:38 +00:00
{
2011-06-14 15:58:22 +00:00
if ( const IfcLocalPlacement * const local = place . ToPtr < IfcLocalPlacement > ( ) ) {
2012-02-02 22:50:24 +00:00
IfcMatrix4 tmp ;
ConvertAxisPlacement ( tmp , * local - > RelativePlacement , conv ) ;
m = static_cast < aiMatrix4x4 > ( tmp ) ;
2011-06-14 15:58:22 +00:00
if ( local - > PlacementRelTo ) {
aiMatrix4x4 tmp ;
ResolveObjectPlacement ( tmp , local - > PlacementRelTo . Get ( ) , conv ) ;
m = tmp * m ;
}
}
else {
IFCImporter : : LogWarn ( " skipping unknown IfcObjectPlacement entity, type is " + place . GetClassName ( ) ) ;
2011-05-06 23:55:38 +00:00
}
}
// ------------------------------------------------------------------------------------------------
2011-06-14 15:58:22 +00:00
void GetAbsTransform ( aiMatrix4x4 & out , const aiNode * nd , ConversionData & conv )
2011-05-06 23:55:38 +00:00
{
2011-06-14 15:58:22 +00:00
aiMatrix4x4 t ;
if ( nd - > mParent ) {
GetAbsTransform ( t , nd - > mParent , conv ) ;
2011-05-06 23:55:38 +00:00
}
2011-06-14 15:58:22 +00:00
out = t * nd - > mTransformation ;
2011-05-06 23:55:38 +00:00
}
// ------------------------------------------------------------------------------------------------
2011-06-14 15:58:22 +00:00
bool ProcessMappedItem ( const IfcMappedItem & mapped , aiNode * nd_src , std : : vector < aiNode * > & subnodes_src , ConversionData & conv )
2011-05-06 23:55:38 +00:00
{
2011-06-14 15:58:22 +00:00
// insert a custom node here, the cartesian transform operator is simply a conventional transformation matrix
std : : auto_ptr < aiNode > nd ( new aiNode ( ) ) ;
nd - > mName . Set ( " IfcMappedItem " ) ;
2011-07-17 01:09:56 +00:00
// handle the Cartesian operator
2012-02-02 02:09:49 +00:00
IfcMatrix4 m ;
2011-06-14 15:58:22 +00:00
ConvertTransformOperator ( m , * mapped . MappingTarget ) ;
2011-05-06 23:55:38 +00:00
2012-02-02 02:09:49 +00:00
IfcMatrix4 msrc ;
2011-06-14 15:58:22 +00:00
ConvertAxisPlacement ( msrc , * mapped . MappingSource - > MappingOrigin , conv ) ;
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
msrc = m * msrc ;
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
std : : vector < unsigned int > meshes ;
const size_t old_openings = conv . collect_openings ? conv . collect_openings - > size ( ) : 0 ;
if ( conv . apply_openings ) {
2012-02-02 02:09:49 +00:00
IfcMatrix4 minv = msrc ;
2011-06-14 15:58:22 +00:00
minv . Inverse ( ) ;
BOOST_FOREACH ( TempOpening & open , * conv . apply_openings ) {
open . Transform ( minv ) ;
}
}
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
const IfcRepresentation & repr = mapped . MappingSource - > MappedRepresentation ;
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
bool got = false ;
BOOST_FOREACH ( const IfcRepresentationItem & item , repr . Items ) {
if ( ! ProcessRepresentationItem ( item , meshes , conv ) ) {
2011-07-17 01:09:56 +00:00
IFCImporter : : LogWarn ( " skipping mapped entity of type " + item . GetClassName ( ) + " , no representations could be generated " ) ;
2011-06-14 15:58:22 +00:00
}
else got = true ;
2011-05-06 23:55:38 +00:00
}
2011-06-14 15:58:22 +00:00
if ( ! got ) {
return false ;
2011-05-06 23:55:38 +00:00
}
2011-06-14 15:58:22 +00:00
AssignAddedMeshes ( meshes , nd . get ( ) , conv ) ;
if ( conv . collect_openings ) {
// if this pass serves us only to collect opening geometry,
// make sure we transform the TempMesh's which we need to
// preserve as well.
if ( const size_t diff = conv . collect_openings - > size ( ) - old_openings ) {
for ( size_t i = 0 ; i < diff ; + + i ) {
( * conv . collect_openings ) [ old_openings + i ] . Transform ( msrc ) ;
}
}
}
2011-05-06 23:55:38 +00:00
2012-02-02 02:09:49 +00:00
nd - > mTransformation = nd_src - > mTransformation * static_cast < aiMatrix4x4 > ( msrc ) ;
2011-06-14 15:58:22 +00:00
subnodes_src . push_back ( nd . release ( ) ) ;
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
return true ;
2011-05-06 23:55:38 +00:00
}
// ------------------------------------------------------------------------------------------------
2011-06-14 15:58:22 +00:00
struct RateRepresentationPredicate {
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
int Rate ( const IfcRepresentation * r ) const {
// the smaller, the better
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
if ( ! r - > RepresentationIdentifier ) {
// neutral choice if no extra information is specified
return 0 ;
}
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
const std : : string & name = r - > RepresentationIdentifier . Get ( ) ;
if ( name = = " MappedRepresentation " ) {
if ( ! r - > Items . empty ( ) ) {
// take the first item and base our choice on it
const IfcMappedItem * const m = r - > Items . front ( ) - > ToPtr < IfcMappedItem > ( ) ;
if ( m ) {
return Rate ( m - > MappingSource - > MappedRepresentation ) ;
}
}
return 100 ;
}
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
return Rate ( name ) ;
2011-05-10 12:30:13 +00:00
}
2011-06-14 15:58:22 +00:00
int Rate ( const std : : string & r ) const {
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
if ( r = = " SolidModel " ) {
return - 3 ;
2011-05-06 23:55:38 +00:00
}
2011-06-14 15:58:22 +00:00
2013-03-29 12:33:18 +00:00
// give strong preference to extruded geometry.
2011-06-14 15:58:22 +00:00
if ( r = = " SweptSolid " ) {
return - 10 ;
}
if ( r = = " Clipping " ) {
return - 5 ;
2011-05-06 23:55:38 +00:00
}
2011-06-14 15:58:22 +00:00
// 'Brep' is difficult to get right due to possible voids in the
// polygon boundaries, so take it only if we are forced to (i.e.
// if the only alternative is (non-clipping) boolean operations,
// which are not supported at all).
if ( r = = " Brep " ) {
return - 2 ;
2011-05-06 23:55:38 +00:00
}
2011-06-14 15:58:22 +00:00
// Curves, bounding boxes - those will most likely not be loaded
// as we can't make any use out of this data. So consider them
// last.
if ( r = = " BoundingBox " | | r = = " Curve2D " ) {
return 100 ;
}
return 0 ;
2011-05-06 23:55:38 +00:00
}
2011-06-14 15:58:22 +00:00
bool operator ( ) ( const IfcRepresentation * a , const IfcRepresentation * b ) const {
2012-01-15 13:51:34 +00:00
return Rate ( a ) < Rate ( b ) ;
2011-05-13 00:52:50 +00:00
}
2011-06-14 15:58:22 +00:00
} ;
2011-05-06 23:55:38 +00:00
// ------------------------------------------------------------------------------------------------
2011-06-14 15:58:22 +00:00
void ProcessProductRepresentation ( const IfcProduct & el , aiNode * nd , std : : vector < aiNode * > & subnodes , ConversionData & conv )
2011-05-06 23:55:38 +00:00
{
2011-06-14 15:58:22 +00:00
if ( ! el . Representation ) {
return ;
2011-05-06 23:55:38 +00:00
}
2011-05-29 19:16:52 +00:00
2011-06-14 15:58:22 +00:00
std : : vector < unsigned int > meshes ;
2011-05-09 18:00:55 +00:00
2011-06-14 15:58:22 +00:00
// we want only one representation type, so bring them in a suitable order (i.e try those
// that look as if we could read them quickly at first). This way of reading
// representation is relatively generic and allows the concrete implementations
// for the different representation types to make some sensible choices what
// to load and what not to load.
const STEP : : ListOf < STEP : : Lazy < IfcRepresentation > , 1 , 0 > & src = el . Representation . Get ( ) - > Representations ;
std : : vector < const IfcRepresentation * > repr_ordered ( src . size ( ) ) ;
std : : copy ( src . begin ( ) , src . end ( ) , repr_ordered . begin ( ) ) ;
std : : sort ( repr_ordered . begin ( ) , repr_ordered . end ( ) , RateRepresentationPredicate ( ) ) ;
BOOST_FOREACH ( const IfcRepresentation * repr , repr_ordered ) {
bool res = false ;
BOOST_FOREACH ( const IfcRepresentationItem & item , repr - > Items ) {
if ( const IfcMappedItem * const geo = item . ToPtr < IfcMappedItem > ( ) ) {
res = ProcessMappedItem ( * geo , nd , subnodes , conv ) | | res ;
}
else {
res = ProcessRepresentationItem ( item , meshes , conv ) | | res ;
}
}
// if we got something meaningful at this point, skip any further representations
if ( res ) {
break ;
}
2011-05-09 18:00:55 +00:00
}
2011-06-14 15:58:22 +00:00
AssignAddedMeshes ( meshes , nd , conv ) ;
2011-05-06 23:55:38 +00:00
}
2013-03-28 22:28:49 +00:00
typedef std : : map < std : : string , std : : string > Metadata ;
// ------------------------------------------------------------------------------------------------
2013-03-29 13:21:06 +00:00
void ProcessMetadata ( const ListOf < Lazy < IfcProperty > , 1 , 0 > & set , ConversionData & conv , Metadata & properties ,
const std : : string & prefix = " " ,
unsigned int nest = 0 )
2013-03-28 22:28:49 +00:00
{
2013-03-29 13:07:28 +00:00
BOOST_FOREACH ( const IfcProperty & property , set ) {
const std : : string & key = prefix . length ( ) > 0 ? ( prefix + " . " + property . Name ) : property . Name ;
if ( const IfcPropertySingleValue * const singleValue = property . ToPtr < IfcPropertySingleValue > ( ) ) {
if ( singleValue - > NominalValue ) {
if ( const EXPRESS : : STRING * str = singleValue - > NominalValue . Get ( ) - > ToPtr < EXPRESS : : STRING > ( ) ) {
std : : string value = static_cast < std : : string > ( * str ) ;
properties [ key ] = value ;
2013-03-28 22:28:49 +00:00
}
2013-03-29 13:07:28 +00:00
else if ( const EXPRESS : : REAL * val = singleValue - > NominalValue . Get ( ) - > ToPtr < EXPRESS : : REAL > ( ) ) {
float value = static_cast < float > ( * val ) ;
std : : stringstream s ;
s < < value ;
properties [ key ] = s . str ( ) ;
2013-03-28 22:28:49 +00:00
}
2013-03-29 13:07:28 +00:00
else if ( const EXPRESS : : INTEGER * val = singleValue - > NominalValue . Get ( ) - > ToPtr < EXPRESS : : INTEGER > ( ) ) {
int64_t value = static_cast < int64_t > ( * val ) ;
std : : stringstream s ;
s < < value ;
properties [ key ] = s . str ( ) ;
2013-03-28 22:28:49 +00:00
}
}
}
2013-03-29 13:07:28 +00:00
else if ( const IfcPropertyListValue * const listValue = property . ToPtr < IfcPropertyListValue > ( ) ) {
std : : stringstream ss ;
ss < < " [ " ;
unsigned index = 0 ;
BOOST_FOREACH ( const IfcValue : : Out & v , listValue - > ListValues ) {
if ( ! v ) continue ;
if ( const EXPRESS : : STRING * str = v - > ToPtr < EXPRESS : : STRING > ( ) ) {
std : : string value = static_cast < std : : string > ( * str ) ;
ss < < " ' " < < value < < " ' " ;
}
else if ( const EXPRESS : : REAL * val = v - > ToPtr < EXPRESS : : REAL > ( ) ) {
float value = static_cast < float > ( * val ) ;
ss < < value ;
}
else if ( const EXPRESS : : INTEGER * val = v - > ToPtr < EXPRESS : : INTEGER > ( ) ) {
int64_t value = static_cast < int64_t > ( * val ) ;
ss < < value ;
}
if ( index + 1 < listValue - > ListValues . size ( ) ) {
ss < < " , " ;
}
index + + ;
}
ss < < " ] " ;
properties [ key ] = ss . str ( ) ;
}
else if ( const IfcComplexProperty * const complexProp = property . ToPtr < IfcComplexProperty > ( ) ) {
2013-03-29 13:21:06 +00:00
if ( nest > 2 ) { // mostly arbitrary limit to prevent stack overflow vulnerabilities
IFCImporter : : LogError ( " maximum nesting level for IfcComplexProperty reached, skipping this property. " ) ;
}
else {
ProcessMetadata ( complexProp - > HasProperties , conv , properties , key , nest + 1 ) ;
}
2013-03-29 13:07:28 +00:00
}
else {
properties [ key ] = " " ;
}
}
}
// ------------------------------------------------------------------------------------------------
void ProcessMetadata ( uint64_t relDefinesByPropertiesID , ConversionData & conv , Metadata & properties )
{
if ( const IfcRelDefinesByProperties * const pset = conv . db . GetObject ( relDefinesByPropertiesID ) - > ToPtr < IfcRelDefinesByProperties > ( ) ) {
if ( const IfcPropertySet * const set = conv . db . GetObject ( pset - > RelatingPropertyDefinition - > GetID ( ) ) - > ToPtr < IfcPropertySet > ( ) ) {
ProcessMetadata ( set - > HasProperties , conv , properties ) ;
}
2013-03-28 22:28:49 +00:00
}
}
2011-05-21 15:38:39 +00:00
// ------------------------------------------------------------------------------------------------
2011-06-14 15:58:22 +00:00
aiNode * ProcessSpatialStructure ( aiNode * parent , const IfcProduct & el , ConversionData & conv , std : : vector < TempOpening > * collect_openings = NULL )
2011-05-21 15:38:39 +00:00
{
2011-06-14 15:58:22 +00:00
const STEP : : DB : : RefMap & refs = conv . db . GetRefs ( ) ;
2011-05-21 15:38:39 +00:00
2011-06-14 15:58:22 +00:00
// skip over space and annotation nodes - usually, these have no meaning in Assimp's context
if ( conv . settings . skipSpaceRepresentations ) {
if ( const IfcSpace * const space = el . ToPtr < IfcSpace > ( ) ) {
IFCImporter : : LogDebug ( " skipping IfcSpace entity due to importer settings " ) ;
return NULL ;
2011-05-29 19:16:52 +00:00
}
}
2011-06-14 15:58:22 +00:00
if ( conv . settings . skipAnnotations ) {
if ( const IfcAnnotation * const ann = el . ToPtr < IfcAnnotation > ( ) ) {
IFCImporter : : LogDebug ( " skipping IfcAnnotation entity due to importer settings " ) ;
return NULL ;
2011-05-29 19:16:52 +00:00
}
}
2011-05-06 23:55:38 +00:00
// add an output node for this spatial structure
std : : auto_ptr < aiNode > nd ( new aiNode ( ) ) ;
2013-01-27 14:44:57 +00:00
nd - > mName . Set ( el . GetClassName ( ) + " _ " + ( el . Name ? el . Name . Get ( ) : " Unnamed " ) + " _ " + el . GlobalId ) ;
2011-05-06 23:55:38 +00:00
nd - > mParent = parent ;
2013-03-29 12:33:18 +00:00
conv . already_processed . insert ( el . GetID ( ) ) ;
2013-03-28 22:28:49 +00:00
// check for node metadata
STEP : : DB : : RefMapRange children = refs . equal_range ( el . GetID ( ) ) ;
if ( children . first ! = refs . end ( ) ) {
Metadata properties ;
if ( children . first = = children . second ) {
// handles single property set
ProcessMetadata ( ( * children . first ) . second , conv , properties ) ;
}
else {
2013-03-29 12:33:18 +00:00
// handles multiple property sets (currently all property sets are merged,
2013-03-28 22:28:49 +00:00
// which may not be the best solution in the long run)
for ( STEP : : DB : : RefMap : : const_iterator it = children . first ; it ! = children . second ; + + it ) {
ProcessMetadata ( ( * it ) . second , conv , properties ) ;
}
}
if ( ! properties . empty ( ) ) {
aiMetadata * data = new aiMetadata ( ) ;
data - > mNumProperties = properties . size ( ) ;
2014-02-22 11:06:34 +00:00
data - > mKeys = new aiString [ data - > mNumProperties ] ( ) ;
2014-03-15 07:37:45 +00:00
data - > mValues = new aiMetadataEntry [ data - > mNumProperties ] ( ) ;
2014-03-14 13:17:34 +00:00
unsigned int index = 0 ;
BOOST_FOREACH ( const Metadata : : value_type & kv , properties )
data - > Set ( index + + , kv . first , aiString ( kv . second ) ) ;
2013-03-28 22:28:49 +00:00
nd - > mMetaData = data ;
}
}
2011-05-06 23:55:38 +00:00
if ( el . ObjectPlacement ) {
ResolveObjectPlacement ( nd - > mTransformation , el . ObjectPlacement . Get ( ) , conv ) ;
}
2011-05-13 00:52:50 +00:00
std : : vector < TempOpening > openings ;
2012-02-02 02:09:49 +00:00
IfcMatrix4 myInv ;
2011-05-13 00:52:50 +00:00
bool didinv = false ;
2011-05-06 23:55:38 +00:00
// convert everything contained directly within this structure,
// this may result in more nodes.
std : : vector < aiNode * > subnodes ;
try {
// locate aggregates and 'contained-in-here'-elements of this spatial structure and add them in recursively
2011-05-13 00:52:50 +00:00
// on our way, collect openings in *this* element
2011-05-06 23:55:38 +00:00
STEP : : DB : : RefMapRange range = refs . equal_range ( el . GetID ( ) ) ;
2011-05-13 00:52:50 +00:00
for ( STEP : : DB : : RefMapRange range2 = range ; range2 . first ! = range . second ; + + range2 . first ) {
2013-03-29 12:33:18 +00:00
// skip over meshes that have already been processed before. This is strictly necessary
// because the reverse indices also include references contained in argument lists and
// therefore every element has a back-reference hold by its parent.
if ( conv . already_processed . find ( ( * range2 . first ) . second ) ! = conv . already_processed . end ( ) ) {
continue ;
}
2011-05-13 00:52:50 +00:00
const STEP : : LazyObject & obj = conv . db . MustGetObject ( ( * range2 . first ) . second ) ;
// handle regularly-contained elements
2011-06-14 15:58:22 +00:00
if ( const IfcRelContainedInSpatialStructure * const cont = obj - > ToPtr < IfcRelContainedInSpatialStructure > ( ) ) {
2013-03-29 12:33:18 +00:00
if ( cont - > RelatingStructure - > GetID ( ) ! = el . GetID ( ) ) {
continue ;
}
2011-06-14 15:58:22 +00:00
BOOST_FOREACH ( const IfcProduct & pro , cont - > RelatedElements ) {
if ( const IfcOpeningElement * const open = pro . ToPtr < IfcOpeningElement > ( ) ) {
2011-05-13 00:52:50 +00:00
// IfcOpeningElement is handled below. Sadly we can't use it here as is:
2013-03-29 12:33:18 +00:00
// The docs say that opening elements are USUALLY attached to building storey,
// but we want them for the building elements to which they belong.
2011-05-13 00:52:50 +00:00
continue ;
}
2011-06-14 15:58:22 +00:00
aiNode * const ndnew = ProcessSpatialStructure ( nd . get ( ) , pro , conv , NULL ) ;
if ( ndnew ) {
subnodes . push_back ( ndnew ) ;
}
2011-05-13 00:52:50 +00:00
}
}
// handle openings, which we collect in a list rather than adding them to the node graph
2011-06-14 15:58:22 +00:00
else if ( const IfcRelVoidsElement * const fills = obj - > ToPtr < IfcRelVoidsElement > ( ) ) {
2011-05-13 00:52:50 +00:00
if ( fills - > RelatingBuildingElement - > GetID ( ) = = el . GetID ( ) ) {
2011-06-14 15:58:22 +00:00
const IfcFeatureElementSubtraction & open = fills - > RelatedOpeningElement ;
2011-05-13 00:52:50 +00:00
// move opening elements to a separate node since they are semantically different than elements that are just 'contained'
std : : auto_ptr < aiNode > nd_aggr ( new aiNode ( ) ) ;
nd_aggr - > mName . Set ( " $RelVoidsElement " ) ;
nd_aggr - > mParent = nd . get ( ) ;
nd_aggr - > mTransformation = nd - > mTransformation ;
std : : vector < TempOpening > openings_local ;
2011-06-14 15:58:22 +00:00
aiNode * const ndnew = ProcessSpatialStructure ( nd_aggr . get ( ) , open , conv , & openings_local ) ;
if ( ndnew ) {
2011-05-13 00:52:50 +00:00
2011-06-14 15:58:22 +00:00
nd_aggr - > mNumChildren = 1 ;
nd_aggr - > mChildren = new aiNode * [ 1 ] ( ) ;
2011-05-13 00:52:50 +00:00
2011-06-14 15:58:22 +00:00
nd_aggr - > mChildren [ 0 ] = ndnew ;
if ( openings_local . size ( ) ) {
if ( ! didinv ) {
myInv = aiMatrix4x4 ( nd - > mTransformation ) . Inverse ( ) ;
didinv = true ;
}
// we need all openings to be in the local space of *this* node, so transform them
BOOST_FOREACH ( TempOpening & op , openings_local ) {
op . Transform ( myInv * nd_aggr - > mChildren [ 0 ] - > mTransformation ) ;
openings . push_back ( op ) ;
}
2011-05-13 00:52:50 +00:00
}
2011-06-14 15:58:22 +00:00
subnodes . push_back ( nd_aggr . release ( ) ) ;
2011-05-13 00:52:50 +00:00
}
2011-05-06 23:55:38 +00:00
}
}
}
for ( ; range . first ! = range . second ; + + range . first ) {
2013-03-29 12:33:18 +00:00
// see note in loop above
if ( conv . already_processed . find ( ( * range . first ) . second ) ! = conv . already_processed . end ( ) ) {
continue ;
}
2011-06-14 15:58:22 +00:00
if ( const IfcRelAggregates * const aggr = conv . db . GetObject ( ( * range . first ) . second ) - > ToPtr < IfcRelAggregates > ( ) ) {
2013-03-29 12:33:18 +00:00
if ( aggr - > RelatingObject - > GetID ( ) ! = el . GetID ( ) ) {
continue ;
}
2011-05-06 23:55:38 +00:00
2011-05-13 00:52:50 +00:00
// move aggregate elements to a separate node since they are semantically different than elements that are just 'contained'
2011-05-06 23:55:38 +00:00
std : : auto_ptr < aiNode > nd_aggr ( new aiNode ( ) ) ;
2011-05-13 00:52:50 +00:00
nd_aggr - > mName . Set ( " $RelAggregates " ) ;
2011-05-06 23:55:38 +00:00
nd_aggr - > mParent = nd . get ( ) ;
2011-05-13 00:52:50 +00:00
nd_aggr - > mTransformation = nd - > mTransformation ;
2011-05-06 23:55:38 +00:00
nd_aggr - > mChildren = new aiNode * [ aggr - > RelatedObjects . size ( ) ] ( ) ;
2011-06-14 15:58:22 +00:00
BOOST_FOREACH ( const IfcObjectDefinition & def , aggr - > RelatedObjects ) {
if ( const IfcProduct * const prod = def . ToPtr < IfcProduct > ( ) ) {
aiNode * const ndnew = ProcessSpatialStructure ( nd_aggr . get ( ) , * prod , conv , NULL ) ;
if ( ndnew ) {
nd_aggr - > mChildren [ nd_aggr - > mNumChildren + + ] = ndnew ;
}
2011-05-06 23:55:38 +00:00
}
}
subnodes . push_back ( nd_aggr . release ( ) ) ;
}
}
2011-05-13 00:52:50 +00:00
conv . collect_openings = collect_openings ;
if ( ! conv . collect_openings ) {
conv . apply_openings = & openings ;
}
2011-05-21 15:38:39 +00:00
ProcessProductRepresentation ( el , nd . get ( ) , subnodes , conv ) ;
2011-05-13 00:52:50 +00:00
conv . apply_openings = conv . collect_openings = NULL ;
2011-05-06 23:55:38 +00:00
if ( subnodes . size ( ) ) {
nd - > mChildren = new aiNode * [ subnodes . size ( ) ] ( ) ;
BOOST_FOREACH ( aiNode * nd2 , subnodes ) {
nd - > mChildren [ nd - > mNumChildren + + ] = nd2 ;
nd2 - > mParent = nd . get ( ) ;
}
}
}
catch ( . . . ) {
// it hurts, but I don't want to pull boost::ptr_vector into -noboost only for these few spots here
std : : for_each ( subnodes . begin ( ) , subnodes . end ( ) , delete_fun < aiNode > ( ) ) ;
throw ;
}
2013-03-29 12:33:18 +00:00
ai_assert ( conv . already_processed . find ( el . GetID ( ) ) ! = conv . already_processed . end ( ) ) ;
conv . already_processed . erase ( conv . already_processed . find ( el . GetID ( ) ) ) ;
2011-05-06 23:55:38 +00:00
return nd . release ( ) ;
}
// ------------------------------------------------------------------------------------------------
void ProcessSpatialStructures ( ConversionData & conv )
{
2011-05-31 14:23:43 +00:00
// XXX add support for multiple sites (i.e. IfcSpatialStructureElements with composition == COMPLEX)
2011-05-06 23:55:38 +00:00
// process all products in the file. it is reasonable to assume that a
// file that is relevant for us contains at least a site or a building.
const STEP : : DB : : ObjectMapByType & map = conv . db . GetObjectsByType ( ) ;
2011-05-21 15:38:39 +00:00
ai_assert ( map . find ( " ifcsite " ) ! = map . end ( ) ) ;
const STEP : : DB : : ObjectSet * range = & map . find ( " ifcsite " ) - > second ;
if ( range - > empty ( ) ) {
ai_assert ( map . find ( " ifcbuilding " ) ! = map . end ( ) ) ;
range = & map . find ( " ifcbuilding " ) - > second ;
if ( range - > empty ( ) ) {
// no site, no building - fail;
IFCImporter : : ThrowException ( " no root element found (expected IfcBuilding or preferably IfcSite) " ) ;
2011-05-06 23:55:38 +00:00
}
}
2011-05-07 20:53:51 +00:00
2011-05-21 15:38:39 +00:00
BOOST_FOREACH ( const STEP : : LazyObject * lz , * range ) {
2011-06-14 15:58:22 +00:00
const IfcSpatialStructureElement * const prod = lz - > ToPtr < IfcSpatialStructureElement > ( ) ;
2011-05-06 23:55:38 +00:00
if ( ! prod ) {
continue ;
}
IFCImporter : : LogDebug ( " looking at spatial structure ` " + ( prod - > Name ? prod - > Name . Get ( ) : " unnamed " ) + " ` " + ( prod - > ObjectType ? " which is of type " + prod - > ObjectType . Get ( ) : " " ) ) ;
// the primary site is referenced by an IFCRELAGGREGATES element which assigns it to the IFCPRODUCT
const STEP : : DB : : RefMap & refs = conv . db . GetRefs ( ) ;
STEP : : DB : : RefMapRange range = refs . equal_range ( conv . proj . GetID ( ) ) ;
for ( ; range . first ! = range . second ; + + range . first ) {
2011-06-14 15:58:22 +00:00
if ( const IfcRelAggregates * const aggr = conv . db . GetObject ( ( * range . first ) . second ) - > ToPtr < IfcRelAggregates > ( ) ) {
2011-05-06 23:55:38 +00:00
2011-06-14 15:58:22 +00:00
BOOST_FOREACH ( const IfcObjectDefinition & def , aggr - > RelatedObjects ) {
2011-05-06 23:55:38 +00:00
// comparing pointer values is not sufficient, we would need to cast them to the same type first
// as there is multiple inheritance in the game.
2011-05-13 00:52:50 +00:00
if ( def . GetID ( ) = = prod - > GetID ( ) ) {
2011-05-06 23:55:38 +00:00
IFCImporter : : LogDebug ( " selecting this spatial structure as root structure " ) ;
// got it, this is the primary site.
2011-05-13 00:52:50 +00:00
conv . out - > mRootNode = ProcessSpatialStructure ( NULL , * prod , conv , NULL ) ;
2011-05-06 23:55:38 +00:00
return ;
}
}
}
}
}
2011-05-07 20:53:51 +00:00
2011-05-31 14:23:43 +00:00
IFCImporter : : LogWarn ( " failed to determine primary site element, taking the first IfcSite " ) ;
BOOST_FOREACH ( const STEP : : LazyObject * lz , * range ) {
2011-06-14 15:58:22 +00:00
const IfcSpatialStructureElement * const prod = lz - > ToPtr < IfcSpatialStructureElement > ( ) ;
2011-05-31 14:23:43 +00:00
if ( ! prod ) {
continue ;
}
conv . out - > mRootNode = ProcessSpatialStructure ( NULL , * prod , conv , NULL ) ;
return ;
}
2011-05-07 20:53:51 +00:00
2011-05-21 15:38:39 +00:00
IFCImporter : : ThrowException ( " failed to determine primary site element " ) ;
2011-05-06 23:55:38 +00:00
}
// ------------------------------------------------------------------------------------------------
void MakeTreeRelative ( aiNode * start , const aiMatrix4x4 & combined )
{
// combined is the parent's absolute transformation matrix
2012-02-02 02:09:49 +00:00
const aiMatrix4x4 old = start - > mTransformation ;
2011-05-06 23:55:38 +00:00
if ( ! combined . IsIdentity ( ) ) {
start - > mTransformation = aiMatrix4x4 ( combined ) . Inverse ( ) * start - > mTransformation ;
}
// All nodes store absolute transformations right now, so we need to make them relative
for ( unsigned int i = 0 ; i < start - > mNumChildren ; + + i ) {
MakeTreeRelative ( start - > mChildren [ i ] , old ) ;
}
}
// ------------------------------------------------------------------------------------------------
void MakeTreeRelative ( ConversionData & conv )
{
2012-02-02 02:09:49 +00:00
MakeTreeRelative ( conv . out - > mRootNode , IfcMatrix4 ( ) ) ;
2011-05-06 23:55:38 +00:00
}
} // !anon
# endif