#4973 - Connection between molecule and monomer lost after opening an…

…d saving to ket - fixed monomer atoms ids mapping during struct clone
epam · Jul 3, 2024 · ce018cb · ce018cb
1 parent 98d6c08
commit ce018cb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/packages/ketcher-core/src/domain/entities/monomerMicromolecule.ts b/packages/ketcher-core/src/domain/entities/monomerMicromolecule.ts
@@ -30,13 +30,18 @@ export class MonomerMicromolecule extends SGroup {
     return { position: this.pp, atomId: sgroupContractedPosition.atomId };
   }
 
-  public static clone(monomerMicromolecule: MonomerMicromolecule) {
+  public static clone(
+    monomerMicromolecule: MonomerMicromolecule,
+    atomIdMap?: Map<number, number>,
+  ) {
     const monomerMicromoleculeClone = new MonomerMicromolecule(
       monomerMicromolecule.type,
       monomerMicromolecule.monomer,
     );
     monomerMicromoleculeClone.pp = monomerMicromolecule.pp;
-    monomerMicromoleculeClone.atoms = monomerMicromolecule.atoms;
+    monomerMicromoleculeClone.atoms = atomIdMap
+      ? monomerMicromolecule.atoms.map((elem) => atomIdMap.get(elem))
+      : monomerMicromolecule.atoms;
 
     return monomerMicromoleculeClone;
   }

diff --git a/packages/ketcher-core/src/domain/entities/struct.ts b/packages/ketcher-core/src/domain/entities/struct.ts
@@ -293,7 +293,7 @@ export class Struct {
 
       sg =
         oldSgroup instanceof MonomerMicromolecule
-          ? MonomerMicromolecule.clone(oldSgroup)
+          ? MonomerMicromolecule.clone(oldSgroup, aidMap!)
           : SGroup.clone(sg, aidMap!);
 
       const id = cp.sgroups.add(sg);