mirror of
https://github.com/vcmi/vcmi.git
synced 2024-11-24 08:32:34 +02:00
592 lines
14 KiB
C++
592 lines
14 KiB
C++
#include "stdafx.h"
|
|
#include "CLua.h"
|
|
#include "CLuaHandler.h"
|
|
#include "hch/CHeroHandler.h"
|
|
#include "lua.h"
|
|
#include "lualib.h"
|
|
#include "lauxlib.h"
|
|
#include "lobject.h"
|
|
#include "lgc.h"
|
|
#include "lapi.h"
|
|
#include "CGameInfo.h"
|
|
#include "CGameState.h"
|
|
#include <sstream>
|
|
#include "hch/CObjectHandler.h"
|
|
#include "CCallback.h"
|
|
#include "hch/CGeneralTextHandler.h"
|
|
#include <sstream>
|
|
#include "CPlayerInterface.h"
|
|
#include <boost/algorithm/string.hpp>
|
|
#include <boost/algorithm/string/replace.hpp>
|
|
#pragma warning (disable : 4311)
|
|
bool getGlobalFunc(lua_State * L, std::string fname)
|
|
{
|
|
unsigned int hash = lua_calchash(fname.c_str(), fname.size());
|
|
lua_pushhstring(L, hash, fname.c_str(), fname.size());
|
|
lua_gettable(L, LUA_GLOBALSINDEX);
|
|
return lua_isfunction(L, -1);
|
|
}
|
|
|
|
CObjectScript::CObjectScript()
|
|
{
|
|
language = ESLan::UNDEF;
|
|
//std::cout << "Tworze obiekt objectscript "<<this<<std::endl;
|
|
}
|
|
|
|
CObjectScript::~CObjectScript()
|
|
{
|
|
//std::cout << "Usuwam obiekt objectscript "<<this<<std::endl;
|
|
}
|
|
|
|
CScript::CScript()
|
|
{
|
|
//std::cout << "Tworze obiekt CScript "<<this<<std::endl;
|
|
}
|
|
CScript::~CScript()
|
|
{
|
|
//std::cout << "Usuwam obiekt CScript "<<this<<std::endl;
|
|
}
|
|
|
|
#define LST (is)
|
|
CLua::CLua(std::string initpath)
|
|
{
|
|
opened=false;
|
|
open(initpath);
|
|
}
|
|
CLua::CLua()
|
|
{
|
|
//std::cout << "Tworze obiekt clua "<<this<<std::endl;
|
|
opened=false;
|
|
}
|
|
void CLua::open(std::string initpath)
|
|
{
|
|
LST = lua_open();
|
|
opened = true;
|
|
LUA_OPEN_LIB(LST, luaopen_base);
|
|
LUA_OPEN_LIB(LST, luaopen_io);
|
|
if ((luaL_loadfile (LST, initpath.c_str())) == 0)
|
|
{
|
|
lua_pcall (LST, 0, LUA_MULTRET, 0);
|
|
}
|
|
else
|
|
{
|
|
std::string temp = "Cannot open script ";
|
|
temp += initpath;
|
|
throw std::exception(temp.c_str());
|
|
}
|
|
}
|
|
void CLua::registerCLuaCallback()
|
|
{
|
|
}
|
|
|
|
CLua::~CLua()
|
|
{
|
|
//std::cout << "Usuwam obiekt clua "<<this<<std::endl;
|
|
if (opened)
|
|
{
|
|
std::cout << "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"<<std::endl;
|
|
lua_close(LST);
|
|
}
|
|
}
|
|
|
|
void CLua::findF(std::string fname)
|
|
{
|
|
lua_getfield(is, LUA_GLOBALSINDEX, fname.c_str()); /* function to be called */
|
|
}
|
|
void CLua::findF2(std::string fname)
|
|
{
|
|
lua_pushstring (is, fname.c_str());
|
|
lua_gettable (is, LUA_GLOBALSINDEX);
|
|
}
|
|
void CLua::findFS(std::string fname)
|
|
{
|
|
lua_settop(is, 0);
|
|
if (!getGlobalFunc(is,fname))
|
|
{
|
|
lua_settop(is, 0);
|
|
throw new std::exception((fname + ": function not defined").c_str()); // the call is not defined
|
|
}
|
|
}
|
|
#undef LST
|
|
|
|
CLuaObjectScript::CLuaObjectScript(std::string filename)
|
|
{
|
|
language = ESLan::LUA;
|
|
open(filename);
|
|
//binit = bnewobject = bonherovisit = brightext = false;
|
|
//std::cout << "Tworze obiekt CLuaObjectScript "<<this<<std::endl;
|
|
}
|
|
CLuaObjectScript::~CLuaObjectScript()
|
|
{
|
|
//std::cout << "Usuwam obiekt CLuaObjectScript "<<this<<std::endl;
|
|
}
|
|
|
|
void CLuaObjectScript::init()
|
|
{
|
|
}
|
|
|
|
std::string CLuaObjectScript::genFN(std::string base, int ID)
|
|
{
|
|
std::stringstream sts;
|
|
sts<<base<<"_"<<ID;
|
|
return sts.str();
|
|
}
|
|
|
|
void CLuaObjectScript::newObject(CGObjectInstance *os)
|
|
{
|
|
findF(genFN("newObject",os->ID));
|
|
lua_pushinteger(is, (int)os);
|
|
if (lua_pcall (is, 1, 0, 0))
|
|
{
|
|
lua_settop(is, 0);
|
|
throw new std::exception(("Failed to call "+genFN("newObject",os->ID)+" function in lua script.").c_str());
|
|
}
|
|
lua_settop(is, 0);
|
|
return;
|
|
}
|
|
void CLuaObjectScript::onHeroVisit(CGObjectInstance *os, int heroID)
|
|
{
|
|
findF(genFN("heroVisit",os->ID));
|
|
lua_pushinteger(is, (int)os);
|
|
lua_pushinteger(is, heroID);
|
|
if (lua_pcall (is, 2, 0, 0))
|
|
{
|
|
lua_settop(is, 0);
|
|
throw new std::exception(("Failed to call "+genFN("heroVisit",os->ID)+" function in lua script.").c_str());
|
|
}
|
|
lua_settop(is, 0);
|
|
}
|
|
std::string CLuaObjectScript::hoverText(CGObjectInstance *os)
|
|
{
|
|
findF(genFN("hoverText",os->ID));
|
|
lua_pushinteger(is, (int)os);
|
|
if (lua_pcall (is, 1, 1, 0))
|
|
{
|
|
lua_settop(is, 0);
|
|
throw new std::exception(("Failed to call "+genFN("hoverText",os->ID)+" function in lua script.").c_str());
|
|
}
|
|
std::string ret = lua_tostring(is,1);
|
|
lua_settop(is, 0);
|
|
return ret;
|
|
}
|
|
|
|
std::string CCPPObjectScript::hoverText(CGObjectInstance *os)
|
|
{
|
|
return CGI->objh->objects[os->defInfo->id].name;
|
|
}
|
|
|
|
void CVisitableOPH::newObject(CGObjectInstance *os)
|
|
{
|
|
visitors.insert
|
|
(std::pair<CGObjectInstance*,std::set<int> >(os,std::set<int>()));
|
|
};
|
|
|
|
void CVisitableOPH::onHeroVisit(CGObjectInstance *os, int heroID)
|
|
{
|
|
if (visitors.find(os)!=visitors.end())
|
|
{
|
|
if(visitors[os].find(heroID)==visitors[os].end())
|
|
{
|
|
onNAHeroVisit(os,heroID, false);
|
|
visitors[os].insert(heroID);
|
|
}
|
|
else
|
|
{
|
|
onNAHeroVisit(os,heroID, true);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
throw new std::exception("Skrypt nie zainicjalizowal instancji tego obiektu. :(");
|
|
}
|
|
};
|
|
void CVisitableOPH::onNAHeroVisit(CGObjectInstance *os, int heroID, bool alreadyVisited)
|
|
{
|
|
int w=0, ot=0, vvv=1;
|
|
switch(os->ID)
|
|
{
|
|
case 51:
|
|
w=0;
|
|
ot=80;
|
|
break;
|
|
case 23:
|
|
w=1;
|
|
ot=39;
|
|
break;
|
|
case 61:
|
|
w=2;
|
|
ot=100;
|
|
break;
|
|
case 32:
|
|
w=3;
|
|
ot=59;
|
|
break;
|
|
case 100:
|
|
w=4;
|
|
ot=143;
|
|
vvv=1000;
|
|
break;
|
|
}
|
|
if (!alreadyVisited)
|
|
{
|
|
switch (os->ID)
|
|
{
|
|
case 51:
|
|
case 23:
|
|
case 61:
|
|
case 32:
|
|
{
|
|
cb->changePrimSkill(heroID,w,vvv);
|
|
std::vector<SComponent*> weko;
|
|
weko.push_back(new SComponent(SComponent::primskill,w,vvv));
|
|
cb->showInfoDialog(cb->getHeroOwner(heroID),CGI->objh->advobtxt[ot],&weko);
|
|
//for (int ii=0; ii<weko.size();ii++)
|
|
// delete weko[ii];
|
|
break;
|
|
}
|
|
case 100:
|
|
{
|
|
cb->changePrimSkill(heroID,w,vvv);
|
|
std::vector<SComponent*> weko;
|
|
weko.push_back(new SComponent(SComponent::experience,0,vvv));
|
|
cb->showInfoDialog(cb->getHeroOwner(heroID),CGI->objh->advobtxt[ot],&weko);
|
|
//for (int ii=0; ii<weko.size();ii++)
|
|
// delete weko[ii];
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ot++;
|
|
cb->showInfoDialog(cb->getHeroOwner(heroID),CGI->objh->advobtxt[ot],&std::vector<SComponent*>());
|
|
}
|
|
}
|
|
|
|
std::vector<int> CVisitableOPH::yourObjects()
|
|
{
|
|
std::vector<int> ret(5);
|
|
ret.push_back(51);
|
|
ret.push_back(23);
|
|
ret.push_back(61);
|
|
ret.push_back(32);
|
|
ret.push_back(100);
|
|
return ret;
|
|
}
|
|
|
|
std::string CVisitableOPH::hoverText(CGObjectInstance *os)
|
|
{
|
|
std::string add;
|
|
int pom;
|
|
switch(os->ID)
|
|
{
|
|
case 51:
|
|
pom = 8;
|
|
break;
|
|
case 23:
|
|
pom = 7;
|
|
break;
|
|
case 61:
|
|
pom = 11;
|
|
break;
|
|
case 32:
|
|
pom = 4;
|
|
break;
|
|
case 100:
|
|
pom = 5;
|
|
break;
|
|
default:
|
|
throw new std::exception("Unsupported ID in CVisitableOPH::hoverText");
|
|
}
|
|
add = " " + CGI->objh->xtrainfo[pom] + " ";
|
|
int heroID = cb->getSelectedHero();
|
|
if (heroID>=0)
|
|
{
|
|
add += ( (visitors[os].find(heroID) == visitors[os].end())
|
|
?
|
|
(CGI->generaltexth->allTexts[353]) //not visited
|
|
:
|
|
( CGI->generaltexth->allTexts[352]) ); //visited
|
|
}
|
|
return CGI->objh->objects[os->defInfo->id].name + add;
|
|
}
|
|
|
|
void CVisitableOPW::onNAHeroVisit(CGObjectInstance *os, int heroID, bool alreadyVisited)
|
|
{
|
|
int mid;
|
|
switch (os->ID)
|
|
{
|
|
case 55:
|
|
mid = 92;
|
|
break;
|
|
case 112:
|
|
mid = 170;
|
|
break;
|
|
case 109:
|
|
mid = 164;
|
|
break;
|
|
}
|
|
if (alreadyVisited)
|
|
{
|
|
if (os->ID!=112)
|
|
mid++;
|
|
else
|
|
mid--;
|
|
cb->showInfoDialog(cb->getHeroOwner(heroID),CGI->objh->advobtxt[mid],&std::vector<SComponent*>()); //TODO: maybe we have memory leak with these windows
|
|
}
|
|
else
|
|
{
|
|
int type, sub, val;
|
|
type = SComponent::resource;
|
|
switch (os->ID)
|
|
{
|
|
case 55:
|
|
if (rand()%2)
|
|
{
|
|
sub = 5;
|
|
val = 5;
|
|
}
|
|
else
|
|
{
|
|
sub = 6;
|
|
val = 500;
|
|
}
|
|
break;
|
|
case 112:
|
|
mid = 170;
|
|
sub = rand() % 6;
|
|
val = (rand() % 4) + 3;
|
|
break;
|
|
case 109:
|
|
mid = 164;
|
|
sub = 6;
|
|
if(cb->getDate(2)<2)
|
|
val = 500;
|
|
else
|
|
val = 1000;
|
|
}
|
|
SComponent * com = new SComponent((SComponent::Etype)type,sub,val);
|
|
std::vector<SComponent*> weko;
|
|
weko.push_back(com);
|
|
cb->giveResource(cb->getHeroOwner(heroID),sub,val);
|
|
cb->showInfoDialog(cb->getHeroOwner(heroID),CGI->objh->advobtxt[mid],&weko);
|
|
visited[os] = true;
|
|
}
|
|
}
|
|
void CVisitableOPW::newTurn ()
|
|
{
|
|
if (cb->getDate(1)==1)
|
|
{
|
|
for (std::map<CGObjectInstance*,bool>::iterator i = visited.begin(); i != visited.end(); i++)
|
|
{
|
|
(*i).second = false;
|
|
}
|
|
}
|
|
}
|
|
void CVisitableOPW::newObject(CGObjectInstance *os)
|
|
{
|
|
visited.insert(std::pair<CGObjectInstance*,bool>(os,false));
|
|
}
|
|
|
|
void CVisitableOPW::onHeroVisit(CGObjectInstance *os, int heroID)
|
|
{
|
|
if(visited[os])
|
|
onNAHeroVisit(os,heroID,true);
|
|
else
|
|
onNAHeroVisit(os,heroID,false);
|
|
}
|
|
|
|
std::vector<int> CVisitableOPW::yourObjects() //returns IDs of objects which are handled by script
|
|
{
|
|
std::vector<int> ret(3);
|
|
ret.push_back(55); //mystical garden
|
|
ret.push_back(112); //windmill
|
|
ret.push_back(109); //water wheel
|
|
return ret;
|
|
}
|
|
|
|
std::string CVisitableOPW::hoverText(CGObjectInstance *os)
|
|
{
|
|
return CGI->objh->objects[os->defInfo->id].name + " " + ( (visited[os]) ? (CGI->generaltexth->allTexts[352]) : (CGI->generaltexth->allTexts[353])) ;
|
|
}
|
|
|
|
void CMines::newObject(CGObjectInstance *os)
|
|
{
|
|
ourObjs.push_back(os);
|
|
os->tempOwner = NEUTRAL_PLAYER;
|
|
}
|
|
void CMines::onHeroVisit(CGObjectInstance *os, int heroID)
|
|
{
|
|
int vv = 1;
|
|
if (os->subID==0 || os->subID==2)
|
|
vv++;
|
|
else if (os->subID==6)
|
|
vv = 1000;
|
|
if (os->tempOwner == cb->getHeroOwner(heroID))
|
|
{
|
|
//TODO: garrison
|
|
}
|
|
else
|
|
{
|
|
if (os->subID==7)
|
|
return; //TODO: support for abandoned mine
|
|
os->tempOwner = cb->getHeroOwner(heroID);
|
|
SComponent * com = new SComponent(SComponent::Etype::resource,os->subID,vv);
|
|
com->subtitle+=CGI->generaltexth->allTexts[3].substr(2,CGI->generaltexth->allTexts[3].length()-2);
|
|
std::vector<SComponent*> weko;
|
|
weko.push_back(com);
|
|
cb->showInfoDialog(cb->getHeroOwner(heroID),CGI->objh->mines[os->subID].second,&weko);
|
|
}
|
|
}
|
|
std::vector<int> CMines::yourObjects()
|
|
{
|
|
std::vector<int> ret(1);
|
|
ret.push_back(53);
|
|
return ret;
|
|
}
|
|
std::string CMines::hoverText(CGObjectInstance *os)
|
|
{
|
|
if (os->tempOwner == NEUTRAL_PLAYER)
|
|
return CGI->objh->mines[os->subID].first;
|
|
else
|
|
return CGI->objh->mines[os->subID].first + " " + CGI->generaltexth->arraytxt[23+os->tempOwner];
|
|
|
|
}
|
|
void CMines::newTurn ()
|
|
{
|
|
for (int i=0;i<ourObjs.size();i++)
|
|
{
|
|
if (ourObjs[i]->tempOwner == NEUTRAL_PLAYER)
|
|
continue;
|
|
int vv = 1;
|
|
if (ourObjs[i]->subID==0 || ourObjs[i]->subID==2)
|
|
vv++;
|
|
else if (ourObjs[i]->subID==6)
|
|
vv = 1000;
|
|
cb->giveResource(ourObjs[i]->tempOwner,ourObjs[i]->subID,vv);
|
|
}
|
|
}
|
|
|
|
|
|
void CPickable::newObject(CGObjectInstance *os)
|
|
{
|
|
os->blockVisit = true;
|
|
}
|
|
void CPickable::onHeroVisit(CGObjectInstance *os, int heroID)
|
|
{
|
|
switch(os->ID)
|
|
{
|
|
case 79:
|
|
{
|
|
//TODO: handle guards (when battles are finished)
|
|
CResourceObjInfo * t2 = static_cast<CResourceObjInfo *>(os->info);
|
|
int val;
|
|
if(t2->amount)
|
|
val = t2->amount;
|
|
else
|
|
{
|
|
switch(os->subID)
|
|
{
|
|
case 6:
|
|
val = 500 + (rand()%6)*100;
|
|
break;
|
|
case 0: case 2:
|
|
val = 6 + (rand()%5);
|
|
break;
|
|
default:
|
|
val = 3 + (rand()%3);
|
|
break;
|
|
}
|
|
}
|
|
if(t2->message.length())
|
|
cb->showInfoDialog(cb->getHeroOwner(heroID),t2->message,&std::vector<SComponent*>());
|
|
SComponent ccc(SComponent::resource,os->subID,val);
|
|
ccc.description = CGI->objh->advobtxt[113];
|
|
boost::algorithm::replace_first(ccc.description,"%s",CGI->objh->restypes[os->subID]);
|
|
cb->giveResource(cb->getHeroOwner(heroID),os->subID,val);
|
|
cb->showCompInfo(cb->getHeroOwner(heroID),&ccc);
|
|
break;
|
|
}
|
|
case 101:
|
|
{
|
|
if (os->subID)
|
|
break; //not OH3 treasure chest
|
|
int wyn = rand()%100;
|
|
if (wyn<32)
|
|
{
|
|
tempStore.push_back(new CSelectableComponent(SComponent::resource,6,1000));
|
|
tempStore.push_back(new CSelectableComponent(SComponent::experience,0,500));
|
|
}//1k/0.5k
|
|
else if(wyn<64)
|
|
{
|
|
tempStore.push_back(new CSelectableComponent(SComponent::resource,6,1500));
|
|
tempStore.push_back(new CSelectableComponent(SComponent::experience,0,1000));
|
|
}//1.5k/1k
|
|
else if(wyn<95)
|
|
{
|
|
tempStore.push_back(new CSelectableComponent(SComponent::resource,6,2000));
|
|
tempStore.push_back(new CSelectableComponent(SComponent::experience,0,1500));
|
|
}//2k/1.5k
|
|
else
|
|
{
|
|
if (1/*TODO: backpack is full*/)
|
|
{
|
|
tempStore.push_back(new CSelectableComponent(SComponent::resource,6,1000));
|
|
tempStore.push_back(new CSelectableComponent(SComponent::experience,0,500));
|
|
}
|
|
else
|
|
{
|
|
//TODO: give treasure artifact
|
|
break;
|
|
}
|
|
}//random treasure artifact, or (if backapack is full) 1k/0.5k
|
|
tempStore[1]->ID = heroID;
|
|
player = cb->getHeroOwner(heroID);
|
|
cb->showSelDialog(player,CGI->objh->advobtxt[146],&tempStore,this);
|
|
break;
|
|
}
|
|
}
|
|
CGI->mh->removeObject(os);
|
|
}
|
|
void CPickable::chosen(int which)
|
|
{
|
|
switch(tempStore[which]->type)
|
|
{
|
|
case SComponent::resource:
|
|
cb->giveResource(player,tempStore[which]->subtype,tempStore[which]->val);
|
|
break;
|
|
case SComponent::experience:
|
|
cb->changePrimSkill(tempStore[which]->ID,4,tempStore[which]->val);
|
|
break;
|
|
default:
|
|
throw new std::exception("Unhandled choice");
|
|
|
|
}
|
|
for (int i=0;i<tempStore.size();i++)
|
|
delete tempStore[i];
|
|
tempStore.clear();
|
|
}
|
|
|
|
std::string CPickable::hoverText(CGObjectInstance *os)
|
|
{
|
|
switch (os->ID)
|
|
{
|
|
case 79:
|
|
return CGI->objh->restypes[os->subID];
|
|
break;
|
|
case 5:
|
|
return CGI->arth->artifacts[os->subID].name;
|
|
break;
|
|
default:
|
|
return CGI->objh->objects[os->defInfo->id].name;
|
|
break;
|
|
}
|
|
}
|
|
std::vector<int> CPickable::yourObjects() //returns IDs of objects which are handled by script
|
|
{
|
|
std::vector<int> ret(3);
|
|
ret.push_back(79); //resource
|
|
ret.push_back(5); //artifact
|
|
ret.push_back(101); //treasure chest / commander stone
|
|
return ret;
|
|
} |