diff options
Diffstat (limited to 'NvBlast/sdk/extensions/physx/source/sync/NvBlastExtSync.cpp')
| -rw-r--r-- | NvBlast/sdk/extensions/physx/source/sync/NvBlastExtSync.cpp | 235 |
1 files changed, 235 insertions, 0 deletions
diff --git a/NvBlast/sdk/extensions/physx/source/sync/NvBlastExtSync.cpp b/NvBlast/sdk/extensions/physx/source/sync/NvBlastExtSync.cpp new file mode 100644 index 0000000..5f018d9 --- /dev/null +++ b/NvBlast/sdk/extensions/physx/source/sync/NvBlastExtSync.cpp @@ -0,0 +1,235 @@ +/* +* Copyright (c) 2016-2017, NVIDIA CORPORATION. All rights reserved. +* +* NVIDIA CORPORATION and its licensors retain all intellectual property +* and proprietary rights in and to this software, related documentation +* and any modifications thereto. Any use, reproduction, disclosure or +* distribution of this software and related documentation without an express +* license agreement from NVIDIA CORPORATION is strictly prohibited. +*/ + + +#include "NvBlastExtSync.h" +#include "NvBlastAssert.h" +#include "NvBlast.h" +#include "NvBlastExtDefs.h" +#include "NvBlastExtPxManager.h" +#include "NvBlastExtPxFamily.h" +#include "NvBlastExtPxActor.h" +#include "PxRigidDynamic.h" + +#include <chrono> +using namespace std::chrono; + +namespace Nv +{ +namespace Blast +{ + +/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// +// ExtSyncImpl Definition +/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// + +class ExtSyncImpl : public ExtSync +{ + NV_NOCOPY(ExtSyncImpl) + +public: + //////// ctor //////// + + ExtSyncImpl(); + + ~ExtSyncImpl(); + + + //////// TkEventListener interface //////// + + virtual void receive(const TkEvent* events, uint32_t eventCount) override; + + + //////// ExtSync interface //////// + + virtual void release() override; + + virtual void syncFamily(const TkFamily& family) override; + virtual void syncFamily(const ExtPxFamily& family) override; + + virtual uint32_t getSyncBufferSize() const override; + virtual void acquireSyncBuffer(const ExtSyncEvent*const*& buffer, uint32_t& size) const override; + virtual void releaseSyncBuffer() override; + + virtual void applySyncBuffer(TkFramework& framework, const ExtSyncEvent** buffer, uint32_t size, TkGroup* groupForNewActors, ExtPxManager* manager) override; + + +private: + //////// data //////// + + std::vector<ExtSyncEvent*> m_syncEvents; +}; + + +/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// +// ExtSyncEvent Implementation +/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// + +void ExtSyncEvent::release() +{ + NVBLASTEXT_DELETE(this, ExtSyncEvent); +} + + +/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// +// ExtSyncImpl Implementation +/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// + +ExtSync* ExtSync::create() +{ + return NVBLASTEXT_NEW(ExtSyncImpl) (); +} + +void ExtSyncImpl::release() +{ + NVBLASTEXT_DELETE(this, ExtSyncImpl); +} + +ExtSyncImpl::ExtSyncImpl() +{ +} + +ExtSyncImpl::~ExtSyncImpl() +{ + releaseSyncBuffer(); +} + +void ExtSyncImpl::receive(const TkEvent* events, uint32_t eventCount) +{ + for (uint32_t i = 0; i < eventCount; ++i) + { + const TkEvent& tkEvent = events[i]; + if (tkEvent.type == TkEvent::FractureCommand) + { + const TkFractureCommands* fracEvent = tkEvent.getPayload<TkFractureCommands>(); + ExtSyncEventFracture* e = NVBLASTEXT_NEW(ExtSyncEventFracture) (); + e->timestamp = duration_cast<milliseconds>(steady_clock::now().time_since_epoch()).count(); + e->familyID = fracEvent->tkActorData.family->getID(); + e->bondFractures.resize(fracEvent->buffers.bondFractureCount); + e->chunkFractures.resize(fracEvent->buffers.chunkFractureCount); + memcpy(e->bondFractures.data(), fracEvent->buffers.bondFractures, e->bondFractures.size() * sizeof(NvBlastBondFractureData)); + memcpy(e->chunkFractures.data(), fracEvent->buffers.chunkFractures, e->chunkFractures.size() * sizeof(NvBlastChunkFractureData)); + m_syncEvents.push_back(e); + } + } +} + +void ExtSyncImpl::syncFamily(const TkFamily& family) +{ + ExtSyncEventFamilySync* e = NVBLASTEXT_NEW(ExtSyncEventFamilySync) (); + e->timestamp = duration_cast<milliseconds>(steady_clock::now().time_since_epoch()).count(); + e->familyID = family.getID(); + const NvBlastFamily* familyLL = family.getFamilyLL(); + const uint32_t size = NvBlastFamilyGetSize(familyLL, NvBlastTkFrameworkGet()->getLogFn()); + e->family = std::vector<char>((char*)familyLL, (char*)familyLL + size); + m_syncEvents.push_back(e); +} + +void ExtSyncImpl::syncFamily(const ExtPxFamily& family) +{ + const TkFamily& tkFamily = family.getTkFamily(); + + syncFamily(tkFamily); + + ExtSyncEventPhysicsSync* e = NVBLASTEXT_NEW(ExtSyncEventPhysicsSync) (); + e->timestamp = duration_cast<milliseconds>(steady_clock::now().time_since_epoch()).count(); + e->familyID = tkFamily.getID(); + std::vector<ExtPxActor*> actors(family.getActorCount()); + family.getActors(actors.data(), static_cast<uint32_t>(actors.size())); + e->data.reserve(actors.size()); + for (ExtPxActor* actor : actors) + { + ExtSyncEventPhysicsSync::ActorData data; + data.transform = actor->getPhysXActor().getGlobalPose(); + data.actorIndex = actor->getTkActor().getIndex(); + e->data.push_back(data); + } + + m_syncEvents.push_back(e); +} + +uint32_t ExtSyncImpl::getSyncBufferSize() const +{ + return static_cast<uint32_t>(m_syncEvents.size()); +} + +void ExtSyncImpl::acquireSyncBuffer(const ExtSyncEvent* const*& buffer, uint32_t& size) const +{ + buffer = m_syncEvents.data(); + size = static_cast<uint32_t>(m_syncEvents.size()); +} + +void ExtSyncImpl::releaseSyncBuffer() +{ + for (uint32_t i = 0; i < m_syncEvents.size(); ++i) + { + NVBLASTEXT_DELETE(m_syncEvents[i], ExtSyncEvent); + } + m_syncEvents.clear(); +} + +void ExtSyncImpl::applySyncBuffer(TkFramework& framework, const ExtSyncEvent** buffer, uint32_t size, TkGroup* groupForNewActors, ExtPxManager* manager) +{ + const TkType* familyType = framework.getType(TkTypeIndex::Family); + NVBLAST_ASSERT(familyType); + + for (uint32_t i = 0; i < size; ++i) + { + const ExtSyncEvent* e = buffer[i]; + const NvBlastID& id = e->familyID; + TkIdentifiable* object = framework.findObjectByID(id); + if (object && object->getType() == *familyType) + { + TkFamily* family = static_cast<TkFamily*>(object); + + if (e->type == ExtSyncEventFracture::EVENT_TYPE) + { + const ExtSyncEventFracture* fractureEvent = e->getEvent<ExtSyncEventFracture>(); + const NvBlastFractureBuffers commands = + { + static_cast<uint32_t>(fractureEvent->bondFractures.size()), + static_cast<uint32_t>(fractureEvent->chunkFractures.size()), + const_cast<NvBlastBondFractureData*>(fractureEvent->bondFractures.data()), + const_cast<NvBlastChunkFractureData*>(fractureEvent->chunkFractures.data()) + }; + family->applyFracture(&commands); + } + else if (e->type == ExtSyncEventFamilySync::EVENT_TYPE) + { + const ExtSyncEventFamilySync* familyEvent = e->getEvent<ExtSyncEventFamilySync>(); + family->reinitialize((NvBlastFamily*)familyEvent->family.data(), groupForNewActors); + } + else if (e->type == ExtSyncEventPhysicsSync::EVENT_TYPE && manager) + { + const ExtSyncEventPhysicsSync* physicsEvent = e->getEvent<ExtSyncEventPhysicsSync>(); + ExtPxFamily* pxFamily = manager->getFamilyFromTkFamily(*family); + if (pxFamily) + { + std::vector<ExtPxActor*> actors(pxFamily->getActorCount()); + pxFamily->getActors(actors.data(), static_cast<uint32_t>(actors.size())); + + for (auto data : physicsEvent->data) + { + for (ExtPxActor* physicsaActor : actors) + { + if (data.actorIndex == physicsaActor->getTkActor().getIndex()) + { + physicsaActor->getPhysXActor().setGlobalPose(data.transform); + } + } + } + } + } + } + } +} + +} // namespace Blast +} // namespace Nv |