#include <Message.hxx>
#include <STEPCAFControl_Reader.hxx>
#include <TDataStd_Name.hxx>
#include <TopExp_Explorer.hxx>
#include <XCAFApp_Application.hxx>
#include <XCAFDoc_ColorTool.hxx>
#include <XCAFPrs_DocumentExplorer.hxx>
#include <XCAFDoc_VisMaterial.hxx>
#include <XCAFDoc_ShapeTool.hxx>

static TCollection_AsciiString getLabelName (const TDF_Label& theLabel)
{
  Handle(TDataStd_Name) aNodeName;
  return theLabel.FindAttribute (TDataStd_Name::GetID(), aNodeName)
       ? TCollection_AsciiString (aNodeName->Get())
       : TCollection_AsciiString();
}

static const int THE_NB_COLOR_CAT = 3;
static const XCAFDoc_ColorType THE_COLOR_CATS     [THE_NB_COLOR_CAT] = { XCAFDoc_ColorGen, XCAFDoc_ColorSurf, XCAFDoc_ColorCurv };
static const char*             THE_COLOR_CAT_NAMES[THE_NB_COLOR_CAT] = { "Generic color: ", "Surface color: ", "Curves color: " };

int main (int theNbArgs, char** theArgVec)
{
  Handle(XCAFApp_Application) anApp = XCAFApp_Application::GetApplication();
  Handle(TDocStd_Document) aDoc;
  anApp->NewDocument ("BinXCAF", aDoc);
  STEPCAFControl_Reader aReader;
  if (!aReader.Perform (theArgVec[1], aDoc))
  //if (!aReader.Perform ("c:/11PI0391_F001-Dima_x_B006_B007.stp", aDoc))
  {
    Message::SendFail() << "STEP file '" << theArgVec[1] << "' reading failed";
    return 1;
  }

  for (XCAFPrs_DocumentExplorer aDocExp (aDoc, XCAFPrs_DocumentExplorerFlags_None); aDocExp.More(); aDocExp.Next())
  {
    const XCAFPrs_DocumentNode& aNode = aDocExp.Current();
    TCollection_AsciiString anIndent (aDocExp.CurrentDepth() * 2, ' ');
    TopoDS_Shape aShape;
    XCAFDoc_ShapeTool::GetShape (aNode.RefLabel, aShape);
    std::cout << anIndent << aNode.Id << " " << (!aShape.IsNull() ? TopAbs::ShapeTypeToString (aShape.ShapeType()) : "") << "\n";

    std::cout << anIndent << "Reference name: '" << getLabelName (aNode.RefLabel) << "'\n";
    if (aNode.RefLabel != aNode.Label)
    {
      std::cout << anIndent << "Instance name:  '" << getLabelName (aNode.Label) << "'\n";
    }

    const XCAFPrs_Style& aStyle = aNode.Style;
    if (aStyle.IsSetColorSurf())
    {
      std::cout << anIndent << "Surface color:  "
                << Quantity_ColorRGBA::ColorToHex (aStyle.GetColorSurfRGBA())
                << " [" << Quantity_Color::StringName (aStyle.GetColorSurf().Name()) << "]"
                << "\n";
    }
    if (aStyle.IsSetColorCurv())
    {
      std::cout << anIndent << "Curves color:   "
                << Quantity_Color::ColorToHex (aStyle.GetColorCurv())
                << " [" << Quantity_Color::StringName (aStyle.GetColorCurv().Name()) << "]"
                << "\n";
    }
    if (!aStyle.Material().IsNull())
    {
      std::cout << anIndent << "Material:       "
                << Quantity_ColorRGBA::ColorToHex (aStyle.Material()->BaseColor())
                << " [" << Quantity_Color::StringName (aStyle.Material()->BaseColor().GetRGB().Name()) << "]"
                << "\n";
    }

    if (!aNode.IsAssembly)
    {
      TDF_LabelSequence aLabSeq;
      XCAFDoc_ShapeTool::GetSubShapes (aNode.RefLabel, aLabSeq);
      Quantity_ColorRGBA aColor;
      anIndent = TCollection_AsciiString ((aDocExp.CurrentDepth() + 1) * 2, ' ');
      int aSubShapeIndex = 0;
      for (TDF_LabelSequence::Iterator aShapeIter (aLabSeq); aShapeIter.More(); aShapeIter.Next())
      {
        const TDF_Label& aSubShapeLab = aShapeIter.Value();
        TopoDS_Shape aSubShape;
        if (!XCAFDoc_ShapeTool::GetShape (aSubShapeLab, aSubShape)) { continue; }
        std::cout << anIndent << "Subshape " << TopAbs::ShapeTypeToString (aSubShape.ShapeType()) << " " << ++aSubShapeIndex << "\n";
        for (int aCatIter = 0; aCatIter < THE_NB_COLOR_CAT; ++aCatIter)
        {
          if (aDocExp.ColorTool()->GetColor (aSubShapeLab, THE_COLOR_CATS[aCatIter], aColor))
          {
            std::cout << anIndent << THE_COLOR_CAT_NAMES[aCatIter]
                      << Quantity_ColorRGBA::ColorToHex (aColor)
                      << " [" << Quantity_Color::StringName (aColor.GetRGB().Name()) << "]"
                      << "\n";
          }
        }
      }

      /*for (TopExp_Explorer aFaceExp (aShape, TopAbs_FACE); aFaceExp.More(); aFaceExp.Next())
      {
        TopoDS_Shape aSubShape = aFaceExp.Current();
        std::cout << anIndent << "Subshape " << TopAbs::ShapeTypeToString (aSubShape.ShapeType()) << " " << ++aSubShapeIndex << "\n";
        for (int aCatIter = 0; aCatIter < THE_NB_COLOR_CAT; ++aCatIter)
        {
          if (aDocExp.ColorTool()->GetColor (aSubShape, THE_COLOR_CATS[aCatIter], aColor))
          {
            std::cout << anIndent << THE_COLOR_CAT_NAMES[aCatIter]
                      << Quantity_ColorRGBA::ColorToHex (aColor)
                      << " [" << Quantity_Color::StringName (aColor.GetRGB().Name()) << "]"
                      << "\n";
          }
        }
      }*/
    }

    std::cout << "\n";
  }

  return 0;
}

0:1:1:1. COMPOUND
Reference name: '11PI0391 F001-Dima x B006 B007'

  0:1:1:1./0:1:1:1:1. SOLID
  Reference name: '11PI0391 B006-Stecca 1'
  Instance name:  '11PI0391 B006-Stecca 1.1'
  Surface color:  #FFFF00FF [YELLOW]
    Subshape FACE 1
    Surface color: #008000FF [GREEN4]
    Subshape FACE 2
    Surface color: #008000FF [GREEN4]
    Subshape FACE 3
    Surface color: #FFFFFFFF [WHITE]
    Subshape FACE 4
    Surface color: #FFFFFFFF [WHITE]
    Subshape FACE 5
    Surface color: #FFFFFFFF [WHITE]
    Subshape FACE 6
    Surface color: #FFFFFFFF [WHITE]
...