164 lines
6.1 KiB
C++
164 lines
6.1 KiB
C++
//-*****************************************************************************
|
|
//
|
|
// Copyright (c) 2009-2014,
|
|
// Sony Pictures Imageworks Inc. and
|
|
// Industrial Light & Magic, a division of Lucasfilm Entertainment Company Ltd.
|
|
//
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use 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.
|
|
// * Neither the name of Sony Pictures Imageworks, nor
|
|
// Industrial Light & Magic, nor the names of their contributors may be used
|
|
// to endorse or promote products derived from this software without specific
|
|
// prior written permission.
|
|
//
|
|
// 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.
|
|
//
|
|
//-*****************************************************************************
|
|
|
|
#include <Foundation.h>
|
|
|
|
using namespace boost::python;
|
|
|
|
const std::string kHDF5 = "HDF5";
|
|
const std::string kOgawa = "Ogawa";
|
|
const std::string kLayer = "Layer";
|
|
const std::string kUnknown = "Unknown";
|
|
|
|
//-*****************************************************************************
|
|
static Abc::IArchive* mkIArchive( const std::string &iName )
|
|
{
|
|
Abc::IArchive archive;
|
|
AbcF::IFactory factory;
|
|
factory.setPolicy(Abc::ErrorHandler::kQuietNoopPolicy);
|
|
AbcF::IFactory::CoreType coreType;
|
|
archive = factory.getArchive(iName, coreType);
|
|
|
|
if ( coreType == AbcF::IFactory::kUnknown ) {
|
|
throwPythonException( "Unknown core type" );
|
|
}
|
|
#ifndef ALEMBIC_WITH_HDF5
|
|
else if ( coreType == AbcF::IFactory::kHDF5 ) {
|
|
throwPythonException( "Unsupported core type: HDF5" );
|
|
}
|
|
#endif
|
|
return new Abc::IArchive( archive );
|
|
}
|
|
|
|
//-*****************************************************************************
|
|
static Abc::IArchive* mkLayeredIArchive( boost::python::list &iNames )
|
|
{
|
|
std::vector< std::string > files;
|
|
for ( int i = 0; i < len(iNames); ++i )
|
|
{
|
|
files.push_back( boost::python::extract<std::string>( iNames[i] ) );
|
|
}
|
|
|
|
Abc::IArchive archive;
|
|
AbcF::IFactory factory;
|
|
factory.setPolicy(Abc::ErrorHandler::kQuietNoopPolicy);
|
|
AbcF::IFactory::CoreType coreType;
|
|
archive = factory.getArchive( files, coreType );
|
|
|
|
if ( coreType == AbcF::IFactory::kUnknown ) {
|
|
throwPythonException( "Unknown core type" );
|
|
}
|
|
#ifndef ALEMBIC_WITH_HDF5
|
|
else if ( coreType == AbcF::IFactory::kHDF5 ) {
|
|
throwPythonException( "Unsupported core type: HDF5" );
|
|
}
|
|
#endif
|
|
return new Abc::IArchive( archive );
|
|
}
|
|
|
|
//-*****************************************************************************
|
|
static std::string getCoreType( Abc::IArchive& archive )
|
|
{
|
|
AbcF::IFactory factory;
|
|
factory.setPolicy(Abc::ErrorHandler::kQuietNoopPolicy);
|
|
AbcF::IFactory::CoreType coreType;
|
|
archive = factory.getArchive(archive.getName(), coreType);
|
|
|
|
if ( coreType == AbcF::IFactory::kOgawa ) {
|
|
return kOgawa;
|
|
} else if ( coreType == AbcF::IFactory::kHDF5 ) {
|
|
return kHDF5;
|
|
} else {
|
|
return kUnknown;
|
|
};
|
|
}
|
|
|
|
//-*****************************************************************************
|
|
void register_iarchive()
|
|
{
|
|
|
|
// export the CoreType constants
|
|
//
|
|
scope().attr("kHDF5") = kHDF5;
|
|
scope().attr("kOgawa") = kOgawa;
|
|
scope().attr("kLayer") = kLayer;
|
|
scope().attr("kUnknown") = kUnknown;
|
|
|
|
// IArchive
|
|
//
|
|
class_< Abc::IArchive >(
|
|
"IArchive",
|
|
"The IArchive class opens an existing Alembic archive for read access",
|
|
no_init )
|
|
.def( "__init__",
|
|
make_constructor(
|
|
mkIArchive,
|
|
default_call_policies(),
|
|
( arg( "fileName" ) ) ),
|
|
"Create an IArchive with the given file name" )
|
|
.def( "__init__",
|
|
make_constructor(
|
|
mkLayeredIArchive,
|
|
default_call_policies(),
|
|
( arg( "fileList" ) ) ),
|
|
"Create a layered IArchive with the given file name list" )
|
|
.def( "getName",
|
|
&Abc::IArchive::getName,
|
|
"Return the file name" )
|
|
.def( "getTop",
|
|
&Abc::IArchive::getTop,
|
|
"Return the single top-level IObject",
|
|
with_custodian_and_ward_postcall<0,1>() )
|
|
.def( "getCoreType",
|
|
&getCoreType,
|
|
"Return the archive CoreType" )
|
|
.def( "getTimeSampling",
|
|
&Abc::IArchive::getTimeSampling,
|
|
( arg( "index" ) ),
|
|
"Return the TimeSampling with the given index" )
|
|
.def( "getNumTimeSamplings",
|
|
&Abc::IArchive::getNumTimeSamplings,
|
|
"Return the total number of TimeSampling in the Archive" )
|
|
.def( "getMaxNumSamplesForTimeSamplingIndex",
|
|
&Abc::IArchive::getMaxNumSamplesForTimeSamplingIndex,
|
|
( arg("index" ) ),
|
|
"Returns the max number of samples set for the TimeSampling at the given index." )
|
|
.def( "valid", &Abc::IArchive::valid )
|
|
.def( ALEMBIC_PYTHON_BOOL_NAME, &Abc::IArchive::valid )
|
|
.def( "__str__", &Abc::IArchive::getName )
|
|
;
|
|
}
|