/////////////////////////////////////////////////////////////////////////////// 
// 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 "StdAfx.h"
#include "DbEntityGripPoints.h"
#include "DbEntity.h"
#include "OdDbAssocPersSubentIdPE.h"
#include <DbAssocActionModule.h>


OdResult OdDbEntityGripPointsPE::getGripPoints( const OdDbEntity* pEnt, OdGePoint3dArray& gripPoints )const
{
  OdGeExtents3d exts;
  OdResult res = pEnt->getGeomExtents(exts);
  if (res == eOk)
  {
    unsigned int size = gripPoints.size();
    gripPoints.resize( size + 1 );
    gripPoints[size] =
        exts.minPoint() + (exts.maxPoint() - exts.minPoint()) / 2.;
  }
  return res;
}

OdResult OdDbEntityGripPointsPE::moveGripPointsAt( OdDbEntity* pEnt, const OdIntArray& indices, const OdGeVector3d& offset )
{
  if ( indices.empty() )
    return eOk;
  return pEnt->transformBy( OdGeMatrix3d::translation(offset) );
}

OdResult OdDbEntityGripPointsPE::getStretchPoints( const OdDbEntity* pEnt, OdGePoint3dArray& stretchPoints ) const
{
  return getGripPoints( pEnt, stretchPoints );
}


OdResult OdDbEntityGripPointsPE::moveStretchPointsAt( OdDbEntity* pEnt, const OdIntArray& indices, const OdGeVector3d& offset )
{
  return moveGripPointsAt( pEnt, indices, offset );
}

OdResult OdDbEntityGripPointsPE::getOsnapPoints(const OdDbEntity* ent,
  OdDb::OsnapMode osnapMode, 
  OdGsMarker /*gsSelectionMark*/,
  const OdGePoint3d& /*pickPoint*/,
  const OdGePoint3d& /*lastPoint*/,
  const OdGeMatrix3d& /*xWorldToEye*/,
  OdGePoint3dArray& snapPoints) const
{
  OdResult res = eOk;
  OdDbAssocActionModulePtr aam = odrxLoadApp(DbAssocActionModuleName);
  if (aam.isNull())
    return eNoInterface;

  OdRxClass* apspeClss = odrxGetClassDesc(L"AcDbAssocPersSubentIdPE");
  if (!apspeClss)
    return eNoInterface;
  OdDbAssocPersSubentIdPEPtr assocPersSubentIdPE = static_cast<OdDbAssocPersSubentIdPE*>(ent->isA()->getX(apspeClss).get());
  if (assocPersSubentIdPE.isNull())
    return eNoInterface;

  switch (osnapMode) {
  case OdDb::kOsModeEnd:
  {
    OdArray<OdDbSubentId> subentIds;
    res = assocPersSubentIdPE->getAllSubentities(ent, OdDb::kVertexSubentType, subentIds);
    snapPoints.reserve(snapPoints.size()+subentIds.size());
    for (const OdDbSubentId& subentId : subentIds)
      assocPersSubentIdPE->getVertexSubentityGeometry(ent, subentId, *snapPoints.append());
  }
  break;
  case OdDb::kOsModeMid:
  {
    OdArray<OdDbSubentId> subentIds;
    res = assocPersSubentIdPE->getAllSubentities(ent, OdDb::kEdgeSubentType, subentIds);
    snapPoints.reserve(snapPoints.size() + subentIds.size());
    for (const OdDbSubentId& subentId : subentIds)
    {
      OdGeCurve3d* pEdgeCurve(nullptr);
      res = assocPersSubentIdPE->getEdgeSubentityGeometry(ent, subentId, pEdgeCurve);
      if (!pEdgeCurve)
        return res;

      std::unique_ptr<OdGeCurve3d> pCurve(pEdgeCurve);
      OdGePoint3d midPoint;
      if (pEdgeCurve->isKindOf(OdGe::kLinearEnt3d) && pEdgeCurve->hasMidPoint(midPoint))
        snapPoints.append(midPoint);
    }   
  }
  break;
  default:
    res = eNotImplementedYet;
    break;
  }

  return res;
}