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
|
/*
* Copyright (C) 2005-2020 Centre National d'Etudes Spatiales (CNES)
*
* This file is part of Orfeo Toolbox
*
* https://www.orfeo-toolbox.org/
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "otbLineSegmentDetector.h"
#include "otbImageFileReader.h"
#include "otbVectorDataFileWriter.h"
#include "otbImageFileWriter.h"
int otbLineSegmentDetector(int argc, char* argv[])
{
const char* infname = argv[1];
const char* outfname = argv[2];
/** Typedefs */
typedef unsigned char PixelType;
typedef otb::Image<PixelType> ImageType;
typedef otb::ImageFileReader<ImageType> ReaderType;
typedef otb::LineSegmentDetector<ImageType, double> LSDFilterType;
typedef LSDFilterType::VectorDataType VectorDataType;
typedef otb::VectorDataFileWriter<VectorDataType> VectorDataWriterType;
typedef otb::ImageFileWriter<ImageType> WriterType;
/** Instantiation of smart pointer*/
ReaderType::Pointer reader = ReaderType::New();
LSDFilterType::Pointer lsdFilter = LSDFilterType::New();
// Reade the input image
reader->SetFileName(infname);
reader->GenerateOutputInformation();
// LSD Detection
lsdFilter->SetInput(reader->GetOutput());
VectorDataWriterType::Pointer vdWriter = VectorDataWriterType::New();
vdWriter->SetFileName(outfname);
vdWriter->SetInput(lsdFilter->GetOutput());
vdWriter->Update();
if (argc > 3)
{
// Write the status map
const char* regfname = argv[3];
WriterType::Pointer writer0 = WriterType::New();
writer0->SetFileName(regfname);
writer0->SetInput(lsdFilter->GetMap());
writer0->Update();
// Write the gradient modulus image
typedef otb::ImageFileWriter<otb::Image<double>> WriterDoubleType;
const char* gradModfname = argv[4];
WriterDoubleType::Pointer writer1 = WriterDoubleType::New();
writer1->SetFileName(gradModfname);
writer1->SetInput(lsdFilter->GetGradMod());
writer1->Update();
// Write the gradient orientation image
const char* gradOrifname = argv[5];
WriterDoubleType::Pointer writer2 = WriterDoubleType::New();
writer1->SetFileName(gradOrifname);
writer1->SetInput(lsdFilter->GetGradOri());
writer1->Update();
}
return EXIT_SUCCESS;
}
int otbLineSegmentDetector_8b_16b_compare(int itkNotUsed(argc), char* argv[])
{
const char* infname8 = argv[1];
const char* infname16 = argv[2];
const char* outfname8 = argv[3];
const char* outfname16 = argv[4];
/** Typedefs */
typedef unsigned char PixelType8;
typedef otb::Image<PixelType8> ImageType8;
typedef otb::ImageFileReader<ImageType8> ReaderType8;
typedef otb::LineSegmentDetector<ImageType8, double> LSDFilterType8;
typedef unsigned short PixelType16;
typedef otb::Image<PixelType16> ImageType16;
typedef otb::ImageFileReader<ImageType16> ReaderType16;
typedef otb::LineSegmentDetector<ImageType16, double> LSDFilterType16;
typedef LSDFilterType8::VectorDataType VectorDataType;
typedef otb::VectorDataFileWriter<VectorDataType> VectorDataWriterType;
/** 8bits */
// Instantiation of smart pointer
ReaderType8::Pointer reader8 = ReaderType8::New();
LSDFilterType8::Pointer lsdFilter8 = LSDFilterType8::New();
// Reade the input image
reader8->SetFileName(infname8);
reader8->GenerateOutputInformation();
// LSD Detection
lsdFilter8->SetInput(reader8->GetOutput());
VectorDataWriterType::Pointer vdWriter8 = VectorDataWriterType::New();
vdWriter8->SetFileName(outfname8);
vdWriter8->SetInput(lsdFilter8->GetOutput());
vdWriter8->Update();
/** 16bits */
// Instantiation of smart pointer
ReaderType16::Pointer reader16 = ReaderType16::New();
LSDFilterType16::Pointer lsdFilter16 = LSDFilterType16::New();
// Reade the input image
reader16->SetFileName(infname16);
reader16->GenerateOutputInformation();
// LSD Detection
lsdFilter16->SetInput(reader16->GetOutput());
VectorDataWriterType::Pointer vdWriter16 = VectorDataWriterType::New();
vdWriter16->SetFileName(outfname16);
vdWriter16->SetInput(lsdFilter16->GetOutput());
vdWriter16->Update();
return EXIT_SUCCESS;
}
|