/*
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 * 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.
 */

#include "native_image.h"
#include "surface.h"
#include "surface_image.h"
#include "window.h"
#include "surface_utils.h"

using namespace OHOS;

struct OH_NativeImage {
    OHOS::sptr<OHOS::SurfaceImage> consumer;
    OHOS::sptr<OHOS::IBufferProducer> producer;
    OHOS::sptr<OHOS::Surface> pSurface = nullptr;
    struct NativeWindow* nativeWindow = nullptr;
};

OH_NativeImage* OH_NativeImage_Create(uint32_t textureId, uint32_t textureTarget)
{
    OHOS::sptr<OHOS::SurfaceImage> surfaceImage = new SurfaceImage(textureId, textureTarget);
    sptr<OHOS::IBufferProducer> producer = surfaceImage->GetProducer();
    OH_NativeImage* nativeImage = new OH_NativeImage();
    nativeImage->consumer = surfaceImage;
    nativeImage->producer = producer;
    sptr<IBufferConsumerListener> listener = new SurfaceImageListener(surfaceImage);
    nativeImage->consumer->RegisterConsumerListener(listener);
    return nativeImage;
}

OH_NativeImage* OH_ConsumerSurface_Create()
{
    OHOS::sptr<OHOS::SurfaceImage> surfaceImage = new SurfaceImage();
    sptr<OHOS::IBufferProducer> producer = surfaceImage->GetProducer();
    OH_NativeImage* nativeImage = new OH_NativeImage();
    nativeImage->consumer = surfaceImage;
    nativeImage->producer = producer;
    sptr<IBufferConsumerListener> listener = new SurfaceImageListener(surfaceImage);
    nativeImage->consumer->RegisterConsumerListener(listener);
    return nativeImage;
}

OHNativeWindow* OH_NativeImage_AcquireNativeWindow(OH_NativeImage* image)
{
    if (image == nullptr) {
        BLOGE("parameter error");
        return nullptr;
    }

    if (image->nativeWindow == nullptr) {
        if (image->pSurface == nullptr) {
            image->pSurface = Surface::CreateSurfaceAsProducer(image->producer);
        }
        BLOGE_CHECK_AND_RETURN_RET(image->pSurface != nullptr, nullptr, "pSurface is null");
        image->nativeWindow = CreateNativeWindowFromSurface(&(image->pSurface));
    }

    return image->nativeWindow;
}

int32_t OH_NativeImage_AttachContext(OH_NativeImage* image, uint32_t textureId)
{
    if (image == nullptr || image->consumer == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->AttachContext(textureId);
}

int32_t OH_NativeImage_DetachContext(OH_NativeImage* image)
{
    if (image == nullptr || image->consumer == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->DetachContext();
}

int32_t OH_NativeImage_UpdateSurfaceImage(OH_NativeImage* image)
{
    if (image == nullptr || image->consumer == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->UpdateSurfaceImage();
}

int64_t OH_NativeImage_GetTimestamp(OH_NativeImage* image)
{
    if (image == nullptr || image->consumer == nullptr) {
        BLOGE("parameter error");
        return -1;
    }
    return image->consumer->GetTimeStamp();
}

int32_t OH_NativeImage_GetTransformMatrix(OH_NativeImage* image, float matrix[16])
{
    if (image == nullptr || image->consumer == nullptr || matrix == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->GetTransformMatrix(matrix);
}

int32_t OH_NativeImage_GetTransformMatrixV2(OH_NativeImage* image, float matrix[16])
{
    if (image == nullptr || image->consumer == nullptr || matrix == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->GetTransformMatrixV2(matrix);
}

int32_t OH_NativeImage_GetSurfaceId(OH_NativeImage* image, uint64_t* surfaceId)
{
    if (image == nullptr || surfaceId == nullptr || image->consumer == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    *surfaceId = image->consumer->GetUniqueId();

    if (image->pSurface == nullptr) {
        image->pSurface = Surface::CreateSurfaceAsProducer(image->producer);
    }
    BLOGE_CHECK_AND_RETURN_RET(image->pSurface != nullptr, SURFACE_ERROR_UNKOWN, "pSurface is null");
    return SURFACE_ERROR_OK;
}

int32_t OH_NativeImage_SetOnFrameAvailableListener(OH_NativeImage* image, OH_OnFrameAvailableListener listener)
{
    if (image == nullptr || image->consumer == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->SetOnBufferAvailableListener(listener.context, listener.onFrameAvailable);
}

int32_t OH_NativeImage_UnsetOnFrameAvailableListener(OH_NativeImage* image)
{
    if (image == nullptr || image->consumer == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->UnsetOnBufferAvailableListener();
}

void OH_NativeImage_Destroy(OH_NativeImage** image)
{
    if (image == nullptr || *image == nullptr) {
        BLOGE("parameter error");
        return;
    }
    if ((*image)->consumer != nullptr) {
        (void)(*image)->consumer->UnsetOnBufferAvailableListener();
    }

    if ((*image)->nativeWindow != nullptr) {
        DestoryNativeWindow((*image)->nativeWindow);
        (*image)->nativeWindow = nullptr;
    }

    delete *image;
    *image = nullptr;
}

int32_t OH_NativeImage_AcquireNativeWindowBuffer(OH_NativeImage* image,
    OHNativeWindowBuffer** nativeWindowBuffer, int32_t* fenceFd)
{
    if (image == nullptr || image->consumer == nullptr || nativeWindowBuffer == nullptr || fenceFd == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->AcquireNativeWindowBuffer(nativeWindowBuffer, fenceFd);
}

int32_t OH_NativeImage_ReleaseNativeWindowBuffer(OH_NativeImage* image,
    OHNativeWindowBuffer* nativeWindowBuffer, int32_t fenceFd)
{
    if (image == nullptr || image->consumer == nullptr || nativeWindowBuffer == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->ReleaseNativeWindowBuffer(nativeWindowBuffer, fenceFd);
}


int32_t OH_ConsumerSurface_SetDefaultUsage(OH_NativeImage* image, uint64_t usage)
{
    if (image == nullptr || image->consumer == nullptr) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->SetDefaultUsage(usage);
}

int32_t OH_ConsumerSurface_SetDefaultSize(OH_NativeImage* image, int32_t width, int32_t height)
{
    if (image == nullptr || image->consumer == nullptr || width <= 0 || height <= 0) {
        BLOGE("parameter error");
        return SURFACE_ERROR_INVALID_PARAM;
    }
    return image->consumer->SetDefaultSize(width, height);
}