Compare commits

...

10 Commits

Author SHA1 Message Date
Peter Stockings
65ffcb767f Add raw and filtered projection graphs 2025-05-22 21:58:50 +10:00
Peter Stockings
bcb7f1bce3 Improve look of app, ive broken accordion content so will need to fix 2025-05-18 18:04:04 +10:00
Peter Stockings
41c3c76218 Draw the angle overlay on sinogram 2025-05-18 12:07:57 +10:00
Peter Stockings
fe6c00c0fa Show both unfiltered and filtered sinogram 2025-05-18 11:49:37 +10:00
Peter Stockings
9f12966b12 Add function that applies different filters 2025-05-18 11:40:14 +10:00
Peter Stockings
26e860d168 Make generated singogram easier to see 2025-05-17 09:52:48 +10:00
Peter Stockings
ad74375567 Change default numbers of angles to 40, to speed it up 2025-05-17 09:40:03 +10:00
Peter Stockings
7541a472ed Move files to public folder 2025-05-17 09:37:05 +10:00
Peter Stockings
d8eaf675cf Update static.json 2025-05-17 09:35:05 +10:00
Peter Stockings
f12aa03fc5 Attempt to resolve static deployment error 2025-05-17 09:29:26 +10:00
11 changed files with 1192 additions and 369 deletions

1
.buildpacks Normal file
View File

@@ -0,0 +1 @@
https://github.com/heroku/heroku-buildpack-static.git

View File

View File

@@ -1,326 +0,0 @@
import {
generateSinogram,
reconstructImageFromSinogram,
convertToGrayscale,
} from "./sinogram.js";
import { StepAccordion } from "./CTAccordion .js";
export const UploadImageComponent = {
hasLoadedInitialImage: false,
angleCount: 180,
imageUrl:
"https://upload.wikimedia.org/wikipedia/commons/e/e5/Shepp_logan.png",
sinogramUrl: null,
reconstructedUrl: null,
defaultImageUrl:
"https://upload.wikimedia.org/wikipedia/commons/e/e5/Shepp_logan.png",
reconstructionFrames: [],
currentFrameIndex: 0,
renderMode: "grayscale", // or "heatmap"
useFBP: true,
drawAngleOverlay(theta) {
const canvas = this.overlayCanvas;
if (!canvas || !this.imageElement) return;
const ctx = canvas.getContext("2d");
const w = canvas.width;
const h = canvas.height;
const cx = w / 2;
const cy = h / 2;
const len = Math.max(w, h);
ctx.clearRect(0, 0, w, h);
const dx = len * Math.cos(theta);
const dy = len * Math.sin(theta);
ctx.strokeStyle = "rgba(255,0,0,0.8)";
ctx.lineWidth = 2;
ctx.beginPath();
ctx.moveTo(cx - dx, cy - dy);
ctx.lineTo(cx + dx, cy + dy);
ctx.stroke();
},
isOverlayReady() {
return (
this.overlayCanvas &&
this.imageElement &&
this.imageElement.complete &&
this.imageElement.naturalWidth > 0
);
},
async loadAndProcess(url, isUploaded = false) {
this.imageUrl = url;
this.sinogramUrl = "loading";
this.reconstructedUrl = null;
m.redraw();
this.imageUrl = url;
let finalUrl = url;
if (isUploaded) {
finalUrl = await convertToGrayscale(url);
this.imageUrl = finalUrl;
}
this.sinogramUrl = await generateSinogram(
finalUrl,
this.angleCount,
this.drawAngleOverlay.bind(this)
);
m.redraw();
this.reconstructionFrames = [];
this.currentFrameIndex = 0;
this.reconstructedUrl = await reconstructImageFromSinogram(
this.sinogramUrl,
undefined,
(angle, frameUrl) => {
this.reconstructionFrames.push(frameUrl);
this.currentFrameIndex = this.reconstructionFrames.length - 1;
this.reconstructedUrl = frameUrl;
m.redraw();
},
this.renderMode,
this.useFBP
);
},
oninit() {
this.loadAndProcessDebounced = debounce((url) => {
this.loadAndProcess(url);
}, 300);
},
view() {
return m(
"div",
{
class: "flex flex-col items-center min-h-screen bg-gray-100 py-10 px-4",
},
[
// Header
m("header", { class: "mb-10 text-center" }, [
m(
"h1",
{ class: "text-4xl font-bold text-gray-800 mb-2" },
"Sinogram Generator"
),
m(
"p",
{ class: "text-gray-600 text-lg" },
"Upload a grayscale image to simulate CT scan projections"
),
]),
m(StepAccordion, { index: 0 }),
// Upload Box
m(
"div",
{
class:
"w-full max-w-lg border-4 border-dashed border-gray-400 bg-white rounded-xl p-6 text-center hover:bg-gray-50 cursor-pointer transition",
ondragover: (e) => e.preventDefault(),
ondrop: (e) => {
e.preventDefault();
const file = e.dataTransfer.files[0];
if (file && file.type.startsWith("image/")) {
const url = URL.createObjectURL(file);
this.loadAndProcess(url, true);
}
},
onclick: () => document.getElementById("fileInput").click(),
},
[
m(
"p",
{ class: "text-gray-500" },
"Click or drag a grayscale image here"
),
m("input", {
id: "fileInput",
type: "file",
class: "hidden",
accept: "image/*",
onchange: (e) => {
const file = e.target.files[0];
if (file && file.type.startsWith("image/")) {
const url = URL.createObjectURL(file);
this.loadAndProcess(url);
}
},
}),
]
),
m(StepAccordion, { index: 1 }),
// Image Preview
m("div", { class: "relative mt-6 w-full max-w-md" }, [
m("img", {
src: this.imageUrl,
class: "rounded shadow max-w-full h-auto mx-auto",
onload: (e) => {
this.imageElement = e.target;
// Only start once both image and canvas are ready
if (this.isOverlayReady() && !this.hasLoadedInitialImage) {
this.hasLoadedInitialImage = true;
this.loadAndProcess(this.imageUrl);
}
},
}),
m("canvas", {
width: this.imageElement?.width || 0,
height: this.imageElement?.height || 0,
style: "position:absolute; top:0; left:0; pointer-events:none;",
oncreate: ({ dom }) => {
this.overlayCanvas = dom;
// Trigger load if image was already ready
if (this.isOverlayReady() && !this.hasLoadedInitialImage) {
this.hasLoadedInitialImage = true;
this.loadAndProcess(this.imageUrl);
}
},
}),
]),
// Angle Slider
m("div", { class: "mt-6 w-full max-w-md" }, [
m(
"label",
{ class: "block text-sm font-medium text-gray-700 mb-1" },
`Number of angles: ${this.angleCount}`
),
m("input", {
type: "range",
min: 5,
max: 360,
value: this.angleCount,
step: 1,
class: "w-full",
oninput: (e) => {
this.angleCount = parseInt(e.target.value, 10);
this.loadAndProcessDebounced(this.imageUrl); // reprocess with new angle count
},
}),
]),
m(StepAccordion, { index: 2 }),
// Sinogram
this.sinogramUrl &&
m("div", { class: "mt-10 w-full max-w-md text-center" }, [
m(
"h2",
{ class: "text-xl font-semibold text-gray-700 mb-4" },
"Generated Sinogram"
),
this.sinogramUrl === "loading"
? m("p", "Processing...")
: m("img", {
src: this.sinogramUrl,
alt: "Sinogram",
class: "rounded shadow max-w-full h-auto mx-auto",
}),
]),
m(StepAccordion, { index: 3 }),
// Reconstructed
this.reconstructedUrl &&
m("div", { class: "mt-10 w-full max-w-md text-center" }, [
m(
"h2",
{ class: "text-xl font-semibold text-gray-700 mb-4" },
"Reconstructed Image (Back Projection)"
),
m("img", {
src: this.reconstructionFrames[this.currentFrameIndex],
alt: "Reconstructed",
class: "rounded shadow max-w-full h-auto mx-auto",
}),
m("div", { class: "mt-6 w-full max-w-md text-center" }, [
m(
"label",
{ class: "text-sm text-gray-600 mr-2" },
"Render style:"
),
m(
"select",
{
value: this.renderMode,
onchange: (e) => {
this.renderMode = e.target.value;
this.loadAndProcess(this.imageUrl); // re-render using selected mode
},
},
[
m("option", { value: "heatmap" }, "Heatmap"),
m("option", { value: "grayscale" }, "Grayscale"),
]
),
]),
m("div", { class: "mt-4 w-full max-w-md text-left" }, [
m("label", [
m("input", {
type: "checkbox",
checked: this.useFBP,
onchange: (e) => {
this.useFBP = e.target.checked;
this.loadAndProcess(this.imageUrl); // regenerate with or without FBP
},
}),
m(
"span",
{ class: "ml-2 text-gray-700" },
"Use Filtered Back Projection (Ramp)"
),
]),
]),
this.reconstructionFrames.length > 1 &&
m("div", { class: "mt-4" }, [
m("input", {
type: "range",
min: 0,
max: this.reconstructionFrames.length - 1,
value: this.currentFrameIndex,
step: 1,
oninput: (e) => {
this.currentFrameIndex = +e.target.value;
this.reconstructedUrl =
this.reconstructionFrames[this.currentFrameIndex];
},
}),
m(
"p",
{ class: "text-sm text-gray-500 mt-1" },
`Angle ${this.currentFrameIndex + 1} / ${
this.reconstructionFrames.length
}`
),
]),
m(StepAccordion, { index: 4 }),
]),
]
);
},
};
function debounce(fn, delay) {
let timeout;
return (...args) => {
clearTimeout(timeout);
timeout = setTimeout(() => fn(...args), delay);
};
}

