1 /* 2 * Copyright (C) 2022 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef FRAMEWORKS_SERVICES_THUMBNAIL_SERVICE_INCLUDE_ITHUMBNAIL_HELPER_H_ 17 #define FRAMEWORKS_SERVICES_THUMBNAIL_SERVICE_INCLUDE_ITHUMBNAIL_HELPER_H_ 18 19 #include <map> 20 #include <shared_mutex> 21 22 #include "ability_connect_callback_stub.h" 23 #include "ability_context.h" 24 #include "datashare_proxy.h" 25 #include "datashare_values_bucket.h" 26 #include "medialibrary_async_worker.h" 27 #include "medialibrary_sync_operation.h" 28 #include "result_set_bridge.h" 29 #include "thumbnail_utils.h" 30 #include "pixel_map.h" 31 32 namespace OHOS { 33 namespace Media { 34 class GenerateAsyncTaskData : public AsyncTaskData { 35 public: 36 GenerateAsyncTaskData() = default; 37 virtual ~GenerateAsyncTaskData() override = default; 38 ThumbRdbOpt opts; 39 ThumbnailData thumbnailData; 40 }; 41 42 enum WaitStatus { 43 INSERT, 44 WAIT_SUCCESS, 45 TIMEOUT, 46 }; 47 48 using ThumbnailMap = std::map<std::string, std::shared_ptr<SyncStatus>>; 49 class ThumbnailWait { 50 public: 51 ThumbnailWait(bool release); 52 ~ThumbnailWait(); 53 54 WaitStatus InsertAndWait(const std::string &id, bool isLcd); 55 void CheckAndWait(const std::string &id, bool isLcd); 56 57 private: 58 void Notify(); 59 std::string id_; 60 bool needRelease_{false}; 61 static ThumbnailMap thumbnailMap_; 62 static std::shared_mutex mutex_; 63 }; 64 65 class IThumbnailHelper { 66 public: 67 IThumbnailHelper() = default; 68 virtual ~IThumbnailHelper() = default; 69 virtual int32_t CreateThumbnail(ThumbRdbOpt &opts, bool isSync = false); 70 virtual int32_t GetThumbnailPixelMap(ThumbRdbOpt &opts); 71 static void DeleteThumbnailKv(ThumbRdbOpt &opts); 72 static void CreateLcd(AsyncTaskData *data); 73 static void CreateThumbnail(AsyncTaskData *data); 74 static void AddAsyncTask(MediaLibraryExecute executor, ThumbRdbOpt &opts, ThumbnailData &data, bool isFront); 75 protected: 76 static void GetThumbnailInfo(ThumbRdbOpt &opts, ThumbnailData &outData); 77 static std::unique_ptr<PixelMap> GetPixelMap(const std::vector<uint8_t> &image, Size &size); 78 static bool DoCreateLcd(ThumbRdbOpt &opts, ThumbnailData &data); 79 static bool DoCreateThumbnail(ThumbRdbOpt &opts, ThumbnailData &data); 80 private: 81 static bool GenThumbnail(ThumbRdbOpt &opts, ThumbnailData &data, const ThumbnailType type); 82 static bool TryLoadSource(ThumbRdbOpt &opts, ThumbnailData &data, const Size &size, const std::string &suffix); 83 }; 84 85 // copy from foundation/distributeddatamgr/data_share/frameworks/native/common/include/idatashare.h 86 class IDataShareThumb : public IRemoteBroker { 87 public: 88 DECLARE_INTERFACE_DESCRIPTOR(u"OHOS.DataShare.IDataShare"); 89 enum { 90 CMD_INSERT = 4, 91 }; 92 virtual int Insert(const Uri &uri, const DataShare::DataShareValuesBucket &value) = 0; 93 }; 94 } // namespace Media 95 } // namespace OHOS 96 97 #endif // FRAMEWORKS_SERVICES_THUMBNAIL_SERVICE_INCLUDE_ITHUMBNAIL_HELPER_H_ 98