Job.h

/*
https://github.com/peterix/dfhack
Copyright (c) 2009-2012 Petr Mrázek (peterix@gmail.com)

This software is provided 'as-is', without any express or implied
warranty. In no event will the authors be held liable for any
damages arising from the use of this software.

Permission is granted to anyone to use this software for any
purpose, including commercial applications, and to alter it and
redistribute it freely, subject to the following restrictions:

1. The origin of this software must not be misrepresented; you must
not claim that you wrote the original software. If you use this
software in a product, an acknowledgment in the product documentation
would be appreciated but is not required.

2. Altered source versions must be plainly marked as such, and
must not be misrepresented as being the original software.

3. This notice may not be removed or altered from any source
distribution.
*/

#pragma once
#ifndef CL_MOD_JOB
#define CL_MOD_JOB

#include "Export.h"
#include "Types.h"
#include "DataDefs.h"

#include "df/building_workshopst.h"
#include "df/item_type.h"
#include "df/job_item_ref.h"

namespace df
{
    struct job;
    struct job_item;
    struct job_item_filter;
    struct building;
    struct unit;
    struct manager_order;
}

namespace DFHack
{
    class color_ostream;

    namespace Job {

        // add a general reference with the specified general type and id
        DFHACK_EXPORT bool addGeneralRef(df::job *job, df::general_ref_type type, int32_t id);

        // Duplicate the job structure. It is not linked into any DF lists.
        DFHACK_EXPORT df::job *cloneJobStruct(df::job *job, bool keepEverything=false);

        // Delete a cloned structure.
        DFHACK_EXPORT void deleteJobStruct(df::job *job, bool keptEverything=false);

        DFHACK_EXPORT void printItemDetails(color_ostream &out, df::job_item *item, int idx);
        DFHACK_EXPORT void printJobDetails(color_ostream &out, df::job *job);

        DFHACK_EXPORT df::general_ref *getGeneralRef(df::job *job, df::general_ref_type type);
        DFHACK_EXPORT df::specific_ref *getSpecificRef(df::job *job, df::specific_ref_type type);

        DFHACK_EXPORT df::building *getHolder(df::job *job);
        DFHACK_EXPORT df::unit *getWorker(df::job *job);

        DFHACK_EXPORT void setJobCooldown(df::building *workshop, df::unit *worker, int cooldown = 100);

        // assign the job to a worker, clearing up the job posting (if it exists)
        DFHACK_EXPORT bool addWorker(df::job *job, df::unit *unit);
        DFHACK_EXPORT bool removeWorker(df::job *job, int cooldown = 100);

        // This helpful method only removes the backref from the item to the job, but it doesn't
        // remove the item ref from the job's vector, or delete it or anything.  Think of it as a method
        // that does all the needful to make an item ref ready to delete.
        DFHACK_EXPORT void disconnectJobItem(df::job *job, df::job_item_ref *item);
        // This helpful method only removes the backref from whatever the general_ref points to,
        // it doesn't remove the general_ref from the job's vector, or delete it or anything.
        // Think of it as a method that does all the needful to make a ref ready to delete.
        // If it returns false, you've found a ref that the method doesn't know how to handle.  Congratulations!
        // You should report that and/or check in a fix.
        DFHACK_EXPORT bool disconnectJobGeneralRef(df::job *job, df::general_ref *ref);
        // Delete a job & remove all refs from everywhere.
        // This method DELETES the job object!  Everything related to it will be wiped
        // clean from the earth, so make sure you pull what you need out before calling this!
        DFHACK_EXPORT bool removeJob(df::job *job);

        // Instruct the game to check and assign workers
        DFHACK_EXPORT void checkBuildingsNow();
        DFHACK_EXPORT void checkDesignationsNow();

        // link the job into the global job list, passing ownership to DF
        DFHACK_EXPORT bool linkIntoWorld(df::job *job, bool new_id = true);
        // create a job and immediately link it into the global job list
        DFHACK_EXPORT df::job* createLinked();

        // assign job to workshop, returns false if workshop already has the maximum of ten jobs
        DFHACK_EXPORT bool assignToWorkshop(df::job *job, df::building_workshopst *workshop);

        // Flag this job's posting as "dead" and set its posting_index to -1
        // If remove_all is true, flag all postings pointing to this job
        // Returns true if any postings were removed
        DFHACK_EXPORT bool removePostings(df::job *job, bool remove_all = false);

        // lists jobs with ids >= *id_var, and sets *id_var = *job_next_id;
        DFHACK_EXPORT bool listNewlyCreated(std::vector<df::job*> *pvec, int *id_var);

        DFHACK_EXPORT bool attachJobItem(df::job *job, df::item *item,
                                         df::job_role_type role,
                                         int filter_idx = -1, int insert_idx = -1);

        DFHACK_EXPORT bool isSuitableItem(const df::job_item *item, df::item_type itype, int isubtype);
        DFHACK_EXPORT bool isSuitableMaterial(const df::job_item *item, int mat_type,
                                              int mat_index,
                                              df::item_type itype);
        DFHACK_EXPORT std::string getName(df::job *job);
        DFHACK_EXPORT std::string getManagerOrderName(df::manager_order *order);

        struct JobDeleter {
            void operator()(df::job *ptr) const {
                if (ptr)
                    Job::deleteJobStruct(ptr, true);
            }
        };
        using JobUniquePtr = std::unique_ptr<df::job, JobDeleter>;
    }

    DFHACK_EXPORT bool operator== (const df::job_item &a, const df::job_item &b);
    DFHACK_EXPORT bool operator== (const df::job &a, const df::job &b);
}
#endif