From a6b7d6beeb4caf87cf2308e628b572189d08f103 Mon Sep 17 00:00:00 2001
From: hobotron-df <hobotron@netlag.com>
Date: Thu, 13 Nov 2014 12:13:23 -0600
Subject: [PATCH] New script for viewing or setting cavern adapation of
 citizens.

---
 NEWS                  |  2 +
 scripts/adaptation.rb | 95 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 97 insertions(+)
 create mode 100644 scripts/adaptation.rb

diff --git a/NEWS b/NEWS
index dada05bda..5248302e0 100644
--- a/NEWS
+++ b/NEWS
@@ -1,4 +1,6 @@
 DFHack Future
+    New scripts:
+        - adaptation: view or set the cavern adaptation level of your citizens
 
 DFHack 0.40.15-r1
     Fixes:
diff --git a/scripts/adaptation.rb b/scripts/adaptation.rb
new file mode 100644
index 000000000..7ed7302d0
--- /dev/null
+++ b/scripts/adaptation.rb
@@ -0,0 +1,95 @@
+# View or set level of cavern adaptation for the selected unit or the whole fort
+# based on removebadthoughts.rb
+
+# Color constants, values mapped to color_value enum in include/ColorText.h
+COLOR_GREEN  = 2
+COLOR_RED    = 4
+COLOR_YELLOW = 14
+COLOR_WHITE  = 15
+
+def usage(s)
+    if nil != s
+        puts(s)
+    end
+    puts "Usage: adaptation <show|set> <him|all> [value]"
+    throw :script_finished
+end
+
+mode  = $script_args[0] || 'help'
+who   = $script_args[1]
+value = $script_args[2]
+
+if 'help' == mode
+    usage(nil)
+elsif 'show' != mode && 'set' != mode
+    usage("Invalid mode '#{mode}': must be either 'show' or 'set'")
+end
+
+if nil == who
+    usage("Target not specified")
+elsif 'him' != who && 'all' != who
+    usage("Invalid target '#{who}'")
+end
+
+if 'set' == mode
+    if nil == value
+        usage("Value not specified")
+    elsif !/[[:digit:]]/.match(value)
+        usage("Invalid value '#{value}'")
+    end
+
+    if 0 > value.to_i || 800000 < value.to_i
+        usage("Value must be between 0 and 800000")
+    end
+    value = value.to_i
+end
+
+num_set = 0
+
+set_adaptation_value = lambda { |u,v|
+    next if !df.unit_iscitizen(u)
+    next if u.flags1.dead
+    u.status.misc_traits.each { |t|
+        if t.id == :CaveAdapt
+            # TBD: expose the color_ostream console and color values of
+            # t.value based on adaptation level
+            if mode == 'show'
+                if df.respond_to?(:print_color)
+                    print "Unit #{u.id} (#{u.name}) has an adaptation of "
+                    case t.value
+                    when 0..399999
+                        #df.print_color(COLOR_GREEN, "#{t.value}\n")
+                        print "#{t.value}\n"
+                    when 400000..599999
+                        df.print_color(COLOR_YELLOW, "#{t.value}\n")
+                    else
+                        df.print_color(COLOR_RED, "#{t.value}\n")
+                    end
+                else
+                    puts "Unit #{u.id} (#{u.name}) has an adaptation of #{t.value}"
+                end
+            elsif mode == 'set'
+                puts "Unit #{u.id} (#{u.name}) changed from #{t.value} to #{v}"
+                t.value = v
+                num_set += 1
+            end
+        end
+    }
+}
+
+case who
+when 'him'
+    if u = df.unit_find
+        set_adaptation_value[u,value]
+    else
+        puts 'Please select a dwarf ingame'
+    end
+when 'all'
+    df.unit_citizens.each { |uu|
+        set_adaptation_value[uu,value]
+    }
+end
+
+if 'set' == mode
+    puts "#{num_set} unit#{'s' if num_set != 1} updated."
+end