33 lines
477 B
C
33 lines
477 B
C
|
#pragma once
|
||
|
|
|
||
|
|
#include <map>
|
||
|
|
|
||
|
|
#include <Core.h>
|
||
|
|
|
||
|
|
using namespace DFHack;
|
||
|
|
using namespace df::enums;
|
||
|
|
|
||
|
|
#include "df/job.h"
|
||
|
|
#include "df/job_type.h"
|
||
|
|
#include "df/unit_labor.h"
|
||
|
|
|
||
|
|
|
||
|
|
class jlfunc;
|
||
|
|
|
||
|
|
class JobLaborMapper {
|
||
|
|
|
||
|
|
private:
|
||
|
|
std::map<df::job_type, jlfunc*> job_to_labor_table;
|
||
|
|
std::map<df::unit_labor, jlfunc*> jlf_cache;
|
||
|
|
|
||
|
|
jlfunc* jlf_const(df::unit_labor l);
|
||
|
|
|
||
|
|
public:
|
||
|
|
~JobLaborMapper();
|
||
|
|
JobLaborMapper();
|
||
|
|
|
||
|
|
df::unit_labor find_job_labor(df::job* j);
|
||
|
|
|
||
|
|
|
||
|
|
};
|