Temperature color schemes

Piriziwè

Thank you, Sir. Sorry for not precising in my post. I would like to do it programmatically. By code, I load a pdb or xyz file and then color atoms according to their temperature factor.

Stephane

Here is essentially the code we use in SAMSON when the user does what @Dmitriy-Marin suggested:


// index the current selection 

SBPointerIndexer<SBNode> const* selectedNodes = SAMSON::getActiveDocument()->getSelectedNodes();

SBNodeIndexer nodeIndexer;
SB_FOR(SBNode* node, *selectedNodes) nodeIndexer.addNode(node);

// make the color change undoable (start holding changes to the document)

SAMSON::beginHolding("Color per temperature factor");

// remove existing materials from selected nodes and their descendants

SB_FOR(SBNode* node, *selectedNodes) {

    SBNodeIndexer nodeIndexer;
    node->getNodes(nodeIndexer);

    SB_FOR(SBNode* descendantNode, nodeIndexer) descendantNode->removeMaterial();

}

// apply the new color scheme 

SB_FOR(SBNode* node, *selectedNodes) {

    SBNodeMaterial* nodeMaterial = new SBNodeMaterial();
    if (!node->addMaterial(nodeMaterial)) delete nodeMaterial;

    SBNodeColorScheme* colorScheme = new SBDColorSchemePerTemperatureFactor(nodeIndexer);
    node->getMaterial()->setColorScheme(colorScheme);

}

// stop holding changes to the document

SAMSON::endHolding();

The reason we index the current selection in nodeIndexer (which is passed to the color scheme) is
so that the color scheme can traverse these nodes, and their descendants, and determine the minimum and maximum values of the temperature factor.

Other color schemes are usable:

SBDColorSchemeConstant
SBDColorSchemeCPK
SBDColorSchemePerChainID
SBDColorSchemePerFormalCharge
SBDColorSchemePerOccupancy
SBDColorSchemePerPartialCharge
SBDColorSchemePerResidueSequenceNumber
SBDColorSchemePerResidueType
SBDColorSchemePerSecondaryStructureType

Piriziwè

@stephane-redon Thank you very much. I am going to try it and give you a feedback about the experience.

Stephane

By the way, sorry, I just fixed the code above (replaced Core and Data with SAMSON) (thanks @Dmitriy-Marin !)

Piriziwè

@stephane-redon Thank you

DmitriyMarin

If you want a custom colorization according to the temperature factor, you can use the same algorithm which @Stephane-Redon showed above and colorize atoms according to the node->getTemperatureFactor() using your own color scheme:

float red = node->getTemperatureFactor() / maximalTemperatureFactor; // should be scaled to 1.0
SBDColorSchemeConstant *colorScheme = new SBDColorSchemeConstant(red, 0.0, 0.0, 1.0); //SBDColorSchemeConstant (float red, float green, float blue, float alpha=1.0f)
node->getMaterial()->setColorScheme(colorScheme);

You can create more advanced color scheme by setting the HSV color space based on the temperatureFactor and converting it to the RGB color space.

Piriziwè

@dmitriy-marin Thank you very much. That's great !

Piriziwè

@stephane-redon Hi Sir,
I am trying the example above and I have an error when I compile: error: expected type-specifier before ‘SBDColorSchemePerTemperatureFactor’
SBNodeColorScheme* colorScheme = new SBDColorSchemePerTemperatureFactor(nodeIndexer);

Would you mind helping me with this error, please?

Piriziwè

@piriziwè It's done. I just included "SBDColorSchemePerTemperatureFactor.hpp"

Stephane

Sorry, should have mentioned that. Great!