spacetac/src/core/Universe.ts

module TK.SpaceTac {
    /**
     * Main game universe
     */
    export class Universe {
        // List of star systems
        stars: Star[] = []

        // List of links between star systems
        starlinks: StarLink[] = []

        // Collection of all star locations
        locations = new RObjectContainer<StarLocation>()

        // Radius of the universe
        radius = 5

        // Source of randomness
        random = RandomGenerator.global

        /**
         * Add a single star
         */
        addStar(level = 1, name?: string, x = 0, y = 0): Star {
            let result = new Star(this, x, y, name || `Star ${this.stars.length + 1}`);
            result.level = level;
            this.stars.push(result);
            return result;
        }

        /**
         * Update the locations list
         */
        updateLocations(): void {
            this.locations = new RObjectContainer(flatten(this.stars.map(star => star.locations)));
        }

        /**
         * Generates a random universe, with star systems and locations of interest
         * 
         * This will also :
         * - create a network of jump links between star systems
         * - add random shops
         * - define a progressive gradient of enemy levels
         */
        generate(starcount = 50): void {
            if (starcount < 4) {
                starcount = 4;
            }

            // Links between stars
            while (this.stars.length == 0 || any(this.stars, star => star.getLinks().length == 0)) {
                this.stars = this.generateStars(starcount);

                let links = this.getPotentialLinks();
                this.starlinks = this.filterRedundantLinks(this.filterCrossingLinks(links));
            }
            this.generateWarpLocations();

            // Encounter levels
            this.setEncounterLevels();

            // Locations
            this.stars.forEach((star: Star) => {
                star.generate(this.random);
            });
            this.updateLocations();
            this.addShops();
        }

        // Generate a given number of stars, not too crowded
        generateStars(count: number): Star[] {
            var result: Star[] = [];

            var names = new NameGenerator(Star.NAMES_POOL);

            while (count) {
                var x = this.random.random() * this.radius * 2.0 - this.radius;
                var y = this.random.random() * this.radius * 2.0 - this.radius;
                var star = new Star(this, x, y);

                var nearest = this.getNearestTo(star, result);
                if (nearest && nearest.getDistanceTo(star) < this.radius * 0.1) {
                    continue;
                }

                star.name = names.getName() || "Star";
                result.push(star);

                count--;
            }

            return result;
        }

        // Get a list of potential links between the stars
        getPotentialLinks(): StarLink[] {
            var result: StarLink[] = [];

            this.stars.forEach((first: Star, idx1: number) => {
                this.stars.forEach((second: Star, idx2: number) => {
                    if (idx1 < idx2) {
                        if (first.getDistanceTo(second) < this.radius * 0.6) {
                            result.push(new StarLink(first, second));
                        }
                    }
                });
            });

            return result;
        }

        /**
         * Filter a list of potential links to avoid crossing ones.
         * 
         * Returned list of links should be free of crossings.
         * This should not alter the universe connectivity.
         */
        filterCrossingLinks(links: StarLink[]): StarLink[] {
            var result: StarLink[] = [];

            links.forEach((link1: StarLink) => {
                var crossed = false;
                links.forEach((link2: StarLink) => {
                    if (link1 !== link2 && link1.isCrossing(link2) && link1.getLength() >= link2.getLength()) {
                        crossed = true;
                    }
                });
                if (!crossed) {
                    result.push(link1);
                }
            });

            return result;
        }

        /**
         * Filter a list of potential links to remove redundant ones.
         * 
         * This will remove direct links that are also achieved by a similar two-jump couple.
         * This should not alter the universe connectivity.
         */
        filterRedundantLinks(links: StarLink[]): StarLink[] {
            let result: StarLink[] = [];

            links = sortedBy(links, link => link.getLength(), true);

            links.forEach(link => {
                let alternative_passages = intersection(
                    link.first.getNeighbors(links).filter(n => n != link.second),
                    link.second.getNeighbors(links).filter(n => n != link.first)
                );
                let alternative_lengths = alternative_passages.map(
                    passage => nn(link.first.getLinkTo(passage, links)).getLength() + nn(link.second.getLinkTo(passage, links)).getLength()
                );
                if (!any(alternative_lengths, length => length < link.getLength() * 1.2)) {
                    result.push(link);
                }
            });

            return result;
        }

        // Generate warp locations for the links between stars
        generateWarpLocations() {
            this.starlinks.forEach(link => {
                let warp1 = link.first.generateWarpLocationTo(link.second, this.random);
                let warp2 = link.second.generateWarpLocationTo(link.first, this.random);

                warp1.setJumpDestination(warp2);
                warp2.setJumpDestination(warp1);
            });

        }

        // Get the star nearest to another
        getNearestTo(star: Star, others = this.stars): Star | null {
            if (others.length === 0) {
                return null;
            } else {
                var mindist = this.radius * 2.0;
                var nearest: Star | null = null;
                others.forEach((istar: Star) => {
                    if (istar !== star) {
                        var dist = star.getDistanceTo(istar);
                        if (dist < mindist) {
                            nearest = istar;
                            mindist = dist;
                        }
                    }
                });
                return nearest;
            }
        }

        // Check if a link exists between two stars
        areLinked(first: Star, second: Star): boolean {
            var result = false;
            this.starlinks.forEach((link: StarLink) => {
                if (link.isLinking(first, second)) {
                    result = true;
                }
            });
            return result;
        }

        // Add a link between two stars
        addLink(first: Star, second: Star): void {
            if (!this.areLinked(first, second)) {
                this.starlinks.push(new StarLink(first, second));
            }
        }

        /**
         * Set the average level of encounters in each system
         */
        setEncounterLevels(maximal?: number) {
            if (!maximal) {
                maximal = Math.min(99, Math.ceil(Math.sqrt(this.stars.length)));
            }

            // Reset levels
            this.stars.forEach(star => star.level = 1);

            // Choose two systems to be the lowest and highest danger zones (not connected directly)
            let lowest = this.random.choice(this.stars.filter(star => star.getLinks().length > 1));
            let highest_choices = this.stars.filter(star => star != lowest && !star.getLinkTo(lowest));
            if (highest_choices.length == 0) {
                highest_choices = this.stars.filter(star => star != lowest);
            }
            let highest = this.random.choice(highest_choices);
            highest.level = maximal;

            // Make danger gradients
            range(this.stars.length).forEach(() => {
                this.stars.forEach(star => {
                    if (star != lowest && star != highest) {
                        let neighbors = star.getLinks().map(link => nn(link.getPeer(star)));
                        let minlevel = min(neighbors.map(neighbor => neighbor.level));
                        let maxlevel = max(neighbors.map(neighbor => neighbor.level));
                        star.level = (minlevel + maxlevel) / 2;
                    }
                });
            });

            // Round levels
            this.stars.forEach(star => star.level = Math.round(star.level));
        }

        /**
         * Add random shops
         */
        addShops(): void {
            this.stars.forEach(star => {
                star.locations.forEach(location => {
                    if (this.random.random() > 0.6) {
                        location.addShop(star.level);
                    }
                });
            });
        }

        /**
         * Get a good start location
         */
        getStartLocation(): StarLocation {
            let star = minBy(this.stars, star => star.level);
            return star.locations[0];
        }

        /**
         * Get a location from its ID
         */
        getLocation(id: RObjectId | null): StarLocation | null {
            return id === null ? null : this.locations.get(id);
        }
    }
}