Added optional downscaled option

This commit is contained in:
Vahid 2021-10-23 22:20:08 +11:00 коммит произвёл GitHub
Родитель 6dfab009ba
Коммит 5f0d380920
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
3 изменённых файлов: 32 добавлений и 10 удалений

Просмотреть файл

@ -65,7 +65,10 @@ namespace UnityVolumeRendering
// Load all DICOM files
List<DICOMSliceFile> files = new List<DICOMSliceFile>();
foreach (string filePath in fileCandidates)
IEnumerable<string> sortedFiles = fileCandidates.OrderBy(s => s);
foreach (string filePath in sortedFiles)
{
DICOMSliceFile sliceFile = ReadDICOMFile(filePath);
if(sliceFile != null)
@ -90,13 +93,10 @@ namespace UnityVolumeRendering
return new List<DICOMSeries>(seriesByUID.Values);
}
public VolumeDataset ImportDICOMSeries(DICOMSeries series)
public VolumeDataset ImportDICOMSeries(DICOMSeries series, bool forceDownScaling = false)
{
List<DICOMSliceFile> files = series.dicomFiles;
// Sort files by slice location
files.Sort((DICOMSliceFile a, DICOMSliceFile b) => { return a.location.CompareTo(b.location); });
// Check if the series is missing the slice location tag
bool needsCalcLoc = false;
foreach (DICOMSliceFile file in files)
@ -108,6 +108,9 @@ namespace UnityVolumeRendering
if (needsCalcLoc)
CalcSliceLocFromPos(files);
// Sort files by slice location
files.Sort((DICOMSliceFile a, DICOMSliceFile b) => { return a.location.CompareTo(b.location); });
Debug.Log($"Importing {files.Count} DICOM slices");
if (files.Count <= 1)
@ -156,6 +159,11 @@ namespace UnityVolumeRendering
dataset.scaleZ = Mathf.Abs(files[files.Count - 1].location - files[0].location);
}
if (forceDownScaling)
{
dataset.DownScaleData();
}
dataset.FixDimensions();
return dataset;

Просмотреть файл

@ -11,15 +11,18 @@ namespace UnityVolumeRendering
public class ImageSequenceImporter
{
private string directoryPath;
private bool forceDownScaling;
private string[] supportedImageTypes = new string[]
{
"*.png",
"*.jpg",
"*.jpeg"
};
public ImageSequenceImporter(string directoryPath)
public ImageSequenceImporter(string directoryPath, bool forceDownScaling = false)
{
this.directoryPath = directoryPath;
this.forceDownScaling = forceDownScaling;
}
public VolumeDataset Import()
@ -36,6 +39,11 @@ namespace UnityVolumeRendering
int[] data = FillSequentialData(dimensions, imagePaths);
VolumeDataset dataset = FillVolumeDataset(data, dimensions);
if (forceDownScaling)
{
dataset.DownScaleData();
}
dataset.FixDimensions();
return dataset;

Просмотреть файл

@ -29,8 +29,8 @@ namespace UnityVolumeRendering
private DataContentFormat contentFormat;
private Endianness endianness;
private int skipBytes;
public RawDatasetImporter(string filePath, int dimX, int dimY, int dimZ, DataContentFormat contentFormat, Endianness endianness, int skipBytes)
private bool forceDownScaling;
public RawDatasetImporter(string filePath, int dimX, int dimY, int dimZ, DataContentFormat contentFormat, Endianness endianness, int skipBytes, bool forceDownScaling = false)
{
this.filePath = filePath;
this.dimX = dimX;
@ -39,12 +39,13 @@ namespace UnityVolumeRendering
this.contentFormat = contentFormat;
this.endianness = endianness;
this.skipBytes = skipBytes;
this.forceDownScaling = forceDownScaling;
}
public VolumeDataset Import()
{
// Check that the file exists
if(!File.Exists(filePath))
if (!File.Exists(filePath))
{
Debug.LogError("The file does not exist: " + filePath);
return null;
@ -86,6 +87,11 @@ namespace UnityVolumeRendering
reader.Close();
fs.Close();
if (forceDownScaling)
{
dataset.DownScaleData();
}
dataset.FixDimensions();
return dataset;