You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

435 lines
14 KiB
JavaScript

import AssociativeArray from "../Core/AssociativeArray.js";
import Color from "../Core/Color.js";
import ColorGeometryInstanceAttribute from "../Core/ColorGeometryInstanceAttribute.js";
import defaultValue from "../Core/defaultValue.js";
import defined from "../Core/defined.js";
import DistanceDisplayCondition from "../Core/DistanceDisplayCondition.js";
import DistanceDisplayConditionGeometryInstanceAttribute from "../Core/DistanceDisplayConditionGeometryInstanceAttribute.js";
import ShowGeometryInstanceAttribute from "../Core/ShowGeometryInstanceAttribute.js";
import GroundPolylinePrimitive from "../Scene/GroundPolylinePrimitive.js";
import PolylineColorAppearance from "../Scene/PolylineColorAppearance.js";
import PolylineMaterialAppearance from "../Scene/PolylineMaterialAppearance.js";
import BoundingSphereState from "./BoundingSphereState.js";
import ColorMaterialProperty from "./ColorMaterialProperty.js";
import MaterialProperty from "./MaterialProperty.js";
import Property from "./Property.js";
var scratchColor = new Color();
var distanceDisplayConditionScratch = new DistanceDisplayCondition();
var defaultDistanceDisplayCondition = new DistanceDisplayCondition();
// Encapsulates a Primitive and all the entities that it represents.
function Batch(
orderedGroundPrimitives,
classificationType,
materialProperty,
zIndex,
asynchronous
) {
var appearanceType;
if (materialProperty instanceof ColorMaterialProperty) {
appearanceType = PolylineColorAppearance;
} else {
appearanceType = PolylineMaterialAppearance;
}
this.orderedGroundPrimitives = orderedGroundPrimitives; // scene level primitive collection
this.classificationType = classificationType;
this.appearanceType = appearanceType;
this.materialProperty = materialProperty;
this.updaters = new AssociativeArray();
this.createPrimitive = true;
this.primitive = undefined; // a GroundPolylinePrimitive encapsulating all the entities
this.oldPrimitive = undefined;
this.geometry = new AssociativeArray();
this.material = undefined;
this.updatersWithAttributes = new AssociativeArray();
this.attributes = new AssociativeArray();
this.invalidated = false;
this.removeMaterialSubscription = materialProperty.definitionChanged.addEventListener(
Batch.prototype.onMaterialChanged,
this
);
this.subscriptions = new AssociativeArray();
this.showsUpdated = new AssociativeArray();
this.zIndex = zIndex;
this._asynchronous = asynchronous;
}
Batch.prototype.onMaterialChanged = function () {
this.invalidated = true;
};
// Check if the given updater's material is compatible with this batch
Batch.prototype.isMaterial = function (updater) {
var material = this.materialProperty;
var updaterMaterial = updater.fillMaterialProperty;
if (
updaterMaterial === material ||
(updaterMaterial instanceof ColorMaterialProperty &&
material instanceof ColorMaterialProperty)
) {
return true;
}
return defined(material) && material.equals(updaterMaterial);
};
Batch.prototype.add = function (time, updater, geometryInstance) {
var id = updater.id;
this.updaters.set(id, updater);
this.geometry.set(id, geometryInstance);
// Updaters with dynamic attributes must be tracked separately, may exit the batch
if (
!updater.hasConstantFill ||
!updater.fillMaterialProperty.isConstant ||
!Property.isConstant(updater.distanceDisplayConditionProperty)
) {
this.updatersWithAttributes.set(id, updater);
} else {
var that = this;
// Listen for show changes. These will be synchronized in updateShows.
this.subscriptions.set(
id,
updater.entity.definitionChanged.addEventListener(function (
entity,
propertyName,
newValue,
oldValue
) {
if (propertyName === "isShowing") {
that.showsUpdated.set(updater.id, updater);
}
})
);
}
this.createPrimitive = true;
};
Batch.prototype.remove = function (updater) {
var id = updater.id;
this.createPrimitive = this.geometry.remove(id) || this.createPrimitive;
if (this.updaters.remove(id)) {
this.updatersWithAttributes.remove(id);
var unsubscribe = this.subscriptions.get(id);
if (defined(unsubscribe)) {
unsubscribe();
this.subscriptions.remove(id);
}
return true;
}
return false;
};
Batch.prototype.update = function (time) {
var isUpdated = true;
var primitive = this.primitive;
var orderedGroundPrimitives = this.orderedGroundPrimitives;
var geometries = this.geometry.values;
var i;
if (this.createPrimitive) {
var geometriesLength = geometries.length;
if (geometriesLength > 0) {
if (defined(primitive)) {
// Keep a handle to the old primitive so it can be removed when the updated version is ready.
if (!defined(this.oldPrimitive)) {
this.oldPrimitive = primitive;
} else {
// For if the new primitive changes again before it is ready.
orderedGroundPrimitives.remove(primitive);
}
}
primitive = new GroundPolylinePrimitive({
show: false,
asynchronous: this._asynchronous,
geometryInstances: geometries.slice(),
appearance: new this.appearanceType(),
classificationType: this.classificationType,
});
if (this.appearanceType === PolylineMaterialAppearance) {
this.material = MaterialProperty.getValue(
time,
this.materialProperty,
this.material
);
primitive.appearance.material = this.material;
}
orderedGroundPrimitives.add(primitive, this.zIndex);
isUpdated = false;
} else {
if (defined(primitive)) {
orderedGroundPrimitives.remove(primitive);
primitive = undefined;
}
var oldPrimitive = this.oldPrimitive;
if (defined(oldPrimitive)) {
orderedGroundPrimitives.remove(oldPrimitive);
this.oldPrimitive = undefined;
}
}
this.attributes.removeAll();
this.primitive = primitive;
this.createPrimitive = false;
} else if (defined(primitive) && primitive.ready) {
primitive.show = true;
if (defined(this.oldPrimitive)) {
orderedGroundPrimitives.remove(this.oldPrimitive);
this.oldPrimitive = undefined;
}
if (this.appearanceType === PolylineMaterialAppearance) {
this.material = MaterialProperty.getValue(
time,
this.materialProperty,
this.material
);
this.primitive.appearance.material = this.material;
}
var updatersWithAttributes = this.updatersWithAttributes.values;
var length = updatersWithAttributes.length;
for (i = 0; i < length; i++) {
var updater = updatersWithAttributes[i];
var entity = updater.entity;
var instance = this.geometry.get(updater.id);
var attributes = this.attributes.get(instance.id.id);
if (!defined(attributes)) {
attributes = primitive.getGeometryInstanceAttributes(instance.id);
this.attributes.set(instance.id.id, attributes);
}
if (!updater.fillMaterialProperty.isConstant) {
var colorProperty = updater.fillMaterialProperty.color;
var resultColor = Property.getValueOrDefault(
colorProperty,
time,
Color.WHITE,
scratchColor
);
if (!Color.equals(attributes._lastColor, resultColor)) {
attributes._lastColor = Color.clone(
resultColor,
attributes._lastColor
);
attributes.color = ColorGeometryInstanceAttribute.toValue(
resultColor,
attributes.color
);
}
}
var show =
entity.isShowing && (updater.hasConstantFill || updater.isFilled(time));
var currentShow = attributes.show[0] === 1;
if (show !== currentShow) {
attributes.show = ShowGeometryInstanceAttribute.toValue(
show,
attributes.show
);
}
var distanceDisplayConditionProperty =
updater.distanceDisplayConditionProperty;
if (!Property.isConstant(distanceDisplayConditionProperty)) {
var distanceDisplayCondition = Property.getValueOrDefault(
distanceDisplayConditionProperty,
time,
defaultDistanceDisplayCondition,
distanceDisplayConditionScratch
);
if (
!DistanceDisplayCondition.equals(
distanceDisplayCondition,
attributes._lastDistanceDisplayCondition
)
) {
attributes._lastDistanceDisplayCondition = DistanceDisplayCondition.clone(
distanceDisplayCondition,
attributes._lastDistanceDisplayCondition
);
attributes.distanceDisplayCondition = DistanceDisplayConditionGeometryInstanceAttribute.toValue(
distanceDisplayCondition,
attributes.distanceDisplayCondition
);
}
}
}
this.updateShows(primitive);
} else if (defined(primitive) && !primitive.ready) {
isUpdated = false;
}
return isUpdated;
};
Batch.prototype.updateShows = function (primitive) {
var showsUpdated = this.showsUpdated.values;
var length = showsUpdated.length;
for (var i = 0; i < length; i++) {
var updater = showsUpdated[i];
var entity = updater.entity;
var instance = this.geometry.get(updater.id);
var attributes = this.attributes.get(instance.id.id);
if (!defined(attributes)) {
attributes = primitive.getGeometryInstanceAttributes(instance.id);
this.attributes.set(instance.id.id, attributes);
}
var show = entity.isShowing;
var currentShow = attributes.show[0] === 1;
if (show !== currentShow) {
attributes.show = ShowGeometryInstanceAttribute.toValue(
show,
attributes.show
);
instance.attributes.show.value[0] = attributes.show[0];
}
}
this.showsUpdated.removeAll();
};
Batch.prototype.contains = function (updater) {
return this.updaters.contains(updater.id);
};
Batch.prototype.getBoundingSphere = function (updater, result) {
var primitive = this.primitive;
if (!primitive.ready) {
return BoundingSphereState.PENDING;
}
var attributes = primitive.getGeometryInstanceAttributes(updater.entity);
if (
!defined(attributes) ||
!defined(attributes.boundingSphere) ||
(defined(attributes.show) && attributes.show[0] === 0)
) {
return BoundingSphereState.FAILED;
}
attributes.boundingSphere.clone(result);
return BoundingSphereState.DONE;
};
Batch.prototype.destroy = function () {
var primitive = this.primitive;
var orderedGroundPrimitives = this.orderedGroundPrimitives;
if (defined(primitive)) {
orderedGroundPrimitives.remove(primitive);
}
var oldPrimitive = this.oldPrimitive;
if (defined(oldPrimitive)) {
orderedGroundPrimitives.remove(oldPrimitive);
}
this.removeMaterialSubscription();
};
/**
* @private
*/
function StaticGroundPolylinePerMaterialBatch(
orderedGroundPrimitives,
classificationType,
asynchronous
) {
this._items = [];
this._orderedGroundPrimitives = orderedGroundPrimitives;
this._classificationType = classificationType;
this._asynchronous = defaultValue(asynchronous, true);
}
StaticGroundPolylinePerMaterialBatch.prototype.add = function (time, updater) {
var items = this._items;
var length = items.length;
var geometryInstance = updater.createFillGeometryInstance(time);
var zIndex = Property.getValueOrDefault(updater.zIndex, 0);
// Check if the Entity represented by the updater has the same material or a material representable with per-instance color.
for (var i = 0; i < length; ++i) {
var item = items[i];
if (item.isMaterial(updater) && item.zIndex === zIndex) {
item.add(time, updater, geometryInstance);
return;
}
}
// If a compatible batch wasn't found, create a new batch.
var batch = new Batch(
this._orderedGroundPrimitives,
this._classificationType,
updater.fillMaterialProperty,
zIndex,
this._asynchronous
);
batch.add(time, updater, geometryInstance);
items.push(batch);
};
StaticGroundPolylinePerMaterialBatch.prototype.remove = function (updater) {
var items = this._items;
var length = items.length;
for (var i = length - 1; i >= 0; i--) {
var item = items[i];
if (item.remove(updater)) {
if (item.updaters.length === 0) {
items.splice(i, 1);
item.destroy();
}
break;
}
}
};
StaticGroundPolylinePerMaterialBatch.prototype.update = function (time) {
var i;
var items = this._items;
var length = items.length;
for (i = length - 1; i >= 0; i--) {
var item = items[i];
if (item.invalidated) {
items.splice(i, 1);
var updaters = item.updaters.values;
var updatersLength = updaters.length;
for (var h = 0; h < updatersLength; h++) {
this.add(time, updaters[h]);
}
item.destroy();
}
}
var isUpdated = true;
for (i = 0; i < items.length; i++) {
isUpdated = items[i].update(time) && isUpdated;
}
return isUpdated;
};
StaticGroundPolylinePerMaterialBatch.prototype.getBoundingSphere = function (
updater,
result
) {
var items = this._items;
var length = items.length;
for (var i = 0; i < length; i++) {
var item = items[i];
if (item.contains(updater)) {
return item.getBoundingSphere(updater, result);
}
}
return BoundingSphereState.FAILED;
};
StaticGroundPolylinePerMaterialBatch.prototype.removeAllPrimitives = function () {
var items = this._items;
var length = items.length;
for (var i = 0; i < length; i++) {
items[i].destroy();
}
this._items.length = 0;
};
export default StaticGroundPolylinePerMaterialBatch;