File: variable_scaling.cpp

package info (click to toggle)
mrtrix3 3.0.4-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 13,712 kB
  • sloc: cpp: 129,776; python: 9,494; sh: 593; makefile: 234; xml: 47
file content (69 lines) | stat: -rw-r--r-- 2,362 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
/* Copyright (c) 2008-2022 the MRtrix3 contributors.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Covered Software is provided under this License on an "as is"
 * basis, without warranty of any kind, either expressed, implied, or
 * statutory, including, without limitation, warranties that the
 * Covered Software is free of defects, merchantable, fit for a
 * particular purpose or non-infringing.
 * See the Mozilla Public License v. 2.0 for more details.
 *
 * For more details, see http://www.mrtrix.org/.
 */

#include <limits>

#include "app.h"
#include "progressbar.h"
#include "header.h"
#include "image_io/variable_scaling.h"

namespace MR
{
  namespace ImageIO
  {


    void VariableScaling::load (const Header& header, size_t)
    {
      if (files.empty())
        throw Exception ("no files specified in header for image \"" + header.name() + "\"");

      assert (header.datatype().is_floating_point() && header.datatype().bits() == 32 && header.datatype().is_byte_order_native());
      assert (files.size() == scale_factors.size());
      assert (header.intensity_offset() == 0 && header.intensity_scale() == 1.0);

      size_t voxels_per_segment = segsize / files.size();

      DEBUG ("loading variable-scaling DICOM image \"" + header.name() + "\"...");
      addresses.resize (1);
      addresses[0].reset (new uint8_t [segsize * sizeof(float32)]);
      if (!addresses[0])
        throw Exception ("failed to allocate memory for image \"" + header.name() + "\"");

      ProgressBar progress ("rescaling DICOM images", files.size());
      float32* data = reinterpret_cast<float32*> (addresses[0].get());
      for (size_t n = 0; n < files.size(); n++) {
        const float offset = scale_factors[n].offset;
        const float scale = scale_factors[n].scale;
        File::MMap file (files[n], false, false, sizeof(uint16_t) * voxels_per_segment);
        const uint16_t* from = reinterpret_cast<uint16_t*> (file.address());

        for (size_t i = 0; i < voxels_per_segment; i++)
          data[i] = offset + scale * from[i];

        data += voxels_per_segment;
        ++progress;
      }

    }

    void VariableScaling::unload (const Header& header) { }

  }
}