diff --git a/classes/Visualizer/Module.php b/classes/Visualizer/Module.php index 77fd5af5..c9ad1f2d 100644 --- a/classes/Visualizer/Module.php +++ b/classes/Visualizer/Module.php @@ -67,7 +67,7 @@ public function __construct( Visualizer_Plugin $plugin ) { $this->_addFilter( Visualizer_Plugin::FILTER_UNDO_REVISIONS, 'undoRevisions', 10, 2 ); $this->_addFilter( Visualizer_Plugin::FILTER_HANDLE_REVISIONS, 'handleExistingRevisions', 10, 2 ); $this->_addFilter( Visualizer_Plugin::FILTER_GET_CHART_DATA_AS, 'getDataAs', 10, 3 ); - $this->_addFilter( Visualizer_Plugin::FILTER_CHART_TITLE, 'filterChartTitle', 10, 2 ); + $this->_addAction( 'pre_get_posts', 'PreGetPosts' ); register_shutdown_function( array($this, 'onShutdown') ); } @@ -793,6 +793,21 @@ public function _getImage( $chart = null ) { ); } + /** + * Filter chart title if visualizer post type. + * + * @param object $query WP Query object. + * @return void + */ + public function PreGetPosts( $query ) { + if ( ! $query->is_main_query() ) { + $post_type = $query->get( 'post_type' ); + if ( 'visualizer' === $post_type ) { + $this->_addFilter( Visualizer_Plugin::FILTER_CHART_TITLE, 'filterChartTitle', 10, 2 ); + } + } + } + /** * Filter chart title. *