#include "drape/data_buffer.hpp"
#include "drape/data_buffer_impl.hpp"

namespace dp
{

DataBuffer::DataBuffer(uint8_t elementSize, uint32_t capacity)
  : m_impl(make_unique_dp<CpuBufferImpl>(elementSize, capacity))
{
}

ref_ptr<DataBufferBase> DataBuffer::GetBuffer() const
{
  ASSERT(m_impl != nullptr, ());
  return make_ref(m_impl);
}

void DataBuffer::MoveToGPU(GPUBuffer::Target target)
{
  // if currentSize is 0 buffer hasn't been filled on preparation stage, let it be filled further
  uint32_t const currentSize = m_impl->GetCurrentSize();
  if (currentSize != 0)
    m_impl = make_unique_dp<GpuBufferImpl>(target, m_impl->Data(),m_impl->GetElementSize(), currentSize);
  else
    m_impl = make_unique_dp<GpuBufferImpl>(target, nullptr, m_impl->GetElementSize(), m_impl->GetAvailableSize());
}


DataBufferMapper::DataBufferMapper(ref_ptr<DataBuffer> buffer)
  : m_buffer(buffer)
{
  m_buffer->GetBuffer()->Bind();
  m_ptr = m_buffer->GetBuffer()->Map();
}

DataBufferMapper::~DataBufferMapper()
{
  m_buffer->GetBuffer()->Unmap();
}

void DataBufferMapper::UpdateData(void const * data, uint32_t elementOffset, uint32_t elementCount)
{
  m_buffer->GetBuffer()->UpdateData(m_ptr, data, elementOffset, elementCount);
}

}