ModelElementLists.scala

package de.tud.swt.rms

import scala.collection.immutable.Set
import de.tud.swt.rms.models.modelB.Family
import de.tud.swt.rms.models.modelB.Member
import de.tud.swt.rms.models.modelA.Person
import de.tud.swt.rms.models.modelC.SimplePerson

/**
  * Helper object to hold instances of all model elements.
  */
object ModelElementLists {

  private var families = Set.empty[Family]
  private var members = Set.empty[Member]
  private var persons = Set.empty[Person]
  private var registers = Set.empty[SimplePerson]

  /**
    * Add a new element to the lists.
    */
  def addElement(obj: Object) {
    if (obj == null)
      return
    if (obj.isInstanceOf[Family])
      this.addFamily(obj.asInstanceOf[Family])
    if (obj.isInstanceOf[Member])
      this.addMember(obj.asInstanceOf[Member])
    if (obj.isInstanceOf[Person])
      this.addPersons(obj.asInstanceOf[Person])
    if (obj.isInstanceOf[SimplePerson])
      this.addRegister(obj.asInstanceOf[SimplePerson])
  }

  private def addFamily(f: Family): Unit = {
    families += f
  }

  private def addMember(m: Member): Unit = {
    members += m
  }

  private def addPersons(p: Person): Unit = {
    persons += p
  }

  private def addRegister(r: SimplePerson): Unit = {
    registers += r
  }

  def setAllFamilyNames(s: String): Unit = {
    families.foreach {
      m => m.setLastName(s) //.setCompleteName(s)
    }
  }

  def setAllPersonNames(s: String): Unit = {
    persons.foreach {
      m => m.setFullName(s)
    }
  }

  def changeRegisterNames(): Unit = {
    registers.foreach {
      m => m.changeFirstAndLastName()
    }
  }

  def setAllRegisterNames(s: String): Unit = {
    registers.foreach {
      m => m.setCompleteName(s)
    }
  }

  /**
    * Print all elements from all lists.
    */
  def printALL(): Unit = {
    println("************************************************")
    println("Members: ");
    members.foreach {
      m => println("**Member: " + m);
    }
    println("Families: ");
    families.foreach {
      m => println("**Family: " + m);
    }
    println("Persons: ");
    persons.foreach {
      m => println("**Person: " + m);
    }
    println("Registers: ");
    registers.foreach {
      m => println("**Register: " + m);
    }
    println("************************************************")
  }
}