1 /* 2 * Copyright (C) 2022-2024 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 "fa_ability_context.h" 25 #include "datashare_proxy.h" 26 #include "datashare_values_bucket.h" 27 #include "medialibrary_sync_operation.h" 28 #include "result_set_bridge.h" 29 #include "thumbnail_const.h" 30 #include "thumbnail_generate_worker.h" 31 #include "thumbnail_utils.h" 32 #include "pixel_map.h" 33 34 namespace OHOS { 35 namespace Media { 36 enum WaitStatus { 37 INSERT, 38 WAIT_SUCCESS, 39 WAIT_CONTINUE, 40 WAIT_FAILED, 41 TIMEOUT, 42 }; 43 44 enum CloudLoadType { 45 NONE, 46 CLOUD_READ_THUMB, 47 CLOUD_READ_LCD, 48 CLOUD_DOWNLOAD, 49 }; 50 51 enum CloudReadStatus { 52 START, 53 SUCCESS, 54 FAIL, 55 }; 56 57 class ThumbnailSyncStatus { 58 public: 59 bool CheckSavedFileMap(const std::string &id, ThumbnailType type, const std::string &dateModified); 60 bool UpdateSavedFileMap(const std::string &id, ThumbnailType type, const std::string &dateModified); 61 std::condition_variable cond_; 62 std::mutex mtx_; 63 bool isSyncComplete_{false}; 64 bool isCreateThumbnailSuccess_{false}; 65 std::string latestDateModified_{"0"}; 66 std::map<std::string, std::string> latestSavedFileMap_; 67 std::atomic<CloudReadStatus> CloudLoadThumbnailStatus_{START}; 68 std::atomic<CloudReadStatus> CloudLoadLcdStatus_{START}; 69 std::atomic<CloudLoadType> cloudLoadType_{NONE}; 70 }; 71 72 using ThumbnailMap = std::map<std::string, std::shared_ptr<ThumbnailSyncStatus>>; 73 class ThumbnailWait { 74 public: 75 ThumbnailWait(bool release); 76 ~ThumbnailWait(); 77 78 WaitStatus InsertAndWait(const std::string &id, ThumbnailType type, const std::string &dateModified); 79 WaitStatus CloudInsertAndWait(const std::string &id, CloudLoadType cloudLoadType); 80 void CheckAndWait(const std::string &id, bool isLcd); 81 void UpdateThumbnailMap(); 82 void UpdateCloudLoadThumbnailMap(CloudLoadType cloudLoadType, bool isLoadSuccess); 83 84 bool TrySaveCurrentPixelMap(ThumbnailData &data, ThumbnailType type); 85 bool TrySaveCurrentPicture(ThumbnailData &data, bool isSourceEx, const std::string &tempOutputPath); 86 87 private: 88 void Notify(); 89 std::string id_; 90 std::string dateModified_; 91 bool needRelease_{false}; 92 static ThumbnailMap thumbnailMap_; 93 static std::shared_mutex mutex_; 94 }; 95 96 class IThumbnailHelper { 97 public: 98 IThumbnailHelper() = default; 99 virtual ~IThumbnailHelper() = default; 100 EXPORT static void CreateLcdAndThumbnail(std::shared_ptr<ThumbnailTaskData> &data); 101 EXPORT static bool DoCreateLcdAndThumbnail(ThumbRdbOpt &opts, ThumbnailData &data, WaitStatus &ret); 102 EXPORT static void CreateLcd(std::shared_ptr<ThumbnailTaskData> &data); 103 EXPORT static void CreateThumbnail(std::shared_ptr<ThumbnailTaskData> &data); 104 EXPORT static void CreateAstc(std::shared_ptr<ThumbnailTaskData> &data); 105 EXPORT static void CreateAstcEx(std::shared_ptr<ThumbnailTaskData> &data); 106 EXPORT static void CloudSyncOnGenerationComplete(std::shared_ptr<ThumbnailTaskData> &data); 107 EXPORT static void AddThumbnailGenerateTask(ThumbnailGenerateExecute executor, 108 const ThumbnailTaskType &taskType, const ThumbnailTaskPriority &priority); 109 EXPORT static void AddThumbnailGenerateTask(ThumbnailGenerateExecute executor, ThumbRdbOpt &opts, 110 ThumbnailData &thumbData, const ThumbnailTaskType &taskType, const ThumbnailTaskPriority &priority, 111 std::shared_ptr<ExecuteParamBuilder> param = nullptr); 112 EXPORT static void AddThumbnailGenBatchTask(ThumbnailGenerateExecute executor, 113 ThumbRdbOpt &opts, ThumbnailData &thumbData, int32_t requestId = 0); 114 EXPORT static std::unique_ptr<PixelMap> GetPixelMap(const std::vector<uint8_t> &image, Size &size); 115 EXPORT static bool DoCreateLcd(ThumbRdbOpt &opts, ThumbnailData &data, WaitStatus &ret); 116 EXPORT static bool DoCreateThumbnail(ThumbRdbOpt &opts, ThumbnailData &data, WaitStatus &ret); 117 EXPORT static bool DoCreateAstc(ThumbRdbOpt &opts, ThumbnailData &data); 118 EXPORT static bool DoCreateAstcEx(ThumbRdbOpt &opts, ThumbnailData &data, WaitStatus &ret); 119 EXPORT static bool DoCreateAstcMthAndYear(ThumbRdbOpt &opts, ThumbnailData &data); 120 EXPORT static bool DoRotateThumbnail(ThumbRdbOpt &opts, ThumbnailData &data); 121 EXPORT static bool DoRotateThumbnailEx(ThumbRdbOpt &opts, ThumbnailData &data, int32_t fd, ThumbnailType thumbType); 122 EXPORT static bool IsPureCloudImage(ThumbRdbOpt &opts); 123 EXPORT static void DeleteMonthAndYearAstc(std::shared_ptr<ThumbnailTaskData> &data); 124 EXPORT static void UpdateAstcDateTaken(std::shared_ptr<ThumbnailTaskData> &data); 125 EXPORT static bool UpdateThumbnailState(const ThumbRdbOpt &opts, ThumbnailData &data, const bool isSuccess); 126 EXPORT static void UpdateHighlightDbState(ThumbRdbOpt &opts, ThumbnailData &data); 127 private: 128 EXPORT static bool TrySavePixelMap(ThumbnailData &data, ThumbnailType type); 129 EXPORT static bool TrySavePicture(ThumbnailData &data, bool isSourceEx, const std::string &tempOutputPath); 130 EXPORT static bool SaveLcdPixelMapSource(ThumbRdbOpt &opts, ThumbnailData &data, bool isSourceEx); 131 EXPORT static bool SaveLcdPictureSource(ThumbRdbOpt &opts, ThumbnailData &data, bool isSourceEx); 132 EXPORT static bool GenThumbnail(ThumbRdbOpt &opts, ThumbnailData &data, const ThumbnailType type); 133 EXPORT static bool GenThumbnailEx(ThumbRdbOpt &opts, ThumbnailData &data); 134 EXPORT static bool TryLoadSource(ThumbRdbOpt &opts, ThumbnailData &data); 135 EXPORT static bool GenMonthAndYearAstcData(ThumbnailData &data, const ThumbnailType type); 136 EXPORT static bool UpdateSuccessState(const ThumbRdbOpt &opts, const ThumbnailData &data); 137 EXPORT static bool UpdateFailState(const ThumbRdbOpt &opts, const ThumbnailData &data); 138 EXPORT static int32_t UpdateThumbDbState(const ThumbRdbOpt &opts, const ThumbnailData &data); 139 EXPORT static bool IsCreateThumbnailSuccess(ThumbRdbOpt &opts, ThumbnailData &data); 140 EXPORT static bool IsCreateThumbnailExSuccess(ThumbRdbOpt &opts, ThumbnailData &data); 141 EXPORT static bool IsCreateLcdSuccess(ThumbRdbOpt &opts, ThumbnailData &data); 142 EXPORT static bool IsCreateLcdExSuccess(ThumbRdbOpt &opts, ThumbnailData &data); 143 }; 144 } // namespace Media 145 } // namespace OHOS 146 147 #endif // FRAMEWORKS_SERVICES_THUMBNAIL_SERVICE_INCLUDE_ITHUMBNAIL_HELPER_H_ 148