SchemaGraphicObject.cpp

Go to the documentation of this file.

 
#include <QApplication>
#include <QCursor>
#include <QEvent>
#include <QGraphicsScene>
#include <QGraphicsSceneHoverEvent>
#include <QGraphicsView>
#include <QPainter>
#include <QToolTip>
#include"oks/method.hpp"
#include "dbe/SchemaClassEditor.hpp"
#include "dbe/SchemaGraphicObject.hpp"
#include "dbe/SchemaGraphicSegmentedArrow.hpp"
#include "dbe/SchemaGraphicsScene.hpp"
#include "dbe/SchemaKernelWrapper.hpp"
#include "dbe/SchemaStyle.hpp"
 
using namespace dunedaq::oks;
 
dbse::SchemaGraphicObject::SchemaGraphicObject ( QString & ClassName,
                                                 SchemaGraphicsScene* scene,
                                                 QGraphicsObject * parent )
  : QGraphicsObject ( parent ),
    m_scene(scene),
    LineOffsetX ( 0 ),
    LineOffsetY ( 0 )
{
  setAcceptHoverEvents(true);
 
  setFlag ( ItemIsMovable );
  setFlag ( ItemSendsGeometryChanges, true );
  setFlag ( ItemSendsScenePositionChanges, true );
 
  connect ( &KernelWrapper::GetInstance(), SIGNAL ( ClassUpdated ( QString ) ), this,
            SLOT ( UpdateObject ( QString ) ) );
  connect ( &KernelWrapper::GetInstance(), SIGNAL ( ClassRemoved ( QString ) ), this,
            SLOT ( RemoveObject ( QString ) ) );
  m_class_info = KernelWrapper::GetInstance().FindClass ( ClassName.toStdString() );
  m_class_object_name = ClassName;
 
  GetInfo();
 
  set_font();
}
 
dbse::SchemaGraphicObject::~SchemaGraphicObject()
{
}
 
 
 
void dbse::SchemaGraphicObject::hoverEnterEvent ( QGraphicsSceneHoverEvent* he) {
  // std::cout << "hover event at scenePos" << he->scenePos().x() << "," << he->scenePos().y()
  //           << ", pos=" << he->pos().x() << "," << he->pos().y() << "\n";
 
  auto text = m_class_info->get_name();
  auto desc = m_class_info->get_description();
  if (!desc.empty()) {
    text += "\n" + desc;
  }
  text += "\n-----------";
 
  std::string attr_descriptions;
  auto attributes = m_class_info->direct_attributes();
  if (attributes != nullptr) {
    for (auto attr: *attributes) {
      if (!attr->get_description().empty()) {
        attr_descriptions += "\n " + attr->get_name() + ": " +
          attr->get_description();
      }
    }
  }
  std::string rel_descriptions;
  auto relationships = m_class_info->direct_relationships();
  if (relationships != nullptr) {
    for (auto rel: *relationships) {
      if (!rel->get_description().empty()) {
        rel_descriptions += "\n " + rel->get_name() + ": " +
          rel->get_description();
      }
    }
  }
  std::string method_descriptions;
  auto methods = m_class_info->direct_methods();
  if (methods != nullptr) {
    for (auto method: *methods) {
      if (!method->get_description().empty()) {
        method_descriptions += "\n " + method->get_name() + ": " +
          method->get_description();
      }
    }
  }
 
  std::string sep="\n-----";
  if (!attr_descriptions.empty()) {
    text += "\nAttributes:" + attr_descriptions;
    if (!(rel_descriptions.empty() && method_descriptions.empty())) {
      text += sep;
    }
  }
  if (!rel_descriptions.empty()) {
    text += "\nRelationships:" + rel_descriptions;
    if (!method_descriptions.empty()) {
      text += sep;
    }
  }
  if (method_descriptions != "") {
    text += "\nMethods:" + method_descriptions;
  }
  //std::cout << "text=<" << text << ">\n"; std::cout.flush();
  QToolTip::showText( he->screenPos(),
                      QString::fromStdString(text),
                      nullptr, QRect(), 10000);
  he->ignore();
}
void dbse::SchemaGraphicObject::hoverLeaveEvent ( QGraphicsSceneHoverEvent* he) {
  // std::cout << "hover leave\n";
  QToolTip::hideText();
  he->ignore();
}
 
