Unknown residue names

lib/app/src/app/component/chemistry/residue.cpp

@@ -69,10 +69,41 @@ namespace VTX::App::Component::Chemistry
 
 	const std::string & Residue::getShortName() const
 	{
-		return ChemDB::Residue::SYMBOL_SHORT_STR[ int( getSymbol() ) ];
+		ChemDB::Residue::SYMBOL symbol = getSymbol();
+		if ( symbol == ChemDB::Residue::SYMBOL::UNKNOWN )
+		{
+			return _moleculePtr->_moleculeStruct.residueUnknownNames[ _index ];
+		}
+		else
+		{
+			return ChemDB::Residue::SYMBOL_SHORT_STR[ int( symbol ) ];
+		}
+	}
+	const std::string & Residue::getName() const
+	{
+		ChemDB::Residue::SYMBOL symbol = getSymbol();
+		if ( symbol == ChemDB::Residue::SYMBOL::UNKNOWN )
+		{
+			return _moleculePtr->_moleculeStruct.residueUnknownNames[ _index ];
+		}
+		else
+		{
+			return ChemDB::Residue::SYMBOL_STR[ int( symbol ) ];
+		}
+	}
+
+	const std::string & Residue::getLongName() const
+	{
+		ChemDB::Residue::SYMBOL symbol = getSymbol();
+		if ( symbol == ChemDB::Residue::SYMBOL::UNKNOWN )
+		{
+			return _moleculePtr->_moleculeStruct.residueUnknownNames[ _index ];
+		}
+		else
+		{
+			return ChemDB::Residue::SYMBOL_NAME[ int( symbol ) ];
+		}
 	}
-	const std::string & Residue::getName() const { return ChemDB::Residue::SYMBOL_STR[ int( getSymbol() ) ]; }
-	const std::string & Residue::getLongName() const { return ChemDB::Residue::SYMBOL_NAME[ int( getSymbol() ) ]; }
 
 	ChemDB::Atom::TYPE Residue::getAtomType() const
 	{

lib/core/include/core/struct/molecule.hpp

@@ -53,6 +53,7 @@ namespace VTX::Core::Struct
 		std::vector<size_t>							  residueBondCounts;
 		std::vector<size_t>							  residueOriginalIds;
 		std::vector<ChemDB::SecondaryStructure::TYPE> residueSecondaryStructureTypes;
+		std::vector<std::string>					  residueUnknownNames;
 
 		void   initResidues( const size_t p_count );
 		size_t getResidueCount() const;

lib/core/src/core/chemdb/residue.cpp

@@ -30,6 +30,7 @@ namespace VTX::Core::ChemDB::Residue
 
 		return symbol.value_or( SYMBOL::UNKNOWN );
 	}
+
 	const SYMBOL getSymbolFromLongName( const std::string & p_residueName )
 	{
 		std::string formattedName = std::string( p_residueName.begin(), p_residueName.end() );
@@ -51,6 +52,7 @@ namespace VTX::Core::ChemDB::Residue
 
 		return SYMBOL::UNKNOWN;
 	}
+
 	const SYMBOL getSymbolFromAnyName( const std::string & p_residueName )
 	{
 		const size_t residueNameSize = p_residueName.size();

lib/core/src/core/struct/molecule.cpp

@@ -47,6 +47,7 @@ namespace VTX::Core::Struct
 		residueBondCounts.resize( p_count, 0 );
 		residueOriginalIds.resize( p_count, 0 );
 		residueSecondaryStructureTypes.resize( p_count, ChemDB::SecondaryStructure::TYPE::UNKNOWN );
+		residueUnknownNames.resize( p_count );
 	}
 	size_t Molecule::getResidueCount() const { return residueSymbols.size(); }
 

lib/io/src/io/reader/molecule.cpp

@@ -3,10 +3,10 @@
 #include <core/chemdb/secondary_structure.hpp>
 #include <core/struct/molecule.hpp>
 #include <core/struct/trajectory.hpp>
-#include <util/enum.hpp>
 #include <map>
 #include <util/chrono.hpp>
 #include <util/constants.hpp>
+#include <util/enum.hpp>
 #include <util/logger.hpp>
 
 namespace VTX::IO::Reader
@@ -100,6 +100,7 @@ namespace VTX::IO::Reader
 
 			const ChemDB::Residue::SYMBOL residueSymbol = VTX::Core::ChemDB::Residue::getSymbolFromName( residueName );
 
+			// TODO: delete?
 			// int symbolValue;
 
 			// if ( residueSymbol == ChemDB::Residue::SYMBOL::UNKNOWN )
@@ -129,6 +130,10 @@ namespace VTX::IO::Reader
 			//}
 
 			p_molecule.residueSymbols[ residueIdx ] = residueSymbol;
+			if ( residueSymbol == ChemDB::Residue::SYMBOL::UNKNOWN )
+			{
+				p_molecule.residueUnknownNames[ residueIdx ] = residueName;
+			}
 
 			const std::string secondaryStructure
 				= p_chemfileStruct.getCurrentResidueStringProperty( "secondary_structure" );

src/main.cpp

@@ -10,8 +10,6 @@
 #include <vector>
 
 #ifdef _WIN32
-// #include <Windows.h>
-
 extern "C"
 {
 	__declspec( dllexport ) uint32_t NvOptimusEnablement			 = 0x00000001;