source: GTP/trunk/Lib/Vis/Preprocessing/src/Preprocessor.h @ 2743

#ifndef _Preprocessor_H__
#define _Preprocessor_H__

#include <string>
//
#include "Containers.h"
#include "Mesh.h"
#include "KdTree.h"
#include "SimpleRay.h"

#define USE_RAY_POOL

namespace GtpVisibilityPreprocessor {

class PreprocessorThread;  
class RenderSimulator;
class SceneGraph;
class Exporter;
class ViewCellsManager;
class BspTree;
class VspOspTree;
class VspBspTree;
class RenderSimulator;
struct VssRayContainer;
class SamplingStrategy;
class GlRendererBuffer;
class VspTree;
class HierarchyManager;
class BvHierarchy;
class Intersectable;
class VssRay;
class RayCaster;
class GlobalLinesRenderer;
class SceneGraphLeaf;
class GlRendererWidget;
class IntersectableGroup;
class TriangleIntersectable;


/** Hack: Visualization structure for the GVS algorithm.
*/
struct VizStruct
{
        VertexContainer enlargedTriangle;
        TriangleIntersectable *originalTriangle;
        VssRay *ray;
};


/** Namespace for the external visibility preprocessor

    This namespace includes all classes which are created by the VUT (Vienna University
    of Technology) for the External Visibility Preprocessor of the GTP (GameTools Project)
    (www.gametools.org).
*/
  
/** Main class of the visibility preprocessor. Responsible for loading and
    saving of the input and output files. Initiates construction of the kD-tree,
    viewcell loading/generation and the visibility computation itself.
*/
class Preprocessor
{
        friend class RayCaster;
        friend class IntelRayCaster;
        friend class InternalRayCaster;
        friend class HavranRayCaster;

public:
        /** Default constructor initialising e.g., KD tree and BSP tree.
        */
        Preprocessor();

        virtual ~Preprocessor();
        /** Load the input scene. 
                @param filename file to load
                @return true on success
        */
        virtual bool LoadScene(const string &filename);
        /** Export all preprocessed data in a XML format understandable by the
                PreprocessingInterface of the GtpVisibilityPreprocessor Module. 
                The file can be compressed depending on the environement settings.
                @return true on successful export
        */
        virtual bool ExportPreprocessedData(const string &filename);
        /** Build the KdTree of currently loaded occluders/occludees/viewcells. The construction
                is driven by the environment settings, which also sais which of the three types of
                entities should be used to drive the heuristical construction (only occluders by default)
        */
        virtual bool BuildKdTree();
        /** Compute visibility method. This method has to be reimplemented by the actual
                Preprocessor implementation (e.g. SamplingPreprocessor, ExactPreprocessor,
                GlobalSamplingpreprocessor)
        */
        virtual bool ComputeVisibility() = 0;
        /** Post Process the computed visibility. By default applys the visibility filter
                (if specified in the environment and export the preprocessed data 
        */
        virtual bool PostProcessVisibility();
        /** View cells are either loaded or prepared for generation, according to the chosen environment
                object. Important evironment options are, e.g, the view cell type.
                Should be done after scene loading (i.e., some options are based on scene type).
        */
        bool PrepareViewCells();
        /** Construct viewcells from the scratch 
        */
        bool ConstructViewCells();
        /** Returns the specified sample strategy, NULL if no valid strategy.
        */
        SamplingStrategy *GenerateSamplingStrategy(const int strategyId);
        /** Export preprocessor data.
        */
        bool Export(const string &filename, const bool scene, const bool kdtree);  

        virtual void KdTreeStatistics(ostream &s);
        virtual void BspTreeStatistics(ostream &s);

        /** Loads samples from file.
        @param samples returns the stored sample rays
        @param objects needed to associate the objects ids
        @returns true if samples were loaded successfully
        */
        bool LoadSamples(VssRayContainer &samples, ObjectContainer &objects) const;

        /** Exports samples to file.
                @returns true if samples were written successfully
        */
        bool ExportSamples(const VssRayContainer &samples) const;

        bool LoadKdTree(const string &filename);
        bool ExportKdTree(const string &filename);

        virtual bool
        ExportRays(const char *filename,
                           const VssRayContainer &vssRays,
                           const int number,
                           const bool exportScene = false
                           );

        bool ExportRayAnimation(const char *filename,
                                                        const vector<VssRayContainer> &vssRays);

        virtual int     GenerateRays(const int number,
                                                         SamplingStrategy &strategy,
                                                         SimpleRayContainer &rays);

        virtual int     GenerateRays(const int number,
                                                         SamplingStrategy &strategy,
                                                         SimpleRayContainer &rays, 
                                                         int &invalidSamples);

        virtual int GenerateRays(const int number,
                                                         const int raysType,
                                                         SimpleRayContainer &rays);

        bool GenerateJitteredRays(SimpleRayContainer &rayBundle,
                                                         const SimpleRay &mainRay,
                                                         int number,
                                                         int shuffleType,
                                                         float scale) const;

        virtual void CastRays(SimpleRayContainer &rays,
                                                  VssRayContainer &vssRays,
                                                  const bool castDoubleRays,
                                                  const bool pruneInvalidRays = true);

        virtual void
                CastRaysWithHwGlobalLines(
                SimpleRayContainer &rays,
                VssRayContainer &vssRays,
                const bool castDoubleRays,
                const bool pruneInvalidRays
                );

