unsigned int optimize(PanoramaData& pano,
const char * userScript)
{
char * script = 0;
unsigned int retval = 0;
if (userScript == 0) {
std::ostringstream scriptbuf;
UIntSet allImg;
fill_set(allImg,0, unsigned(pano.getNrOfImages()-1));
pano.printPanoramaScript(scriptbuf, pano.getOptimizeVector(), pano.getOptions(), allImg, true);
script = strdup(scriptbuf.str().c_str());
} else {
script = const_cast<char *>(userScript);
}
OptInfo opt;
AlignInfo ainf;
if (ParseScript( script, &ainf ) == 0)
{
if( CheckParams( &ainf ) == 0 )
{
ainf.fcn = fcnPano;
SetGlobalPtr( &ainf );
opt.numVars = ainf.numParam;
opt.numData = ainf.numPts;
opt.SetVarsToX = SetLMParams;
opt.SetXToVars = SetAlignParams;
opt.fcn = ainf.fcn;
*opt.message = 0;
RunLMOptimizer( &opt );
ainf.data = opt.message;
// get results from align info.
#ifdef DEBUG_WRITE_OPTIM_OUTPUT
fullPath path;
StringtoFullPath(&path, DEBUG_WRITE_OPTIM_OUTPUT_FILE );
ainf.data = opt.message;
WriteResults( script, &path, &ainf, distSquared, 0);
#endif
pano.updateVariables( GetAlignInfoVariables(ainf) );
pano.updateCtrlPointErrors( GetAlignInfoCtrlPoints(ainf) );
} else {
std::cerr << "Bad params" << std::endl;
retval = 2;
}
DisposeAlignInfo( &ainf );
} else {
std::cerr << "Bad params" << std::endl;
retval = 1;
}
if (! userScript) {
free(script);
}
return retval;
}
UIntSet getImagesinROI (const PanoramaData& pano, const UIntSet activeImages)
{
UIntSet images;
PanoramaOptions opts = pano.getOptions();
for (UIntSet::const_iterator it = activeImages.begin(); it != activeImages.end(); ++it)
{
vigra::Rect2D roi = estimateOutputROI(pano, opts, *it);
if (! (roi.isEmpty())) {
images.insert(*it);
}
}
return images;
}
void SmartPhotometricOptimizer::smartOptimizePhotometric(PanoramaData & pano, PhotometricOptimizeMode mode,
const std::vector<vigra_ext::PointPairRGB> & correspondences,
AppBase::ProgressReporter & progress,
double & error)
{
std::vector<SrcPanoImage> ret;
PanoramaOptions opts = pano.getOptions();
if (mode == OPT_PHOTOMETRIC_LDR || mode == OPT_PHOTOMETRIC_LDR_WB) {
// optimize exposure
int vars = OPT_EXP;
optimizePhotometric(pano,
createOptVars(pano, vars, opts.colorReferenceImage),
correspondences, progress, error);
// optimize vignetting & exposure
vars = OPT_EXP | OPT_VIG;
optimizePhotometric(pano,
createOptVars(pano, vars, opts.colorReferenceImage),
correspondences, progress, error);
if (mode == OPT_PHOTOMETRIC_LDR_WB) {
// optimize vignetting & exposure & wb & response
vars = OPT_EXP | OPT_VIG | OPT_RESP | OPT_WB;
} else {
vars = OPT_EXP | OPT_VIG | OPT_RESP;
}
// optimize vignetting & exposure & wb & response
optimizePhotometric(pano,
createOptVars(pano, vars, opts.colorReferenceImage),
correspondences, progress, error);
} else if (mode == OPT_PHOTOMETRIC_HDR || mode == OPT_PHOTOMETRIC_HDR_WB) {
// optimize vignetting
int vars = OPT_VIG;
optimizePhotometric(pano,
createOptVars(pano, vars, opts.colorReferenceImage),
correspondences, progress, error);
// optimize vignetting, wb and response
if (mode == OPT_PHOTOMETRIC_HDR_WB) {
vars = OPT_VIG | OPT_RESP | OPT_WB;
} else {
vars = OPT_VIG | OPT_RESP;
}
optimizePhotometric(pano,
createOptVars(pano, vars, opts.colorReferenceImage),
correspondences, progress, error);
} else {
assert(0 && "Unknown photometric optimisation mode");
}
// adjust
}
void calcCtrlPointErrors (PanoramaData& pano)
{
if(pano.getNrOfImages()>0 && pano.getNrOfCtrlPoints()>0)
{
char * p=setlocale(LC_ALL,NULL);
#ifndef ANDROID
char * oldlocale=strdup(p);
#else
char * oldlocale="";
#endif
setlocale(LC_ALL,"C");
UIntSet allImg;
std::ostringstream scriptbuf;
fill_set(allImg,0, unsigned(pano.getNrOfImages()-1));
//create temporary non-empty optimize vector
OptimizeVector optVec;
std::set<std::string> opt;
opt.insert("y");
for(unsigned int i=0;i<pano.getNrOfImages();i++)
{
optVec.push_back(opt);
};
pano.printPanoramaScript(scriptbuf, optVec,
pano.getOptions(), allImg, true);
char * script = 0;
script = strdup(scriptbuf.str().c_str());
AlignInfo ainf;
if (ParseScript( script, &ainf ) == 0)
{
if( CheckParams( &ainf ) == 0 )
{
ainf.fcn = fcnPano;
SetGlobalPtr( &ainf );
pano.updateCtrlPointErrors( GetAlignInfoCtrlPoints(ainf) );
}
}
setlocale(LC_ALL,oldlocale);
free(oldlocale);
};
}
FDiff2D CalculateFOV::calcFOV(const PanoramaData& panorama)
{
if (panorama.getNrOfImages() == 0) {
// no change
return FDiff2D(panorama.getOptions().getHFOV(), panorama.getOptions().getVFOV());
}
vigra::Size2D panoSize(360*2,180*2);
// remap into minature pano.
PanoramaOptions opts;
opts.setHFOV(360);
opts.setProjection(PanoramaOptions::EQUIRECTANGULAR);
opts.setWidth(panoSize.x);
opts.setHeight(panoSize.y);
// remap image
// DGSW - make sure the type is correct
vigra::BImage panoAlpha(panoSize.x, panoSize.y,static_cast< unsigned char >(0));
// vigra::BImage panoAlpha(panoSize.x, panoSize.y,0);
Nona::RemappedPanoImage<vigra::BImage, vigra::BImage> remapped;
UIntSet activeImgs = panorama.getActiveImages();
for (UIntSet::iterator it = activeImgs.begin(); it != activeImgs.end(); ++it) {
// for (unsigned int imgNr=0; imgNr < getNrOfImages(); imgNr++) {
// DGSW FIXME - Unreferenced
// const PanoImage & img = getImage(*it);
remapped.setPanoImage(panorama.getSrcImage(*it), opts, vigra::Rect2D(0,0,panoSize.x,panoSize.y));
//remapped.setPanoImage(*this, *it, vigra::Size2D(img.getWidth(), img.getHeight()), opts);
// calculate alpha channel
remapped.calcAlpha();
// copy into global alpha channel.
vigra::copyImageIf(vigra_ext::applyRect(remapped.boundingBox(),
vigra_ext::srcMaskRange(remapped)),
vigra_ext::applyRect(remapped.boundingBox(),
vigra_ext::srcMask(remapped)),
vigra_ext::applyRect(remapped.boundingBox(),
destImage(panoAlpha)));
// vigra::ImageExportInfo imge2("c:/hugin_calcfov_alpha.png");
// exportImage(vigra::srcImageRange(panoAlpha), imge2);
}
// get field of view
FDiff2D ul,lr;
bool found = false;
ul.x = DBL_MAX;
ul.y = DBL_MAX;
lr.x = -DBL_MAX;
lr.y = -DBL_MAX;
for (int v=0; v< panoSize.y; v++) {
for (int h=0; h < panoSize.x; h++) {
if (panoAlpha(h,v)) {
// pixel is valid
if ( ul.x > h ) {
found=true;
ul.x = h;
}
if ( ul.y > v ) {
found=true;
ul.y = v;
}
if ( lr.x < h) {
found=true;
lr.x = h;
}
if ( lr.y < v) {
found=true;
lr.y = v;
}
}
}
}
if (!found) {
// if nothing found, return current fov
return FDiff2D(panorama.getOptions().getHFOV(), panorama.getOptions().getVFOV());
}
ul=ul/2.0;
lr=lr/2.0;
ul.x = ul.x - 180;
ul.y = ul.y - 90;
lr.x = lr.x - 180;
lr.y = lr.y - 90;
FDiff2D fov (2*std::max(fabs(ul.x), fabs(lr.x)), 2*std::max(fabs(ul.y), fabs(lr.y)));
if(fov.x<40)
{
fov.x+=1;
};
return fov;
}