View File

@@ -97,12 +97,19 @@ export const StepAccordion = {
},
};
// You must provide a markdownToHTML() function or use a library like marked.js
// Use marked.js for robust Markdown to HTML conversion
const renderer = new marked.Renderer();
renderer.image = (href, title, text) => {
// Add Tailwind classes for styling images
return `<img src="${href}" alt="${text}" title="${
title || ""
}" class="my-2 rounded shadow max-w-full h-auto mx-auto">`;
};
marked.setOptions({ renderer });
function markdownToHTML(text) {
return text
.replace(/\n/g, "<br>")
.replace(
/!\[(.*?)\]\((.*?)\)/g,
'<img alt="$1" src="$2" class="my-2 rounded shadow max-w-full">'
);
// Replace escaped newlines from template literals with actual newlines for marked
const processedText = text.replace(/\\n/g, "\n");
return marked.parse(processedText);
}

View File

@@ -0,0 +1,160 @@
export const ProjectionGraphComponent = {
view: function (vnode) {
const {
data,
title,
width = 300,
height = 150,
color = "steelblue",
} = vnode.attrs;
if (!data || data.length === 0) {
return m(
"div.projection-graph-container",
{
style: {
width: `${width}px`,
height: `${height}px`,
border: "1px solid #ccc",
display: "flex",
"align-items": "center",
"justify-content": "center",
"background-color": "#f9f9f9",
"margin-bottom": "10px",
},
},
m("p.text-gray-500", title ? `${title}: No data` : "No data available")
);
}
const padding = { top: 20, right: 20, bottom: 30, left: 40 };
const chartWidth = width - padding.left - padding.right;
const chartHeight = height - padding.top - padding.bottom;
const maxVal = Math.max(...data, 0); // Ensure maxVal is at least 0
const minVal = Math.min(...data, 0); // Ensure minVal is at most 0
// Adjust scale if all values are 0 or very close to 0
let yMax = maxVal;
let yMin = minVal;
if (maxVal === 0 && minVal === 0) {
yMax = 1; // Avoid division by zero if all data points are 0
yMin = -1;
} else if (maxVal === minVal) {
// All values are the same but not zero
yMax = maxVal + Math.abs(maxVal * 0.1) + 1; // Add some padding
yMin = minVal - Math.abs(minVal * 0.1) - 1;
}
const xScale = (index) => (index / (data.length - 1)) * chartWidth;
const yScale = (value) =>
chartHeight - ((value - yMin) / (yMax - yMin)) * chartHeight;
const linePath = data
.map((d, i) => `${i === 0 ? "M" : "L"}${xScale(i)},${yScale(d)}`)
.join(" ");
// Generate Y-axis ticks (e.g., 3 ticks: min, mid, max)
const yTicks = [];
if (yMin !== yMax) {
yTicks.push({ value: yMin, y: yScale(yMin) });
yTicks.push({ value: (yMin + yMax) / 2, y: yScale((yMin + yMax) / 2) });
yTicks.push({ value: yMax, y: yScale(yMax) });
} else {
// Handle case where all values are the same
yTicks.push({ value: yMin - 1, y: yScale(yMin - 1) });
yTicks.push({ value: yMin, y: yScale(yMin) });
yTicks.push({ value: yMin + 1, y: yScale(yMin + 1) });
}
return m(
"div.projection-graph-container",
{
style: {
"margin-bottom": "15px",
padding: "10px",
border: "1px solid #e2e8f0",
"border-radius": "0.375rem",
"background-color": "#fff",
},
},
[
m(
"h4.text-md.font-semibold.text-gray-700.mb-2.text-center",
title || "Projection Data"
),
m("svg", { width: width, height: height }, [
m("g", { transform: `translate(${padding.left}, ${padding.top})` }, [
// Y-axis
m("line", { x1: 0, y1: 0, x2: 0, y2: chartHeight, stroke: "#ccc" }),
yTicks.map((tick) =>
m("g", { transform: `translate(0, ${tick.y})` }, [
m("line", { x1: -5, y1: 0, x2: 0, y2: 0, stroke: "#ccc" }),
m(
"text",
{
x: -10,
y: 0,
"font-size": "10px",
"text-anchor": "end",
dy: ".32em",
},
parseFloat(tick.value).toFixed(1)
),
])
),
// X-axis
m("line", {
x1: 0,
y1: chartHeight,
x2: chartWidth,
y2: chartHeight,
stroke: "#ccc",
}),
// X-axis labels (optional, e.g., start, mid, end)
m(
"text",
{
x: 0,
y: chartHeight + 15,
"font-size": "10px",
"text-anchor": "start",
},
"0"
),
m(
"text",
{
x: chartWidth / 2,
y: chartHeight + 15,
"font-size": "10px",
"text-anchor": "middle",
},
`${Math.floor(data.length / 2)}`
),
m(
"text",
{
x: chartWidth,
y: chartHeight + 15,
"font-size": "10px",
"text-anchor": "end",
},
`${data.length - 1}`
),
// Data line
m("path", {
d: linePath,
stroke: color,
"stroke-width": 2,
fill: "none",
}),
]),
]),
]
);
},
};

