Add api for enabling liquid and temperature updates for blocks.

angavrilov · angavrilov · commit 60bb486abac5 · 2012-05-12T20:12:09.000+04:00
Now updates also have to be enabled for the z level to work.
diff --git a/LUA_API.rst b/LUA_API.rst
@@ -666,6 +666,14 @@ Job module
 
   Returns the unit performing the job.
 
+* ``dfhack.job.checkBuildingsNow()``
+
+  Instructs the game to check buildings for jobs next frame and assign workers.
+
+* ``dfhack.job.checkDesignationsNow()``
+
+  Instructs the game to check designations for jobs next frame and assign workers.
+
 * ``dfhack.job.is_equal(job1,job2)``
 
   Compares important fields in the job and nested item structures.
@@ -803,6 +811,10 @@ Maps module
 
   Returns the biome info struct for the given global map region.
 
+* ``dfhack.maps.enableBlockUpdates(block[,flow,temperature])``
+
+  Enables updates for liquid flow or temperature, unless already active.
+
 * ``dfhack.maps.getGlobalInitFeature(index)``
 
   Returns the global feature object with the given index.
diff --git a/Lua API.html b/Lua API.html
@@ -912,6 +912,12 @@ <h3><a class="toc-backref" href="#id15">Job module</a></h3>
 <li><p class="first"><tt class="docutils literal">dfhack.job.getWorker(job)</tt></p>
 <p>Returns the unit performing the job.</p>
 </li>
+<li><p class="first"><tt class="docutils literal">dfhack.job.checkBuildingsNow()</tt></p>
+<p>Instructs the game to check buildings for jobs next frame and assign workers.</p>
+</li>
+<li><p class="first"><tt class="docutils literal">dfhack.job.checkDesignationsNow()</tt></p>
+<p>Instructs the game to check designations for jobs next frame and assign workers.</p>
+</li>
 <li><p class="first"><tt class="docutils literal">dfhack.job.is_equal(job1,job2)</tt></p>
 <p>Compares important fields in the job and nested item structures.</p>
 </li>
@@ -1023,6 +1029,9 @@ <h3><a class="toc-backref" href="#id18">Maps module</a></h3>
 <li><p class="first"><tt class="docutils literal">dfhack.maps.getRegionBiome(region_coord2d)</tt></p>
 <p>Returns the biome info struct for the given global map region.</p>
 </li>
+<li><p class="first"><tt class="docutils literal"><span class="pre">dfhack.maps.enableBlockUpdates(block[,flow,temperature])</span></tt></p>
+<p>Enables updates for liquid flow or temperature, unless already active.</p>
+</li>
 <li><p class="first"><tt class="docutils literal">dfhack.maps.getGlobalInitFeature(index)</tt></p>
 <p>Returns the global feature object with the given index.</p>
 </li>
