/*
 * Copyright (C) 2016 Intel Corporation. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "PooledFrameAllocator.h"
#include <common/log.h>
#include <vaapi/VaapiUtils.h>

namespace YamiMediaCodec {

struct SurfaceDestoryer
{
private:
    SharedPtr<VADisplay> m_display;
    std::vector<VASurfaceID> m_surfaces;
public:
    SurfaceDestoryer(const SharedPtr<VADisplay>& display, std::vector<VASurfaceID>& surfaces)
        :m_display(display)
    {
        m_surfaces.swap(surfaces);
    }
    void operator()(VideoPool<VideoFrame>* pool)
    {
        if (m_surfaces.size())
            vaDestroySurfaces(*m_display, &m_surfaces[0], m_surfaces.size());
        delete pool;
    }

};


PooledFrameAllocator::PooledFrameAllocator(const SharedPtr<VADisplay>& display, int poolsize)
    :m_display(display)
     , m_poolsize(poolsize)
{
}
bool PooledFrameAllocator::setFormat(uint32_t fourcc, int width, int height)
{
    std::vector<VASurfaceID> surfaces;
    surfaces.resize(m_poolsize);

    uint32_t rtFormat = getRtFormat(fourcc);
    if (!rtFormat) {
        return false;
    }

    uint32_t vaFourcc = fourcc;
    if (fourcc == YAMI_FOURCC_R210) {
        //workaround for libva, currently libva will use ARGB as fourcc for 10 bits RGB
        //it's not good,  we need change this. add dedicate fourcc for 10 bits
        vaFourcc = YAMI_FOURCC_ARGB;
    }

    VASurfaceAttrib attrib;
    attrib.flags = VA_SURFACE_ATTRIB_SETTABLE;
    attrib.type = VASurfaceAttribPixelFormat;
    attrib.value.type = VAGenericValueTypeInteger;
    attrib.value.value.i = vaFourcc;

    VAStatus status = vaCreateSurfaces(*m_display, rtFormat, width, height,
            &surfaces[0], surfaces.size(),
            &attrib, 1);

    if (status != VA_STATUS_SUCCESS) {
        ERROR("create surface failed, %s", vaErrorStr(status));
        return false;
    }
    std::deque<SharedPtr<VideoFrame> > buffers;
    for (size_t i = 0;  i < surfaces.size(); i++) {
        SharedPtr<VideoFrame> f(new VideoFrame);
        memset(f.get(), 0, sizeof(VideoFrame));
        f->crop.width = width;
        f->crop.height = height;
        f->surface = (intptr_t)surfaces[i];
        f->fourcc = fourcc;
        buffers.push_back(f);
    }
    m_pool.reset(new VideoPool<VideoFrame>(buffers), SurfaceDestoryer(m_display, surfaces));
    return true;
}

SharedPtr<VideoFrame> PooledFrameAllocator::alloc()
{
    return  m_pool->alloc();
}

} //namespace