1

/* $Id$ */

     2

     3

/** @file ai_controller.cpp Implementation of AIControler. */

     4

     5

#include "../../stdafx.h"

     6

#include "../../openttd.h"

     7

#include "../../player_func.h"

     8

#include "../../core/alloc_func.hpp"

     9

#include "table/strings.h"

    10

    11

#include <squirrel.h>

    12

#include "../../squirrel.hpp"

    13

#include "../../squirrel_helper.hpp"

    14

#include "../../squirrel_class.hpp"

    15

#include "../../squirrel_std.hpp"

    16

#include "ai_controller.hpp"

    17

#include "../ai_threads.h"

    18

    19

/* Convert all AI related classes to Squirrel data.

    20

 * Note: this line a marker in squirrel_export.sh. Do not change! */

    21

#include "ai_abstractlist.hpp.sq"

    22

#include "ai_accounting.hpp.sq"

    23

#include "ai_airport.hpp.sq"

    24

#include "ai_base.hpp.sq"

    25

#include "ai_bridge.hpp.sq"

    26

#include "ai_bridgelist.hpp.sq"

    27

#include "ai_cargo.hpp.sq"

    28

#include "ai_cargolist.hpp.sq"

    29

#include "ai_company.hpp.sq"

    30

#include "ai_controller.hpp.sq"

    31

#include "ai_date.hpp.sq"

    32

#include "ai_engine.hpp.sq"

    33

#include "ai_enginelist.hpp.sq"

    34

#include "ai_error.hpp.sq"

    35

#include "ai_event.hpp.sq"

    36

#include "ai_event_types.hpp.sq"

    37

#include "ai_execmode.hpp.sq"

    38

#include "ai_industry.hpp.sq"

    39

#include "ai_industrylist.hpp.sq"

    40

#include "ai_list.hpp.sq"

    41

#include "ai_log.hpp.sq"

    42

#include "ai_map.hpp.sq"

    43

#include "ai_marine.hpp.sq"

    44

#include "ai_order.hpp.sq"

    45

#include "ai_road.hpp.sq"

    46

#include "ai_settings.hpp.sq"

    47

#include "ai_sign.hpp.sq"

    48

#include "ai_station.hpp.sq"

    49

#include "ai_stationlist.hpp.sq"

    50

#include "ai_subsidy.hpp.sq"

    51

#include "ai_subsidylist.hpp.sq"

    52

#include "ai_testmode.hpp.sq"

    53

#include "ai_tile.hpp.sq"

    54

#include "ai_tilelist.hpp.sq"

    55

#include "ai_town.hpp.sq"

    56

#include "ai_townlist.hpp.sq"

    57

#include "ai_transactionmode.hpp.sq"

    58

#include "ai_tunnel.hpp.sq"

    59

#include "ai_vehicle.hpp.sq"

    60

#include "ai_vehiclelist.hpp.sq"

    61

    62

/* static */ void AIController::Sleep(uint ticks)

    63

{

    64

	AI_SuspendPlayer(_current_player, ticks);

    65

}

    66

    67

/* static */ void AIController::Print(bool error_msg, const char *message)

    68

{

    69

	AILog::Log(error_msg ? AILog::LOG_SQ_ERROR : AILog::LOG_SQ_INFO, message);

    70

}

    71

    72

void AIController::RegisterClasses()

    73

