overte/tools/oven/src/Oven.cpp

//
//  Oven.cpp
//  tools/oven/src
//
//  Created by Stephen Birarda on 4/5/17.
//  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 "Oven.h"

#include <QtCore/QDebug>
#include <QtCore/QThread>

#include <image/Image.h>

#include <DependencyManager.h>
#include <StatTracker.h>
#include <ResourceManager.h>

Oven* Oven::_staticInstance { nullptr };

Oven::Oven() {
    _staticInstance = this;

    // enable compression in image library
    image::setColorTexturesCompressionEnabled(true);
    image::setGrayscaleTexturesCompressionEnabled(true);
    image::setNormalTexturesCompressionEnabled(true);
    image::setCubeTexturesCompressionEnabled(true);

    // setup our worker threads
    setupWorkerThreads(QThread::idealThreadCount());

    // Initialize dependencies for OBJ Baker
    DependencyManager::set<StatTracker>();
    DependencyManager::set<ResourceManager>(false);
}

Oven::~Oven() {
    // quit all worker threads and wait on them
    for (auto& thread : _workerThreads) {
        thread->quit();
    }

    for (auto& thread: _workerThreads) {
        thread->wait();
    }

    _staticInstance = nullptr;
}

void Oven::setupWorkerThreads(int numWorkerThreads) {
    _workerThreads.reserve(numWorkerThreads);

    for (auto i = 0; i < numWorkerThreads; ++i) {
        // setup a worker thread yet and add it to our concurrent vector
        auto newThread = std::unique_ptr<QThread> { new QThread };
        newThread->setObjectName("Oven Worker Thread " + QString::number(i + 1));

        _workerThreads.push_back(std::move(newThread));
    }
}

QThread* Oven::getNextWorkerThread() {
    // Here we replicate some of the functionality of QThreadPool by giving callers an available worker thread to use.
    // We can't use QThreadPool because we want to put QObjects with signals/slots on these threads.
    // So instead we setup our own list of threads, up to one less than the ideal thread count
    // (for the FBX Baker Thread to have room), and cycle through them to hand a usable running thread back to our callers.

    auto nextIndex = ++_nextWorkerThreadIndex;
    auto& nextThread = _workerThreads[nextIndex % _workerThreads.size()];

    // start the thread if it isn't running yet
    if (!nextThread->isRunning()) {
        nextThread->start();
    }

    return nextThread.get();
}