neural-amp-modeler-lv2/src/nam_plugin.h

#pragma once

#include <array>
#include <cstddef>
#include <cstdint>
#include <random>
#include <string_view>

// LV2
#include <lv2/core/lv2.h>
#include <lv2/atom/atom.h>
#include <lv2/log/log.h>
#include <lv2/log/logger.h>
#include <lv2/urid/urid.h>
#include <lv2/atom/forge.h>
#include <lv2/patch/patch.h>
#include <lv2/worker/worker.h>

#include "dsp.h"

#define PlUGIN_URI "http://github.com/mikeoliphant/neural-amp-modeler-lv2"
#define MODEL_URI PlUGIN_URI "#model"

namespace NAM {

	enum LV2WorkType {
		kWorkTypeLoad,
		kWorkTypeSwitch
	};

	struct LV2LoadModelMsg {
		LV2WorkType type;
		char path[1024];
	};

	class Plugin {
	public:
		struct Ports {
			const LV2_Atom_Sequence* control;
			LV2_Atom_Sequence* notify;
			const float* audio_in;
			float* audio_out;
			float* input_level;
			float* output_level;
		};

		Ports ports = {};

		LV2_URID_Map* map;
		LV2_Log_Logger logger;
		LV2_Worker_Schedule* schedule;

		std::unique_ptr<::DSP> currentModel;
		std::unique_ptr<::DSP> stagedModel;
		std::unique_ptr<::DSP> deleteModel;

		std::unordered_map<std::string, double> mNAMParams = {};

		Plugin();
		~Plugin() = default;

		bool initialize(double rate, const LV2_Feature* const* features) noexcept;
		void process(uint32_t n_samples) noexcept;


		static LV2_Worker_Status work(LV2_Handle instance, LV2_Worker_Respond_Function respond, LV2_Worker_Respond_Handle handle,
			uint32_t size, const void* data);
		static LV2_Worker_Status work_response(LV2_Handle instance, uint32_t size, const void* data);

	private:
		struct URIs {
			LV2_URID atom_Object;
			LV2_URID atom_Float;
			LV2_URID atom_Int;
			LV2_URID atom_Path;
			LV2_URID atom_URID;
			LV2_URID patch_Set;
			LV2_URID patch_property;
			LV2_URID patch_value;
			LV2_URID model_Path;
		};

		URIs uris = {};
		LV2_Atom_Forge atom_forge = {};

		std::vector<double> dblData;

		float m_rate;
	};
}