/*=============================================================================
    Name    : GravWellGenerator.c
    Purpose : Specifics for the Gravity Well Generator

    Created 01/06/1998 by bpasechn
    Copyright Relic Entertainment, Inc.  All rights reserved.
=============================================================================*/

#include <string.h>
#include "types.h"
#include "spaceobj.h"
#include "GravWellGenerator.h"
#include "SoundEvent.h"
#include "vector.h"
#include "universe.h"
#include "aitrack.h"
#include "select.h"
#include "SaveGame.h"
#include "madlinkin.h"
#include "univupdate.h"
#include "attack.h"
#include "randy.h"
#include "battle.h"
#include "fastmath.h"
#include "blobs.h"
#include "etg.h"

GravWellGeneratorStatics GravWellGeneratorStatic;

GravWellGeneratorStatics GravWellGeneratorStaticRace1;
GravWellGeneratorStatics GravWellGeneratorStaticRace2;

//after effect is spawned, wait between these two numbers before re-spawinging
#define gravwellEffectLifeLow       3.0f
#define gravwellEffectLifeHigh      5.0f

#define GRAV_TIME_BEFORE_EXPLODE    3.5f

#define GRAV_BEGIN      0
#define GRAV_STOPPED    2
#define GRAV_STAGE2     1

scriptStructEntry GravWellGeneratorStaticScriptTable[] =
{
    { "GravWellRadius",    scriptSetReal32CB, (udword) &(GravWellGeneratorStatic.GravWellRadius), (udword) &(GravWellGeneratorStatic) },
    { "OperationTime",    scriptSetReal32CB, (udword) &(GravWellGeneratorStatic.OperationTime), (udword) &(GravWellGeneratorStatic) },
    { "EffectConstant",    scriptSetReal32CB, (udword) &(GravWellGeneratorStatic.EffectConstant), (udword) &(GravWellGeneratorStatic) },
    { "scanrate",    scriptSetUdwordCB, (udword) &(GravWellGeneratorStatic.scanrate), (udword) &(GravWellGeneratorStatic) },
    { "repulseForce",    scriptSetReal32CB, (udword) &(GravWellGeneratorStatic.repulseForce), (udword) &(GravWellGeneratorStatic) },
    { "warmupdowntime",    scriptSetReal32CB, (udword) &(GravWellGeneratorStatic.warmupdowntime), (udword) &(GravWellGeneratorStatic) },
    { "xrot",    scriptSetReal32CB, (udword) &(GravWellGeneratorStatic.xrot), (udword) &(GravWellGeneratorStatic) },
    { "yrot",    scriptSetReal32CB, (udword) &(GravWellGeneratorStatic.yrot), (udword) &(GravWellGeneratorStatic) },
    { "zrot",    scriptSetReal32CB, (udword) &(GravWellGeneratorStatic.zrot), (udword) &(GravWellGeneratorStatic) },



    { NULL,NULL,0,0 }
};

void GravWellGeneratorStaticInit(char *directory,char *filename,struct ShipStaticInfo *statinfo)
{
    GravWellGeneratorStatics *GravWellGeneratorstat = (statinfo->shiprace == R1) ? &GravWellGeneratorStaticRace1 : &GravWellGeneratorStaticRace2;

    statinfo->custstatinfo = GravWellGeneratorstat;
    scriptSetStruct(directory,filename,GravWellGeneratorStaticScriptTable,(ubyte *)GravWellGeneratorstat);
    GravWellGeneratorstat->GravWellRadiusSqr = GravWellGeneratorstat->GravWellRadius*GravWellGeneratorstat->GravWellRadius;
}

void GravWellGeneratorInit(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    spec->GravFieldOn = FALSE;
    spec->GravFired = FALSE;
    spec->ready = TRUE;
    spec->TimeOn = 0.0f;
    spec->GravDerelict = FALSE;
    spec->gravityEffect = NULL;
    listInit(&spec->GravList);
}

