Определение значения из функции для шорткода

Всем добрый день.
Подскажите пожалуйста, всю голову уже сломал переделыванием плагина woocommerce...
Для получения категории/подкатегории и всех данных в коде вызываю шорткод:

Код (Text):

1. <div class="woo_shop_cat"><?php echo do_shortcode('[product_categories columns="6" number="0" parent="0"]'); ?></div>

Использовать шорткод полноценно не могу, так как при переходе из категории в подкатегорию на странице по прежнему отображается категория.

Нашел функцию, которая полноценно справляется с задачей:

Код (Text):

1. <div class="woo_shop_cat">
2. <?php
3. function mynew_product_subcategories( $args = array() ) {
4.     $parentid = get_queried_object_id();
5.     $args = array(
6.         'parent' => $parentid
7.     );
8.     $terms = get_terms( 'product_cat', $args );
9.     if ( $terms ) {
10.         echo '<ul class="product-cats">';
11.             foreach ( $terms as $term ) {            
12.                 echo '<li class="category">';                      
13.                     woocommerce_subcategory_thumbnail( $term );
14.                         echo '<a href="' .  esc_url( get_term_link( $term ) ) . '" class="' . $term->slug . '">';
15.                             echo $term->name;
16.                         echo '</a>';                                                    
17.                 echo '</li>';                                                      
18.         }
19.         echo '</ul>';
20.     }
21. }
22. add_action( 'woocommerce_before_shop_loop', 'mynew_product_subcategories', 50 );
23. ?>
24. </div>

В функции происходит всё четко, при переходе отображаются подкатегории, но теряется вся вёрстка сайта и прописывать в CSS потребуется слишком много.

Каким образом можно соединить оба кода так, что бы при вызове функции определялся parent?

 

Функция плагина wordpress, которая выводит/отображает информацию.

Мне необходимо определять параметр parent