← All changes | classes/Visualizer/Module.php +319 -36 3.1.3 → 3.10.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,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.
		@@ -418,11 +523,16 @@
418	523	require_once( ABSPATH . 'wp-admin/includes/file.php' );
419	524	WP_Filesystem();
420	525	global $wp_filesystem;
421	526
422		- $dir = $wp_filesystem->wp_content_dir() . 'uploads/visualizer';
423		- $file = $wp_filesystem->wp_content_dir() . 'uploads/visualizer/customization.js';
	527	+ $multisite_arg = '/';
	528	+ if ( is_multisite() && ! is_main_site() ) {
	529	+ $multisite_arg = '/sites/' . get_current_blog_id() . '/';
	530	+ }
424	531
	532	+ $dir = $wp_filesystem->wp_content_dir() . 'uploads' . $multisite_arg . 'visualizer';
	533	+ $file = $wp_filesystem->wp_content_dir() . 'uploads' . $multisite_arg . 'visualizer/customization.js';
	534	+
425	535	if ( $wp_filesystem->is_readable( $file ) ) {
426	536	return $specific;
427	537	}
428	538
		@@ -431,8 +541,9 @@
431	541	return $default;
432	542	}
433	543
434	544	if ( ! $wp_filesystem->exists( $dir ) ) {
	545	+ // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.Found
435	546	if ( ( $done = $wp_filesystem->mkdir( $dir ) ) === false ) {
436	547	do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'Unable to create directory %s', $dir ), 'error', __FILE__, __LINE__ );
437	548	return $default;
438	549	}
		@@ -440,8 +551,9 @@
440	551
441	552	// if file does not exist, copy.
442	553	if ( ! $wp_filesystem->exists( $file ) ) {
443	554	$src = str_replace( ABSPATH, $wp_filesystem->abspath(), VISUALIZER_ABSPATH . '/js/customization.js' );
	555	+ // phpcs:ignore WordPress.CodeAnalysis.AssignmentInCondition.Found
444	556	if ( ( $done = $wp_filesystem->copy( $src, $file ) ) === false ) {
445	557	do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'Unable to copy file %s to %s', $src, $file ), 'error', __FILE__, __LINE__ );
446	558	return $default;
447	559	}
		@@ -453,18 +565,21 @@
453	565	/**
454	566	* Load the class for the given chart's chart type so that its assets can be loaded.
455	567	*/
456	568	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 );
	569	+ $name = $this->load_chart_class_name( $chart_id );
459	570	$class = null;
460	571	if ( class_exists( $name ) || true === apply_filters( 'visualizer_load_chart', false, $name ) ) {
	572	+ if ( 'Visualizer_Render_Sidebar_Type_DataTable_DataTable' === $name ) {
	573	+ $name = 'Visualizer_Render_Sidebar_Type_DataTable_Tabular';
	574	+ }
461	575	$class = new $name;
462	576	}
463	577
464		- if ( is_null( $class ) && VISUALIZER_PRO ) {
	578	+ if ( is_null( $class ) && Visualizer_Module::is_pro() ) {
465	579	// 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 ) ) );
	580	+ $type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
	581	+ $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	582	}
468	583
469	584	return is_null( $class ) ? null : $class->getLibrary();
470	585	}
		@@ -469,8 +584,26 @@
