native/android/surface_control.cpp - LeafOS-Project/android_frameworks_base - Gitiles

 /*
  * Copyright 2018 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.
  */

 #include <android/hardware/configstore/1.0/ISurfaceFlingerConfigs.h>
 #include <android/native_window.h>
 #include <android/surface_control.h>

 #include <configstore/Utils.h>

 #include <gui/HdrMetadata.h>
 #include <gui/ISurfaceComposer.h>
 #include <gui/Surface.h>
 #include <gui/SurfaceComposerClient.h>
 #include <gui/SurfaceControl.h>

 #include <ui/HdrCapabilities.h>

 #include <utils/Timers.h>

 using namespace android::hardware::configstore;
 using namespace android::hardware::configstore::V1_0;
 using namespace android;
 using android::hardware::configstore::V1_0::ISurfaceFlingerConfigs;

 using Transaction = SurfaceComposerClient::Transaction;

 #define CHECK_NOT_NULL(name) \
     LOG_ALWAYS_FATAL_IF(name == nullptr, "nullptr passed as " #name " argument");

 #define CHECK_VALID_RECT(name)                                     \
     LOG_ALWAYS_FATAL_IF(!static_cast<const Rect&>(name).isValid(), \
                         "invalid arg passed as " #name " argument");

 static bool getWideColorSupport(const sp<SurfaceControl>& surfaceControl) {
     sp<SurfaceComposerClient> client = surfaceControl->getClient();
     sp<IBinder> display(client->getBuiltInDisplay(ISurfaceComposer::eDisplayIdMain));

     Vector<ui::ColorMode> colorModes;
     status_t err = client->getDisplayColorModes(display, &colorModes);
     if (err) {
         ALOGE("unable to get wide color support");
         return false;
     }

     bool wideColorBoardConfig =
         getBool<ISurfaceFlingerConfigs,
                 &ISurfaceFlingerConfigs::hasWideColorDisplay>(false);

     for (android::ui::ColorMode colorMode : colorModes) {
         switch (colorMode) {
             case ui::ColorMode::DISPLAY_P3:
             case ui::ColorMode::ADOBE_RGB:
             case ui::ColorMode::DCI_P3:
                 if (wideColorBoardConfig) {
                     return true;
                 }
                 break;
             default:
                 break;
         }
     }
     return false;
 }

 static bool getHdrSupport(const sp<SurfaceControl>& surfaceControl) {
     sp<SurfaceComposerClient> client = surfaceControl->getClient();
     sp<IBinder> display(client->getBuiltInDisplay(ISurfaceComposer::eDisplayIdMain));

     HdrCapabilities hdrCapabilities;
     status_t err = client->getHdrCapabilities(display, &hdrCapabilities);
     if (err) {
         ALOGE("unable to get hdr capabilities");
         return false;
     }

     return !hdrCapabilities.getSupportedHdrTypes().empty();
 }

 static bool isDataSpaceValid(const sp<SurfaceControl>& surfaceControl, ADataSpace dataSpace) {
     static_assert(static_cast<int>(ADATASPACE_UNKNOWN) == static_cast<int>(HAL_DATASPACE_UNKNOWN));
     static_assert(static_cast<int>(ADATASPACE_SCRGB_LINEAR) == static_cast<int>(HAL_DATASPACE_V0_SCRGB_LINEAR));
     static_assert(static_cast<int>(ADATASPACE_SRGB) == static_cast<int>(HAL_DATASPACE_V0_SRGB));
     static_assert(static_cast<int>(ADATASPACE_SCRGB) == static_cast<int>(HAL_DATASPACE_V0_SCRGB));
     static_assert(static_cast<int>(ADATASPACE_DISPLAY_P3) == static_cast<int>(HAL_DATASPACE_DISPLAY_P3));
     static_assert(static_cast<int>(ADATASPACE_BT2020_PQ) == static_cast<int>(HAL_DATASPACE_BT2020_PQ));

     switch (static_cast<android_dataspace_t>(dataSpace)) {
         case HAL_DATASPACE_UNKNOWN:
         case HAL_DATASPACE_V0_SRGB:
             return true;
         // These data space need wide gamut support.
         case HAL_DATASPACE_V0_SCRGB_LINEAR:
         case HAL_DATASPACE_V0_SCRGB:
         case HAL_DATASPACE_DISPLAY_P3:
             return getWideColorSupport(surfaceControl);
         // These data space need HDR support.
         case HAL_DATASPACE_BT2020_PQ:
             return getHdrSupport(surfaceControl);
         default:
             return false;
     }
 }

 Transaction* ASurfaceTransaction_to_Transaction(ASurfaceTransaction* aSurfaceTransaction) {
     return reinterpret_cast<Transaction*>(aSurfaceTransaction);
 }

 SurfaceControl* ASurfaceControl_to_SurfaceControl(ASurfaceControl* aSurfaceControl) {
     return reinterpret_cast<SurfaceControl*>(aSurfaceControl);
 }

 void SurfaceControl_acquire(SurfaceControl* surfaceControl) {
     // incStrong/decStrong token must be the same, doesn't matter what it is
     surfaceControl->incStrong((void*)SurfaceControl_acquire);
 }

 void SurfaceControl_release(SurfaceControl* surfaceControl) {
     // incStrong/decStrong token must be the same, doesn't matter what it is
     surfaceControl->decStrong((void*)SurfaceControl_acquire);
 }

 ASurfaceControl* ASurfaceControl_createFromWindow(ANativeWindow* window, const char* debug_name) {
     CHECK_NOT_NULL(window);
     CHECK_NOT_NULL(debug_name);

     sp<SurfaceComposerClient> client = new SurfaceComposerClient();
     if (client->initCheck() != NO_ERROR) {
         return nullptr;
     }

     uint32_t flags = ISurfaceComposerClient::eFXSurfaceBufferState;
     sp<SurfaceControl> surfaceControl =
             client->createWithSurfaceParent(String8(debug_name), 0 /* width */, 0 /* height */,
                                             // Format is only relevant for buffer queue layers.
                                             PIXEL_FORMAT_UNKNOWN /* format */, flags,
                                             static_cast<Surface*>(window));
     if (!surfaceControl) {
         return nullptr;
     }

     SurfaceControl_acquire(surfaceControl.get());
     return reinterpret_cast<ASurfaceControl*>(surfaceControl.get());
 }

 ASurfaceControl* ASurfaceControl_create(ASurfaceControl* parent, const char* debug_name) {
     CHECK_NOT_NULL(parent);
     CHECK_NOT_NULL(debug_name);

     SurfaceComposerClient* client = ASurfaceControl_to_SurfaceControl(parent)->getClient().get();

     SurfaceControl* surfaceControlParent = ASurfaceControl_to_SurfaceControl(parent);

     uint32_t flags = ISurfaceComposerClient::eFXSurfaceBufferState;
     sp<SurfaceControl> surfaceControl =
             client->createSurface(String8(debug_name), 0 /* width */, 0 /* height */,
                                   // Format is only relevant for buffer queue layers.
                                   PIXEL_FORMAT_UNKNOWN /* format */, flags,
                                   surfaceControlParent);
     if (!surfaceControl) {
         return nullptr;
     }

     SurfaceControl_acquire(surfaceControl.get());
     return reinterpret_cast<ASurfaceControl*>(surfaceControl.get());
 }

 void ASurfaceControl_release(ASurfaceControl* aSurfaceControl) {
     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);

     SurfaceControl_release(surfaceControl.get());
 }

 ASurfaceTransaction* ASurfaceTransaction_create() {
     Transaction* transaction = new Transaction;
     return reinterpret_cast<ASurfaceTransaction*>(transaction);
 }

 void ASurfaceTransaction_delete(ASurfaceTransaction* aSurfaceTransaction) {
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);
     delete transaction;
 }

 void ASurfaceTransaction_apply(ASurfaceTransaction* aSurfaceTransaction) {
     CHECK_NOT_NULL(aSurfaceTransaction);

     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     transaction->apply();
 }

 typedef struct ASurfaceControlStats {
     int64_t acquireTime;
     sp<Fence> previousReleaseFence;
 } ASurfaceControlStats;

 struct ASurfaceTransactionStats {
     std::unordered_map<ASurfaceControl*, ASurfaceControlStats> aSurfaceControlStats;
     int64_t latchTime;
     sp<Fence> presentFence;
 };

 int64_t ASurfaceTransactionStats_getLatchTime(ASurfaceTransactionStats* aSurfaceTransactionStats) {
     CHECK_NOT_NULL(aSurfaceTransactionStats);
     return aSurfaceTransactionStats->latchTime;
 }

 int ASurfaceTransactionStats_getPresentFenceFd(ASurfaceTransactionStats* aSurfaceTransactionStats) {
     CHECK_NOT_NULL(aSurfaceTransactionStats);
     auto& presentFence = aSurfaceTransactionStats->presentFence;
     return (presentFence) ? presentFence->dup() : -1;
 }

 void ASurfaceTransactionStats_getASurfaceControls(ASurfaceTransactionStats* aSurfaceTransactionStats,
                                                   ASurfaceControl*** outASurfaceControls,
                                                   size_t* outASurfaceControlsSize) {
     CHECK_NOT_NULL(aSurfaceTransactionStats);
     CHECK_NOT_NULL(outASurfaceControls);
     CHECK_NOT_NULL(outASurfaceControlsSize);

     size_t size = aSurfaceTransactionStats->aSurfaceControlStats.size();

     SurfaceControl** surfaceControls = new SurfaceControl*[size];
     ASurfaceControl** aSurfaceControls = reinterpret_cast<ASurfaceControl**>(surfaceControls);

     size_t i = 0;
     for (auto& [aSurfaceControl, aSurfaceControlStats] : aSurfaceTransactionStats->aSurfaceControlStats) {
         aSurfaceControls[i] = aSurfaceControl;
         i++;
     }

     *outASurfaceControls = aSurfaceControls;
     *outASurfaceControlsSize = size;
 }

 int64_t ASurfaceTransactionStats_getAcquireTime(ASurfaceTransactionStats* aSurfaceTransactionStats,
                                                 ASurfaceControl* aSurfaceControl) {
     CHECK_NOT_NULL(aSurfaceTransactionStats);
     CHECK_NOT_NULL(aSurfaceControl);

     const auto& aSurfaceControlStats =
             aSurfaceTransactionStats->aSurfaceControlStats.find(aSurfaceControl);
     LOG_ALWAYS_FATAL_IF(
             aSurfaceControlStats == aSurfaceTransactionStats->aSurfaceControlStats.end(),
             "ASurfaceControl not found");

     return aSurfaceControlStats->second.acquireTime;
 }

 int ASurfaceTransactionStats_getPreviousReleaseFenceFd(
             ASurfaceTransactionStats* aSurfaceTransactionStats, ASurfaceControl* aSurfaceControl) {
     CHECK_NOT_NULL(aSurfaceTransactionStats);
     CHECK_NOT_NULL(aSurfaceControl);

     const auto& aSurfaceControlStats =
             aSurfaceTransactionStats->aSurfaceControlStats.find(aSurfaceControl);
     LOG_ALWAYS_FATAL_IF(
             aSurfaceControlStats == aSurfaceTransactionStats->aSurfaceControlStats.end(),
             "ASurfaceControl not found");

     auto& previousReleaseFence = aSurfaceControlStats->second.previousReleaseFence;
     return (previousReleaseFence) ? previousReleaseFence->dup() : -1;
 }

 void ASurfaceTransactionStats_releaseASurfaceControls(ASurfaceControl** aSurfaceControls) {
     CHECK_NOT_NULL(aSurfaceControls);

     SurfaceControl** surfaceControls = reinterpret_cast<SurfaceControl**>(aSurfaceControls);
     delete[] surfaceControls;
 }

 void ASurfaceTransaction_setOnComplete(ASurfaceTransaction* aSurfaceTransaction, void* context,
                                        ASurfaceTransaction_OnComplete func) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(context);
     CHECK_NOT_NULL(func);

     TransactionCompletedCallbackTakesContext callback = [func](void* callback_context,
                                                                nsecs_t latchTime,
                                                                const sp<Fence>& presentFence,
                                                                const std::vector<SurfaceControlStats>& surfaceControlStats) {
         ASurfaceTransactionStats aSurfaceTransactionStats;

         aSurfaceTransactionStats.latchTime = latchTime;
         aSurfaceTransactionStats.presentFence = presentFence;

         auto& aSurfaceControlStats = aSurfaceTransactionStats.aSurfaceControlStats;

         for (const auto& [surfaceControl, acquireTime, previousReleaseFence] : surfaceControlStats) {
             ASurfaceControl* aSurfaceControl = reinterpret_cast<ASurfaceControl*>(surfaceControl.get());
             aSurfaceControlStats[aSurfaceControl].acquireTime = acquireTime;
             aSurfaceControlStats[aSurfaceControl].previousReleaseFence = previousReleaseFence;
         }

         (*func)(callback_context, &aSurfaceTransactionStats);
     };

     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     transaction->addTransactionCompletedCallback(callback, context);
 }

 void ASurfaceTransaction_reparent(ASurfaceTransaction* aSurfaceTransaction,
                                   ASurfaceControl* aSurfaceControl,
                                   ASurfaceControl* newParentASurfaceControl) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     sp<SurfaceControl> newParentSurfaceControl = ASurfaceControl_to_SurfaceControl(
             newParentASurfaceControl);
     sp<IBinder> newParentHandle = (newParentSurfaceControl)? newParentSurfaceControl->getHandle() : nullptr;
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     transaction->reparent(surfaceControl, newParentHandle);
 }

 void ASurfaceTransaction_setVisibility(ASurfaceTransaction* aSurfaceTransaction,
                                        ASurfaceControl* aSurfaceControl,
                                        int8_t visibility) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     switch (visibility) {
     case ASURFACE_TRANSACTION_VISIBILITY_SHOW:
         transaction->show(surfaceControl);
         break;
     case ASURFACE_TRANSACTION_VISIBILITY_HIDE:
         transaction->hide(surfaceControl);
         break;
     default:
         LOG_ALWAYS_FATAL("invalid visibility %d", visibility);
     }
 }

 void ASurfaceTransaction_setZOrder(ASurfaceTransaction* aSurfaceTransaction,
                                    ASurfaceControl* aSurfaceControl,
                                    int32_t z_order) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     transaction->setLayer(surfaceControl, z_order);
 }

 void ASurfaceTransaction_setBuffer(ASurfaceTransaction* aSurfaceTransaction,
                                    ASurfaceControl* aSurfaceControl,
                                    AHardwareBuffer* buffer, int acquire_fence_fd) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     sp<GraphicBuffer> graphic_buffer(reinterpret_cast<GraphicBuffer*>(buffer));

     transaction->setBuffer(surfaceControl, graphic_buffer);
     if (acquire_fence_fd != -1) {
         sp<Fence> fence = new Fence(acquire_fence_fd);
         transaction->setAcquireFence(surfaceControl, fence);
     }
 }

 void ASurfaceTransaction_setGeometry(ASurfaceTransaction* aSurfaceTransaction,
                                      ASurfaceControl* aSurfaceControl, const ARect& source,
                                      const ARect& destination, int32_t transform) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);
     CHECK_VALID_RECT(source);
     CHECK_VALID_RECT(destination);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     transaction->setCrop(surfaceControl, static_cast<const Rect&>(source));
     transaction->setFrame(surfaceControl, static_cast<const Rect&>(destination));
     transaction->setTransform(surfaceControl, transform);
 }

 void ASurfaceTransaction_setBufferTransparency(ASurfaceTransaction* aSurfaceTransaction,
                                                ASurfaceControl* aSurfaceControl,
                                                int8_t transparency) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     uint32_t flags = (transparency == ASURFACE_TRANSACTION_TRANSPARENCY_OPAQUE) ?
                       layer_state_t::eLayerOpaque : 0;
     transaction->setFlags(surfaceControl, flags, layer_state_t::eLayerOpaque);
 }

 void ASurfaceTransaction_setDamageRegion(ASurfaceTransaction* aSurfaceTransaction,
                                          ASurfaceControl* aSurfaceControl,
                                          const ARect rects[], uint32_t count) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     Region region;
     for (uint32_t i = 0; i < count; ++i) {
         region.merge(static_cast<const Rect&>(rects[i]));
     }

     transaction->setSurfaceDamageRegion(surfaceControl, region);
 }

 void ASurfaceTransaction_setDesiredPresentTime(ASurfaceTransaction* aSurfaceTransaction,
                                          int64_t desiredPresentTime) {
     CHECK_NOT_NULL(aSurfaceTransaction);

     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     transaction->setDesiredPresentTime(static_cast<nsecs_t>(desiredPresentTime));
 }

 void ASurfaceTransaction_setBufferAlpha(ASurfaceTransaction* aSurfaceTransaction,
                                          ASurfaceControl* aSurfaceControl,
                                          float alpha) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     LOG_ALWAYS_FATAL_IF(alpha < 0.0 || alpha > 1.0, "invalid alpha");

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     transaction->setAlpha(surfaceControl, alpha);
 }

 void ASurfaceTransaction_setHdrMetadata_smpte2086(ASurfaceTransaction* aSurfaceTransaction,
                                                   ASurfaceControl* aSurfaceControl,
                                                   struct AHdrMetadata_smpte2086* metadata) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     HdrMetadata hdrMetadata;

     if (metadata) {
         hdrMetadata.smpte2086.displayPrimaryRed.x = metadata->displayPrimaryRed.x;
         hdrMetadata.smpte2086.displayPrimaryRed.y = metadata->displayPrimaryRed.y;
         hdrMetadata.smpte2086.displayPrimaryGreen.x = metadata->displayPrimaryGreen.x;
         hdrMetadata.smpte2086.displayPrimaryGreen.y = metadata->displayPrimaryGreen.y;
         hdrMetadata.smpte2086.displayPrimaryBlue.x = metadata->displayPrimaryBlue.x;
         hdrMetadata.smpte2086.displayPrimaryBlue.y = metadata->displayPrimaryBlue.y;
         hdrMetadata.smpte2086.whitePoint.x = metadata->whitePoint.x;
         hdrMetadata.smpte2086.whitePoint.y = metadata->whitePoint.y;
         hdrMetadata.smpte2086.minLuminance = metadata->minLuminance;
         hdrMetadata.smpte2086.maxLuminance = metadata->maxLuminance;

         hdrMetadata.validTypes |= HdrMetadata::SMPTE2086;
     } else {
         hdrMetadata.validTypes &= ~HdrMetadata::SMPTE2086;
     }

     transaction->setHdrMetadata(surfaceControl, hdrMetadata);
 }

 void ASurfaceTransaction_setHdrMetadata_cta861_3(ASurfaceTransaction* aSurfaceTransaction,
                                                  ASurfaceControl* aSurfaceControl,
                                                  struct AHdrMetadata_cta861_3* metadata) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     HdrMetadata hdrMetadata;

     if (metadata) {
         hdrMetadata.cta8613.maxContentLightLevel = metadata->maxContentLightLevel;
         hdrMetadata.cta8613.maxFrameAverageLightLevel = metadata->maxFrameAverageLightLevel;

         hdrMetadata.validTypes |= HdrMetadata::CTA861_3;
     } else {
         hdrMetadata.validTypes &= ~HdrMetadata::CTA861_3;
     }

     transaction->setHdrMetadata(surfaceControl, hdrMetadata);
 }

 void ASurfaceTransaction_setColor(ASurfaceTransaction* aSurfaceTransaction,
                                   ASurfaceControl* aSurfaceControl,
                                   float r, float g, float b, float alpha,
                                   ADataSpace dataspace) {
     CHECK_NOT_NULL(aSurfaceTransaction);
     CHECK_NOT_NULL(aSurfaceControl);

     sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
     LOG_ALWAYS_FATAL_IF(!isDataSpaceValid(surfaceControl, dataspace), "invalid dataspace");
     Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

     half3 color;
     color.r = r;
     color.g = g;
     color.b = b;

     transaction->setColor(surfaceControl, color)
                 .setColorAlpha(surfaceControl, alpha)
                 .setColorDataspace(surfaceControl, static_cast<ui::Dataspace>(dataspace));
 }
	/*
	* Copyright 2018 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.
	*/

	#include <android/hardware/configstore/1.0/ISurfaceFlingerConfigs.h>
	#include <android/native_window.h>
	#include <android/surface_control.h>

	#include <configstore/Utils.h>

	#include <gui/HdrMetadata.h>
	#include <gui/ISurfaceComposer.h>
	#include <gui/Surface.h>
	#include <gui/SurfaceComposerClient.h>
	#include <gui/SurfaceControl.h>

	#include <ui/HdrCapabilities.h>

	#include <utils/Timers.h>

	using namespace android::hardware::configstore;
	using namespace android::hardware::configstore::V1_0;
	using namespace android;
	using android::hardware::configstore::V1_0::ISurfaceFlingerConfigs;

	using Transaction = SurfaceComposerClient::Transaction;

	#define CHECK_NOT_NULL(name) \
	LOG_ALWAYS_FATAL_IF(name == nullptr, "nullptr passed as " #name " argument");

	#define CHECK_VALID_RECT(name) \
	LOG_ALWAYS_FATAL_IF(!static_cast<const Rect&>(name).isValid(), \
	"invalid arg passed as " #name " argument");

	static bool getWideColorSupport(const sp<SurfaceControl>& surfaceControl) {
	sp<SurfaceComposerClient> client = surfaceControl->getClient();
	sp<IBinder> display(client->getBuiltInDisplay(ISurfaceComposer::eDisplayIdMain));

	Vector<ui::ColorMode> colorModes;
	status_t err = client->getDisplayColorModes(display, &colorModes);
	if (err) {
	ALOGE("unable to get wide color support");
	return false;
	}

	bool wideColorBoardConfig =
	getBool<ISurfaceFlingerConfigs,
	&ISurfaceFlingerConfigs::hasWideColorDisplay>(false);

	for (android::ui::ColorMode colorMode : colorModes) {
	switch (colorMode) {
	case ui::ColorMode::DISPLAY_P3:
	case ui::ColorMode::ADOBE_RGB:
	case ui::ColorMode::DCI_P3:
	if (wideColorBoardConfig) {
	return true;
	}
	break;
	default:
	break;
	}
	}
	return false;
	}

	static bool getHdrSupport(const sp<SurfaceControl>& surfaceControl) {
	sp<SurfaceComposerClient> client = surfaceControl->getClient();
	sp<IBinder> display(client->getBuiltInDisplay(ISurfaceComposer::eDisplayIdMain));

	HdrCapabilities hdrCapabilities;
	status_t err = client->getHdrCapabilities(display, &hdrCapabilities);
	if (err) {
	ALOGE("unable to get hdr capabilities");
	return false;
	}

	return !hdrCapabilities.getSupportedHdrTypes().empty();
	}

	static bool isDataSpaceValid(const sp<SurfaceControl>& surfaceControl, ADataSpace dataSpace) {
	static_assert(static_cast<int>(ADATASPACE_UNKNOWN) == static_cast<int>(HAL_DATASPACE_UNKNOWN));
	static_assert(static_cast<int>(ADATASPACE_SCRGB_LINEAR) == static_cast<int>(HAL_DATASPACE_V0_SCRGB_LINEAR));
	static_assert(static_cast<int>(ADATASPACE_SRGB) == static_cast<int>(HAL_DATASPACE_V0_SRGB));
	static_assert(static_cast<int>(ADATASPACE_SCRGB) == static_cast<int>(HAL_DATASPACE_V0_SCRGB));
	static_assert(static_cast<int>(ADATASPACE_DISPLAY_P3) == static_cast<int>(HAL_DATASPACE_DISPLAY_P3));
	static_assert(static_cast<int>(ADATASPACE_BT2020_PQ) == static_cast<int>(HAL_DATASPACE_BT2020_PQ));

	switch (static_cast<android_dataspace_t>(dataSpace)) {
	case HAL_DATASPACE_UNKNOWN:
	case HAL_DATASPACE_V0_SRGB:
	return true;
	// These data space need wide gamut support.
	case HAL_DATASPACE_V0_SCRGB_LINEAR:
	case HAL_DATASPACE_V0_SCRGB:
	case HAL_DATASPACE_DISPLAY_P3:
	return getWideColorSupport(surfaceControl);
	// These data space need HDR support.
	case HAL_DATASPACE_BT2020_PQ:
	return getHdrSupport(surfaceControl);
	default:
	return false;
	}
	}

	Transaction* ASurfaceTransaction_to_Transaction(ASurfaceTransaction* aSurfaceTransaction) {
	return reinterpret_cast<Transaction*>(aSurfaceTransaction);
	}

	SurfaceControl* ASurfaceControl_to_SurfaceControl(ASurfaceControl* aSurfaceControl) {
	return reinterpret_cast<SurfaceControl*>(aSurfaceControl);
	}

	void SurfaceControl_acquire(SurfaceControl* surfaceControl) {
	// incStrong/decStrong token must be the same, doesn't matter what it is
	surfaceControl->incStrong((void*)SurfaceControl_acquire);
	}

	void SurfaceControl_release(SurfaceControl* surfaceControl) {
	// incStrong/decStrong token must be the same, doesn't matter what it is
	surfaceControl->decStrong((void*)SurfaceControl_acquire);
	}

	ASurfaceControl* ASurfaceControl_createFromWindow(ANativeWindow* window, const char* debug_name) {
	CHECK_NOT_NULL(window);
	CHECK_NOT_NULL(debug_name);

	sp<SurfaceComposerClient> client = new SurfaceComposerClient();
	if (client->initCheck() != NO_ERROR) {
	return nullptr;
	}

	uint32_t flags = ISurfaceComposerClient::eFXSurfaceBufferState;
	sp<SurfaceControl> surfaceControl =
	client->createWithSurfaceParent(String8(debug_name), 0 /* width /, 0 / height */,
	// Format is only relevant for buffer queue layers.
	PIXEL_FORMAT_UNKNOWN /* format */, flags,
	static_cast<Surface*>(window));
	if (!surfaceControl) {
	return nullptr;
	}

	SurfaceControl_acquire(surfaceControl.get());
	return reinterpret_cast<ASurfaceControl*>(surfaceControl.get());
	}

	ASurfaceControl* ASurfaceControl_create(ASurfaceControl* parent, const char* debug_name) {
	CHECK_NOT_NULL(parent);
	CHECK_NOT_NULL(debug_name);

	SurfaceComposerClient* client = ASurfaceControl_to_SurfaceControl(parent)->getClient().get();

	SurfaceControl* surfaceControlParent = ASurfaceControl_to_SurfaceControl(parent);

	uint32_t flags = ISurfaceComposerClient::eFXSurfaceBufferState;
	sp<SurfaceControl> surfaceControl =
	client->createSurface(String8(debug_name), 0 /* width /, 0 / height */,
	// Format is only relevant for buffer queue layers.
	PIXEL_FORMAT_UNKNOWN /* format */, flags,
	surfaceControlParent);
	if (!surfaceControl) {
	return nullptr;
	}

	SurfaceControl_acquire(surfaceControl.get());
	return reinterpret_cast<ASurfaceControl*>(surfaceControl.get());
	}

	void ASurfaceControl_release(ASurfaceControl* aSurfaceControl) {
	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);

	SurfaceControl_release(surfaceControl.get());
	}

	ASurfaceTransaction* ASurfaceTransaction_create() {
	Transaction* transaction = new Transaction;
	return reinterpret_cast<ASurfaceTransaction*>(transaction);
	}

	void ASurfaceTransaction_delete(ASurfaceTransaction* aSurfaceTransaction) {
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);
	delete transaction;
	}

	void ASurfaceTransaction_apply(ASurfaceTransaction* aSurfaceTransaction) {
	CHECK_NOT_NULL(aSurfaceTransaction);

	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	transaction->apply();
	}

	typedef struct ASurfaceControlStats {
	int64_t acquireTime;
	sp<Fence> previousReleaseFence;
	} ASurfaceControlStats;

	struct ASurfaceTransactionStats {
	std::unordered_map<ASurfaceControl*, ASurfaceControlStats> aSurfaceControlStats;
	int64_t latchTime;
	sp<Fence> presentFence;
	};

	int64_t ASurfaceTransactionStats_getLatchTime(ASurfaceTransactionStats* aSurfaceTransactionStats) {
	CHECK_NOT_NULL(aSurfaceTransactionStats);
	return aSurfaceTransactionStats->latchTime;
	}

	int ASurfaceTransactionStats_getPresentFenceFd(ASurfaceTransactionStats* aSurfaceTransactionStats) {
	CHECK_NOT_NULL(aSurfaceTransactionStats);
	auto& presentFence = aSurfaceTransactionStats->presentFence;
	return (presentFence) ? presentFence->dup() : -1;
	}

	void ASurfaceTransactionStats_getASurfaceControls(ASurfaceTransactionStats* aSurfaceTransactionStats,
	ASurfaceControl*** outASurfaceControls,
	size_t* outASurfaceControlsSize) {
	CHECK_NOT_NULL(aSurfaceTransactionStats);
	CHECK_NOT_NULL(outASurfaceControls);
	CHECK_NOT_NULL(outASurfaceControlsSize);

	size_t size = aSurfaceTransactionStats->aSurfaceControlStats.size();

	SurfaceControl** surfaceControls = new SurfaceControl*[size];
	ASurfaceControl aSurfaceControls = reinterpret_cast<ASurfaceControl>(surfaceControls);

	size_t i = 0;
	for (auto& [aSurfaceControl, aSurfaceControlStats] : aSurfaceTransactionStats->aSurfaceControlStats) {
	aSurfaceControls[i] = aSurfaceControl;
	i++;
	}

	*outASurfaceControls = aSurfaceControls;
	*outASurfaceControlsSize = size;
	}

	int64_t ASurfaceTransactionStats_getAcquireTime(ASurfaceTransactionStats* aSurfaceTransactionStats,
	ASurfaceControl* aSurfaceControl) {
	CHECK_NOT_NULL(aSurfaceTransactionStats);
	CHECK_NOT_NULL(aSurfaceControl);

	const auto& aSurfaceControlStats =
	aSurfaceTransactionStats->aSurfaceControlStats.find(aSurfaceControl);
	LOG_ALWAYS_FATAL_IF(
	aSurfaceControlStats == aSurfaceTransactionStats->aSurfaceControlStats.end(),
	"ASurfaceControl not found");

	return aSurfaceControlStats->second.acquireTime;
	}

	int ASurfaceTransactionStats_getPreviousReleaseFenceFd(
	ASurfaceTransactionStats* aSurfaceTransactionStats, ASurfaceControl* aSurfaceControl) {
	CHECK_NOT_NULL(aSurfaceTransactionStats);
	CHECK_NOT_NULL(aSurfaceControl);

	const auto& aSurfaceControlStats =
	aSurfaceTransactionStats->aSurfaceControlStats.find(aSurfaceControl);
	LOG_ALWAYS_FATAL_IF(
	aSurfaceControlStats == aSurfaceTransactionStats->aSurfaceControlStats.end(),
	"ASurfaceControl not found");

	auto& previousReleaseFence = aSurfaceControlStats->second.previousReleaseFence;
	return (previousReleaseFence) ? previousReleaseFence->dup() : -1;
	}

	void ASurfaceTransactionStats_releaseASurfaceControls(ASurfaceControl** aSurfaceControls) {
	CHECK_NOT_NULL(aSurfaceControls);

	SurfaceControl surfaceControls = reinterpret_cast<SurfaceControl>(aSurfaceControls);
	delete[] surfaceControls;
	}

	void ASurfaceTransaction_setOnComplete(ASurfaceTransaction* aSurfaceTransaction, void* context,
	ASurfaceTransaction_OnComplete func) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(context);
	CHECK_NOT_NULL(func);

	TransactionCompletedCallbackTakesContext callback = [func](void* callback_context,
	nsecs_t latchTime,
	const sp<Fence>& presentFence,
	const std::vector<SurfaceControlStats>& surfaceControlStats) {
	ASurfaceTransactionStats aSurfaceTransactionStats;

	aSurfaceTransactionStats.latchTime = latchTime;
	aSurfaceTransactionStats.presentFence = presentFence;

	auto& aSurfaceControlStats = aSurfaceTransactionStats.aSurfaceControlStats;

	for (const auto& [surfaceControl, acquireTime, previousReleaseFence] : surfaceControlStats) {
	ASurfaceControl* aSurfaceControl = reinterpret_cast<ASurfaceControl*>(surfaceControl.get());
	aSurfaceControlStats[aSurfaceControl].acquireTime = acquireTime;
	aSurfaceControlStats[aSurfaceControl].previousReleaseFence = previousReleaseFence;
	}

	(*func)(callback_context, &aSurfaceTransactionStats);
	};

	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	transaction->addTransactionCompletedCallback(callback, context);
	}

	void ASurfaceTransaction_reparent(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	ASurfaceControl* newParentASurfaceControl) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	sp<SurfaceControl> newParentSurfaceControl = ASurfaceControl_to_SurfaceControl(
	newParentASurfaceControl);
	sp<IBinder> newParentHandle = (newParentSurfaceControl)? newParentSurfaceControl->getHandle() : nullptr;
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	transaction->reparent(surfaceControl, newParentHandle);
	}

	void ASurfaceTransaction_setVisibility(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	int8_t visibility) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	switch (visibility) {
	case ASURFACE_TRANSACTION_VISIBILITY_SHOW:
	transaction->show(surfaceControl);
	break;
	case ASURFACE_TRANSACTION_VISIBILITY_HIDE:
	transaction->hide(surfaceControl);
	break;
	default:
	LOG_ALWAYS_FATAL("invalid visibility %d", visibility);
	}
	}

	void ASurfaceTransaction_setZOrder(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	int32_t z_order) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	transaction->setLayer(surfaceControl, z_order);
	}

	void ASurfaceTransaction_setBuffer(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	AHardwareBuffer* buffer, int acquire_fence_fd) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	sp<GraphicBuffer> graphic_buffer(reinterpret_cast<GraphicBuffer*>(buffer));

	transaction->setBuffer(surfaceControl, graphic_buffer);
	if (acquire_fence_fd != -1) {
	sp<Fence> fence = new Fence(acquire_fence_fd);
	transaction->setAcquireFence(surfaceControl, fence);
	}
	}

	void ASurfaceTransaction_setGeometry(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl, const ARect& source,
	const ARect& destination, int32_t transform) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);
	CHECK_VALID_RECT(source);
	CHECK_VALID_RECT(destination);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	transaction->setCrop(surfaceControl, static_cast<const Rect&>(source));
	transaction->setFrame(surfaceControl, static_cast<const Rect&>(destination));
	transaction->setTransform(surfaceControl, transform);
	}

	void ASurfaceTransaction_setBufferTransparency(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	int8_t transparency) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	uint32_t flags = (transparency == ASURFACE_TRANSACTION_TRANSPARENCY_OPAQUE) ?
	layer_state_t::eLayerOpaque : 0;
	transaction->setFlags(surfaceControl, flags, layer_state_t::eLayerOpaque);
	}

	void ASurfaceTransaction_setDamageRegion(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	const ARect rects[], uint32_t count) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	Region region;
	for (uint32_t i = 0; i < count; ++i) {
	region.merge(static_cast<const Rect&>(rects[i]));
	}

	transaction->setSurfaceDamageRegion(surfaceControl, region);
	}

	void ASurfaceTransaction_setDesiredPresentTime(ASurfaceTransaction* aSurfaceTransaction,
	int64_t desiredPresentTime) {
	CHECK_NOT_NULL(aSurfaceTransaction);

	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	transaction->setDesiredPresentTime(static_cast<nsecs_t>(desiredPresentTime));
	}

	void ASurfaceTransaction_setBufferAlpha(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	float alpha) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	LOG_ALWAYS_FATAL_IF(alpha < 0.0 \|\| alpha > 1.0, "invalid alpha");

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	transaction->setAlpha(surfaceControl, alpha);
	}

	void ASurfaceTransaction_setHdrMetadata_smpte2086(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	struct AHdrMetadata_smpte2086* metadata) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	HdrMetadata hdrMetadata;

	if (metadata) {
	hdrMetadata.smpte2086.displayPrimaryRed.x = metadata->displayPrimaryRed.x;
	hdrMetadata.smpte2086.displayPrimaryRed.y = metadata->displayPrimaryRed.y;
	hdrMetadata.smpte2086.displayPrimaryGreen.x = metadata->displayPrimaryGreen.x;
	hdrMetadata.smpte2086.displayPrimaryGreen.y = metadata->displayPrimaryGreen.y;
	hdrMetadata.smpte2086.displayPrimaryBlue.x = metadata->displayPrimaryBlue.x;
	hdrMetadata.smpte2086.displayPrimaryBlue.y = metadata->displayPrimaryBlue.y;
	hdrMetadata.smpte2086.whitePoint.x = metadata->whitePoint.x;
	hdrMetadata.smpte2086.whitePoint.y = metadata->whitePoint.y;
	hdrMetadata.smpte2086.minLuminance = metadata->minLuminance;
	hdrMetadata.smpte2086.maxLuminance = metadata->maxLuminance;

	hdrMetadata.validTypes \|= HdrMetadata::SMPTE2086;
	} else {
	hdrMetadata.validTypes &= ~HdrMetadata::SMPTE2086;
	}

	transaction->setHdrMetadata(surfaceControl, hdrMetadata);
	}

	void ASurfaceTransaction_setHdrMetadata_cta861_3(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	struct AHdrMetadata_cta861_3* metadata) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	HdrMetadata hdrMetadata;

	if (metadata) {
	hdrMetadata.cta8613.maxContentLightLevel = metadata->maxContentLightLevel;
	hdrMetadata.cta8613.maxFrameAverageLightLevel = metadata->maxFrameAverageLightLevel;

	hdrMetadata.validTypes \|= HdrMetadata::CTA861_3;
	} else {
	hdrMetadata.validTypes &= ~HdrMetadata::CTA861_3;
	}

	transaction->setHdrMetadata(surfaceControl, hdrMetadata);
	}

	void ASurfaceTransaction_setColor(ASurfaceTransaction* aSurfaceTransaction,
	ASurfaceControl* aSurfaceControl,
	float r, float g, float b, float alpha,
	ADataSpace dataspace) {
	CHECK_NOT_NULL(aSurfaceTransaction);
	CHECK_NOT_NULL(aSurfaceControl);

	sp<SurfaceControl> surfaceControl = ASurfaceControl_to_SurfaceControl(aSurfaceControl);
	LOG_ALWAYS_FATAL_IF(!isDataSpaceValid(surfaceControl, dataspace), "invalid dataspace");
	Transaction* transaction = ASurfaceTransaction_to_Transaction(aSurfaceTransaction);

	half3 color;
	color.r = r;
	color.g = g;
	color.b = b;

	transaction->setColor(surfaceControl, color)
	.setColorAlpha(surfaceControl, alpha)
	.setColorDataspace(surfaceControl, static_cast<ui::Dataspace>(dataspace));
	}