diff --git a/library/LuaApi.cpp b/library/LuaApi.cpp
@@ -637,6 +637,8 @@ static const LuaWrapper::FunctionReg dfhack_job_module[] = {
     WRAPM(Job,printJobDetails),
     WRAPM(Job,getHolder),
     WRAPM(Job,getWorker),
+    WRAPM(Job,checkBuildingsNow),
+    WRAPM(Job,checkDesignationsNow),
     WRAPN(is_equal, jobEqual),
     WRAPN(is_item_equal, jobItemEqual),
     { NULL, NULL }
@@ -753,6 +755,7 @@ static const luaL_Reg dfhack_items_funcs[] = {
 static const LuaWrapper::FunctionReg dfhack_maps_module[] = {
     WRAPN(getBlock, (df::map_block* (*)(int32_t,int32_t,int32_t))Maps::getBlock),
     WRAPM(Maps, getRegionBiome),
+    WRAPM(Maps, enableBlockUpdates),
     WRAPM(Maps, getGlobalInitFeature),
     WRAPM(Maps, getLocalInitFeature),
     WRAPM(Maps, canWalkBetween),
diff --git a/library/include/modules/Job.h b/library/include/modules/Job.h
@@ -57,6 +57,10 @@ namespace DFHack
         DFHACK_EXPORT df::building *getHolder(df::job *job);
         DFHACK_EXPORT df::unit *getWorker(df::job *job);
 
+        // Instruct the game to check and assign workers
+        DFHACK_EXPORT void checkBuildingsNow();
+        DFHACK_EXPORT void checkDesignationsNow();
+
         DFHACK_EXPORT bool linkIntoWorld(df::job *job, bool new_id = true);
 
         // lists jobs with ids >= *id_var, and sets *id_var = *job_next_id;
diff --git a/library/include/modules/MapCache.h b/library/include/modules/MapCache.h
@@ -93,6 +93,12 @@ class DFHACK_EXPORT Block
     Block(MapCache *parent, DFCoord _bcoord);
     ~Block();
 
+    DFCoord getCoord() { return bcoord; }
+
+    void enableBlockUpdates(bool flow = false, bool temp = false) {
+        Maps::enableBlockUpdates(block, flow, temp);
+    }
+
     /*
      * All coordinates are taken mod 16.
      */
@@ -208,11 +214,8 @@ class DFHACK_EXPORT Block
         dirty_designations = true;
         //printf("setting block %d/%d/%d , %d %d\n",x,y,z, p.x, p.y);
         index_tile<df::tile_designation&>(designation,p) = des;
-        if(des.bits.dig)
-        {
-            dirty_blockflags = true;
-            blockflags.bits.designated = true;
-        }
+        if(des.bits.dig && block)
+            block->flags.bits.designated = true;
         return true;
     }
 
@@ -236,15 +239,7 @@ class DFHACK_EXPORT Block
 
     t_blockflags BlockFlags()
     {
-        return blockflags;
-    }
-    bool setBlockFlags(t_blockflags des)
-    {
-        if(!valid) return false;
-        dirty_blockflags = true;
-        //printf("setting block %d/%d/%d , %d %d\n",x,y,z, p.x, p.y);
-        blockflags = des;
-        return true;
+        return block ? block->flags : t_blockflags();
     }
 
     bool Write();
@@ -273,7 +268,6 @@ class DFHACK_EXPORT Block
     bool dirty_designations:1;
     bool dirty_tiles:1;
     bool dirty_temperatures:1;
-    bool dirty_blockflags:1;
     bool dirty_occupancies:1;
 
     DFCoord bcoord;
@@ -328,7 +322,6 @@ class DFHACK_EXPORT Block
 
     designations40d designation;
     occupancies40d occupancy;
-    t_blockflags blockflags;
 
     t_temperatures temp1;
     t_temperatures temp2;
diff --git a/library/include/modules/Maps.h b/library/include/modules/Maps.h
@@ -257,6 +257,9 @@ inline df::tile_occupancy *getTileOccupancy(df::coord pos) {
 
 DFHACK_EXPORT df::world_data::T_region_map *getRegionBiome(df::coord2d rgn_pos);
 
+// Enables per-frame updates for liquid flow and/or temperature.
+DFHACK_EXPORT void enableBlockUpdates(df::map_block *blk, bool flow = false, bool temperature = false);
+
 /// sorts the block event vector into multiple vectors by type
 /// mineral veins, what's under ice, blood smears and mud
 extern DFHACK_EXPORT bool SortBlockEvents(df::map_block *block,
diff --git a/library/modules/Buildings.cpp b/library/modules/Buildings.cpp
@@ -709,8 +709,7 @@ static void linkBuilding(df::building *bld)
 
     linkRooms(bld);
 
-    if (process_jobs)
-        *process_jobs = true;
+    Job::checkBuildingsNow();
 }
 
 static void createDesign(df::building *bld, bool rough)
@@ -900,8 +899,6 @@ bool Buildings::deconstruct(df::building *bld)
 {
     using df::global::ui;
     using df::global::world;
-    using df::global::process_jobs;
-    using df::global::process_dig;
     using df::global::ui_look_list;
 
     CHECK_NULL_POINTER(bld);
@@ -952,8 +949,8 @@ bool Buildings::deconstruct(df::building *bld)
         }
     }
 
-    if (process_dig) *process_dig = true;
-    if (process_jobs) *process_jobs = true;
+    Job::checkBuildingsNow();
+    Job::checkDesignationsNow();
 
     return true;
 }
diff --git a/library/modules/Job.cpp b/library/modules/Job.cpp
@@ -256,6 +256,18 @@ df::unit *DFHack::Job::getWorker(df::job *job)
     return NULL;
 }
 
