package com.supermap.interfaces.ar.rajawali;

import android.opengl.GLES20;

import org.rajawali3d.Geometry3D;
import org.rajawali3d.Object3D;
import org.rajawali3d.materials.Material;
import org.rajawali3d.math.vector.Vector3;

import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.nio.FloatBuffer;

/**
 * Created by yanhang on 1/10/17.
 */

public class TrajectoryPointCloud extends Object3D {

    private static final int MAX_NUMBER_OF_VERTICES = 100000;
    private Vector3 mLastPoint = new Vector3(0,0,0);
    private static final double mMinDistance = 0.05;
    private FloatBuffer mVertexBuffer;
    private int mTrajectoryCount;



    public TrajectoryPointCloud(int color, float thickness){
        super();
        init(true);

        Material m = new Material();
        m.setColor(color);


        setMaterial(m);
        mVertexBuffer = ByteBuffer.allocateDirect(MAX_NUMBER_OF_VERTICES * Geometry3D.FLOAT_SIZE_BYTES * 3).order(ByteOrder.nativeOrder()).asFloatBuffer();
    }

    protected void init(boolean createVBOs){


        float[] vertices = new float[MAX_NUMBER_OF_VERTICES * 3];
        int[] indices = new int[MAX_NUMBER_OF_VERTICES];
        for(int i=0; i<indices.length; ++i){
            indices[i] = i;
        }

        setData(vertices, GLES20.GL_STATIC_DRAW,
                null, GLES20.GL_STATIC_DRAW,
                null, GLES20.GL_STATIC_DRAW,
                null,  GLES20.GL_STATIC_DRAW,
                indices, GLES20.GL_STATIC_DRAW,
                createVBOs);
    }

    public void addSegmentTo(Vector3 vertex){
        if(mTrajectoryCount >= MAX_NUMBER_OF_VERTICES){
            return;
        }
        if(mLastPoint.length() > 1e-05 && vertex.distanceTo(mLastPoint) < mMinDistance){
            return;
        }
        mVertexBuffer.position(mTrajectoryCount * 3);
        mVertexBuffer.put((float) vertex.x);
        mVertexBuffer.put((float) vertex.y);
        mVertexBuffer.put((float) vertex.z);
        mTrajectoryCount++;
        mLastPoint = vertex.clone();
        mGeometry.setNumIndices(mTrajectoryCount);
        mGeometry.getVertices().position(0);
        mGeometry.changeBufferData(mGeometry.getVertexBufferInfo(), mVertexBuffer, 0, mTrajectoryCount * 3);



    }

    public void resetTrajectory(){
        mTrajectoryCount = 0;
    }

    public void preRender(){

        super.preRender();
        setDrawingMode(GLES20.GL_POINTS);

    }

    public Vector3 getLastPoint(){
        return mLastPoint;
    }


}