protected function getElementData()
    {
        $arResult = [
            'category_id_list' => $this->obElement->category()->lists('id'),
            'product_id_list'  => $this->obElement->product()->lists('id'),
        ];

        if (PluginManager::instance()->hasPlugin('RainLab.Location')) {
            $arResult['country_id_list'] = $this->obElement->country()->lists('id');
            $arResult['state_id_list'] = $this->obElement->state()->lists('id');
        }

        return $arResult;
    }