Trees.h

//
// Trees.h
//
// Copyright (c) 2001-2009 Virtual Terrain Project
// Free for all uses, see license.txt for details.
//

#ifndef TERRAINTREESH
#define TERRAINTREESH


#include "vtdata/Plants.h"

class vtHeightField3d;

#if SUPPORT_XFROG
// Greenworks Xfrog:  See http://www.xfrog.com for details.  Not currently
//  supported as we don't currently have a way to interface to their engine.
#include "\dism\xfrog2dism\xfrog2dism.h"
#endif

class vtPlantAppearance3d : public vtPlantAppearance
{
public:
    vtPlantAppearance3d(AppearType type, const char *filename,
        float width, float height,
        float shadow_radius, float shadow_darkness);
    ~vtPlantAppearance3d();
    vtPlantAppearance3d(const vtPlantAppearance &v);

    void CheckAvailability();
    void LoadAndCreate();
    bool GenerateGeom(vtTransform *container);
    bool IsAvailable() { return m_bAvailable; }

    static float s_fPlantScale;
    static bool  s_bPlantShadows;

protected:
    vtMesh *CreateTreeMesh(float fTreeScale, bool bShadows);
    void _Defaults();

    osg::ref_ptr<vtMaterialArray> m_pMats;
    vtGeodePtr      m_pGeode;
    osg::ref_ptr<vtMesh> m_pMesh;
    int             m_iMatIdx;
    osg::ref_ptr<osg::Node> m_pExternal;
#if SUPPORT_XFROG
    CFrogModel  *m_pFrogModel;
#endif
    bool m_bAvailable;
    bool m_bCreated;
};

class vtPlantSpecies3d : public vtPlantSpecies
{
public:
    vtPlantSpecies3d();

    // copy
    vtPlantSpecies3d &operator=(const vtPlantSpecies &v);

    vtPlantAppearance3d *GetAppearanceByHeight(float fHeight);
    vtPlantAppearance3d *GetRandomAppearance();
    vtPlantAppearance3d *GetAppearance(int i) const { return (vtPlantAppearance3d *)m_Apps[i]; };

    void CheckAvailability();
    int NumAvailableInstances();

    virtual void AddAppearance(AppearType type, const char *filename, float width, float height,
        float shadow_radius, float shadow_darkness);
};


class vtSpeciesList3d : public vtSpeciesList
{
public:
    vtSpeciesList3d();

    // copy operator
    vtSpeciesList3d &operator=(const vtSpeciesList &v);

    int CheckAvailability();
    void CreatePlantSurfaces();

    // override / replace a few methods of vtSpeciesList
    vtPlantSpecies3d *GetSpecies(uint i) const;

//  vtGeode *plant_nursery(vtHeightField *pHeightField, float lat, float lon);
};

class vtPlantInstance3d
{
public:
    vtPlantInstance3d();
    ~vtPlantInstance3d();

    void ShowBounds(bool bShow);
    void ReleaseContents();

    vtTransform *m_pContainer;
    vtGeode     *m_pHighlight;  // The wireframe highlight
};

class vtPlantInstanceArray3d : public vtPlantInstanceArray
{
public:
    vtPlantInstanceArray3d();
    ~vtPlantInstanceArray3d();

    int CreatePlantNodes(bool progress_dialog(int) = NULL);
    bool CreatePlantNode(uint i);
    int NumOffTerrain() { return m_iOffTerrain; }

    vtTransform *GetPlantNode(uint i) const;
    vtPlantInstance3d *GetInstance3d(uint i) const;

    void SetHeightField(vtHeightField3d *hf) { m_pHeightField = hf; }

    vtSpeciesList3d *GetPlantList() { return (vtSpeciesList3d *) m_pPlantList; }

    void VisualDeselectAll();

    void VisualSelect(uint i);

    void OffsetSelectedPlants(const DPoint2 &offset);

    void ReleasePlantGeometry(uint i);
    void DeletePlant(uint i);

    void UpdateTransform(uint i);

    bool FindPlantFromNode(osg::Node *pNode, int &iOffset);

protected:
    vtArray<vtPlantInstance3d*> m_Instances3d;
    vtHeightField3d     *m_pHeightField;
    int                 m_iOffTerrain;
};
    // Group veg

#endif  // TERRAINTREESH