//
//  EntitySimulation.h
//  libraries/entities/src
//
//  Created by Andrew Meadows on 2014.11.24
//  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
//

#ifndef hifi_EntitySimulation_h
#define hifi_EntitySimulation_h

#include <QtCore/QObject>
#include <QSet>
#include <QVector>

#include <PerfStat.h>

#include "EntityItem.h"
#include "EntityTree.h"

typedef QSet<EntityItem*> SetOfEntities;
typedef QVector<EntityItem*> VectorOfEntities;

// the EntitySimulation needs to know when these things change on an entity, 
// so it can sort EntityItem or relay its state to the PhysicsEngine.
const int DIRTY_SIMULATION_FLAGS = 
        EntityItem::DIRTY_POSITION |
        EntityItem::DIRTY_ROTATION |
        EntityItem::DIRTY_LINEAR_VELOCITY |
        EntityItem::DIRTY_ANGULAR_VELOCITY |
        EntityItem::DIRTY_MASS |
        EntityItem::DIRTY_COLLISION_GROUP |
        EntityItem::DIRTY_MOTION_TYPE |
        EntityItem::DIRTY_SHAPE |
        EntityItem::DIRTY_LIFETIME |
        EntityItem::DIRTY_UPDATEABLE |
        EntityItem::DIRTY_MATERIAL;

class EntitySimulation : public QObject {
Q_OBJECT
public:
    EntitySimulation() : _mutex(QMutex::Recursive), _entityTree(NULL), _nextExpiry(quint64(-1)) { }
    virtual ~EntitySimulation() { setEntityTree(NULL); }

    void lock() { _mutex.lock(); }
    void unlock() { _mutex.unlock(); }

    /// \param tree pointer to EntityTree which is stored internally
    void setEntityTree(EntityTree* tree);

    void updateEntities();

    friend class EntityTree;

protected: // these only called by the EntityTree?
    /// \param entity pointer to EntityItem to be added
    /// \sideeffect sets relevant backpointers in entity, but maybe later when appropriate data structures are locked
    void addEntity(EntityItem* entity);

    /// \param entity pointer to EntityItem to be removed
    /// \brief the actual removal may happen later when appropriate data structures are locked
    /// \sideeffect nulls relevant backpointers in entity
    void removeEntity(EntityItem* entity);

    /// \param entity pointer to EntityItem to that may have changed in a way that would affect its simulation
    /// call this whenever an entity was changed from some EXTERNAL event (NOT by the EntitySimulation itself)
    void changeEntity(EntityItem* entity);

    void clearEntities();

    void moveSimpleKinematics(const quint64& now);

public:

    EntityTree* getEntityTree() { return _entityTree; }

    void getEntitiesToDelete(VectorOfEntities& entitiesToDelete);

signals:
    void entityCollisionWithEntity(const EntityItemID& idA, const EntityItemID& idB, const Collision& collision);

protected:

    // These pure virtual methods are protected because they are not to be called will-nilly. The base class
    // calls them in the right places.
    virtual void updateEntitiesInternal(const quint64& now) = 0;
    virtual void addEntityInternal(EntityItem* entity);
    virtual void removeEntityInternal(EntityItem* entity) = 0;
    virtual void changeEntityInternal(EntityItem* entity);
    virtual void clearEntitiesInternal() = 0;

    void expireMortalEntities(const quint64& now);
    void callUpdateOnEntitiesThatNeedIt(const quint64& now);
    void sortEntitiesThatMoved();

    QMutex _mutex;

    // back pointer to EntityTree structure
    EntityTree* _entityTree;

    // We maintain multiple lists, each for its distinct purpose.
    // An entity may be in more than one list.
    SetOfEntities _mortalEntities; // entities that have an expiry
    quint64 _nextExpiry;
    SetOfEntities _entitiesToUpdate; // entities that need to call EntityItem::update()
    SetOfEntities _entitiesToSort; // entities moved by simulation (and might need resort in EntityTree)
    SetOfEntities _entitiesToDelete; // entities simulation decided needed to be deleted (EntityTree will actually delete)
    SetOfEntities _simpleKinematicEntities; // entities undergoing non-colliding kinematic motion

private:
    void moveSimpleKinematics();
};

#endif // hifi_EntitySimulation_h