/*
 * Copyright (C) 2022 The Android Open Source Project
 *
 * 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.
 */

#ifndef ANDROID_HWC_HOSTUTILS_H
#define ANDROID_HWC_HOSTUTILS_H

#include "Common.h"
#include "HostConnection.h"

namespace aidl::android::hardware::graphics::composer3::impl {

HostConnection* createOrGetHostConnection();

inline HWC3::Error getAndValidateHostConnection(HostConnection** ppHostCon,
                                                ExtendedRCEncoderContext** ppRcEnc) {
    *ppHostCon = nullptr;
    *ppRcEnc = nullptr;

    HostConnection* hostCon = createOrGetHostConnection();
    if (!hostCon) {
        ALOGE("%s: Failed to get host connection\n", __FUNCTION__);
        return HWC3::Error::NoResources;
    }
    ExtendedRCEncoderContext* rcEnc = hostCon->rcEncoder();
    if (!rcEnc) {
        ALOGE("%s: Failed to get renderControl encoder context\n", __FUNCTION__);
        return HWC3::Error::NoResources;
    }

    *ppHostCon = hostCon;
    *ppRcEnc = rcEnc;
    return HWC3::Error::None;
}

#define DEFINE_AND_VALIDATE_HOST_CONNECTION                               \
    HostConnection* hostCon;                                              \
    ExtendedRCEncoderContext* rcEnc;                                      \
    {                                                                     \
        HWC3::Error res = getAndValidateHostConnection(&hostCon, &rcEnc); \
        if (res != HWC3::Error::None) {                                   \
            return res;                                                   \
        }                                                                 \
    }
}  // namespace aidl::android::hardware::graphics::composer3::impl

#endif