agerecnik/mammography-plugin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated dicom metadata and pdf document title

Browse Source
This commit is contained in:
Aljaž Gerečnik 2025-03-07 12:17:03 +01:00
parent 57e16bbc43
commit 1098d76180
2 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
dicom_sr.py
View File

@ -95,14 +95,16 @@ def apply(retina_net, dicom,
instance_number = 1 instance_number = 1
) )
sr_object.StudyDate = dicom.StudyDate sr_object.StudyDate = dicom.get("StudyDate", "")
sr_object.StudyTime = dicom.StudyTime sr_object.StudyTime = dicom.get("StudyTime", "")
sr_object.StudyDescription = dicom.get("StudyDescription", "")
sr_object.SeriesDate = datetime.now().strftime("%Y%m%d") sr_object.SeriesDate = datetime.now().strftime("%Y%m%d")
sr_object.SeriesTime = datetime.now().strftime("%H%M%S") sr_object.SeriesTime = datetime.now().strftime("%H%M%S")
sr_object.PatientID = dicom.PatientID sr_object.SeriesDescription = "Mammography Structured Report"
sr_object.PatientName = dicom.PatientName sr_object.PatientID = dicom.get("PatientID", "")
sr_object.PatientSex = dicom.PatientSex sr_object.PatientName = dicom.get("PatientName", "")
sr_object.PatientBirthDate = dicom.PatientBirthDate sr_object.PatientSex = dicom.get("PatientSex", "")
sr_object.ReferringPhysicianName = sr_object.ReferringPhysicianName sr_object.PatientBirthDate = dicom.get("PatientBirthDate", "")
sr_object.ReferringPhysicianName = dicom.get("ReferringPhysicianName", "")
return sr_object return sr_object

9
dicom_sr_to_pdf.py
View File

@ -52,7 +52,10 @@ def overlay_measurements(dcm_image_pixels, measurements, probabilities, image_pa
def create_pdf(image_path, measurements, sr, pdf_path): def create_pdf(image_path, measurements, sr, pdf_path):
"""Creates a PDF with the mammography image and extracted measurements.""" """Creates a PDF with the mammography image and extracted measurements."""
c = canvas.Canvas(pdf_path) c = canvas.Canvas(pdf_path)
# Set PDF metadata
c.setTitle("Mammography Report")
# Set font for the title # Set font for the title
c.setFont("Helvetica-Bold", 16) c.setFont("Helvetica-Bold", 16)
@ -112,6 +115,7 @@ def create_dcm_pdf(sr, pdf_path, instance_uid):
ds.StudyInstanceUID = sr.StudyInstanceUID ds.StudyInstanceUID = sr.StudyInstanceUID
ds.StudyDate = sr.StudyDate ds.StudyDate = sr.StudyDate
ds.StudyTime = sr.StudyTime ds.StudyTime = sr.StudyTime
ds.StudyDescription = sr.StudyDescription
ds.AccessionNumber = sr.AccessionNumber ds.AccessionNumber = sr.AccessionNumber
ds.ReferringPhysicianName = sr.ReferringPhysicianName ds.ReferringPhysicianName = sr.ReferringPhysicianName
ds.StudyID = sr.StudyID ds.StudyID = sr.StudyID
@ -119,11 +123,12 @@ def create_dcm_pdf(sr, pdf_path, instance_uid):
ds.SeriesInstanceUID = generate_uid() ds.SeriesInstanceUID = generate_uid()
ds.SeriesDate = sr.SeriesDate ds.SeriesDate = sr.SeriesDate
ds.SeriesTime = sr.SeriesTime ds.SeriesTime = sr.SeriesTime
ds.SeriesDescription = "Mammography PDF Report"
ds.SeriesNumber = 1 ds.SeriesNumber = 1
ds.Modality = "DOC" ds.Modality = "DOC"
ds.Manufacturer = "MammographyAI" ds.Manufacturer = "MammographyAI"
ds.ConversionType = "DI" ds.ConversionType = "SD"
ds.SOPInstanceUID = instance_uid ds.SOPInstanceUID = instance_uid
ds.SOPClassUID = EncapsulatedPDFStorage ds.SOPClassUID = EncapsulatedPDFStorage