179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// Ceres Solver - A fast non-linear least squares minimizer 279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// Copyright 2014 Google Inc. All rights reserved. 379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// http://code.google.com/p/ceres-solver/ 479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// 579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// Redistribution and use in source and binary forms, with or without 679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// modification, are permitted provided that the following conditions are met: 779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// 879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// * Redistributions of source code must retain the above copyright notice, 979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// this list of conditions and the following disclaimer. 1079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// * Redistributions in binary form must reproduce the above copyright notice, 1179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// this list of conditions and the following disclaimer in the documentation 1279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// and/or other materials provided with the distribution. 1379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// * Neither the name of Google Inc. nor the names of its contributors may be 1479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// used to endorse or promote products derived from this software without 1579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// specific prior written permission. 1679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// 1779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 1879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 1979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 2079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 2179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 2279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 2379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 2479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 2579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 2679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 2779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// POSSIBILITY OF SUCH DAMAGE. 2879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// 2979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez// Author: richie.stebbing@gmail.com (Richard Stebbing) 3079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 3179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez#include "ceres/compressed_row_jacobian_writer.h" 3279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez#include "ceres/dynamic_compressed_row_jacobian_writer.h" 3379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez#include "ceres/casts.h" 3479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez#include "ceres/dynamic_compressed_row_sparse_matrix.h" 3579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez#include "ceres/parameter_block.h" 3679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez#include "ceres/program.h" 3779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez#include "ceres/residual_block.h" 3879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 3979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandeznamespace ceres { 4079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandeznamespace internal { 4179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 4279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos HernandezScratchEvaluatePreparer* 4379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos HernandezDynamicCompressedRowJacobianWriter::CreateEvaluatePreparers(int num_threads) { 4479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez return ScratchEvaluatePreparer::Create(*program_, num_threads); 4579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez} 4679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 4779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos HernandezSparseMatrix* DynamicCompressedRowJacobianWriter::CreateJacobian() const { 4879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez // Initialize `jacobian` with zero number of `max_num_nonzeros`. 4979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez const int num_residuals = program_->NumResiduals(); 5079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez const int num_effective_parameters = program_->NumEffectiveParameters(); 5179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 5279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez DynamicCompressedRowSparseMatrix* jacobian = 5379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez new DynamicCompressedRowSparseMatrix(num_residuals, 5479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez num_effective_parameters, 5579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 0); 5679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 5779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez CompressedRowJacobianWriter::PopulateJacobianRowAndColumnBlockVectors( 5879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez program_, jacobian); 5979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 6079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez return jacobian; 6179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez} 6279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 6379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandezvoid DynamicCompressedRowJacobianWriter::Write(int residual_id, 6479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez int residual_offset, 6579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez double **jacobians, 6679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez SparseMatrix* base_jacobian) { 6779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez DynamicCompressedRowSparseMatrix* jacobian = 6879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez down_cast<DynamicCompressedRowSparseMatrix*>(base_jacobian); 6979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 7079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez // Get the `residual_block` of interest. 7179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez const ResidualBlock* residual_block = 7279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez program_->residual_blocks()[residual_id]; 7379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez const int num_residuals = residual_block->NumResiduals(); 7479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 7579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez vector<pair<int, int> > evaluated_jacobian_blocks; 7679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez CompressedRowJacobianWriter::GetOrderedParameterBlocks( 7779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez program_, residual_id, &evaluated_jacobian_blocks); 7879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 7979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez // `residual_offset` is the residual row in the global jacobian. 8079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez // Empty the jacobian rows. 8179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez jacobian->ClearRows(residual_offset, num_residuals); 8279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 8379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez // Iterate over each parameter block. 8479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez for (int i = 0; i < evaluated_jacobian_blocks.size(); ++i) { 8579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez const ParameterBlock* parameter_block = 8679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez program_->parameter_blocks()[evaluated_jacobian_blocks[i].first]; 8779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez const int parameter_block_jacobian_index = 8879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez evaluated_jacobian_blocks[i].second; 8979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez const int parameter_block_size = parameter_block->LocalSize(); 9079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 9179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez // For each parameter block only insert its non-zero entries. 9279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez for (int r = 0; r < num_residuals; ++r) { 9379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez for (int c = 0; c < parameter_block_size; ++c) { 9479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez const double& v = jacobians[parameter_block_jacobian_index][ 9579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez r * parameter_block_size + c]; 9679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez // Only insert non-zero entries. 9779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez if (v != 0.0) { 9879397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez jacobian->InsertEntry( 9979397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez residual_offset + r, parameter_block->delta_offset() + c, v); 10079397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez } 10179397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez } 10279397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez } 10379397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez } 10479397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez} 10579397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez 10679397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez} // namespace internal 10779397c21138f54fcff6ec067b44b847f1f7e0e98Carlos Hernandez} // namespace ceres 108