luaFeatureSet.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2012 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/
#pragma once

#include <vrfLua/vrfLuaDefines.h>
#include <vrfLua/luaFeatureGeometry.h>
#include <features/featureSet.h>
#include <features/query.h>

#include "vrfutil/asciiSerializable.h"
#include "vrfutil/scriptGeometry.h"
#include "vrfutil/objectCounter.h"

#include <boost/shared_ptr.hpp>
#include <boost/optional.hpp>

#include <luabind/luabind.hpp>

class DtSimManager;

class DT_DLL_vrfLua DtLuaFeatureSet
   : public DtAsciiSerializable
   , public DtObjectDebugger<DtLuaFeatureSet>
{
public:
   typedef std::vector<DtLocation3D> LocationVector;
   typedef std::map<std::string, std::string> AttributeMap;

   DtLuaFeatureSet(DtSimManager* = 0, const std::string& type="");

   DtLuaFeatureSet(const DtLuaFeatureSet&);

   DtLuaFeatureSet& operator=(const DtLuaFeatureSet&);
   
   DtLuaFeatureSet getFeaturesWithinArea(luabind::object const& points, luabind::object const& attributes);
   
   DtLuaFeatureSet getFeaturesWithinRange(const DtLocation3D&, double range, luabind::object const& attributes);

   DtLuaFeatureSet filter(const DtLuaFeatureGeometry& geom, luabind::object const& attributes);
   DtLuaFeatureSet filterGeometry(const DtLuaFeatureGeometry& geom);
   DtLuaFeatureSet filterAttributes(luabind::object const& attributes);

public:

   virtual void setSimManager(DtSimManager* simManager);

   virtual DtString objectType() const;

   DtString serialize();

   bool deserialize(const DtString& data);

   static DtAsciiSerializable* creator();

public:

   bool getIsLoaded() const;

   int getFeatureCount();

   bool getIsPointFeature(int);

   bool getIsLinearFeature(int);

   bool getIsArealFeature(int);

   std::string getFeatureType(int);

   std::vector<std::string> getFeatureAttributeNames(int);

   bool getHasAttributeNamed(int, const std::string);

   DtLocation3D getLocation3D(int);

   std::vector<DtLocation3D> getLocations3D(int);

   static luabind::object getFeatureAttributeValue(DtLuaFeatureSet&, int, const std::string, lua_State*);

   MAKVRinTerra::DtFeature::CPtr feature(int);

   DtLuaFeatureGeometry geometry(int);

   DtLuaFeatureSet load();

protected:
   virtual void gatherFeatures(const MAKVRinTerra::DtFeature&);

   virtual void testLoaded();

   void update(MAKVRinTerra::DtFeatureGeometry);
   void update(const LocationVector& locations);
   void update(const AttributeMap& attributes);
   void updateRange(const DtLocation3D& center, double range);
   void updatePoints(luabind::object const& points);
   void updateAttributes(luabind::object const& attributes);

   void filterFeatures();

protected:
   typedef std::vector<MAKVRinTerra::DtFeature::CPtr> Features;

   DtSimManager* mySimManager;

   std::vector<DtLocation3D> myLocations;
   std::map<std::string, std::string> myAttributes;

   boost::shared_ptr<MAKVRinTerra::DtFeatureSet> myFeatureSet;
   bool myProcessed;
   Features myFeatures;

   std::string myType;
   boost::optional<MAKVRinTerra::DtQuery> myQuery;
   boost::optional<MAKVRinTerra::DtFeatureGeometry> myGeomFilter;
};

DT_DEFINE_OBJECT_DEBUGGER_NAME(DtLuaFeatureSet);