Add existing to tracked
This commit is contained in:
Generated
Vendored
+437
@@ -0,0 +1,437 @@
|
||||
import AssociativeArray from "../Core/AssociativeArray.js";
|
||||
import Color from "../Core/Color.js";
|
||||
import ColorGeometryInstanceAttribute from "../Core/ColorGeometryInstanceAttribute.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";
|
||||
|
||||
const scratchColor = new Color();
|
||||
const distanceDisplayConditionScratch = new DistanceDisplayCondition();
|
||||
const defaultDistanceDisplayCondition = new DistanceDisplayCondition();
|
||||
|
||||
// Encapsulates a Primitive and all the entities that it represents.
|
||||
function Batch(
|
||||
orderedGroundPrimitives,
|
||||
classificationType,
|
||||
materialProperty,
|
||||
zIndex,
|
||||
asynchronous,
|
||||
) {
|
||||
let 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) {
|
||||
const material = this.materialProperty;
|
||||
const 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) {
|
||||
const 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 {
|
||||
const 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) {
|
||||
const id = updater.id;
|
||||
this.createPrimitive = this.geometry.remove(id) || this.createPrimitive;
|
||||
if (this.updaters.remove(id)) {
|
||||
this.updatersWithAttributes.remove(id);
|
||||
const unsubscribe = this.subscriptions.get(id);
|
||||
if (defined(unsubscribe)) {
|
||||
unsubscribe();
|
||||
this.subscriptions.remove(id);
|
||||
this.showsUpdated.remove(id);
|
||||
}
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
};
|
||||
|
||||
Batch.prototype.update = function (time) {
|
||||
let isUpdated = true;
|
||||
let primitive = this.primitive;
|
||||
const orderedGroundPrimitives = this.orderedGroundPrimitives;
|
||||
const geometries = this.geometry.values;
|
||||
let i;
|
||||
|
||||
if (this.createPrimitive) {
|
||||
const 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);
|
||||
}
|
||||
}
|
||||
|
||||
const AppearanceType = this.appearanceType;
|
||||
|
||||
primitive = new GroundPolylinePrimitive({
|
||||
show: false,
|
||||
asynchronous: this._asynchronous,
|
||||
geometryInstances: geometries.slice(),
|
||||
appearance: new 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;
|
||||
}
|
||||
const 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;
|
||||
}
|
||||
const updatersWithAttributes = this.updatersWithAttributes.values;
|
||||
const length = updatersWithAttributes.length;
|
||||
for (i = 0; i < length; i++) {
|
||||
const updater = updatersWithAttributes[i];
|
||||
const entity = updater.entity;
|
||||
const instance = this.geometry.get(updater.id);
|
||||
|
||||
let 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) {
|
||||
const colorProperty = updater.fillMaterialProperty.color;
|
||||
const 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,
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
const show =
|
||||
entity.isShowing && (updater.hasConstantFill || updater.isFilled(time));
|
||||
const currentShow = attributes.show[0] === 1;
|
||||
if (show !== currentShow) {
|
||||
attributes.show = ShowGeometryInstanceAttribute.toValue(
|
||||
show,
|
||||
attributes.show,
|
||||
);
|
||||
}
|
||||
|
||||
const distanceDisplayConditionProperty =
|
||||
updater.distanceDisplayConditionProperty;
|
||||
if (!Property.isConstant(distanceDisplayConditionProperty)) {
|
||||
const 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) {
|
||||
const showsUpdated = this.showsUpdated.values;
|
||||
const length = showsUpdated.length;
|
||||
for (let i = 0; i < length; i++) {
|
||||
const updater = showsUpdated[i];
|
||||
const entity = updater.entity;
|
||||
const instance = this.geometry.get(updater.id);
|
||||
|
||||
let attributes = this.attributes.get(instance.id.id);
|
||||
if (!defined(attributes)) {
|
||||
attributes = primitive.getGeometryInstanceAttributes(instance.id);
|
||||
this.attributes.set(instance.id.id, attributes);
|
||||
}
|
||||
|
||||
const show = entity.isShowing;
|
||||
const 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) {
|
||||
const primitive = this.primitive;
|
||||
if (!primitive.ready) {
|
||||
return BoundingSphereState.PENDING;
|
||||
}
|
||||
const 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 () {
|
||||
const primitive = this.primitive;
|
||||
const orderedGroundPrimitives = this.orderedGroundPrimitives;
|
||||
if (defined(primitive)) {
|
||||
orderedGroundPrimitives.remove(primitive);
|
||||
}
|
||||
const 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 = asynchronous ?? true;
|
||||
}
|
||||
|
||||
StaticGroundPolylinePerMaterialBatch.prototype.add = function (time, updater) {
|
||||
const items = this._items;
|
||||
const length = items.length;
|
||||
const geometryInstance = updater.createFillGeometryInstance(time);
|
||||
const 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 (let i = 0; i < length; ++i) {
|
||||
const 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.
|
||||
const 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) {
|
||||
const items = this._items;
|
||||
const length = items.length;
|
||||
for (let i = length - 1; i >= 0; i--) {
|
||||
const item = items[i];
|
||||
if (item.remove(updater)) {
|
||||
if (item.updaters.length === 0) {
|
||||
items.splice(i, 1);
|
||||
item.destroy();
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
StaticGroundPolylinePerMaterialBatch.prototype.update = function (time) {
|
||||
let i;
|
||||
const items = this._items;
|
||||
const length = items.length;
|
||||
|
||||
for (i = length - 1; i >= 0; i--) {
|
||||
const item = items[i];
|
||||
if (item.invalidated) {
|
||||
items.splice(i, 1);
|
||||
const updaters = item.updaters.values;
|
||||
const updatersLength = updaters.length;
|
||||
for (let h = 0; h < updatersLength; h++) {
|
||||
this.add(time, updaters[h]);
|
||||
}
|
||||
item.destroy();
|
||||
}
|
||||
}
|
||||
|
||||
let isUpdated = true;
|
||||
for (i = 0; i < items.length; i++) {
|
||||
isUpdated = items[i].update(time) && isUpdated;
|
||||
}
|
||||
return isUpdated;
|
||||
};
|
||||
|
||||
StaticGroundPolylinePerMaterialBatch.prototype.getBoundingSphere = function (
|
||||
updater,
|
||||
result,
|
||||
) {
|
||||
const items = this._items;
|
||||
const length = items.length;
|
||||
for (let i = 0; i < length; i++) {
|
||||
const item = items[i];
|
||||
if (item.contains(updater)) {
|
||||
return item.getBoundingSphere(updater, result);
|
||||
}
|
||||
}
|
||||
return BoundingSphereState.FAILED;
|
||||
};
|
||||
|
||||
StaticGroundPolylinePerMaterialBatch.prototype.removeAllPrimitives =
|
||||
function () {
|
||||
const items = this._items;
|
||||
const length = items.length;
|
||||
for (let i = 0; i < length; i++) {
|
||||
items[i].destroy();
|
||||
}
|
||||
this._items.length = 0;
|
||||
};
|
||||
export default StaticGroundPolylinePerMaterialBatch;
|
||||
Reference in New Issue
Block a user