#region License
/* 
 * Copyright (C) 1999-2024 John Källén.
 *
 * 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, 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; see the file COPYING.  If not, write to
 * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
 */
#endregion

using NUnit.Framework;
using Reko.Arch.RiscV;
using Reko.Core.Machine;
using Reko.Core.Types;
using System;
using System.Collections.Generic;
using System.ComponentModel.Design;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace Reko.UnitTests.Arch.RiscV
{
    [TestFixture]
    public class RiscVCallingConventionTests
    {
        private static PrimitiveType i32 = PrimitiveType.Int32;
        private static PrimitiveType i64 = PrimitiveType.Int64;
        private static PrimitiveType w128 = PrimitiveType.Word128;
        private static PrimitiveType r32 = PrimitiveType.Real32;
        private static PrimitiveType r64 = PrimitiveType.Real64;

        private RiscVArchitecture arch;
        private RiscVCallingConvention cc;
        private ICallingConventionEmitter ccr;

        [SetUp]
        public void Setup()
        {
            Given_ArchWithFloatAbi(64);
        }

        private void Given_ArchWithFloatAbi(int floatAbi)
        { 
            this.arch = new RiscVArchitecture(
                new ServiceContainer(),
                "riscV",
                new Dictionary<string, object>
                {
                    { "FloatAbi", floatAbi }
                });
            this.cc = new RiscVCallingConvention(arch);
            this.ccr = new CallingConventionEmitter();
        }

        [Test]
        public void RiscVCc_SingleArg()
        {
            cc.Generate(ccr, 0, null, null, new List<DataType> { i32 });
            Assert.AreEqual("Stk: 0 void (a0)", ccr.ToString());
        }

        [Test]
        public void RiscVCc_ReturnInt()
        {
            cc.Generate(ccr, 0, i64, null, new List<DataType> { r64 });
            Assert.AreEqual("Stk: 0 a0 (fa0)", ccr.ToString());
        }

        [Test]
        public void RiscVCc_ReturnLongLong()
        {
            cc.Generate(ccr, 0, r64, null, new List<DataType> { i32, w128 });
            Assert.AreEqual("Stk: 0 fa0 (a0, Sequence a3:a2)", ccr.ToString());
        }

        [Test]
        public void RiscVcc_ArgsOnStack()
        {
            cc.Generate(ccr, 0, null, null,
                new List<DataType> { w128, w128, w128, i32, w128, r64 });

            Assert.AreEqual(
                "Stk: 0 void (Sequence a1:a0, " +
                             "Sequence a3:a2, " +
                             "Sequence a5:a4, " +
                             "a6, " +
                             "Stack +0000, " +
                             "Stack +0010)",
                ccr.ToString());
        }

        [Test]
        public void RiscVcc_MixedRegs()
        {
            cc.Generate(ccr, 0, null, null,
                new List<DataType> { r64, i32, r32, i64, r64, i32, r32, i64, r32, i32 });
            Assert.AreEqual(
                "Stk: 0 void (fa0, a1, fa2, a3, fa4, a5, fa6, a7, Stack +0000, Stack +0008)",
                ccr.ToString());
        }

        [Test]
        public void RiscVCc_RealArgsReturn_NoFpuSupport()
        {
            Given_ArchWithFloatAbi(0);  // No FPU support.
            cc.Generate(ccr, 0, r64, null, new List<DataType> { r64 });
            Assert.AreEqual("Stk: 0 a0 (a0)", ccr.ToString());
        }

    }
}
