From 710b50eb1f7f92c11c89f085e100c6a03e206083 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Petr=20Mr=C3=A1zek?= <peterix@gmail.com>
Date: Mon, 21 Feb 2011 03:39:29 +0100
Subject: [PATCH] Vegetation vector for 31.19

---
 data/Memory-ng.xml                    |   7 +-
 tools/supported/SegmentedFinder.h     |  11 +-
 tools/supported/incrementalsearch.cpp | 207 +++++++++++++++++---------
 tools/supported/vdig.cpp              |   4 +
 4 files changed, 152 insertions(+), 77 deletions(-)

diff --git a/data/Memory-ng.xml b/data/Memory-ng.xml
index 285e57350..0b9148a13 100644
--- a/data/Memory-ng.xml
+++ b/data/Memory-ng.xml
@@ -1944,6 +1944,9 @@
             <Group name="GUI" valid="false">
                 <Address name="pause_state" value="0x14c9be1" valid="true" />
             </Group>
+            <Group name="Vegetation" valid="true">
+                <Address name="vector" value="0x16db478" />
+            </Group>
             <!--
             <Address name="WORLD" valid="false" />
             <Group name="Buildings">
@@ -1977,10 +1980,6 @@
                 <Address name="language_vector" value="0x016e553c"/>
                 <Address name="translation_vector" value="0x016e551c"/>
             </Group>
-            <Group name="Vegetation" valid="false">
-                <Address name="vector"/>
-                <Offset name="tree_desc_offset" valid="true"/>
-            </Group>
             <Group name="Items" valid="false">
                 <Address name="items_vector" />
                 List of offsets in the VTable :
diff --git a/tools/supported/SegmentedFinder.h b/tools/supported/SegmentedFinder.h
index f8911e96f..b4ec7c804 100644
--- a/tools/supported/SegmentedFinder.h
+++ b/tools/supported/SegmentedFinder.h
@@ -386,14 +386,15 @@ public:
             return false;
         }
     }
-    bool insert( char what )
+    template < class T >
+    bool insert( T what )
     {
         if(constant)
             return false;
-        if(d->length+1 == d->allocated)
-            Allocate(d->allocated * 2);
-        ((char *) d->object)[d->length] = what;
-        d->length ++;
+        if(d->length+sizeof(T) >= d->allocated)
+            Allocate((d->length+sizeof(T)) * 2);
+        (*(T *)(d->object + d->length)) = what;
+        d->length += sizeof(T);
     }
     Bytestreamdata * d;
     bool constant;
diff --git a/tools/supported/incrementalsearch.cpp b/tools/supported/incrementalsearch.cpp
index 4c85b181e..d2f89ec1c 100644
--- a/tools/supported/incrementalsearch.cpp
+++ b/tools/supported/incrementalsearch.cpp
@@ -179,10 +179,14 @@ bool Incremental ( vector <uint64_t> &found, const char * what, T& output,
         }
         goto incremental_more;
     }
-    else if(select.empty())
+    else if(select == "q")
     {
         return false;
     }
+    else if(select.empty())
+    {
+        goto incremental_more;
+    }
     else
     {
         if(numberz)
@@ -399,29 +403,91 @@ void FindData(DFHack::ContextManager & DFMgr, vector <DFHack::t_memrange>& range
         DF->Detach();
     }
 }
