Show pre-selected data groups

packages/core/demo/data/bar.ts

@@ -36,6 +36,25 @@ export const groupedBarOptions = {
 	},
 };
 
+export const groupedBarSelectedGroupsData = groupedBarData;
+
+// Grouped bar with selected groups option
+export const groupedBarSelectedGroupsOptions = {
+	title: "Grouped bar (selected groups)",
+	data: {
+		selectedGroups: ["Dataset 1", "Dataset 3"]
+	},
+	axes: {
+		left: {
+			mapsTo: "value",
+		},
+		bottom: {
+			scaleType: "labels",
+			mapsTo: "key",
+		},
+	},
+};
+
 // Horizontal Grouped
 export const groupedHorizontalBarData = groupedBarData;
 

packages/core/demo/data/index.ts

@@ -176,6 +176,11 @@ let allDemoGroups = [
 				data: barDemos.groupedBarData,
 				chartType: chartTypes.GroupedBarChart
 			},
+			{
+				options: barDemos.groupedBarSelectedGroupsOptions,
+				data: barDemos.groupedBarSelectedGroupsData,
+				chartType: chartTypes.GroupedBarChart
+			},
 			{
 				options: barDemos.groupedBarEmptyStateOptions,
 				data: barDemos.groupedBarEmptyStateData,
@@ -377,6 +382,11 @@ let allDemoGroups = [
 				data: lineDemos.lineData,
 				chartType: chartTypes.LineChart,
 			},
+			{
+				options: lineDemos.lineSelectedGroupsOptions,
+				data: lineDemos.lineData,
+				chartType: chartTypes.LineChart,
+			},
 			{
 				options: lineDemos.lineTimeSeriesRotatedTicksOptions,
 				data: lineDemos.lineTimeSeriesDataRotatedTicks,

packages/core/demo/data/line.ts

@@ -85,7 +85,7 @@ export const lineCustomDomainOptions = {
 			title: "2019 Annual Sales Figures",
 			mapsTo: "key",
 			scaleType: "labels",
-			domain: ['Qty', 'More', 'Misc']
+			domain: ["Qty", "More", "Misc"]
 		},
 		left: {
 			domain: [10000, 50000],
@@ -96,6 +96,25 @@ export const lineCustomDomainOptions = {
 	},
 };
 
+export const lineSelectedGroupsOptions = {
+	title: "Line (selected groups)",
+	data: {
+		selectedGroups: ["Dataset 1", "Dataset 3"]
+	},
+	axes: {
+		bottom: {
+			title: "2019 Annual Sales Figures",
+			mapsTo: "key",
+			scaleType: "labels",
+		},
+		left: {
+			mapsTo: "value",
+			title: "Conversion rate",
+			scaleType: "linear",
+		},
+	},
+};
+
 export const lineTimeSeriesData = [
 	{ group: "Dataset 1", date: new Date(2019, 0, 1), value: 50000 },
 	{ group: "Dataset 1", date: new Date(2019, 0, 5), value: 65000 },

packages/core/src/components/essentials/legend.ts

@@ -27,7 +27,10 @@ export class Legend extends Component {
 		const addedLegendItems = legendItems
 			.enter()
 			.append("g")
-			.classed("legend-item", true);
+			.classed("legend-item", true)
+			.classed("active", function (d, i) {
+				return d.status === options.legend.items.status.ACTIVE;
+			});
 
 		// Configs
 		const checkboxRadius = options.legend.checkbox.radius;
@@ -208,14 +211,24 @@ export class Legend extends Component {
 				legendItem
 					.select("text")
 					.attr("x", startingPoint + spaceNeededForCheckbox)
-					.attr("y", yOffset + yPosition + 2);
+					.attr("y", yOffset + yPosition + 3);
 
 				lastYPosition = yPosition;
 
+				// Test if legendItems are placed in the correct direction
+				const testHorizontal = (!legendOrientation ||
+					legendOrientation === LegendOrientations.HORIZONTAL) &&
+					legendItem.select("rect.checkbox").attr("y") === '0';
+
+				const testVertical = legendOrientation === LegendOrientations.VERTICAL &&
+					legendItem.select("rect.checkbox").attr("x") === '0';
+
+				const hasCorrectLegendDirection = testHorizontal || testVertical;
+
 				// Render checkbox check icon
-				if (
-					hasDeactivatedItems &&
-					legendItem.select("g.check").empty()
+				if (hasDeactivatedItems &&
+					legendItem.select("g.check").empty() &&
+					hasCorrectLegendDirection
 				) {
 					legendItem.append("g").classed("check", true).html(`
 							<svg focusable="false" preserveAspectRatio="xMidYMid meet"
@@ -273,7 +286,6 @@ export class Legend extends Component {
 
 				// Configs
 				const checkboxRadius = options.legend.checkbox.radius;
-
 				const hoveredItem = select(this);
 				hoveredItem
 					.append("rect")

packages/core/src/components/layout/layout.ts

@@ -162,6 +162,15 @@ export class LayoutComponent extends Component {
 				select(this),
 				{ useBBox: true }
 			);
+
+			if (d.data.id === "legend") {
+				const svgSize = DOMUtils.getSVGElementSize(
+					select(this),
+					{ useAttrs: true }
+				);
+				matchingSVGDimensions.height = svgSize.height;
+			};
+
 			if (growth === LayoutGrowth.PREFERRED) {
 				const matchingSVGWidth = horizontal
 					? matchingSVGDimensions.width

packages/core/src/configuration.ts

@@ -169,7 +169,8 @@ const chart: BaseChartOptions = {
 	},
 	data: {
 		groupMapsTo: "group",
-		loading: false
+		loading: false,
+		selectedGroups: []
 	},
 	color: {
 		scale: null

packages/core/src/interfaces/charts.ts

@@ -97,6 +97,11 @@ export interface BaseChartOptions {
 		 * used to simulate data loading
 		 */
 		loading?: Boolean;
+		/**
+		 * options related to pre-selected data groups
+		 * Remains empty if every legend item is active or dataset doesn't have the data groups.
+		 */
+		selectedGroups?: string[];
 	};
 	/**
 	 * options related to color scales

packages/core/src/model.ts

@@ -39,12 +39,11 @@ export class ChartModel {
 		this.services = services;
 	}
 
-	getDisplayData() {
+	getAllDataFromDomain() {
 		if (!this.get("data")) {
 			return null;
 		}
 
-		const { ACTIVE } = Configuration.legend.items.status;
 		const dataGroups = this.getDataGroups();
 
 		// Remove datasets that have been disabled
@@ -76,6 +75,24 @@ export class ChartModel {
 		}
 
 		return displayData.filter((datum) => {
+			return dataGroups.find(
+				(group) => group.name === datum[groupMapsTo]
+			);
+		});
+	}
+
+	getDisplayData() {
+		if (!this.get("data")) {
+			return null;
+		}
+
+		const { ACTIVE } = Configuration.legend.items.status;
+		const dataGroups = this.getDataGroups();
+		const { groupMapsTo } = this.getOptions().data;
+
+		const allDataFromDomain = this.getAllDataFromDomain();
+
+		return allDataFromDomain.filter((datum) => {
 			const group = dataGroups.find(
 				(group) => group.name === datum[groupMapsTo]
 			);
@@ -338,6 +355,22 @@ export class ChartModel {
 			});
 		}
 
+		// Updates selected groups
+		const updatedActiveItems = dataGroups.filter(group => group.status === ACTIVE);
+		const options = this.getOptions();
+
+		const hasUpdatedDeactivatedItems = dataGroups.some(
+			group => group.status === DISABLED
+		);
+
+		// If there are deactivated items, map the item name into selected groups
+		if (hasUpdatedDeactivatedItems) {
+			options.data.selectedGroups = updatedActiveItems.map(activeItem => activeItem.name);
+		} else {
+			// If every item is active, clear array
+			options.data.selectedGroups = [];
+		};
+
 		// dispatch legend filtering event with the status of all the dataLabels
 		this.services.events.dispatchEvent(Events.Legend.ITEMS_UPDATE, {
 			dataGroups
@@ -478,15 +511,32 @@ export class ChartModel {
 
 	protected generateDataGroups(data) {
 		const { groupMapsTo } = this.getOptions().data;
-		const { ACTIVE } = Configuration.legend.items.status;
+		const { ACTIVE, DISABLED } = Configuration.legend.items.status;
+		const options = this.getOptions();
 
 		const uniqueDataGroups = map(
 			data,
 			(datum) => datum[groupMapsTo]
 		).keys();
-		return uniqueDataGroups.map((groupName) => ({
+
+		// check if selectedGroups can be applied to chart with current data groups
+		if (options.data.selectedGroups.length) {
+			const hasAllSelectedGroups = options.data.selectedGroups
+				.every(groupName => uniqueDataGroups.includes(groupName));
+			if (!hasAllSelectedGroups) {
+				options.data.selectedGroups = [];
+			};
+		}
+
+		// Get group status based on items in selected groups
+		const getStatus = (groupName) => 
+			!options.data.selectedGroups.length || options.data.selectedGroups.includes(groupName)
+				? ACTIVE
+				: DISABLED;
+
+		return uniqueDataGroups.map(groupName => ({
 			name: groupName,
-			status: ACTIVE
+			status: getStatus(groupName)
 		}));
 	}
 

packages/core/src/selectedGroups.spec.ts

@@ -0,0 +1,56 @@
+import { TestEnvironment } from "./tests/index";
+
+// import the settings for the css prefixes
+import settings from "carbon-components/es/globals/js/settings";
+
+import { options } from "./configuration";
+import { Events } from "./interfaces";
+
+import { select } from "d3-selection";
+
+describe("selectedGroups option", () => {
+	beforeEach(function () {
+		const testEnvironment = new TestEnvironment();
+		testEnvironment.render();
+
+		this.chart = testEnvironment.getChartReference();
+		this.testEnvironment = testEnvironment;
+	});
+
+	describe("selected legend labels", () => {
+		it("should match the selected groups provided in options", function (done) {
+			const sampleSelectedGroups = ["Dataset 1", "Dataset 3"];
+
+			const chartEventsService = this.chart.services.events;
+
+			const renderCb = () => {
+				// Remove render event listener
+				chartEventsService.removeEventListener(
+					Events.Chart.RENDER_FINISHED,
+					renderCb
+				);
+
+				const selectedLegendLabels = select(
+					`g.${settings.prefix}--${options.chart.style.prefix}--legend`
+				)
+					.selectAll("g.legend-item.active > text")
+					.nodes()
+					.map((item) => item["innerHTML"]);
+
+				expect(selectedLegendLabels).toEqual(sampleSelectedGroups);
+
+				done();
+			};
+
+			// Add event listener for when chart render is finished
+			chartEventsService.addEventListener(
+				Events.Chart.RENDER_FINISHED,
+				renderCb
+			);
+		});
+	});
+
+	afterEach(function () {
+		this.testEnvironment.destroy();
+	});
+});

packages/core/src/services/scales-cartesian.ts

@@ -336,6 +336,7 @@ export class CartesianScales extends Service {
 			return [];
 		}
 
+		const allDataFromDomain = this.model.getAllDataFromDomain();
 		const displayData = this.model.getDisplayData();
 		const { mapsTo, percentage } = axisOptions;
 
@@ -368,7 +369,7 @@ export class CartesianScales extends Service {
 				sum(values(dataValues) as any)
 			);
 		} else {
-			allDataValues = displayData.map((datum) => datum[mapsTo]);
+			allDataValues = allDataFromDomain.map((datum) => datum[mapsTo]);
 		}
 
 		if (scaleType !== ScaleTypes.TIME && includeZero) {

packages/core/src/tests/test-environment.ts

@@ -10,6 +10,9 @@ import { groupedBarData, groupedBarOptions } from "../../demo/data";
 export const data = groupedBarData as ChartData;
 export const options = Object.assign(groupedBarOptions, {
 	title: "My chart",
+	data: {
+		selectedGroups: ["Dataset 1", "Dataset 3"]
+	}
 }) as any;
 
 export class TestEnvironment {