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"

    39

#include "ai_industry.hpp.sq"

    40

#include "ai_industrylist.hpp.sq"

    41

#include "ai_list.hpp.sq"

    42

#include "ai_log.hpp.sq"

    43

#include "ai_map.hpp.sq"

    44

#include "ai_marine.hpp.sq"

    45

#include "ai_order.hpp.sq"

    46

#include "ai_road.hpp.sq"

    47

#include "ai_settings.hpp.sq"

    48

#include "ai_sign.hpp.sq"

    49

#include "ai_station.hpp.sq"

    50

#include "ai_stationlist.hpp.sq"

    51

#include "ai_subsidy.hpp.sq"

    52

#include "ai_subsidylist.hpp.sq"

    53

#include "ai_testmode.hpp.sq"

    54

#include "ai_tile.hpp.sq"

    55

#include "ai_tilelist.hpp.sq"

    56

#include "ai_town.hpp.sq"

    57

#include "ai_townlist.hpp.sq"

    58

#include "ai_transactionmode.hpp.sq"

    59

#include "ai_tunnel.hpp.sq"

    60

#include "ai_vehicle.hpp.sq"

    61

#include "ai_vehiclelist.hpp.sq"

    62

    63

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

    64

{

    65

	AI_SuspendPlayer(_current_player, ticks);

    66

}

    67

    68

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

    69

{

    70

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

    71

}

    72

    73

void AIController::RegisterClasses()

    74

{

    75

	/* Register all classes */

    76

	squirrel_register_std(this->engine);

    77

	SQAIAbstractList_Register(this->engine);

    78

	SQAIAccounting_Register(this->engine);

    79

	SQAIAirport_Register(this->engine);

    80

	SQAIBase_Register(this->engine);

    81

	SQAIBridge_Register(this->engine);

    82

	SQAIBridgeList_Register(this->engine);

    83

	SQAIBridgeList_Length_Register(this->engine);

    84

	SQAICargo_Register(this->engine);

    85

	SQAICargoList_Register(this->engine);

    86

	SQAICompany_Register(this->engine);

    87

	SQAIController_Register(this->engine);

    88

	SQAIDate_Register(this->engine);

    89

	SQAIEngine_Register(this->engine);

    90

	SQAIEngineList_Register(this->engine);

    91

	SQAIError_Register(this->engine);

    92

	SQAIEvent_Register(this->engine);

    93

	SQAIEventCompanyBankrupt_Register(this->engine);

    94

	SQAIEventCompanyInTrouble_Register(this->engine);

    95

	SQAIEventCompanyMerger_Register(this->engine);

    96

	SQAIEventCompanyNew_Register(this->engine);

    97

	SQAIEventController_Register(this->engine);

    98

	SQAIEventEnginePreview_Register(this->engine);

    99

	SQAIEventSubsidyAwarded_Register(this->engine);

   100

	SQAIEventSubsidyExpired_Register(this->engine);

   101

	SQAIEventSubsidyOffer_Register(this->engine);

   102

	SQAIEventSubsidyOfferExpired_Register(this->engine);

   103

	SQAIEventTest_Register(this->engine);

   104

	SQAIEventVehicleCrash_Register(this->engine);

   105

	SQAIEventVehicleLost_Register(this->engine);

   106

	SQAIEventVehicleUnprofitable_Register(this->engine);

   107

	SQAIEventVehicleWaitingInDepot_Register(this->engine);

   108

	SQAIExecMode_Register(this->engine);

   110

	SQAIIndustry_Register(this->engine);

   111

	SQAIIndustryList_Register(this->engine);

   112

	SQAIIndustryList_CargoAccepting_Register(this->engine);

   113

	SQAIIndustryList_CargoProducing_Register(this->engine);

   114

	SQAIList_Register(this->engine);

   115

	SQAILog_Register(this->engine);

   116

	SQAIMap_Register(this->engine);

   117

	SQAIMarine_Register(this->engine);

   118

	SQAIOrder_Register(this->engine);

   119

	SQAIRoad_Register(this->engine);

   120

	SQAISettings_Register(this->engine);

   121

	SQAISign_Register(this->engine);

   122

	SQAIStation_Register(this->engine);

   123

	SQAIStationList_Register(this->engine);

   124

	SQAIStationList_Vehicle_Register(this->engine);

   125

	SQAISubsidy_Register(this->engine);

   126

	SQAISubsidyList_Register(this->engine);

   127

	SQAITestMode_Register(this->engine);

   128

	SQAITile_Register(this->engine);

   129

	SQAITileList_Register(this->engine);

   130

	SQAITileList_IndustryAccepting_Register(this->engine);

   131

	SQAITileList_IndustryProducing_Register(this->engine);

   132

	SQAITown_Register(this->engine);

   133

	SQAITownList_Register(this->engine);

   134

	SQAITransactionMode_Register(this->engine);

   135

	SQAITunnel_Register(this->engine);

   136

	SQAIVehicle_Register(this->engine);

   137

	SQAIVehicleList_Register(this->engine);

   138

	SQAIVehicleList_Station_Register(this->engine);

   139

   140

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

   141

}

   142

   143

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

   144

{

   145

	/* Convert to OpenTTD internal capable string */

   146

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

   147

}

   148

   149

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

   150

{

   151

	this->tick = 0;

   152

	this->SQ_instance = NULL;

   153

   154

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

   155

	this->engine = new Squirrel();

   156

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

   157

	this->RegisterClasses();

   158

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

   159

		delete this->engine;

   160

		this->engine = NULL;

   161

		return;

   162

	}

   163

	/* Create the main-class */

   164

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

   165

	if (!this->engine->CreateClassInstance(class_name, this, this->SQ_instance)) {

   166

		delete this->engine;

   167

		this->engine = NULL;

   168

		return;

   169

	}

   170

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

   171

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

   172

}

   173

   174

void AIController::Start()

   175

{

   176

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

   177

}

   178

   179

void AIController::Stop()

   180

{

   181

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

   182

}

   183

   184

AIController::~AIController()

   185

{

   186

	if (this->engine != NULL) delete this->engine;

   187

	free(this->SQ_instance);

   188

}

   189

   190

void AIController::IncreaseTick()

   191

{

   192

	this->tick++;

   193

}

   194

   195

uint AIController::GetTick()

   196

{

   197

	return this->tick;

   198

}