void dbse::SchemaGraphicObject::mouseDoubleClickEvent ( QGraphicsSceneMouseEvent* ) {
  QString class_name = QString::fromStdString ( m_class_info->get_name() );
  SchemaClassEditor::launch(class_name);
}
 
OksClass * dbse::SchemaGraphicObject::GetClass() const
{
  return m_class_info;
}
 
QString dbse::SchemaGraphicObject::GetClassName() const
{
  return m_class_object_name;
}
 
void dbse::SchemaGraphicObject::GetInfo()
{
  m_class_attributes.clear();
  m_class_relationhips.clear();
  m_class_methods.clear();
 
  std::list<OksAttribute *> direct_attributes = {};
  if (m_class_info->direct_attributes()) {
    direct_attributes = *m_class_info->direct_attributes();
  }
  std::list<OksMethod *> direct_methods = {};
  if (m_class_info->direct_methods()) direct_methods = *m_class_info->direct_methods();
 
  std::list<OksRelationship *> direct_relationships = {};
  if (m_class_info->direct_relationships()) direct_relationships = *m_class_info->direct_relationships();
 
  std::list<OksAttribute *> all_attributes = {};
  if (m_class_info->all_attributes()) all_attributes = *m_class_info->all_attributes();
 
  std::list<OksMethod *> all_methods = {};
  if (m_class_info->all_methods()) all_methods = *m_class_info->all_methods();
 
  std::list<OksRelationship *> all_relationships = {};
  if (m_class_info->all_relationships()) all_relationships = *m_class_info->all_relationships();
 
  // Prepare indirect relationship list
  std::list<OksAttribute *> indirect_attributes = all_attributes;
  std::list<OksMethod *> indirect_methods = all_methods;
  std::list<OksRelationship *> indirect_relationships = all_relationships;
 
  for ( OksAttribute * attribute : direct_attributes ) {
    indirect_attributes.remove(attribute);
  }
 
  for ( OksMethod * method : direct_methods ) {
    indirect_methods.remove(method);
  } 
 
  for ( OksRelationship * relationship : direct_relationships ) {
    indirect_relationships.remove(relationship);
  } 
 
  std::map<OksRelationship::CardinalityConstraint, std::string> m = {
    {OksRelationship::Zero, "0"},
    {OksRelationship::One, "1"},
    {OksRelationship::Many, "n"}
  };
 
  for ( OksAttribute * attribute : direct_attributes )
  {
    QString AttributeString (
      QString::fromStdString ( attribute->get_name() ) + " : "
      + QString::fromStdString ( attribute->get_type() ) + (attribute->get_is_multi_values() ? "[]" : "") );
    m_class_attributes.append ( AttributeString );
 
    QString value;
    if (!attribute->get_init_value().empty()) {
      value = QString::fromStdString (" = " + attribute->get_init_value());
    }
    m_class_attribute_values.append (value);
  }
 
  for ( OksRelationship * relationship : direct_relationships )
  {
    QString relationship_string ( QString::fromStdString ( relationship->get_name() ) + " : " 
    + QString::fromStdString ( relationship->get_type() ) + " - "
    + QString::fromStdString ( m[ relationship->get_low_cardinality_constraint() ] ) + ":"
    + QString::fromStdString ( m[ relationship->get_high_cardinality_constraint() ] )  );
    m_class_relationhips.append ( relationship_string );
  }
 
  for ( OksMethod * Method : direct_methods )
  {
    QString MethodString ( QString::fromStdString ( Method->get_name() ) + "()" );
    m_class_methods.append ( MethodString );
  }
 
  for ( OksAttribute * attribute : indirect_attributes )
  {
    QString attribute_string (
      QString::fromStdString ( attribute->get_name() ) + " : "
      + QString::fromStdString ( attribute->get_type() ) );
    m_class_inherited_attributes.append ( attribute_string );
  }
 
  for ( OksRelationship * relationship : indirect_relationships )
  {
    QString relationship_string ( QString::fromStdString ( relationship->get_name() ) + " : " 
                                  + QString::fromStdString ( relationship->get_type() ) + " - "
                                  + QString::fromStdString ( m[ relationship->get_low_cardinality_constraint() ] ) + ":"
                                  + QString::fromStdString ( m[ relationship->get_high_cardinality_constraint() ] )  );
    m_class_inherited_relationhips.append ( relationship_string );
  }
 
  for ( OksMethod * method : indirect_methods )
  {
    QString method_string ( QString::fromStdString ( method->get_name() ) + "()" );
    m_class_inherited_methods.append ( method_string );
  }
}
 
