Условие для шаблон вывода wp_nav_menu(). Walker?

Приветствую Вас.
Помогите пожалуйста, как скорректировать шаблон вывода wp_nav_menu(), а именно возможно ли оборачивать вложенные в sub-menu элементы li в дополнительный div.wrapper, при проверке условия, если какой-то из элементов li имеет класс sub-rubric, то его самого и следующие за ним элементы li (до следующего li.sub-rubric) нужно обернуть в div.wrapper.

Немного подробнее.
Есть навигационное меню, состоящее из родительских и дочерних каталогов (таксономия catalog). Родительские каталоги — это элементы меню первого уровня, их дочерние — вложенные sub-menu.

Обратите внимание, вложенное sub-menu разделено на условные, назовем их подрубрики (li.sub-rubriс) — это произвольные ссылки, «ПО ЦЕНЕ», «ПО СТИЛЮ», «ПО НАЗНАЧЕНИЮ» и т.д.

Выводится меню в таком виде.

<ul class="menu">
  <li class="menu-item-has-children">
    <a href="http://xxx/kitchens/">Кухни</a>
    <ul class="sub-menu">

      <li class="sub-rubriс menu-item"><a href="#">ПО ЦЕНЕ</a></li>
      <li class="menu-item"><a href="/link/">Эконом</a></li>

      <li class="sub-rubriс menu-item"><a href="#">ПО СТИЛЮ</a></li>
      <li class="menu-item"><a href="/link/">Хай-тек</a></li>
      <li class="menu-item"><a href="/link/">Современные</a></li>
      <li class="menu-item"><a href="/link/">Классические</a></li>
      <li class="menu-item"><a href="/link/">Маленькие</a></li>

      <li class="sub-rubriс menu-item"><a href="#">ПО НАЗНАЧЕНИЮ</a></li>
      <li class="menu-item"><a href="/link/">С барной стойкой</a></li>
      <li class="menu-item"><a href="/link/">С островом</a></li>

      <li class="sub-rubriс menu-item"><a href="#">ПО ЦВЕТУ</a></li>
      <li class="menu-item"><a href="/link/">Зеленая</a></li>

      <li class="sub-rubriс menu-item"><a href="#">ПО МАТЕРИАЛУ</a></li>
      <li class="menu-item"><a href="/link/">Из пластика</a></li>
      <li class="menu-item"><a href="/link/">Из массива</a></li>

      <li class="sub-rubriс menu-item"><a href="#">ПО ФОРМЕ</a></li>
      <li class="menu-item"><a href="/link/">Угловые</a></li>
      <li class="menu-item"><a href="/link/">Прямые</a></li>
      <li class="menu-item"><a href="/link/">Радиусные</a></li>
    </ul>
  </li>
</ul>

Задача обернуть эти подрубрики каждую в свой дополнительный div.wrapper.
Чтобы получилось так:

<ul class="menu">
  <li class="menu-item-has-children">
    <a href="http://xxx/kitchens/">Кухни</a>
    <ul class="sub-menu">

      <div class="sub-rubrik-wrap"
        <li class="sub-rubrik menu-item"><a href="#">ПО ЦЕНЕ</a></li>
        <li class="menu-item"><a href="/link/">Эконом</a></li>
      </div>

      <div class="sub-rubrik-wrap"
        <li class="sub-rubrik menu-item"><a href="#">ПО СТИЛЮ</a></li>
        <li class="menu-item"><a href="/link/">Хай-тек</a></li>
        <li class="menu-item"><a href="/link/">Современные</a></li>
        <li class="menu-item"><a href="/link/">Классические</a></li>
        <li class="menu-item"><a href="/link/">Маленькие</a></li>
      </div>

      <div class="sub-rubrik-wrap"
        <li class="sub-rubrik menu-item"><a href="#">ПО НАЗНАЧЕНИЮ</a></li>
        <li class="menu-item"><a href="/link/">С барной стойкой</a></li>
        <li class="menu-item"><a href="/link/">С островом</a></li>
      </div>

      <div class="sub-rubrik-wrap"
        <li class="sub-rubrik menu-item"><a href="#">ПО ЦВЕТУ</a></li>
        <li class="menu-item"><a href="/link/">Зеленая</a></li>
      </div>

      <div class="sub-rubrik-wrap"
        <li class="sub-rubrik menu-item"><a href="#">ПО МАТЕРИАЛУ</a></li>
        <li class="menu-item"><a href="/link/">Из пластика</a></li>
        <li class="menu-item"><a href="/link/">Из массива</a></li>
      </div>

      <div class="sub-rubrik-wrap"
        <li class="sub-rubrik menu-item"><a href="#">ПО ФОРМЕ</a></li>
        <li class="menu-item"><a href="/link/">Угловые</a></li>
        <li class="menu-item"><a href="/link/">Прямые</a></li>
        <li class="menu-item"><a href="/link/">Радиусные</a></li>
      </div>
    </ul>
  </li>
</ul>

Думаю для этого нужно корректировать шаблон Walker, но опыта сделать это самому не достаточно, помогите пожалуйста.
Спасибо!