package com.eturkis.flattener;

import java.util.ArrayList;
import java.util.List;

/*
 * This class contains the methods for flattening an arbitrary array.
 *
 * @author : Emre TURKIS
 * created on : 26/11/2015
 */
public class ArrayFlattener implements IFlattenArray{
    @Override
    public Object[] flatten(Object[] arbitraryArray) {
        List flattenedList = new ArrayList();

        for(Object arrayElement : arbitraryArray){
            if(arrayElement instanceof Integer)
                flattenedList.add(arrayElement);
            else if (arrayElement instanceof Object[])
                addSubElements(flattenedList, (Object[]) arrayElement);
        }

        return flattenedList.toArray();
    }

    private void addSubElements(List flattenedList, Object[] arrayElement) {
        Object[] flattenedArray = flatten(arrayElement);
        for(Object o : flattenedArray){
            flattenedList.add(o);
        }
    }
}