-/*
-    while(Incremental(found, "integer",test1))
+
+bool TriggerIncremental ( vector <uint64_t> &found )
+{
+    string select;
+    if(found.empty())
+    {
+        cout << "search ready - position the DF cursor and hit enter when ready" << endl;
+    }
+    else if( found.size() == 1 )
+    {
+        cout << "Found single coord!" << endl;
+        cout << hex << "0x" << found[0] << endl;
+    }
+    else
+    {
+        cout << "Found " << dec << found.size() << " coords." << endl;
+    }
+    incremental_more:
+    cout << ">>";
+    std::getline(cin, select);
+    size_t num = 0;
+    if( sscanf(select.c_str(),"p %d", &num) && num > 0)
+    {
+        cout << "Found coords:" << endl;
+        for(int i = 0; i < min(found.size(), num);i++)
+        {
+            cout << hex << "0x" << found[i] << endl;
+        }
+        goto incremental_more;
+    }
+    else if(select == "p")
+    {
+        cout << "Found coords:" << endl;
+        for(int i = 0; i < found.size();i++)
+        {
+            cout << hex << "0x" << found[i] << endl;
+        }
+        goto incremental_more;
+    }
+    else if(select == "q")
+    {
+        return false;
+    }
+    else return true;
+}
+
+
+void FindCoords(DFHack::ContextManager & DFMgr, vector <DFHack::t_memrange>& ranges)
+{
+    vector <uint64_t> found;
+    int size = 4;
+    do
+    {
+        getNumber("Select coord size (2,4 bytes)",size, 4);
+    } while (size != 2 && size != 4);
+    while (TriggerIncremental(found))
     {
         DFMgr.Refresh();
         DFHack::Context * DF = DFMgr.getSingleContext();
         DF->Attach();
-        SegmentedFinder sf(ranges,DF);
-        switch(size)
+        DFHack::Position * pos = DF->getPosition();
+        pos->Start();
+        int32_t x, y, z;
+        pos->getCursorCoords(x,y,z);
+        cout << "Searching for: " << dec << x << ":" << y << ":" << z << endl;
+        Bytestream select;
+        if(size == 2)
         {
-            case 1:
-                sf.Incremental<uint8_t,uint8_t>(test1,alignment,found, equalityP<uint8_t>);
-                break;
-            case 2:
-                sf.Incremental<uint16_t,uint16_t>(test1,alignment,found, equalityP<uint16_t>);
-                break;
-            case 4:
-                sf.Incremental<uint32_t,uint32_t>(test1,alignment,found, equalityP<uint32_t>);
-                break;
+            select.insert<uint16_t>(x);
+            select.insert<uint16_t>(y);
+            select.insert<uint16_t>(z);
+        }
+        else
+        {
+            select.insert<uint32_t>(x);
+            select.insert<uint32_t>(y);
+            select.insert<uint32_t>(z);
         }
+        cout << select << endl;
+        SegmentedFinder sf(ranges,DF);
+        sf.Incremental< Bytestream ,uint32_t>(select,1,found, findBytestream);
         DF->Detach();
     }
 }
-*/
+
 void PtrTrace(DFHack::ContextManager & DFMgr, vector <DFHack::t_memrange>& ranges)
 {
     int element_size;
@@ -841,64 +907,69 @@ int main (void)
     "Select search type: 1=number(default), 2=vector by length, 3=vector>object>string,\n"
     "                    4=string, 5=automated offset search, 6=vector by address in its array,\n"
     "                    7=pointer vector by address of an object, 8=vector>first object>string\n"
-    "                    9=string buffers, 10=known data, 11=backpointers, 12=data+backpointers\n";
+    "                    9=string buffers, 10=known data, 11=backpointers, 12=data+backpointers\n"
+    "                    13=coord lookup\n";
     int mode;
     do
     {
         getNumber(prompt,mode, 1, false);
+        switch (mode)
+        {
+            case 1:
+                DF->Detach();
+                FindIntegers(DFMgr, selected_ranges);
+                break;
+            case 2:
+                DF->Detach();
+                FindVectorByLength(DFMgr, selected_ranges);
+                break;
+            case 3:
+                DF->Detach();
+                FindVectorByObjectRawname(DFMgr, selected_ranges);
+                break;
+            case 4:
+                DF->Detach();
+                FindStrings(DFMgr, selected_ranges);
+                break;
+            case 5:
+                autoSearch(DF,selected_ranges);
+                break;
+            case 6:
+                DF->Detach();
+                FindVectorByBounds(DFMgr,selected_ranges);
+                break;
+            case 7:
+                DF->Detach();
+                FindPtrVectorsByObjectAddress(DFMgr,selected_ranges);
+                break;
+            case 8:
+                DF->Detach();
+                FindVectorByFirstObjectRawname(DFMgr, selected_ranges);
+                break;
+            case 9:
+                DF->Detach();
+                FindStrBufs(DFMgr, selected_ranges);
+                break;
+            case 10:
+                DF->Detach();
+                FindData(DFMgr, selected_ranges);
+                break;
+            case 11:
+                DF->Detach();
+                PtrTrace(DFMgr, selected_ranges);
+                break;
+            case 12:
+                DF->Detach();
+                DataPtrTrace(DFMgr, selected_ranges);
+                break;
+            case 13:
+                DF->Detach();
+                FindCoords(DFMgr, selected_ranges);
+                break;
+            default:
+                cout << "not implemented :(" << endl;
+        }
     } while (mode < 1 || mode > 12 );
-    switch (mode)
-    {
-        case 1:
-            DF->Detach();
-            FindIntegers(DFMgr, selected_ranges);
-            break;
-        case 2:
-            DF->Detach();
-            FindVectorByLength(DFMgr, selected_ranges);
-            break;
-        case 3:
-            DF->Detach();
-            FindVectorByObjectRawname(DFMgr, selected_ranges);
-            break;
-        case 4:
-            DF->Detach();
-            FindStrings(DFMgr, selected_ranges);
-            break;
-        case 5:
-            autoSearch(DF,selected_ranges);
-            break;
-        case 6:
-            DF->Detach();
-            FindVectorByBounds(DFMgr,selected_ranges);
-            break;
-        case 7:
-            DF->Detach();
-            FindPtrVectorsByObjectAddress(DFMgr,selected_ranges);
-            break;
-        case 8:
-            DF->Detach();
-            FindVectorByFirstObjectRawname(DFMgr, selected_ranges);
-            break;
-        case 9:
-            DF->Detach();
-            FindStrBufs(DFMgr, selected_ranges);
-            break;
-        case 10:
-            DF->Detach();
-            FindData(DFMgr, selected_ranges);
-            break;
-        case 11:
-            DF->Detach();
-            PtrTrace(DFMgr, selected_ranges);
-            break;
-        case 12:
-            DF->Detach();
-            DataPtrTrace(DFMgr, selected_ranges);
-            break;
-        default:
-            cout << "not implemented :(" << endl;
-    }
     #ifndef LINUX_BUILD
         cout << "Done. Press any key to continue" << endl;
         cin.ignore();
diff --git a/tools/supported/vdig.cpp b/tools/supported/vdig.cpp
index 17a7c2f2c..2b235d4a3 100644
--- a/tools/supported/vdig.cpp
+++ b/tools/supported/vdig.cpp
@@ -13,6 +13,8 @@ using namespace std;
 #include <argstream.h>
 
 #define MAX_DIM 0x300
+
+//TODO: turn into the official coord class for DFHack/DF
 class Vertex
 {
     public:
@@ -25,8 +27,10 @@ class Vertex
     {
         return (other.x == x && other.y == y && other.z == z);
     }
+    // FIXME: <tomprince> peterix_: you could probably get away with not defining operator< if you defined a std::less specialization for Vertex.
     bool operator<(const Vertex &other) const
     {
+        // FIXME: could be changed to eliminate MAX_DIM and make std::map lookups faster?
         return ( (z*MAX_DIM*MAX_DIM + y*MAX_DIM + x) < (other.z*MAX_DIM*MAX_DIM + other.y*MAX_DIM + other.x));
     }
     Vertex operator/(int number) const