PluginProbe
Visualizer – Tables & Charts Manager with Built-in AI Generator / 3.10.8
Visualizer – Tables & Charts Manager with Built-in AI Generator v3.10.8
4.0.7 4.0.6 4.0.5 4.0.4 4.0.3 3.0.5 3.0.6 3.0.7 3.0.8 3.0.9 3.1.0 3.1.1 3.1.2 3.1.3 3.10.0 3.10.1 3.10.10 3.10.11 3.10.12 3.10.13 3.10.14 3.10.15 3.10.2 3.10.3 3.10.4 All 148 releases
← All changes | classes/Visualizer/Module.php +323 -36 3.1.33.10.8 View file →
@@ -65,12 +65,29 @@
65 65 $this->_plugin = $plugin;
66 66
67 67 $this->_addFilter( Visualizer_Plugin::FILTER_UNDO_REVISIONS, 'undoRevisions', 10, 2 );
68 68 $this->_addFilter( Visualizer_Plugin::FILTER_HANDLE_REVISIONS, 'handleExistingRevisions', 10, 2 );
69 + $this->_addFilter( Visualizer_Plugin::FILTER_GET_CHART_DATA_AS, 'getDataAs', 10, 3 );
70 + $this->_addAction( 'pre_get_posts', 'PreGetPosts' );
71 + register_shutdown_function( array($this, 'onShutdown') );
69 72
70 73 }
71 74
72 75 /**
76 + * Register a shutdown hook to catch fatal errors.
77 + *
78 + * @since ?
79 + *
80 + * @access public
81 + */
82 + public function onShutdown() {
83 + $error = error_get_last();
84 + if ( $error && $error['type'] === E_ERROR && false !== strpos( $error['file'], 'Visualizer/' ) ) {
85 + do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'Critical error %s', print_r( $error, true ) ), 'error', __FILE__, __LINE__ );
86 + }
87 + }
88 +
89 + /**
73 90 * Registers an action hook.
74 91 *
75 92 * @since 1.0.0
76 93 * @uses add_action() To register action hook.
@@ -147,8 +164,17 @@
147 164 return $this;
148 165 }
149 166
150 167 /**
168 + * A wrapper around the actual function _getDataAs. This function is invoked as a filter.
169 + *
170 + * @since 3.2.0
171 + */
172 + public function getDataAs( $final, $chart_id, $type ) {
173 + return $this->_getDataAs( $chart_id, $type );
174 + }
175 +
176 + /**
151 177 * Extracts the data for a chart and prepares it for the given type.
152 178 *
153 179 * @access public
154 180 * @param int $chart_id The chart id.
@@ -158,15 +184,15 @@
158 184 $final = null;
159 185 $success = false;
160 186 if ( $chart_id ) {
161 187 $chart = get_post( $chart_id );
162 - $success = $chart && $chart->post_type == Visualizer_Plugin::CPT_VISUALIZER;
188 + $success = $chart && $chart->post_type === Visualizer_Plugin::CPT_VISUALIZER;
163 189 }
164 190 if ( $success ) {
165 191 $settings = get_post_meta( $chart_id, Visualizer_Plugin::CF_SETTINGS, true );
166 192 $rows = array();
167 193 $series = get_post_meta( $chart_id, Visualizer_Plugin::CF_SERIES, true );
168 - $data = unserialize( $chart->post_content );
194 + $data = self::get_chart_data( $chart, $type, false );
169 195 if ( ! empty( $series ) ) {
170 196 $row = array();
171 197 foreach ( $series as $array ) {
172 198 $row[] = $array['label'];
@@ -195,14 +221,29 @@
195 221 }
196 222 }
197 223 }
198 224
199 - $filename = isset( $settings['title'] ) && ! empty( $settings['title'] ) ? $settings['title'] : 'visualizer#' . $chart_id;
225 + $title = 'visualizer#' . $chart_id;
226 + if ( ! empty( $settings['title'] ) ) {
227 + $title = $settings['title'];
228 + }
229 + // for ChartJS, title is an array.
230 + if ( is_array( $title ) && isset( $title['text'] ) ) {
231 + $title = $title['text'];
232 + }
233 + if ( empty( $title ) ) {
234 + $title = 'visualizer#' . $chart_id;
235 + }
200 236
237 + $filename = $title;
238 +
201 239 switch ( $type ) {
202 240 case 'csv':
203 - $final = $this->_getCSV( $rows, $filename );
241 + $final = $this->_getCSV( $rows, $filename, false );
204 242 break;
243 + case 'csv-display':
244 + $final = $this->_getCSV( $rows, $filename, true );
245 + break;
205 246 case 'xls':
206 247 $final = $this->_getExcel( $rows, $filename );
207 248 break;
208 249 case 'print':
@@ -207,8 +248,11 @@
207 248 break;
208 249 case 'print':
209 250 $final = $this->_getHTML( $rows );
210 251 break;
252 + case 'image':
253 + $final = $this->_getImage( $chart );
254 + break;
211 255 }
212 256 }
213 257 return $final;
214 258 }
@@ -218,24 +262,50 @@
218 262 *
219 263 * @access private
220 264 * @param array $rows The array of data.
221 265 * @param string $filename The name of the file to use.
266 + * @param bool $enclose Enclose strings that have commas in them in double quotes.
222 267 */
223 - private function _getCSV( $rows, $filename ) {
268 + private function _getCSV( $rows, $filename, $enclose ) {
224 269 $filename .= '.csv';
225 270
226 - $fp = tmpfile();
271 + $bom = chr( 0xEF ) . chr( 0xBB ) . chr( 0xBF );
272 + // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
273 + $fp = function_exists( 'tmpfile' ) ? @tmpfile() : null;
274 + if ( null === $fp ) {
275 + if ( ! function_exists( 'wp_tempnam' ) ) {
276 + require_once ABSPATH . 'wp-admin/includes/file.php';
277 + }
278 + $fp = fopen( wp_tempnam(), 'w+' );
279 + }
280 + if ( ! apply_filters( 'vizualizer_export_include_series_type', true ) ) {
281 + unset( $rows[1] );
282 + $rows = array_values( $rows );
283 + }
227 284 // support for MS Excel
228 - fprintf( $fp, $bom = ( chr( 0xEF ) . chr( 0xBB ) . chr( 0xBF ) ) );
285 + fprintf( $fp, $bom );
229 286 foreach ( $rows as $row ) {
230 287 fputcsv( $fp, $row );
231 288 }
232 289 rewind( $fp );
233 290 $csv = '';
291 + // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.FoundInWhileCondition
234 292 while ( ( $array = fgetcsv( $fp ) ) !== false ) {
235 293 if ( strlen( $csv ) > 0 ) {
236 294 $csv .= PHP_EOL;
237 295 }
296 + // if enclosure is required, check every item of this line
297 + // if a comma exists in the item, add enclosure.
298 + if ( $enclose ) {
299 + $temp_array = array();
300 + foreach ( $array as $item ) {
301 + if ( strpos( $item, ',' ) !== false ) {
302 + $item = VISUALIZER_CSV_ENCLOSURE . $item . VISUALIZER_CSV_ENCLOSURE;
303 + }
304 + $temp_array[] = $item;
305 + }
306 + $array = $temp_array;
307 + }
238 308 $csv .= implode( ',', $array );
239 309 }
240 310 fclose( $fp );
241 311
@@ -241,8 +311,9 @@
241 311
242 312 return array(
243 313 'csv' => $csv,
244 314 'name' => $filename,
315 + 'string' => str_replace( $bom, '', $csv ),
245 316 );
246 317 }
247 318
248 319 /**
@@ -255,14 +326,22 @@
255 326 private function _getExcel( $rows, $filename ) {
256 327 // PHPExcel did not like sheet names longer than 31 characters and we will assume the same with PhpSpreadsheet
257 328 $chart = substr( $filename, 0, 30 );
258 329 $filename .= '.xlsx';
259 -
330 + if ( ! apply_filters( 'vizualizer_export_include_series_type', true ) ) {
331 + unset( $rows[1] );
332 + $rows = array_values( $rows );
333 + $rows = array_map(
334 + function( $r ) {
335 + return array_map( 'strval', $r );
336 + },
337 + $rows
338 + );
339 + }
260 340 $vendor_file = VISUALIZER_ABSPATH . '/vendor/autoload.php';
261 341 if ( is_readable( $vendor_file ) ) {
262 342 include_once( $vendor_file );
263 343 }
264 -
265 344 $xlsData = '';
266 345 if ( class_exists( 'PhpOffice\PhpSpreadsheet\Spreadsheet' ) ) {
267 346 $doc = new PhpOffice\PhpSpreadsheet\Spreadsheet();
268 347 $doc->getActiveSheet()->fromArray( $rows, null, 'A1' );
@@ -279,8 +358,9 @@
279 358 }
280 359 return array(
281 360 'csv' => 'data:application/vnd.ms-excel;base64,' . base64_encode( $xlsData ),
282 361 'name' => $filename,
362 + 'raw' => base64_encode( $xlsData ),
283 363 );
284 364 }
285 365
286 366 /**
@@ -310,8 +390,14 @@
310 390
311 391 $table = '<table class="visualizer-print">';
312 392 $index = 0;
313 393 foreach ( $rows as $row ) {
394 + // skip the data type row.
395 + if ( 1 === $index ) {
396 + $index++;
397 + continue;
398 + }
399 +
314 400 $table .= '<tr>';
315 401 foreach ( $row as $col ) {
316 402 if ( $index === 0 ) {
317 403 $table .= '<th>' . $col . '</th>';
@@ -332,8 +418,22 @@
332 418 );
333 419 }
334 420
335 421 /**
422 + * Disable revisions temporarily for visualizer post type.
423 + */
424 + protected final function disableRevisionsTemporarily() {
425 + add_filter(
426 + 'wp_revisions_to_keep', function( $num, $post ) {
427 + if ( $post->post_type === Visualizer_Plugin::CPT_VISUALIZER ) {
428 + return 0;
429 + }
430 + return $num;
431 + }, 10, 2
432 + );
433 + }
434 +
435 + /**
336 436 * Undo revisions for the chart, and if necessary, restore the earliest version.
337 437 *
338 438 * @return bool If any revisions were found.
339 439 */
@@ -338,9 +438,11 @@
338 438 * @return bool If any revisions were found.
339 439 */
340 440 public final function undoRevisions( $chart_id, $restore = false ) {
341 441 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'undoRevisions for %d with%s restore', $chart_id, ( $restore ? '' : 'out' ) ), 'debug', __FILE__, __LINE__ );
342 -
442 + if ( get_post_type( $chart_id ) !== Visualizer_Plugin::CPT_VISUALIZER ) {
443 + return false;
444 + }
343 445 $revisions = wp_get_post_revisions( $chart_id, array( 'order' => 'ASC' ) );
344 446 if ( count( $revisions ) > 1 ) {
345 447 $revision_ids = array_keys( $revisions );
346 448
@@ -346,9 +448,9 @@
346 448
347 449 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'found %d revisions = %s', count( $revisions ), print_r( $revision_ids, true ) ), 'debug', __FILE__, __LINE__ );
348 450
349 451 // when we restore, a new revision is likely to be created. so, let's disable revisions for the time being.
350 - add_filter( 'wp_revisions_to_keep', '__return_false' );
452 + $this->disableRevisionsTemporarily();
351 453
352 454 if ( $restore ) {
353 455 // restore to the oldest one i.e. the first one.
354 456 wp_restore_post_revision( array_shift( $revision_ids ) );
@@ -367,10 +469,13 @@
367 469 /**
368 470 * If existing revisions exist for the chart, restore the earliest version and then create a new revision to initiate editing.
369 471 */
370 472 public final function handleExistingRevisions( $chart_id, $chart ) {
473 +
371 474 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'handleExistingRevisions for %d', $chart_id ), 'debug', __FILE__, __LINE__ );
372 -
475 + if ( get_post_type( $chart_id ) !== Visualizer_Plugin::CPT_VISUALIZER ) {
476 + return $chart_id;
477 + }
373 478 // undo revisions.
374 479 $revisions_found = $this->undoRevisions( $chart_id, true );
375 480
376 481 // create revision for the edit action.
@@ -417,12 +522,21 @@
417 522
418 523 require_once( ABSPATH . 'wp-admin/includes/file.php' );
419 524 WP_Filesystem();
420 525 global $wp_filesystem;
526 + if ( ! is_a( $wp_filesystem, 'WP_Filesystem_Base' ) ) {
527 + $creds = request_filesystem_credentials( site_url() );
528 + wp_filesystem( $creds );
529 + }
421 530
422 - $dir = $wp_filesystem->wp_content_dir() . 'uploads/visualizer';
423 - $file = $wp_filesystem->wp_content_dir() . 'uploads/visualizer/customization.js';
531 + $multisite_arg = '/';
532 + if ( is_multisite() && ! is_main_site() ) {
533 + $multisite_arg = '/sites/' . get_current_blog_id() . '/';
534 + }
424 535
536 + $dir = $wp_filesystem->wp_content_dir() . 'uploads' . $multisite_arg . 'visualizer';
537 + $file = $wp_filesystem->wp_content_dir() . 'uploads' . $multisite_arg . 'visualizer/customization.js';
538 +
425 539 if ( $wp_filesystem->is_readable( $file ) ) {
426 540 return $specific;
427 541 }
428 542
@@ -431,8 +545,9 @@
431 545 return $default;
432 546 }
433 547
434 548 if ( ! $wp_filesystem->exists( $dir ) ) {
549 + // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.Found
435 550 if ( ( $done = $wp_filesystem->mkdir( $dir ) ) === false ) {
436 551 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'Unable to create directory %s', $dir ), 'error', __FILE__, __LINE__ );
437 552 return $default;
438 553 }
@@ -440,8 +555,9 @@
440 555
441 556 // if file does not exist, copy.
442 557 if ( ! $wp_filesystem->exists( $file ) ) {
443 558 $src = str_replace( ABSPATH, $wp_filesystem->abspath(), VISUALIZER_ABSPATH . '/js/customization.js' );
559 + // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.Found
444 560 if ( ( $done = $wp_filesystem->copy( $src, $file ) ) === false ) {
445 561 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'Unable to copy file %s to %s', $src, $file ), 'error', __FILE__, __LINE__ );
446 562 return $default;
447 563 }
@@ -453,18 +569,21 @@
453 569 /**
454 570 * Load the class for the given chart's chart type so that its assets can be loaded.
455 571 */
456 572 protected function load_chart_type( $chart_id ) {
457 - $type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
458 - $name = 'Visualizer_Render_Sidebar_Type_' . ucwords( $type );
573 + $name = $this->load_chart_class_name( $chart_id );
459 574 $class = null;
460 575 if ( class_exists( $name ) || true === apply_filters( 'visualizer_load_chart', false, $name ) ) {
576 + if ( 'Visualizer_Render_Sidebar_Type_DataTable_DataTable' === $name ) {
577 + $name = 'Visualizer_Render_Sidebar_Type_DataTable_Tabular';
578 + }
461 579 $class = new $name;
462 580 }
463 581
464 - if ( is_null( $class ) && VISUALIZER_PRO ) {
582 + if ( is_null( $class ) && Visualizer_Module::is_pro() ) {
465 583 // lets see if this type exists in pro. New Lite(3.1.0+) & old Pro(1.8.0-).
466 - $class = apply_filters( 'visualizer_pro_chart_type_sidebar', null, array( 'type' => $type, 'settings' => get_post_meta( $chart_id, Visualizer_Plugin::CF_SETTINGS, true ) ) );
584 + $type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
585 + $class = apply_filters( 'visualizer_pro_chart_type_sidebar', null, array( 'id' => $chart_id, 'type' => $type, 'settings' => get_post_meta( $chart_id, Visualizer_Plugin::CF_SETTINGS, true ) ) );
467 586 }
468 587
469 588 return is_null( $class ) ? null : $class->getLibrary();
470 589 }
@@ -469,8 +588,26 @@
469 588 return is_null( $class ) ? null : $class->getLibrary();
470 589 }
471 590
472 591 /**
592 + * Returns the class name for the given chart's chart type.
593 + */
594 + protected function load_chart_class_name( $chart_id ) {
595 + $type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
596 + $lib = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_LIBRARY, true );
597 +
598 + // backward compatibility.
599 + if ( empty( $lib ) ) {
600 + $lib = 'GoogleCharts';
601 + if ( 'dataTable' === $type ) {
602 + $lib = 'DataTable';
603 + }
604 + }
605 + $name = 'Visualizer_Render_Sidebar_Type_' . $lib . '_' . ucwords( $type );
606 + return $name;
607 + }
608 +
609 + /**
473 610 * Generates the inline CSS to apply to the chart and adds these classes to the settings.
474 611 *
475 612 * @access public
476 613 * @param int $id The id of the chart.
@@ -478,32 +615,31 @@
478 615 */
479 616 protected function get_inline_custom_css( $id, $settings ) {
480 617 $css = '';
481 618
482 - $arguments = array( '', $settings );
483 - if ( ! isset( $settings['customcss'] ) ) {
484 - return $arguments;
485 - }
486 -
487 619 $classes = array();
488 620 $css = '<style type="text/css" name="visualizer-custom-css" id="customcss-' . $id . '">';
489 - foreach ( $settings['customcss'] as $name => $element ) {
490 - $attributes = array();
491 - foreach ( $element as $property => $value ) {
492 - $attributes[] = $this->handle_css_property( $property, $value );
621 + if ( ! empty( $settings['customcss'] ) ) {
622 + foreach ( $settings['customcss'] as $name => $element ) {
623 + $attributes = array();
624 + foreach ( $element as $property => $value ) {
625 + $attributes[] = $this->handle_css_property( $property, $value );
626 + }
627 + $class_name = $id . $name;
628 + $properties = implode( ' !important; ', array_filter( $attributes ) );
629 + if ( ! empty( $properties ) ) {
630 + $css .= '.' . $class_name . ' {' . $properties . ' !important;}';
631 + $classes[ $name ] = $class_name;
632 + }
493 633 }
494 - $class_name = $id . $name;
495 - $properties = implode( '; ', array_filter( $attributes ) );
496 - if ( ! empty( $properties ) ) {
497 - $css .= '.' . $class_name . ' {' . $properties . ' !important;}';
498 - $classes[ $name ] = $class_name;
499 - }
634 + $settings['cssClassNames'] = $classes;
500 635 }
501 - $css .= '</style>';
502 636
503 - $settings['cssClassNames'] = $classes;
637 + $img_path = VISUALIZER_ABSURL . 'images';
638 + $css .= ".locker,.locker-loader{position:absolute;top:0;left:0;width:100%;height:100%}.locker{z-index:1000;opacity:.8;background-color:#fff;-ms-filter:\"progid:DXImageTransform.Microsoft.Alpha(Opacity=80)\";filter:alpha(opacity=80)}.locker-loader{z-index:1001;background:url($img_path/ajax-loader.gif) no-repeat center center}.dt-button{display:none!important}.visualizer-front-container.visualizer-lazy-render{content-visibility: auto;}.google-visualization-controls-categoryfilter label.google-visualization-controls-label {vertical-align: middle;}.google-visualization-controls-categoryfilter li.goog-inline-block {margin: 0 0.2em;}.google-visualization-controls-categoryfilter li {padding: 0 0.2em;}.visualizer-front-container .dataTables_scrollHeadInner{margin: 0 auto;}";
639 + $css .= '</style>';
504 640
505 - $arguments = array( $css, $settings );
641 + $arguments = array( $css, $settings );
506 642 apply_filters_ref_array( 'visualizer_inline_css', array( &$arguments ) );
507 643
508 644 return $arguments;
509 645 }
@@ -547,5 +683,156 @@
547 683 $q = new WP_Query( $args );
548 684 return $q->found_posts > 0;
549 685 }
550 686
687 + /**
688 + * Determines how many charts have been created.
689 + */
690 + protected static function numberOfCharts() {
691 + $args = array(
692 + 'post_type' => Visualizer_Plugin::CPT_VISUALIZER,
693 + 'fields' => 'ids',
694 + 'post_status' => 'publish',
695 + 'posts_per_page' => 300,
696 + );
697 +
698 + $q = new WP_Query( $args );
699 + return $q->found_posts;
700 + }
701 +
702 + /**
703 + * Checks if the PRO version is active.
704 + *
705 + * @since 3.3.0
706 + */
707 + public static function is_pro() {
708 + // versions of pro before 1.9.0 will use the constant VISUALIZER_PRO
709 + // versions of pro 1.9.0 onwards will use the filter
710 + return apply_filters( 'visualizer_is_pro', VISUALIZER_PRO );
711 + }
712 +
713 + /**
714 + * Checks if the PRO version is older than a particular version.
715 + *
716 + * @since 3.3.0
717 + */
718 + public static function is_pro_older_than( $version ) {
719 + return version_compare( VISUALIZER_PRO_VERSION, $version, '<' );
720 + }
721 +
722 + /**
723 + * Should we show some specific feature on the basis of the version?
724 + *
725 + * @since 3.4.0
726 + */
727 + public static function can_show_feature( $feature ) {
728 + switch ( $feature ) {
729 + case 'simple-editor':
730 + // if user has pro but an older version, then don't load the simple editor functionality
731 + // as the select box will not behave as expected because the pro editor's functionality will supercede.
732 + return ! Visualizer_Module::is_pro() || ! Visualizer_Module::is_pro_older_than( '1.9.2' );
733 + }
734 + return false;
735 + }
736 +
737 + /**
738 + * Gets the features for the provided license type.
739 + */
740 + public static final function get_features_for_license( $plan ) {
741 + switch ( $plan ) {
742 + case 1:
743 + return array( 'import-wp', 'db-query', 'import-wc-report' );
744 + case 2:
745 + return array( 'schedule-chart', 'chart-permissions', 'import-chart', 'manual-data', 'data-filter-configuration', 'frontend-actions' );
746 + }
747 + }
748 +
749 + /**
750 + * Gets the chart content after common manipulations.
751 + */
752 + public static function get_chart_data( $chart, $type, $run_filter = true ) {
753 + // change HTML entities
754 + $post_content = html_entity_decode( htmlentities( $chart->post_content ) );
755 + $post_content = preg_replace_callback(
756 + '!s:(\d+):"(.*?)";!s',
757 + function ( $matches ) {
758 + if ( isset( $matches[2] ) ) {
759 + return 's:' . strlen( $matches[2] ) . ':"' . $matches[2] . '";';
760 + }
761 + },
762 + $post_content
763 + );
764 + $data = unserialize( $post_content );
765 + $altered = array();
766 + if ( ! empty( $data ) ) {
767 + foreach ( $data as $index => $array ) {
768 + if ( ! is_array( $index ) && is_array( $array ) ) {
769 + foreach ( $array as &$datum ) {
770 + if ( is_string( $datum ) ) {
771 + $datum = stripslashes( $datum );
772 + }
773 + }
774 + $altered[ $index ] = $array;
775 + }
776 + }
777 + }
778 + // if something goes wrong and the end result is empty, be safe and use the original data
779 + if ( empty( $altered ) ) {
780 + $altered = $data;
781 + }
782 + if ( $run_filter ) {
783 + return apply_filters( Visualizer_Plugin::FILTER_GET_CHART_DATA, $altered, $chart->ID, $type );
784 + }
785 + return $altered;
786 + }
787 +
788 + /**
789 + * Get chart image.
790 + *
791 + * @param object $chart Chart data.
792 + * @return string
793 + */
794 + public function _getImage( $chart = null ) {
795 + $image = '';
796 + if ( $chart ) {
797 + $chart_image = get_post_meta( $chart->ID, Visualizer_Plugin::CF_CHART_IMAGE, true );
798 + if ( ! empty( $chart_image ) ) {
799 + $image = wp_get_attachment_image( $chart_image, 'full' );
800 + }
801 + }
802 + return array(
803 + 'csv' => $image,
804 + );
805 + }
806 +
807 + /**
808 + * Filter chart title if visualizer post type.
809 + *
810 + * @param object $query WP Query object.
811 + * @return void
812 + */
813 + public function PreGetPosts( $query ) {
814 + if ( ! $query->is_main_query() ) {
815 + $post_type = $query->get( 'post_type' );
816 + if ( 'visualizer' === $post_type ) {
817 + $this->_addFilter( Visualizer_Plugin::FILTER_CHART_TITLE, 'filterChartTitle', 10, 2 );
818 + }
819 + }
820 + }
821 +
822 + /**
823 + * Filter chart title.
824 + *
825 + * @access public
826 + * @param string $post_title Post title.
827 + * @param int $post_id Post ID.
828 + * @return string
829 + */
830 + public function filterChartTitle( $post_title, $post_id ) {
831 + $post_type = get_post_type( $post_id );
832 + $post_title = trim( $post_title );
833 + if ( 'visualizer' === $post_type && 'Visualization' === $post_title ) {
834 + return sprintf( '%s #%d', $post_title, $post_id );
835 + }
836 + return $post_title;
837 + }
551 838 }