LIB3MFMETHODIMP CCOMModelTexture2D::WriteToFile(_In_z_ LPCWSTR pwszFilename)
{
try {
if (pwszFilename == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPOINTER);
CModelTexture2DResource * pTextureResource = getTexture2D();
__NMRASSERT(pTextureResource);
PImportStream pTextureStream = pTextureResource->getTextureStream();
if (pTextureStream.get() != nullptr) {
pTextureStream->writeToFile(pwszFilename);
}
else {
throw CNMRException(NMR_ERROR_NOTEXTURESTREAM);
}
return handleSuccess();
}
catch (CNMRException & Exception) {
return handleNMRException(&Exception);
}
catch (...) {
return handleGenericException();
}
}
LIB3MFMETHODIMP CCOMModelTexture2D::WriteToBuffer(_Out_ BYTE * pBuffer, _In_ ULONG64 cbBufferSize)
{
try {
if (pBuffer == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPOINTER);
CModelTexture2DResource * pTextureResource = getTexture2D();
__NMRASSERT(pTextureResource);
PImportStream pTextureStream = pTextureResource->getTextureStream();
if (pTextureStream.get() != nullptr) {
nfUint64 cbStreamSize = pTextureStream->retrieveSize();
if (cbStreamSize > cbBufferSize)
throw CNMRException(NMR_ERROR_INSUFFICIENTBUFFERSIZE);
pTextureStream->seekPosition(0, true);
pTextureStream->readBuffer(pBuffer, cbStreamSize, true);
}
else {
throw CNMRException(NMR_ERROR_NOTEXTURESTREAM);
}
return handleSuccess();
}
catch (CNMRException & Exception) {
return handleNMRException(&Exception);
}
catch (...) {
return handleGenericException();
}
}
LIB3MFMETHODIMP CCOMModelTexture2D::WriteToCallback(_In_ void * pWriteCallback, _In_opt_ void * pUserData)
{
try {
if (pWriteCallback == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPOINTER);
CModelTexture2DResource * pTextureResource = getTexture2D();
__NMRASSERT(pTextureResource);
PImportStream pTextureAttachment = pTextureResource->getTextureStream();
ExportStream_WriteCallbackType pTypedWriteCallback = (ExportStream_WriteCallbackType)pWriteCallback;
if (pTextureAttachment.get() != nullptr) {
PExportStream pExportStream = std::make_shared<CExportStream_Callback>(pTypedWriteCallback, nullptr, pUserData);
pExportStream->copyFrom(pTextureAttachment.get(), pTextureAttachment->retrieveSize(), MODELTEXTURE2D_BUFFERSIZE);
}
else {
throw CNMRException(NMR_ERROR_NOTEXTURESTREAM);
}
return handleSuccess();
}
catch (CNMRException & Exception) {
return handleNMRException(&Exception);
}
catch (...) {
return handleGenericException();
}
}
LIB3MFMETHODIMP CCOMModelTexture2D::GetStreamSize(_Out_ ULONG64 * pcbStreamSize)
{
try {
CModelTexture2DResource * pTextureResource = getTexture2D();
__NMRASSERT(pTextureResource);
PImportStream pTextureStream = pTextureResource->getTextureStream();
if (pTextureStream.get() != nullptr) {
*pcbStreamSize = pTextureStream->retrieveSize();
}
else {
*pcbStreamSize = 0;
}
return handleSuccess();
}
catch (CNMRException & Exception) {
return handleNMRException(&Exception);
}
catch (...) {
return handleGenericException();
}
}
void CModelWriter_3MF_Native::addAttachments(_In_ CModel * pModel, _In_ POpcPackageWriter pPackageWriter, _In_ POpcPackagePart pModelPart)
{
__NMRASSERT(pModel != nullptr);
__NMRASSERT(pModelPart.get() != nullptr);
__NMRASSERT(pPackageWriter.get() != nullptr);
nfUint32 nCount = pModel->getAttachmentCount();
nfUint32 nIndex;
if (nCount > 0) {
for (nIndex = 0; nIndex < nCount; nIndex++) {
PModelAttachment pAttachment = pModel->getModelAttachment(nIndex);
PImportStream pStream = pAttachment->getStream();
std::wstring sPath = fnIncludeLeadingPathDelimiter(pAttachment->getPathURI());
std::wstring sRelationShipType = pAttachment->getRelationShipType();
if (pStream.get() == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPARAM);
// create Texture Part
POpcPackagePart pAttachmentPart = pPackageWriter->addPart(sPath);
PExportStream pExportStream = pAttachmentPart->getExportStream();
// Copy data
pStream->seekPosition(0, true);
pExportStream->copyFrom(pStream.get(), pStream->retrieveSize(), MODELWRITER_NATIVE_BUFFERSIZE);
// add relationships
pModelPart->addRelationship(generateRelationShipID(), sRelationShipType.c_str(), pAttachmentPart->getURI());
}
}
}
LIB3MFMETHODIMP CCOMModelTexture2D::WriteToFileUTF8(_In_z_ LPCSTR pszFilename)
{
try {
if (pszFilename == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPOINTER);
CModelTexture2DResource * pTextureResource = getTexture2D();
__NMRASSERT(pTextureResource);
PImportStream pTextureAttachment = pTextureResource->getTextureStream();
if (pTextureAttachment.get() != nullptr) {
std::string sUTF8FileName(pszFilename);
std::wstring sUTF16FileName = fnUTF8toUTF16(sUTF8FileName);
pTextureAttachment->writeToFile(sUTF16FileName.c_str());
}
else {
throw CNMRException(NMR_ERROR_NOTEXTURESTREAM);
}
return handleSuccess();
}
catch (CNMRException & Exception) {
return handleNMRException(&Exception);
}
catch (...) {
return handleGenericException();
}
}
void CModelWriter_3MF_Native::addTextureParts(_In_ CModel * pModel, _In_ POpcPackageWriter pPackageWriter, _In_ POpcPackagePart pModelPart)
{
__NMRASSERT(pModel != nullptr);
__NMRASSERT(pModelPart.get() != nullptr);
__NMRASSERT(pPackageWriter.get() != nullptr);
nfUint32 nCount = pModel->getTextureStreamCount();
nfUint32 nIndex;
if (nCount > 0) {
for (nIndex = 0; nIndex < nCount; nIndex++) {
PImportStream pStream = pModel->getTextureStream(nIndex);
std::wstring sPath = fnIncludeLeadingPathDelimiter (pModel->getTextureStreamPath(nIndex));
if (pStream.get() == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPARAM);
// create Texture Part
POpcPackagePart pTexturePart = pPackageWriter->addPart(sPath);
PExportStream pExportStream = pTexturePart->getExportStream();
// Copy data
pStream->seekPosition(0, true);
pExportStream->copyFrom(pStream.get(), pStream->retrieveSize(), MODELWRITER_NATIVE_BUFFERSIZE);
// add relationships
pModelPart->addRelationship(generateRelationShipID(), PACKAGE_TEXTURE_RELATIONSHIP_TYPE, pTexturePart.get());
}
}
}
void CModelWriter_3MF_Native::writePackageToStream(_In_ PExportStream pStream)
{
if (pStream.get() == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPARAM);
if (m_pModel == nullptr)
throw CNMRException(NMR_ERROR_NOMODELTOWRITE);
// Write Model Stream
POpcPackageWriter pPackageWriter = std::make_shared<COpcPackageWriter>(pStream);
POpcPackagePart pModelPart = pPackageWriter->addPart(PACKAGE_3D_MODEL_URI);
PXmlWriter_Native pXMLWriter = std::make_shared<CXmlWriter_Native>(pModelPart->getExportStream());
writeModelStream(pXMLWriter.get(), m_pModel);
// add Root relationships
pPackageWriter->addRootRelationship(generateRelationShipID(), PACKAGE_START_PART_RELATIONSHIP_TYPE, pModelPart.get());
PModelAttachment pPackageThumbnail = m_pModel->getPackageThumbnail();
if (pPackageThumbnail.get() != nullptr)
{
// create Package Thumbnail Part
POpcPackagePart pThumbnailPart = pPackageWriter->addPart(pPackageThumbnail->getPathURI());
PExportStream pExportStream = pThumbnailPart->getExportStream();
// Copy data
PImportStream pPackageThumbnailStream = pPackageThumbnail->getStream();
pPackageThumbnailStream->seekPosition(0, true);
pExportStream->copyFrom(pPackageThumbnailStream.get(), pPackageThumbnailStream->retrieveSize(), MODELWRITER_NATIVE_BUFFERSIZE);
// add root relationship
pPackageWriter->addRootRelationship(generateRelationShipID(), pPackageThumbnail->getRelationShipType(), pThumbnailPart.get());
}
// add slicestacks that reference other files
addSlicerefAttachments(m_pModel);
// add Attachments
addAttachments(m_pModel, pPackageWriter, pModelPart);
// add Content Types
pPackageWriter->addContentType(PACKAGE_3D_RELS_EXTENSION, PACKAGE_3D_RELS_CONTENT_TYPE);
pPackageWriter->addContentType(PACKAGE_3D_MODEL_EXTENSION, PACKAGE_3D_MODEL_CONTENT_TYPE);
pPackageWriter->addContentType(PACKAGE_3D_TEXTURE_EXTENSION, PACKAGE_TEXTURE_CONTENT_TYPE);
pPackageWriter->addContentType(PACKAGE_3D_PNG_EXTENSION, PACKAGE_PNG_CONTENT_TYPE);
pPackageWriter->addContentType(PACKAGE_3D_JPEG_EXTENSION, PACKAGE_JPG_CONTENT_TYPE);
pPackageWriter->addContentType(PACKAGE_3D_JPG_EXTENSION, PACKAGE_JPG_CONTENT_TYPE);
std::map<std::wstring, std::wstring> CustomContentTypes = m_pModel->getCustomContentTypes();
std::map<std::wstring, std::wstring>::iterator iContentTypeIterator;
for (iContentTypeIterator = CustomContentTypes.begin(); iContentTypeIterator != CustomContentTypes.end(); iContentTypeIterator++) {
if (!m_pModel->contentTypeIsDefault(iContentTypeIterator->first)) {
pPackageWriter->addContentType(iContentTypeIterator->first, iContentTypeIterator->second);
}
}
}
LIB3MFMETHODIMP CCOMModelTexture2D::WriteToStream(_In_ IStream * pStream)
{
try {
if (pStream == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPOINTER);
CModelTexture2DResource * pTextureResource = getTexture2D();
__NMRASSERT(pTextureResource);
PImportStream pTextureStream = pTextureResource->getTextureStream();
if (pTextureStream.get() != nullptr) {
nfUint64 cbStreamSize = pTextureStream->retrieveSize();
pTextureStream->seekPosition(0, true);
std::array<nfByte, NMR_IMPORTSTREAM_COPYBUFFERSIZE> pBuffer;
nfUint64 cbBytesLeft = cbStreamSize;
while (cbBytesLeft > 0) {
nfUint64 cbLength = cbBytesLeft;
if (cbLength > NMR_IMPORTSTREAM_COPYBUFFERSIZE)
cbLength = NMR_IMPORTSTREAM_COPYBUFFERSIZE;
ULONG cbWrittenBytes = 0;
pTextureStream->readBuffer(&pBuffer[0], cbLength, true);
HRESULT hResult = pStream->Write(&pBuffer[0], (nfUint32) cbLength, &cbWrittenBytes);
if (hResult != S_OK)
throw CNMRException_Windows(NMR_ERROR_COULDNOTWRITESTREAM, hResult);
if (cbWrittenBytes != cbLength)
throw CNMRException(NMR_ERROR_COULDNOTWRITEFULLDATA);
cbBytesLeft -= cbLength;
}
}
else {
throw CNMRException(NMR_ERROR_NOTEXTURESTREAM);
}
return handleSuccess();
}
catch (CNMRException & Exception) {
return handleNMRException(&Exception);
}
catch (...) {
return handleGenericException();
}
}
void CModel::mergeTextureStreams(_In_ CModel * pSourceModel)
{
if (pSourceModel == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPARAM);
nfUint32 nCount = pSourceModel->getTextureStreamCount();
nfUint32 nIndex;
for (nIndex = 0; nIndex < nCount; nIndex++) {
std::wstring sPath = pSourceModel->getTextureStreamPath(nIndex);
PImportStream pTextureStream = pSourceModel->getTextureStream(nIndex);
PImportStream pCopiedStream = pTextureStream->copyToMemory();
addTextureStream(sPath, pCopiedStream);
}
}
void CModelReader_3MF_Native::extractTexturesFromRelationships(_In_ COpcPackagePart * pModelPart)
{
if (pModelPart == nullptr)
throw CNMRException(NMR_ERROR_INVALIDPARAM);
std::list<POpcPackageRelationship> RelationShips = pModelPart->getRelationShips();
std::list<POpcPackageRelationship>::iterator iIterator;
for (iIterator = RelationShips.begin(); iIterator != RelationShips.end(); iIterator++) {
std::wstring sType = (*iIterator)->getType();
if (wcscmp(sType.c_str(), PACKAGE_TEXTURE_RELATIONSHIP_TYPE) == 0) {
std::wstring sURI = (*iIterator)->getTargetPartURI();
POpcPackagePart pTexturePart = m_pPackageReader->createPart(sURI);
PImportStream pTextureStream = pTexturePart->getImportStream();
PImportStream pMemoryStream = pTextureStream->copyToMemory();
addTextureStream(sURI, pMemoryStream);
}
}
}
POpcPackagePart COpcPackageReader::createPart(_In_ std::wstring sPath)
{
std::wstring sRealPath = fnRemoveLeadingPathDelimiter (sPath);
auto iPartIterator = m_Parts.find(sRealPath);
if (iPartIterator != m_Parts.end()) {
return iPartIterator->second;
}
PImportStream pStream = openZIPEntry(sRealPath);
if (pStream.get() == nullptr)
throw CNMRException(NMR_ERROR_COULDNOTCREATEOPCPART);
POpcPackagePart pPart = std::make_shared<COpcPackagePart>(sRealPath, pStream);
m_Parts.insert(std::make_pair(sRealPath, pPart));
std::wstring sRelationShipName = fnExtractFileName(sRealPath);
std::wstring sRelationShipPath = sRealPath.substr(0, sRealPath.length() - sRelationShipName.length());
sRelationShipPath += L"_rels/";
sRelationShipPath += sRelationShipName;
sRelationShipPath += L".rels";
PImportStream pRelStream = openZIPEntry(sRelationShipPath);
if (pRelStream.get() != nullptr) {
POpcPackageRelationshipReader pReader = std::make_shared<COpcPackageRelationshipReader>(pRelStream);
nfUint32 nCount = pReader->getCount();
nfUint32 nIndex;
for (nIndex = 0; nIndex < nCount; nIndex++) {
POpcPackageRelationship pRelationShip = pReader->getRelationShip(nIndex);
pPart->addRelationship(pRelationShip->getID(), pRelationShip->getType(), pRelationShip->getTargetPartURI());
}
}
return pPart;
}
