forked from xiangruili/dicm2nii
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathnii_tool.m
executable file
·1255 lines (1159 loc) · 48.8 KB
/
nii_tool.m
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
function varargout = nii_tool(cmd, varargin)
% Basic function to create, load and save NIfTI file.
%
% rst = nii_tool('cmd', para);
%
% To list all command, type
% nii_tool ?
%
% To get help information for each command, include '?' in cmd, for example:
% nii_tool init?
% nii_tool('init?')
%
% Here is a list of all command:
%
% nii_tool('default', 'version', 1, 'rgb_dim', 1);
% nii = nii_tool('init', img);
% nii = nii_tool('update', nii, mat);
% nii_tool('save', nii, filename, force_3D);
% hdr = nii_tool('hdr', filename);
% img = nii_tool('img', filename_or_hdr);
% ext = nii_tool('ext', filename_or_hdr);
% nii = nii_tool('load', filename_or_hdr);
% nii = nii_tool('cat3D', filenames);
% nii_tool('RGBStyle', 'afni');
%
% Detail for each command is described below.
%
% oldVal = nii_tool('default', 'version', 1, 'rgb_dim', 1);
% oldVal = nii_tool('default', struct('version', 1, 'rgb_dim', 1));
%
% - Set/query default NIfTI version and/or rgb_dim. To check the setting, run
% nii_tool('default') without other input. The input for 'default' command can
% be either a struct with fields of 'version' and/or 'rgb_dim', or
% parameter/value pairs. See nii_tool('RGBstyle') for meaning of rgb_dim.
%
% Note that the setting will be saved for future use. If one wants to change the
% settting temporarily, it is better to return the oldVal, and to restore it
% after done:
%
% oldVal = nii_tool('default', 'version', 2); % set version 2 as default
% % 'init' and 'save' NIfTI using above version
% nii_tool('default', 'version', oldVal); % restore default setting
%
% The default version setting affects 'init' command only. If you 'load' a NIfTI
% file, modify it, and then 'save' it, the version will be the same as the
% original file, unless it is changed explicitly (see help for 'save' command).
% All 'load' command ('load', 'hdr', 'ext', 'img') will read any version
% correctly, regardless of version setting.
%
%
% nii = nii_tool('init', img, RGB_dim);
%
% - Initialize nii struct based on img, normally 3D or 4D array. Most fields in
% the returned nii.hdr contain default values, and need to be updated based on
% dicom or other information. Important ones include pixdim, s/qform_code and
% related parameters.
%
% The NIfTI datatype will depend on data type of img. Most Matlab data types are
% supported, including 8/16/32/64 bit signed and unsigned integers, single and
% double floating numbers. Single/double complex and logical array are also
% supported.
%
% The optional third input, RGB_dim, is needed only if img contains RGB/RGBA
% data. It specifies which dimension in img encodes RGB or RGBA. In other words,
% if a non-empty RGB_dim is provided, img will be interpreted as RGB/RGBA data.
%
% Another way to signify RGB/RGBA data is to permute color dim to 8th-dim of img
% (RGB_dim of 8 can be omitted then). Since NIfTI img can have up to 7 dim,
% nii_tool chooses to store RGB/RGBA in 8th dim. Although this looks lengthy
% (4th to 7th dim are often all ones), nii_tool can deal with up to 7 dim
% without causing any confusion. This is why the returned nii.img always stores
% RGB in 8th dim.
%
%
% nii = nii_tool('update', nii, mat);
%
% - Update nii.hdr according to nii.img. This is useful if one changes nii.img
% type or dimension. The 'save' command calls this internally, so it is not
% necessary to call this before 'save'. A useful case to call 'update' is that
% one likes to use nii struct without saving it to a file, and 'update' will
% make nii.hdr.dim and others correct.
%
% If the 3rd input, new transformation matrix is provided, it will be set as the
% sform transformation matrix.
%
%
% hdr = nii_tool('hdr', filename);
%
% - Return hdr struct of the provided NIfTI file. This is useful to check NIfTI
% hdr, and it is much faster than 'load', especially for .gz file.
%
%
% img = nii_tool('img', filename_or_hdr);
%
% - Return image data in a NIfTI file. The second input can be NIfTI file name,
% or hdr struct returned by nii_tool('hdr', filename).
%
%
% ext = nii_tool('ext', filename_or_hdr);
%
% - Return NIfTI extension in a NIfTI file. The second input can be NIfTI file
% name, or hdr struct returned by nii_tool('hdr', filename). The returned ext
% will have field 'edata_decoded' if 'ecode' is of known type, such as dicom
% (2), text (4 or 6) or Matlab (40).
%
% Here is an example to add data in myFile.mat as extension to nii struct, which
% can be from 'init' or 'load':
%
% fid = fopen('myFile.mat'); % open the MAT file
% myEdata = fread(fid, inf, '*uint8'); % load all bytes as byte column
% fclose(fid);
% len = int32(numel(myEdata)); % number of bytes in int32
% myEdata = [typecast(len, 'uint8')'; myEdata]; % include len in myEdata
% nii.ext.ecode = 40; % 40 for Matlab extension
% nii.ext.edata = myEdata; % myEdata must be uint8 array
%
% nii_tool will take care of rest when you 'save' nii to a file.
%
% In case a NIfTI ext causes problem (for example, some FSL builds have problem
% in reading NIfTI img with ecode>30), one can remove the ext easily:
%
% nii = nii_tool('load', 'file_with_ext.nii'); % load the file with ext
% nii.ext = []; % or nii = rmfield(nii, 'ext'); % remove ext
% nii_tool('save', nii, 'file_without_ext.nii'); % save it
%
%
% nii = nii_tool('load', filename_or_hdr);
%
% - Load NIfTI file into nii struct. The returned struct includes NIfTI 'hdr'
% and 'img', as well as 'ext' if the file contains NIfTI extension.
%
% nii_tool returns nii.img with the same data type as stored in the file, while
% numeric values in hdr are in double precision for convenience.
%
%
% nii_tool('save', nii, filename, force_3D);
%
% - Save struct nii into filename. The format of the file is determined by the
% file extension, such as .img, .nii, .img.gz, .nii.gz etc. If filename is not
% provided, nii.hdr.file_name must contain a file name. Note that 'save' command
% always overwrites file in case of name conflict.
%
% If filename has no extension, '.nii' will be used as default.
%
% If the 4th input, force_3D, is true (default false), the output file will be
% 3D only, which means multiple volume data will be split into multiple files.
% This is the format SPM likes. You can use this command to convert 4D into 3D
% by 'load' a 4D file, then 'save' it as 3D files. The 3D file names will have
% 5-digit like '_00001' appended to indicate volume index.
%
% The NIfTI version can be set by nii_tool('default'). One can override the
% default version by specifying it in nii.hdr.version. To convert between
% versions, load a NIfTI file, specify new version, and save it. For example:
%
% nii = nii_tool('load', 'file_nifti1.nii'); % load version 1 file
% nii.hdr.version = 2; % force to NIfTI-2
% nii_tool('save', nii, 'file_nifti2.nii'); % save as version 2 file
%
% Following example shows how to change data type of a nii file:
% nii = nii_tool('load', 'file_int16.nii'); % load int16 type file
% nii.img = single(nii.img); % change data type to single/float32
% nii_tool('save', nii, 'file_float.nii'); % nii_tool will take care of hdr
%
%
% nii = nii_tool('cat3D', files);
%
% - Concatenate SPM 3D files into a 4D dataset. The input 'files' can be cellstr
% with file names, or char with wildcards (* or ?). If it is cellstr, the volume
% order in the 4D data corresponds to those files. If wildcards are used, the
% volume order is based on alphabetical order of file names.
%
% Note that the files to be concatenated must have the same datatype, dim, voxel
% size, scaling slope and intercept, transformation matrix, etc. This is
% normally true if files are for the same dicom series.
%
% Following example shows how to convert a series of 3D files into a 4D file:
%
% nii = nii_tool('cat3D', './data/fSubj2-0003*.nii'); % load files for series 3
% nii_tool('save', nii, './data/fSubj2-0003_4D.nii'); % save as a 4D file
%
%
% oldStyle = nii_tool('RGBStyle', 'afni');
%
% - Set/query the method to read/save RGB or RGBA NIfTI file. The default method
% can be set by nii_tool('default', 'rgb_dim', dimN), where dimN can be 1, 3 or
% 4, or 'afni', 'mricron' or 'fsl', as explained below.
%
% The default is 'afni' style (or 1), which is defined by NIfTI standard, but is
% not well supported by fslview till v5.0.8 or mricron till v20140804.
%
% If the second input is set to 'mricron' (or 3), nii_tool will save file using
% the old RGB fashion (dim 3 for RGB). This works for mricron v20140804 or
% earlier.
%
% If the second input is set to 'fsl' (or 4), nii_tool will save RGB or RGBA
% layer into 4th dimension, and the file is not encoded as RGB data, but as
% normal 4D NIfTI. This violates the NIfTI rule, but it seems it is the only way
% to work for fslview (at least till fsl v5.0.8).
%
% If no new style (second input) is provided, it means to query the current
% style (one of 'afni', 'mricron' and 'fsl').
%
% The GUI method to convert between different RGB style can be found in
% nii_viewer. Following shows how to convert other style into fsl style:
%
% nii_tool('RGBStyle', 'afni'); % we are loading afni style RGB
% nii = nii_tool('load', 'afni_style.nii'); % load RGB file
% nii_tool('RGBStyle', 'fsl'); % switch to fsl style for later save
% nii_tool('save', nii, 'fslRGB.nii'); % fsl can read it as RGB
%
% Note that, if one wants to convert fsl style (non-RGB file by NIfTI standard)
% to other styles, an extra step is needed to change the RGB dim from 4th to 8th
% dim before 'save':
%
% nii = nii_tool('load', 'fslStyleFile.nii'); % it is normal NIfTI
% nii.img = permute(nii.img, [1:3 5:8 4]); % force it to be RGB data
% nii_tool('RGBStyle', 'afni'); % switch to NIfTI RGB style if needed
% nii_tool('save', nii, 'afni_RGB.nii'); % now AFNI can read it as RGB
%
% Also note that the setting by nii_tool('RGBStyle') is effective only for
% current Matlab session. If one clears all or starts a new Matlab session, the
% default style by nii_tool('default') will take effect.
%
% See also NII_VIEWER, NII_XFORM, DICM2NII
% More information for NIfTI format:
% Official NIfTI website: http://nifti.nimh.nih.gov/
% Another excellent site: http://brainder.org/2012/09/23/the-nifti-file-format/
% History (yymmdd)
% 150109 Write it based on Jimmy Shen's NIfTI tool ([email protected])
% 150202 Include renamed pigz files for Windows
% 150203 Fix closeFile and deleteTmpFile order
% 150205 Add hdr.machine: needed for .img fopen
% 150208 Add 4th input for 'save', allowing to save SPM 3D files
% 150210 Add 'cat3D' to load SPM 3D files
% 150226 Assign all 8 char for 'magic' (version 2 needs it)
% 150321 swapbytes(nByte) for ecode=40 with big endian
% 150401 Add 'default' to set/query version and rgb_dim default setting
% 150514 read_ext: decode txt edata by dicm2nii.m
% 150517 func_handle: provide a way to use gunzipOS etc from outside
% 150617 auto detect rgb_dim 1&3 for 'load' etc using ChrisR method
% 151025 Change subfunc img2datatype as 'update' for outside access
% 151109 Include dd.exe from WinAVR-20100110 for partial gz unzip
% 151205 Partial gunzip: fix fname with space & unknown pigz | dd error.
% 151222 Take care of img for intent_code 2003/2004: anyone uses it?
% 160110 Use matlab pref method to replace para file.
% 160120 check_gzip: use "" for included pigz; ignore dd error if err is false.
% 160326 fix setpref for older Octave: set each parameter separately.
% 160531 fopen uses 'W' for 'w': performance benefit according to Yair.
% 160701 subFuncHelp: bug fix for mfile case.
% 161018 gunzipOS: use unique name for outName, to avoid problem with parfor.
% 161025 Make included linux pigz executible; fix "dd" for windows.
% 161031 gunzip_mem(), nii_bytes() for hdr/ext read: read uint8 then parse;
% Replace hdr.machine with hdr.swap_endian.
% 170212 Extract decode_ext() from 'ext' cmd so call it in 'update' cmd.
% 170215 gunzipOS: use -c > rather than copyfile for performance.
% 170322 gzipOS: stop using background gz to avoid file not exist error.
% 170410 read_img(): turn off auto RGB dim detection, and use rgb_dim.
% 170714 'save': force to version 2 if img dim exceeds 2^15-1.
% 170716 Add functionSignatures.json file for tab auto-completion.
% 171031 'LocalFunc' makes eaiser to call local functions.
% 171206 Allow file name ext other than .nii, .hdr, .img.
% 180104 check_gzip: add /usr/local/bin to PATH for unix if needed.
% 180119 use jsystem for better speed.
% 180710 bug fix for cal_max/cal_min in 'update'.
% 210302 take care of unicode char in hdr (Thx Yong).
persistent C para; % C columns: name, length, format, value, offset
if isempty(C)
[C, para] = niiHeader;
if exist('OCTAVE_VERSION', 'builtin')
warning('off', 'Octave:fopen-mode'); % avoid 'W' warning
more off;
end
end
if ~ischar(cmd)
error('Provide a string command as the first input for nii_tool');
end
if any(cmd=='?'), subFuncHelp(mfilename, cmd); return; end
if strcmpi(cmd, 'init')
if nargin<2, error('nii_tool(''%s'') needs second input', cmd); end
nii.hdr = cell2struct(C(:,4), C(:,1));
nii.img = varargin{1};
if numel(size(nii.img))>8
error('NIfTI img can have up to 7 dimension');
end
if nargin>2
i = varargin{2};
if i<0 || i>8 || mod(i,1)>0, error('Invalid RGB_dim number'); end
nii.img = permute(nii.img, [1:i-1 i+1:8 i]); % RGB to dim8
end
nii.hdr.file_name = inputname(2);
if isempty(nii.hdr.file_name), nii.hdr.file_name = 'tmpNII'; end
varargout{1} = nii_tool('update', nii); % set datatype etc
elseif strcmpi(cmd, 'save')
if nargin<2, error('nii_tool(''%s'') needs second input', cmd); end
nii = varargin{1};
if ~isstruct(nii) || ~isfield(nii, 'hdr') || ~isfield(nii, 'img')
error(['nii_tool(''save'') needs a struct from nii_tool(''init'')' ...
' or nii_tool(''load'') as the second input']);
end
% Check file name to save
if nargin>2
fname = varargin{2};
if ~ischar(fname), error('Invalid name for NIfTI file: %s', fname); end
elseif isfield(nii.hdr, 'file_name')
fname = nii.hdr.file_name;
else
error('Provide a valid file name as the third input');
end
if ~ispc && strncmp(fname, '~/', 2) % matlab may err with this abbrevation
fname = [getenv('HOME') fname(2:end)];
end
[pth, fname, fext] = fileparts(fname);
do_gzip = strcmpi(fext, '.gz');
if do_gzip
[~, fname, fext] = fileparts(fname); % get .nii .img .hdr
end
if isempty(fext), fext = '.nii'; end % default single .nii file
fname = fullfile(pth, fname); % without file ext
if nargout, varargout{1} = []; end
isNii = strcmpi(fext, '.nii'); % will use .img/.hdr if not .nii
% Deal with NIfTI version and sizeof_hdr
niiVer = para.version;
if isfield(nii.hdr, 'version'), niiVer = nii.hdr.version; end
if niiVer<2 && any(nii.hdr.dim(2:end) > 32767), niiVer = 2; end
if niiVer == 1
nii.hdr.sizeof_hdr = 348; % in case it was loaded from other version
elseif niiVer == 2
nii.hdr.sizeof_hdr = 540; % version 2
else
error('Unsupported NIfTI version: %g', niiVer);
end
if niiVer ~= para.version
C0 = niiHeader(niiVer);
else
C0 = C;
end
% Update datatype/bitpix/dim in case nii.img is changed
[nii, fmt] = nii_tool('update', nii);
% This 'if' block: lazy implementation to split into 3D SPM files
if nargin>3 && ~isempty(varargin{3}) && varargin{3} && nii.hdr.dim(5)>1
if do_gzip, fext = [fext '.gz']; end
nii0 = nii;
for i = 1:nii.hdr.dim(5)
fname0 = sprintf('%s_%05g%s', fname, i, fext);
nii0.img = nii.img(:,:,:,i,:,:,:,:); % one vol
if i==1 && isfield(nii, 'ext'), nii0.ext = nii.ext;
elseif i==2 && isfield(nii0, 'ext'), nii0 = rmfield(nii0, 'ext');
end
nii_tool('save', nii0, fname0);
end
return;
end
% re-arrange img for special datatype: RGB/RGBA/Complex.
if any(nii.hdr.datatype == [128 511 2304]) % RGB or RGBA
if para.rgb_dim == 1 % AFNI style
nii.img = permute(nii.img, [8 1:7]);
elseif para.rgb_dim == 3 % old mricron style
nii.img = permute(nii.img, [1 2 8 3:7]);
elseif para.rgb_dim == 4 % for fslview
nii.img = permute(nii.img, [1:3 8 4:7]); % violate nii rule
dim = size(nii.img);
if numel(dim)>6 % dim7 is not 1
i = find(dim(5:7)==1, 1, 'last') + 4;
nii.img = permute(nii.img, [1:i-1 i+1:8 i]);
end
nii = nii_tool('update', nii); % changed to non-RGB datatype
end
elseif any(nii.hdr.datatype == [32 1792]) % complex single/double
nii.img = [real(nii.img(:))'; imag(nii.img(:))'];
end
% Check nii extension: update esize to x16
nExt = 0; esize = 0;
nii.hdr.extension = [0 0 0 0]; % no nii ext
if isfield(nii, 'ext') && isstruct(nii.ext) ...
&& isfield(nii.ext(1), 'edata') && ~isempty(nii.ext(1).edata)
nExt = numel(nii.ext);
nii.hdr.extension = [1 0 0 0]; % there is nii ext
for i = 1:nExt
if ~isfield(nii.ext(i), 'ecode') || ~isfield(nii.ext(i), 'edata')
error('NIfTI header ext struct must have ecode and edata');
end
n0 = numel(nii.ext(i).edata) + 8; % 8 byte for esize and ecode
n1 = ceil(n0/16) * 16; % esize: multiple of 16
nii.ext(i).esize = n1;
nii.ext(i).edata(end+(1:n1-n0)) = 0; % pad zeros
esize = esize + n1;
end
end
% Set magic, vox_offset, and open file for .nii or .hdr
if isNii
% version 1 will take only the first 4
nii.hdr.magic = sprintf('n+%g%s', niiVer, char([0 13 10 26 10]));
nii.hdr.vox_offset = nii.hdr.sizeof_hdr + 4 + esize;
fid = fopen(strcat(fname, fext), 'W');
else
nii.hdr.magic = sprintf('ni%g%s', niiVer, char([0 13 10 26 10]));
nii.hdr.vox_offset = 0;
fid = fopen(strcat(fname, '.hdr'), 'W');
end
% Write nii hdr
for i = 1:size(C0,1)
if isfield(nii.hdr, C0{i,1})
val = nii.hdr.(C0{i,1});
else % niiVer=2 omit some fields, also take care of other cases
val = C0{i,4};
end
fmt0 = C0{i,3};
if strcmp(C0{i,3}, 'char') && ~isempty(val)
if ~ischar(val), val = char(val); end % avoid val=[] error etc
val = unicode2native(val); % may have more bytes than numel(val)
fmt0 = 'uint8';
end
n = numel(val);
len = C0{i,2};
if n>len
val(len+1:n) = []; % remove extra, normally for char
elseif n<len
val(n+1:len) = 0; % pad 0, normally for char
end
fwrite(fid, val, fmt0);
end
% Write nii ext: extension is in hdr
for i = 1:nExt % nExt may be 0
fwrite(fid, nii.ext(i).esize, 'int32');
fwrite(fid, nii.ext(i).ecode, 'int32');
fwrite(fid, nii.ext(i).edata, 'uint8');
end
if isNii
n = nii.hdr.vox_offset - ftell(fid);
if n<0 % seen n=-1 for unknown reason
fseek(fid, n, 'cof');
elseif n>0
fwrite(fid, zeros(n,1), 'uint8');
end
else
fclose(fid); % done with .hdr
fid = fopen(strcat(fname, '.img'), 'W');
end
% Write nii image
fwrite(fid, nii.img, fmt);
fclose(fid); % all written
% gzip if asked
if do_gzip
if isNii
gzipOS(strcat(fname, '.nii'));
else
gzipOS(strcat(fname, '.hdr')); % better not to compress .hdr
gzipOS(strcat(fname, '.img'));
end
end
elseif strcmpi(cmd, 'hdr')
if nargin<2, error('nii_tool(''%s'') needs second input', cmd); end
if ~ischar(varargin{1})
error('nii_tool(''hdr'') needs nii file name as second input');
end
fname = nii_name(varargin{1}, '.hdr'); % get .hdr if it is .img
[b, fname] = nii_bytes(fname, 600); % v2: 544+10 gzip header
varargout{1} = read_hdr(b, C, fname);
elseif any(strcmpi(cmd, {'img' 'load'}))
if nargin<2, error('nii_tool(''%s'') needs second input', cmd); end
if ischar(varargin{1}) % file name input
fname = nii_name(varargin{1}, '.hdr');
nii = struct;
elseif isstruct(varargin{1}) && isfield(varargin{1}, 'file_name')
nii.hdr = varargin{1};
fname = nii.hdr.file_name;
else
error(['nii_tool(''%s'') needs a file name or hdr struct from ' ...
'nii_tool(''hdr'') as second input'], cmd);
end
if strcmpi(cmd, 'load')
[ext, nii.hdr] = nii_tool('ext', varargin{1});
if ~isempty(ext), nii.ext = ext; end
elseif ~isfield(nii, 'hdr')
nii.hdr = nii_tool('hdr', fname);
end
nii.img = read_img(nii.hdr, para);
if strcmpi(cmd, 'load')
varargout{1} = nii;
else % img
varargout{1} = nii.img;
end
elseif strcmpi(cmd, 'ext')
if nargin<2, error('nii_tool(''%s'') needs second input', cmd); end
if ischar(varargin{1}) % file name input
fname = nii_name(varargin{1}, '.hdr');
hdr = nii_tool('hdr', fname);
elseif isstruct(varargin{1}) && isfield(varargin{1}, 'file_name')
hdr = varargin{1};
fname = hdr.file_name;
else
error(['nii_tool(''%s'') needs a file name or hdr struct from ' ...
'nii_tool(''hdr'') as second input'], cmd);
end
if isempty(hdr.extension) || hdr.extension(1)==0
varargout{1} = [];
else
if hdr.vox_offset>0, nByte = hdr.vox_offset + 64; % .nii arbituary +64
else, nByte = inf;
end
b = nii_bytes(fname, nByte);
varargout{1} = read_ext(b, hdr);
end
if nargout>1, varargout{2} = hdr; end
elseif strcmpi(cmd, 'RGBStyle')
styles = {'afni' '' 'mricron' 'fsl'};
curStyle = styles{para.rgb_dim};
if nargin<2, varargout{1} = curStyle; return; end % query only
irgb = varargin{1};
if isempty(irgb), irgb = 1; end % default as 'afni'
if ischar(irgb)
if strncmpi(irgb, 'fsl', 3), irgb = 4;
elseif strncmpi(irgb, 'mricron', 4), irgb = 3;
else, irgb = 1;
end
end
if ~any(irgb == [1 3 4])
error('nii_tool(''RGBStyle'') can have 1, 3, or 4 as second input');
end
if nargout, varargout{1} = curStyle; end % return old one
para.rgb_dim = irgb; % no save to pref
elseif strcmpi(cmd, 'cat3D')
if nargin<2, error('nii_tool(''%s'') needs second input', cmd); end
fnames = varargin{1};
if ischar(fnames) % guess it is like run1*.nii
f = dir(fnames);
f = sort({f.name});
fnames = strcat([fileparts(fnames) '/'], f);
end
n = numel(fnames);
if n<2 || (~iscellstr(fnames) && (exist('strings', 'builtin') && ~isstring(fnames)))
error('Invalid input for nii_tool(''cat3D''): %s', varargin{1});
end
nii = nii_tool('load', fnames{1}); % all for first file
nii.img(:,:,:,2:n) = 0; % pre-allocate
% For now, omit all consistence check between files
for i = 2:n, nii.img(:,:,:,i) = nii_tool('img', fnames{i}); end
varargout{1} = nii_tool('update', nii); % update dim
elseif strcmpi(cmd, 'default')
flds = {'version' 'rgb_dim'}; % may add more in the future
pf = getpref('nii_tool_para');
for i = 1:numel(flds), val.(flds{i}) = pf.(flds{i}); end
if nargin<2, varargout{1} = val; return; end % query only
if nargout, varargout{1} = val; end % return old val
in2 = varargin;
if ~isstruct(in2), in2 = struct(in2{:}); end
nam = fieldnames(in2);
for i = 1:numel(nam)
ind = strcmpi(nam{i}, flds);
if isempty(ind), continue; end
para.(flds{ind}) = in2.(nam{i});
setpref('nii_tool_para', flds{ind}, in2.(nam{i}));
end
if val.version ~= para.version, C = niiHeader(para.version); end
elseif strcmpi(cmd, 'update') % old img2datatype subfunction
if nargin<2, error('nii_tool(''%s'') needs second input', cmd); end
nii = varargin{1};
if ~isstruct(nii) || ~isfield(nii, 'hdr') || ~isfield(nii, 'img')
error(['nii_tool(''update'') needs a struct from nii_tool(''init'')' ...
' or nii_tool(''load'') as the second input']);
end
dim = size(nii.img);
ndim = numel(dim);
dim(ndim+1:7) = 1;
if nargin>2 % set new sform mat
R = varargin{2};
if size(R,2)~=4, error('Invalid matrix dimension.'); end
nii.hdr.srow_x = R(1,:);
nii.hdr.srow_y = R(2,:);
nii.hdr.srow_z = R(3,:);
end
if ndim == 8 % RGB/RGBA data. Change img type to uint8/single if needed
valpix = dim(8);
if valpix == 4 % RGBA
typ = 'RGBA'; % error info only
nii.img = uint8(nii.img); % NIfTI only support uint8 for RGBA
elseif valpix == 3 % RGB, must be single or uint8
typ = 'RGB';
if max(nii.img(:))>1, nii.img = uint8(nii.img);
else, nii.img = single(nii.img);
end
else
error('Color dimension must have length of 3 for RGB or 4 for RGBA');
end
dim(8) = []; % remove color-dim so numel(dim)=7 for nii.hdr
ndim = find(dim>1, 1, 'last'); % update it
elseif isreal(nii.img)
typ = 'real';
valpix = 1;
else
typ = 'complex';
valpix = 2;
end
if islogical(nii.img), imgFmt = 'ubit1';
else, imgFmt = class(nii.img);
end
ind = find(strcmp(para.format, imgFmt) & para.valpix==valpix);
if isempty(ind) % only RGB and complex can have this problem
error('nii_tool does not support %s image of ''%s'' type', typ, imgFmt);
elseif numel(ind)>1 % unlikely
error('Non-unique datatype found for %s image of ''%s'' type', typ, imgFmt);
end
fmt = para.format{ind};
nii.hdr.datatype = para.datatype(ind);
nii.hdr.bitpix = para.bitpix(ind);
nii.hdr.dim = [ndim dim];
mx = double(max(nii.img(:)));
mn = double(min(nii.img(:)));
if nii.hdr.cal_min>mx || nii.hdr.cal_max<mn % reset wrong value
nii.hdr.cal_min = 0;
nii.hdr.cal_max = 0;
end
if nii.hdr.sizeof_hdr == 348
nii.hdr.glmax = round(mx); % we may remove these
nii.hdr.glmin = round(mn);
end
if isfield(nii, 'ext')
try swap = nii.hdr.swap_endian; catch, swap = false; end
nii.ext = decode_ext(nii.ext, swap);
end
varargout{1} = nii;
if nargout>1, varargout{2} = fmt; end
elseif strcmp(cmd, 'func_handle') % make a local function avail to outside
varargout{1} = str2func(varargin{1});
elseif strcmp(cmd, 'LocalFunc') % call local function from outside
[varargout{1:nargout}] = feval(varargin{:});
else
error('Invalid command for nii_tool: %s', cmd);
end
% End of main function
%% Subfunction: all nii header in the order in NIfTI-1/2 file
function [C, para] = niiHeader(niiVer)
pf = getpref('nii_tool_para');
if isempty(pf)
setpref('nii_tool_para', 'version', 1);
setpref('nii_tool_para', 'rgb_dim', 1);
pf = getpref('nii_tool_para');
end
if nargin<1 || isempty(niiVer), niiVer = pf.version; end
if niiVer == 1
C = {
% name len format value offset
'sizeof_hdr' 1 'int32' 348 0
'data_type' 10 'char' '' 4
'db_name' 18 'char' '' 14
'extents' 1 'int32' 16384 32
'session_error' 1 'int16' 0 36
'regular' 1 'char' 'r' 38
'dim_info' 1 'uint8' 0 39
'dim' 8 'int16' ones(1,8) 40
'intent_p1' 1 'single' 0 56
'intent_p2' 1 'single' 0 60
'intent_p3' 1 'single' 0 64
'intent_code' 1 'int16' 0 68
'datatype' 1 'int16' 0 70
'bitpix' 1 'int16' 0 72
'slice_start' 1 'int16' 0 74
'pixdim' 8 'single' zeros(1,8) 76
'vox_offset' 1 'single' 352 108
'scl_slope' 1 'single' 1 112
'scl_inter' 1 'single' 0 116
'slice_end' 1 'int16' 0 120
'slice_code' 1 'uint8' 0 122
'xyzt_units' 1 'uint8' 0 123
'cal_max' 1 'single' 0 124
'cal_min' 1 'single' 0 128
'slice_duration' 1 'single' 0 132
'toffset' 1 'single' 0 136
'glmax' 1 'int32' 0 140
'glmin' 1 'int32' 0 144
'descrip' 80 'char' '' 148
'aux_file' 24 'char' '' 228
'qform_code' 1 'int16' 0 252
'sform_code' 1 'int16' 0 254
'quatern_b' 1 'single' 0 256
'quatern_c' 1 'single' 0 260
'quatern_d' 1 'single' 0 264
'qoffset_x' 1 'single' 0 268
'qoffset_y' 1 'single' 0 272
'qoffset_z' 1 'single' 0 276
'srow_x' 4 'single' [1 0 0 0] 280
'srow_y' 4 'single' [0 1 0 0] 296
'srow_z' 4 'single' [0 0 1 0] 312
'intent_name' 16 'char' '' 328
'magic' 4 'char' 'n+1' 344
'extension' 4 'uint8' [0 0 0 0] 348
};
elseif niiVer == 2
C = {
'sizeof_hdr' 1 'int32' 540 0
'magic' 8 'char' 'n+2' 4
'datatype' 1 'int16' 0 12
'bitpix' 1 'int16' 0 14
'dim' 8 'int64' ones(1,8) 16
'intent_p1' 1 'double' 0 80
'intent_p2' 1 'double' 0 88
'intent_p3' 1 'double' 0 96
'pixdim' 8 'double' zeros(1,8) 104
'vox_offset' 1 'int64' 544 168
'scl_slope' 1 'double' 1 176
'scl_inter' 1 'double' 0 184
'cal_max' 1 'double' 0 192
'cal_min' 1 'double' 0 200
'slice_duration' 1 'double' 0 208
'toffset' 1 'double' 0 216
'slice_start' 1 'int64' 0 224
'slice_end' 1 'int64' 0 232
'descrip' 80 'char' '' 240
'aux_file' 24 'char' '' 320
'qform_code' 1 'int32' 0 344
'sform_code' 1 'int32' 0 348
'quatern_b' 1 'double' 0 352
'quatern_c' 1 'double' 0 360
'quatern_d' 1 'double' 0 368
'qoffset_x' 1 'double' 0 376
'qoffset_y' 1 'double' 0 384
'qoffset_z' 1 'double' 0 392
'srow_x' 4 'double' [1 0 0 0] 400
'srow_y' 4 'double' [0 1 0 0] 432
'srow_z' 4 'double' [0 0 1 0] 464
'slice_code' 1 'int32' 0 496
'xyzt_units' 1 'int32' 0 500
'intent_code' 1 'int32' 0 504
'intent_name' 16 'char' '' 508
'dim_info' 1 'uint8' 0 524
'unused_str' 15 'char' '' 525
'extension' 4 'uint8' [0 0 0 0] 540
};
else
error('Nifti version %g is not supported', niiVer);
end
if nargout<2, return; end
% class datatype bitpix valpix
D = {
'ubit1' 1 1 1 % neither mricron nor fsl support this
'uint8' 2 8 1
'int16' 4 16 1
'int32' 8 32 1
'single' 16 32 1
'single' 32 64 2 % complex
'double' 64 64 1
'uint8' 128 24 3 % RGB
'int8' 256 8 1
'single' 511 96 3 % RGB, not in NIfTI standard?
'uint16' 512 16 1
'uint32' 768 32 1
'int64' 1024 64 1
'uint64' 1280 64 1
% 'float128' 1536 128 1 % long double, for 22nd century?
'double' 1792 128 2 % complex
% 'float128' 2048 256 2 % long double complex
'uint8' 2304 32 4 % RGBA
};
para.format = D(:,1)';
para.datatype = [D{:,2}];
para.bitpix = [D{:,3}];
para.valpix = [D{:,4}];
para.rgb_dim = pf.rgb_dim; % dim of RGB/RGBA in NIfTI FILE
para.version = niiVer;
%% Subfunction: use pigz or system gzip if available (faster)
function gzipOS(fname)
persistent cmd; % command to gzip
if isempty(cmd)
cmd = check_gzip('gzip');
if ischar(cmd)
cmd = @(nam){cmd '-nf' nam};
elseif islogical(cmd) && ~cmd
fprintf(2, ['None of system pigz, gzip or Matlab gzip available. ' ...
'Files are not compressed into gz.\n']);
end
end
if islogical(cmd)
if cmd, gzip(fname); deleteFile(fname); end
return;
end
[err, str] = jsystem(cmd(fname));
if err && ~exist(strcat(fname, '.gz'), 'file')
try
gzip(fname); deleteFile(fname);
catch
fprintf(2, 'Error during compression: %s\n', str);
end
end
%% Deal with pigz/gzip on path or in nii_tool folder, and matlab gzip/gunzip
function cmd = check_gzip(gz_unzip)
m_dir = fileparts(mfilename('fullpath'));
% away from pwd, so use OS pigz if both exist. Avoid error if pwd changed later
if strcmpi(pwd, m_dir), cd ..; clnObj = onCleanup(@() cd(m_dir)); end
if isunix
pth1 = getenv('PATH');
if isempty(strfind(pth1, '/usr/local/bin'))
pth1 = [pth1 ':/usr/local/bin'];
setenv('PATH', pth1);
end
end
% first, try system pigz
[err, ~] = jsystem({'pigz' '-V'});
if ~err, cmd = 'pigz'; return; end
% next, try pigz included with nii_tool
cmd = [m_dir '/pigz'];
if ismac % pigz for mac is not included in the package
if strcmp(gz_unzip, 'gzip')
fprintf(2, [' Please install pigz for fast compression: ' ...
'http://macappstore.org/pigz/\n']);
end
elseif isunix % linux
[st, val] = fileattrib(cmd);
if st && ~val.UserExecute, fileattrib(cmd, '+x'); end
end
[err, ~] = jsystem({cmd '-V'});
if ~err, return; end
% Third, try system gzip/gunzip
[err, ~] = jsystem({gz_unzip '-V'}); % gzip/gunzip on system path?
if ~err, cmd = gz_unzip; return; end
% Lastly, use Matlab gzip/gunzip if java avail
cmd = usejava('jvm');
%% check dd command, return empty if not available
function dd = check_dd
m_dir = fileparts(mfilename('fullpath'));
if strcmpi(pwd, m_dir), cd ..; clnObj = onCleanup(@() cd(m_dir)); end
[err, ~] = jsystem({'dd' '--version'});
if ~err, dd = 'dd'; return; end % dd with linix/mac, and maybe windows
if ispc % rename it as exe
dd = [m_dir '\dd'];
[err, ~] = jsystem({dd '--version'});
if ~err, return; end
end
dd = '';
%% Try to use in order of pigz, system gunzip, then matlab gunzip
function outName = gunzipOS(fname, nByte)
persistent cmd dd pth uid; % command to run gupzip, dd tool, and temp_path
if isempty(cmd)
cmd = check_gzip('gunzip'); % gzip -dc has problem in PC
if ischar(cmd)
cmd = @(nam)sprintf('"%s" -nfdc "%s" ', cmd, nam); % f for overwrite
elseif islogical(cmd) && ~cmd
cmd = [];
error('None of system pigz, gunzip or Matlab gunzip is available');
end
dd = check_dd;
if ~isempty(dd)
dd = @(n,out)sprintf('| "%s" count=%g of="%s"', dd, ceil(n/512), out);
end
if ispc % matlab tempdir could be slow due to cd in and out
pth = getenv('TEMP');
if isempty(pth), pth = pwd; end
else
pth = getenv('TMP');
if isempty(pth), pth = getenv('TMPDIR'); end
if isempty(pth), pth = '/tmp'; end % last resort
end
uid = @()sprintf('_%s_%03x', datestr(now, 'yymmddHHMMSSfff'), randi(999));
end
fname = char(fname);
if islogical(cmd)
outName = gunzip(fname, pth);
outName = outName{1};
return;
end
[~, outName, ext] = fileparts(fname);
if strcmpi(ext, '.gz') % likely always true
[~, outName, ext1] = fileparts(outName);
outName = [outName uid() ext1];
else
outName = [outName uid()];
end
outName = fullfile(pth, outName);
if ~isempty(dd) && nargin>1 && ~isinf(nByte) % unzip only part of data
try
[err, ~] = system([cmd(fname) dd(nByte, outName)]);
if err==0, return; end
end
end
[err, str] = system([cmd(fname) '> "' outName '"']);
% [err, str] = jsystem({'pigz' '-nfdc' fname '>' outName});
if err
try
outName = gunzip(fname, pth);
catch
error('Error during gunzip:\n%s', str);
end
end
%% cast bytes into a type, swapbytes if needed
function out = cast_swap(b, typ, swap)
out = typecast(b, typ);
if swap, out = swapbytes(out); end
out = double(out); % for convenience
%% subfunction: read hdr
function hdr = read_hdr(b, C, fname)
n = typecast(b(1:4), 'int32');
if n==348, niiVer = 1; swap = false;
elseif n==540, niiVer = 2; swap = false;
else
n = swapbytes(n);
if n==348, niiVer = 1; swap = true;
elseif n==540, niiVer = 2; swap = true;
else, error('Not valid NIfTI file: %s', fname);
end
end
if niiVer>1, C = niiHeader(niiVer); end % C defaults for version 1
for i = 1:size(C,1)
try a = b(C{i,5}+1 : C{i+1,5});
catch
if C{i,5}==numel(b), a = [];
else, a = b(C{i,5} + (1:C{i,2})); % last item extension is in bytes
end
end
if strcmp(C{i,3}, 'char')
a = deblank(native2unicode(a));
else
a = cast_swap(a, C{i,3}, swap);
a = double(a);
end
hdr.(C{i,1}) = a;
end
hdr.version = niiVer; % for 'save', unless user asks to change
hdr.swap_endian = swap;
hdr.file_name = fname;
%% subfunction: read ext, and decode it if known ecode
function ext = read_ext(b, hdr)
ext = []; % avoid error if no ext but hdr.extension(1) was set
nEnd = hdr.vox_offset;
if nEnd == 0, nEnd = numel(b); end % .hdr file
swap = hdr.swap_endian;
j = hdr.sizeof_hdr + 4; % 4 for hdr.extension
while j < nEnd