//ship is the ship to have the gravwell effect played around
void runShipEffect(Ship *ship)
{
    sdword LOD;
    etglod *etgLOD;
    etgeffectstatic *stat;

    etgLOD = etgSpecialPurposeEffectTable[EGT_CAUGHT_GRAVWELL];
     if (etgLOD != NULL)
     {
         LOD = ship->currentLOD;
         if (LOD >= etgLOD->nLevels)
         {
             stat = NULL;
         }
         else
         {
             stat = etgLOD->level[LOD];
         }
     }
     else
     {
         stat = NULL;
     }
     if (univSpaceObjInRenderList((SpaceObj *)ship))
     {
    #if ETG_DISABLEABLE
         if (stat != NULL && etgEffectsEnabled && !etgFrequencyExceeded(stat))
    #else
         if (stat != NULL)
    #endif
         {
             udword colSizeDword;
             dbgAssert(ship->objtype == OBJ_ShipType);
             colSizeDword = TreatAsUdword(((ShipStaticInfo *)(ship->staticinfo))->staticheader.staticCollInfo.collspheresize);
             etgEffectCreate(stat, ship, &ship->posinfo.position, &ship->posinfo.velocity, &ship->rotinfo.coordsys, ship->magnitudeSquared, EAF_Full, 1, colSizeDword);
			 {
				real32 realtemp;
				udword temp;
				temp = etgFRandom(gravwellEffectLifeLow,gravwellEffectLifeHigh);
				realtemp = *((real32 *)(&temp));
				ship->gravwellTimeEffect = universe.totaltimeelapsed + realtemp;
			 }
         }
     }

}
void GravWellGeneratorDied(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    Node *gravnode;
    GravStruct *gravstruct;

    gravnode = spec->GravList.head;

    while(gravnode != NULL)
    {
       gravstruct = (GravStruct *)listGetStructOfNode(gravnode);
       //do maintenance here if any needed, otherwise remove loop :)
       bitClear(gravstruct->ship->dontapplyforceever,1);
       bitClear(gravstruct->ship->dontrotateever,1);
       bitClear(gravstruct->ship->specialFlags2,SPECIAL_2_ShipInGravwell);
       gravnode = gravnode->next;
    }
    listDeleteAll(&spec->GravList);        //delete the list
}

bool gravwellIsShipStuckForHyperspaceing(Ship *ship)
{
    //multiplayer game...lets test if in grav well..if we are, we'll pause and let us get pummeled!
    Ship *posShip;
    sdword i;
    for(i=0;i<ship->collMyBlob->blobShips->numShips;i++)
    {
        posShip = ship->collMyBlob->blobShips->ShipPtr[i];
        if(posShip->shiptype == GravWellGenerator)
        {
            //SHIP IS A GRAV WELL!
            if(((GravWellGeneratorSpec *)(posShip->ShipSpecifics))->GravFieldOn)
            {
                //GRAV WELL IS ON!
                //probably in it, but lets see...
                vector sep;
                real32 distsqr;
                vecSub(sep,ship->posinfo.position,posShip->posinfo.position);
                distsqr = vecMagnitudeSquared(sep);
                if(distsqr < (((GravWellGeneratorStatics *)posShip->staticinfo->custstatinfo)->GravWellRadiusSqr))
                {
                    //we are inside it...so don't update this ships HSing!
                    return TRUE;
                }
            }
        }
    }
    return FALSE;
}



