int main(int argc, char* argv[])
{
// parse arguments
const char* optstring = "o:i:l:h";
static struct option longOptions[] =
{
{"output", required_argument, NULL, 'o' },
{"image", required_argument, NULL, 'i' },
{"lines", required_argument, NULL, 'l' },
{"help", no_argument, NULL, 'h' },
0
};
UIntSet cmdlineImages;
int c;
int optionIndex = 0;
int nrLines = 5;
string output;
while ((c = getopt_long (argc, argv, optstring, longOptions,&optionIndex)) != -1)
{
switch (c)
{
case 'o':
output = optarg;
break;
case 'h':
usage(argv[0]);
return 0;
case 'i':
{
int imgNr=atoi(optarg);
if((imgNr==0) && (strcmp(optarg,"0")!=0))
{
cerr << "Could not parse image number.";
return 1;
};
cmdlineImages.insert(imgNr);
};
break;
case 'l':
nrLines=atoi(optarg);
if(nrLines<1)
{
cerr << "Could not parse number of lines.";
return 1;
};
break;
case ':':
cerr <<"Option " << longOptions[optionIndex].name << " requires a number" << endl;
return 1;
break;
case '?':
break;
default:
abort ();
}
}
if (argc - optind != 1)
{
cout << "Warning: " << argv[0] << " can only work on one project file at one time" << endl << endl;
usage(argv[0]);
return 1;
};
string input=argv[optind];
// read panorama
Panorama pano;
ifstream prjfile(input.c_str());
if (!prjfile.good())
{
cerr << "could not open script : " << input << endl;
return 1;
}
pano.setFilePrefix(hugin_utils::getPathPrefix(input));
DocumentData::ReadWriteError err = pano.readData(prjfile);
if (err != DocumentData::SUCCESSFUL)
{
cerr << "error while parsing panos tool script: " << input << endl;
cerr << "DocumentData::ReadWriteError code: " << err << endl;
return 1;
}
if(pano.getNrOfImages()==0)
{
cerr << "error: project file does not contains any image" << endl;
cerr << "aborting processing" << endl;
return 1;
};
std::vector<size_t> imagesToProcess;
if(cmdlineImages.size()==0)
{
//no image given, process all
for(size_t i=0;i<pano.getNrOfImages();i++)
{
imagesToProcess.push_back(i);
};
}
else
{
//check, if given image numbers are valid
for(UIntSet::const_iterator it=cmdlineImages.begin();it!=cmdlineImages.end();it++)
{
if((*it)>=0 && (*it)<pano.getNrOfImages())
{
imagesToProcess.push_back(*it);
};
};
};
if(imagesToProcess.size()==0)
{
cerr << "No image to process found" << endl << "Stopping processing" << endl;
return 1;
};
PT_setProgressFcn(ptProgress);
PT_setInfoDlgFcn(ptinfoDlg);
cout << argv[0] << " is searching for vertical lines" << endl;
#if _WINDOWS
//multi threading of image loading results sometime in a race condition
//try to prevent this by initialisation of codecManager before
//running multi threading part
std::string s=vigra::impexListExtensions();
#endif
#ifdef HAS_PPL
size_t nrCPS=pano.getNrOfCtrlPoints();
Concurrency::parallel_for<size_t>(0,imagesToProcess.size(),[&pano,imagesToProcess,nrLines](size_t i)
#else
for(size_t i=0;i<imagesToProcess.size();i++)
#endif
{
unsigned int imgNr=imagesToProcess[i];
cout << "Working on image " << pano.getImage(imgNr).getFilename() << endl;
// now load and process all images
vigra::ImageImportInfo info(pano.getImage(imgNr).getFilename().c_str());
HuginBase::CPVector foundLines;
if(info.isGrayscale())
{
foundLines=LoadGrayImageAndFindLines(info, pano, imgNr, nrLines);
}
else
{
if(info.isColor())
{
//colour images
foundLines=LoadImageAndFindLines(info, pano, imgNr, nrLines);
}
else
{
std::cerr << "Image " << pano.getImage(imgNr).getFilename().c_str() << " has "
<< info.numBands() << " channels." << std::endl
<< "Linefind works only with grayscale or color images." << std::endl
<< "Skipping image." << std::endl;
};
};
#ifndef HAS_PPL
cout << "Found " << foundLines.size() << " vertical lines" << endl;
#endif
if(foundLines.size()>0)
{
for(CPVector::const_iterator cpIt=foundLines.begin(); cpIt!=foundLines.end(); cpIt++)
{
pano.addCtrlPoint(*cpIt);
};
};
}
#ifdef HAS_PPL
);
