package com.javarush.test.level08.lesson11.home02;

import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

/* Множество всех животных
1. Внутри класса Solution создать public static классы Cat, Dog.
2. Реализовать метод createCats, котороый должен возвращать множество с 4 котами.
3. Реализовать метод createDogs, котороый должен возвращать множество с 3 собаками.
4. Реализовать метод join, котороый должен возвращать объединенное множество всех животных - всех котов и собак.

5. Реализовать метод removeCats, котороый должен удалять из множества pets всех котов, которые есть в множестве cats.

6. Реализовать метод printPets, котороый должен выводить на экран всех животных, которые в нем есть.
   Каждое животное с новой строки
*/

public class Solution
{
    public static void main(String[] args)
    {
        Set<Cat> cats = createCats();
        Set<Dog> dogs = createDogs();

        Set<Object> pets = join(cats, dogs);

        printPets(pets);

        removeCats(pets, cats);
        printPets(pets);
    }

    public static Set<Cat> createCats()
    {
        HashSet<Cat> result = new HashSet<Cat>();

        //Написать тут ваш код
        for (int i =0; i<4; i++) {
            result.add(new Cat());
        }
        return result;
    }

    public static Set<Dog> createDogs()
    {
        //Написать тут ваш код
        HashSet<Dog> result = new HashSet<Dog>();
        for (int i1 =0; i1<3; i1++) {
            result.add(new Dog());
        }
        return result;
    }

    public static Set<Object> join(Set<Cat> cats, Set<Dog> dogs)
    {
        //Написать тут ваш код
        Set<Object> myPets = new HashSet<Object>();
        myPets.addAll(cats);
        myPets.addAll(dogs);
        return myPets;
    }

    public static void removeCats(Set<Object> pets, Set<Cat> cats)
    {
        //Написать тут ваш код
/*        for (Cat x : cats) {
            pets.remove(x);
        }*/
        pets.removeAll(cats);
        //System.out.println("Pets " + pets.size());
    }

    public static void printPets(Set<Object> pets)
    {
        //Написать тут ваш код
        //System.out.println("pringPets starts");
        Iterator<Object> myCatsAndDogs = pets.iterator();
        while (myCatsAndDogs.hasNext()) {
            System.out.println(myCatsAndDogs.next());
        }

    }

    //Написать тут ваш код
    public static class Cat {

    }

    public static class Dog {

    }
}