View File

@@ -0,0 +1,855 @@
import {
generateSinogram,
reconstructImageFromSinogram,
convertToGrayscale,
} from "./sinogram.js";
import { StepAccordion } from "./CTAccordion .js";
import { ProjectionGraphComponent } from "./ProjectionGraphComponent.js";
export const UploadImageComponent = {
hasLoadedInitialImage: false,
angleCount: 40,
imageUrl:
"https://upload.wikimedia.org/wikipedia/commons/e/e5/Shepp_logan.png",
sinogramUrl: null,
filteredSinogramUrl: null, // Added to store the filtered sinogram
reconstructedUrl: null,
defaultImageUrl:
"https://upload.wikimedia.org/wikipedia/commons/e/e5/Shepp_logan.png",
reconstructionFrames: [],
currentFrameIndex: 0,
manualAngleIndex: 0, // For the new angle control slider
sinogramImageElement: null, // To store the sinogram img DOM element
sinogramHighlightCanvas: null, // Canvas for sinogram highlight
rawProjections: [], // To store raw projections from generateSinogram
filteredProjections: [], // To store filtered projections from reconstructImageFromSinogram
currentRawProjectionData: null, // Data for the raw projection graph
currentFilteredProjectionData: null, // Data for the filtered projection graph
renderMode: "grayscale", // or "heatmap"
filterType: "ramp", // Added to manage selected filter
drawAngleOverlay(theta) {
const canvas = this.overlayCanvas;
if (!canvas || !this.imageElement) return;
const ctx = canvas.getContext("2d");
const w = canvas.width;
const h = canvas.height;
const cx = w / 2;
const cy = h / 2;
const len = Math.max(w, h);
ctx.clearRect(0, 0, w, h);
const dx = len * Math.cos(theta);
const dy = len * Math.sin(theta);
ctx.strokeStyle = "rgba(255,0,0,0.8)";
ctx.lineWidth = 2;
ctx.beginPath();
ctx.moveTo(cx - dx, cy - dy);
ctx.lineTo(cx + dx, cy + dy);
ctx.stroke();
},
drawSinogramHighlight(angleIndex) {
if (
!this.sinogramHighlightCanvas ||
!this.sinogramImageElement ||
!this.sinogramImageElement.complete ||
this.sinogramImageElement.naturalWidth === 0
) {
if (this.sinogramHighlightCanvas) {
const ctx = this.sinogramHighlightCanvas.getContext("2d");
ctx.clearRect(
0,
0,
this.sinogramHighlightCanvas.width,
this.sinogramHighlightCanvas.height
);
}
return;
}
const canvas = this.sinogramHighlightCanvas;
const ctx = canvas.getContext("2d");
const dispWidth = this.sinogramImageElement.width; // Displayed width of sinogram image
const dispHeight = this.sinogramHighlightCanvas.height; // Use canvas display height
const naturalW = this.sinogramImageElement.naturalWidth;
const naturalH = this.sinogramImageElement.naturalHeight;
if (canvas.width !== dispWidth) canvas.width = dispWidth;
// Ensure canvas height matches the displayed sinogram image's aspect ratio, or a fixed value if not loaded
const targetCanvasHeight =
this.sinogramImageElement.complete && naturalH > 0
? (dispWidth * naturalH) / naturalW
: 150;
if (canvas.height !== targetCanvasHeight)
canvas.height = targetCanvasHeight;
ctx.clearRect(0, 0, canvas.width, canvas.height);
if (this.angleCount <= 0 || naturalW === 0 || naturalH === 0) return;
let anglesAlongImageHeight = true;
if (naturalW === this.angleCount && naturalH !== this.angleCount) {
anglesAlongImageHeight = false;
} else if (naturalH === this.angleCount && naturalW !== this.angleCount) {
anglesAlongImageHeight = true;
} else if (naturalH === this.angleCount && naturalW === this.angleCount) {
anglesAlongImageHeight = true;
}
ctx.strokeStyle = "rgba(0, 255, 0, 0.7)";
if (anglesAlongImageHeight) {
const highlightRowHeight = canvas.height / this.angleCount;
const yPos = (angleIndex + 0.5) * highlightRowHeight;
ctx.lineWidth = Math.max(1, highlightRowHeight * 0.7);
ctx.beginPath();
ctx.moveTo(0, yPos);
ctx.lineTo(canvas.width, yPos);
ctx.stroke();
} else {
const highlightColWidth = canvas.width / this.angleCount;
const xPos = (angleIndex + 0.5) * highlightColWidth;
ctx.lineWidth = Math.max(1, highlightColWidth * 0.7);
ctx.beginPath();
ctx.moveTo(xPos, 0);
ctx.lineTo(xPos, canvas.height);
ctx.stroke();
}
},
isOverlayReady() {
return (
this.overlayCanvas &&
this.imageElement &&
this.imageElement.complete &&
this.imageElement.naturalWidth > 0
);
},
async loadAndProcess(url, isUploaded = false) {
this.imageUrl = url;
this.sinogramUrl = "loading";
this.filteredSinogramUrl = null;
this.reconstructedUrl = "loading"; // Indicate reconstruction is also loading
this.manualAngleIndex = 0;
this.rawProjections = [];
this.filteredProjections = [];
this.currentRawProjectionData = null;
this.currentFilteredProjectionData = null;
m.redraw();
let finalUrl = url;
if (isUploaded) {
finalUrl = await convertToGrayscale(url);
this.imageUrl = finalUrl; // Update imageUrl to the grayscaled one
}
// Ensure imageElement is set for overlay drawing if using default URL
if (!isUploaded && !this.imageElement && this.defaultImageUrl === url) {
// Attempt to get the image element if it's the default one and not yet set
// This might happen if loadAndProcess is called before the default image's onload
const img = document.querySelector(`img[src="${url}"]`);
if (img) this.imageElement = img;
}
const sinogramResult = await generateSinogram(
finalUrl,
this.angleCount,
this.drawAngleOverlay.bind(this)
);
this.sinogramUrl = sinogramResult.sinogramUrl;
this.rawProjections = sinogramResult.projections; // Store raw projections
if (
this.rawProjections &&
this.rawProjections.length > this.manualAngleIndex
) {
this.currentRawProjectionData =
this.rawProjections[this.manualAngleIndex];
}
m.redraw();
this.reconstructionFrames = [];
this.currentFrameIndex = 0;
const reconstructionResult = await reconstructImageFromSinogram(
this.sinogramUrl,
this.imageElement
? Math.max(
this.imageElement.naturalWidth,
this.imageElement.naturalHeight
)
: 256, // Pass original image size
(angle, frameUrl) => {
this.reconstructionFrames.push(frameUrl);
this.currentFrameIndex = this.reconstructionFrames.length - 1;
this.reconstructedUrl = frameUrl;
m.redraw();
},
this.renderMode,
this.filterType
);
this.reconstructedUrl = reconstructionResult.reconstructedUrl;
this.filteredSinogramUrl = reconstructionResult.filteredSinogramUrl;
// Assuming reconstructImageFromSinogram now returns allFilteredProjections
this.filteredProjections =
reconstructionResult.allFilteredProjections || [];
if (
this.filteredProjections &&
this.filteredProjections.length > this.manualAngleIndex
) {
this.currentFilteredProjectionData =
this.filteredProjections[this.manualAngleIndex];
}
// It seems rawProjectionsFromSinogram is also returned, let's decide which raw projections to use.
// For now, sticking with the ones from generateSinogram for the "raw projection" graph.
// If `reconstructionResult.rawProjectionsFromSinogram` is preferred, adjust here.
m.redraw();
},
oninit() {
this.loadAndProcessDebounced = debounce((url, isUploaded = false) => {
// Pass isUploaded to debounced function
this.loadAndProcess(url, isUploaded);
}, 500); // Increased debounce time slightly
// Load initial image after a brief delay to ensure DOM is ready for canvas
// and imageElement to be potentially available.
setTimeout(() => {
if (!this.hasLoadedInitialImage && this.defaultImageUrl) {
this.loadAndProcess(this.defaultImageUrl, false); // Process default image (not uploaded)
this.hasLoadedInitialImage = true; // Prevent re-loading
}
}, 100);
},
view() {
return m(
"div",
{
class:
"flex flex-col items-center min-h-screen bg-gray-200 py-8 px-4 sm:px-6 lg:px-8", // Slightly darker bg, more padding
},
[
// Header
m(
"header",
{
class:
"mb-8 pb-6 border-b border-gray-300 text-center w-full max-w-3xl",
},
[
m(
"h1",
{ class: "text-3xl sm:text-4xl font-bold text-gray-800 mb-3" }, // Responsive text size
"Sinogram & CT Reconstruction Explorer"
),
m(
"p",
{ class: "text-gray-600 text-md sm:text-lg" }, // Responsive text size
"Upload an image, see its sinogram, and explore the reconstruction process."
),
]
),
// --- Section 1: Upload, Original Image, and Initial Controls ---
m(
"div",
{ class: "bg-white shadow-xl rounded-lg p-6 my-6 w-full max-w-xl" },
[
m(StepAccordion, { index: 0 }), // "1. Upload / Input Image"
// Upload Box
m(
"div",
{
class:
"mt-4 w-full border-4 border-dashed border-gray-300 hover:border-blue-500 bg-gray-50 rounded-xl p-6 sm:p-8 text-center hover:bg-gray-100 cursor-pointer transition duration-150 ease-in-out", // Added duration
ondragover: (e) => e.preventDefault(),
ondrop: (e) => {
e.preventDefault();
const file = e.dataTransfer.files[0];
if (file && file.type.startsWith("image/")) {
const url = URL.createObjectURL(file);
this.loadAndProcessDebounced(url, true); // Use debounced version
}
},
onclick: () => document.getElementById("fileInput").click(),
},
[
m(
"svg",
{
class: "mx-auto h-12 w-12 text-gray-400",
stroke: "currentColor",
fill: "none",
viewBox: "0 0 48 48",
"aria-hidden": "true",
},
m("path", {
d: "M28 8H12a4 4 0 00-4 4v20m32-12v8m0 0v8a4 4 0 01-4 4H12a4 4 0 01-4-4v-4m32-4l-3.172-3.172a4 4 0 00-5.656 0L28 28M8 32l9.172-9.172a4 4 0 015.656 0L28 28m0 0l4 4m4-24h8m-4-4v8m-12 4h.02",
"stroke-width": "2",
"stroke-linecap": "round",
"stroke-linejoin": "round",
})
),
m(
"p",
{ class: "mt-2 text-sm font-medium text-gray-700" }, // Slightly bolder text
"Click to upload or drag and drop an image"
),
m(
"p",
{ class: "text-xs text-gray-500 mt-1" },
"PNG, JPG, GIF recommended"
),
m("input", {
id: "fileInput",
type: "file",
class: "hidden",
accept: "image/*",
onchange: (e) => {
const file = e.target.files[0];
if (file && file.type.startsWith("image/")) {
const url = URL.createObjectURL(file);
this.loadAndProcessDebounced(url, true); // Use debounced version
}
},
}),
]
),
m("div", { class: "mt-4 text-center" }, [
m(
"button",
{
class: "text-sm text-blue-600 hover:text-blue-800 underline",
onclick: () => {
this.loadAndProcessDebounced(this.defaultImageUrl, false); // Load default, not as uploaded
},
},
"Load Default Shepp-Logan Phantom"
),
]),
m(StepAccordion, { index: 1 }),
// Image Preview
m(
"div",
{ class: "relative mt-6 border rounded-md p-4 bg-gray-50" },
[
m(
"h3",
{
class:
"text-lg font-semibold text-gray-700 mb-3 text-center",
}, // Bolder
"Original Image"
),
m(
"div",
{
class:
"relative w-64 h-64 mx-auto bg-gray-200 flex items-center justify-center rounded",
},
[
// Container for image and canvas
m("img", {
src: this.imageUrl,
class:
"rounded shadow-md max-w-full max-h-full h-auto mx-auto block", // max-h-full
// key: this.imageUrl, // Removed to fix Mithril key error
onload: (e) => {
this.imageElement = e.target;
// Adjust canvas to image size
if (this.overlayCanvas) {
this.overlayCanvas.width =
this.imageElement.naturalWidth;
this.overlayCanvas.height =
this.imageElement.naturalHeight;
// Update style to match image display size if different from natural
this.overlayCanvas.style.width = `${this.imageElement.width}px`;
this.overlayCanvas.style.height = `${this.imageElement.height}px`;
}
if (this.isOverlayReady()) {
// Check if overlay can be drawn
this.drawAngleOverlay(
(this.manualAngleIndex * Math.PI) / this.angleCount
);
}
if (
!this.hasLoadedInitialImage &&
this.imageUrl === this.defaultImageUrl
) {
this.hasLoadedInitialImage = true; // Mark that initial (default) image has been processed by onload
}
},
onerror: () => {
// Handle image load error, e.g., show a placeholder or message
if (this.overlayCanvas) {
// Clear overlay if image fails
const ctx = this.overlayCanvas.getContext("2d");
ctx.clearRect(
0,
0,
this.overlayCanvas.width,
this.overlayCanvas.height
);
}
},
}),
m("canvas", {
// Overlay Canvas
// Width/height set by onload of image or default
class:
"absolute top-1/2 left-1/2 transform -translate-x-1/2 -translate-y-1/2 pointer-events-none",
oncreate: ({ dom }) => {
this.overlayCanvas = dom;
// Attempt to set initial size based on current imageElement if available
if (this.imageElement && this.imageElement.complete) {
dom.width = this.imageElement.naturalWidth;
dom.height = this.imageElement.naturalHeight;
dom.style.width = `${this.imageElement.width}px`;
dom.style.height = `${this.imageElement.height}px`;
if (this.isOverlayReady()) {
this.drawAngleOverlay(
(this.manualAngleIndex * Math.PI) /
this.angleCount
);
}
} else {
// Default placeholder size if no image yet
dom.width = 256;
dom.height = 256;
dom.style.width = `256px`;
dom.style.height = `256px`;
}
},
}),
]
),
]
),
// Angle Slider
m("div", { class: "mt-6" }, [
m(
"label",
{
for: "angleCountSlider",
class: "block text-sm font-medium text-gray-700 mb-1",
},
`Number of Angles: ${this.angleCount}`
),
m("input", {
id: "angleCountSlider",
type: "range",
min: 5,
max: 360,
value: this.angleCount,
step: 1,
class:
"w-full h-2 bg-gray-300 rounded-lg appearance-none cursor-pointer accent-blue-600", // Styled slider
oninput: (e) => {
this.angleCount = parseInt(e.target.value, 10);
this.manualAngleIndex = 0;
const newTheta = (0 * Math.PI) / this.angleCount; // Angle for index 0
this.drawAngleOverlay(newTheta);
if (this.sinogramHighlightCanvas) {
this.drawSinogramHighlight(this.manualAngleIndex);
}
this.loadAndProcessDebounced(
this.imageUrl,
this.imageUrl !== this.defaultImageUrl
); // Pass isUploaded
},
}),
]),
// Manual Angle Slider for Preview
m("div", { class: "mt-4" }, [
m(
"label",
{
for: "manualAngleSlider",
class: "block text-sm font-medium text-gray-700 mb-1",
},
`Preview Angle: ${this.manualAngleIndex} (θ ≈ ${(
(this.manualAngleIndex * 180) /
(this.angleCount || 1)
) // Avoid division by zero
.toFixed(1)}°)`
),
m("input", {
id: "manualAngleSlider",
type: "range",
min: 0,
max: Math.max(0, this.angleCount - 1),
value: this.manualAngleIndex,
step: 1,
class:
"w-full h-2 bg-gray-300 rounded-lg appearance-none cursor-pointer accent-blue-600", // Styled slider
disabled: !this.imageUrl || this.sinogramUrl === "loading", // Disable if no image or loading
oninput: (e) => {
this.manualAngleIndex = parseInt(e.target.value, 10);
const theta =
(this.manualAngleIndex * Math.PI) / (this.angleCount || 1);
this.drawAngleOverlay(theta);
this.drawSinogramHighlight(this.manualAngleIndex);
// Update current projection data for graphs
if (
this.rawProjections &&
this.rawProjections.length > this.manualAngleIndex
) {
this.currentRawProjectionData =
this.rawProjections[this.manualAngleIndex];
} else {
this.currentRawProjectionData = null;
}
if (
this.filteredProjections &&
this.filteredProjections.length > this.manualAngleIndex
) {
this.currentFilteredProjectionData =
this.filteredProjections[this.manualAngleIndex];
} else {
this.currentFilteredProjectionData = null;
}
// Redraw will be handled by Mithril automatically if currentRawProjectionData/currentFilteredProjectionData are used in view
},
}),
]),
// Projection Graphs
(this.currentRawProjectionData ||
this.currentFilteredProjectionData) &&
m("div.mt-6.flex.flex-col.space-y-4", [
this.currentRawProjectionData &&
m(ProjectionGraphComponent, {
title: "Current Raw Projection",
data: this.currentRawProjectionData,
width: 450, // Adjusted width
height: 200, // Adjusted height
color: "dodgerblue",
}),
this.currentFilteredProjectionData &&
m(ProjectionGraphComponent, {
title: "Current Filtered Projection",
data: this.currentFilteredProjectionData,
width: 450, // Adjusted width
height: 200, // Adjusted height
color: "orangered",
}),
]),
// Filter Type Selector
m("div", { class: "mt-6" }, [
// Using mt-6 for spacing
m(
"label",
{
for: "filterType",
class: "block text-sm font-medium text-gray-700 mb-1",
},
"Filter Type:"
),
m(
"select",
{
id: "filterType",
class:
"mt-1 block w-full pl-3 pr-10 py-2 text-base border-gray-300 focus:outline-none focus:ring-blue-500 focus:border-blue-500 sm:text-sm rounded-md",
value: this.filterType,
onchange: (e) => {
this.filterType = e.target.value;
// Reload and reprocess when filter changes
this.loadAndProcessDebounced(
this.imageUrl,
this.imageUrl !== this.defaultImageUrl
);
},
},
[
m("option", { value: "none" }, "None"),
m("option", { value: "ramp" }, "Ramp (Shepp-Logan)"),
m("option", { value: "cosine" }, "Cosine"),
m("option", { value: "hamming" }, "Hamming"),
m("option", { value: "hann" }, "Hann"),
]
),
]),
]
),
// --- Section 2: Sinogram Generation & Filtering ---
(this.sinogramUrl || this.filteredSinogramUrl) &&
m(
"div",
{ class: "bg-white shadow-xl rounded-lg p-6 my-6 w-full max-w-xl" },
[
m(StepAccordion, { index: 2 }),
this.sinogramUrl &&
m(
"div",
{
class:
"mt-6 border rounded-md p-4 bg-gray-50 text-center relative",
},
[
m(
"h3",
{ class: "text-lg font-semibold text-gray-700 mb-3" },
"Generated Sinogram"
),
m(
"div",
{
class: "relative inline-block w-full",
style: "max-width:100%;",
},
[
this.sinogramUrl === "loading"
? m(
"div",
{
class:
"flex items-center justify-center h-48 text-gray-500",
},
[
m("div", {
class:
"animate-spin rounded-full h-12 w-12 border-t-2 border-b-2 border-blue-500 mr-3",
}),
"Processing Sinogram...",
]
)
: m("img", {
src: this.sinogramUrl,
alt: "Sinogram",
class:
"rounded shadow-md max-w-full h-auto block w-full", // Removed mx-auto, parent is inline-block
style: "max-height: 250px;",
// key: this.sinogramUrl, // Removed to fix Mithril key error
onload: ({ target }) => {
this.sinogramImageElement = target;
if (this.sinogramHighlightCanvas) {
this.sinogramHighlightCanvas.width =
target.naturalWidth; // Use natural for accuracy
this.sinogramHighlightCanvas.height =
target.naturalHeight;
// Style canvas to match displayed image size
this.sinogramHighlightCanvas.style.width = `${target.width}px`;
this.sinogramHighlightCanvas.style.height = `${target.height}px`;
this.drawSinogramHighlight(
this.manualAngleIndex
);
}
},
onerror: () => {
this.sinogramImageElement = null;
if (this.sinogramHighlightCanvas) {
// Clear highlight
const sctx =
this.sinogramHighlightCanvas.getContext(
"2d"
);
sctx.clearRect(
0,
0,
this.sinogramHighlightCanvas.width,
this.sinogramHighlightCanvas.height
);
}
},
}),
m("canvas", {
class: "absolute top-0 left-0 pointer-events-none",
oncreate: ({ dom }) => {
this.sinogramHighlightCanvas = dom;
if (
this.sinogramImageElement &&
this.sinogramImageElement.complete
) {
dom.width =
this.sinogramImageElement.naturalWidth;
dom.height =
this.sinogramImageElement.naturalHeight;
dom.style.width = `${this.sinogramImageElement.width}px`;
dom.style.height = `${this.sinogramImageElement.height}px`;
this.drawSinogramHighlight(this.manualAngleIndex);
} else {
// Default placeholder size
dom.width = 300;
dom.height = 150;
dom.style.width = `300px`;
dom.style.height = `150px`;
}
},
}),
]
),
]
),
m(StepAccordion, { index: 3 }),
this.filteredSinogramUrl &&
m(
"div",
{
class: "mt-6 border rounded-md p-4 bg-gray-50 text-center",
},
[
m(
"h3",
{ class: "text-lg font-semibold text-gray-700 mb-3" },
"Filtered Sinogram"
),
m("img", {
src: this.filteredSinogramUrl,
alt: "Filtered Sinogram",
class:
"rounded shadow-md max-w-full h-auto mx-auto block w-full",
style: "max-height: 250px;",
// key: this.filteredSinogramUrl, // Removed to fix Mithril key error
}),
]
),
]
),
// --- Section 3: Reconstruction ---
this.reconstructedUrl &&
m(
"div",
{ class: "bg-white shadow-xl rounded-lg p-6 my-6 w-full max-w-xl" },
[
m(StepAccordion, { index: 4 }),
m(StepAccordion, { index: 5 }),
m(
"div",
{ class: "mt-6 border rounded-md p-4 bg-gray-50 text-center" },
[
m(
"h3",
{ class: "text-lg font-semibold text-gray-700 mb-3" },
"Reconstructed Image (Filtered Back Projection)"
),
this.reconstructedUrl === "loading"
? m(
"div",
{
class:
"flex items-center justify-center h-48 text-gray-500",
},
[
m("div", {
class:
"animate-spin rounded-full h-12 w-12 border-t-2 border-b-2 border-blue-500 mr-3",
}),
"Reconstructing Image...",
]
)
: m("img", {
src:
this.reconstructionFrames[this.currentFrameIndex] ||
this.reconstructedUrl,
alt: "Reconstructed Image",
class:
"rounded shadow-md max-w-full h-auto mx-auto block",
// key: // Removed to fix Mithril key error
// this.reconstructionFrames[this.currentFrameIndex] ||
// this.reconstructedUrl,
}),
]
),
m(
"div",
{
class: "mt-6",
},
[
m("div", [
m(
"label",
{
for: "renderStyleSelect",
class: "block text-sm font-medium text-gray-700 mb-1",
},
"Render Style:"
),
m(
"select",
{
id: "renderStyleSelect",
class:
"mt-1 block w-full pl-3 pr-10 py-2 text-base border-gray-300 focus:outline-none focus:ring-blue-500 focus:border-blue-500 sm:text-sm rounded-md shadow-sm",
value: this.renderMode,
onchange: (e) => {
this.renderMode = e.target.value;
this.loadAndProcessDebounced(
this.imageUrl,
this.imageUrl !== this.defaultImageUrl
);
},
},
[
m("option", { value: "heatmap" }, "Heatmap"),
m("option", { value: "grayscale" }, "Grayscale"),
]
),
]),
]
),
this.reconstructionFrames.length > 1 &&
m("div", { class: "mt-6" }, [
m(
"label",
{
for: "reconstructionFrameSlider",
class: "block text-sm font-medium text-gray-700 mb-1",
},
`Reconstruction Progress (Frame: ${
this.currentFrameIndex + 1
} / ${this.reconstructionFrames.length})`
),
m("input", {
id: "reconstructionFrameSlider",
type: "range",
min: 0,
max: this.reconstructionFrames.length - 1,
value: this.currentFrameIndex,
step: 1,
class:
"w-full h-2 bg-gray-300 rounded-lg appearance-none cursor-pointer accent-blue-600",
oninput: (e) => {
this.currentFrameIndex = +e.target.value;
m.redraw();
},
}),
]),
]
),
]
);
},
};
// Debounce utility
function debounce(fn, delay) {
let timeout;
return (...args) => {
clearTimeout(timeout);
timeout = setTimeout(() => fn(...args), delay);
};
}

