package main

// takeChildren recursively takes the children of the class.
func takeChildren(c *CppClass) []CppClass {
	if len(c.ChildClassdefs) == 0 {
		return []CppClass{}
	}

	var ret []CppClass
	for _, child := range c.ChildClassdefs {
		ret = append(ret, takeChildren(&child)...)

		child.ChildClassdefs = nil
		ret = append(ret, child)
	}
	c.ChildClassdefs = nil

	return ret
}

// astTransformChildClasses expands all classes with child class definitions
// into new top-level entries within the header.
func astTransformChildClasses(parsed *CppParsedHeader) {
	var taken []CppClass

	for i, _ := range parsed.Classes {
		taken = append(taken, takeChildren(&parsed.Classes[i])...)
	}

	parsed.Classes = append(parsed.Classes, taken...)
}