Merge pull request #12021 from elisa-lj11/poly_api

Google Poly API

interface/src/Application.cpp

@@ -191,6 +191,7 @@
 #include <GPUIdent.h>
 #include <gl/GLHelpers.h>
 #include <src/scripting/LimitlessVoiceRecognitionScriptingInterface.h>
+#include <src/scripting/GooglePolyScriptingInterface.h>
 #include <EntityScriptClient.h>
 #include <ModelScriptingInterface.h>
 
@@ -698,6 +699,7 @@ bool setupEssentials(int& argc, char** argv, bool runningMarkerExisted) {
     DependencyManager::set<EntityScriptClient>();
     DependencyManager::set<EntityScriptServerLogClient>();
     DependencyManager::set<LimitlessVoiceRecognitionScriptingInterface>();
+    DependencyManager::set<GooglePolyScriptingInterface>();
     DependencyManager::set<OctreeStatsProvider>(nullptr, qApp->getOcteeSceneStats());
     DependencyManager::set<AvatarBookmarks>();
     DependencyManager::set<LocationBookmarks>();
@@ -5941,6 +5943,7 @@ void Application::registerScriptEngineWithApplicationServices(ScriptEnginePointe
     scriptEngine->registerGlobalObject("Users", DependencyManager::get<UsersScriptingInterface>().data());
 
     scriptEngine->registerGlobalObject("LimitlessSpeechRecognition", DependencyManager::get<LimitlessVoiceRecognitionScriptingInterface>().data());
+    scriptEngine->registerGlobalObject("GooglePoly", DependencyManager::get<GooglePolyScriptingInterface>().data());
 
     if (auto steamClient = PluginManager::getInstance()->getSteamClientPlugin()) {
         scriptEngine->registerGlobalObject("Steam", new SteamScriptingInterface(scriptEngine.data(), steamClient.get()));

interface/src/scripting/GooglePolyScriptingInterface.cpp

@@ -0,0 +1,181 @@
+//
+//  GooglePolyScriptingInterface.cpp
+//  interface/src/scripting
+//
+//  Created by Elisa Lupin-Jimenez on 12/3/2017.
+//  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 <QEventLoop>
+#include <QtGlobal>
+#include <QJsonArray>
+#include <QJsonDocument>
+#include <QJsonObject>
+#include <QNetworkAccessManager>
+#include <QNetworkReply>
+#include <QNetworkRequest>
+#include <QString>
+#include <QTime>
+#include <QUrl>
+#include <random>
+
+#include "GooglePolyScriptingInterface.h"
+#include "ScriptEngineLogging.h"
+
+const QString LIST_POLY_URL = "https://poly.googleapis.com/v1/assets?";
+const QString GET_POLY_URL = "https://poly.googleapis.com/v1/assets/model?";
+
+const QStringList VALID_FORMATS = QStringList() << "BLOCKS" << "FBX" << "GLTF" << "GLTF2" << "OBJ" << "TILT" << "";
+const QStringList VALID_CATEGORIES = QStringList() << "animals" << "architecture" << "art" << "food" << 
+    "nature" << "objects" << "people" << "scenes" << "technology" << "transport" << "";
+
+GooglePolyScriptingInterface::GooglePolyScriptingInterface() {
+    // nothing to be implemented
+}
+
+void GooglePolyScriptingInterface::setAPIKey(const QString& key) {
+    _authCode = key;
+}
+
+QString GooglePolyScriptingInterface::getAssetList(const QString& keyword, const QString& category, const QString& format) {
+    QUrl url = formatURLQuery(keyword, category, format);
+    if (!url.isEmpty()) {
+        QByteArray json = parseJSON(url, 0).toJsonDocument().toJson();
+        return (QString) json;
+    } else {
+        qCDebug(scriptengine) << "Invalid filters were specified.";
+        return "";
+    }
+}
+
+QString GooglePolyScriptingInterface::getFBX(const QString& keyword, const QString& category) {
+    QUrl url = formatURLQuery(keyword, category, "FBX");
+    return getModelURL(url);
+}
+
+QString GooglePolyScriptingInterface::getOBJ(const QString& keyword, const QString& category) {
+    QUrl url = formatURLQuery(keyword, category, "OBJ");
+    return getModelURL(url);
+}
+
+QString GooglePolyScriptingInterface::getBlocks(const QString& keyword, const QString& category) {
+    QUrl url = formatURLQuery(keyword, category, "BLOCKS");
+    return getModelURL(url);
+}
+
+QString GooglePolyScriptingInterface::getGLTF(const QString& keyword, const QString& category) {
+    QUrl url = formatURLQuery(keyword, category, "GLTF");
+    return getModelURL(url);
+}
+
+QString GooglePolyScriptingInterface::getGLTF2(const QString& keyword, const QString& category) {
+    QUrl url = formatURLQuery(keyword, category, "GLTF2");
+    return getModelURL(url);
+}
+ 
+// This method will not be useful until we support Tilt models
+QString GooglePolyScriptingInterface::getTilt(const QString& keyword, const QString& category) {
+    QUrl url = formatURLQuery(keyword, category, "TILT");
+    return getModelURL(url);
+}
+
+// Can provide asset name or full URL to model
+QString GooglePolyScriptingInterface::getModelInfo(const QString& input) {
+    QString name(input);
+    if (input.contains("poly.googleapis") || input.contains("poly.google.com")) {
+        QStringList list = input.split("/");
+        if (input.contains("poly.googleapis")) {
+            name = list[4];
+        } else {
+            name = list.last();
+        }
+    }
+    QString urlString(GET_POLY_URL);
+    urlString = urlString.replace("model", name) + "key=" + _authCode;
+    qCDebug(scriptengine) << "Google URL request: " << urlString;
+    QUrl url(urlString);
+    QString json = parseJSON(url, 2).toString();
+    return json;
+}
+
+int GooglePolyScriptingInterface::getRandIntInRange(int length) {
+    QTime time = QTime::currentTime();
+    qsrand((uint)time.msec());
+    return qrand() % length;
+}
+
+QUrl GooglePolyScriptingInterface::formatURLQuery(const QString& keyword, const QString& category, const QString& format) {
+    QString queries;
+    if (!VALID_FORMATS.contains(format, Qt::CaseInsensitive) || !VALID_CATEGORIES.contains(category, Qt::CaseInsensitive)) {
+        return QUrl("");
+    } else {
+        if (!keyword.isEmpty()) {
+            QString keywords(keyword);
+            keywords.replace(" ", "+");
+            queries.append("&keywords=" + keywords);
+        }
+        if (!category.isEmpty()) {
+            queries.append("&category=" + category);
+        }
+        if (!format.isEmpty()) {
+            queries.append("&format=" + format);
+        }
+        QString urlString(LIST_POLY_URL + "key=" + _authCode + queries);
+        return QUrl(urlString);
+    }
+}
+
+QString GooglePolyScriptingInterface::getModelURL(const QUrl& url) {
+    qCDebug(scriptengine) << "Google URL request: " << url;
+    if (!url.isEmpty()) {
+        return parseJSON(url, 1).toString();
+    } else {
+        qCDebug(scriptengine) << "Invalid filters were specified.";
+        return "";
+    }
+}
+
+// FIXME: synchronous
+QByteArray GooglePolyScriptingInterface::getHTTPRequest(const QUrl& url) {
+    QNetworkAccessManager manager;
+    QNetworkReply *response = manager.get(QNetworkRequest(url));
+    QEventLoop event;
+    connect(response, SIGNAL(finished()), &event, SLOT(quit()));
+    event.exec();
+
+    return response->readAll();
+}
+
+// 0 = asset list, 1 = model from asset list, 2 = specific model
+QVariant GooglePolyScriptingInterface::parseJSON(const QUrl& url, int fileType) {
+    QByteArray jsonString = getHTTPRequest(url);
+    QJsonDocument doc = QJsonDocument::fromJson(jsonString);
+    QJsonObject obj = doc.object();
+    if (obj.isEmpty()) {
+        qCDebug(scriptengine) << "Assets with specified filters not found";
+        return "";
+    }
+    if (obj.keys().first() == "error") {
+        QString error = obj.value("error").toObject().value("message").toString();
+        qCDebug(scriptengine) << error;
+        return "";
+    }
+    if (fileType == 0 || fileType == 1) {
+        QJsonArray arr = obj.value("assets").toArray();
+        // return model url
+        if (fileType == 1) {
+            int random = getRandIntInRange(arr.size());
+            QJsonObject json = arr.at(random).toObject();
+            // nested JSONs
+            return json.value("formats").toArray().at(0).toObject().value("root").toObject().value("url");
+        }
+        // return whole asset list
+        return QJsonDocument(arr);
+    // return specific object
+    } else {
+        return jsonString;
+    }
+}

interface/src/scripting/GooglePolyScriptingInterface.h

@@ -0,0 +1,47 @@
+//
+//  GooglePolyScriptingInterface.h
+//  interface/src/scripting
+//
+//  Created by Elisa Lupin-Jimenez on 12/3/2017.
+//  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
+//
+
+#ifndef hifi_GooglePolyScriptingInterface_h
+#define hifi_GooglePolyScriptingInterface_h
+
+#include <QObject>
+#include <DependencyManager.h>
+
+class GooglePolyScriptingInterface : public QObject, public Dependency {
+    Q_OBJECT
+
+public:
+    GooglePolyScriptingInterface();
+
+public slots:
+    void setAPIKey(const QString& key);
+
+    QString getAssetList(const QString& keyword, const QString& category, const QString& format);
+    QString getFBX(const QString& keyword, const QString& category);
+    QString getOBJ(const QString& keyword, const QString& category);
+    QString getBlocks(const QString& keyword, const QString& categoryy);
+    QString getGLTF(const QString& keyword, const QString& category);
+    QString getGLTF2(const QString& keyword, const QString& category);
+    QString getTilt(const QString& keyword, const QString& category);
+    QString getModelInfo(const QString& input);
+
+private:
+    QString _authCode;
+
+    QUrl formatURLQuery(const QString& keyword, const QString& category, const QString& format);
+    QString getModelURL(const QUrl& url);
+    QByteArray getHTTPRequest(const QUrl& url);
+    QVariant parseJSON(const QUrl& url, int fileType);
+    int getRandIntInRange(int length);
+
+};
+
+#endif // hifi_GooglePolyScriptingInterface_h