281 lines · 8.9 KB · C++
RawDownload
////  Created by Bradley Austin Davis on 2014/04/13.//  Copyright 2015 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 "OculusLegacyDisplayPlugin.h" #include <memory> #include <gl/Config.h> #include <QThread>#include <QtWidgets/QMainWindow>#include <QtOpenGL/QGLWidget>#include <GLMHelpers.h>#include <QEvent>#include <QtGui/QResizeEvent>#include <QtGui/QGuiApplication>#include <QtGui/QScreen>#include <gl/GLWindow.h>#include <gl/GLWidget.h>#include <MainWindow.h> #include <gl/QOpenGLContextWrapper.h>#include <PerfStat.h>#include <ViewFrustum.h>#include <gpu/gl/GLBackend.h> #include <ui-plugins/PluginContainer.h> #include "OculusHelpers.h" const char* OculusLegacyDisplayPlugin::NAME { "Oculus Rift" }; OculusLegacyDisplayPlugin::OculusLegacyDisplayPlugin() {} void OculusLegacyDisplayPlugin::init() {    Plugin::init();     emit deviceConnected(getName());} void OculusLegacyDisplayPlugin::resetSensors() {    ovrHmd_RecenterPose(_hmd);} bool OculusLegacyDisplayPlugin::beginFrameRender(uint32_t frameIndex) {     _currentRenderFrameInfo = FrameInfo();    _currentRenderFrameInfo.predictedDisplayTime = _currentRenderFrameInfo.sensorSampleTime = ovr_GetTimeInSeconds();    _trackingState = ovrHmd_GetTrackingState(_hmd, _currentRenderFrameInfo.predictedDisplayTime);    _currentRenderFrameInfo.renderPose = toGlm(_trackingState.HeadPose.ThePose);    withNonPresentThreadLock([&]{        _frameInfos[frameIndex] = _currentRenderFrameInfo;    });    return Parent::beginFrameRender(frameIndex);} bool OculusLegacyDisplayPlugin::isSupported() const {    if (!ovr_Initialize(nullptr)) {        return false;    }    bool result = false;    if (ovrHmd_Detect() > 0) {        result = true;    }     auto hmd = ovrHmd_Create(0);     // The Oculus SDK seems to have trouble finding the right screen sometimes, so we have to guess    // Guesses, in order of best match:    //  - resolution and position match    //  - resolution and one component of position match    //  - resolution matches    //  - position matches    // If it still picks the wrong screen, you'll have to mess with your monitor configuration    QList<int> matches({ -1, -1, -1, -1 });    if (hmd) {        QPoint targetPosition{ hmd->WindowsPos.x, hmd->WindowsPos.y };        QSize targetResolution{ hmd->Resolution.w, hmd->Resolution.h };        auto screens = qApp->screens();        for(int i = 0; i < screens.size(); ++i) {            auto screen = screens[i];            QPoint position = screen->geometry().topLeft();            QSize resolution = screen->geometry().size();             if (position == targetPosition && resolution == targetResolution) {                matches[0] = i;            } else if ((position.x() == targetPosition.x() || position.y() == targetPosition.y()) &&                resolution == targetResolution) {                matches[1] = i;            } else if (resolution == targetResolution) {                matches[2] = i;            } else if (position == targetPosition) {                matches[3] = i;            }        }    }     for (int screen : matches) {        if (screen != -1) {            _hmdScreen = screen;            break;        }    }     if (_hmdScreen == -1) {        qDebug() << "Could not find Rift screen";        result = false;    }      ovr_Shutdown();    return result;} bool OculusLegacyDisplayPlugin::internalActivate() {    if (!Parent::internalActivate()) {        return false;    }     if (!(ovr_Initialize(nullptr))) {        Q_ASSERT(false);        qFatal("Failed to Initialize SDK");        return false;    }            _hmdWindow = new GLWindow();    _hmdWindow->create();    _hmdWindow->createContext(_container->getPrimaryContext());    auto hmdScreen = qApp->screens()[_hmdScreen];    auto hmdGeometry = hmdScreen->geometry();    _hmdWindow->setGeometry(hmdGeometry);    _hmdWindow->showFullScreen();      _hmdWindow->makeCurrent();    glClearColor(1, 0, 0, 1);    glClear(GL_COLOR_BUFFER_BIT);    _hmdWindow->swapBuffers();        _container->makeRenderingContextCurrent();        QOpenGLContextWrapper(_hmdWindow->context()).moveToThread(_presentThread);     _hswDismissed = false;    _hmd = ovrHmd_Create(0);    if (!_hmd) {        qFatal("Failed to acquire HMD");        return false;    }        _ipd = ovrHmd_GetFloat(_hmd, OVR_KEY_IPD, _ipd);     glm::uvec2 eyeSizes[2];    ovr_for_each_eye([&](ovrEyeType eye) {        _eyeFovs[eye] = _hmd->MaxEyeFov[eye];        ovrEyeRenderDesc erd = _eyeRenderDescs[eye] = ovrHmd_GetRenderDesc(_hmd, eye, _eyeFovs[eye]);        ovrMatrix4f ovrPerspectiveProjection =            ovrMatrix4f_Projection(erd.Fov, DEFAULT_NEAR_CLIP, DEFAULT_FAR_CLIP, ovrProjection_RightHanded);        _eyeProjections[eye] = toGlm(ovrPerspectiveProjection);        _ovrEyeOffsets[eye] = erd.HmdToEyeViewOffset;        _eyeOffsets[eye] = glm::translate(mat4(), -1.0f * toGlm(_ovrEyeOffsets[eye]));        eyeSizes[eye] = toGlm(ovrHmd_GetFovTextureSize(_hmd, eye, erd.Fov, 1.0f));    });        auto combinedFov = _eyeFovs[0];    combinedFov.LeftTan = combinedFov.RightTan = std::max(combinedFov.LeftTan, combinedFov.RightTan);    _cullingProjection = toGlm(ovrMatrix4f_Projection(combinedFov, DEFAULT_NEAR_CLIP, DEFAULT_FAR_CLIP, ovrProjection_RightHanded));     _renderTargetSize = uvec2(        eyeSizes[0].x + eyeSizes[1].x,        std::max(eyeSizes[0].y, eyeSizes[1].y));        if (!ovrHmd_ConfigureTracking(_hmd,                                  ovrTrackingCap_Orientation | ovrTrackingCap_Position | ovrTrackingCap_MagYawCorrection, 0)) {        qFatal("Could not attach to sensor device");    }     return true;} void OculusLegacyDisplayPlugin::internalDeactivate() {    Parent::internalDeactivate();    ovrHmd_Destroy(_hmd);    _hmd = nullptr;    ovr_Shutdown();    _hmdWindow->showNormal();    _hmdWindow->destroy();    _hmdWindow->deleteLater();    _hmdWindow = nullptr;    _container->makeRenderingContextCurrent();} // DLL based display plugins MUST initialize GL inside the DLL code.void OculusLegacyDisplayPlugin::customizeContext() {    gl::initModuleGl();    _hmdWindow->requestActivate();    QThread::msleep(1000);    Parent::customizeContext();    ovrGLConfig config; memset(&config, 0, sizeof(ovrRenderAPIConfig));    auto& header = config.Config.Header;    header.API = ovrRenderAPI_OpenGL;    header.BackBufferSize = _hmd->Resolution;    header.Multisample = 1;    int distortionCaps = ovrDistortionCap_TimeWarp | ovrDistortionCap_Vignette;        memset(_eyeTextures, 0, sizeof(ovrTexture) * 2);    ovr_for_each_eye([&](ovrEyeType eye) {        auto& header = _eyeTextures[eye].Header;        header.API = ovrRenderAPI_OpenGL;        header.TextureSize = { (int)_renderTargetSize.x, (int)_renderTargetSize.y };        header.RenderViewport.Size = header.TextureSize;        header.RenderViewport.Size.w /= 2;        if (eye == ovrEye_Right) {            header.RenderViewport.Pos.x = header.RenderViewport.Size.w;        }    });        if (_hmdWindow->makeCurrent()) {#ifndef NDEBUG        ovrBool result =#endif        ovrHmd_ConfigureRendering(_hmd, &config.Config, distortionCaps, _eyeFovs, _eyeRenderDescs);        assert(result);        _hmdWindow->doneCurrent();    }     } void OculusLegacyDisplayPlugin::uncustomizeContext() {    _hmdWindow->doneCurrent();    QOpenGLContextWrapper(_hmdWindow->context()).moveToThread(qApp->thread());    Parent::uncustomizeContext();} void OculusLegacyDisplayPlugin::hmdPresent() {    if (!_hswDismissed) {        ovrHSWDisplayState hswState;        ovrHmd_GetHSWDisplayState(_hmd, &hswState);        if (hswState.Displayed) {            ovrHmd_DismissHSWDisplay(_hmd);        }     }    auto r = glm::quat_cast(_currentPresentFrameInfo.presentPose);    ovrQuatf ovrRotation = { r.x, r.y, r.z, r.w };    ovrPosef eyePoses[2];    memset(eyePoses, 0, sizeof(ovrPosef) * 2);    eyePoses[0].Orientation = eyePoses[1].Orientation = ovrRotation;        GLint texture = getGLBackend()->getTextureID(_compositeFramebuffer->getRenderBuffer(0));    auto sync = glFenceSync(GL_SYNC_GPU_COMMANDS_COMPLETE, 0);    glFlush();    if (_hmdWindow->makeCurrent()) {        glClearColor(0, 0.4, 0.8, 1);        glClear(GL_COLOR_BUFFER_BIT);        ovrHmd_BeginFrame(_hmd, _currentFrame->frameIndex);        glWaitSync(sync, 0, GL_TIMEOUT_IGNORED);        glDeleteSync(sync);        ovr_for_each_eye([&](ovrEyeType eye) {            reinterpret_cast<ovrGLTexture&>(_eyeTextures[eye]).OGL.TexId = texture;        });        ovrHmd_EndFrame(_hmd, eyePoses, _eyeTextures);        _hmdWindow->doneCurrent();    }    static auto widget = _container->getPrimaryWidget();    widget->makeCurrent();} int OculusLegacyDisplayPlugin::getHmdScreen() const {    return _hmdScreen;} float OculusLegacyDisplayPlugin::getTargetFrameRate() const {    return TARGET_RATE_OculusLegacy;}