View File

@@ -90,3 +90,57 @@ export function applyRampFilter(projection) {
ifft1D(re, im);
return Array.from(re.slice(0, projection.length));
}
export function applyFilter(projection, type = "ramp") {
const N = nextPow2(projection.length);
const re = new Float32Array(N);
const im = new Float32Array(N);
for (let i = 0; i < projection.length; i++) {
re[i] = projection[i];
}
fft1D(re, im);
for (let i = 0; i < N / 2; i++) {
const normFreq = i / N;
const ramp = normFreq;
let filter = 1;
switch (type) {
case "ramp":
filter = ramp;
break;
case "shepp-logan":
const sinc =
normFreq === 0
? 1
: Math.sin(Math.PI * normFreq) / (Math.PI * normFreq);
filter = ramp * sinc;
break;
case "hann":
filter = ramp * (0.5 + 0.5 * Math.cos(Math.PI * normFreq));
break;
case "hamming":
filter = ramp * (0.54 + 0.46 * Math.cos(Math.PI * normFreq));
break;
case "cosine":
filter = ramp * Math.cos((Math.PI * normFreq) / 2);
break;
case "none":
filter = 1;
break;
default:
console.warn(`Unknown filter type: ${type}. Defaulting to ramp.`);
filter = ramp;
}
re[i] *= filter;
im[i] *= filter;
re[N - i - 1] *= filter;
im[N - i - 1] *= filter;
}
ifft1D(re, im);
return Array.from(re.slice(0, projection.length));
}

