/////////////////////////////////////////////////////////////////////////////// 
// Copyright (C) 2002-2025, Open Design Alliance (the "Alliance"). 
// All rights reserved. 
// 
// This software and its documentation and related materials are owned by 
// the Alliance. The software may only be incorporated into application 
// programs owned by members of the Alliance, subject to a signed 
// Membership Agreement and Supplemental Software License Agreement with the
// Alliance. The structure and organization of this software are the valuable  
// trade secrets of the Alliance and its suppliers. The software is also 
// protected by copyright law and international treaty provisions. Application  
// programs incorporating this software must include the following statement 
// with their copyright notices:
//   
//   This application incorporates Open Design Alliance software pursuant to a license 
//   agreement with Open Design Alliance.
//   Open Design Alliance Copyright (C) 2002-2025 by Open Design Alliance. 
//   All rights reserved.
//
// By use of this software, its documentation or related materials, you 
// acknowledge and accept the above terms.
///////////////////////////////////////////////////////////////////////////////

#include "OdaCommon.h"

#include "DbAssocGeomDependency.h"
#include "DbAssoc2dConstraintGroup.h"
#include "OdConstrainedRigidSetImpl.h"
#include "DbBlockReference.h"
#include "DbFiler.h"

OdConstrainedRigidSetImpl::OdConstrainedRigidSetImpl()
  : m_unk_Bool(false)
{}

OdResult OdConstrainedRigidSetImpl::init(OdDbObjectId depId)
{
  OdDbAssocGeomDependencyPtr pDep = OdDbAssocGeomDependency::cast(depId.openObject());
  if (pDep.isNull() || !pDep->isDelegatingToOwningAction())
    return eBadObjType;

  OdDbBlockReferencePtr pBref = OdDbBlockReference::cast(pDep->dependentOnObject().openObject());
  if (pBref.isNull())
    return eBadObjType;

  m_groupObjectId = pDep->owningAction();
  OdDbAssoc2dConstraintGroupPtr pGr = OdDbAssoc2dConstraintGroup::cast(m_groupObjectId.openObject());
  if (pGr.isNull())
    return eOwnerNotInDatabase;

  const OdGeMatrix3d matrFromWrd = OdGeMatrix3d::worldToPlane(pGr->getWorkPlane());
  m_matrix = pBref->blockTransform().preMultBy(matrFromWrd);
  m_origMatrix = m_matrix;
  m_origGeomSet = false;

  return eOk;
}

OdResult OdConstrainedRigidSetImpl::addConstrainedGeom(OdConstraintGroupNodeId geomId)
{
  OdDbAssoc2dConstraintGroupPtr pGr = OdDbAssoc2dConstraintGroup::cast(owningConstraintGroupId().openObject());
  OdConstraintGroupNode* pGrNode = pGr->getGroupNodePtr(geomId);
  if (!pGrNode || !pGrNode->isKindOf(OdConstrainedGeometry::desc()))
    return eBadObjType;

  OdConstrainedGeometry* pConsGeom = static_cast<OdConstrainedGeometry*>(pGrNode);
  if (pConsGeom->getOwningRigidSet())
    return eAmbiguousInput;

  m_connections.push_back(geomId);
  return eOk;
}

OdResult OdConstrainedRigidSetImpl::removeConstrainedGeom(OdConstraintGroupNodeId geomId)
{
  return m_connections.remove(geomId) ? eOk : eInvalidInput;
}

int OdConstrainedRigidSetImpl::numOfConstrainedGeoms() const
{
  return m_connections.length();
}

OdConstrainedGeometry* OdConstrainedRigidSetImpl::getConstrainedGeomAt(int index) const
{
  OdDbAssoc2dConstraintGroupPtr pGr = OdDbAssoc2dConstraintGroup::cast(owningConstraintGroupId().openObject());
  OdConstraintGroupNode *pGrNode = pGr->getGroupNodePtr(m_connections[index]);
  
  OdSmartPtr<OdConstrainedGeometry> pRigidSet = OdConstrainedGeometry::cast(pGrNode);
  return pRigidSet;
}

OdGeMatrix3d OdConstrainedRigidSetImpl::transform() const
{
  return m_matrix;
}

OdGeMatrix3d OdConstrainedRigidSetImpl::originalTransform() const
{
  return m_origGeomSet ? m_origMatrix : m_matrix;
}

void OdConstrainedRigidSetImpl::dwgOutFields(OdDbDwgFiler* pFiler) const
{
  OdConstrainedGeometryImpl::dwgOutFields(pFiler);

  pFiler->wrBool(m_unk_Bool);

  int i, j;
  for ( i = 0; i < 4; i++ )
  {
    for ( j = 0; j < 4; j++ )
    {
      pFiler->wrDouble(m_matrix.entry[i][j]);
    }
  }

  int logicalLen;
  pFiler->wrInt32((logicalLen = m_connections.length()));

  const OdConstraintGroupNodeId* pConnections = m_connections.asArrayPtr();
  while (logicalLen--)
  {
    pFiler->wrInt32(*pConnections++);
  }
}

