#pragma once

#include "definition_global.h"

#include <vector>
#include "DefinitionNode.h"
#include "Sphere.h"
#include "CappedCylinder.h"
#include "Disk.h"

namespace paysages {
namespace definition {

/**
 * Model of vegetation.
 */
class DEFINITIONSHARED_EXPORT VegetationModelDefinition : public DefinitionNode {
  public:
    VegetationModelDefinition(DefinitionNode *parent);
    virtual ~VegetationModelDefinition();

    inline const SurfaceMaterial &getSolidMaterial() const {
        return *solid_material;
    }
    inline const SurfaceMaterial &getFoliageMaterial() const {
        return *foliage_material;
    }

    inline const vector<CappedCylinder> &getSolidVolumes() const {
        return solid_volumes;
    }
    inline const vector<Sphere> &getFoliageGroups() const {
        return foliage_groups;
    }
    inline const vector<Disk> &getFoliageItems() const {
        return foliage_items;
    }

    virtual void save(PackStream *stream) const override;
    virtual void load(PackStream *stream) override;
    virtual void copy(DefinitionNode *destination) const override;
    virtual void validate() override;

    /**
     * Randomize the model geometry.
     */
    void randomize(RandomGenerator &random = RandomGeneratorDefault);

  private:
    SurfaceMaterial *solid_material;
    SurfaceMaterial *foliage_material;
    vector<CappedCylinder> solid_volumes;
    vector<Sphere> foliage_groups;
    vector<Disk> foliage_items;
};
}
}