public static VariableVectorsWriter Write(this VariableVectorsWriter writer, SnappedCylinder cylinder)
{
return(writer
.Write(cylinder.BottomCenter)
.Write(cylinder.Axis)
.Write(cylinder.Length)
.Write(cylinder.Radius));
}
public static VariableVectorsWriter Write(this VariableVectorsWriter writer, SnappedCone cone)
{
return(writer
.Write(cone.BottomCenter)
.Write(cone.Axis)
.Write(cone.Length)
.Write(cone.TopRadius)
.Write(cone.BottomRadius));
}
public static VariableVectorsWriter Write(this VariableVectorsWriter writer, SnappedCuboid cuboid)
{
return(writer
.Write(cuboid.Center)
.Write(cuboid.Width)
.Write(cuboid.Height)
.Write(cuboid.Depth)
.Write(cuboid.Wv)
.Write(cuboid.Hv)
.Write(cuboid.Dv));
}
public static VariableVectorsWriter Write(this VariableVectorsWriter writer, SnappedStraightGenCylinder sgc)
{
writer = writer
.Write(sgc.BottomCenter)
.Write(sgc.Axis)
.Write(sgc.Length);
foreach (var component in sgc.Components)
{
writer = writer
.Write(component.Radius);
}
return(writer);
}
public static VariableVectorsWriter Write(this VariableVectorsWriter writer, SnappedBendedGenCylinder bgc)
{
writer = writer
.Write(bgc.BottomCenter)
.Write(bgc.NPtop)
.Write(bgc.NPbot)
.Write(bgc.U)
.Write(bgc.V);
foreach (var component in bgc.Components)
{
writer = writer
.Write(component.Radius)
.Write(component.vS)
.Write(component.vT);
}
return(writer);
}
private void OptimizeAll()
{
#region Write all variables and their current values to a vector
var variablesWriter = new VariableVectorsWriter();
var startVectorWriter = new VectorsWriter();
// write cylinders
foreach (var snappedCylinder in sessionData.SnappedPrimitives.OfType <SnappedCylinder>())
{
variablesWriter.Write(snappedCylinder);
startVectorWriter.Write(snappedCylinder);
}
// write cones
foreach (var snappedCone in sessionData.SnappedPrimitives.OfType <SnappedCone>())
{
variablesWriter.Write(snappedCone);
startVectorWriter.Write(snappedCone);
}
#endregion
// all objective functions. Will be summed eventually to form one big objective.
var objectives = new List <Term>();
// all equality constraints.
var constraints = new List <Term>();
var curvesToAnnotations = new Dictionary <FeatureCurve, ISet <Annotation> >();
#region Get mapping of curves to annotations
foreach (var fc in sessionData.FeatureCurves)
{
curvesToAnnotations[fc] = new HashSet <Annotation>();
}
foreach (var annotation in sessionData.Annotations)
{
IEnumerable <FeatureCurve> curves = null;
annotation.MatchClass <Parallelism>(pa => curves = pa.Elements);
annotation.MatchClass <Coplanarity>(ca => curves = ca.Elements);
annotation.MatchClass <Cocentrality>(ca => curves = ca.Elements);
Debug.Assert(curves != null);
foreach (var fc in curves)
{
curvesToAnnotations[fc].Add(annotation);
}
}
#endregion
#region get objectives and constraints for primitives
foreach (var snappedPrimitive in sessionData.SnappedPrimitives)
{
var objectiveAndConstraints = snappersManager.Reconstruct(snappedPrimitive, curvesToAnnotations);
objectives.Add(objectiveAndConstraints.Item1);
constraints.AddRange(objectiveAndConstraints.Item2);
}
#endregion
#region get constraints for annotations
foreach (var annotation in sessionData.Annotations)
{
var constraintTerms = GetAnnotationConstraints(annotation);
constraints.AddRange(constraintTerms);
}
#endregion
#region perform optimization
var finalObjective = TermUtils.SafeSum(objectives);
var vars = variablesWriter.ToArray();
var vals = startVectorWriter.ToArray();
var optimum = Optimizer.MinAugmentedLagrangian(finalObjective, constraints.ToArray(), vars, vals, mu: 10, tolerance: 1E-5);
#endregion
#region read data back from the optimized vector
var resultReader = new VectorsReader(optimum);
foreach (var snappedCylinder in sessionData.SnappedPrimitives.OfType <SnappedCylinder>())
{
resultReader.Read(snappedCylinder);
}
foreach (var snappedCone in sessionData.SnappedPrimitives.OfType <SnappedCone>())
{
resultReader.Read(snappedCone);
}
#endregion
#region Update feature curves
foreach (var snappedPrimitive in sessionData.SnappedPrimitives)
{
snappedPrimitive.UpdateFeatureCurves();
}
#endregion
}
public static VariableVectorsWriter Write(this VariableVectorsWriter writer, SnappedSphere sphere)
{
return(writer
.Write(sphere.Center)
.Write(sphere.Radius));
}
public PrimitivesWriter()
{
variablesWriter = new VariableVectorsWriter();
valuesWriter = new VectorsWriter();
}