View File

@@ -17,6 +17,7 @@
};
</script>
<script src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-svg.js"></script>
<script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
</head>

View File

@@ -1,4 +1,4 @@
import { applyRampFilter } from "./fbp.js";
import { applyFilter } from "./fbp.js";
export async function generateSinogram(
imageUrl,
@@ -18,51 +18,49 @@ export async function generateSinogram(
for (let angle = 0; angle < angles; angle++) {
const theta = (angle * Math.PI) / angles;
// 🔁 Call visual overlay for this angle
if (drawAngleCallback) drawAngleCallback(theta);
// await new Promise((r) => setTimeout(r, 0.01)); // Removed for performance
// (Optional: add delay for animation)
await new Promise((r) => setTimeout(r, 0.01));
// Clear canvas
ctx.clearRect(0, 0, size, size);
// Transform and draw rotated image
ctx.save();
ctx.translate(size / 2, size / 2);
ctx.rotate(theta);
ctx.drawImage(image, -image.width / 2, -image.height / 2);
ctx.restore();
// Read pixel data
const { data } = ctx.getImageData(0, 0, size, size);
// Sum brightness vertically (simulate X-ray projection)
const projection = [];
for (let x = 0; x < size; x++) {
let sum = 0;
for (let y = 0; y < size; y++) {
const i = (y * size + x) * 4;
const gray = data[i]; // red channel (since grayscale)
sum += gray;
sum += data[i]; // grayscale from red channel
}
projection.push(sum / size); // normalize
projection.push(sum / size);
}
projections.push(projection);
}
// Create sinogram canvas
const isHorizontal = angles >= size;
// Create rotated canvas accordingly
const sinogramCanvas = Object.assign(document.createElement("canvas"), {
width: size,
height: angles,
width: isHorizontal ? size : angles,
height: isHorizontal ? angles : size,
});
const sinCtx = sinogramCanvas.getContext("2d");
const imgData = sinCtx.createImageData(size, angles);
const imgData = sinCtx.createImageData(
sinogramCanvas.width,
sinogramCanvas.height
);
for (let y = 0; y < angles; y++) {
for (let angle = 0; angle < angles; angle++) {
for (let x = 0; x < size; x++) {
const val = projections[y][x];
const i = (y * size + x) * 4;
const val = projections[angle][x];
const px = isHorizontal ? x : angle;
const py = isHorizontal ? angle : x;
const i = (py * sinogramCanvas.width + px) * 4;
imgData.data[i + 0] = val;
imgData.data[i + 1] = val;
imgData.data[i + 2] = val;
@@ -71,7 +69,10 @@ export async function generateSinogram(
}
sinCtx.putImageData(imgData, 0, 0);
return sinogramCanvas.toDataURL();
return {
sinogramUrl: sinogramCanvas.toDataURL(),
projections: projections, // Return the raw projections
};
}
export async function reconstructImageFromSinogram(
@@ -79,7 +80,7 @@ export async function reconstructImageFromSinogram(
size = 256,
onFrame = null,
renderMode = "heatmap",
useFBP = true
filterType = "ramp"
) {
const sinogramImage = await loadImage(sinogramUrl);
const canvas = Object.assign(document.createElement("canvas"), {
@@ -95,7 +96,18 @@ export async function reconstructImageFromSinogram(
sinogramImage.height
).data;
size = sinogramImage.width; // match size to sinogram resolution
// Detect orientation
let width, angles;
const isVertical = sinogramImage.height > sinogramImage.width;
if (isVertical) {
angles = sinogramImage.width;
width = sinogramImage.height;
} else {
angles = sinogramImage.height;
width = sinogramImage.width;
}
size = width;
const outputCanvas = Object.assign(document.createElement("canvas"), {
width: size,
height: size,
@@ -104,36 +116,43 @@ export async function reconstructImageFromSinogram(
const accum = new Float32Array(size * size);
const center = size / 2;
const angles = sinogramImage.height;
const width = sinogramImage.width;
const rawProjectionsFromSinogram = []; // To store raw projections extracted from sinogram image
const allFilteredProjections = []; // To store filtered projections
for (let angle = 0; angle < angles; angle++) {
const theta = (angle * Math.PI) / angles;
let projection = [];
let currentRawProjection = [];
for (let x = 0; x < width; x++) {
const i = (angle * width + x) * 4;
projection.push(sinogramData[i]);
}
if (useFBP) {
projection = applyRampFilter(projection);
const i = isVertical
? (x * angles + angle) * 4 // transposed layout
: (angle * width + x) * 4; // normal layout
currentRawProjection.push(sinogramData[i]);
}
rawProjectionsFromSinogram.push(currentRawProjection); // Store the raw projection
// Apply the selected filter
const filteredProjection = applyFilter(
[...currentRawProjection],
filterType
); // Apply filter to a copy
allFilteredProjections.push(filteredProjection); // Store the filtered projection
for (let y = 0; y < size; y++) {
for (let x = 0; x < size; x++) {
const x0 = x - center;
const y0 = center - y; // flip y
const y0 = center - y;
const s = Math.round(
x0 * Math.cos(theta) + y0 * Math.sin(theta) + width / 2
);
if (s >= 0 && s < width) {
accum[y * size + x] += projection[s];
accum[y * size + x] += filteredProjection[s]; // Use the filtered projection for accumulation
}
}
}
if (onFrame) {
// normalize and draw current frame
let maxVal = 0;
for (let i = 0; i < accum.length; i++) {
if (accum[i] > maxVal) maxVal = accum[i];
@@ -154,13 +173,58 @@ export async function reconstructImageFromSinogram(
imageData.data[i * 4 + 2] = b;
imageData.data[i * 4 + 3] = 255;
}
outputCtx.putImageData(imageData, 0, 0);
await new Promise((r) => setTimeout(r, 1));
// await new Promise((r) => setTimeout(r, 1)); // Removed for performance
onFrame(angle, outputCanvas.toDataURL());
}
}
return outputCanvas.toDataURL();
// Create filtered sinogram image
const filteredSinogramCanvas = Object.assign(
document.createElement("canvas"),
{
width: isVertical ? angles : width, // Same dimensions as original sinogram
height: isVertical ? width : angles,
}
);
const filteredSinCtx = filteredSinogramCanvas.getContext("2d");
const filteredImgData = filteredSinCtx.createImageData(
filteredSinogramCanvas.width,
filteredSinogramCanvas.height
);
let maxFilteredVal = 0;
for (const proj of allFilteredProjections) {
for (const val of proj) {
if (val > maxFilteredVal) maxFilteredVal = val;
}
}
// Avoid division by zero if maxFilteredVal is 0
if (maxFilteredVal === 0) maxFilteredVal = 1;
for (let angle = 0; angle < angles; angle++) {
const currentProjection = allFilteredProjections[angle];
for (let x = 0; x < width; x++) {
const val = (currentProjection[x] / maxFilteredVal) * 255; // Normalize and scale
const px = isVertical ? angle : x;
const py = isVertical ? x : angle;
const i = (py * filteredSinogramCanvas.width + px) * 4;
filteredImgData.data[i + 0] = val;
filteredImgData.data[i + 1] = val;
filteredImgData.data[i + 2] = val;
filteredImgData.data[i + 3] = 255;
}
}
filteredSinCtx.putImageData(filteredImgData, 0, 0);
const filteredSinogramUrl = filteredSinogramCanvas.toDataURL();
return {
reconstructedUrl: outputCanvas.toDataURL(),
filteredSinogramUrl: filteredSinogramUrl,
rawProjectionsFromSinogram: rawProjectionsFromSinogram,
allFilteredProjections: allFilteredProjections,
};
}
// Heatmap mapping: blue → green → yellow → red

7
static.json Normal file
View File

@@ -0,0 +1,7 @@
{
"root": "public",
"clean_urls": true,
"routes": {
"/**": "index.html"
}
}