GEO/GEO_PrimConnector.h

Go to the documentation of this file.

/*
 * PROPRIETARY INFORMATION.  This software is proprietary to
 * Side Effects Software Inc., and is not to be reproduced,
 * transmitted, or disclosed in any way without written permission.
 *
 * Produced by:
 *      Ondrej Kos
 *      Side Effects
 *      477 Richmond Street West
 *      Toronto, Ontario
 *      Canada   M5V 3E7
 *      416-504-9876
 *
 * NAME:        GEO Library (C++)
 *
 * COMMENTS:  This class builds a useful structure for determining the 
 *            connectivity of primitives.
 *             
 */

#ifndef __GEO_PrimConnector_h__
#define __GEO_PrimConnector_h__

#include "GEO_API.h"
#include <UT/UT_BitArray.h>
#include <UT/UT_IntArray.h>
#include <UT/UT_PtrArray.h>
#include "GEO_Detail.h"


typedef UT_PtrArray<GEO_Primitive *>    geo_ConnectedPrimArray;
typedef UT_PtrArray<GEO_Point *>        geo_ConnectedPointArray;

class GEO_PrimClassifier;

class GEO_API GEO_PrimConnector
{
public:
    GEO_PrimConnector(GEO_Detail &gdp,
                      const GEO_PrimClassifier &classifier,
                      int build_connected_points = 0);
    virtual ~GEO_PrimConnector();


    int                          getTreeCount() const 
                                 { return (int)myForest.entries(); }
    int                          getMaxTreeCount() const;

    // Query either the prims or points list of tree i.  Note that the
    // points will only be filled in if this connector is flagged to
    // do so.
    const geo_ConnectedPointArray       &connectedPoints(int i) const 
                 { return myForest((unsigned)i)->myConnectedPoints; }

    const geo_ConnectedPrimArray        &connectedPrims(int i) const
                 { return myForest((unsigned)i)->myConnectedPrims; }

    // Clears the connections built
    virtual void                 clear();

    // This method returns the tree index for primitives connected to
    // the supplied primitive.  Returns -1 if we have not yet built the
    // tree for this connected group.
    int                          getPrimTree(const GEO_Primitive &prim);

    int                          buildTree(const GEO_Primitive &prim);

    // This method fills the trees array with all the indices of any
    // primitive trees which reference the given point.  The return
    // value is the number of trees found.  This method will construct
    // the trees if they do not already exist.
    virtual int                  getPrimTrees(const GEO_Point &point, 
                                              UT_IntArray &trees) = 0;

protected:
    int                          buildClassTree(int class_num);

    class GEO_API geo_ConnectedTree
    {
        public:
#ifdef INTEL_COMPILER
                geo_ConnectedTree() { }
#endif          
            geo_ConnectedPointArray     myConnectedPoints;
            geo_ConnectedPrimArray      myConnectedPrims;
        private:
#ifdef INTEL_COMPILER           
                geo_ConnectedTree(const geo_ConnectedTree& ) { }
#endif
                
    };

    // Members:
    GEO_Detail                          &myGdp;
    const GEO_PrimClassifier            *myClassifier;

    // The forest contains the lists of connected points (trees)
    UT_PtrArray<geo_ConnectedTree *>     myForest;
    int                                  myBuildConnectedPoints;
    UT_BitArray                          myVisitedPoints;
};

class GEO_PointClassifier;

class GEO_API GEO_PrimConnector3D : public GEO_PrimConnector
{
public:
    GEO_PrimConnector3D(GEO_Detail &gdp,
                        const GEO_PrimClassifier &classifier,
                        const GEO_PointClassifier &ptclassifier,
                        int build_connected_points = 0);
    virtual ~GEO_PrimConnector3D();

    // This method fills the trees array with all the indices of any
    // primitive trees which reference the given point.  The return
    // value is the number of trees found.  This method will construct
    // the trees if they do not already exist.
    virtual int                  getPrimTrees(const GEO_Point &point, 
                                              UT_IntArray &trees);

private:
    const GEO_PointClassifier           *myPointClassifier;
};

class GEO_PointRefArray;

class GEO_API GEO_PrimConnectorUV : public GEO_PrimConnector
{
public:
    GEO_PrimConnectorUV(GEO_Detail &gdp,
                        const GEO_PrimClassifier &classifier,
                        int build_connected_points = 0);
    virtual ~GEO_PrimConnectorUV();

    // This method fills the trees array with all the indices of any
    // primitive trees which reference the given point.  The return
    // value is the number of trees found.  This method will construct
    // the trees if they do not already exist.
    virtual int                  getPrimTrees(const GEO_Point &point, 
                                              UT_IntArray &trees);

    GEO_PointRefArray           &getPointRefArray();

private:
    GEO_PointRefArray           *myPointRefArray;
};

#endif

---