//
//  VoxelTree.h
//  hifi
//
//  Created by Stephen Birarda on 3/13/13.
//  Copyright (c) 2013 High Fidelity, Inc. All rights reserved.
//

#ifndef __hifi__VoxelTree__
#define __hifi__VoxelTree__

#include <PointerStack.h>
#include <SimpleMovingAverage.h>

#include "CoverageMap.h"
#include "ViewFrustum.h"
#include "VoxelNode.h"
#include "VoxelNodeBag.h"
#include "VoxelSceneStats.h"

// Callback function, for recuseTreeWithOperation
typedef bool (*RecurseVoxelTreeOperation)(VoxelNode* node, void* extraData);
typedef enum {GRADIENT, RANDOM, NATURAL} creationMode;

#define NO_EXISTS_BITS         false
#define WANT_EXISTS_BITS       true
#define NO_COLOR               false
#define WANT_COLOR             true
#define IGNORE_VIEW_FRUSTUM    NULL
#define JUST_STAGE_DELETION    true
#define ACTUALLY_DELETE        false
#define COLLAPSE_EMPTY_TREE    true
#define DONT_COLLAPSE          false
#define NO_OCCLUSION_CULLING   false
#define WANT_OCCLUSION_CULLING true
#define IGNORE_COVERAGE_MAP    NULL
#define DONT_CHOP              0
#define NO_BOUNDARY_ADJUST     0
#define LOW_RES_MOVING_ADJUST  1
#define IGNORE_LAST_SENT       0
#define IGNORE_SCENE_STATS     NULL

class EncodeBitstreamParams {
public:
    int                 maxEncodeLevel;
    int                 maxLevelReached;
    const ViewFrustum*  viewFrustum;
    bool                includeColor;
    bool                includeExistsBits;
    int                 chopLevels;
    bool                deltaViewFrustum;
    const ViewFrustum*  lastViewFrustum;
    bool                wantOcclusionCulling;
    int                 boundaryLevelAdjust;
    uint64_t            lastViewFrustumSent;
    bool                forceSendScene;
    VoxelSceneStats*    stats;
    CoverageMap*        map;
    
    EncodeBitstreamParams(
        int                 maxEncodeLevel      = INT_MAX, 
        const ViewFrustum*  viewFrustum         = IGNORE_VIEW_FRUSTUM,
        bool                includeColor        = WANT_COLOR, 
        bool                includeExistsBits   = WANT_EXISTS_BITS,
        int                 chopLevels          = 0, 
        bool                deltaViewFrustum    = false, 
        const ViewFrustum*  lastViewFrustum     = IGNORE_VIEW_FRUSTUM,
        bool                wantOcclusionCulling= NO_OCCLUSION_CULLING,
        CoverageMap*        map                 = IGNORE_COVERAGE_MAP,
        int                 boundaryLevelAdjust = NO_BOUNDARY_ADJUST,
        uint64_t            lastViewFrustumSent = IGNORE_LAST_SENT,
        bool                forceSendScene      = true,
        VoxelSceneStats*    stats               = IGNORE_SCENE_STATS) :
            maxEncodeLevel          (maxEncodeLevel),
            maxLevelReached         (0),
            viewFrustum             (viewFrustum),
            includeColor            (includeColor),
            includeExistsBits       (includeExistsBits),
            chopLevels              (chopLevels),
            deltaViewFrustum        (deltaViewFrustum),
            lastViewFrustum         (lastViewFrustum),
            wantOcclusionCulling    (wantOcclusionCulling),
            boundaryLevelAdjust     (boundaryLevelAdjust),
            lastViewFrustumSent     (lastViewFrustumSent),
            forceSendScene          (forceSendScene),
            stats                   (stats),
            map                     (map)
    {}
};

class VoxelTree {
public:
    // when a voxel is created in the tree (object new'd)
    long voxelsCreated;
    // when a voxel is colored/set in the tree (object may have already existed)
    long voxelsColored;
    long voxelsBytesRead;

    SimpleMovingAverage voxelsCreatedStats;
    SimpleMovingAverage voxelsColoredStats;
    SimpleMovingAverage voxelsBytesReadStats;

    VoxelTree(bool shouldReaverage = false);
    ~VoxelTree();

    VoxelNode* rootNode;
    int leavesWrittenToBitstream;

    void eraseAllVoxels();

    void processRemoveVoxelBitstream(unsigned char* bitstream, int bufferSizeBytes);
    void readBitstreamToTree(unsigned char* bitstream,  unsigned long int bufferSizeBytes, 
                             bool includeColor = WANT_COLOR, bool includeExistsBits = WANT_EXISTS_BITS, 
                             VoxelNode* destinationNode = NULL);
    void readCodeColorBufferToTree(unsigned char* codeColorBuffer, bool destructive = false);
    void deleteVoxelCodeFromTree(unsigned char* codeBuffer, bool stage = ACTUALLY_DELETE, 
                                 bool collapseEmptyTrees = DONT_COLLAPSE);
    void printTreeForDebugging(VoxelNode* startNode);
    void reaverageVoxelColors(VoxelNode* startNode);

