2013-01-03 16:14:16 -07:00
# include "assignJob.h"
# include "modules/Buildings.h"
# include "modules/Items.h"
# include "modules/Job.h"
# include "df/building.h"
# include "df/coord.h"
# include "df/general_ref.h"
# include "df/general_ref_building_holderst.h"
# include "df/general_ref_unit.h"
//#include "df/general_ref_unit_holderst.h"
# include "df/general_ref_unit_workerst.h"
# include "df/item.h"
# include "df/itemdef_weaponst.h"
# include "df/item_quality.h"
# include "df/item_weaponst.h"
# include "df/job.h"
# include "df/job_type.h"
# include "df/unit.h"
# include "df/unit_inventory_item.h"
2013-01-03 20:46:17 -07:00
void getRidOfOldJob ( df : : unit * unit ) {
if ( unit - > job . current_job = = NULL ) {
return ;
}
df : : job * job = unit - > job . current_job ;
unit - > job . current_job = NULL ;
if ( job - > list_link - > prev ! = NULL ) {
job - > list_link - > prev - > next = job - > list_link - > next ;
}
if ( job - > list_link - > next ! = NULL ) {
job - > list_link - > next - > prev = job - > list_link - > prev ;
}
//TODO: consider building pointers?
//for now, just let the memory leak TODO: fix
//delete job->list_link;
//delete job;
}
2013-01-03 21:18:40 -07:00
int32_t assignJob ( color_ostream & out , Edge firstImportantEdge , unordered_map < df : : coord , df : : coord , PointHash > parentMap , unordered_map < df : : coord , int64_t , PointHash > & costMap , vector < df : : unit * > & invaders , unordered_set < df : : coord , PointHash > & requiresZNeg , unordered_set < df : : coord , PointHash > & requiresZPos , MapExtras : : MapCache & cache , unordered_set < int32_t > & diggingRaces ) {
2013-01-03 16:14:16 -07:00
df : : unit * firstInvader = invaders [ 0 ] ;
//do whatever you need to do at the first important edge
df : : coord pt1 = firstImportantEdge . p1 ;
df : : coord pt2 = firstImportantEdge . p2 ;
if ( costMap [ pt1 ] > costMap [ pt2 ] ) {
df : : coord temp = pt1 ;
pt1 = pt2 ;
pt2 = temp ;
}
2013-05-31 18:39:43 -06:00
//out.print("first important edge: (%d,%d,%d) -> (%d,%d,%d)\n", pt1.x,pt1.y,pt1.z, pt2.x,pt2.y,pt2.z);
2013-01-03 16:14:16 -07:00
int32_t jobId = - 1 ;
df : : map_block * block1 = Maps : : getTileBlock ( pt1 ) ;
df : : map_block * block2 = Maps : : getTileBlock ( pt2 ) ;
2013-05-31 18:39:43 -06:00
bool passable1 = block1 - > walkable [ pt1 . x & 0xF ] [ pt1 . y & 0xF ] ;
bool passable2 = block2 - > walkable [ pt2 . x & 0xF ] [ pt2 . y & 0xF ] ;
2013-01-03 16:14:16 -07:00
2013-01-04 16:11:38 -07:00
df : : coord location ;
2013-01-03 16:14:16 -07:00
df : : building * building = Buildings : : findAtTile ( pt2 ) ;
df : : coord buildingPos = pt2 ;
if ( pt1 . z > pt2 . z ) {
building = Buildings : : findAtTile ( df : : coord ( pt2 . x , pt2 . y , pt2 . z + 1 ) ) ;
buildingPos = df : : coord ( pt2 . x , pt2 . y , pt2 . z + 1 ) ;
}
if ( building ! = NULL ) {
df : : coord destroyFrom = parentMap [ buildingPos ] ;
2013-05-31 18:39:43 -06:00
if ( destroyFrom . z ! = buildingPos . z ) {
//TODO: deal with this
}
2013-01-03 22:43:39 -07:00
//out.print("%s, line %d: Destroying building %d at (%d,%d,%d) from (%d,%d,%d).\n", __FILE__, __LINE__, building->id, buildingPos.x,buildingPos.y,buildingPos.z, destroyFrom.x,destroyFrom.y,destroyFrom.z);
2013-01-03 16:14:16 -07:00
df : : job * job = new df : : job ;
job - > job_type = df : : enums : : job_type : : DestroyBuilding ;
2013-01-03 22:08:54 -07:00
//job->flags.bits.special = 1;
2013-01-03 16:14:16 -07:00
df : : general_ref_building_holderst * buildingRef = new df : : general_ref_building_holderst ;
buildingRef - > building_id = building - > id ;
job - > general_refs . push_back ( buildingRef ) ;
df : : general_ref_unit_workerst * workerRef = new df : : general_ref_unit_workerst ;
workerRef - > unit_id = firstInvader - > id ;
job - > general_refs . push_back ( workerRef ) ;
2013-01-03 20:46:17 -07:00
getRidOfOldJob ( firstInvader ) ;
2013-01-03 16:14:16 -07:00
firstInvader - > job . current_job = job ;
firstInvader - > path . path . x . clear ( ) ;
firstInvader - > path . path . y . clear ( ) ;
firstInvader - > path . path . z . clear ( ) ;
firstInvader - > path . dest = destroyFrom ;
2013-01-04 16:11:38 -07:00
location = destroyFrom ;
2013-01-03 16:14:16 -07:00
firstInvader - > job . hunt_target = NULL ;
firstInvader - > job . destroy_target = NULL ;
building - > jobs . clear ( ) ;
building - > jobs . push_back ( job ) ;
Job : : linkIntoWorld ( job ) ;
jobId = job - > id ;
} else {
df : : tiletype * type1 = Maps : : getTileType ( pt1 ) ;
df : : tiletype * type2 = Maps : : getTileType ( pt2 ) ;
df : : tiletype_shape shape1 = ENUM_ATTR ( tiletype , shape , * type1 ) ;
df : : tiletype_shape shape2 = ENUM_ATTR ( tiletype , shape , * type2 ) ;
bool construction2 = ENUM_ATTR ( tiletype , material , * type2 ) = = df : : enums : : tiletype_material : : CONSTRUCTION ;
if ( construction2 ) {
df : : job * job = new df : : job ;
job - > job_type = df : : enums : : job_type : : RemoveConstruction ;
df : : general_ref_unit_workerst * workerRef = new df : : general_ref_unit_workerst ;
workerRef - > unit_id = firstInvader - > id ;
job - > general_refs . push_back ( workerRef ) ;
job - > pos = pt2 ;
2013-01-03 20:46:17 -07:00
getRidOfOldJob ( firstInvader ) ;
2013-01-03 16:14:16 -07:00
firstInvader - > job . current_job = job ;
firstInvader - > path . path . x . clear ( ) ;
firstInvader - > path . path . y . clear ( ) ;
firstInvader - > path . path . z . clear ( ) ;
firstInvader - > path . dest = pt1 ;
2013-01-04 16:11:38 -07:00
location = pt1 ;
2013-01-03 16:14:16 -07:00
firstInvader - > job . hunt_target = NULL ;
firstInvader - > job . destroy_target = NULL ;
Job : : linkIntoWorld ( job ) ;
jobId = job - > id ;
} else {
2013-01-04 19:02:09 -07:00
bool walkable_low1 = shape1 = = df : : tiletype_shape : : STAIR_DOWN | | shape1 = = df : : tiletype_shape : : STAIR_UPDOWN ;
bool walkable_low2 = shape2 = = df : : tiletype_shape : : STAIR_DOWN | | shape2 = = df : : tiletype_shape : : STAIR_UPDOWN ;
bool walkable_high1 = shape1 = = df : : tiletype_shape : : STAIR_UP | | shape1 = = df : : tiletype_shape : : STAIR_UPDOWN ;
bool walkable_high2 = shape2 = = df : : tiletype_shape : : STAIR_UP | | shape2 = = df : : tiletype_shape : : STAIR_UPDOWN ;
2013-01-03 16:14:16 -07:00
//must be a dig job
2013-01-04 19:02:09 -07:00
bool up1 = ! walkable_high1 & & requiresZPos . find ( pt1 ) ! = requiresZPos . end ( ) ;
bool up2 = ! walkable_high2 & & requiresZPos . find ( pt2 ) ! = requiresZPos . end ( ) ;
bool down1 = ! walkable_low1 & & requiresZNeg . find ( pt1 ) ! = requiresZNeg . end ( ) ;
bool down2 = ! walkable_low2 & & requiresZNeg . find ( pt2 ) ! = requiresZNeg . end ( ) ;
bool up ;
bool down ;
df : : coord goHere ;
df : : coord workHere ;
if ( pt1 . z = = pt2 . z ) {
up = up2 ;
down = down2 ;
goHere = pt1 ;
workHere = pt2 ;
} else {
if ( up1 | | down1 ) {
up = up1 ;
down = down1 ;
goHere = pt1 ;
workHere = pt1 ;
} else {
up = up2 ;
down = down2 ;
goHere = pt1 ;
workHere = pt2 ;
}
}
2013-01-03 16:14:16 -07:00
df : : job * job = new df : : job ;
if ( up & & down ) {
job - > job_type = df : : enums : : job_type : : CarveUpDownStaircase ;
2013-05-31 18:39:43 -06:00
//out.print("%s, line %d: type = up/down\n", __FILE__, __LINE__);
2013-01-03 16:14:16 -07:00
} else if ( up & & ! down ) {
job - > job_type = df : : enums : : job_type : : CarveUpwardStaircase ;
2013-05-31 18:39:43 -06:00
//out.print("%s, line %d: type = up\n", __FILE__, __LINE__);
2013-01-03 16:14:16 -07:00
} else if ( ! up & & down ) {
job - > job_type = df : : enums : : job_type : : CarveDownwardStaircase ;
2013-05-31 18:39:43 -06:00
//out.print("%s, line %d: type = down\n", __FILE__, __LINE__);
2013-01-03 16:14:16 -07:00
} else {
job - > job_type = df : : enums : : job_type : : Dig ;
2013-05-31 18:39:43 -06:00
//out.print("%s, line %d: type = dig\n", __FILE__, __LINE__);
2013-01-03 16:14:16 -07:00
}
2013-05-31 18:39:43 -06:00
//out.print("%s, line %d: up=%d,up1=%d,up2=%d, down=%d,down1=%d,down2=%d\n", __FILE__, __LINE__, up,up1,up2, down,down1,down2);
2013-01-04 19:02:09 -07:00
job - > pos = workHere ;
firstInvader - > path . dest = goHere ;
location = goHere ;
2013-01-03 16:14:16 -07:00
df : : general_ref_unit_workerst * ref = new df : : general_ref_unit_workerst ;
ref - > unit_id = firstInvader - > id ;
job - > general_refs . push_back ( ref ) ;
firstInvader - > job . hunt_target = NULL ;
firstInvader - > job . destroy_target = NULL ;
2013-01-03 20:46:17 -07:00
getRidOfOldJob ( firstInvader ) ;
2013-01-03 16:14:16 -07:00
firstInvader - > job . current_job = job ;
firstInvader - > path . path . x . clear ( ) ;
firstInvader - > path . path . y . clear ( ) ;
firstInvader - > path . path . z . clear ( ) ;
Job : : linkIntoWorld ( job ) ;
jobId = job - > id ;
//TODO: test if he already has a pick
//create and give a pick
df : : item_weaponst * pick = new df : : item_weaponst ;
pick - > pos = firstInvader - > pos ;
pick - > flags . bits . forbid = 1 ;
pick - > flags . bits . on_ground = 1 ;
pick - > id = ( * df : : global : : item_next_id ) + + ;
pick - > ignite_point = - 1 ;
pick - > heatdam_point = - 1 ;
pick - > colddam_point = - 1 ;
pick - > boiling_point = 11000 ;
pick - > melting_point = 10500 ;
pick - > fixed_temp = - 1 ;
pick - > weight = 0 ;
pick - > weight_fraction = 0 ;
pick - > stack_size = 1 ;
pick - > temperature . whole = 10059 ;
pick - > temperature . fraction = 0 ;
pick - > mat_type = 0 ;
pick - > mat_index = 5 ;
pick - > maker_race = 0 ; //hehe
pick - > quality = ( df : : enums : : item_quality : : item_quality ) 0 ;
pick - > skill_used = ( df : : enums : : job_skill : : job_skill ) 0 ;
pick - > maker = - 1 ;
df : : itemdef_weaponst * itemdef = NULL ;
for ( size_t a = 0 ; a < df : : global : : world - > raws . itemdefs . weapons . size ( ) ; a + + ) {
df : : itemdef_weaponst * candidate = df : : global : : world - > raws . itemdefs . weapons [ a ] ;
if ( candidate - > id = = " ITEM_WEAPON_PICK " ) {
itemdef = candidate ;
break ;
}
}
if ( itemdef = = NULL ) {
out . print ( " %s, %d: null itemdef. \n " , __FILE__ , __LINE__ ) ;
return - 1 ;
}
pick - > subtype = itemdef ;
pick - > sharpness = 5000 ;
int32_t part = - 1 ;
part = firstInvader - > body . weapon_bp ; //weapon_bp
if ( part = = - 1 ) {
out . print ( " %s, %d: no grasp part. \n " , __FILE__ , __LINE__ ) ;
return - 1 ;
}
//check for existing item there
for ( size_t a = 0 ; a < firstInvader - > inventory . size ( ) ; a + + ) {
df : : unit_inventory_item * inv_item = firstInvader - > inventory [ a ] ;
if ( false | | inv_item - > body_part_id = = part ) {
//throw it on the GROUND
Items : : moveToGround ( cache , inv_item - > item , firstInvader - > pos ) ;
}
}
Items : : moveToInventory ( cache , pick , firstInvader , df : : unit_inventory_item : : T_mode : : Weapon , part ) ;
}
}
2013-01-04 16:11:38 -07:00
#if 0
//tell EVERYONE to move there
for ( size_t a = 0 ; a < invaders . size ( ) ; a + + ) {
df : : unit * invader = invaders [ a ] ;
invader - > path . path . x . clear ( ) ;
invader - > path . path . y . clear ( ) ;
invader - > path . path . z . clear ( ) ;
invader - > path . dest = location ;
//invader->flags1.bits.invades = true;
//invader->flags1.bits.marauder = true;
//invader->flags2.bits.visitor_uninvited = true;
invader - > relations . group_leader_id = invader - > id ;
}
# endif
2013-01-03 22:08:54 -07:00
return firstInvader - > id ;
2013-01-03 16:14:16 -07:00
}