void turnoffGravwell(Ship *ship)
{
    //turning the gravwell generator off
    /////////////////////////
    //Speech Event here:
    //event num: COMM_Grav_Off
    //use battle chatter
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    sdword LOD;
    etglod *etgLOD;
    etgeffectstatic *stat;
    if(ship->playerowner->playerIndex == universe.curPlayerIndex)
    {
        if (battleCanChatterAtThisTime(BCE_COMM_Grav_Off, ship))
        {
            battleChatterAttempt(SOUND_EVENT_DEFAULT, BCE_COMM_Grav_Off, ship, SOUND_EVENT_DEFAULT);
        }
    }


    //////////////////////////

    madLinkInCloseSpecialShip(ship);
    spec->GravFieldOn = FALSE;
    GravWellGeneratorDied(ship); //not literal...just cleans up linked list of ships
    //make the continuous effect time out by modifying the duration parameter directly
    if (spec->gravityEffect != NULL)
    {                                               //if there is a continuous effect playing here
        ((real32 *)spec->gravityEffect->variable)[ETG_SpecialDurationParam] =
            spec->gravityEffect->timeElapsed;       //time the effect out
        spec->gravityEffect = NULL;
    }
    //spawn a gravwell off effect
    etgLOD = etgSpecialPurposeEffectTable[EGT_GRAVWELL_OFF];
    if (etgLOD != NULL)
    {
        LOD = ship->currentLOD;
        if (LOD >= etgLOD->nLevels)
        {
            stat = NULL;
        }
        else
        {
            stat = etgLOD->level[LOD];
        }
    }
    else
    {
        stat = NULL;
    }
    if (univSpaceObjInRenderList((SpaceObj *)ship))
    {
#if ETG_DISABLEABLE
        if (stat != NULL && etgEffectsEnabled)
#else
        if (stat != NULL)
#endif
        {                                       //if there is a gravwell off effect for this LOD
            dbgAssert(ship->objtype == OBJ_ShipType);
            etgEffectCreate(stat, ship, &ship->posinfo.position, NULL, &ship->rotinfo.coordsys, ship->magnitudeSquared, EAF_Full, 1, ship->shiprace);
        }
    }
}
bool GravWellGeneratorSpecialActivate(Ship *ship)
{
    sdword LOD;
    etglod *etgLOD;
    etgeffectstatic *stat;
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    GravWellGeneratorStatics *gravwellgeneratorstatics;
    gravwellgeneratorstatics = (GravWellGeneratorStatics *) ((ShipStaticInfo *)(ship->staticinfo))->custstatinfo;

    //SelectCommand selection;
    if (spec->GravFired == FALSE)
    {   //Grav Generator hasn't expired yet
        spec->ready = FALSE;
        if(spec->GravFieldOn)
        {
            turnoffGravwell(ship);

        }
        else
        {                                                   //else turning it on
            //selection.numShips = 1;
            //selection.ShipPtr[0] = ship;
            //clHalt(&universe.mainCommandLayer,&selection);
            if(spec->TimeOn < gravwellgeneratorstatics->OperationTime)
            {
                madLinkInOpenSpecialShip(ship);
                speechEvent(ship, COMM_Grav_On, 0);
                spec->GravFieldOn = TRUE;
                //create the looping "gravity on" effect
                etgLOD = etgSpecialPurposeEffectTable[EGT_GRAVWELL_ON];
                if (etgLOD != NULL)
                {
                    LOD = ship->currentLOD;
                    if (LOD >= etgLOD->nLevels)
                    {
                        stat = NULL;
                    }
                    else
                    {
                        stat = etgLOD->level[LOD];
                    }
                }
                else
                {
                    stat = NULL;
                }
                if (univSpaceObjInRenderList((SpaceObj *)ship))
                {
#if ETG_DISABLEABLE
                    if (stat != NULL && etgEffectsEnabled)
#else
                    if (stat != NULL)
#endif
                    {                                       //if there is a gravwell on effect
                        dbgAssert(ship->objtype == OBJ_ShipType);
                        spec->gravityEffect = etgEffectCreate(stat, ship, &ship->posinfo.position, NULL, &ship->rotinfo.coordsys, ship->magnitudeSquared, EAF_Full, 1, ship->shiprace);
                        if (((real32 *)spec->gravityEffect->variable)[ETG_SpecialDurationParam] != 999999.0f)
                        {                                   //if it's not one of those free-running effects
                            spec->gravityEffect = NULL;     //don't try to time it out later
                        }
                    }
                }
            }
        }
    }

    return TRUE;
}

