#pragma once

#include <td/render/shader/CameraShaderProgram.h>

namespace td {
namespace shader {

class EntityShader : public CameraShaderProgram {
  private:
	unsigned int m_LocationPosition = 0;
	unsigned int m_LocationColorEffect = 0;

  protected:
	virtual void GetAllUniformLocation();

  public:
	EntityShader();

	void SetColorEffect(const Vec3f& color);
	void SetModelPos(const Vec3f& pos) const;
};

}  // namespace shader
}  // namespace td