| 1 | /**************************************************************************/ |
| 2 | /* audio_stream_wav.h */ |
| 3 | /**************************************************************************/ |
| 4 | /* This file is part of: */ |
| 5 | /* GODOT ENGINE */ |
| 6 | /* https://godotengine.org */ |
| 7 | /**************************************************************************/ |
| 8 | /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */ |
| 9 | /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */ |
| 10 | /* */ |
| 11 | /* Permission is hereby granted, free of charge, to any person obtaining */ |
| 12 | /* a copy of this software and associated documentation files (the */ |
| 13 | /* "Software"), to deal in the Software without restriction, including */ |
| 14 | /* without limitation the rights to use, copy, modify, merge, publish, */ |
| 15 | /* distribute, sublicense, and/or sell copies of the Software, and to */ |
| 16 | /* permit persons to whom the Software is furnished to do so, subject to */ |
| 17 | /* the following conditions: */ |
| 18 | /* */ |
| 19 | /* The above copyright notice and this permission notice shall be */ |
| 20 | /* included in all copies or substantial portions of the Software. */ |
| 21 | /* */ |
| 22 | /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ |
| 23 | /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ |
| 24 | /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */ |
| 25 | /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ |
| 26 | /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ |
| 27 | /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ |
| 28 | /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ |
| 29 | /**************************************************************************/ |
| 30 | |
| 31 | #ifndef AUDIO_STREAM_WAV_H |
| 32 | #define AUDIO_STREAM_WAV_H |
| 33 | |
| 34 | #include "servers/audio/audio_stream.h" |
| 35 | |
| 36 | class AudioStreamWAV; |
| 37 | |
| 38 | class AudioStreamPlaybackWAV : public AudioStreamPlayback { |
| 39 | GDCLASS(AudioStreamPlaybackWAV, AudioStreamPlayback); |
| 40 | enum { |
| 41 | MIX_FRAC_BITS = 13, |
| 42 | MIX_FRAC_LEN = (1 << MIX_FRAC_BITS), |
| 43 | MIX_FRAC_MASK = MIX_FRAC_LEN - 1, |
| 44 | }; |
| 45 | |
| 46 | struct IMA_ADPCM_State { |
| 47 | int16_t step_index = 0; |
| 48 | int32_t predictor = 0; |
| 49 | /* values at loop point */ |
| 50 | int16_t loop_step_index = 0; |
| 51 | int32_t loop_predictor = 0; |
| 52 | int32_t last_nibble = 0; |
| 53 | int32_t loop_pos = 0; |
| 54 | int32_t window_ofs = 0; |
| 55 | } ima_adpcm[2]; |
| 56 | |
| 57 | int64_t offset = 0; |
| 58 | int sign = 1; |
| 59 | bool active = false; |
| 60 | friend class AudioStreamWAV; |
| 61 | Ref<AudioStreamWAV> base; |
| 62 | |
| 63 | template <class Depth, bool is_stereo, bool is_ima_adpcm> |
| 64 | void do_resample(const Depth *p_src, AudioFrame *p_dst, int64_t &p_offset, int32_t &p_increment, uint32_t p_amount, IMA_ADPCM_State *p_ima_adpcm); |
| 65 | |
| 66 | public: |
| 67 | virtual void start(double p_from_pos = 0.0) override; |
| 68 | virtual void stop() override; |
| 69 | virtual bool is_playing() const override; |
| 70 | |
| 71 | virtual int get_loop_count() const override; //times it looped |
| 72 | |
| 73 | virtual double get_playback_position() const override; |
| 74 | virtual void seek(double p_time) override; |
| 75 | |
| 76 | virtual int mix(AudioFrame *p_buffer, float p_rate_scale, int p_frames) override; |
| 77 | |
| 78 | virtual void tag_used_streams() override; |
| 79 | |
| 80 | AudioStreamPlaybackWAV(); |
| 81 | }; |
| 82 | |
| 83 | class AudioStreamWAV : public AudioStream { |
| 84 | GDCLASS(AudioStreamWAV, AudioStream); |
| 85 | RES_BASE_EXTENSION("sample" ) |
| 86 | |
| 87 | public: |
| 88 | enum Format { |
| 89 | FORMAT_8_BITS, |
| 90 | FORMAT_16_BITS, |
| 91 | FORMAT_IMA_ADPCM |
| 92 | }; |
| 93 | |
| 94 | // Keep the ResourceImporterWAV `edit/loop_mode` enum hint in sync with these options. |
| 95 | enum LoopMode { |
| 96 | LOOP_DISABLED, |
| 97 | LOOP_FORWARD, |
| 98 | LOOP_PINGPONG, |
| 99 | LOOP_BACKWARD |
| 100 | }; |
| 101 | |
| 102 | private: |
| 103 | friend class AudioStreamPlaybackWAV; |
| 104 | |
| 105 | enum { |
| 106 | DATA_PAD = 16 //padding for interpolation |
| 107 | }; |
| 108 | |
| 109 | Format format = FORMAT_8_BITS; |
| 110 | LoopMode loop_mode = LOOP_DISABLED; |
| 111 | bool stereo = false; |
| 112 | int loop_begin = 0; |
| 113 | int loop_end = 0; |
| 114 | int mix_rate = 44100; |
| 115 | void *data = nullptr; |
| 116 | uint32_t data_bytes = 0; |
| 117 | |
| 118 | protected: |
| 119 | static void _bind_methods(); |
| 120 | |
| 121 | public: |
| 122 | void set_format(Format p_format); |
| 123 | Format get_format() const; |
| 124 | |
| 125 | void set_loop_mode(LoopMode p_loop_mode); |
| 126 | LoopMode get_loop_mode() const; |
| 127 | |
| 128 | void set_loop_begin(int p_frame); |
| 129 | int get_loop_begin() const; |
| 130 | |
| 131 | void set_loop_end(int p_frame); |
| 132 | int get_loop_end() const; |
| 133 | |
| 134 | void set_mix_rate(int p_hz); |
| 135 | int get_mix_rate() const; |
| 136 | |
| 137 | void set_stereo(bool p_enable); |
| 138 | bool is_stereo() const; |
| 139 | |
| 140 | virtual double get_length() const override; //if supported, otherwise return 0 |
| 141 | |
| 142 | virtual bool is_monophonic() const override; |
| 143 | |
| 144 | void set_data(const Vector<uint8_t> &p_data); |
| 145 | Vector<uint8_t> get_data() const; |
| 146 | |
| 147 | Error save_to_wav(const String &p_path); |
| 148 | |
| 149 | virtual Ref<AudioStreamPlayback> instantiate_playback() override; |
| 150 | virtual String get_stream_name() const override; |
| 151 | |
| 152 | AudioStreamWAV(); |
| 153 | ~AudioStreamWAV(); |
| 154 | }; |
| 155 | |
| 156 | VARIANT_ENUM_CAST(AudioStreamWAV::Format) |
| 157 | VARIANT_ENUM_CAST(AudioStreamWAV::LoopMode) |
| 158 | |
| 159 | #endif // AUDIO_STREAM_WAV_H |
| 160 | |