void GravWellGeneratorJustDisabled(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;

    if(spec->GravFieldOn)
    {                                                   //turning the gravwell generator off
        GravWellGeneratorSpecialActivate(ship);
    }
}

bool ShipInGravField(Ship *ship, Ship *objinquestion)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    Node *gravnode;
    GravStruct *gravstruct;

    gravnode = spec->GravList.head;

    while(gravnode != NULL)
    {
        gravstruct = (GravStruct *)listGetStructOfNode(gravnode);
        if (gravstruct->ship == objinquestion)
        {     //if ship is already in list
            return(FALSE);             //
        }

        gravnode = gravnode->next;
    }
    return(TRUE);
}

void GravWellGeneratorAddObj(Ship *ship,  Ship *objtoadd)
{                       //Adds shiptoadd to 'ship's cloaklist
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;  //maybe don't need
    GravStruct *newgravstruct;
    GravWellGeneratorStatics *gravwellgeneratorstatics;


    gravwellgeneratorstatics = (GravWellGeneratorStatics *) ((ShipStaticInfo *)(ship->staticinfo))->custstatinfo;

    if(bitTest(objtoadd->flags,SOF_Dead))
    {
        return;     //don't add dead objects...
    }

    //Need to check if already in list!!
    if (!ShipInGravField(ship,objtoadd))
    {   //Ship is already in field, so don't add
        //re set variable in case it was claered b4.
        bitSet(objtoadd->specialFlags2,SPECIAL_2_ShipInGravwell);
        return;
    }

    //////////////////////
    //speech event ships stuck in gravwell
    //need to definatly limit with battle chatter!
    //event define:   STAT_Strike_StuckInGrav
    if(objtoadd->playerowner->playerIndex == universe.curPlayerIndex)
    {
        if (battleCanChatterAtThisTime(BCE_STAT_Strike_StuckInGrav, objtoadd))
        {
            battleChatterAttempt(SOUND_EVENT_DEFAULT, BCE_STAT_Strike_StuckInGrav, objtoadd, SOUND_EVENT_DEFAULT);
        }
    }

    ////////////////////////

    newgravstruct = memAlloc(sizeof(GravStruct),"GravStruct",0);

    newgravstruct->ship = objtoadd;
    newgravstruct->stoppingstate = GRAV_BEGIN;
    newgravstruct->xangle = frandombetween(-gravwellgeneratorstatics->xrot,gravwellgeneratorstatics->xrot);
    newgravstruct->yangle = frandombetween(-gravwellgeneratorstatics->yrot,gravwellgeneratorstatics->yrot);
    newgravstruct->zangle = frandombetween(-gravwellgeneratorstatics->zrot,gravwellgeneratorstatics->zrot);
    listAddNode(&spec->GravList,&newgravstruct->objnode,newgravstruct);


    bitSet(objtoadd->specialFlags2,SPECIAL_2_ShipInGravwell);

}

void GravAddObjectsInProximity(Ship *gravship)
{
    Node *objnode = universe.ShipList.head;
    udword num = universe.ShipList.num;
    GravWellGeneratorStatics *gravwellgeneratorstatics;
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)gravship->ShipSpecifics;  //maybe don't need
    Ship *ship;

    real32 distanceSqr;
    vector diff;

    gravwellgeneratorstatics = (GravWellGeneratorStatics *) ((ShipStaticInfo *)(gravship->staticinfo))->custstatinfo;

    while (objnode != NULL)
    {
        ship = (Ship *)listGetStructOfNode(objnode);

        if(ship->staticinfo->shipclass == CLASS_Corvette ||
           ship->staticinfo->shipclass == CLASS_Fighter)
        {
            if(ship->shiptype == SalCapCorvette)
                goto nextnode;
            //later, do a cubular check instead?
            vecSub(diff,ship->posinfo.position,gravship->posinfo.position);
            distanceSqr = diff.x*diff.x + diff.y*diff.y + diff.z*diff.z;
            if(distanceSqr <= gravwellgeneratorstatics->GravWellRadiusSqr)
            {
                GravWellGeneratorAddObj(gravship,  ship);
            }
        }
nextnode:
        objnode = objnode->next;
    }
}

