zPowerBaseVariation.cs

using System;
using System.Numerics;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Model;
 
namespace Waher.Script.Fractals.IFS.Variations.ComplexVariations
{
    public class ZPowerBaseVariation : FlameVariationOneComplexParameter
    {
        public ZPowerBaseVariation(ScriptNode Parameter1, ScriptNode Parameter2, int Start, int Length, Expression Expression)
            : base(Parameter1, Parameter2, Start, Length, Expression)
        {
        }
 
        public ZPowerBaseVariation(ScriptNode Parameter1, int Start, int Length, Expression Expression)
            : base(Parameter1, null, Start, Length, Expression)
        {
        }
 
        private ZPowerBaseVariation(Complex z, ScriptNode Parameter, int Start, int Length, Expression Expression)
            : base(z, Parameter, Start, Length, Expression)
        {
        }
 
        private ZPowerBaseVariation(double Re, double Im, ScriptNode Parameter1, ScriptNode Parameter2,
            int Start, int Length, Expression Expression)
            : base(Re, Im, Parameter1, Parameter2, Start, Length, Expression)
        {
        }
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            if (Arguments[1] is null || Arguments[1].AssociatedObjectValue is null)
                return new ZPowerBaseVariation(Expression.ToComplex(Arguments[0].AssociatedObjectValue), this.Arguments[0], this.Start, this.Length, this.Expression);
            else
            {
                return new ZPowerBaseVariation(Expression.ToDouble(Arguments[0].AssociatedObjectValue), Expression.ToDouble(Arguments[1].AssociatedObjectValue),
                    this.Arguments[0], this.Arguments[1], this.Start, this.Length, this.Expression);
            }
        }
 
        public override void Operate(ref double x, ref double y)
        {
            // z^(x+iy)
 
            double lnz = Math.Log(Math.Sqrt(this.re * this.re + this.im * this.im));
            double argz = Math.Atan2(this.im, this.re);
            double amp = Math.Exp(x * lnz - y * argz);
            double phi = y * lnz + x * argz;
 
            x = amp * Math.Cos(phi);
            y = amp * Math.Sin(phi);
        }
 
        public override string FunctionName => nameof(ZPowerBaseVariation);
    }
}