/**File RealOrComplex.scala, by KWR for CSE250, Spring 2022
   Illustrates setters and case classes and value typesaout of Ch. 3
   Resembles the text's Vector2D example but does different stuff
 */

import io.StdIn._
import io.Source
import scala.math._         //for cos(...) and sin(...)
import java.io.File
import java.io.FileWriter   //makes it easy to append
import java.io.PrintWriter  //makes "print" and "println" available

/** Complex numbers as re^{i.theta}
    Argument variables need type declarations
 */
class ComplexPolar(private var _r: Double, private var _theta: Double) {   
    def r = _r
    def theta = _theta
    def r_=(newMag: Double): Unit = if (newMag < 0.0) {_r = 0.0} else {_r = newMag}
    def theta_=(newTheta: Double): Unit = _theta = newTheta
    //makes other code in the class normalize 0 <= theta < 2pi if it wants to.
    def x = r*cos(theta)
    def y = r*sin(theta)
}


/** Complex numbers as re^{i.theta}
    Args are automatically values.  Still need type declarations.
 */
case class ComplexPolarVal(r: Double, theta: Double) {
    def x = r*cos(theta)
    def y = r*sin(theta)
}

case class RealVal(x: Double)

object RealOrComplex extends App {
   //val z = new ComplexPolar(1.0,Pi)
   //val z = ComplexPolar(1.0,Pi)
   val z = ComplexPolarVal(1.0,Pi)
   //z.r = 2.0
   println(s"The polar number $z has real part ${z.x} and complex part ${z.y}")

   
   //Return imaginary part of number
   //The allowability of "Any" here makes me very uncomfortable, IMPHO.  
   //
   def imagPart(w: AnyVal) = w match {
      //case RealVal => 0.0         //causes error: match not found (unless default is commented in)
      case RealVal(x) => 0.0
      //case ComplexPolarVal(z) => z.y    //error: both arguments needed
      case ComplexPolarVal(r,theta) => ComplexPolarVal(r,theta).y
      case _ => 0.0                     //default case---should we use it?
   }


   val u = RealVal(3.0)
   val v = ComplexPolarVal(1.0,Pi/2.0)
   val w = new ComplexPolar(1.0,Pi/4.0)
   println(s"The imaginary part of u is ${imagPart(u)} and of v is ${imagPart(v)}")
   println(s"Also the imag part of w is ${imagPart(w)}")
}