#pragma once

#include "GeneratorTemplate.h"

class TreeTemplate : public GeneratorTemplate
{
private:
    int woodBlockTypeId;
    int leaveBlockType;
    int minHeight;
    int maxHeight;

public:
    TreeTemplate(float propability,
        int woodBlockTypeId,
        int leaveBlockType,
        int minHeight,
        int maxHeight);
    virtual GeneratedStructure* generateAt(
        Framework::Vec3<int> location, Noise* zNoise, int dimensionId) override;
    const BlockType* getWoodType() const;
    const BlockType* getLeavesType() const;
};

class TreeTemplateFactory : public GeneratorTemplateFactory
{
public:
	TreeTemplateFactory();
    GeneratorTemplate* createTemplate(
		Framework::JSON::JSONValue* zConfig) override;
    Framework::JSON::Validator::JSONValidator* getValidator() override;
};