//
//  EntityTreeHeadlessViewer.cpp
//  libraries/models/src
//
//  Created by Brad Hefta-Gaub on 2/26/14.
//  Copyright 2014 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 "EntityTreeHeadlessViewer.h"
#include "SimpleEntitySimulation.h"

EntityTreeHeadlessViewer::EntityTreeHeadlessViewer()
    :   OctreeHeadlessViewer(), _simulation(NULL) {
}

EntityTreeHeadlessViewer::~EntityTreeHeadlessViewer() {
}

void EntityTreeHeadlessViewer::init() {
    OctreeHeadlessViewer::init();
    if (!_simulation) {
        SimpleEntitySimulation* simpleSimulation = new SimpleEntitySimulation();
        EntityTree* entityTree = static_cast<EntityTree*>(_tree);
        simpleSimulation->setEntityTree(entityTree);
        entityTree->setSimulation(simpleSimulation);
        _simulation = simpleSimulation;
    }
}

void EntityTreeHeadlessViewer::update() {
    if (_tree) {
        EntityTree* tree = static_cast<EntityTree*>(_tree);
        if (tree->tryLockForWrite()) {
            tree->update();
            tree->unlock();
        }
    }
}

void EntityTreeHeadlessViewer::processEraseMessage(const QByteArray& dataByteArray, const SharedNodePointer& sourceNode) {
    static_cast<EntityTree*>(_tree)->processEraseMessage(dataByteArray, sourceNode);
}