Skip to content

Instantly share code, notes, and snippets.

@dgant

dgant/Purple[Wavelet|Broodling|Essence]

Created November 2, 2018 00:25
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dgant/e37f8bd561228730717589033b2275d7 to your computer and use it in GitHub Desktop.
Save dgant/e37f8bd561228730717589033b2275d7 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Purple[Wavelet|Broodling|Essence]
PurpleWavelet (Protoss)
#include<BWAPI.h>
#include<cmath>
#define A auto
#define C continue;
#define G using
#define N G namespace
#define R return
#define PP Position
#define TP TilePosition
#define UP(u) u->getPosition()
#define MP(t) t.mineralPrice()
#define GT(u) u->getType()
#define I(u) u->getID()
#define J(u) !k[I(u)]
#define K(u) k[I(u)]=1
#define Z(t) Protoss_##t
N BWAPI;N UnitTypes;N Filter;N std;G I=int;G U=Unit;G UT=UnitType;vector<U>mi;U m,b,c;PP d;TP bd;I i,t,cW,cP,cB,cE,lW,lP,lB,lE;A&g=BroodwarPtr;
struct ExampleAIModule:AIModule{void onFrame(){
g->setLatCom(cW=cP=cB=cE=0);
if(++i%6!=2)R;
A*s=g->self();
A ou=s->getUnits();
A sl=s->getStartLocation();
I k[99999]={},ms=45+s->minerals(),su=0,sc=0+g->isVisible(TP(d));
U n=g->getClosestUnit(PP(sl),IsResourceDepot);
A np=n?UP(n):PP(sl);
if(!mi.size())for(U u:g->getUnitsInRadius(PP(sl),320,IsMineralField))mi.push_back(u);
std::sort(mi.begin(),mi.end(),[&](U x,U y){R np.getDistance(UP(x))<np.getDistance(UP(y));});
for(A u:ou)su+=GT(u).supplyProvided()-GT(u).supplyRequired();
A&st=g->getStartLocations();
d=(sc&&!lE||d==PP())?PP(st[(++t)%st.size()])+(i<1e5?PP():PP(i%994-497,(i*su)%994-497)):d;
bd=bd==TP()?sl:bd;
b=b&&b->exists()?b:g->getClosestUnit(PP(bd),IsOwned&&IsWorker);
c=c?c:b->getClosestUnit(IsWorker);
A B=[&](UT x){A r=(ms-=MP(x))>0;if(r){su+=x.supplyProvided();if(b){bd=g->getBuildLocation(x,TP(sl),100);J(b)?(g->isVisible(bd)&&ms>45)?b->build(x,bd):b->move(PP(bd)):0;K(b);}}R r;};
A T=[&](UT x,UT y){I r=ms>MP(x)&&su>0;if(r){A u=find_if(ou.begin(),ou.end(),[&](A z){R z->isIdle()&&GT(z)==y;});if(r=u!=ou.end()&&J((*u))&&(*u)->train(x)){K((*u));ms-=MP(x);su-=x.supplyRequired();};}R r;};
while(ms>0){
if(lW>7&&lP<1&&B(Z(Pylon)))C
if(lW>8&&lP>0&&lB<2&&B(Z(Gateway)))C
if(T(Z(Zealot),Z(Gateway)))C
if(lW<22&&T(Z(Probe),Z(Nexus)))C
if(su<2+4*lB&&B(Z(Pylon)))C
if(ms>250&&lB<5&&B(Z(Gateway)))C
break;
}
for(U u:g->getAllUnits()){
A p=u->getPlayer();
A t=GT(u);
A up=UP(u);
U tr=u->getOrderTarget();
tr=tr?tr:u->getTarget();
I ag=!!t.maxGroundHits(),
hp=u->getHitPoints(),
iW=t.isWorker(),
rw=1;
I iF=ag&&(!iW||(u==c&&lB>1)),cs=(iW?5.:100.)*hp/(1.+t.maxHitPoints());
m=m&&m->exists()?m:t.isMineralField()?u:m;
cE+=p->isEnemy(s);
if(p==s){
cW+=iW;cP+=t==Z(Pylon);cB+=t==Z(Gateway);
if(!J(u))C
if(iF)for(U n:u->getUnitsInRadius(512,CanAttack)){
A q=GT(n);A r=GT(n).groundWeapon().maxRange();
rw=rw&&r<32;
cs+=(n->getPlayer()==s?1.:-1)
*u->isCompleted()
*(u->getDistance(n)-r<320)
*(q.gasPrice()+MP(q)+(q==Zerg_Sunken_Colony?150:0))/(1+q.isTwoUnitsInOneEgg())/(1+10*q.isWorker())
*n->getHitPoints()/(1.+q.maxHitPoints());
}
if((cs<0||(rw&&hp+u->getShields()<16))&&up.getDistance(PP(sl))>224&&u->move(PP(sl)))C
#define TR u->getClosestUnit(IsEnemy&&!IsCloaked&&!IsFlying
U v=(v=TR&&CanAttack,iF?1e3:99))?v:TR,iW?0:1e3);
if(ag&&v){tr==v||(u->isAttacking()&&tr&&tr->canAttack())||u->attack(UP(v));C}
iF?u->attack(d):u->isGatheringMinerals()||u->gather(mi.size()?mi[0]:m);
if(mi.size())mi.erase(mi.begin());
}}
lW=cW;lP=cP;lB=cB;lE=cE;
}};
---
PurpleBroodling (Zerg)
#include<BWAPI.h>
#define A auto
#define C continue;
#define G using
#define N G namespace
#define R return
#define Y Position
#define X TilePosition
#define D(u) u->getPosition()
#define L(t) t.mineralPrice()
#define V(u) u->getType()
#define I(u) u->getID()
#define J(u) !k[I(u)]
#define K(u) k[I(u)]=1
#define Z(t) Zerg_##t
#define O u->getClosestUnit(IsEnemy&&!IsCloaked&&!IsFlying
N BWAPI;N UnitTypes;N Filter;N std;G I=int;G U=Unit;G UT=UnitType;vector<U>mi;U m,b=0,c;Y d;X bd;I i,t,cW,cF,cP,cB,cE,lW,lF,lP,lB,lE,go;A&g=BroodwarPtr;
struct ExampleAIModule:AIModule{void onFrame(){g->setLatCom(cW=cF=cP=cB=cE=0);if(++i%6!=2)R;A*s=g->self();A ou=s->getUnits();A sl=s->getStartLocation();I k[99999]={},ms=24+s->minerals(),sc=0+g->isVisible(X(d));U n=g->getClosestUnit(Y(sl),IsResourceDepot);A np=n?D(n):Y(sl);go=go||(ms>124&&lB&&lW>3);if(!mi.size())for(U u:g->getUnitsInRadius(Y(sl),320,IsMineralField))mi.push_back(u);std::sort(mi.begin(),mi.end(),[&](U x,U y){R np.getDistance(D(x))<np.getDistance(D(y));});A&st=g->getStartLocations();d=(sc&&!lE||d==Y())?Y(st[(++t)%st.size()])+(i<1e5?Y():Y(i%994-497,(i*i)%994-497)):d;bd=bd==X()?sl:bd;A B=[&](UT x){A r=(ms-=L(x))>0;if(r){if(b){bd=i%11>0?bd:g->getBuildLocation(x,b?X(D(b)):X(sl),100);J(b)?(g->isVisible(bd)&&ms>24)?b->build(x,bd):b->move(Y(bd)):0;K(b);}}R r;};A T=[&](UT x,UT y){I r=ms>L(x);if(r){A u=find_if(ou.begin(),ou.end(),[&](A z){R V(z)==y;});if(r=u!=ou.end()&&J((*u))&&(*u)->morph(x)){K((*u));ms-=L(x);};}R r;};while(ms>0){if(lB<1&&B(Z(Spawning_Pool)))C;if(T(Z(Zergling),Z(Hatchery)))C;if(lW<4&&T(Z(Drone),Z(Larva)))C;break;}for(U u:g->getAllUnits()){A p=u->getPlayer();A t=V(u);A up=D(u);U tr=u->getOrderTarget();tr=tr?tr:u->getTarget();I ag=!!t.maxGroundHits(),hp=u->getHitPoints(),iW=t.isWorker(),rw=1;I iF=(ag&&(!iW||(u==c&&lB>0&&lW>3&&go))),cs=(iW?20.:50.)*hp/(1.+t.maxHitPoints());m=m&&m->exists()?m:t.isMineralField()?u:m;cE+=p->isEnemy(s);if(p==s){b=b&&b->exists()&&b!=c&&!b->getBuildUnit()?b:iW?u:b;c=c&&c!=b?c:iW?u:c;cW+=iW;cF+=ag&&!iW;cP+=t==Z(Overlord);cB+=t==Z(Spawning_Pool);if(!J(u))C
if(iF)for(U n:u->getUnitsInRadius(512,CanAttack)){A q=V(n);A r=V(n).groundWeapon().maxRange();rw=rw&&r<32;cs+=(n->getPlayer()==s?1.:-1)*u->isCompleted()*(u->getDistance(n)-r<320)*(q.gasPrice()+L(q)+(q==Zerg_Sunken_Colony?150:0))/(1+q.isTwoUnitsInOneEgg())/(1+10*q.isWorker())*n->getHitPoints()/(1.+q.maxHitPoints());}if((cs<0||(!iW&&u->getGroundWeaponCooldown()&&u->isUnderAttack()))&&up.getDistance(Y(sl))>224&&u->move(Y(sl)))C
U v=(v=O&&(IsWorker||CanAttack),iF?1e4:99))?v:O,iF?1e4:0);if(ag&&v){tr==v||(u->isAttacking()&&tr&&tr->canAttack())||u->attack(D(v));C}if(u->isFlying()){u->move(lE?Y(sl):d);C}iF?u->attack(d):u->isGatheringMinerals()||u->gather(mi.size()?mi[0]:m);if(mi.size())mi.erase(mi.begin());}}lW=cW;lF=cF;lP=cP;lB=cB;lE=cE;}};
---
PurpleEssence (Terran)
#include<BWAPI.h>
#include<cmath>
#include<set>
#define A auto
#define C continue;
#define G using
#define N G namespace
#define R return
#define PP Position
#define TP TilePosition
#define UP(u) u->getPosition()
#define MP(t) t.mineralPrice()
#define GT(u) u->getType()
#define I(u) u->getID()
#define J(u) !k[I(u)]
#define K(u) k[I(u)]=1
#define Z(t) Terran_##t
N BWAPI;N UnitTypes;N Filter;N std;G I=int;G U=Unit;G UT=UnitType;set<U>mr;vector<U>mi;U m,b=0,c;PP d;TP bd;I i,t,cW,cF,cP,cB,cE,lW,lF,lP,lB,lE,go;A&g=BroodwarPtr;
struct ExampleAIModule:AIModule{void onFrame(){
g->setLatCom(cW=cF=cP=cB=cE=0);
if(++i%6!=2)R;
A*s=g->self();A ou=s->getUnits();A sl=s->getStartLocation();I k[99999]={},ms=45+s->minerals(),sc=0+g->isVisible(TP(d));go=go||lW+lF>=16;U n=g->getClosestUnit(PP(sl),IsResourceDepot);A np=n?UP(n):PP(sl);
if(!mi.size())for(U u:g->getUnitsInRadius(PP(sl),320,IsMineralField))mi.push_back(u);
for(U u:ou)if(mr.size()<6&&GT(u).isWorker())mr.insert(u);
std::sort(mi.begin(),mi.end(),[&](U x,U y){R np.getDistance(UP(x))<np.getDistance(UP(y));});
A&st=g->getStartLocations();
d=(sc&&!lE||d==PP())?PP(st[(++t)%st.size()])+(i<1e5?PP():PP(i%994-497,(i*i)%994-497)):d;
bd=bd==TP()?sl:bd;
A B=[&](UT x){A r=(ms-=MP(x))>0;if(r){if(b){bd=i%23>0?bd:g->getBuildLocation(x,TP(sl),100);J(b)?(g->isVisible(bd)&&ms>45)?b->build(x,bd):b->move(PP(bd)):0;K(b);}}R r;};
A T=[&](UT x,UT y){I r=ms>MP(x);if(r){A u=find_if(ou.begin(),ou.end(),[&](A z){R z->isIdle()&&GT(z)==y;});if(r=u!=ou.end()&&J((*u))&&(*u)->train(x)){K((*u));ms-=MP(x);};}R r;};
while(ms>0){if(lW>7&&lB<2&&B(Z(Barracks)))C;if(lW>8&&lP<1&&B(Z(Supply_Depot)))C;if(T(Z(Marine),Z(Barracks)))C;if(T(Z(SCV),Z(Command_Center)))C;break;}
for(U u:g->getAllUnits()){
A p=u->getPlayer();A t=GT(u);A up=UP(u);U tr=u->getOrderTarget();tr=tr?tr:u->getTarget();I ag=!!t.maxGroundHits(),hp=u->getHitPoints(),iW=t.isWorker(),rw=1;
I iF=ag&&(!iW||(u==c&&lP>0||(go&&mr.find(u)==mr.end()))),cs=(iW?20.:50.)*hp/(1.+t.maxHitPoints());m=m&&m->exists()?m:t.isMineralField()?u:m;cE+=p->isEnemy(s);
if(p==s){b=b&&b->exists()&&b!=c&&!b->getBuildUnit()?b:iW?u:b;c=c&&c!=b?c:iW&&mr.find(u)==mr.end()?u:c;cW+=iW;cF+=ag&&!iW;cP+=t==Z(Supply_Depot);cB+=t==Z(Barracks);if(!J(u))C
if(iF)for(U n:u->getUnitsInRadius(512,CanAttack)){A q=GT(n);A r=GT(n).groundWeapon().maxRange();rw=rw&&r<32;
cs+=(n->getPlayer()==s?1.:-1)*u->isCompleted()*(u->getDistance(n)-r<320)*(q.gasPrice()+MP(q)+(q==Zerg_Sunken_Colony?150:0))/(1+q.isTwoUnitsInOneEgg())/(1+3*q.isWorker())*n->getHitPoints()/(1.+q.maxHitPoints());}
if((cs<0||(!iW&&u->getGroundWeaponCooldown()&&u->isUnderAttack()))&&up.getDistance(PP(sl))>224&&u->move(PP(sl)))C
#define TR u->getClosestUnit(IsEnemy&&!IsCloaked&&!IsFlying
U v=(v=TR&&(IsWorker||CanAttack),iF?1e4:99))?v:TR,iF?1e4:0);
if(ag&&v){tr==v||(u->isAttacking()&&tr&&tr->canAttack())||u->attack(UP(v));C}
iF?u->attack(d):u->isGatheringMinerals()||u->gather(mi.size()?mi[0]:m);
if(mi.size())mi.erase(mi.begin());
}}
lW=cW;lF=cF;lP=cP;lB=cB;lE=cE;
}};
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment