OpenSubdiv/opensubdiv/vtr/maskInterfaces.h

//
//   Copyright 2014 DreamWorks Animation LLC.
//
//   Licensed under the Apache License, Version 2.0 (the "Apache License")
//   with the following modification; you may not use this file except in
//   compliance with the Apache License and the following modification to it:
//   Section 6. Trademarks. is deleted and replaced with:
//
//   6. Trademarks. This License does not grant permission to use the trade
//      names, trademarks, service marks, or product names of the Licensor
//      and its affiliates, except as required to comply with Section 4(c) of
//      the License and to reproduce the content of the NOTICE file.
//
//   You may obtain a copy of the Apache License at
//
//       http://www.apache.org/licenses/LICENSE-2.0
//
//   Unless required by applicable law or agreed to in writing, software
//   distributed under the Apache License with the above modification is
//   distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
//   KIND, either express or implied. See the Apache License for the specific
//   language governing permissions and limitations under the Apache License.
//
#ifndef VTR_INTERFACES_H
#define VTR_INTERFACES_H

#include "../version.h"

#include "../sdc/type.h"
#include "../sdc/crease.h"
#include "../vtr/types.h"

#include <vector>


namespace OpenSubdiv {
namespace OPENSUBDIV_VERSION {

namespace Vtr {

//
//  Simple classes supporting the interfaces required of generic types in the Scheme mask
//  queries, e.g. <typename FACE, MASK, etc.>
//
//  These were added solely to support the temporary Refinement::computeMasks(), which
//  is not expected to persist in its current form.  So these are for illustration purposes
//  now and may eventually be moved elsewhere (likely into Far).
//

//
//  For <typename MASK>, where the mask weights are stored:
//
class MaskInterface {
public:
    typedef float Weight;  //  Also part of the expected interface

public:
    MaskInterface(Weight* v, Weight* e, Weight* f) : _vertWeights(v), _edgeWeights(e), _faceWeights(f) { }
    ~MaskInterface() { }

public:  //  Generic interface expected of <typename MASK>:
    int GetNumVertexWeights() const { return _vertCount; }
    int GetNumEdgeWeights()   const { return _edgeCount; }
    int GetNumFaceWeights()   const { return _faceCount; }

    void SetNumVertexWeights(int count) { _vertCount = count; }
    void SetNumEdgeWeights(  int count) { _edgeCount = count; }
    void SetNumFaceWeights(  int count) { _faceCount = count; }

    Weight const& VertexWeight(int index) const { return _vertWeights[index]; }
    Weight const& EdgeWeight(  int index) const { return _edgeWeights[index]; }
    Weight const& FaceWeight(  int index) const { return _faceWeights[index]; }

    Weight& VertexWeight(int index) { return _vertWeights[index]; }
    Weight& EdgeWeight(  int index) { return _edgeWeights[index]; }
    Weight& FaceWeight(  int index) { return _faceWeights[index]; }

    bool AreFaceWeightsForFaceCenters() const  { return _faceWeightsForFaceCenters; }
    void SetFaceWeightsForFaceCenters(bool on) { _faceWeightsForFaceCenters = on; }

private:
    Weight* _vertWeights;
    Weight* _edgeWeights;
    Weight* _faceWeights;

    int _vertCount;
    int _edgeCount;
    int _faceCount;

    bool _faceWeightsForFaceCenters;
};


//
//  For <typename FACE>, which provides information int the neighborhood of a face:
//
class FaceInterface {
public:
    FaceInterface() { }
    FaceInterface(int vertCount) : _vertCount(vertCount) { }
    ~FaceInterface() { }

public:  //  Generic interface expected of <typename FACE>:
    int GetNumVertices() const { return _vertCount; }

private:
    int _vertCount;
};


//
//  For <typename EDGE>, which provides information in the neighborhood of an edge:
//
class EdgeInterface {
public:
    EdgeInterface() { }
    EdgeInterface(Level const& level) : _level(&level) { }
    ~EdgeInterface() { }

    void SetIndex(int edgeIndex) { _eIndex = edgeIndex; }

public:  //  Generic interface expected of <typename EDGE>:
    int   GetNumFaces() const { return _level->getEdgeFaces(_eIndex).size(); }
    float GetSharpness() const { return _level->getEdgeSharpness(_eIndex); }

    void GetChildSharpnesses(Sdc::Crease const&, float s[2]) const {
        //  Need to use the Refinement here to identify the two child edges:
        s[0] = s[1] = GetSharpness() - 1.0f;
    }

    void GetNumVerticesPerFace(int vertsPerFace[]) const {
        IndexArray const eFaces = _level->getEdgeFaces(_eIndex);
        for (int i = 0; i < eFaces.size(); ++i) {
            vertsPerFace[i] = _level->getFaceVertices(eFaces[i]).size();
        }
    }

private:
    const Level* _level;

    int _eIndex;
};


//
//  For <typename VERTEX>, which provides information in the neighborhood of a vertex:
//
class VertexInterface {
public:
    VertexInterface() { }
    VertexInterface(Level const& parent, Level const& child) : _parent(&parent), _child(&child) { }
    ~VertexInterface() { }

    void SetIndex(int parentIndex, int childIndex) {
        _pIndex = parentIndex;
        _cIndex = childIndex;
        _eCount = _parent->getVertexEdges(_pIndex).size();
        _fCount = _parent->getVertexFaces(_pIndex).size();
    }

public:  //  Generic interface expected of <typename VERT>:
    int GetNumEdges() const { return _eCount; }
    int GetNumFaces() const { return _fCount; }

    float  GetSharpness() const { return _parent->getVertexSharpness(_pIndex); }
    float* GetSharpnessPerEdge(float pSharpness[]) const {
        IndexArray const pEdges = _parent->getVertexEdges(_pIndex);
        for (int i = 0; i < _eCount; ++i) {
            pSharpness[i] = _parent->getEdgeSharpness(pEdges[i]);
        }
        return pSharpness;
    }

    float  GetChildSharpness(Sdc::Crease const&) const { return _child->getVertexSharpness(_cIndex); }
    float* GetChildSharpnessPerEdge(Sdc::Crease const& crease, float cSharpness[]) const {
        float * pSharpness = (float *)alloca(_eCount*sizeof(float));
        GetSharpnessPerEdge(pSharpness);
        crease.SubdivideEdgeSharpnessesAroundVertex(_eCount, pSharpness, cSharpness);
        return cSharpness;
    }

private:
    const Level* _parent;
    const Level* _child;

    int _pIndex;
    int _cIndex;
    int _eCount;
    int _fCount;
};

} // end namespace Vtr

} // end namespace OPENSUBDIV_VERSION
using namespace OPENSUBDIV_VERSION;
} // end namespace OpenSubdiv

#endif /* VTR_INTERFACES_H */