#include <TopExp_Explorer.hxx>
#include <TopoDS_Wire.hxx>
#include <TopoDS_Vertex.hxx>
+#include <TopoDS_Edge.hxx>
#include <BRep_Tool.hxx>
#include <Precision.hxx>
-
+#include <Handle_Geom_Curve.hxx>
+#include <Handle_Geom_Line.hxx>
+#include <Handle_Geom_TrimmedCurve.hxx>
+#include <Geom_TrimmedCurve.hxx>
HYDROGUI_ExportFileOp::HYDROGUI_ExportFileOp( HYDROGUI_Module* theModule )
: HYDROGUI_Operation( theModule )
{
HYDROGUI_Operation::startOperation();
- QString aFilter( tr("SHP_FILTER") ); //temp ext-n; replace with filter; TODO
+ QString aFilter( tr("SHP_FILTER") );
Handle(HYDROData_PolylineXY) aPolyXY;
Handle(HYDROData_Polyline3D) aPoly3D;
}
if (!aPolyXYSeq.IsEmpty() && !aPoly3DSeq.IsEmpty())
- SUIT_MessageBox::warning( module()->getApp()->desktop(), "Export Polyline", "Cannot export polylines of different kind");
+ SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "EXPORT_TO_SHAPE_FILE" ), tr("ERROR_POLYLINES_OF_DIFF_KIND"));
else
{
- QString aFileName = SUIT_FileDlg::getFileName( module()->getApp()->desktop(), "", aFilter, tr( "EXPORT_TO_SHAPE_FILE" ), false );
+ QString aName = "";
+ if (aPolyXYSeq.Size() == 1 && aPoly3DSeq.IsEmpty())
+ aName = aPolyXYSeq(1)->GetName();
+ if (aPoly3DSeq.Size() == 1 && aPolyXYSeq.IsEmpty())
+ aName = aPoly3DSeq(1)->GetName();
+ if (aLCSeq.Size() == 1 && aPolyXYSeq.IsEmpty() && aPoly3DSeq.IsEmpty())
+ aName = aLCSeq(1)->GetName();
+ QString aFileName = SUIT_FileDlg::getFileName( module()->getApp()->desktop(), aName, aFilter, tr( "EXPORT_TO_SHAPE_FILE" ), false );
if (!aFileName.isEmpty())
{
SHPHandle hSHPHandle;
+ QStringList aNonExpList;
if (!aPolyXYSeq.IsEmpty() && aPoly3DSeq.IsEmpty())
{
- hSHPHandle = SHPCreate( aFileName.toAscii().data(), SHPT_ARC );
+ hSHPHandle = SHPCreate( aFileName.toAscii().data(), SHPT_ARC );
for (int i = 1; i <= aPolyXYSeq.Size(); i++)
- WriteObjectPolyXY(hSHPHandle, aPolyXYSeq(i));
+ if (WriteObjectPolyXY(hSHPHandle, aPolyXYSeq(i)) != 1)
+ aNonExpList.append(aPolyXYSeq(i)->GetName());
+
}
else if (aPolyXYSeq.IsEmpty() && !aPoly3DSeq.IsEmpty())
{
hSHPHandle = SHPCreate( aFileName.toAscii().data(), SHPT_ARCZ );
for (int i = 1; i <= aPoly3DSeq.Size(); i++)
- WriteObjectPoly3D(hSHPHandle, aPoly3DSeq(i));
+ if (WriteObjectPoly3D(hSHPHandle, aPoly3DSeq(i)) != 1)
+ aNonExpList.append(aPoly3DSeq(i)->GetName());
}
else if (aPolyXYSeq.IsEmpty() && aPoly3DSeq.IsEmpty() && !aLCSeq.IsEmpty())
{
hSHPHandle = SHPCreate( aFileName.toAscii().data(), SHPT_POLYGON );
for (int i = 1; i <= aLCSeq.Size(); i++)
- WriteObjectLC(hSHPHandle, aLCSeq(i));
+ if (WriteObjectLC(hSHPHandle, aLCSeq(i)) != 1)
+ aNonExpList.append(aLCSeq(i)->GetName());
}
SHPClose( hSHPHandle );
+ if (!aNonExpList.empty())
+ {
+ QString aMessage = tr("CANNOT_BE_EXPORTED") + "\n";
+ foreach (QString anNonExpEntName, aNonExpList)
+ aMessage += anNonExpEntName + "\n";
+ SUIT_MessageBox::warning( module()->getApp()->desktop(), tr( "EXPORT_TO_SHAPE_FILE" ), aMessage);
+ }
commit();
}
else
}
-void HYDROGUI_ExportFileOp::WriteObjectPolyXY(SHPHandle theShpHandle, Handle_HYDROData_PolylineXY thePoly )
+int HYDROGUI_ExportFileOp::WriteObjectPolyXY(SHPHandle theShpHandle, Handle_HYDROData_PolylineXY thePoly )
{
SHPObject *aSHPObj;
std::vector<double> x, y;
std::vector<int> anPartStart;
+ for (int i = 0; i < thePoly->NbSections(); i++)
+ if (thePoly->GetSectionType(i) == HYDROData_IPolyline::SECTION_SPLINE)
+ return -1;
+
for (int i = 0; i < thePoly->NbSections(); i++)
{
anPartStart.push_back(x.size());
aSHPObj = SHPCreateObject( SHPT_ARC, -1, thePoly->NbSections(), &anPartStart[0], NULL, x.size(), &x[0], &y[0], NULL, NULL );
SHPWriteObject( theShpHandle, -1, aSHPObj );
SHPDestroyObject( aSHPObj );
+ return 1;
}
-void HYDROGUI_ExportFileOp::WriteObjectPoly3D(SHPHandle theShpHandle, Handle_HYDROData_Polyline3D thePoly )
+int HYDROGUI_ExportFileOp::WriteObjectPoly3D(SHPHandle theShpHandle, Handle_HYDROData_Polyline3D thePoly )
{
SHPObject *aSHPObj;
std::vector<double> x, y, z;
std::vector<int> anPartStart;
+
+ for (int i = 0; i < thePoly->GetPolylineXY()->NbSections(); i++)
+ if (thePoly->GetPolylineXY()->GetSectionType(i) == HYDROData_IPolyline::SECTION_SPLINE)
+ return -1;
for (int i = 0; i < thePoly->GetPolylineXY()->NbSections(); i++)
{
aSHPObj = SHPCreateObject( SHPT_ARCZ, -1, thePoly->GetPolylineXY()->NbSections(), &anPartStart[0], NULL, x.size(), &x[0], &y[0], &z[0], NULL );
SHPWriteObject( theShpHandle, -1, aSHPObj );
SHPDestroyObject( aSHPObj );
+ return 1;
}
-void HYDROGUI_ExportFileOp::WriteObjectLC(SHPHandle theShpHandle, Handle_HYDROData_LandCover theLC )
+int HYDROGUI_ExportFileOp::WriteObjectLC(SHPHandle theShpHandle, Handle_HYDROData_LandCover theLC )
{
-
TopoDS_Shape aSh = theLC->GetShape();
if (aSh.IsNull())
- return;
+ return 0;
+ TopExp_Explorer anEdgeEx(aSh, TopAbs_EDGE);
+ for (; anEdgeEx.More(); anEdgeEx.Next())
+ {
+ TopoDS_Edge E = TopoDS::Edge(anEdgeEx.Current());
+ double aFP, aLP;
+ Handle_Geom_Curve aCur = BRep_Tool::Curve(E, aFP, aLP);
+ Handle(Geom_Line) aLine = Handle(Geom_Line)::DownCast(aCur);
+ if (aLine.IsNull())
+ {
+ Handle(Geom_TrimmedCurve) aTC = Handle(Geom_TrimmedCurve)::DownCast(aCur);
+ if (!aTC.IsNull())
+ {
+ Handle(Geom_Line) aLine = Handle(Geom_Line)::DownCast(aTC->BasisCurve());
+ if (aLine.IsNull())
+ return -1;
+ }
+ else
+ return -1;
+ }
+
+ }
if (aSh.ShapeType() == TopAbs_FACE)
{
ProcessFace(TopoDS::Face(aSh), theShpHandle);
}
}
else
- return;
+ return 0;
+
+ return 1;
}