    void deleteVoxelAt(float x, float y, float z, float s, bool stage = false);
    VoxelNode* getVoxelAt(float x, float y, float z, float s) const;
    void createVoxel(float x, float y, float z, float s, 
                     unsigned char red, unsigned char green, unsigned char blue, bool destructive = false);
    void createLine(glm::vec3 point1, glm::vec3 point2, float unitSize, rgbColor color, bool destructive = false);
    void createSphere(float r,float xc, float yc, float zc, float s, bool solid, 
                      creationMode mode, bool destructive = false, bool debug = false);

    void recurseTreeWithOperation(RecurseVoxelTreeOperation operation, void* extraData=NULL);
    void recurseTreeWithOperationDistanceSorted(RecurseVoxelTreeOperation operation, 
                                                const glm::vec3& point, void* extraData=NULL);

    int encodeTreeBitstream(VoxelNode* node, unsigned char* outputBuffer, int availableBytes, VoxelNodeBag& bag, 
                            EncodeBitstreamParams& params) const;

    bool isDirty() const { return _isDirty; };
    void clearDirtyBit() { _isDirty = false; };
    void setDirtyBit() { _isDirty = true; };
    unsigned long int getNodesChangedFromBitstream() const { return _nodesChangedFromBitstream; };

    bool findRayIntersection(const glm::vec3& origin, const glm::vec3& direction,
                             VoxelNode*& node, float& distance, BoxFace& face);

    bool findSpherePenetration(const glm::vec3& center, float radius, glm::vec3& penetration);
    bool findCapsulePenetration(const glm::vec3& start, const glm::vec3& end, float radius, glm::vec3& penetration);

    // Note: this assumes the fileFormat is the HIO individual voxels code files
    void loadVoxelsFile(const char* fileName, bool wantColorRandomizer);

    // these will read/write files that match the wireformat, excluding the 'V' leading
    void writeToSVOFile(const char* filename, VoxelNode* node = NULL) const;
    bool readFromSVOFile(const char* filename);
    // reads voxels from square image with alpha as a Y-axis
    bool readFromSquareARGB32Pixels(const uint32_t* pixels, int dimension);
    bool readFromSchematicFile(const char* filename);
    void computeBlockColor(int id, int data, int& r, int& g, int& b, int& create);

    unsigned long getVoxelCount();

    void copySubTreeIntoNewTree(VoxelNode* startNode, VoxelTree* destinationTree, bool rebaseToRoot);
    void copyFromTreeIntoSubTree(VoxelTree* sourceTree, VoxelNode* destinationNode);
    
    bool getShouldReaverage() const { return _shouldReaverage; }

    void recurseNodeWithOperation(VoxelNode* node, RecurseVoxelTreeOperation operation, void* extraData);
    void recurseNodeWithOperationDistanceSorted(VoxelNode* node, RecurseVoxelTreeOperation operation, 
                const glm::vec3& point, void* extraData);


    void recurseTreeWithOperationDistanceSortedTimed(PointerStack* stackOfNodes, long allowedTime,
                                                            RecurseVoxelTreeOperation operation, 
                                                            const glm::vec3& point, void* extraData);
    
private:
    void deleteVoxelCodeFromTreeRecursion(VoxelNode* node, void* extraData);
    void readCodeColorBufferToTreeRecursion(VoxelNode* node, void* extraData);

    int encodeTreeBitstreamRecursion(VoxelNode* node, unsigned char* outputBuffer, int availableBytes, VoxelNodeBag& bag, 
                                     EncodeBitstreamParams& params, int& currentEncodeLevel) const;

    static bool countVoxelsOperation(VoxelNode* node, void* extraData);

    VoxelNode* nodeForOctalCode(VoxelNode* ancestorNode, unsigned char* needleCode, VoxelNode** parentOfFoundNode) const;
    VoxelNode* createMissingNode(VoxelNode* lastParentNode, unsigned char* deepestCodeToCreate);
    int readNodeData(VoxelNode *destinationNode, unsigned char* nodeData, int bufferSizeBytes, 
                     bool includeColor = WANT_COLOR, bool includeExistsBits = WANT_EXISTS_BITS);
    
    bool _isDirty;
    unsigned long int _nodesChangedFromBitstream;
    bool _shouldReaverage;
};

float boundaryDistanceForRenderLevel(unsigned int renderLevel);
float boundaryDistanceSquaredForRenderLevel(unsigned int renderLevel);

#endif /* defined(__hifi__VoxelTree__) */