{

    74

	/* Register all classes */

    75

	squirrel_register_std(this->engine);

    76

	SQAIAbstractList_Register(this->engine);

    77

	SQAIAccounting_Register(this->engine);

    78

	SQAIAirport_Register(this->engine);

    79

	SQAIBase_Register(this->engine);

    80

	SQAIBridge_Register(this->engine);

    81

	SQAIBridgeList_Register(this->engine);

    82

	SQAIBridgeList_Length_Register(this->engine);

    83

	SQAICargo_Register(this->engine);

    84

	SQAICargoList_Register(this->engine);

    85

	SQAICompany_Register(this->engine);

    86

	SQAIController_Register(this->engine);

    87

	SQAIDate_Register(this->engine);

    88

	SQAIEngine_Register(this->engine);

    89

	SQAIEngineList_Register(this->engine);

    90

	SQAIError_Register(this->engine);

    91

	SQAIEvent_Register(this->engine);

    92

	SQAIEventCompanyBankrupt_Register(this->engine);

    93

	SQAIEventCompanyInTrouble_Register(this->engine);

    94

	SQAIEventCompanyMerger_Register(this->engine);

    95

	SQAIEventCompanyNew_Register(this->engine);

    96

	SQAIEventController_Register(this->engine);

    97

	SQAIEventEnginePreview_Register(this->engine);

    98

	SQAIEventSubsidyAwarded_Register(this->engine);

    99

	SQAIEventSubsidyExpired_Register(this->engine);

   100

	SQAIEventSubsidyOffer_Register(this->engine);

   101

	SQAIEventSubsidyOfferExpired_Register(this->engine);

   102

	SQAIEventTest_Register(this->engine);

   103

	SQAIEventVehicleCrash_Register(this->engine);

   104

	SQAIEventVehicleLost_Register(this->engine);

   105

	SQAIEventVehicleUnprofitable_Register(this->engine);

   106

	SQAIEventVehicleWaitingInDepot_Register(this->engine);

   107

	SQAIExecMode_Register(this->engine);

   108

	SQAIIndustry_Register(this->engine);

   109

	SQAIIndustryList_Register(this->engine);

   110

	SQAIIndustryList_CargoAccepting_Register(this->engine);

   111

	SQAIIndustryList_CargoProducing_Register(this->engine);

   112

	SQAIList_Register(this->engine);

   113

	SQAILog_Register(this->engine);

   114

	SQAIMap_Register(this->engine);

   115

	SQAIMarine_Register(this->engine);

   116

	SQAIOrder_Register(this->engine);

   117

	SQAIRoad_Register(this->engine);

   118

	SQAISettings_Register(this->engine);

   119

	SQAISign_Register(this->engine);

   120

	SQAIStation_Register(this->engine);

   121

	SQAIStationList_Register(this->engine);

   122

	SQAIStationList_Vehicle_Register(this->engine);

   123

	SQAISubsidy_Register(this->engine);

   124

	SQAISubsidyList_Register(this->engine);

   125

	SQAITestMode_Register(this->engine);

   126

	SQAITile_Register(this->engine);

   127

	SQAITileList_Register(this->engine);

   128

	SQAITileList_IndustryAccepting_Register(this->engine);

   129

	SQAITileList_IndustryProducing_Register(this->engine);

   130

	SQAITown_Register(this->engine);

   131

	SQAITownList_Register(this->engine);

   132

	SQAITransactionMode_Register(this->engine);

   133

	SQAITunnel_Register(this->engine);

   134

	SQAIVehicle_Register(this->engine);

   135

	SQAIVehicleList_Register(this->engine);

   136

	SQAIVehicleList_Station_Register(this->engine);

   137

   138

	this->engine->SetGlobalPointer(this->engine);

   139

}

   140

   141

static void PrintFunc(bool error_msg, const SQChar *message)

   142

{

   143

	/* Convert to OpenTTD internal capable string */

   144

	AIController::Print(error_msg, FS2OTTD(message));

   145

}

   146

   147

AIController::AIController(const char *script, const char *class_name)

   148

{

   149

	this->tick = 0;

   151

   152

	/* Create the Squirrel Engine and assign all required classes and settings */

   153

	this->engine = new Squirrel();

   154

	this->engine->SetPrintFunction(&PrintFunc);

   155

	this->RegisterClasses();

   156

	if (!this->engine->LoadScript(script)) {

   157

		delete this->engine;

   158

		this->engine = NULL;

   159

		return;

   160

	}

   161

	/* Create the main-class */

   162

	this->SQ_instance = MallocT<SQObject>(sizeof(SQObject));

   168

	if (this->engine->MethodExists(*this->SQ_instance, "constructor"))

   169

		this->engine->CallMethod(*this->SQ_instance, "constructor");

   170

}

   171

   172

void AIController::Start()

   173

{

   174

	if (this->engine != NULL) this->engine->CallMethod(*this->SQ_instance, "Start");

   175

}

   176

   177

void AIController::Stop()

   178

{

   179

	if (this->engine != NULL) this->engine->CallMethod(*this->SQ_instance, "Stop");

   180

}

   181

   182

AIController::~AIController()

   183

{

   185

	free(this->SQ_instance);

   186

}

   187

   188

void AIController::IncreaseTick()

   189

{

   190

	this->tick++;

   191

}

   192

   193

uint AIController::GetTick()

   194

{

   195

	return this->tick;

   196

}