QRectF dbse::SchemaGraphicObject::boundingRect() const
{
  double SpaceX = 3;
  double TotalBoundingHeight = 0;
  double TotalBoundingWidth = 0;
 
  QFontMetrics FontMetrics ( m_font );
 
  TotalBoundingHeight += SpaceX * 5;
  TotalBoundingHeight += FontMetrics.boundingRect ( m_class_object_name ).height();
  TotalBoundingWidth += FontMetrics.boundingRect ( m_class_object_name ).width();
 
 
  for ( int entry=0; entry<m_class_attributes.size(); entry++)
  {
    QString attribute_name = m_class_attributes[entry];
    if (m_scene->show_defaults()) {
      attribute_name.append(m_class_attribute_values[entry]);
    }
    TotalBoundingHeight += FontMetrics.boundingRect ( attribute_name ).height();
 
    if ( FontMetrics.boundingRect (attribute_name).width() > TotalBoundingWidth )
    {
      TotalBoundingWidth =
        FontMetrics.boundingRect (attribute_name).width();
    }
  }
 
  if (m_scene->inherited_properties_visible()) {
    for ( auto & AttributeName : m_class_inherited_attributes )
    {
      TotalBoundingHeight += FontMetrics.boundingRect ( AttributeName ).height();
 
      if ( FontMetrics.boundingRect ( AttributeName ).width() > TotalBoundingWidth )
        TotalBoundingWidth =
          FontMetrics.boundingRect ( AttributeName ).width();
    }
  }
 
  for ( auto & relationship_name : m_class_relationhips )
  {
    TotalBoundingHeight += FontMetrics.boundingRect ( relationship_name ).height();
 
    if ( FontMetrics.boundingRect ( relationship_name ).width() > TotalBoundingWidth )
      TotalBoundingWidth =
        FontMetrics.boundingRect ( relationship_name ).width();
  }
 
  if (m_scene->inherited_properties_visible()) {
    for ( auto & relationship_name : m_class_inherited_relationhips )
    {
      TotalBoundingHeight += FontMetrics.boundingRect ( relationship_name ).height();
 
      if ( FontMetrics.boundingRect ( relationship_name ).width() > TotalBoundingWidth )
        TotalBoundingWidth =
          FontMetrics.boundingRect ( relationship_name ).width();
    }
  }
 
  for ( auto & MethodName : m_class_methods )
  {
    TotalBoundingHeight += FontMetrics.boundingRect ( MethodName ).height();
 
    if ( FontMetrics.boundingRect ( MethodName ).width() > TotalBoundingWidth )
      TotalBoundingWidth =
        FontMetrics.boundingRect ( MethodName ).width();
  }
 
  if (m_scene->inherited_properties_visible()) {
    for ( auto & MethodName : m_class_inherited_methods )
    {
      TotalBoundingHeight += FontMetrics.boundingRect ( MethodName ).height();
 
      if ( FontMetrics.boundingRect ( MethodName ).width() > TotalBoundingWidth )
        TotalBoundingWidth =
          FontMetrics.boundingRect ( MethodName ).width();
    }
  }
 
  TotalBoundingWidth += 15;
  return QRectF ( 0, 0, TotalBoundingWidth, TotalBoundingHeight );
}
 
QPainterPath dbse::SchemaGraphicObject::shape() const
{
  QPainterPath path;
  path.addRect ( boundingRect() );
  return path;
}
 
