/*
Copyright (c) 2018 tevador

This file is part of RandomX.

RandomX 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 3 of the License, or
(at your option) any later version.

RandomX 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 RandomX.  If not, see<http://www.gnu.org/licenses/>.
*/

#pragma once

#include <cstdint>
#include <iostream>

namespace RandomX {

	using addr_t = uint32_t;

	constexpr int RoundToNearest = 0;
	constexpr int RoundDown = 1;
	constexpr int RoundUp = 2;
	constexpr int RoundToZero = 3;

	constexpr int SeedSize = 32;
	constexpr int ResultSize = 32;

	constexpr int ArgonIterations = 3;
	constexpr uint32_t ArgonMemorySize = 262144; //KiB
	constexpr int ArgonLanes = 1;
	const char ArgonSalt[] = "Monero\x1A$";
	constexpr int ArgonSaltSize = sizeof(ArgonSalt) - 1;

	constexpr int CacheLineSize = 64;
	constexpr uint64_t DatasetSize = 4ULL * 1024 * 1024 * 1024; //4 GiB
	constexpr uint32_t CacheSize = ArgonMemorySize * 1024;
	constexpr int CacheBlockCount = CacheSize / CacheLineSize;
	constexpr int BlockExpansionRatio = DatasetSize / CacheSize;
	constexpr int DatasetBlockCount = BlockExpansionRatio * CacheBlockCount;
	constexpr int DatasetIterations = 3;


#ifdef TRACE
	constexpr bool trace = true;
#else
	constexpr bool trace = false;
#endif

	union convertible_t {
		double f64;
		int64_t i64;
		uint64_t u64;
		int32_t i32;
		uint32_t u32;
		struct {
			int32_t i32lo;
			int32_t i32hi;
		};
	};

	struct fpu_reg_t {
		convertible_t lo;
		convertible_t hi;
	};

	constexpr int ProgramLength = 128;
	constexpr uint32_t InstructionCount = 1024;
	constexpr uint32_t ScratchpadSize = 1024 * 1024;
	constexpr uint32_t ScratchpadLength = ScratchpadSize / sizeof(convertible_t);
	constexpr uint32_t ScratchpadL1 = ScratchpadSize / 64 / sizeof(convertible_t);
	constexpr uint32_t ScratchpadL2 = ScratchpadSize / 4 / sizeof(convertible_t);
	constexpr uint32_t ScratchpadL3 = ScratchpadSize / sizeof(convertible_t);
	constexpr int ScratchpadL1Mask = (ScratchpadL1 - 1) * 8;
	constexpr int ScratchpadL2Mask = (ScratchpadL2 - 1) * 8;
	constexpr int ScratchpadL1Mask16 = (ScratchpadL1 / 2 - 1) * 16;
	constexpr int ScratchpadL2Mask16 = (ScratchpadL2 / 2 - 1) * 16;
	constexpr uint32_t TransformationCount = 90;
	constexpr int RegistersCount = 8;

	class Cache;

	inline int wrapInstr(int i) {
		return i % RandomX::ProgramLength;
	}

	class ILightClientAsyncWorker {
	public:
		virtual void prepareBlock(addr_t) = 0;
		virtual void prepareBlocks(void* out, uint32_t startBlock, uint32_t blockCount) = 0;
		virtual const uint64_t* getBlock(addr_t) = 0;
		virtual void getBlocks(void* out, uint32_t startBlock, uint32_t blockCount) = 0;
		virtual void sync() = 0;
		const Cache* getCache() {
			return cache;
		}
	protected:
		ILightClientAsyncWorker(const Cache* c) : cache(c) {}
		const Cache* cache;
	};

	union dataset_t {
		uint8_t* dataset;
		Cache* cache;
		ILightClientAsyncWorker* asyncWorker;
	};

	struct MemoryRegisters {
		addr_t mx, ma;
		dataset_t ds;
	};

	static_assert(sizeof(MemoryRegisters) == 2 * sizeof(addr_t) + sizeof(uintptr_t), "Invalid alignment of struct RandomX::MemoryRegisters");

	struct RegisterFile {
		convertible_t r[RegistersCount];
		fpu_reg_t f[RegistersCount / 2];
		fpu_reg_t g[RegistersCount / 2];
		fpu_reg_t a[RegistersCount / 2];
	};

	static_assert(sizeof(RegisterFile) == 256, "Invalid alignment of struct RandomX::RegisterFile");

	typedef void(*DatasetReadFunc)(addr_t, MemoryRegisters&, RegisterFile&);

	typedef void(*ProgramFunc)(RegisterFile&, MemoryRegisters&, convertible_t*, uint64_t);

	extern "C" {
		void executeProgram(RegisterFile&, MemoryRegisters&, convertible_t*, uint64_t);
	}
}

std::ostream& operator<<(std::ostream& os, const RandomX::RegisterFile& rf);