OdResult OdConstrainedRigidSetImpl::dwgInFields(OdDbDwgFiler* pFiler)
{
  OdResult res = OdConstrainedGeometryImpl::dwgInFields(pFiler);
  if ( res != eOk )
    return res;

  m_unk_Bool = pFiler->rdBool();
  ODA_ASSERT(m_unk_Bool == false);

  int i, j;
  for ( i = 0; i < 4; i++ )
  {
    for ( j = 0; j < 4; j++ )
    {
      m_matrix.entry[i][j] = pFiler->rdDouble();
    }
  }

  int logicalLen = pFiler->rdInt32();
  m_connections.resize(logicalLen);

  OdConstraintGroupNodeId* pConnections = m_connections.asArrayPtr();
  while (logicalLen--)
  {
    *pConnections++ = pFiler->rdInt32();
  }

  return res;
}

void OdConstrainedRigidSetImpl::dxfOutFields(OdDbDxfFiler* pFiler) const
{
  OdConstrainedGeometryImpl::dxfOutFields(pFiler);
  pFiler->wrBool(290, m_unk_Bool);

  int i, j;
  for ( i = 0; i < 4; i++ )
  {
    for ( j = 0; j < 4; j++ )
    {
      pFiler->wrDouble(40, m_matrix.entry[i][j]);
    }
  }

  int logicalLen;
  pFiler->wrInt32(90, (logicalLen = m_connections.length()));

  const OdConstraintGroupNodeId* pConnections = m_connections.asArrayPtr();
  while (logicalLen--)
  {
    pFiler->wrInt32(90, *pConnections++);
  }
}

OdResult OdConstrainedRigidSetImpl::dxfInFields(OdDbDxfFiler* pFiler)
{
  OdResult res = OdConstrainedGeometryImpl::dxfInFields(pFiler);
  if ( res != eOk )
    return res;

  NEXT_CODE(290)
  m_unk_Bool = pFiler->rdBool();
  ODA_ASSERT(m_unk_Bool == false);

  int i, j;
  for ( i = 0; i < 4; i++ )
  {
    for ( j = 0; j < 4; j++ )
    {
      NEXT_CODE(40)
      m_matrix.entry[i][j] = pFiler->rdDouble();
    }
  }

  NEXT_CODE(90)
  int logicalLen = pFiler->rdInt32();
  m_connections.resize(logicalLen);

  OdConstraintGroupNodeId* pConnections = m_connections.asArrayPtr();
  while (logicalLen--)
  {
    NEXT_CODE(90)
    *pConnections++ = pFiler->rdInt32();
  }

  return res;
}

void OdConstrainedRigidSetImpl::setTransform(const OdGeMatrix3d &val)
{
  if (!m_origGeomSet)
  {
    m_origMatrix = m_matrix;
    m_origGeomSet = true;
  }
  for ( int i = 0; i < 4; i++ )
    for ( int j = 0; j < 4; j++ )
      m_matrix.entry[i][j] = val.entry[i][j];
}

OdResult OdConstrainedRigidSetImpl::setDataFromGeomDependency(const OdDbObjectId geomDependencyId,
                                             const OdGeMatrix3d& matrFromWrd)
{
  OdDbAssocGeomDependencyPtr pDep = OdDbAssocGeomDependency::cast(geomDependencyId.openObject());
  if (pDep.isNull())
    return eNullObjectPointer;

  OdDbObjectPtr pObj = pDep->dependentOnObject().openObject();
  if (pObj.isNull())
    return eNullObjectPointer;
  if (pObj->isErased())
    return eWasErased;

  if (!pObj.isNull() && pObj->isKindOf(OdDbBlockReference::desc()) )
  {
    if (!m_origGeomSet)
    {
      m_origMatrix = m_matrix;
      m_origGeomSet = true;
    }

    OdDbBlockReferencePtr pBlockRef = OdDbBlockReference::cast(pObj);
    setTransform(pBlockRef->blockTransform().preMultBy(matrFromWrd));
    return eOk;
  }
  else
  {
    ODA_FAIL();
  }

  return eNotImplementedYet;
}

OdResult OdConstrainedRigidSetImpl::updateDbObject(OdDbObject *pObj, OdDbAssocGeomDependency* /*pDep*/, const OdGeMatrix3d& matrToWrd)
{
  OdDbBlockReferencePtr pBlockRef = OdDbBlockReference::cast(pObj);
  if (pBlockRef.get())
  {
    OdGeMatrix3d rsOldTr = originalTransform();
    OdGeMatrix3d rsNewTr = transform();

    m_origGeomSet = false;

    OdGeMatrix3d oldInverted;
    if (rsOldTr.inverse(oldInverted, OdGeTol().equalPoint()))
    {
      OdGeMatrix3d solverTr = rsNewTr * oldInverted;
      if (!solverTr.isEqualTo(OdGeMatrix3d::kIdentity))
      {
        OdGeMatrix3d desiredTr = pBlockRef->blockTransform();
        OdGeMatrix3d desiredInverted;
        if (!desiredTr.inverse(desiredInverted, OdGeTol().equalPoint()))
        {
          ODA_FAIL_ONCE();
          return eOk;
        }
        OdGeMatrix3d backTr = rsOldTr * desiredInverted; // moves to original position

        // Apply solver transformation both to the block reference and to the entities (attributes and xdata) associated with it
        solverTr.preMultBy(backTr);
        pBlockRef->transformBy(solverTr);

        // We still need to set the transformation to preserve precision
        OdGeMatrix3d blockRefTr = matrToWrd * rsNewTr;
        pBlockRef->setBlockTransform(blockRefTr);
      }
    }
    else
    {
      ODA_FAIL_ONCE(); // unreachable
    }

    return eOk;
  }
  else
  {
    ODA_FAIL();
  }

  return eNotImplementedYet;
}