469	584	return is_null( $class ) ? null : $class->getLibrary();
470	585	}
471	586
472	587	/**
	588	+ * Returns the class name for the given chart's chart type.
	589	+ */
	590	+ protected function load_chart_class_name( $chart_id ) {
	591	+ $type = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_TYPE, true );
	592	+ $lib = get_post_meta( $chart_id, Visualizer_Plugin::CF_CHART_LIBRARY, true );
	593	+
	594	+ // backward compatibility.
	595	+ if ( empty( $lib ) ) {
	596	+ $lib = 'GoogleCharts';
	597	+ if ( 'dataTable' === $type ) {
	598	+ $lib = 'DataTable';
	599	+ }
	600	+ }
	601	+ $name = 'Visualizer_Render_Sidebar_Type_' . $lib . '_' . ucwords( $type );
	602	+ return $name;
	603	+ }
	604	+
	605	+ /**
473	606	* Generates the inline CSS to apply to the chart and adds these classes to the settings.
474	607	*
475	608	* @access public
476	609	* @param int $id The id of the chart.
		@@ -478,32 +611,31 @@
478	611	*/
479	612	protected function get_inline_custom_css( $id, $settings ) {
480	613	$css = '';
481	614
482		- $arguments = array( '', $settings );
483		- if ( ! isset( $settings['customcss'] ) ) {
484		- return $arguments;
485		- }
486		-
487	615	$classes = array();
488	616	$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 );
	617	+ if ( ! empty( $settings['customcss'] ) ) {
	618	+ foreach ( $settings['customcss'] as $name => $element ) {
	619	+ $attributes = array();
	620	+ foreach ( $element as $property => $value ) {
	621	+ $attributes[] = $this->handle_css_property( $property, $value );
	622	+ }
	623	+ $class_name = $id . $name;
	624	+ $properties = implode( ' !important; ', array_filter( $attributes ) );
	625	+ if ( ! empty( $properties ) ) {
	626	+ $css .= '.' . $class_name . ' {' . $properties . ' !important;}';
	627	+ $classes[ $name ] = $class_name;
	628	+ }
493	629	}
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		- }
	630	+ $settings['cssClassNames'] = $classes;
500	631	}
501		- $css .= '</style>';
502	632
503		- $settings['cssClassNames'] = $classes;
	633	+ $img_path = VISUALIZER_ABSURL . 'images';
	634	+ $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;}";
	635	+ $css .= '</style>';
504	636
505		- $arguments = array( $css, $settings );
	637	+ $arguments = array( $css, $settings );
506	638	apply_filters_ref_array( 'visualizer_inline_css', array( &$arguments ) );
507	639
508	640	return $arguments;
509	641	}
		@@ -547,5 +679,156 @@
