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