void GravWellGeneratorRemoveShipReferences(Ship *ship,Ship *shiptoremove)
{
  GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
  Node *gravnode;
  GravStruct *gravstruct;

  gravnode = spec->GravList.head;

  while(gravnode != NULL)
  {
    gravstruct = (GravStruct *)listGetStructOfNode(gravnode);
    if (gravstruct->ship == shiptoremove)
    {
        bitClear(gravstruct->ship->dontapplyforceever,1);
        bitClear(gravstruct->ship->dontrotateever,1);
        bitClear(shiptoremove->specialFlags2,SPECIAL_2_ShipInGravwell);
        listDeleteNode(gravnode);     //delete nodes memory...may not work as I think...ask gary..
        break;             //done so break and return
    }
    gravnode = gravnode->next;
  }
}

void GravWellClose(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    listDeleteAll(&spec->GravList);        //delete the list
}

#define speedlimit  0.7f
void GravWellGeneratorHouseKeep(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    GravWellGeneratorStatics *gravwellgeneratorstatics;
    Node *gravnode;
    GravStruct *gravstruct;
    vector diff;
    real32 distanceSqr;
    real32 distance,oneOverDistance;
    vector gravToShip;
    sdword LOD;
    etglod *etgLOD;
    etgeffectstatic *stat;

    //don't do any processing if a derelict now
    if(spec->GravDerelict == TRUE)
    {
        //may need to clear selectionbit each frame,
        //we'll see...
        return;
    }

    if(spec->GravFieldOn == TRUE)
    {
        gravwellgeneratorstatics = (GravWellGeneratorStatics *) ((ShipStaticInfo *)(ship->staticinfo))->custstatinfo;
        spec->ready = TRUE;
        spec->TimeOn += universe.phystimeelapsed;
        if(spec->TimeOn >= gravwellgeneratorstatics->OperationTime)
        {   //Grav Generator has used up its time so shut it down
            speechEvent(ship, STAT_Grav_Collapse, 0);
            GravWellGeneratorDied(ship);
            spec->GravFieldOn = FALSE;
            spec->GravDerelict = TRUE;
            ship->deathtime= universe.totaltimeelapsed+GRAV_TIME_BEFORE_EXPLODE;

            //derelictize ship
            bitSet(ship->flags,SOF_Disabled);
            bitClear(ship->flags,SOF_Selectable);
            clRemoveShipFromSelection(&selSelected,ship);
            //make the effect time out by modifying the duration parameter directly
            if (spec->gravityEffect != NULL)
            {
                ((real32 *)spec->gravityEffect->variable)[ETG_SpecialDurationParam] =
                    spec->gravityEffect->timeElapsed;       //time the effect out
                spec->gravityEffect = NULL;
            }
        }
        else if ((universe.univUpdateCounter & gravwellgeneratorstatics->scanrate) == gravwellgeneratorstatics->scanrate)
        {       //15th frame..so do a search on other ships withing 'proximity'
            if (spec->gravityEffect == NULL)
            {
                //GRAVWELL WASN'T ON WHEN NEARBY...LETS TURN IT ON!
etgLOD = etgSpecialPurposeEffectTable[EGT_GRAVWELL_ON];
                if (etgLOD != NULL)
                {
                    LOD = ship->currentLOD;
                    if (LOD >= etgLOD->nLevels)
                    {
                        stat = NULL;
                    }
                    else
                    {
                        stat = etgLOD->level[LOD];
                    }
                }
                else
                {
                    stat = NULL;
                }
                if (univSpaceObjInRenderList((SpaceObj *)ship))
                {
#if ETG_DISABLEABLE
                    if (stat != NULL && etgEffectsEnabled)
#else
                    if (stat != NULL)
#endif
                    {                                       //if there is a gravwell on effect
                        dbgAssert(ship->objtype == OBJ_ShipType);
                        spec->gravityEffect = etgEffectCreate(stat, ship, &ship->posinfo.position, NULL, &ship->rotinfo.coordsys, ship->magnitudeSquared, EAF_Full, 1, ship->shiprace);
                        if (((real32 *)spec->gravityEffect->variable)[ETG_SpecialDurationParam] != 999999.0f)
                        {                                   //if it's not one of those free-running effects
                            spec->gravityEffect = NULL;     //don't try to time it out later
                        }
                    }
                }

            }
 
			GravAddObjectsInProximity(ship);
        }

        gravnode = spec->GravList.head;

        while(gravnode != NULL)
        {
            gravstruct = (GravStruct *)listGetStructOfNode(gravnode);

            vecSub(diff,gravstruct->ship->posinfo.position,ship->posinfo.position);
            distanceSqr = diff.x*diff.x + diff.y*diff.y + diff.z*diff.z;
            gravToShip = diff;
            if(distanceSqr >= gravwellgeneratorstatics->GravWellRadiusSqr)
            {
                gravnode = gravnode->next;
                GravWellGeneratorRemoveShipReferences(ship,  (Ship *) gravstruct->ship);
                continue;
            }
            if(bitTest(gravstruct->ship->flags,SOF_Dead))
            {
                gravnode = gravnode->next;
                GravWellGeneratorRemoveShipReferences(ship,  (Ship *) gravstruct->ship);
                continue;
            }
            //play gravwell effect
            if(universe.totaltimeelapsed > gravstruct->ship->gravwellTimeEffect)
            {
                //time to run the effect again
                runShipEffect((Ship *)gravstruct->ship);
            }


            bitSet(gravstruct->ship->dontapplyforceever,1);
            switch(gravstruct->stoppingstate)
            {
            case GRAV_BEGIN:
                vecScalarMultiply(gravstruct->ship->posinfo.velocity,gravstruct->ship->posinfo.velocity,gravwellgeneratorstatics->EffectConstant);
                if(vecMagnitudeSquared(gravstruct->ship->posinfo.velocity) < 2500)
                {
                    gravstruct->stoppingstate = GRAV_STAGE2;
                }
                break;
            case GRAV_STAGE2:
                gravstruct->ship->autostabilizeship = FALSE;
                if(gravstruct->ship->rotinfo.torque.x == 0.0 &&
                    gravstruct->ship->rotinfo.torque.y == 0.0 &&
                    gravstruct->ship->rotinfo.torque.z == 0.0)
                {
                    if(gravstruct->ship->rotinfo.rotspeed.x < speedlimit &&
                        gravstruct->ship->rotinfo.rotspeed.x > -speedlimit)
                    {
                        gravstruct->ship->rotinfo.torque.x = gravstruct->xangle;
                    }
                    if(gravstruct->ship->rotinfo.rotspeed.y < speedlimit &&
                        gravstruct->ship->rotinfo.rotspeed.y > -speedlimit)
                    {
                        gravstruct->ship->rotinfo.torque.y = gravstruct->yangle;
                    }
                    if(gravstruct->ship->rotinfo.rotspeed.z < speedlimit &&
                        gravstruct->ship->rotinfo.rotspeed.z > -speedlimit)
                    {
                        gravstruct->ship->rotinfo.torque.z = gravstruct->zangle;
                    }

                }
                distance = fsqrt(distanceSqr);
                oneOverDistance = 1.0f/distance;
                gravToShip.x = gravToShip.x*oneOverDistance;
                gravToShip.y = gravToShip.y*oneOverDistance;
                gravToShip.z = gravToShip.z*oneOverDistance;
                vecScalarMultiply(gravstruct->ship->posinfo.force,gravToShip,gravwellgeneratorstatics->repulseForce);

                break;
            default:
                dbgFatalf(DBG_Loc,"Unknown gravwellgenerator stopping state %d",gravstruct->stoppingstate);
                break;
            }

            gravnode = gravnode->next;
        }
    }
}

