/*
Copyright (C) 2005 The Pentagram team
Copyright (C) 2010-2022 The Exult team

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
*/

#ifndef AUDIOCHANNEL_H_INCLUDED
#define AUDIOCHANNEL_H_INCLUDED

#include "AudioSample.h"

namespace Pentagram {

	class AudioChannel {
		// We have:
		// 1x decompressor size
		// 2x frame size
		std::unique_ptr<uint8[]> playdata;
		void*                    decomp = nullptr;
		uint8*                   frames[2]{};
		size_t                   playdata_size = 0;

		uint32 sample_rate;
		bool   stereo;

		sint32       loop   = 0;
		AudioSample* sample = nullptr;

		// Info for sampling
		uint32 frame_evenodd = 0;    // which buffer is 'frame0'
		uint32 frame0_size   = 0;    // Size of the frame0 buffer in samples
		uint32 frame1_size   = 0;    // Size of the frame1 buffer in samples
		uint32 position      = 0;    // Position in frame0 buffer in source samples
		uint32 overall_position = 0;    // Overall position of playback
		int    lvol, rvol;           // 0-256
		int    distance;             // 0 - 256
		int    balance;              // -256 - 256
		uint32 pitch_shift;          // 0x10000 = no shift
		int    priority;             // anything.
		bool   paused = false;       // true/false

		sint32 instance_id = -1;    // Unique id for this channel

	public:
		AudioChannel(uint32 sample_rate, bool stereo);
		~AudioChannel();
		AudioChannel(const AudioChannel&)            = delete;
		AudioChannel(AudioChannel&&)                 = default;
		AudioChannel& operator=(const AudioChannel&) = delete;
		AudioChannel& operator=(AudioChannel&&)      = default;

		void stop() {
			if (sample) {
				if (playdata) {
					sample->freeDecompressor(decomp);
				}
				sample->Release();
				sample = nullptr;
			}
		}

		void playSample(
				AudioSample* sample, int loop, int priority, bool paused,
				uint32 pitch_shift, int lvol, int rvol, sint32 instance_id);
		void resampleAndMix(sint16* stream, uint32 bytes);

		bool isPlaying() const {
			return sample != nullptr;
		}

		void setPitchShift(int pitch_shift_) {
			pitch_shift = pitch_shift_;
		}

		uint32 getPitchShift() const {
			return pitch_shift;
		}

		void setLoop(int loop_) {
			loop = loop_;
		}

		sint32 getLoop() const {
			return loop;
		}

		uint32 getPlaybackLength() const;

		uint32 getPlaybackPosition() const;

		void setVolume(int lvol_, int rvol_) {
			lvol = lvol_;
			rvol = rvol_;
		}

		void getVolume(int& lvol_, int& rvol_) const {
			lvol_ = lvol;
			rvol_ = rvol;
		}

		void setPriority(int priority_) {
			priority = priority_;
		}

		int getPriority() const {
			return priority;
		}

		void setPaused(bool paused_) {
			paused = paused_;
		}

		bool isPaused() const {
			return paused;
		}

		void set2DPosition(int distance_, int balance_) {
			distance = distance_;
			balance  = balance_;
		}

		void get2DPosition(int& distance_, int& balance_) const {
			distance_ = distance;
			balance_  = balance;
		}

		void calculate2DVolume(int& lvol, int& rvol);

		AudioSample* getSample() const {
			return sample;
		}

		sint32 getInstanceId() const {
			return instance_id;
		}

	private:
		//
		void DecompressNextFrame();

		//
		// Resampling
		//
		class CubicInterpolator {
		protected:
			int x0 = 0, x1 = 0, x2 = 0, x3 = 0;
			int a, b, c, d;

		public:
			CubicInterpolator() {
				updateCoefficients();
			}

			CubicInterpolator(int a0, int a1, int a2, int a3)
					: x0(a0), x1(a1), x2(a2), x3(a3) {
				updateCoefficients();
			}

			CubicInterpolator(int a1, int a2, int a3)
					: x0(2 * a1 - a2), x1(a1), x2(a2), x3(a3) {
				// We use a simple linear interpolation for x0
				updateCoefficients();
			}

			inline void init(int a0, int a1, int a2, int a3) {
				x0 = a0;
				x1 = a1;
				x2 = a2;
				x3 = a3;
				updateCoefficients();
			}

			inline void init(int a1, int a2, int a3) {
				// We use a simple linear interpolation for x0
				x0 = 2 * a1 - a2;
				x1 = a1;
				x2 = a2;
				x3 = a3;
				updateCoefficients();
			}

			inline void feedData() {
				x0 = x1;
				x1 = x2;
				x2 = x3;
				x3 = 2 * x2 - x1;    // Simple linear interpolation
				updateCoefficients();
			}

			inline void feedData(int xNew) {
				x0 = x1;
				x1 = x2;
				x2 = x3;
				x3 = xNew;
				updateCoefficients();
			}

			/* t must be a 16.16 fixed point number between 0 and 1 */
			inline int interpolate(uint32 fp_pos) {
				int       result = 0;
				const int t      = fp_pos >> 8;
				result           = (a * t + b) >> 8;
				result           = (result * t + c) >> 8;
				result           = (result * t + d) >> 8;
				result           = (result / 3 + 1) >> 1;

				return result;
			}

		protected:
			inline void updateCoefficients() {
				a = ((-x0 * 2) + (x1 * 5) - (x2 * 4) + x3);
				b = ((x0 + x2 - (2 * x1)) * 6) << 8;
				c = ((-4 * x0) + x1 + (x2 * 4) - x3) << 8;
				d = (x1 * 6) << 8;
			}
		};

		// Resampler stuff
		CubicInterpolator interp_l;
		CubicInterpolator interp_r;
		int               fp_pos   = 0;
		int               fp_speed = 0;

		void resampleFrameM8toS(sint16*& stream, uint32& bytes);
		void resampleFrameM8toM(sint16*& stream, uint32& bytes);
		void resampleFrameS8toM(sint16*& stream, uint32& bytes);
		void resampleFrameS8toS(sint16*& stream, uint32& bytes);
		void resampleFrameM16toS(sint16*& stream, uint32& bytes);
		void resampleFrameM16toM(sint16*& stream, uint32& bytes);
		void resampleFrameS16toM(sint16*& stream, uint32& bytes);
		void resampleFrameS16toS(sint16*& stream, uint32& bytes);
	};

}    // namespace Pentagram

#endif
