From c10b0389de7ebe6f50f3d9994164d968b9abd201 Mon Sep 17 00:00:00 2001 From: David Kelly Date: Tue, 28 Feb 2017 14:45:34 -0700 Subject: [PATCH] Add a getAvatarGain method to NodeList and UsersScriptingInterface --- libraries/networking/src/NodeList.cpp | 50 ++++++++++++------- libraries/networking/src/NodeList.h | 13 +++-- .../src/UsersScriptingInterface.cpp | 6 ++- .../src/UsersScriptingInterface.h | 8 +++ 4 files changed, 52 insertions(+), 25 deletions(-) diff --git a/libraries/networking/src/NodeList.cpp b/libraries/networking/src/NodeList.cpp index f4a02ad805..7bd6800921 100644 --- a/libraries/networking/src/NodeList.cpp +++ b/libraries/networking/src/NodeList.cpp @@ -49,7 +49,7 @@ NodeList::NodeList(char newOwnerType, int socketListenPort, int dtlsListenPort) setCustomDeleter([](Dependency* dependency){ static_cast(dependency)->deleteLater(); }); - + auto addressManager = DependencyManager::get(); // handle domain change signals from AddressManager @@ -85,8 +85,8 @@ NodeList::NodeList(char newOwnerType, int socketListenPort, int dtlsListenPort) connect(&_domainHandler, &DomainHandler::icePeerSocketsReceived, this, &NodeList::pingPunchForDomainServer); auto accountManager = DependencyManager::get(); - - // assume that we may need to send a new DS check in anytime a new keypair is generated + + // assume that we may need to send a new DS check in anytime a new keypair is generated connect(accountManager.data(), &AccountManager::newKeypair, this, &NodeList::sendDomainServerCheckIn); // clear out NodeList when login is finished @@ -101,7 +101,7 @@ NodeList::NodeList(char newOwnerType, int socketListenPort, int dtlsListenPort) // anytime we get a new node we may need to re-send our set of ignored node IDs to it connect(this, &LimitedNodeList::nodeActivated, this, &NodeList::maybeSendIgnoreSetToNode); - + // setup our timer to send keepalive pings (it's started and stopped on domain connect/disconnect) _keepAlivePingTimer.setInterval(KEEPALIVE_PING_INTERVAL_MS); // 1s, Qt::CoarseTimer acceptable connect(&_keepAlivePingTimer, &QTimer::timeout, this, &NodeList::sendKeepAlivePings); @@ -161,11 +161,11 @@ qint64 NodeList::sendStatsToDomainServer(QJsonObject statsObject) { void NodeList::timePingReply(ReceivedMessage& message, const SharedNodePointer& sendingNode) { PingType_t pingType; - + quint64 ourOriginalTime, othersReplyTime; - + message.seek(0); - + message.readPrimitive(&pingType); message.readPrimitive(&ourOriginalTime); message.readPrimitive(&othersReplyTime); @@ -199,7 +199,7 @@ void NodeList::timePingReply(ReceivedMessage& message, const SharedNodePointer& } void NodeList::processPingPacket(QSharedPointer message, SharedNodePointer sendingNode) { - + // send back a reply auto replyPacket = constructPingReplyPacket(*message); const HifiSockAddr& senderSockAddr = message->getSenderSockAddr(); @@ -329,7 +329,7 @@ void NodeList::sendDomainServerCheckIn() { } auto domainPacket = NLPacket::create(domainPacketType); - + QDataStream packetStream(domainPacket.get()); if (domainPacketType == PacketType::DomainConnectRequest) { @@ -488,7 +488,7 @@ void NodeList::processDomainServerPathResponse(QSharedPointer m qCDebug(networking) << "Could not read query path from DomainServerPathQueryResponse. Bailing."; return; } - + QString pathQuery = QString::fromUtf8(message->getRawMessage() + message->getPosition(), numPathBytes); message->seek(message->getPosition() + numPathBytes); @@ -500,10 +500,10 @@ void NodeList::processDomainServerPathResponse(QSharedPointer m qCDebug(networking) << "Could not read resulting viewpoint from DomainServerPathQueryReponse. Bailing"; return; } - + // pull the viewpoint from the packet QString viewpoint = QString::fromUtf8(message->getRawMessage() + message->getPosition(), numViewpointBytes); - + // Hand it off to the AddressManager so it can handle it as a relative viewpoint if (DependencyManager::get()->goToViewpointForPath(viewpoint, pathQuery)) { qCDebug(networking) << "Going to viewpoint" << viewpoint << "which was the lookup result for path" << pathQuery; @@ -664,16 +664,16 @@ void NodeList::parseNodeFromPacketStream(QDataStream& packetStream) { } void NodeList::sendAssignment(Assignment& assignment) { - + PacketType assignmentPacketType = assignment.getCommand() == Assignment::CreateCommand ? PacketType::CreateAssignment : PacketType::RequestAssignment; auto assignmentPacket = NLPacket::create(assignmentPacketType); - + QDataStream packetStream(assignmentPacket.get()); packetStream << assignment; - + sendPacket(std::move(assignmentPacket), _assignmentServerSocket); } @@ -833,7 +833,7 @@ void NodeList::ignoreNodeBySessionID(const QUuid& nodeID, bool ignoreEnabled) { _ignoredNodeIDs.insert(nodeID); } { - QReadLocker personalMutedSetLocker{ &_personalMutedSetLock }; // read lock for insert + QReadLocker personalMutedSetLocker{ &_personalMutedSetLock }; // read lock for insert // add this nodeID to our set of personal muted IDs _personalMutedNodeIDs.insert(nodeID); } @@ -896,7 +896,7 @@ void NodeList::personalMuteNodeBySessionID(const QUuid& nodeID, bool muteEnabled if (muteEnabled) { - QReadLocker personalMutedSetLocker{ &_personalMutedSetLock }; // read lock for insert + QReadLocker personalMutedSetLocker{ &_personalMutedSetLock }; // read lock for insert // add this nodeID to our set of personal muted IDs _personalMutedNodeIDs.insert(nodeID); } else { @@ -981,7 +981,7 @@ void NodeList::setAvatarGain(const QUuid& nodeID, float gain) { if (audioMixer) { // setup the packet auto setAvatarGainPacket = NLPacket::create(PacketType::PerAvatarGainSet, NUM_BYTES_RFC4122_UUID + sizeof(float), true); - + // write the node ID to the packet setAvatarGainPacket->write(nodeID.toRfc4122()); // We need to convert the gain in dB (from the script) to an amplitude before packing it. @@ -990,6 +990,9 @@ void NodeList::setAvatarGain(const QUuid& nodeID, float gain) { qCDebug(networking) << "Sending Set Avatar Gain packet UUID: " << uuidStringWithoutCurlyBraces(nodeID) << "Gain:" << gain; sendPacket(std::move(setAvatarGainPacket), *audioMixer); + QWriteLocker{ &_avatarGainMapLock }; + _avatarGainMap[nodeID] = gain; + } else { qWarning() << "Couldn't find audio mixer to send set gain request"; } @@ -998,6 +1001,15 @@ void NodeList::setAvatarGain(const QUuid& nodeID, float gain) { } } +float NodeList::getAvatarGain(const QUuid& nodeID) { + QReadLocker{ &_avatarGainMapLock }; + auto it = _avatarGainMap.find(nodeID); + if (it != _avatarGainMap.cend()) { + return it->second; + } + return 0.0f; +} + void NodeList::kickNodeBySessionID(const QUuid& nodeID) { // send a request to domain-server to kick the node with the given session ID // the domain-server will handle the persistence of the kick (via username or IP) @@ -1036,7 +1048,7 @@ void NodeList::muteNodeBySessionID(const QUuid& nodeID) { mutePacket->write(nodeID.toRfc4122()); qCDebug(networking) << "Sending packet to mute node" << uuidStringWithoutCurlyBraces(nodeID); - + sendPacket(std::move(mutePacket), *audioMixer); } else { qWarning() << "Couldn't find audio mixer to send node mute request"; diff --git a/libraries/networking/src/NodeList.h b/libraries/networking/src/NodeList.h index 0e0a2fd6c8..293b0942d6 100644 --- a/libraries/networking/src/NodeList.h +++ b/libraries/networking/src/NodeList.h @@ -68,7 +68,7 @@ public: void setAssignmentServerSocket(const HifiSockAddr& serverSocket) { _assignmentServerSocket = serverSocket; } void sendAssignment(Assignment& assignment); - + void setIsShuttingDown(bool isShuttingDown) { _isShuttingDown = isShuttingDown; } void ignoreNodesInRadius(bool enabled = true); @@ -83,6 +83,7 @@ public: void personalMuteNodeBySessionID(const QUuid& nodeID, bool muteEnabled); bool isPersonalMutingNode(const QUuid& nodeID) const; void setAvatarGain(const QUuid& nodeID, float gain); + float getAvatarGain(const QUuid& nodeID); void kickNodeBySessionID(const QUuid& nodeID); void muteNodeBySessionID(const QUuid& nodeID); @@ -103,7 +104,7 @@ public slots: void processDomainServerPathResponse(QSharedPointer message); void processDomainServerConnectionTokenPacket(QSharedPointer message); - + void processPingPacket(QSharedPointer message, SharedNodePointer sendingNode); void processPingReplyPacket(QSharedPointer message, SharedNodePointer sendingNode); @@ -131,11 +132,11 @@ private slots: void handleNodePingTimeout(); void pingPunchForDomainServer(); - + void sendKeepAlivePings(); void maybeSendIgnoreSetToNode(SharedNodePointer node); - + private: NodeList() : LimitedNodeList(INVALID_PORT, INVALID_PORT) { assert(false); } // Not implemented, needed for DependencyManager templates compile NodeList(char ownerType, int socketListenPort = INVALID_PORT, int dtlsListenPort = INVALID_PORT); @@ -148,7 +149,7 @@ private: void timePingReply(ReceivedMessage& message, const SharedNodePointer& sendingNode); void sendDSPathQuery(const QString& newPath); - + void parseNodeFromPacketStream(QDataStream& packetStream); void pingPunchForInactiveNode(const SharedNodePointer& node); @@ -170,6 +171,8 @@ private: tbb::concurrent_unordered_set _ignoredNodeIDs; mutable QReadWriteLock _personalMutedSetLock; tbb::concurrent_unordered_set _personalMutedNodeIDs; + mutable QReadWriteLock _avatarGainMapLock; + tbb::concurrent_unordered_map _avatarGainMap; void sendIgnoreRadiusStateToNode(const SharedNodePointer& destinationNode); Setting::Handle _ignoreRadiusEnabled { "IgnoreRadiusEnabled", true }; diff --git a/libraries/script-engine/src/UsersScriptingInterface.cpp b/libraries/script-engine/src/UsersScriptingInterface.cpp index 3a3225ec75..6dc3188b3f 100644 --- a/libraries/script-engine/src/UsersScriptingInterface.cpp +++ b/libraries/script-engine/src/UsersScriptingInterface.cpp @@ -47,6 +47,10 @@ void UsersScriptingInterface::setAvatarGain(const QUuid& nodeID, float gain) { DependencyManager::get()->setAvatarGain(nodeID, gain); } +float UsersScriptingInterface::getAvatarGain(const QUuid& nodeID) { + return DependencyManager::get()->getAvatarGain(nodeID); +} + void UsersScriptingInterface::kick(const QUuid& nodeID) { // ask the NodeList to kick the user with the given session ID DependencyManager::get()->kickNodeBySessionID(nodeID); @@ -88,4 +92,4 @@ bool UsersScriptingInterface::getRequestsDomainListData() { } void UsersScriptingInterface::setRequestsDomainListData(bool isRequesting) { DependencyManager::get()->setRequestsDomainListData(isRequesting); -} \ No newline at end of file +} diff --git a/libraries/script-engine/src/UsersScriptingInterface.h b/libraries/script-engine/src/UsersScriptingInterface.h index 608fa937c8..acaa92d9c8 100644 --- a/libraries/script-engine/src/UsersScriptingInterface.h +++ b/libraries/script-engine/src/UsersScriptingInterface.h @@ -70,6 +70,14 @@ public slots: */ void setAvatarGain(const QUuid& nodeID, float gain); + /**jsdoc + * Gets an avatar's gain for you and you only. + * @function Users.getAvatarGain + * @param {nodeID} nodeID The node or session ID of the user whose gain you want to get. + * @return {float} gain (in dB) + */ + float getAvatarGain(const QUuid& nodeID); + /**jsdoc * Kick another user. * @function Users.kick