coriolis/app/js/controllers/controller-outfit.js

angular.module('app').controller('OutfitController', ['$window', '$rootScope', '$scope', '$state', '$stateParams', '$translate', 'ShipsDB', 'Ship', 'Components', 'Serializer', 'Persist', 'calcTotalRange', 'calcSpeed', function($window, $rootScope, $scope, $state, $p, $translate, Ships, Ship, Components, Serializer, Persist, calcTotalRange, calcSpeed) {
  var win = angular.element($window);   // Angularized window object for event triggering
  var data = Ships[$p.shipId];   // Retrieve the basic ship properties, slots and defaults
  var ship = new Ship($p.shipId, data.properties, data.slots); // Create a new Ship instance
  var retrofitShip = new Ship($p.shipId, data.properties, data.slots); // Create a new Ship for retrofit comparison

  //  Update the ship instance with the code (if provided) or the 'factory' defaults.
  if ($p.code) {
    Serializer.toShip(ship, $p.code);  // Populate components from 'code' URL param
    $scope.code = $p.code;
  } else {
    ship.buildWith(data.defaults);  // Populate with default components
  }

  $scope.buildName = $p.bn;
  $scope.ships = Ships;
  $rootScope.title = ship.name + ($scope.buildName ? ' - ' + $scope.buildName : '');
  $scope.ship = ship;
  $scope.pp = ship.common[0];   // Power Plant
  $scope.th = ship.common[1];   // Thruster
  $scope.fsd = ship.common[2];  // Frame Shrift Drive
  $scope.ls = ship.common[3];   // Life Support
  $scope.pd = ship.common[4];   // Power Distributor
  $scope.ss = ship.common[5];   // Sensors
  $scope.ft = ship.common[6];   // Fuel Tank
  $scope.hps = ship.hardpoints;
  $scope.internal = ship.internal;
  $scope.costList = ship.costList;
  $scope.powerList = ship.powerList;
  $scope.priorityBands = ship.priorityBands;
  $scope.availCS = ship.getAvailableComponents();
  $scope.selectedSlot = null;
  $scope.savedCode = Persist.getBuild(ship.id, $scope.buildName);
  $scope.canSave = Persist.isEnabled();
  $scope.allBuilds = Persist.builds;
  $scope.fuel = 0;
  $scope.pwrDesc = false;
  $scope.pwrPredicate = 'type';
  $scope.retroDesc = false;
  $scope.retroPredicate = 'netCost';
  $scope.costDesc = true;
  $scope.costPredicate = 'c.cost';
  $scope.ammoDesc = false;
  $scope.ammoPredicate = 'ammoName';
  $scope.costTab = Persist.getCostTab() || 'costs';

  if ($scope.savedCode) {
    Serializer.toShip(retrofitShip, $scope.savedCode);  // Populate components from last save
    $scope.retrofitBuild = $scope.buildName;
  } else {
    retrofitShip.buildWith(data.defaults);
    $scope.retrofitBuild = null;
  }

  ship.applyDiscounts($rootScope.discounts.ship, $rootScope.discounts.components);
  retrofitShip.applyDiscounts($rootScope.discounts.ship, $rootScope.discounts.components);
  updateRetrofitCosts();

  $scope.jrSeries = {
    xMin: 0,
    xMax: ship.cargoCapacity,
    yMax: ship.unladenRange,
    yMin: 0,
    func: function(cargo) { // X Axis is Cargo
      return ship.getJumpRangeForMass(ship.unladenMass + $scope.fuel + cargo, $scope.fuel);
    }
  };
  $scope.jrChart = {
    labels: {
      xAxis: {
        title: 'cargo',
        unit: 'T'
      },
      yAxis: {
        title: 'jump range',
        unit: 'LY'
      }
    }
  };

  $scope.trSeries = {
    xMin: 0,
    xMax: ship.cargoCapacity,
    yMax: ship.unladenTotalRange,
    yMin: 0,
    func: function(cargo) { // X Axis is Cargo
      return calcTotalRange(ship.unladenMass + cargo, $scope.fsd.c, $scope.fuel);
    }
  };
  $scope.trChart = {
    labels: {
      xAxis: {
        title: 'cargo',
        unit: 'T'
      },
      yAxis: {
        title: 'total range',
        unit: 'LY'
      }
    }
  };

  $scope.speedSeries = {
    xMin: 0,
    xMax: ship.cargoCapacity,
    yMax: calcSpeed(ship.unladenMass, ship.speed, ship.boost, $scope.th.c, ship.pipSpeed).boost,
    yMin: 0,
    series: ['boost', '4 Pips', '2 Pips', '0 Pips'],
    colors: ['#0088d2', '#ff8c0d', '#D26D00', '#c06400'],
    func: function(cargo) { // X Axis is Cargo
      return calcSpeed(ship.unladenMass + $scope.fuel + cargo, ship.speed, ship.boost, $scope.th.c, ship.pipSpeed);
    }
  };
  $scope.speedChart = {
    labels: {
      xAxis: {
        title: 'cargo',
        unit: 'T'
      },
      yAxis: {
        title: 'speed',
        unit: 'm/s'
      }
    }
  };

  /**
   * 'Opens' a select for component selection.
   *
   * @param  {[type]} e    The event object
   * @param  {[type]} slot The slot that is being 'opened' for selection
   */
  $scope.selectSlot = function(e, slot) {
    e.stopPropagation();
    if ($scope.selectedSlot == slot) {
      $scope.selectedSlot = null;
    } else {
      $scope.selectedSlot = slot;
    }
  };

  /**
   * Updates the ships build with the selected component for the
   * specified slot. Prevents the click event from propagation.
   *
   * @param  {string} type Shorthand key/string for identifying the slot & component type
   * @param  {[type]} slot The slot object belonging to the ship instance
   * @param  {[type]} e    The event object
   */
  $scope.select = function(type, slot, e, id) {
    e.stopPropagation();

    if (!id) {  // Find component id if not passed
      var elem = e.target;
      while (elem && elem !== e.currentTarget && !elem.getAttribute('cpid')) {
        elem = elem.parentElement;
      }
      if (elem) {
        id = elem.getAttribute('cpid');
      }
    }

    if (id) {
      if (id == 'empty') {
        ship.use(slot, null, null);
      } else if (type == 'h') {
        ship.use(slot, id, Components.hardpoints(id));
      } else if (type == 'c') {
        ship.use(slot, id, Components.common(ship.common.indexOf(slot), id));
      } else if (type == 'i') {
        ship.use(slot, id, Components.internal(id));
      } else if (type == 'b') {
        ship.useBulkhead(id);
      }
      $scope.selectedSlot = null;
      updateState(Serializer.fromShip(ship));
    }
  };

  /**
   * Reload the build from the last save.
   */
  $scope.reloadBuild = function() {
    if ($scope.buildName && $scope.savedCode) {
      Serializer.toShip(ship, $scope.savedCode);  // Repopulate with components from last save
      updateState($scope.savedCode);
    }
  };

  $scope.resetBuild = function() {
    ship.buildWith(data.defaults);  // Populate with default components
    updateState(null);
  };

  /**
   * Strip ship to A-class and biggest A-class shield generator with military bulkheads
   */
  $scope.aRatedBuild = function() {
    ship
        .useBulkhead(2)     // Military Composite
        .useCommon('A')
        .emptyHardpoints()
        .emptyInternal();

    ship.internal.some(function(slot) {
      if (!slot.eligible || slot.eligible.sg) { // Assuming largest slot can hold an eligible shield
        var sg = Components.findInternal('sg', slot.maxClass, 'A');
        ship.use(slot, sg.id, sg);
        return true;
      }
    });
    updateState(Serializer.fromShip(ship));
  };

  /**
   * Optimize for the lower mass build that can still boost and power the ship
   * without power management.
   */
  $scope.optimizeMassBuild = function() {
    updateState(Serializer.fromShip(ship.optimizeMass()));
  };

  /**
   * Optimize for the lower mass build that can still boost and power the ship
   * without power management.
   */
  $scope.optimizeCommon = function() {
    updateState(Serializer.fromShip(ship.useLightestCommon()));
  };

  $scope.useCommon = function(rating) {
    updateState(Serializer.fromShip(ship.useCommon(rating)));
  };

  $scope.useHardpoint = function(group, mount, missile) {
    updateState(Serializer.fromShip(ship.useWeapon(group, mount, missile)));
  };

  $scope.useUtility = function(group, rating) {
    updateState(Serializer.fromShip(ship.useUtility(group, rating)));
  };

  $scope.emptyInternal = function() {
    updateState(Serializer.fromShip(ship.emptyInternal()));
  };

  $scope.emptyHardpoints = function() {
    updateState(Serializer.fromShip(ship.emptyWeapons()));
  };

  $scope.emptyUtility = function() {
    updateState(Serializer.fromShip(ship.emptyUtility()));
  };

  $scope.fillWithCargo = function() {
    ship.internal.forEach(function(slot) {
      var id = Components.findInternalId('cr', slot.maxClass, 'E');
      ship.use(slot, id, Components.internal(id));
    });
    updateState(Serializer.fromShip(ship));
  };

  $scope.fillWithCells = function() {
    var chargeCap = 0; // Capacity of single activation
    ship.internal.forEach(function(slot) {
      var id = Components.findInternalId('scb', slot.maxClass, 'A');
      if ((!slot.c || (slot.c.grp != 'sg' && slot.c.grp != 'psg')) && (!slot.eligible || slot.eligible.scb)) { // Check eligibility because of Orca, don't overwrite generator
        ship.use(slot, id, Components.internal(id));
        chargeCap += Components.internal(id).recharge;
        ship.setSlotEnabled(slot, chargeCap <= ship.shieldStrength); // Don't waste cell capacity on overcharge
      }
    });
    updateState(Serializer.fromShip(ship));
  };

  /**
   * Fill all internal slots with Cargo Racks, and optmize internal components.
   * Hardpoints are not altered.
   */
  $scope.optimizeCargo = function() {
    ship.internal.forEach(function(slot) {
      var id = Components.findInternalId('cr', slot.maxClass, 'E');
      ship.use(slot, id, Components.internal(id));
    });
    ship.useLightestCommon();
    updateState(Serializer.fromShip(ship));
  };

  /**
   * Optimize common and internal components, hardpoints for exploration
   */
  $scope.optimizeExplorer = function() {
    var intLength = ship.internal.length,
        heatSinkCount = 2,  // Fit 2 heat sinks if possible
        afmUnitCount = 2,   // Fit 2 AFM Units if possible
        sgSlot,
        fuelScoopSlot,
        sgId = $scope.availCS.lightestShieldGenerator(ship.hullMass),
        sg = Components.internal(sgId);

    ship.setSlotEnabled(ship.cargoHatch, false)
        .use(ship.internal[--intLength], '2f', Components.internal('2f'))      // Advanced Discovery Scanner
        .use(ship.internal[--intLength], '2i', Components.internal('2i'));      // Detailed Surface Scanner

    for (var i = 0; i < intLength; i++) {
      var slot = ship.internal[i];
      var nextSlot = (i + 1) < intLength ? ship.internal[i + 1] : null;
      if (!fuelScoopSlot && (!slot.eligible || slot.eligible.fs)) {             // Fit best possible Fuel Scoop
        var fuelScoopId = Components.findInternalId('fs', slot.maxClass, 'A');
        fuelScoopSlot = slot;
        ship.use(fuelScoopSlot, fuelScoopId, Components.internal(fuelScoopId));
        ship.setSlotEnabled(fuelScoopSlot, true);

      // Mount a Shield generator if possible AND an AFM Unit has been mounted already (Guarantees at least 1 AFM Unit)
      } else if (!sgSlot && afmUnitCount < 2 && sg.class <= slot.maxClass && (!slot.eligible || slot.eligible.sg) && (!nextSlot || nextSlot.maxClass < sg.class)) {
        sgSlot = slot;
        ship.use(sgSlot, sgId, sg);
        ship.setSlotEnabled(sgSlot, true);
      } else if (afmUnitCount > 0 && (!slot.eligible || slot.eligible.am)) {
        afmUnitCount--;
        var id = Components.findInternalId('am', slot.maxClass, 'A'); // Best AFM Unit for slot
        ship.use(slot, id, Components.internal(id));
        ship.setSlotEnabled(slot, false);   // Disabled power for AFM Unit

      } else {
        ship.use(slot, null, null);
      }
    }

    ship.hardpoints.forEach(function(s) {
      if (s.maxClass == 0 && heatSinkCount) {       // Mount up to 2 heatsinks
        ship.use(s, '02', Components.hardpoints('02'));
        ship.setSlotEnabled(s, heatSinkCount == 2); // Only enable a single Heatsink
        heatSinkCount--;
      } else {
        ship.use(s, null, null);
      }
    });

    if (sgSlot) {
      // The SG and Fuel scoop to not need to be powered at the same time
      if (sgSlot.c.power > fuelScoopSlot.c.power) { // The Shield generator uses the most power
        ship.setSlotEnabled(fuelScoopSlot, false);
      } else {                                    // The Fuel scoop uses the most power
        ship.setSlotEnabled(sgSlot, false);
      }
    }

    ship.useLightestCommon({ pd: '1D', ppRating: 'A' });
    updateState(Serializer.fromShip(ship));
  };

  /**
   * Save the current build. Will replace the saved build if there is one
   * for this ship & with the exact name.
   */
  $scope.saveBuild = function() {
    if (!$scope.buildName) {
      return;
    }
    // No change hav been made, i.e. save ship default build under a name
    if (!$scope.code) {
      $scope.code = Serializer.fromShip(ship);
    }
    // Only save if there a build name and a change has been made or the build has never been saved
    if ($scope.code != $scope.savedCode) {
      Persist.saveBuild(ship.id, $scope.buildName, $scope.code);
      $scope.savedCode = $scope.code;
      if ($scope.retrofitBuild === $scope.buildName) {
        Serializer.toShip(retrofitShip, $scope.code);
      }
      updateState($scope.code);
    }
  };

  /**
   * Export the build to detailed JSON
   */
  $scope.exportBuild = function(e) {
    e.stopPropagation();

    if ($scope.buildName) {
      $state.go('modal.export', {
        title: $scope.buildName + ' ' + $translate.instant('export'),
        description: $translate.instant('PHRASE_EXPORT_DESC'),
        data: Serializer.toDetailedBuild($scope.buildName, ship, $scope.code || Serializer.fromShip(ship))
      });
    }
  };

  /**
   * Permanently delete the current build and redirect/reload this controller
   * with the 'factory' build of the current ship.
   */
  $scope.deleteBuild = function() {
    Persist.deleteBuild(ship.id, $scope.buildName);
    $state.go('outfit', { shipId: ship.id, code: null, bn: null }, { location: 'replace', reload: true });
  };

  /**
   * On build name change, retrieve the existing saved code if there is one
   */
  $scope.bnChange = function() {
    $scope.savedCode = Persist.getBuild(ship.id, $scope.buildName);
  };

  /**
   * Toggle cost of the selected component
   * @param  {object} item The component being toggled
   */
  $scope.toggleCost = function(item) {
    ship.setCostIncluded(item, !item.incCost);
  };

/**
   * Toggle cost of the selected component for retrofitting comparison
   * @param  {object} item The component being toggled
   */
  $scope.toggleRetrofitCost = function(item) {
    retrofitShip.setCostIncluded(item, !item.incCost);
    updateRetrofitCosts();
  };

  /**
   * [sortCost description]
   * @param  {[type]} key [description]
   * @return {[type]}     [description]
   */
  $scope.sortCost = function(key) {
    $scope.costDesc = $scope.costPredicate == key ? !$scope.costDesc : $scope.costDesc;
    $scope.costPredicate = key;
  };

  $scope.sortPwr = function(key) {
    $scope.pwrDesc = $scope.pwrPredicate == key ? !$scope.pwrDesc : $scope.pwrDesc;
    $scope.pwrPredicate = key;
  };

  $scope.sortRetrofit = function(key) {
    $scope.retroDesc = $scope.retroPredicate == key ? !$scope.retroDesc : $scope.retroDesc;
    $scope.retroPredicate = key;
  };

  $scope.sortAmmo = function(key) {
    $scope.ammoDesc = $scope.ammoPredicate == key ? !$scope.ammoDesc : $scope.ammoDesc;
    $scope.ammoPredicate = key;
  };

  /**
   * Toggle the power on/off for the selected component
   * @param  {object} item The component being toggled
   */
  $scope.togglePwr = function(c) {
    ship.setSlotEnabled(c, !c.enabled);
    updateState(Serializer.fromShip(ship));
  };

  $scope.incPriority = function(c) {
    if (ship.changePriority(c, c.priority + 1)) {
      updateState(Serializer.fromShip(ship));
    }
  };

  $scope.decPriority = function(c) {
    if (ship.changePriority(c, c.priority - 1)) {
      updateState(Serializer.fromShip(ship));
    }
  };

  $scope.fuelChange = function(fuel) {
    $scope.fuel = fuel;
    win.triggerHandler('render');
  };

  $scope.statusRetracted = function(slot) {
    return ship.getSlotStatus(slot, false);
  };

  $scope.statusDeployed = function(slot) {
    return ship.getSlotStatus(slot, true);
  };

  $scope.setRetrofitBase = function() {
    if ($scope.retrofitBuild) {
      Serializer.toShip(retrofitShip, Persist.getBuild(ship.id, $scope.retrofitBuild));
    } else {
      retrofitShip.buildWith(data.defaults);
    }
    updateRetrofitCosts();
  };

  $scope.updateCostTab = function(tab) {
    Persist.setCostTab(tab);
    $scope.costTab = tab;
  };

  $scope.ppWarning = function(pp) {
    return pp.pGen < ship.powerRetracted;
  };

  $scope.pdWarning = function(pd) {
    return pd.enginecapacity < ship.boostEnergy;
  };

  // Utilify functions

  function updateState(code) {
    $scope.code = code;
    $state.go('outfit', { shipId: ship.id, code: $scope.code, bn: $scope.buildName }, { location: 'replace', notify: false });
    $scope.speedSeries.xMax = $scope.trSeries.xMax = $scope.jrSeries.xMax = ship.cargoCapacity;
    $scope.jrSeries.yMax = ship.unladenRange;
    $scope.trSeries.yMax = ship.unladenTotalRange;
    $scope.speedSeries.yMax = calcSpeed(ship.unladenMass, ship.speed, ship.boost, $scope.th.c, ship.pipSpeed).boost;
    updateRetrofitCosts();
    win.triggerHandler('pwrchange');
  }

  function updateRetrofitCosts() {
    var costs = $scope.retrofitList = [];
    var total = 0, i, l, item;

    if (ship.bulkheads.id != retrofitShip.bulkheads.id) {
      item = {
        buyClassRating: ship.bulkheads.c.class + ship.bulkheads.c.rating,
        buyName: ship.bulkheads.c.name,
        sellClassRating: retrofitShip.bulkheads.c.class + retrofitShip.bulkheads.c.rating,
        sellName: retrofitShip.bulkheads.c.name,
        netCost: ship.bulkheads.discountedCost - retrofitShip.bulkheads.discountedCost,
        retroItem: retrofitShip.bulkheads
      };
      costs.push(item);
      if (retrofitShip.bulkheads.incCost) {
        total += item.netCost;
      }
    }

    for (var g in { common: 1, internal: 1, hardpoints: 1 }) {
      var retroSlotGroup = retrofitShip[g];
      var slotGroup = ship[g];
      for (i = 0, l = slotGroup.length; i < l; i++) {
        if (slotGroup[i].id != retroSlotGroup[i].id) {
          item = { netCost: 0, retroItem: retroSlotGroup[i] };
          if (slotGroup[i].id) {
            item.buyName = slotGroup[i].c.name || slotGroup[i].c.grp;
            item.buyClassRating = slotGroup[i].c.class + slotGroup[i].c.rating;
            item.netCost = slotGroup[i].discountedCost;
          }
          if (retroSlotGroup[i].id) {
            item.sellName = retroSlotGroup[i].c.name || retroSlotGroup[i].c.grp;
            item.sellClassRating = retroSlotGroup[i].c.class + retroSlotGroup[i].c.rating;
            item.netCost -= retroSlotGroup[i].discountedCost;
          }
          costs.push(item);
          if (retroSlotGroup[i].incCost) {
            total += item.netCost;
          }
        }
      }
    }
    $scope.retrofitTotal = total;
    updateAmmoCosts();
  }

  function updateAmmoCosts() {
    var costs = $scope.ammoList = [];
    var total = 0, i, l, item;

    for (var g in { common: 1, internal: 1, hardpoints: 1 }) {
      var slotGroup = ship[g];
      for (i = 0, l = slotGroup.length; i < l; i++) {
        var q = 0;
        if (slotGroup[i].id) {
          //special cases needed for fuel, SCB and AFMU since they don't use standard ammo/clip
          if (slotGroup[i].c.grp == 'ft') {
            q = slotGroup[i].c.capacity;
            slotGroup[i].c.ammocost = 50;
          }
          if (slotGroup[i].c.grp == 'scb') {
            q = slotGroup[i].c.cells;
          }
          if (slotGroup[i].c.grp == 'am') {
            q = slotGroup[i].c.ammo;
          }
          //calculate ammo costs only if a cost is specified
          if (slotGroup[i].c.ammocost > 0) {
            if (q == 0) {
              q = slotGroup[i].c.clip + slotGroup[i].c.ammo;
            }
            item = {
              ammoClassRating: slotGroup[i].c.class + slotGroup[i].c.rating,
              ammoName: slotGroup[i].c.name || slotGroup[i].c.grp,
              ammoMax: q,
              ammoUnitCost: slotGroup[i].c.ammocost,
              ammoTotalCost: q * slotGroup[i].c.ammocost
            };
            costs.push(item);
            total += item.ammoTotalCost;
          }
        }
      }
    }
    $scope.ammoTotal = total;
  }

  // Hide any open menu/slot/etc if the background is clicked
  $scope.$on('close', function() {
    $scope.selectedSlot = null;
  });

  // Hide any open menu/slot/etc if the background is clicked
  $scope.$on('languageChanged', function() {
    $scope.selectedSlot = null;
  });

  // Hide any open menu/slot/etc if the background is clicked
  $scope.$on('discountChange', function() {
    ship.applyDiscounts($rootScope.discounts.ship, $rootScope.discounts.components);
    retrofitShip.applyDiscounts($rootScope.discounts.ship, $rootScope.discounts.components);
    updateRetrofitCosts();
  });

}]);