+void DFHack::Job::checkBuildingsNow()
+{
+    if (df::global::process_jobs)
+        *df::global::process_jobs = true;
+}
+
+void DFHack::Job::checkDesignationsNow()
+{
+    if (df::global::process_dig)
+        *df::global::process_dig = true;
+}
+
 bool DFHack::Job::linkIntoWorld(df::job *job, bool new_id)
 {
     using df::global::world;
diff --git a/library/modules/Maps.cpp b/library/modules/Maps.cpp
@@ -54,6 +54,7 @@ using namespace std;
 #include "df/world_region_details.h"
 #include "df/builtin_mats.h"
 #include "df/block_square_event_grassst.h"
+#include "df/z_level_flags.h"
 
 using namespace DFHack;
 using namespace df::enums;
@@ -176,6 +177,30 @@ df::world_data::T_region_map *Maps::getRegionBiome(df::coord2d rgn_pos)
     return &data->region_map[rgn_pos.x][rgn_pos.y];
 }
 
+void Maps::enableBlockUpdates(df::map_block *blk, bool flow, bool temperature)
+{
+    if (!blk || !(flow || temperature)) return;
+
+    if (temperature)
+        blk->flags.bits.update_temperature = true;
+
+    if (flow)
+    {
+        blk->flags.bits.update_liquid = true;
+        blk->flags.bits.update_liquid_twice = true;
+    }
+
+    auto z_flags = world->map.z_level_flags;
+    int z_level = blk->map_pos.z;
+
+    if (z_flags && z_level >= 0 && z_level < world->map.z_count_block)
+    {
+        z_flags += z_level;
+        z_flags->bits.update = true;
+        z_flags->bits.update_twice = true;
+    }
+}
+
 df::feature_init *Maps::getGlobalInitFeature(int32_t index)
 {
     auto data = world->world_data;
@@ -426,7 +451,6 @@ MapExtras::Block::Block(MapCache *parent, DFCoord _bcoord) : parent(parent)
     dirty_designations = false;
     dirty_tiles = false;
     dirty_temperatures = false;
-    dirty_blockflags = false;
     dirty_occupancies = false;
     valid = false;
     bcoord = _bcoord;
@@ -440,7 +464,6 @@ MapExtras::Block::Block(MapCache *parent, DFCoord _bcoord) : parent(parent)
     {
         COPY(designation, block->designation);
         COPY(occupancy, block->occupancy);
-        blockflags = block->flags;
 
         COPY(temp1, block->temperature_1);
         COPY(temp2, block->temperature_2);
@@ -449,7 +472,6 @@ MapExtras::Block::Block(MapCache *parent, DFCoord _bcoord) : parent(parent)
     }
     else
     {
-        blockflags.whole = 0;
         memset(designation,0,sizeof(designation));
         memset(occupancy,0,sizeof(occupancy));
         memset(temp1,0,sizeof(temp1));
@@ -634,11 +656,6 @@ bool MapExtras::Block::Write ()
 {
     if(!valid) return false;
 
-    if(dirty_blockflags)
-    {
-        block->flags = blockflags;
-        dirty_blockflags = false;
-    }
     if(dirty_designations)
     {
         COPY(block->designation, designation);
diff --git a/plugins/liquids.cpp b/plugins/liquids.cpp

Original file line number	Diff line number	Diff line change
`@@ -709,8 +709,7 @@ static void linkBuilding(df::building *bld)`
`709`	`709`
`710`	`710`	`linkRooms(bld);`
`711`	`711`
`712`		`- if (process_jobs)`
`713`		`- *process_jobs = true;`
	`712`	`+ Job::checkBuildingsNow();`
`714`	`713`	`}`
`715`	`714`
`716`	`715`	`static void createDesign(df::building *bld, bool rough)`
`@@ -900,8 +899,6 @@ bool Buildings::deconstruct(df::building *bld)`
`900`	`899`	`{`
`901`	`900`	`using df::global::ui;`
`902`	`901`	`using df::global::world;`
`903`		`- using df::global::process_jobs;`
`904`		`- using df::global::process_dig;`
`905`	`902`	`using df::global::ui_look_list;`
`906`	`903`
`907`	`904`	`CHECK_NULL_POINTER(bld);`
`@@ -952,8 +949,8 @@ bool Buildings::deconstruct(df::building *bld)`
`952`	`949`	`}`
`953`	`950`	`}`
`954`	`951`
`955`		`- if (process_dig) *process_dig = true;`
`956`		`- if (process_jobs) *process_jobs = true;`
	`952`	`+ Job::checkBuildingsNow();`
	`953`	`+ Job::checkDesignationsNow();`
`957`	`954`
`958`	`955`	`return true;`
`959`	`956`	`}`