#pragma warning( 4 : 4047)      // turns off "different levels of indirection warning"

void SaveGravStruct(GravStruct *gravStruct)
{
    SaveChunk *chunk;
    GravStruct *savecontents;

    chunk = CreateChunk(BASIC_STRUCTURE,sizeof(GravStruct),gravStruct);
    savecontents = (GravStruct *)chunkContents(chunk);

    savecontents->ship = (Ship *)SpaceObjRegistryGetID((SpaceObj *)savecontents->ship);

    SaveThisChunk(chunk);
    memFree(chunk);
}

void GravWellGenerator_Save(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    Node *node = spec->GravList.head;
    sdword cur = 0;

    SaveInfoNumber(spec->GravList.num);

    while (node != NULL)
    {
        cur++;
        SaveGravStruct((GravStruct *)listGetStructOfNode(node));
        node = node->next;
    }

    dbgAssert(cur == spec->GravList.num);
}

GravStruct *LoadGravStruct(void)
{
    SaveChunk *chunk;
    GravStruct *gravStruct;

    chunk = LoadNextChunk();
    VerifyChunk(chunk,BASIC_STRUCTURE,sizeof(GravStruct));

    gravStruct = memAlloc(sizeof(GravStruct),"GravStruct",0);
    memcpy(gravStruct,chunkContents(chunk),sizeof(GravStruct));

    memFree(chunk);

    return gravStruct;
}

