/*
 * Copyright (c) 2008-2017, NVIDIA CORPORATION.  All rights reserved.
 *
 * NVIDIA CORPORATION and its licensors retain all intellectual property
 * and proprietary rights in and to this software, related documentation
 * and any modifications thereto.  Any use, reproduction, disclosure or
 * distribution of this software and related documentation without an express
 * license agreement from NVIDIA CORPORATION is strictly prohibited.
 */


#ifndef __APEX_GENERALIZED_MARCHING_CUBES_H__
#define __APEX_GENERALIZED_MARCHING_CUBES_H__

#include "ApexUsingNamespace.h"
#include "PsUserAllocated.h"
#include "PsArray.h"

#include "PxBounds3.h"

namespace nvidia
{
namespace apex
{

class IProgressListener;
class ApexGeneralizedCubeTemplates;

class ApexGeneralizedMarchingCubes : public UserAllocated
{
public:
	ApexGeneralizedMarchingCubes(const PxBounds3& bound, uint32_t subdivision);
	~ApexGeneralizedMarchingCubes();

	void release();

	void registerTriangle(const PxVec3& p0, const PxVec3& p1, const PxVec3& p2);
	bool endRegistration(uint32_t bubleSizeToRemove, IProgressListener* progress);

	uint32_t getNumVertices()
	{
		return mVertices.size();
	}
	uint32_t getNumIndices()
	{
		return mIndices.size();
	}
	const PxVec3* getVertices()
	{
		return mVertices.begin();
	}
	const uint32_t* getIndices()
	{
		return mIndices.begin();
	}
private:

	struct GeneralizedVertRef
	{
		void init()
		{
			vertNr = -1;
			dangling = false;
			deleted = false;
		}
		int32_t vertNr;
		bool dangling;
		bool deleted;
	};

	struct GeneralizedCube
	{
		void init(int32_t xi, int32_t yi, int32_t zi)
		{
			this->xi = xi;
			this->yi = yi;
			this->zi = zi;
			next = -1;
			vertRefs[0].init();
			vertRefs[1].init();
			vertRefs[2].init();
			sideVertexNr[0] = -1;
			sideVertexNr[1] = -1;
			sideVertexNr[2] = -1;
			sideBounds[0].setEmpty();
			sideBounds[1].setEmpty();
			sideBounds[2].setEmpty();
			firstTriangle = -1;
			numTriangles = 0;
			deleted = false;
		}
		int32_t xi, yi, zi;
		int32_t next;
		GeneralizedVertRef vertRefs[3];
		int32_t sideVertexNr[3];
		PxBounds3 sideBounds[3];
		int32_t firstTriangle;
		uint32_t numTriangles;
		bool deleted;
	};

	inline int hashFunction(int xi, int yi, int zi)
	{
		int h = (int)(unsigned int)((xi * 92837111) ^(yi * 689287499) ^(zi * 283923481));
		return h % HASH_INDEX_SIZE;
	}
	int32_t createCube(int32_t xi, int32_t yi, int32_t zi);
	int32_t findCube(int32_t xi, int32_t yi, int32_t zi);
	void completeCells();
	void createTrianglesForCube(int32_t cellNr);
	void createNeighbourInfo();
	void getCubeEdgesAndGroups(int32_t cellNr, GeneralizedVertRef* vertRefs[12], int32_t groups[8]);
	void determineGroups();
	void removeBubbles(int32_t minGroupSize);
	void fixOrientations();
	void compress();


	PxBounds3 mBound;

	float mSpacing;
	float mInvSpacing;

	physx::Array<GeneralizedCube> mCubes;

	enum { HASH_INDEX_SIZE = 170111 };

	int32_t mFirstCube[HASH_INDEX_SIZE];

	physx::Array<PxVec3> mVertices;
	physx::Array<uint32_t> mIndices;

	physx::Array<int32_t> mFirstNeighbour;
	physx::Array<int32_t> mNeighbours;
	physx::Array<uint8_t> mTriangleDeleted;
	physx::Array<int32_t> mGeneralizedTriangles;
	physx::Array<int32_t> mCubeQueue;

	physx::Array<int32_t> mTriangleGroup;
	physx::Array<int32_t> mGroupFirstTriangle;
	physx::Array<int32_t> mGroupTriangles;

	ApexGeneralizedCubeTemplates* mTemplates;

	// for debugging only
public:
	physx::Array<PxVec3> mDebugLines;
};

}
} // end namespace nvidia::apex

#endif