------------------------------------------------------------------------------
| 275 | |
| 276 | //------------------------------------------------------------------------------ |
| 277 | void vtkPhyloXMLTreeWriter::WriteColorElement( |
| 278 | vtkTree* input, vtkIdType vertex, vtkXMLDataElement* element) |
| 279 | { |
| 280 | vtkUnsignedCharArray* colorArray = |
| 281 | vtkArrayDownCast<vtkUnsignedCharArray>(input->GetVertexData()->GetAbstractArray("color")); |
| 282 | if (!colorArray) |
| 283 | { |
| 284 | return; |
| 285 | } |
| 286 | |
| 287 | vtkNew<vtkXMLDataElement> colorElement; |
| 288 | colorElement->SetName("color"); |
| 289 | |
| 290 | vtkNew<vtkXMLDataElement> redElement; |
| 291 | redElement->SetName("red"); |
| 292 | std::string r = vtkVariant(colorArray->GetComponent(vertex, 0)).ToString(); |
| 293 | redElement->SetCharacterData(r.c_str(), static_cast<int>(r.length())); |
| 294 | |
| 295 | vtkNew<vtkXMLDataElement> greenElement; |
| 296 | greenElement->SetName("green"); |
| 297 | std::string g = vtkVariant(colorArray->GetComponent(vertex, 1)).ToString(); |
| 298 | greenElement->SetCharacterData(g.c_str(), static_cast<int>(g.length())); |
| 299 | |
| 300 | vtkNew<vtkXMLDataElement> blueElement; |
| 301 | blueElement->SetName("blue"); |
| 302 | std::string b = vtkVariant(colorArray->GetComponent(vertex, 2)).ToString(); |
| 303 | blueElement->SetCharacterData(b.c_str(), static_cast<int>(b.length())); |
| 304 | |
| 305 | colorElement->AddNestedElement(redElement); |
| 306 | colorElement->AddNestedElement(greenElement); |
| 307 | colorElement->AddNestedElement(blueElement); |
| 308 | |
| 309 | element->AddNestedElement(colorElement); |
| 310 | |
| 311 | if (this->Blacklist->LookupValue("color") == -1) |
| 312 | { |
| 313 | this->IgnoreArray("color"); |
| 314 | } |
| 315 | } |
| 316 | |
| 317 | //------------------------------------------------------------------------------ |
| 318 | void vtkPhyloXMLTreeWriter::WritePropertyElement( |
no test coverage detected