1 /*
2  * Copyright (c) 2021-2021 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 HISTREAMER_AUDIO_CAPTURE_MODULE_H
17 #define HISTREAMER_AUDIO_CAPTURE_MODULE_H
18 
19 #include <string>
20 #include <memory>
21 #include "audio_capturer.h"
22 #include "common/status.h"
23 #include "meta/meta.h"
24 #include "buffer/avbuffer.h"
25 #include "osal/task/mutex.h"
26 
27 namespace OHOS {
28 namespace Media {
29 namespace AudioCaptureModule {
30 using ValueType = Any;
31 
32 class AudioCaptureModuleCallback {
33 public:
34     virtual ~AudioCaptureModuleCallback() = default;
35 
36     virtual void OnInterrupt(const std::string &interruptInfo) = 0;
37 };
38 
39 class AudioCaptureModule {
40 public:
41     explicit AudioCaptureModule();
42     ~AudioCaptureModule();
43     Status Init();
44     Status Deinit();
45     Status Prepare();
46     Status Reset();
47     Status Start();
48     Status Stop();
49     Status GetParameter(std::shared_ptr<Meta> &meta);
50     Status SetParameter(const std::shared_ptr<Meta> &meta);
51     Status Read(std::shared_ptr<AVBuffer> &buffer, size_t expectedLen);
52     Status GetSize(uint64_t &size);
53     Status SetAudioInterruptListener(const std::shared_ptr<AudioCaptureModuleCallback> &callback);
54     Status SetAudioCapturerInfoChangeCallback(
55         const std::shared_ptr<AudioStandard::AudioCapturerInfoChangeCallback> &callback);
56     Status GetCurrentCapturerChangeInfo(AudioStandard::AudioCapturerChangeInfo &changeInfo);
57     int32_t GetMaxAmplitude();
58     void SetAudioSource(AudioStandard::SourceType source);
59     void SetFaultEvent(const std::string &errMsg);
60     void SetFaultEvent(const std::string &errMsg, int32_t ret);
61     void SetCallingInfo(int32_t appUid, int32_t appPid, const std::string &bundleName, uint64_t instanceId);
62 
63 private:
64     Status DoDeinit();
65     bool AssignSampleRateIfSupported(const int32_t value);
66     bool AssignChannelNumIfSupported(const int32_t value);
67     bool AssignSampleFmtIfSupported(const Plugins::AudioSampleFormat value);
68 
69     void TrackMaxAmplitude(int16_t *data, int32_t size);
70 
71     Mutex captureMutex_ {};
72     std::unique_ptr<OHOS::AudioStandard::AudioCapturer> audioCapturer_ {nullptr};
73     std::shared_ptr<AudioStandard::AudioCapturerInfoChangeCallback> audioCapturerInfoChangeCallback_{nullptr};
74     AudioStandard::AudioCapturerOptions options_{};
75     std::shared_ptr<AudioCaptureModuleCallback> audioCaptureModuleCallback_ {nullptr};
76     std::shared_ptr<AudioStandard::AudioCapturerCallback> audioInterruptCallback_ {nullptr};
77     int64_t bitRate_ {0};
78     int32_t appTokenId_ {0};
79     int32_t appUid_ {0};
80     int32_t appPid_ {0};
81     int64_t appFullTokenId_ {0};
82     size_t bufferSize_ {0};
83     int32_t maxAmplitude_ {0};
84     bool isTrackMaxAmplitude {false};
85     std::string bundleName_;
86     uint64_t instanceId_{0};
87 };
88 } // namespace AudioCaptureModule
89 } // namespace Media
90 } // namespace OHOS
91 #endif // HISTREAMER_AUDIO_CAPTURE_MODULE_H
92 
93