void dbse::SchemaGraphicObject::paint ( QPainter * painter,
                                        const QStyleOptionGraphicsItem * option,
                                        QWidget * widget )
{
  Q_UNUSED ( widget )
  Q_UNUSED ( option )
 
  double SpaceX = 3;
  double SpaceY = 3;
 
 
  QColor colour;
  QColor background;
  if (m_highlight_class) {
    colour = SchemaStyle::get_color("foreground", "highlight");
    background = SchemaStyle::get_color("background", "highlight");
  }
  else if (m_scene->highlight_active() &&
           (m_class_info->get_file()->get_full_file_name()
            == KernelWrapper::GetInstance().GetActiveSchema())) {
    colour = SchemaStyle::get_color("foreground", "active_file");
    background = SchemaStyle::get_color("background", "active_file");
  }
  else {
    colour = SchemaStyle::get_color("foreground", "default");
    background = SchemaStyle::get_color("background", "default");
  }
 
  set_font();
 
  const QPen bounding_box_pen = QPen(colour, 2.5);
  const QPen inner_line_pen = QPen(colour, 1.5);
 
  painter->setFont ( m_font );
  painter->setPen ( bounding_box_pen );
  painter->setBackground(background);
  //painter->setBackgroundMode(Qt::OpaqueMode);
  painter->drawRect ( boundingRect() );
 
  const QFontMetrics FontMetrics = painter->fontMetrics();
  const QRectF ClassNameBoundingRect = FontMetrics.boundingRect ( m_class_object_name );
  const QRectF ObjectBoundingRect = boundingRect();
 
  double HeightOffset = ClassNameBoundingRect.height() + SpaceY;
  const auto ClassNamePosition = QPointF(
    (ObjectBoundingRect.width() - ClassNameBoundingRect.width() ) / 2,
    ClassNameBoundingRect.height());
  painter->drawText ( ClassNamePosition, m_class_object_name );
  painter->drawLine ( 0, HeightOffset, ObjectBoundingRect.width(), HeightOffset );
 
  for ( int entry=0; entry<m_class_attributes.size(); entry++)
  {
    QString attribute_name = m_class_attributes[entry];
    if (m_scene->show_defaults()) {
      attribute_name.append(m_class_attribute_values[entry]);
    }
    const QRectF AttributeBoundingRect = FontMetrics.boundingRect (attribute_name);
    HeightOffset += AttributeBoundingRect.height();
    painter->drawText ( QPointF(SpaceX, HeightOffset), attribute_name );
  }
 
  if (m_scene->inherited_properties_visible()) {
    painter->setPen ( SchemaStyle::get_color("foreground", "inherited") );
    for ( QString & AttributeName : m_class_inherited_attributes )
    {
      const QRectF AttributeBoundingRect = FontMetrics.boundingRect ( AttributeName );
      HeightOffset += AttributeBoundingRect.height();
      painter->drawText ( QPointF(SpaceX, HeightOffset), AttributeName );
    }
    painter->setPen ( colour );
  }
 
  HeightOffset += SpaceY;
  painter->setPen ( inner_line_pen );
  painter->drawLine ( 0, HeightOffset, ObjectBoundingRect.width(), HeightOffset );
 
  for (const QString & relationship_name : m_class_relationhips )
  {
    const QRectF relationship_bounding_rect = FontMetrics.boundingRect ( relationship_name );
    HeightOffset += relationship_bounding_rect.height();
    painter->drawText ( QPointF(SpaceX, HeightOffset), relationship_name );
  }
 
  if (m_scene->inherited_properties_visible()) {
    painter->setPen ( SchemaStyle::get_color("foreground", "inherited") );
    for ( const QString & relationship_name : m_class_inherited_relationhips )
    {
      const QRectF relationship_bounding_rect = FontMetrics.boundingRect ( relationship_name );
      HeightOffset += relationship_bounding_rect.height();
      painter->drawText ( QPointF(SpaceX, HeightOffset), relationship_name );
    }
    painter->setPen ( colour );
  }
 
 
  HeightOffset += SpaceY;
  painter->setPen ( inner_line_pen );
  painter->drawLine ( 0, HeightOffset, ObjectBoundingRect.width(), HeightOffset );
  
  for ( const QString & MethodName : m_class_methods )
  {
    const QRectF AttributeBoundingRect = FontMetrics.boundingRect ( MethodName );
    HeightOffset += AttributeBoundingRect.height();
    painter->drawText ( QPointF(SpaceX, HeightOffset), MethodName );
  }
 
  if (m_scene->inherited_properties_visible()) {
    painter->setPen ( SchemaStyle::get_color("foreground", "inherited") );
    for ( const QString & MethodName : m_class_inherited_methods )
    {
      const QRectF AttributeBoundingRect = FontMetrics.boundingRect ( MethodName );
      HeightOffset += AttributeBoundingRect.height();
      painter->drawText (QPointF(SpaceX, HeightOffset), MethodName );
    }
    painter->setPen ( colour );
  }
}
 