void GravWellGenerator_Load(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    sdword num;
    sdword i;
    GravStruct *gravStruct;

    num = LoadInfoNumber();

    listInit(&spec->GravList);

    for (i=0;i<num;i++)
    {
        gravStruct = LoadGravStruct();
        listAddNode(&spec->GravList,&gravStruct->objnode,gravStruct);
    }
}

void GravWellGenerator_Fix(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    Node *node = spec->GravList.head;
    GravStruct *gravStruct;

    while (node != NULL)
    {
        gravStruct = (GravStruct *)listGetStructOfNode(node);

        gravStruct->ship = SpaceObjRegistryGetShip((sdword)gravStruct->ship);

        node = node->next;
    }
}

void GravWellGenerator_PreFix(Ship *ship)
{
    GravWellGeneratorSpec *spec = (GravWellGeneratorSpec *)ship->ShipSpecifics;
    spec->gravityEffect = NULL; //set to NULL!
}

#pragma warning( 2 : 4047)      // turn back on "different levels of indirection warning"

void GravWellAttack(Ship *ship,SpaceObjRotImpTarg *target,real32 maxdist)
{
    attackStraightForward(ship,target,4000.0f,3000.0f);
}

CustShipHeader GravWellGeneratorHeader =
{
    GravWellGenerator,
    sizeof(GravWellGeneratorSpec),
    GravWellGeneratorStaticInit,
    NULL,
    GravWellGeneratorInit,
    GravWellClose,
    GravWellAttack,
    NULL,
    NULL,
    GravWellGeneratorSpecialActivate,
    NULL,
    GravWellGeneratorHouseKeep,
    GravWellGeneratorRemoveShipReferences,
    GravWellGeneratorDied,
    GravWellGenerator_PreFix,
    GravWellGenerator_Save,
    GravWellGenerator_Load,
    GravWellGenerator_Fix
};