File: TestImageFragmentSplitter.cxx

package info (click to toggle)
gdcm 3.0.24-5
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 27,560 kB
  • sloc: cpp: 203,722; ansic: 76,471; xml: 48,131; python: 3,473; cs: 2,308; java: 1,629; lex: 1,290; sh: 334; php: 128; makefile: 97
file content (129 lines) | stat: -rw-r--r-- 4,159 bytes parent folder | download | duplicates (8)
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
/*=========================================================================

  Program: GDCM (Grassroots DICOM). A DICOM library

  Copyright (c) 2006-2011 Mathieu Malaterre
  All rights reserved.
  See Copyright.txt or http://gdcm.sourceforge.net/Copyright.html for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/
#include "gdcmImageFragmentSplitter.h"
#include "gdcmTesting.h"
#include "gdcmSystem.h"
#include "gdcmImageReader.h"
#include "gdcmImageWriter.h"
#include "gdcmTransferSyntax.h"
#include "gdcmImage.h"
#include "gdcmFilename.h"

int TestImageFragmentSplitterFunc(const char *filename, bool verbose = false)
{
  gdcm::ImageReader reader;
  reader.SetFileName( filename );
  if ( !reader.Read() )
    {
    const gdcm::FileMetaInformation &header = reader.GetFile().GetHeader();
    gdcm::MediaStorage ms = header.GetMediaStorage();
    bool isImage = gdcm::MediaStorage::IsImage( ms );
    if( isImage )
      {
      if( reader.GetFile().GetDataSet().FindDataElement( gdcm::Tag(0x7fe0,0x0010) ) )
        {
        std::cerr << "Could not read: " << filename << std::endl;
        return 1;
        }
      }
    return 0;
    }
  const gdcm::Image &image = reader.GetImage();
  const unsigned int *dims = image.GetDimensions();
  if( dims[2] != 1 )
    {
    return 0; // nothing to do
    }
  const gdcm::File &file = reader.GetFile();
  const gdcm::FileMetaInformation &header = file.GetHeader();
  //gdcm::MediaStorage ms = header.GetMediaStorage();
  if(  header.GetDataSetTransferSyntax() == gdcm::TransferSyntax::ImplicitVRLittleEndian
    || header.GetDataSetTransferSyntax() == gdcm::TransferSyntax::ImplicitVRBigEndianPrivateGE
    || header.GetDataSetTransferSyntax() == gdcm::TransferSyntax::ExplicitVRLittleEndian
    || header.GetDataSetTransferSyntax() == gdcm::TransferSyntax::DeflatedExplicitVRLittleEndian
    || header.GetDataSetTransferSyntax() == gdcm::TransferSyntax::ExplicitVRBigEndian
  )
    {
    return 0; // nothing to do
    }

  gdcm::ImageFragmentSplitter splitter;
  splitter.SetInput( image );
  splitter.SetFragmentSizeMax( 65536 );
  bool b = splitter.Split();
  if( !b )
    {
    const gdcm::DataElement &pixeldata = file.GetDataSet().GetDataElement( gdcm::Tag(0x7fe0,0x0010) );
    const gdcm::SequenceOfFragments* sqf = pixeldata.GetSequenceOfFragments();
    if( sqf && dims[2] == 1 )
      {
      return 0;
      }
    gdcm::Filename fn( filename );
    if( fn.GetName() == std::string("JPEGDefinedLengthSequenceOfFragments.dcm" ) )
      {
      // JPEG Fragments are packed in a VR:OB Attribute
      return 0;
      }
    std::cerr << "Could not apply splitter: " << filename << std::endl;
    return 1;
    }

  // Create directory first:
  const char subdir[] = "TestImageFragmentSplitter";
  std::string tmpdir = gdcm::Testing::GetTempDirectory( subdir );
  if( !gdcm::System::FileIsDirectory( tmpdir.c_str() ) )
    {
    gdcm::System::MakeDirectory( tmpdir.c_str() );
    //return 1;
    }
  std::string outfilename = gdcm::Testing::GetTempFilename( filename, subdir );

  gdcm::ImageWriter writer;
  writer.SetFileName( outfilename.c_str() );
  //writer.SetFile( reader.GetFile() ); // increase test goal
  writer.SetImage( splitter.GetOutput() );
  if( !writer.Write() )
    {
    std::cerr << "Failed to write: " << outfilename << std::endl;
    return 1;
    }
  if( verbose )
    std::cout << "Success to write: " << outfilename << std::endl;

  return 0;
}

int TestImageFragmentSplitter(int argc, char *argv[])
{
  if( argc == 2 )
    {
    const char *filename = argv[1];
    return TestImageFragmentSplitterFunc(filename, true);
    }

  // else
  gdcm::Trace::DebugOff();
  gdcm::Trace::WarningOff();
  int r = 0, i = 0;
  const char *filename;
  const char * const *filenames = gdcm::Testing::GetFileNames();
  while( (filename = filenames[i]) )
    {
    r += TestImageFragmentSplitterFunc( filename );
    ++i;
    }

  return r;
}