/* *****************************************************************
    MESQUITE -- The Mesh Quality Improvement Toolkit

    Copyright 2004 Sandia Corporation and Argonne National
    Laboratory.  Under the terms of Contract DE-AC04-94AL85000
    with Sandia Corporation, the U.S. Government retains certain
    rights in this software.

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2.1 of the License, or (at your option) any later version.

    This library 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
    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    (lgpl.txt) along with this library; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    diachin2@llnl.gov, djmelan@sandia.gov, mbrewer@sandia.gov,
    pknupp@sandia.gov, tleurent@mcs.anl.gov, tmunson@mcs.anl.gov

  ***************************************************************** */
// -*- Mode : c++; tab-width: 3; c-tab-always-indent: t; indent-tabs-mode: nil; c-basic-offset: 3
// -*-

/*! \file CompositeOFAdd.hpp

Header file for the MBMesquite:: CompositeOFAdd class

  \author Michael Brewer
  \date   2002-06-24
 */

#ifndef CompositeOFAdd_hpp
#define CompositeOFAdd_hpp

#include "Mesquite.hpp"
#include "ObjectiveFunction.hpp"
#include "MsqHessian.hpp"

namespace MBMesquite
{
/*!\class CompositeOFAdd
  \brief Adds two ObjectiveFunction values together.
*/
class MsqMeshEntity;
class PatchData;
class MsqError;
class Vector3D;

class CompositeOFAdd : public ObjectiveFunction
{
  public:
    MESQUITE_EXPORT
    CompositeOFAdd( ObjectiveFunction*, ObjectiveFunction*, bool delete_OFs = false );

    MESQUITE_EXPORT
    virtual ~CompositeOFAdd();

    //!\brief Called at start of instruction queue processing
    //!
    //! Do any preliminary global initialization, consistency checking,
    //! etc.
    MESQUITE_EXPORT
    virtual void initialize_queue( MeshDomainAssoc* mesh_and_domain, const Settings* settings, MsqError& err );

    MESQUITE_EXPORT
    virtual bool initialize_block_coordinate_descent( MeshDomainAssoc* mesh_and_domain, const Settings* settings,
                                                      PatchSet* user_set, MsqError& err );

    MESQUITE_EXPORT
    virtual bool evaluate( EvalType type, PatchData& pd, double& value_out, bool free, MsqError& err );

    MESQUITE_EXPORT
    virtual bool evaluate_with_gradient( EvalType type, PatchData& pd, double& value_out,
                                         std::vector< Vector3D >& grad_out, MsqError& err );

    MESQUITE_EXPORT
    virtual bool evaluate_with_Hessian_diagonal( EvalType type, PatchData& pd, double& value_out,
                                                 std::vector< Vector3D >& grad_out,
                                                 std::vector< SymMatrix3D >& hess_diag_out, MsqError& err );

    MESQUITE_EXPORT
    virtual bool evaluate_with_Hessian( EvalType type, PatchData& pd, double& value_out,
                                        std::vector< Vector3D >& grad_out, MsqHessian& Hessian_out, MsqError& err );

    MESQUITE_EXPORT
    virtual ObjectiveFunction* clone() const;

    MESQUITE_EXPORT
    virtual void clear();

    MESQUITE_EXPORT
    virtual int min_patch_layers() const;

  private:
    /** Temporary storage for gradient */
    mutable std::vector< Vector3D > mGradient;
    /** Temporary storage for hessian diagonal */
    mutable std::vector< SymMatrix3D > mDiagonal;
    /** Temporary storage for Hessian */
    mutable MsqHessian mHessian;

    ObjectiveFunction* objFunc1;
    ObjectiveFunction* objFunc2;
    bool deleteObjFuncs;
};
}  // namespace MBMesquite
#endif  //  CompositeOFAdd_hpp
