200 lines · 7.1 KB · C++
RawDownload
////  SelectionScriptingInterface.cpp//  interface/src/scripting////  Created by Zach Fox on 2017-08-22.//  Copyright 2017 High Fidelity, Inc.////  Distributed under the Apache License, Version 2.0.//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html// #include "SelectionScriptingInterface.h"#include <QDebug>#include "Application.h" GameplayObjects::GameplayObjects() {} bool GameplayObjects::addToGameplayObjects(const QUuid& avatarID) {    _avatarIDs.push_back(avatarID);    return true;}bool GameplayObjects::removeFromGameplayObjects(const QUuid& avatarID) {    _avatarIDs.erase(std::remove(_avatarIDs.begin(), _avatarIDs.end(), avatarID), _avatarIDs.end());    return true;} bool GameplayObjects::addToGameplayObjects(const EntityItemID& entityID) {    _entityIDs.push_back(entityID);    return true;}bool GameplayObjects::removeFromGameplayObjects(const EntityItemID& entityID) {    _entityIDs.erase(std::remove(_entityIDs.begin(), _entityIDs.end(), entityID), _entityIDs.end());    return true;} bool GameplayObjects::addToGameplayObjects(const OverlayID& overlayID) {    _overlayIDs.push_back(overlayID);    return true;}bool GameplayObjects::removeFromGameplayObjects(const OverlayID& overlayID) {    _overlayIDs.erase(std::remove(_overlayIDs.begin(), _overlayIDs.end(), overlayID), _overlayIDs.end());    return true;} SelectionScriptingInterface::SelectionScriptingInterface(AbstractViewStateInterface* viewState) {    _viewState = viewState;} bool SelectionScriptingInterface::addToSelectedItemsList(const QString& listName, const QString& itemType, const QUuid& id) {    if (itemType == "avatar") {        return addToGameplayObjects(listName, (QUuid)id);    } else if (itemType == "entity") {        return addToGameplayObjects(listName, (EntityItemID)id);    } else if (itemType == "overlay") {        return addToGameplayObjects(listName, (OverlayID)id);    }    return false;}bool SelectionScriptingInterface::removeFromSelectedItemsList(const QString& listName, const QString& itemType, const QUuid& id) {    if (itemType == "avatar") {        return removeFromGameplayObjects(listName, (QUuid)id);    } else if (itemType == "entity") {        return removeFromGameplayObjects(listName, (EntityItemID)id);    } else if (itemType == "overlay") {        return removeFromGameplayObjects(listName, (OverlayID)id);    }    return false;} template <class T> bool SelectionScriptingInterface::addToGameplayObjects(const QString& listName, T idToAdd) {    if (_selectedItemsListMap.contains(listName)) {        auto currentList = _selectedItemsListMap.take(listName);        currentList.addToGameplayObjects(idToAdd);        _selectedItemsListMap.insert(listName, currentList);         updateRendererSelectedList(listName);        return true;    } else {        _selectedItemsListMap.insert(listName, GameplayObjects());        return addToGameplayObjects(listName, idToAdd);    }}template <class T> bool SelectionScriptingInterface::removeFromGameplayObjects(const QString& listName, T idToRemove) {    if (_selectedItemsListMap.contains(listName)) {        auto currentList = _selectedItemsListMap.take(listName);        currentList.removeFromGameplayObjects(idToRemove);        _selectedItemsListMap.insert(listName, currentList);         updateRendererSelectedList(listName);        return true;    } else {        return false;    }}//// END HANDLING GENERIC ITEMS// void SelectionScriptingInterface::printList(const QString& listName) {    if (_selectedItemsListMap.contains(listName)) {        auto currentList = _selectedItemsListMap.value(listName);         qDebug() << "Avatar IDs:";        for (auto i : currentList.getAvatarIDs()) {            qDebug() << i << '; ';        }        qDebug() << "";         qDebug() << "Entity IDs:";        for (auto j : currentList.getEntityIDs()) {            qDebug() << j << '; ';        }        qDebug() << "";         qDebug() << "Overlay IDs:";        for (auto k : currentList.getOverlayIDs()) {            qDebug() << k << '; ';        }        qDebug() << "";    } else {        qDebug() << "List named" << listName << "doesn't exist.";    }} bool SelectionScriptingInterface::removeListFromMap(const QString& listName) {    if (_selectedItemsListMap.remove(listName)) {        updateRendererSelectedList(listName);        return true;    } else {        return false;    }}  SelectionToSceneHandler::SelectionToSceneHandler() {} void SelectionToSceneHandler::initialize(render::ScenePointer mainScene, const QString& listName) {    _mainScene = mainScene;    _listName = listName;} void SelectionToSceneHandler::updateRendererSelectedList() {    if (_mainScene) {        render::Transaction transaction;         if (_selectedItemsListMap.contains(_listName)) {            render::ItemIDs finalList;            render::ItemID currentID;            auto entityTree = qApp->getEntities()->getTree();            auto& overlays = qApp->getOverlays();            auto currentList = _selectedItemsListMap.value(_listName);             for (QUuid& currentAvatarID : currentList.getAvatarIDs()) {                auto avatar = std::static_pointer_cast<Avatar>(DependencyManager::get<AvatarManager>()->getAvatarBySessionID(currentAvatarID));                if (avatar) {                    currentID = avatar->getRenderItemID();                    if (currentID != render::Item::INVALID_ITEM_ID) {                        finalList.push_back(currentID);                    }                }            }             for (EntityItemID& currentEntityID : currentList.getEntityIDs()) {                entityTree->withReadLock([&] {                    auto entityItem = entityTree->findEntityByEntityItemID(currentEntityID);                    if (entityItem != NULL) {                        auto renderableInterface = entityItem->getRenderableInterface();                        if (renderableInterface != NULL) {                            currentID = renderableInterface->getMetaRenderItemID();                            if (currentID != render::Item::INVALID_ITEM_ID) {                                finalList.push_back(currentID);                            }                        }                    }                });            }             for (OverlayID& currentOverlayID : currentList.getOverlayIDs()) {                auto overlay = overlays.getOverlay(currentOverlayID);                if (overlay != NULL) {                    currentID = overlay->getRenderItemID();                    if (currentID != render::Item::INVALID_ITEM_ID) {                        finalList.push_back(currentID);                    }                }            }             render::Selection selection(_listName.toStdString(), finalList);            transaction.resetSelection(selection);             _mainScene->enqueueTransaction(transaction);        } else {            qWarning() << "List of GameplayObjects doesn't exist in _selectedItemsListMap";        }    } else {        qWarning() << "SelectionScriptingInterface::updateRendererSelectedList(), Unexpected null scene, possibly during application shutdown";    }}