fix microsoft#104102.

rebornix · rebornix · commit 408ad5d67dcc · 2020-08-05T12:13:53.000-07:00
diff --git a/src/vs/workbench/contrib/notebook/browser/contrib/coreActions.ts b/src/vs/workbench/contrib/notebook/browser/contrib/coreActions.ts
@@ -1497,3 +1497,16 @@ registerAction2(class extends NotebookCellAction {
 		context.notebookEditor.viewModel!.notebookDocument.changeCellMetadata(context.cell.handle, { inputCollapsed: false });
 	}
 });
+
+registerAction2(class extends NotebookCellAction {
+	constructor() {
+		super({
+			id: 'notebook.inspectLayout',
+			title: localize('notebookActions.inspectLayout', "Inspect Layout")
+		});
+	}
+
+	async runWithContext(accessor: ServicesAccessor, context: INotebookCellActionContext): Promise<void> {
+		context.notebookEditor.viewModel!.inspectLayout();
+	}
+});
diff --git a/src/vs/workbench/contrib/notebook/browser/view/renderers/backLayerWebView.ts b/src/vs/workbench/contrib/notebook/browser/view/renderers/backLayerWebView.ts
@@ -344,6 +344,13 @@ export class BackLayerWebView extends Disposable {
 			return;
 		}
 
+		const cell = this.insetMapping.get(output)!.cell;
+
+		const currCell = this.notebookEditor.viewModel?.viewCells.find(vc => vc.handle === cell.handle);
+		if (currCell !== cell && currCell !== undefined) {
+			this.insetMapping.get(output)!.cell = currCell as CodeCellViewModel;
+		}
+
 		return { cell: this.insetMapping.get(output)!.cell, output };
 	}
 
diff --git a/src/vs/workbench/contrib/notebook/browser/view/renderers/webviewPreloads.ts b/src/vs/workbench/contrib/notebook/browser/view/renderers/webviewPreloads.ts
@@ -449,6 +449,15 @@ function webviewPreloads() {
 					if (output) {
 						output.parentElement!.style.display = 'block';
 						output.style.top = top + 'px';
+
+						vscode.postMessage({
+							__vscode_notebook_message: true,
+							type: 'dimension',
+							id: outputId,
+							data: {
+								height: output.clientHeight
+							}
+						});
 					}
 				});
 				break;
diff --git a/src/vs/workbench/contrib/notebook/browser/viewModel/notebookViewModel.ts b/src/vs/workbench/contrib/notebook/browser/viewModel/notebookViewModel.ts
@@ -349,6 +349,16 @@ export class NotebookViewModel extends Disposable implements EditorFoldingStateD
 		});
 	}
 
+	inspectLayout() {
+		console.log('--- notebook ---\n');
+		console.log(this.layoutInfo);
+		console.log('--- cells ---');
+		this.viewCells.forEach(cell => {
+			console.log(`--- cell: ${cell.handle} ---\n`);
+			console.log((cell as (CodeCellViewModel | MarkdownCellViewModel)).layoutInfo);
+		});
+	}
+
 	setFocus(focused: boolean) {
 		this._focused = focused;
 	}

Original file line number	Diff line number	Diff line change
`@@ -344,6 +344,13 @@ export class BackLayerWebView extends Disposable {`
`344`	`344`	`return;`
`345`	`345`	`}`
`346`	`346`
	`347`	`+ const cell = this.insetMapping.get(output)!.cell;`
	`348`	`+`
	`349`	`+ const currCell = this.notebookEditor.viewModel?.viewCells.find(vc => vc.handle === cell.handle);`
	`350`	`+ if (currCell !== cell && currCell !== undefined) {`
	`351`	`+ this.insetMapping.get(output)!.cell = currCell as CodeCellViewModel;`
	`352`	`+ }`
	`353`	`+`
`347`	`354`	`return { cell: this.insetMapping.get(output)!.cell, output };`
`348`	`355`	`}`
`349`	`356`