        /** Compute pixel error of the current PVS solution by sampling given number of viewpoints.
        */
        virtual void ComputeRenderError();

        /** Returns a view cells manager of the given name.
        */
        ViewCellsManager *CreateViewCellsManager(const char *name);

        GlRendererBuffer *GetRenderer();

        bool InitRayCast(const string &externKdTree, const string &internKdTree);

        bool LoadInternKdTree(const string &internKdTree);

        bool ExportObj(const string &filename, const ObjectContainer &objects);

        Intersectable *GetParentObject(const int index) const;
        Vector3 GetParentNormal(const int index) const;

        /** Sets a preprocessor thread.
        */
        void SetThread(PreprocessorThread *t);

        /** Returns a preprocessor thread.
        */
        PreprocessorThread *GetThread() const;

        Intersectable *GetObjectById(const int id);

        void PrepareHwGlobalLines();

        virtual void DeterminePvsObjects(VssRayContainer &rays);

        static bool LoadObjects(const string &filename,
                                                        ObjectContainer &pvsObjects, 
                                                        const ObjectContainer &preprocessorObject);

        /** Adds dynamic geometry
        */
        SceneGraphLeaf *LoadDynamicGeometry(const string &filename);

        /** Register the dynamic object to be handled by the ray caster for visibility
                computation.
        */
        void RegisterDynamicObject(SceneGraphLeaf *leaf);
        
        void PrepareObjectsForRayCaster(SceneGraphLeaf *l);
        
        void ScheduleUpdateDynamicObjects();
        void UpdateDynamicObjects();
        /** Notify the preprocessor that an object has been deleted
        */
        virtual void ObjectRemoved(SceneGraphLeaf *object);

        SceneGraphLeaf *GenerateBoxGeometry(const AxisAlignedBox3 &box);

        float _HackComputeRenderCost(ViewCell *vc);

        /////////////////////////

        bool mSynchronize;

        /// scene graph loaded from file
        SceneGraph *mSceneGraph;

        /// raw array of objects
        ObjectContainer mObjects;

        /// kD-tree organizing the scene graph (occluders + occludees) + viewcells
        KdTree *mKdTree;
        AxisAlignedBox3 sceneBox;

        /// View space partition bsp tree
        VspBspTree *mVspBspTree;
        /// BSP tree representing the viewcells
        BspTree *mBspTree;

        /// list of all loaded occluders
        ObjectContainer mOccluders;
        /// list of all loaded occludees
        ObjectContainer mOccludees;
        /// the view cells manager
        ViewCellsManager *mViewCellsManager;

        DynamicObjectsContainer mDynamicObjects;

        /// greedy optimized hierarchy for both objects and view cells
        VspOspTree *mVspOspTree;

        bool mUseGlRenderer;
        bool mUseGlDebugger;

        bool mUseHwGlobalLines;
        bool mLoadViewCells;

        bool mDetectEmptyViewSpace;

        bool mQuitOnFinish;
        bool mLoadMeshes;
        bool mComputeVisibility;

        bool mExportVisibility;
        string mVisibilityFileName;

        bool mApplyVisibilityFilter;
        bool mApplyVisibilitySpatialFilter;

        float mVisibilityFilterWidth;

        int mPass;

        bool mStopComputation;

        bool mExportObj;

        bool mExportRays;
        bool mExportAnimation;
        int mExportNumRays;

        ofstream mStats;

        GlRendererBuffer *renderer;
        GlRendererWidget *mRendererWidget;

        int mTotalSamples;
        int mCurrentSamples;

        int mTotalTime;
        int mSamplesPerPass;
        int mSamplesPerEvaluation;
 
        int mTotalRaysCast;

        RayCaster *mRayCaster;

        // triangle pvs for gvs 
        // hack: should be in gvspreprocessor but pulled 
        // out here for visualization purpose
        ObjectContainer mTrianglePvs;
        vector<VizStruct> visTriangles;

        // generic stats vector that can be used for anything
        int mGenericStats[2];

        //std::vector<int> mDummyBuffer;


protected:

        bool LoadBinaryObj(const std::string &filename,
                       SceneGraphLeaf *root,
                                           std::vector<FaceParentInfo> *parents,
                                           float scale = -1.0f);

        bool ExportBinaryObj(const std::string &filename, SceneGraphLeaf *root);

        void SetupRay(Ray &ray, const Vector3 &point,
                      const Vector3 &direction) const;

        void EvalPvsStat();

        virtual void EvalViewCellHistogram();

        /** Notify the preprocessor that an object has moved and must be reevaluated.
        */
        virtual void ObjectMoved(SceneGraphLeaf *object);
        


        /////////////////////////

        GlobalLinesRenderer *mGlobalLinesRenderer;

        /// samples used for construction of the BSP view cells tree.
        int mBspConstructionSamples;
        /// samples used for construction of the VSP OSP tree.
        int mVspOspConstructionSamples;
        /// Simulates rendering of the scene.
        RenderSimulator *mRenderSimulator;

        vector<FaceParentInfo> mFaceParents;

        /// if box around view space should be used
        bool mUseViewSpaceBox;

        PreprocessorThread *mThread;

        bool mUpdateDynamicObjects;

        
};

extern Preprocessor *preprocessor;

}

#endif