summaryrefslogtreecommitdiff
path: root/src/mesh/assimp-master/code/AssetLib/3MF/D3MFOpcPackage.cpp
diff options
context:
space:
mode:
authorsanine <sanine.not@pm.me>2022-03-04 10:47:15 -0600
committersanine <sanine.not@pm.me>2022-03-04 10:47:15 -0600
commit058f98a63658dc1a2579826ba167fd61bed1e21f (patch)
treebcba07a1615a14d943f3af3f815a42f3be86b2f3 /src/mesh/assimp-master/code/AssetLib/3MF/D3MFOpcPackage.cpp
parent2f8028ac9e0812cb6f3cbb08f0f419e4e717bd22 (diff)
add assimp submodule
Diffstat (limited to 'src/mesh/assimp-master/code/AssetLib/3MF/D3MFOpcPackage.cpp')
-rw-r--r--src/mesh/assimp-master/code/AssetLib/3MF/D3MFOpcPackage.cpp256
1 files changed, 256 insertions, 0 deletions
diff --git a/src/mesh/assimp-master/code/AssetLib/3MF/D3MFOpcPackage.cpp b/src/mesh/assimp-master/code/AssetLib/3MF/D3MFOpcPackage.cpp
new file mode 100644
index 0000000..f88039a
--- /dev/null
+++ b/src/mesh/assimp-master/code/AssetLib/3MF/D3MFOpcPackage.cpp
@@ -0,0 +1,256 @@
+/*
+Open Asset Import Library (assimp)
+----------------------------------------------------------------------
+
+Copyright (c) 2006-2022, assimp team
+
+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.
+
+* Neither the name of the assimp team, nor the names of its
+ contributors may be used to endorse or promote products
+ derived from this software without specific prior
+ written permission of the assimp team.
+
+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.
+
+----------------------------------------------------------------------
+*/
+
+#ifndef ASSIMP_BUILD_NO_3MF_IMPORTER
+
+#include "D3MFOpcPackage.h"
+#include <assimp/Exceptional.h>
+#include <assimp/XmlParser.h>
+#include <assimp/ZipArchiveIOSystem.h>
+#include <assimp/ai_assert.h>
+#include <assimp/DefaultLogger.hpp>
+#include <assimp/IOStream.hpp>
+#include <assimp/IOSystem.hpp>
+#include <assimp/texture.h>
+#include "3MFXmlTags.h"
+
+#include <algorithm>
+#include <cassert>
+#include <cstdlib>
+#include <map>
+#include <vector>
+
+namespace Assimp {
+
+namespace D3MF {
+// ------------------------------------------------------------------------------------------------
+
+using OpcPackageRelationshipPtr = std::shared_ptr<OpcPackageRelationship>;
+
+class OpcPackageRelationshipReader {
+public:
+ OpcPackageRelationshipReader(XmlParser &parser) :
+ m_relationShips() {
+ XmlNode root = parser.getRootNode();
+ ParseRootNode(root);
+ }
+
+ void ParseRootNode(XmlNode &node) {
+ ParseAttributes(node);
+ for (XmlNode currentNode = node.first_child(); currentNode; currentNode = currentNode.next_sibling()) {
+ std::string name = currentNode.name();
+ if (name == "Relationships") {
+ ParseRelationsNode(currentNode);
+ }
+ }
+ }
+
+ void ParseAttributes(XmlNode & /*node*/) {
+ // empty
+ }
+
+ bool validateRels(OpcPackageRelationshipPtr &relPtr) {
+ if (relPtr->id.empty() || relPtr->type.empty() || relPtr->target.empty()) {
+ return false;
+ }
+
+ return true;
+ }
+
+ void ParseRelationsNode(XmlNode &node) {
+ if (node.empty()) {
+ return;
+ }
+
+ for (XmlNode currentNode = node.first_child(); currentNode; currentNode = currentNode.next_sibling()) {
+ const std::string name = currentNode.name();
+ if (name == "Relationship") {
+ OpcPackageRelationshipPtr relPtr(new OpcPackageRelationship());
+ relPtr->id = currentNode.attribute(XmlTag::RELS_ATTRIB_ID).as_string();
+ relPtr->type = currentNode.attribute(XmlTag::RELS_ATTRIB_TYPE).as_string();
+ relPtr->target = currentNode.attribute(XmlTag::RELS_ATTRIB_TARGET).as_string();
+ if (validateRels(relPtr)) {
+ m_relationShips.push_back(relPtr);
+ }
+ }
+ }
+ }
+
+ std::vector<OpcPackageRelationshipPtr> m_relationShips;
+};
+
+static bool IsEmbeddedTexture( const std::string &filename ) {
+ const std::string extension = BaseImporter::GetExtension(filename);
+ if (extension == "jpg" || extension == "png") {
+ std::string::size_type pos = filename.find("thumbnail");
+ if (pos == std::string::npos) {
+ return false;
+ }
+ return true;
+ }
+
+ return false;
+}
+// ------------------------------------------------------------------------------------------------
+D3MFOpcPackage::D3MFOpcPackage(IOSystem *pIOHandler, const std::string &rFile) :
+ mRootStream(nullptr),
+ mZipArchive() {
+ mZipArchive = new ZipArchiveIOSystem(pIOHandler, rFile);
+ if (!mZipArchive->isOpen()) {
+ throw DeadlyImportError("Failed to open file ", rFile, ".");
+ }
+
+ std::vector<std::string> fileList;
+ mZipArchive->getFileList(fileList);
+
+ for (auto &file : fileList) {
+ if (file == D3MF::XmlTag::ROOT_RELATIONSHIPS_ARCHIVE) {
+ if (!mZipArchive->Exists(file.c_str())) {
+ continue;
+ }
+
+ IOStream *fileStream = mZipArchive->Open(file.c_str());
+ if (nullptr == fileStream) {
+ ASSIMP_LOG_ERROR("Filestream is nullptr.");
+ continue;
+ }
+
+ std::string rootFile = ReadPackageRootRelationship(fileStream);
+ if (!rootFile.empty() && rootFile[0] == '/') {
+ rootFile = rootFile.substr(1);
+ if (rootFile[0] == '/') {
+ // deal with zip-bug
+ rootFile = rootFile.substr(1);
+ }
+ }
+
+ ASSIMP_LOG_VERBOSE_DEBUG(rootFile);
+
+ mZipArchive->Close(fileStream);
+
+ mRootStream = mZipArchive->Open(rootFile.c_str());
+ ai_assert(mRootStream != nullptr);
+ if (nullptr == mRootStream) {
+ throw DeadlyImportError("Cannot open root-file in archive : " + rootFile);
+ }
+ } else if (file == D3MF::XmlTag::CONTENT_TYPES_ARCHIVE) {
+ ASSIMP_LOG_WARN("Ignored file of unsupported type CONTENT_TYPES_ARCHIVES", file);
+ } else if (IsEmbeddedTexture(file)) {
+ IOStream *fileStream = mZipArchive->Open(file.c_str());
+ LoadEmbeddedTextures(fileStream, file);
+ mZipArchive->Close(fileStream);
+ } else {
+ ASSIMP_LOG_WARN("Ignored file of unknown type: ", file);
+ }
+ }
+}
+
+D3MFOpcPackage::~D3MFOpcPackage() {
+ mZipArchive->Close(mRootStream);
+ delete mZipArchive;
+}
+
+IOStream *D3MFOpcPackage::RootStream() const {
+ return mRootStream;
+}
+
+const std::vector<aiTexture *> &D3MFOpcPackage::GetEmbeddedTextures() const {
+ return mEmbeddedTextures;
+}
+
+static const char *const ModelRef = "3D/3dmodel.model";
+
+bool D3MFOpcPackage::validate() {
+ if (nullptr == mRootStream || nullptr == mZipArchive) {
+ return false;
+ }
+
+ return mZipArchive->Exists(ModelRef);
+}
+
+std::string D3MFOpcPackage::ReadPackageRootRelationship(IOStream *stream) {
+ XmlParser xmlParser;
+ if (!xmlParser.parse(stream)) {
+ return std::string();
+ }
+
+ OpcPackageRelationshipReader reader(xmlParser);
+
+ auto itr = std::find_if(reader.m_relationShips.begin(), reader.m_relationShips.end(), [](const OpcPackageRelationshipPtr &rel) {
+ return rel->type == XmlTag::PACKAGE_START_PART_RELATIONSHIP_TYPE;
+ });
+
+ if (itr == reader.m_relationShips.end()) {
+ throw DeadlyImportError("Cannot find ", XmlTag::PACKAGE_START_PART_RELATIONSHIP_TYPE);
+ }
+
+ return (*itr)->target;
+}
+
+void D3MFOpcPackage::LoadEmbeddedTextures(IOStream *fileStream, const std::string &filename) {
+ if (nullptr == fileStream) {
+ return;
+ }
+
+ const size_t size = fileStream->FileSize();
+ if (0 == size) {
+ return;
+ }
+
+ unsigned char *data = new unsigned char[size];
+ fileStream->Read(data, 1, size);
+ aiTexture *texture = new aiTexture;
+ std::string embName = "*" + filename;
+ texture->mFilename.Set(embName.c_str());
+ texture->mWidth = static_cast<unsigned int>(size);
+ texture->mHeight = 0;
+ texture->achFormatHint[0] = 'p';
+ texture->achFormatHint[1] = 'n';
+ texture->achFormatHint[2] = 'g';
+ texture->achFormatHint[3] = '\0';
+ texture->pcData = (aiTexel*) data;
+ mEmbeddedTextures.emplace_back(texture);
+}
+
+} // Namespace D3MF
+} // Namespace Assimp
+
+#endif //ASSIMP_BUILD_NO_3MF_IMPORTER