From b14f1f263b787217ecb9fe8a1db00a8853b8f1ce Mon Sep 17 00:00:00 2001
From: hamed musallam <hamed.musallam@gmail.com>
Date: Wed, 27 Aug 2025 08:31:57 +0200
Subject: [PATCH] fix: make printing use the same handling logic as export

---
 src/component/main/NMRiumViewer.tsx | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/component/main/NMRiumViewer.tsx b/src/component/main/NMRiumViewer.tsx
index 29e3d0646b..120825db69 100644
--- a/src/component/main/NMRiumViewer.tsx
+++ b/src/component/main/NMRiumViewer.tsx
@@ -10,6 +10,7 @@ import { SVGRootContainer } from '../1d-2d/components/SVGRootContainer.js';
 import Viewer2D from '../2d/Viewer2D.js';
 import { useChartData } from '../context/ChartContext.js';
 import { useExportSettings } from '../elements/export/index.js';
+import { usePrintPage } from '../elements/print/PrintProvider.tsx';
 import {
   useCheckExportStatus,
   useViewportSize,
@@ -33,13 +34,26 @@ export function NMRiumViewer(props: NMRiumViewerProps) {
   const viewPort = useViewportSize();
   const { displayerMode } = useChartData();
   const exportSettings = useExportSettings();
+  const printSettings = usePrintPage();
 
   useOnRender(onRender);
 
   const isExportingProcessStart = useCheckExportStatus();
 
-  if (isExportingProcessStart && exportSettings) {
-    const { width, height, exportHeight, exportWidth } = exportSettings;
+  if (isExportingProcessStart && (exportSettings || printSettings)) {
+    if (!exportSettings && !printSettings) {
+      return <text>Print/export settings are missing.</text>;
+    }
+
+    const { width, height } = exportSettings || printSettings || {};
+    let exportHeight = height;
+    let exportWidth = width;
+
+    if (exportSettings) {
+      exportHeight = exportSettings.exportHeight;
+      exportWidth = exportSettings.exportWidth;
+    }
+
     return (
       <SVGRootContainer
         enableBoxBorder={displayerMode === '2D'}