Files
ar_dso/src/OptimizationBackend/EnergyFunctionalStructs.h
Ivan e4c8529305 v1
2022-06-28 10:36:24 +03:00

169 lines
3.1 KiB
C++

/**
* This file is part of DSO.
*
* Copyright 2016 Technical University of Munich and Intel.
* Developed by Jakob Engel <engelj at in dot tum dot de>,
* for more information see <http://vision.in.tum.de/dso>.
* If you use this code, please cite the respective publications as
* listed on the above website.
*
* DSO is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* DSO is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with DSO. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
#include "util/NumType.h"
#include "vector"
#include <math.h>
#include "OptimizationBackend/RawResidualJacobian.h"
namespace dso
{
class PointFrameResidual;
class CalibHessian;
class FrameHessian;
class PointHessian;
class EFResidual;
class EFPoint;
class EFFrame;
class EnergyFunctional;
class EFResidual
{
public:
EIGEN_MAKE_ALIGNED_OPERATOR_NEW;
inline EFResidual(PointFrameResidual* org, EFPoint* point_, EFFrame* host_, EFFrame* target_) :
data(org), point(point_), host(host_), target(target_)
{
isLinearized=false;
isActiveAndIsGoodNEW=false;
J = new RawResidualJacobian();
assert(((long)this)%16==0);
assert(((long)J)%16==0);
}
inline ~EFResidual()
{
delete J;
}
void takeDataF();
void fixLinearizationF(EnergyFunctional* ef);
// structural pointers
PointFrameResidual* data;
int hostIDX, targetIDX;
EFPoint* point;
EFFrame* host;
EFFrame* target;
int idxInAll;
RawResidualJacobian* J;
VecNRf res_toZeroF;
Vec8f JpJdF;
// status.
bool isLinearized;
// if residual is not OOB & not OUTLIER & should be used during accumulations
bool isActiveAndIsGoodNEW;
inline const bool &isActive() const {return isActiveAndIsGoodNEW;}
};
enum EFPointStatus {PS_GOOD=0, PS_MARGINALIZE, PS_DROP};
class EFPoint
{
public:
EIGEN_MAKE_ALIGNED_OPERATOR_NEW;
EFPoint(PointHessian* d, EFFrame* host_) : data(d),host(host_)
{
takeData();
stateFlag=EFPointStatus::PS_GOOD;
}
void takeData();
PointHessian* data;
float priorF;
float deltaF;
// constant info (never changes in-between).
int idxInPoints;
EFFrame* host;
// contains all residuals.
std::vector<EFResidual*> residualsAll;
float bdSumF;
float HdiF;
float Hdd_accLF;
VecCf Hcd_accLF;
float bd_accLF;
float Hdd_accAF;
VecCf Hcd_accAF;
float bd_accAF;
EFPointStatus stateFlag;
};
class EFFrame
{
public:
EIGEN_MAKE_ALIGNED_OPERATOR_NEW;
EFFrame(FrameHessian* d) : data(d)
{
takeData();
}
void takeData();
Vec8 prior; // prior hessian (diagonal)
Vec8 delta_prior; // = state-state_prior (E_prior = (delta_prior)' * diag(prior) * (delta_prior)
Vec8 delta; // state - state_zero.
std::vector<EFPoint*> points;
FrameHessian* data;
int idx; // idx in frames.
int frameID;
};
}