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 <memory> 21 #include <shared_mutex> 22 23 #include "ability_connect_callback_stub.h" 24 #include "ability_context.h" 25 #include "datashare_proxy.h" 26 #include "datashare_values_bucket.h" 27 #include "medialibrary_async_worker.h" 28 #include "medialibrary_sync_operation.h" 29 #include "result_set_bridge.h" 30 #include "thumbnail_const.h" 31 #include "thumbnail_utils.h" 32 #include "pixel_map.h" 33 34 namespace OHOS { 35 namespace Media { 36 class GenerateAsyncTaskData : public AsyncTaskData { 37 public: 38 GenerateAsyncTaskData() = default; 39 virtual ~GenerateAsyncTaskData() override = default; 40 ThumbRdbOpt opts; 41 ThumbnailData thumbnailData; 42 }; 43 44 enum WaitStatus { 45 INSERT, 46 WAIT_SUCCESS, 47 TIMEOUT, 48 }; 49 50 using ThumbnailMap = std::map<std::string, std::shared_ptr<SyncStatus>>; 51 class ThumbnailWait { 52 public: 53 ThumbnailWait(bool release); 54 ~ThumbnailWait(); 55 56 WaitStatus InsertAndWait(const std::string &id, bool isLcd); 57 void CheckAndWait(const std::string &id, bool isLcd); 58 59 private: 60 void Notify(); 61 std::string id_; 62 bool needRelease_{false}; 63 static ThumbnailMap thumbnailMap_; 64 static std::shared_mutex mutex_; 65 }; 66 67 class IThumbnailHelper { 68 public: 69 IThumbnailHelper() = default; 70 virtual ~IThumbnailHelper() = default; 71 virtual int32_t CreateThumbnail(ThumbRdbOpt &opts, bool isSync = false); 72 virtual int32_t GetThumbnailPixelMap(ThumbRdbOpt &opts, const Size &size, bool isAstc = false); 73 static void CreateLcd(AsyncTaskData *data); 74 static void CreateThumbnail(AsyncTaskData *data); 75 static void CreateAstc(AsyncTaskData *data); 76 static void AddAsyncTask(MediaLibraryExecute executor, ThumbRdbOpt &opts, ThumbnailData &data, bool isFront); 77 protected: 78 static void GetThumbnailInfo(ThumbRdbOpt &opts, ThumbnailData &outData); 79 static std::unique_ptr<PixelMap> GetPixelMap(const std::vector<uint8_t> &image, Size &size); 80 static bool DoCreateLcd(ThumbRdbOpt &opts, ThumbnailData &data, bool forQuery = true); 81 static bool DoCreateThumbnail(ThumbRdbOpt &opts, ThumbnailData &data, bool forQuery = true); 82 static bool DoCreateAstc(ThumbRdbOpt &opts, ThumbnailData &data, bool forQuery = true); 83 static bool IsPureCloudImage(ThumbRdbOpt &opts); 84 private: 85 static bool GenThumbnail(ThumbRdbOpt &opts, ThumbnailData &data, const ThumbnailType type); 86 static bool TryLoadSource(ThumbRdbOpt &opts, ThumbnailData &data, const std::string &suffix, 87 bool isLoadFromSourcePath); 88 static bool GenMonthAndYearAstcData(ThumbnailData &data, const ThumbnailType type); 89 static bool UpdateThumbnailState(const ThumbRdbOpt &opts, const ThumbnailData &data); 90 static int32_t UpdateAstcState(const ThumbRdbOpt &opts, const ThumbnailData &data); 91 }; 92 } // namespace Media 93 } // namespace OHOS 94 95 #endif // FRAMEWORKS_SERVICES_THUMBNAIL_SERVICE_INCLUDE_ITHUMBNAIL_HELPER_H_ 96