PluginProbe
Visualizer – Tables & Charts Manager with Built-in AI Generator / 3.7.12
Visualizer – Tables & Charts Manager with Built-in AI Generator v3.7.12
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 +321 -40 3.1.23.7.12 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,43 @@
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
271 + $bom = chr( 0xEF ) . chr( 0xBB ) . chr( 0xBF );
226 272 $fp = tmpfile();
273 + if ( ! apply_filters( 'vizualizer_export_include_series_type', true ) ) {
274 + unset( $rows[1] );
275 + $rows = array_values( $rows );
276 + }
227 277 // support for MS Excel
228 - fprintf( $fp, $bom = ( chr( 0xEF ) . chr( 0xBB ) . chr( 0xBF ) ) );
278 + fprintf( $fp, $bom );
229 279 foreach ( $rows as $row ) {
230 280 fputcsv( $fp, $row );
231 281 }
232 282 rewind( $fp );
233 283 $csv = '';
284 + // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.FoundInWhileCondition
234 285 while ( ( $array = fgetcsv( $fp ) ) !== false ) {
235 286 if ( strlen( $csv ) > 0 ) {
236 287 $csv .= PHP_EOL;
237 288 }
289 + // if enclosure is required, check every item of this line
290 + // if a comma exists in the item, add enclosure.
291 + if ( $enclose ) {
292 + $temp_array = array();
293 + foreach ( $array as $item ) {
294 + if ( strpos( $item, ',' ) !== false ) {
295 + $item = VISUALIZER_CSV_ENCLOSURE . $item . VISUALIZER_CSV_ENCLOSURE;
296 + }
297 + $temp_array[] = $item;
298 + }
299 + $array = $temp_array;
300 + }
238 301 $csv .= implode( ',', $array );
239 302 }
240 303 fclose( $fp );
241 304
@@ -241,8 +304,9 @@
241 304
242 305 return array(
243 306 'csv' => $csv,
244 307 'name' => $filename,
308 + 'string' => str_replace( $bom, '', $csv ),
245 309 );
246 310 }
247 311
248 312 /**
@@ -252,30 +316,44 @@
252 316 * @param array $rows The array of data.
253 317 * @param string $filename The name of the file to use.
254 318 */
255 319 private function _getExcel( $rows, $filename ) {
256 - // PHPExcel does not like sheet names longer than 31 characters.
320 + // PHPExcel did not like sheet names longer than 31 characters and we will assume the same with PhpSpreadsheet
257 321 $chart = substr( $filename, 0, 30 );
258 322 $filename .= '.xlsx';
259 -
323 + if ( ! apply_filters( 'vizualizer_export_include_series_type', true ) ) {
324 + unset( $rows[1] );
325 + $rows = array_values( $rows );
326 + $rows = array_map(
327 + function( $r ) {
328 + return array_map( 'strval', $r );
329 + },
330 + $rows
331 + );
332 + }
333 + $vendor_file = VISUALIZER_ABSPATH . '/vendor/autoload.php';
334 + if ( is_readable( $vendor_file ) ) {
335 + include_once( $vendor_file );
336 + }
260 337 $xlsData = '';
261 - if ( class_exists( 'PHPExcel' ) ) {
262 - $doc = new PHPExcel();
338 + if ( class_exists( 'PhpOffice\PhpSpreadsheet\Spreadsheet' ) ) {
339 + $doc = new PhpOffice\PhpSpreadsheet\Spreadsheet();
263 340 $doc->getActiveSheet()->fromArray( $rows, null, 'A1' );
264 341 $doc->getActiveSheet()->setTitle( sanitize_title( $chart ) );
265 342 $doc = apply_filters( 'visualizer_excel_doc', $doc );
266 - $writer = PHPExcel_IOFactory::createWriter( $doc, 'Excel2007' );
343 + $writer = PhpOffice\PhpSpreadsheet\IOFactory::createWriter( $doc, 'Xlsx' );
267 344 ob_start();
268 345 $writer->save( 'php://output' );
269 346 $xlsData = ob_get_contents();
270 347 ob_end_clean();
271 348 } else {
272 - do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, 'Class PHPExcel does not exist!', 'error', __FILE__, __LINE__ );
273 - error_log( 'Class PHPExcel does not exist!' );
349 + do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, 'Class PhpOffice\PhpSpreadsheet\Spreadsheet does not exist!', 'error', __FILE__, __LINE__ );
350 + error_log( 'Class PhpOffice\PhpSpreadsheet\Spreadsheet does not exist!' );
274 351 }
275 352 return array(
276 353 'csv' => 'data:application/vnd.ms-excel;base64,' . base64_encode( $xlsData ),
277 354 'name' => $filename,
355 + 'raw' => base64_encode( $xlsData ),
278 356 );
279 357 }
280 358
281 359 /**
@@ -305,8 +383,14 @@
305 383
306 384 $table = '<table class="visualizer-print">';
307 385 $index = 0;
308 386 foreach ( $rows as $row ) {
387 + // skip the data type row.
388 + if ( 1 === $index ) {
389 + $index++;
390 + continue;
391 + }
392 +
309 393 $table .= '<tr>';
310 394 foreach ( $row as $col ) {
311 395 if ( $index === 0 ) {
312 396 $table .= '<th>' . $col . '</th>';
@@ -327,8 +411,22 @@
327 411 );
328 412 }
329 413
330 414 /**
415 + * Disable revisions temporarily for visualizer post type.
416 + */
417 + protected final function disableRevisionsTemporarily() {
418 + add_filter(
419 + 'wp_revisions_to_keep', function( $num, $post ) {
420 + if ( $post->post_type === Visualizer_Plugin::CPT_VISUALIZER ) {
421 + return 0;
422 + }
423 + return $num;
424 + }, 10, 2
425 + );
426 + }
427 +
428 + /**
331 429 * Undo revisions for the chart, and if necessary, restore the earliest version.
332 430 *
333 431 * @return bool If any revisions were found.
334 432 */
@@ -333,9 +431,11 @@
333 431 * @return bool If any revisions were found.
334 432 */
335 433 public final function undoRevisions( $chart_id, $restore = false ) {
336 434 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'undoRevisions for %d with%s restore', $chart_id, ( $restore ? '' : 'out' ) ), 'debug', __FILE__, __LINE__ );
337 -
435 + if ( get_post_type( $chart_id ) !== Visualizer_Plugin::CPT_VISUALIZER ) {
436 + return false;
437 + }
338 438 $revisions = wp_get_post_revisions( $chart_id, array( 'order' => 'ASC' ) );
339 439 if ( count( $revisions ) > 1 ) {
340 440 $revision_ids = array_keys( $revisions );
341 441
@@ -341,9 +441,9 @@
341 441
342 442 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'found %d revisions = %s', count( $revisions ), print_r( $revision_ids, true ) ), 'debug', __FILE__, __LINE__ );
343 443
344 444 // when we restore, a new revision is likely to be created. so, let's disable revisions for the time being.
345 - add_filter( 'wp_revisions_to_keep', '__return_false' );
445 + $this->disableRevisionsTemporarily();
346 446
347 447 if ( $restore ) {
348 448 // restore to the oldest one i.e. the first one.
349 449 wp_restore_post_revision( array_shift( $revision_ids ) );
@@ -362,10 +462,13 @@
362 462 /**
363 463 * If existing revisions exist for the chart, restore the earliest version and then create a new revision to initiate editing.
364 464 */
365 465 public final function handleExistingRevisions( $chart_id, $chart ) {
466 +
366 467 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'handleExistingRevisions for %d', $chart_id ), 'debug', __FILE__, __LINE__ );
367 -
468 + if ( get_post_type( $chart_id ) !== Visualizer_Plugin::CPT_VISUALIZER ) {
469 + return $chart_id;
470 + }
368 471 // undo revisions.
369 472 $revisions_found = $this->undoRevisions( $chart_id, true );
370 473
371 474 // create revision for the edit action.
@@ -413,11 +516,16 @@
413 516 require_once( ABSPATH . 'wp-admin/includes/file.php' );
414 517 WP_Filesystem();
415 518 global $wp_filesystem;
416 519
417 - $dir = $wp_filesystem->wp_content_dir() . 'uploads/visualizer';
418 - $file = $wp_filesystem->wp_content_dir() . 'uploads/visualizer/customization.js';
520 + $multisite_arg = '/';
521 + if ( is_multisite() && ! is_main_site() ) {
522 + $multisite_arg = '/sites/' . get_current_blog_id() . '/';
523 + }
419 524
525 + $dir = $wp_filesystem->wp_content_dir() . 'uploads' . $multisite_arg . 'visualizer';
526 + $file = $wp_filesystem->wp_content_dir() . 'uploads' . $multisite_arg . 'visualizer/customization.js';
527 +
420 528 if ( $wp_filesystem->is_readable( $file ) ) {
421 529 return $specific;
422 530 }
423 531
@@ -426,8 +534,9 @@
426 534 return $default;
427 535 }
428 536
429 537 if ( ! $wp_filesystem->exists( $dir ) ) {
538 + // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.Found
430 539 if ( ( $done = $wp_filesystem->mkdir( $dir ) ) === false ) {
431 540 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'Unable to create directory %s', $dir ), 'error', __FILE__, __LINE__ );
432 541 return $default;
433 542 }
@@ -435,8 +544,9 @@
435 544
436 545 // if file does not exist, copy.
437 546 if ( ! $wp_filesystem->exists( $file ) ) {
438 547 $src = str_replace( ABSPATH, $wp_filesystem->abspath(), VISUALIZER_ABSPATH . '/js/customization.js' );
548 + // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.Found
439 549 if ( ( $done = $wp_filesystem->copy( $src, $file ) ) === false ) {
440 550 do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'Unable to copy file %s to %s', $src, $file ), 'error', __FILE__, __LINE__ );
441 551 return $default;
442 552 }
@@ -448,18 +558,21 @@
448 558 /**
449 559 * Load the class for the given chart's chart type so that its assets can be loaded.
450 560 */
451 561 protected function load_chart_type( $chart_id ) {
452 - $type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
453 - $name = 'Visualizer_Render_Sidebar_Type_' . ucwords( $type );
562 + $name = $this->load_chart_class_name( $chart_id );
454 563 $class = null;
455 564 if ( class_exists( $name ) || true === apply_filters( 'visualizer_load_chart', false, $name ) ) {
565 + if ( 'Visualizer_Render_Sidebar_Type_DataTable_DataTable' === $name ) {
566 + $name = 'Visualizer_Render_Sidebar_Type_DataTable_Tabular';
567 + }
456 568 $class = new $name;
457 569 }
458 570
459 - if ( is_null( $class ) && VISUALIZER_PRO ) {
571 + if ( is_null( $class ) && Visualizer_Module::is_pro() ) {
460 572 // lets see if this type exists in pro. New Lite(3.1.0+) & old Pro(1.8.0-).
461 - $class = apply_filters( 'visualizer_pro_chart_type_sidebar', null, array( 'type' => $type, 'settings' => get_post_meta( $chart_id, Visualizer_Plugin::CF_SETTINGS, true ) ) );
573 + $type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
574 + $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 ) ) );
462 575 }
463 576
464 577 return is_null( $class ) ? null : $class->getLibrary();
465 578 }
@@ -464,8 +577,26 @@
464 577 return is_null( $class ) ? null : $class->getLibrary();
465 578 }
466 579
467 580 /**
581 + * Returns the class name for the given chart's chart type.
582 + */
583 + protected function load_chart_class_name( $chart_id ) {
584 + $type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
585 + $lib = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_LIBRARY, true );
586 +
587 + // backward compatibility.
588 + if ( empty( $lib ) ) {
589 + $lib = 'GoogleCharts';
590 + if ( 'dataTable' === $type ) {
591 + $lib = 'DataTable';
592 + }
593 + }
594 + $name = 'Visualizer_Render_Sidebar_Type_' . $lib . '_' . ucwords( $type );
595 + return $name;
596 + }
597 +
598 + /**
468 599 * Generates the inline CSS to apply to the chart and adds these classes to the settings.
469 600 *
470 601 * @access public
471 602 * @param int $id The id of the chart.
@@ -473,32 +604,31 @@
473 604 */
474 605 protected function get_inline_custom_css( $id, $settings ) {
475 606 $css = '';
476 607
477 - $arguments = array( '', $settings );
478 - if ( ! isset( $settings['customcss'] ) ) {
479 - return $arguments;
480 - }
481 -
482 608 $classes = array();
483 609 $css = '<style type="text/css" name="visualizer-custom-css" id="customcss-' . $id . '">';
484 - foreach ( $settings['customcss'] as $name => $element ) {
485 - $attributes = array();
486 - foreach ( $element as $property => $value ) {
487 - $attributes[] = $this->handle_css_property( $property, $value );
610 + if ( ! empty( $settings['customcss'] ) ) {
611 + foreach ( $settings['customcss'] as $name => $element ) {
612 + $attributes = array();
613 + foreach ( $element as $property => $value ) {
614 + $attributes[] = $this->handle_css_property( $property, $value );
615 + }
616 + $class_name = $id . $name;
617 + $properties = implode( ' !important; ', array_filter( $attributes ) );
618 + if ( ! empty( $properties ) ) {
619 + $css .= '.' . $class_name . ' {' . $properties . ' !important;}';
620 + $classes[ $name ] = $class_name;
621 + }
488 622 }
489 - $class_name = $id . $name;
490 - $properties = implode( '; ', array_filter( $attributes ) );
491 - if ( ! empty( $properties ) ) {
492 - $css .= '.' . $class_name . ' {' . $properties . ' !important;}';
493 - $classes[ $name ] = $class_name;
494 - }
623 + $settings['cssClassNames'] = $classes;
495 624 }
496 - $css .= '</style>';
497 625
498 - $settings['cssClassNames'] = $classes;
626 + $img_path = VISUALIZER_ABSURL . 'images';
627 + $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;}";
628 + $css .= '</style>';
499 629
500 - $arguments = array( $css, $settings );
630 + $arguments = array( $css, $settings );
501 631 apply_filters_ref_array( 'visualizer_inline_css', array( &$arguments ) );
502 632
503 633 return $arguments;
504 634 }
@@ -542,5 +672,156 @@
542 672 $q = new WP_Query( $args );
543 673 return $q->found_posts > 0;
544 674 }
545 675
676 + /**
677 + * Determines how many charts have been created.
678 + */
679 + protected static function numberOfCharts() {
680 + $args = array(
681 + 'post_type' => Visualizer_Plugin::CPT_VISUALIZER,
682 + 'fields' => 'ids',
683 + 'post_status' => 'publish',
684 + 'posts_per_page' => 300,
685 + );
686 +
687 + $q = new WP_Query( $args );
688 + return $q->found_posts;
689 + }
690 +
691 + /**
692 + * Checks if the PRO version is active.
693 + *
694 + * @since 3.3.0
695 + */
696 + public static function is_pro() {
697 + // versions of pro before 1.9.0 will use the constant VISUALIZER_PRO
698 + // versions of pro 1.9.0 onwards will use the filter
699 + return apply_filters( 'visualizer_is_pro', VISUALIZER_PRO );
700 + }
701 +
702 + /**
703 + * Checks if the PRO version is older than a particular version.
704 + *
705 + * @since 3.3.0
706 + */
707 + public static function is_pro_older_than( $version ) {
708 + return version_compare( VISUALIZER_PRO_VERSION, $version, '<' );
709 + }
710 +
711 + /**
712 + * Should we show some specific feature on the basis of the version?
713 + *
714 + * @since 3.4.0
715 + */
716 + public static function can_show_feature( $feature ) {
717 + switch ( $feature ) {
718 + case 'simple-editor':
719 + // if user has pro but an older version, then don't load the simple editor functionality
720 + // as the select box will not behave as expected because the pro editor's functionality will supercede.
721 + return ! Visualizer_Module::is_pro() || ! Visualizer_Module::is_pro_older_than( '1.9.2' );
722 + }
723 + return false;
724 + }
725 +
726 + /**
727 + * Gets the features for the provided license type.
728 + */
729 + public static final function get_features_for_license( $plan ) {
730 + switch ( $plan ) {
731 + case 1:
732 + return array( 'import-wp', 'db-query' );
733 + case 2:
734 + return array( 'schedule-chart', 'chart-permissions' );
735 + }
736 + }
737 +
738 + /**
739 + * Gets the chart content after common manipulations.
740 + */
741 + public static function get_chart_data( $chart, $type, $run_filter = true ) {
742 + // change HTML entities
743 + $post_content = html_entity_decode( htmlentities( $chart->post_content ) );
744 + $post_content = preg_replace_callback(
745 + '!s:(\d+):"(.*?)";!s',
746 + function ( $matches ) {
747 + if ( isset( $matches[2] ) ) {
748 + return 's:' . strlen( $matches[2] ) . ':"' . $matches[2] . '";';
749 + }
750 + },
751 + $post_content
752 + );
753 + $data = unserialize( $post_content );
754 + $altered = array();
755 + if ( ! empty( $data ) ) {
756 + foreach ( $data as $index => $array ) {
757 + if ( ! is_array( $index ) && is_array( $array ) ) {
758 + foreach ( $array as &$datum ) {
759 + if ( is_string( $datum ) ) {
760 + $datum = stripslashes( $datum );
761 + }
762 + }
763 + $altered[ $index ] = $array;
764 + }
765 + }
766 + }
767 + // if something goes wrong and the end result is empty, be safe and use the original data
768 + if ( empty( $altered ) ) {
769 + $altered = $data;
770 + }
771 + if ( $run_filter ) {
772 + return apply_filters( Visualizer_Plugin::FILTER_GET_CHART_DATA, $altered, $chart->ID, $type );
773 + }
774 + return $altered;
775 + }
776 +
777 + /**
778 + * Get chart image.
779 + *
780 + * @param object $chart Chart data.
781 + * @return string
782 + */
783 + public function _getImage( $chart = null ) {
784 + $image = '';
785 + if ( $chart ) {
786 + $chart_image = get_post_meta( $chart->ID, Visualizer_Plugin::CF_CHART_IMAGE, true );
787 + if ( ! empty( $chart_image ) ) {
788 + $image = wp_get_attachment_image( $chart_image, 'full' );
789 + }
790 + }
791 + return array(
792 + 'csv' => $image,
793 + );
794 + }
795 +
796 + /**
797 + * Filter chart title if visualizer post type.
798 + *
799 + * @param object $query WP Query object.
800 + * @return void
801 + */
802 + public function PreGetPosts( $query ) {
803 + if ( ! $query->is_main_query() ) {
804 + $post_type = $query->get( 'post_type' );
805 + if ( 'visualizer' === $post_type ) {
806 + $this->_addFilter( Visualizer_Plugin::FILTER_CHART_TITLE, 'filterChartTitle', 10, 2 );
807 + }
808 + }
809 + }
810 +
811 + /**
812 + * Filter chart title.
813 + *
814 + * @access public
815 + * @param string $post_title Post title.
816 + * @param int $post_id Post ID.
817 + * @return string
818 + */
819 + public function filterChartTitle( $post_title, $post_id ) {
820 + $post_type = get_post_type( $post_id );
821 + $post_title = trim( $post_title );
822 + if ( 'visualizer' === $post_type && 'Visualization' === $post_title ) {
823 + return sprintf( '%s #%d', $post_title, $post_id );
824 + }
825 + return $post_title;
826 + }
546 827 }