src/Mesh/PatchDataUser.hpp

Go to the documentation of this file.

/* ***************************************************************** 
    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      
   
  ***************************************************************** */
#ifndef PatchDataUser_hpp
#define PatchDataUser_hpp

#include "Mesquite.hpp"
#include "MsqError.hpp"
#include "PatchData.hpp"

#ifndef MSQ_USE_OLD_C_HEADERS
#include <cstddef>
#include <cstdlib>
#include <cstdio>
#else
#include <stddef.h>
#include <stdlib.h>
#include <stdio.h>
#endif


namespace Mesquite
{

  class PatchDataParameters
  {
  public:
    PatchDataParameters() :
      mType(PatchData::UNDEFINED_PATCH_TYPE),
      mParam1(0),
      mParam2(0),
      cullingMethodBits(0),
      globalPatch(0)
    {}

    PatchDataParameters(const PatchDataParameters &A) {
      mType = A.mType;
      mParam1 = A.mParam1;
      mParam2 = A.mParam2;
      cullingMethodBits = A.cullingMethodBits;
      globalPatch = A.globalPatch;
    }

    friend class PatchDataUser;
    

    inline void set_patch_type(PatchData::PatchType patch_type, MsqError &err,
                        int patch_param1=0, int patch_param2=0);

    PatchData::PatchType get_patch_type()
    {return mType;}
    inline int get_nb_layers(MsqError &err);
    
    inline void set_global_patch_type();
    inline void set_element_on_vertex_patch_type( unsigned num_layers );

    inline void add_culling_method(enum PatchData::culling_method cm);
    inline void no_culling_method();
    inline void remove_culling_method(enum PatchData::culling_method cm);
    long unsigned int get_culling_method_bits() { return cullingMethodBits; }

    inline void set_global_patch(PatchData* pd, MsqError &err);
    PatchData* get_global_patch()
    { return globalPatch; }
    void no_global_patch()
    { globalPatch=0; }



  private:
    PatchData::PatchType mType; 
    int mParam1, mParam2; 
    long unsigned int cullingMethodBits; 
    PatchData* globalPatch; 

  };


  class PatchDataUser
  {
  protected:
    PatchDataUser() : mParams()
      {}
  public:
    virtual ~PatchDataUser()
      {}

    virtual void set_patch_type(PatchData::PatchType patch_type, MsqError &err,
                                int param1=0, int param2=0) {
      mParams.set_patch_type(patch_type, err, param1, param2); }
    PatchData::PatchType get_patch_type() { 
      return mParams.get_patch_type(); }
    int get_nb_layers(MsqError &err) { 
      return mParams.get_nb_layers(err); }
    
    void add_culling_method(enum PatchData::culling_method cm) {
      mParams.add_culling_method(cm); }
    void no_culling_method() {
      mParams.no_culling_method(); }
    void remove_culling_method(enum PatchData::culling_method cm) {
      mParams.remove_culling_method(cm); }
    long unsigned int get_culling_method_bits() { 
      return mParams.get_culling_method_bits(); }

    void set_all_parameters(PatchDataParameters &params)
    { mParams = params; }
    PatchDataParameters& get_all_parameters()
    { return mParams; }

    void set_global_patch(PatchData* pd, MsqError &err)
    { mParams.set_global_patch(pd, err); }
    PatchData* get_global_patch()
    { return mParams.get_global_patch(); }
    void no_global_patch()
    { mParams.no_global_patch(); }


    // *** functions related to the algorithms, no to the patch parameters *** 
    virtual double loop_over_mesh(MeshSet &ms, MsqError &err) = 0;
    virtual msq_std::string get_name() = 0;
    enum AlgorithmType { QUALITY_IMPROVER, QUALITY_ASSESSOR, MESH_TRANSFORM, TARGET_CALCULATOR };
    virtual enum AlgorithmType get_algorithm_type() = 0;

  private:
    PatchDataParameters mParams; 
  };
  

  inline void PatchDataParameters::set_patch_type(PatchData::PatchType patch_type,
                                                  MsqError &err,
                                                  int patch_param1,
                                                  int patch_param2)
  {
    // For now, no support for VERTICES_ON_ELEMENT_PATCH
    if ( patch_type != PatchData::ELEMENTS_ON_VERTEX_PATCH
         && patch_type != PatchData::GLOBAL_PATCH )
      {
        MSQ_SETERR(err)("VERTICES_ON_ELEMENT_PATCH not supported yet.",
                        MsqError::NOT_IMPLEMENTED);
        return;
      }
    
    if (patch_type == PatchData::ELEMENTS_ON_VERTEX_PATCH)
      {
        if (patch_param1 < 0)
        {
          MSQ_SETERR(err)("ELEMENTS_ON_VERTEX_PATCH not supported yet.",
                          MsqError::NOT_IMPLEMENTED);
          return;
        }
      }
    
    mType = patch_type;
    mParam1 = patch_param1;
    mParam2 = patch_param2;
    
    return;
  }
  
  inline void PatchDataParameters::set_global_patch_type()
  {
    mType = PatchData::GLOBAL_PATCH;
  }
  
  inline void PatchDataParameters::set_element_on_vertex_patch_type( unsigned num_layers )
  {
    mType = PatchData::ELEMENTS_ON_VERTEX_PATCH;
    mParam1 = num_layers;
  }
  
  
  inline int PatchDataParameters::get_nb_layers(MsqError &err)
  {
    if (mType == PatchData::GLOBAL_PATCH) {
      MSQ_SETERR(err)("Patch Type is GLOBAL_PATCH.", MsqError::INVALID_STATE);
      return 0;
    }

    return mParam1;
  }
  
  inline void PatchDataParameters::add_culling_method(enum PatchData::culling_method cm)
  {
    cullingMethodBits |= cm;
  }
  
  inline void PatchDataParameters::no_culling_method()
  {
    cullingMethodBits = 0;
  }
  
  inline void PatchDataParameters::remove_culling_method(enum PatchData::culling_method cm)
  {
    cullingMethodBits &= ~cm;
  }
  

  inline void PatchDataParameters::set_global_patch(PatchData* pd, MsqError &err)
  {
    if (get_patch_type() != PatchData::GLOBAL_PATCH)
      MSQ_SETERR(err)("Trying to set a global patch whereas the "
                  "PatchType is set to something else\n."
                  "Consider using the function no_global_patch().",
                  MsqError::INVALID_STATE);
    else { 
      globalPatch = pd;
    }
  }


} // namespace

#endif //  PatchDataUser_hpp