function display_element( $element, &$children_elements, $max_depth, $depth = 0, $args, &$output ) {
static $mainLevelCounter;
if ($depth == 0) {
- ++$mainLevelCounter;
+ $mainLevelCounter++;
}
$element->has_children = ! empty( $children_elements[ $element->ID ] );
$element->classes[] = ( $element->current || $element->current_item_ancestor ) ? 'active' : '';
$element->classes[] = ( $element->has_children && 1 !== $max_depth ) ? 'has-dropdown' : '';
- $element->classes[] = ( $element->post_parent == 0 && ($mainLevelCounter < 6 || $mainLevelCounter != 1 )) ? '' : 'drop-left';
- $element->classes[] = ( $element->post_parent == 0 && ($mainLevelCounter < 6 || $mainLevelCounter != 1 )) ? '' : $mainLevelCounter;
+ $element->classes[] = ( $element->post_parent == 0 && ($mainLevelCounter < 6 )) ? '' : 'drop-left';
+ $element->classes[] = ( $element->post_parent == 0 && ($mainLevelCounter < 6 || $mainLevelCounter == 1 || $mainLevelCounter == 2 )) ? '' : 'TEST123';
+ $element->classes[] = ( $element->post_parent == 0 && ($mainLevelCounter < 6 )) ? '' : $mainLevelCounter;
parent::display_element( $element, $children_elements, $max_depth, $depth, $args, $output );
}