547	679	$q = new WP_Query( $args );
548	680	return $q->found_posts > 0;
549	681	}
550	682
	683	+ /**
	684	+ * Determines how many charts have been created.
	685	+ */
	686	+ protected static function numberOfCharts() {
	687	+ $args = array(
	688	+ 'post_type' => Visualizer_Plugin::CPT_VISUALIZER,
	689	+ 'fields' => 'ids',
	690	+ 'post_status' => 'publish',
	691	+ 'posts_per_page' => 300,
	692	+ );
	693	+
	694	+ $q = new WP_Query( $args );
	695	+ return $q->found_posts;
	696	+ }
	697	+
	698	+ /**
	699	+ * Checks if the PRO version is active.
	700	+ *
	701	+ * @since 3.3.0
	702	+ */
	703	+ public static function is_pro() {
	704	+ // versions of pro before 1.9.0 will use the constant VISUALIZER_PRO
	705	+ // versions of pro 1.9.0 onwards will use the filter
	706	+ return apply_filters( 'visualizer_is_pro', VISUALIZER_PRO );
	707	+ }
	708	+
	709	+ /**
	710	+ * Checks if the PRO version is older than a particular version.
	711	+ *
	712	+ * @since 3.3.0
	713	+ */
	714	+ public static function is_pro_older_than( $version ) {
	715	+ return version_compare( VISUALIZER_PRO_VERSION, $version, '<' );
	716	+ }
	717	+
	718	+ /**
	719	+ * Should we show some specific feature on the basis of the version?
	720	+ *
	721	+ * @since 3.4.0
	722	+ */
	723	+ public static function can_show_feature( $feature ) {
	724	+ switch ( $feature ) {
	725	+ case 'simple-editor':
	726	+ // if user has pro but an older version, then don't load the simple editor functionality
	727	+ // as the select box will not behave as expected because the pro editor's functionality will supercede.
	728	+ return ! Visualizer_Module::is_pro() || ! Visualizer_Module::is_pro_older_than( '1.9.2' );
	729	+ }
	730	+ return false;
	731	+ }
	732	+
	733	+ /**
	734	+ * Gets the features for the provided license type.
	735	+ */
	736	+ public static final function get_features_for_license( $plan ) {
	737	+ switch ( $plan ) {
	738	+ case 1:
	739	+ return array( 'import-wp', 'db-query', 'import-wc-report' );
	740	+ case 2:
	741	+ return array( 'schedule-chart', 'chart-permissions' );
	742	+ }
	743	+ }
	744	+
	745	+ /**
	746	+ * Gets the chart content after common manipulations.
	747	+ */
	748	+ public static function get_chart_data( $chart, $type, $run_filter = true ) {
	749	+ // change HTML entities
	750	+ $post_content = html_entity_decode( htmlentities( $chart->post_content ) );
	751	+ $post_content = preg_replace_callback(
	752	+ '!s:(\d+):"(.*?)";!s',
	753	+ function ( $matches ) {
	754	+ if ( isset( $matches[2] ) ) {
	755	+ return 's:' . strlen( $matches[2] ) . ':"' . $matches[2] . '";';
	756	+ }
	757	+ },
	758	+ $post_content
	759	+ );
	760	+ $data = unserialize( $post_content );
	761	+ $altered = array();
	762	+ if ( ! empty( $data ) ) {
	763	+ foreach ( $data as $index => $array ) {
	764	+ if ( ! is_array( $index ) && is_array( $array ) ) {
	765	+ foreach ( $array as &$datum ) {
	766	+ if ( is_string( $datum ) ) {
	767	+ $datum = stripslashes( $datum );
	768	+ }
	769	+ }
	770	+ $altered[ $index ] = $array;
	771	+ }
	772	+ }
	773	+ }
	774	+ // if something goes wrong and the end result is empty, be safe and use the original data
	775	+ if ( empty( $altered ) ) {
	776	+ $altered = $data;
	777	+ }
	778	+ if ( $run_filter ) {
	779	+ return apply_filters( Visualizer_Plugin::FILTER_GET_CHART_DATA, $altered, $chart->ID, $type );
	780	+ }
	781	+ return $altered;
	782	+ }
	783	+
	784	+ /**
	785	+ * Get chart image.
	786	+ *
	787	+ * @param object $chart Chart data.
	788	+ * @return string
	789	+ */
	790	+ public function _getImage( $chart = null ) {
	791	+ $image = '';
	792	+ if ( $chart ) {
	793	+ $chart_image = get_post_meta( $chart->ID, Visualizer_Plugin::CF_CHART_IMAGE, true );
	794	+ if ( ! empty( $chart_image ) ) {
	795	+ $image = wp_get_attachment_image( $chart_image, 'full' );
	796	+ }
	797	+ }
	798	+ return array(
	799	+ 'csv' => $image,
	800	+ );
	801	+ }
	802	+
	803	+ /**
	804	+ * Filter chart title if visualizer post type.
	805	+ *
	806	+ * @param object $query WP Query object.
	807	+ * @return void
	808	+ */
	809	+ public function PreGetPosts( $query ) {
	810	+ if ( ! $query->is_main_query() ) {
	811	+ $post_type = $query->get( 'post_type' );
	812	+ if ( 'visualizer' === $post_type ) {
	813	+ $this->_addFilter( Visualizer_Plugin::FILTER_CHART_TITLE, 'filterChartTitle', 10, 2 );
	814	+ }
	815	+ }
	816	+ }
	817	+
	818	+ /**
	819	+ * Filter chart title.
	820	+ *
	821	+ * @access public
	822	+ * @param string $post_title Post title.
	823	+ * @param int $post_id Post ID.
	824	+ * @return string
	825	+ */
	826	+ public function filterChartTitle( $post_title, $post_id ) {
	827	+ $post_type = get_post_type( $post_id );
	828	+ $post_title = trim( $post_title );
	829	+ if ( 'visualizer' === $post_type && 'Visualization' === $post_title ) {
	830	+ return sprintf( '%s #%d', $post_title, $post_id );
	831	+ }
	832	+ return $post_title;
	833	+ }
551	834	}