void dbse::SchemaGraphicObject::AddArrow ( SchemaGraphicSegmentedArrow * Arrow )
{
  m_arrows.append ( Arrow );
}
 
void dbse::SchemaGraphicObject::RemoveArrow ( SchemaGraphicSegmentedArrow * Arrow )
{
  const int index = m_arrows.indexOf ( Arrow );
 
  if ( index != -1 )
  {
    m_arrows.removeAt ( index );
  }
}
 
void dbse::SchemaGraphicObject::RemoveArrows()
{
  for ( SchemaGraphicSegmentedArrow * arrow : m_arrows )
  {
    arrow->GetStartItem()->RemoveArrow ( arrow );
    arrow->GetEndItem()->RemoveArrow ( arrow );
    scene()->removeItem ( arrow );
  }
}
 
bool dbse::SchemaGraphicObject::HasArrow ( SchemaGraphicObject * Dest ) const
{
  if ( m_arrows.isEmpty() )
  {
    return false;
  }
 
  for ( SchemaGraphicSegmentedArrow * Arrow : m_arrows )
  {
    SchemaGraphicObject * ArrowSource = Arrow->GetStartItem();
    SchemaGraphicObject * ArrowDest = Arrow->GetEndItem();
 
    if ( ( ArrowSource == this ) && ( ArrowDest == Dest ) )
    {
      return true;
    }
  }
 
  return false;
}
 
void dbse::SchemaGraphicObject::update_arrows()
{
  for ( SchemaGraphicSegmentedArrow * arrow : m_arrows )
  {
    arrow->UpdatePosition();
  }
}
 
void dbse::SchemaGraphicObject::set_font()
{
  if (m_highlight_class) {
    m_font = SchemaStyle::get_font("highlight");
  }
  else {
    m_font = SchemaStyle::get_font("default");
  }
  if (m_class_info->get_is_abstract() && m_scene->highlight_abstract()) {
    // Use selected font but with style from abstract class font 
    m_font.setStyle(SchemaStyle::get_font("abstract").style());
  }
}
void dbse::SchemaGraphicObject::toggle_highlight_class()
{
  m_highlight_class = !m_highlight_class;
  set_font();
  update_arrows();
}
 
 
QVariant dbse::SchemaGraphicObject::itemChange ( GraphicsItemChange change,
                                                 const QVariant & value )
{
  if ( change == ItemPositionChange ) 
    for ( SchemaGraphicSegmentedArrow * arrow : m_arrows )
    {
      arrow->UpdatePosition();
    }
 
  return value;
}
 
void dbse::SchemaGraphicObject::UpdateObject ( QString Name )
{
  if ( Name != m_class_object_name )
  {
    return;
  }
 
  GetInfo();
  SchemaGraphicsScene * Scene = dynamic_cast<SchemaGraphicsScene *> ( scene() );
 
  if ( Scene )
  {
    QStringList ClassesList;
    ClassesList.append ( Name );
 
    QList<QPointF> ClassesPositions;
    QPointF ClassPosition ( this->scenePos() );
    ClassesPositions.append ( ClassPosition );
 
    Scene->RemoveClassObject ( this );
    Scene->AddItemsToScene ( QStringList ( m_class_object_name ), ClassesPositions );
  }
 
  update();
}
 
void dbse::SchemaGraphicObject::RemoveObject ( QString Name )
{
  if ( Name != m_class_object_name )
  {
    return;
  }
 
  SchemaGraphicsScene * Scene = dynamic_cast<SchemaGraphicsScene *> ( scene() );
 
  if ( Scene )
  {
    Scene->RemoveClassObject ( this );
  }
}