[occt.git] / samples / CSharp / OCC / OCCViewer.cpp

#include "StdAfx.h"
#include ".\occviewer.h"

#pragma warning( disable : 4800 )
OCCViewer::OCCViewer(void)
{
	myGraphicDriver=NULL;
	myViewer=NULL;
	myView=NULL;
	myAISContext=NULL;
}

OCCViewer::~OCCViewer(void)
{
	myView->Remove();
}

bool OCCViewer::InitViewer(void* wnd)
{
  try
  {
    Handle(Aspect_DisplayConnection) aDisplayConnection;
    myGraphicDriver = Graphic3d::InitGraphicDriver (aDisplayConnection);
  }
  catch (Standard_Failure)
  {
    return false;
  }

  TCollection_ExtendedString a3DName("Visu3D");
  myViewer = new V3d_Viewer (myGraphicDriver, a3DName.ToExtString(),"", 1000.0, 
                             V3d_XposYnegZpos, Quantity_NOC_GRAY30,
                             V3d_ZBUFFER,V3d_GOURAUD,V3d_WAIT, 
                             Standard_True, Standard_False);

	myViewer->Init();
	myViewer->SetDefaultLights();
	myViewer->SetLightOn();
	myView = myViewer->CreateView();
	Handle(WNT_Window) aWNTWindow = new WNT_Window (reinterpret_cast<HWND> (wnd));
	myView->SetWindow(aWNTWindow);
	if (!aWNTWindow->IsMapped()) 
		 aWNTWindow->Map();
	myAISContext = new AIS_InteractiveContext(myViewer);
	myAISContext->UpdateCurrentViewer();
	myView->Redraw();
	myView->MustBeResized();
	return true;

}

bool OCCViewer::ImportBRep(char* filename)
{
	Standard_CString aFileName = (Standard_CString) filename;
	TopoDS_Shape aShape;
    BRep_Builder aBuilder;
    Standard_Boolean result = BRepTools::Read(aShape,aFileName,aBuilder);
	if (!result)
		return false;
	if(myAISContext->HasOpenedContext())
		myAISContext->CloseLocalContext();
	myAISContext->Display(new AIS_Shape(aShape));
	return true;
}

void OCCViewer::UpdateView(void)
{
	if (!myView.IsNull())
		myView->MustBeResized();
}

void OCCViewer::RedrawView(void)
{
	if (!myView.IsNull())
		myView->Redraw();
}

void OCCViewer::SetDegenerateModeOn(void)
{
	if (!myView.IsNull())
		myView->SetComputedMode (Standard_False);
}

void OCCViewer::SetDegenerateModeOff(void)
{
	if (!myView.IsNull())
		myView->SetComputedMode (Standard_True);
}

void OCCViewer::WindowFitAll(int Xmin, int Ymin, int Xmax, int Ymax)
{
	if (!myView.IsNull())
		myView->WindowFitAll(Xmin, Ymin, Xmax, Ymax);
}

void OCCViewer::Place(int x, int y, float zoomFactor)
{
	Quantity_Factor aZoomFactor = zoomFactor;
	if (!myView.IsNull())
		myView->Place(x, y, aZoomFactor);
}

void OCCViewer::Zoom(int x1, int y1, int x2, int y2)
{
	if (!myView.IsNull())
		myView->Zoom(x1, y1, x2, y2);
}

void OCCViewer::Pan(int x, int y)
{
	if (!myView.IsNull())
		myView->Pan(x, y);
}

void OCCViewer::Rotation(int x, int y)
{
	if (!myView.IsNull())
		myView->Rotation(x, y);
}

void OCCViewer::StartRotation(int x, int y)
{
	if (!myView.IsNull())
		myView->StartRotation(x, y);
}

void OCCViewer::Select(int x1, int y1, int x2, int y2)
{
	if (!myAISContext.IsNull())
		myAISContext->Select(x1, y1, x2, y2, myView);
}

void OCCViewer::Select(void)
{
	if (!myAISContext.IsNull())
		myAISContext->Select();
}

void OCCViewer::MoveTo(int x, int y)
{
	if ((!myAISContext.IsNull()) && (!myView.IsNull()))
		myAISContext->MoveTo(x, y, myView);
}

void OCCViewer::ShiftSelect(int x1, int y1, int x2, int y2)
{
	if ((!myAISContext.IsNull()) && (!myView.IsNull()))
		myAISContext->ShiftSelect(x1, y1, x2, y2, myView);
}

void OCCViewer::ShiftSelect(void)
{
	if (!myAISContext.IsNull())
		myAISContext->ShiftSelect();
}

void OCCViewer::BackgroundColor(int& r, int& g, int& b)
{
	Standard_Real R1;
	Standard_Real G1;
	Standard_Real B1;
	if (!myView.IsNull())
		myView->BackgroundColor(Quantity_TOC_RGB,R1,G1,B1);
	r = (int)R1*255;
	g = (int)G1*255;
	b = (int)B1*255;
}


void OCCViewer::UpdateCurrentViewer(void)
{
	if (!myAISContext.IsNull())
		myAISContext->UpdateCurrentViewer();
}

void OCCViewer::FrontView(void)
{
	if (!myView.IsNull())
		myView->SetProj(V3d_Xpos); 
}

void OCCViewer::TopView(void)
{
	if (!myView.IsNull())
		myView->SetProj(V3d_Zpos); 
}

void OCCViewer::LeftView(void)
{
	if (!myView.IsNull())
		myView->SetProj(V3d_Ypos);
}

void OCCViewer::BackView(void)
{
	if (!myView.IsNull())
		myView->SetProj(V3d_Xneg);
}

void OCCViewer::RightView(void)
{
	if (!myView.IsNull())
		myView->SetProj(V3d_Yneg);
}

void OCCViewer::BottomView(void)
{
	if (!myView.IsNull())
		myView->SetProj(V3d_Zneg); 
}

void OCCViewer::AxoView(void)
{
	if (!myView.IsNull())
		myView->SetProj(V3d_XposYnegZpos);
}

void OCCViewer::ZoomAllView(void)
{
	if (!myView.IsNull())
	{
		myView->FitAll();
		myView->ZFitAll();
	}
}

float OCCViewer::Scale(void)
{
	if (myView.IsNull())
		return -1;
	else
		return (float)myView->Scale();
}

void OCCViewer::ResetView(void)
{
	if (!myView.IsNull())
		myView->Reset();
}

void OCCViewer::SetDisplayMode(int aMode)
{
	if (myAISContext.IsNull())
		return;
	AIS_DisplayMode CurrentMode;
	if (aMode == 0) 
		CurrentMode=AIS_WireFrame;
	else
		CurrentMode=AIS_Shaded;
    if(myAISContext->NbCurrents()==0 || myAISContext->NbSelected()==0)
       myAISContext->SetDisplayMode(CurrentMode);
    else 
	{
       for(myAISContext->InitCurrent();myAISContext->MoreCurrent();myAISContext->NextCurrent())
	           myAISContext->SetDisplayMode(myAISContext->Current(),aMode,Standard_False);
	         
	}
	 myAISContext->UpdateCurrentViewer();
}

void OCCViewer::SetColor(int r, int g, int b)
{
	if (myAISContext.IsNull())
		return;
	Quantity_Color col =  Quantity_Color(r/255.,g/255.,b/255.,Quantity_TOC_RGB);
	for (;myAISContext->MoreCurrent ();myAISContext->NextCurrent ())
	          myAISContext->SetColor (myAISContext->Current(),col.Name());
}

void OCCViewer::ObjectColor(int& r, int& g, int& b)
{
	if (myAISContext.IsNull())
		return;
	r=255;
	g=255;
	b=255;
	Handle_AIS_InteractiveObject Current ;
	Quantity_Color ObjCol;
	myAISContext->InitCurrent();
	if (!myAISContext->MoreCurrent())
		return;
    Current = myAISContext->Current();
	if ( Current->HasColor () ) {
      ObjCol = myAISContext->Color(myAISContext->Current());
	  Quantity_Parameter r1, r2, r3;
	  ObjCol.Values(r1, r2, r3, Quantity_TOC_RGB);
		r=(int)r1*255;
		g=(int)r2*255;
		b=(int)r3*255;
  	}
}

void OCCViewer::SetBackgroundColor(int r, int g, int b)
{
	if (!myView.IsNull())
		myView->SetBackgroundColor(Quantity_TOC_RGB,r/255.,g/255.,b/255.);
}

void OCCViewer::EraseObjects(void)
{
	if (myAISContext.IsNull())
		return;
	for(myAISContext->InitCurrent();myAISContext->MoreCurrent();myAISContext->NextCurrent())
        myAISContext->Erase(myAISContext->Current(),Standard_True);
	myAISContext->ClearCurrents();
}

float OCCViewer::GetVersion(void)
{
	return (float)OCC_VERSION;
}

void OCCViewer::SetMaterial(int theMaterial)
{
	if (myAISContext.IsNull())
		return;
    for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent (); myAISContext->NextCurrent () )
        myAISContext->SetMaterial( myAISContext->Current(), (Graphic3d_NameOfMaterial)theMaterial );
	myAISContext->UpdateCurrentViewer();
}

void OCCViewer::SetTransparency(int theTrans)
{
	if (myAISContext.IsNull())
		return;
	for( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextSelected() )
		myAISContext->SetTransparency( myAISContext->Current(), ((Standard_Real)theTrans) / 10.0 );
}

bool OCCViewer::ImportCsfdb(char* filename)
{
	Standard_CString aFileName = (Standard_CString) filename;
    if ( FSD_File::IsGoodFileType(aFileName) != Storage_VSOk )
	    return false;

    static FSD_File fileDriver;
    TCollection_AsciiString aName( aFileName );
    if ( fileDriver.Open( aName, Storage_VSRead ) != Storage_VSOk )
        return false;

    Handle(ShapeSchema) schema = new ShapeSchema();
    Handle(Storage_Data) data  = schema->Read( fileDriver );
    if ( data->ErrorStatus() != Storage_VSOk )
        return false;
    fileDriver.Close();

    Handle(Storage_HSeqOfRoot) roots = data->Roots();
    for ( int i = 1; i <= roots->Length() ; i++ )
    {
        Handle(Storage_Root) r = roots->Value( i );
        Handle(Standard_Persistent) p = r->Object();
        Handle(PTopoDS_HShape) aPShape = Handle(PTopoDS_HShape)::DownCast(p);
        if ( !aPShape.IsNull() )
        {
	        PTColStd_PersistentTransientMap aMap;
	        TopoDS_Shape aTShape;
            MgtBRep::Translate( aPShape, aMap, aTShape, MgtBRep_WithTriangle );
			myAISContext->Display(new AIS_Shape(aTShape));
        }
    }

	return true;
}

bool OCCViewer::ImportIges(char* filename)
{
	Standard_CString aFileName = (Standard_CString) filename;
    IGESControl_Reader Reader;
    int status = Reader.ReadFile( aFileName );

    if ( status == IFSelect_RetDone )
    {
        Reader.TransferRoots();
        TopoDS_Shape aShape = Reader.OneShape();
        myAISContext->Display(new AIS_Shape(aShape));
    } else
		return false;
	myAISContext->UpdateCurrentViewer();
	return true;
}

bool OCCViewer::ImportStep(char* filename)
{
	Standard_CString aFileName = (Standard_CString) filename;
	STEPControl_Reader aReader;
	IFSelect_ReturnStatus status = aReader.ReadFile(aFileName);
	if ( status == IFSelect_RetDone )
    {
	    bool failsonly = false;
	    aReader.PrintCheckLoad( failsonly, IFSelect_ItemsByEntity );

	    int nbr = aReader.NbRootsForTransfer();
	    aReader.PrintCheckTransfer( failsonly, IFSelect_ItemsByEntity );
	    for ( Standard_Integer n = 1; n <= nbr; n++ )
	    {
	        Standard_Boolean ok = aReader.TransferRoot( n );
	        int nbs = aReader.NbShapes();
	        if ( nbs > 0 )
            {
	            for ( int i = 1; i <= nbs; i++ )
                {
		            TopoDS_Shape shape = aReader.Shape( i );
		            myAISContext->Display(new AIS_Shape(shape));
	            }
            }
        }
	} else
		return false;
	return true;
}

bool OCCViewer::ExportBRep(char* filename)
{
	myAISContext->InitCurrent();
	if (!myAISContext->MoreCurrent())
		return false;
	Handle_AIS_InteractiveObject anIO = myAISContext->Current();
	Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
	return (bool)BRepTools::Write( anIS->Shape(), (Standard_CString)filename ); ;
}


bool OCCViewer::ExportIges(char* filename)
{
	IGESControl_Controller::Init();
	IGESControl_Writer writer( Interface_Static::CVal( "XSTEP.iges.unit" ),
                               Interface_Static::IVal( "XSTEP.iges.writebrep.mode" ) );
 
	
	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent() )
	{
		Handle_AIS_InteractiveObject anIO = myAISContext->Current();
		Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
		TopoDS_Shape shape = anIS->Shape();
		writer.AddShape ( shape );
	}
	writer.ComputeModel();
	return (bool)writer.Write( (Standard_CString)filename );
}

bool OCCViewer::ExpotStep(char* filename)
{
    STEPControl_StepModelType type = STEPControl_AsIs;
    IFSelect_ReturnStatus status;
    STEPControl_Writer writer;
	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent() )
    {
		Handle_AIS_InteractiveObject anIO = myAISContext->Current();
		Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
		TopoDS_Shape shape = anIS->Shape();
		status = writer.Transfer( shape , type );
        if ( status != IFSelect_RetDone )
            return false;
    }

    status = writer.Write( (Standard_CString)filename );
    if ( status != IFSelect_RetDone )
            return false;
	return true;
}

bool OCCViewer::ExportStl(char* filename)
{
	TopoDS_Compound comp;
	BRep_Builder builder;
	builder.MakeCompound( comp );

	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent() )
	{
		Handle_AIS_InteractiveObject anIO = myAISContext->Current();
		Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
		TopoDS_Shape shape = anIS->Shape();
		if ( shape.IsNull() ) 
			return false;    
		builder.Add( comp, shape );
	}

	StlAPI_Writer writer;
	writer.Write( comp, (Standard_CString)filename );
	return true;
}

bool OCCViewer::ExportVrml(char* filename)
{
	TopoDS_Compound res;
	BRep_Builder builder;
	builder.MakeCompound( res );

	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent() )
	{
		Handle_AIS_InteractiveObject anIO = myAISContext->Current();
		Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
		TopoDS_Shape shape = anIS->Shape();
		if ( shape.IsNull() )
			return false;
        
		builder.Add( res, shape );
	}

	VrmlAPI_Writer writer;
	writer.Write( res, (Standard_CString)filename );

	return true;
}

bool OCCViewer::Dump(char *filename)
{
	if (myView.IsNull())
		return false;
	myView->Redraw();
    return (bool)myView->Dump(filename);
}

bool OCCViewer::IsObjectSelected(void)
{
	if (myAISContext.IsNull())
		return false;
	myAISContext->InitCurrent();
	return (bool)myAISContext->MoreCurrent();
}

int OCCViewer::DisplayMode(void)
{
	if (myAISContext.IsNull())
		return -1;
	int mode = -1;
	bool OneOrMoreInShading=false;
	bool OneOrMoreInWireframe=false;
	for (myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent())
	{
		if ( myAISContext->IsDisplayed( myAISContext->Current(), 1 ) )
             OneOrMoreInShading = true;
        if ( myAISContext->IsDisplayed( myAISContext->Current(), 0 ) )
             OneOrMoreInWireframe = true;
	}
	if (OneOrMoreInShading&&OneOrMoreInWireframe)
		mode=10;
	else if(OneOrMoreInShading)
		mode=1;
	else if (OneOrMoreInWireframe)
		mode=0;
	return mode;
}

void OCCViewer::CreateNewView(void* wnd)
{
	if (myAISContext.IsNull())
		return;
	myView = myAISContext->CurrentViewer()->CreateView();
	if (myGraphicDriver.IsNull())
    {
      Handle(Aspect_DisplayConnection) aDisplayConnection;
      myGraphicDriver = Graphic3d::InitGraphicDriver (aDisplayConnection);
    }
	Handle(WNT_Window) aWNTWindow = new WNT_Window (reinterpret_cast<HWND> (wnd));
	myView->SetWindow(aWNTWindow);
	Standard_Integer w=100, h=100;
	aWNTWindow->Size(w,h);
	if (!aWNTWindow->IsMapped()) 
		 aWNTWindow->Map();
}

bool OCCViewer::SetAISContext(OCCViewer* Viewer)
{
	this->myAISContext=Viewer->GetAISContext();
	if (myAISContext.IsNull())
		return false;
	return true;
}

Handle_AIS_InteractiveContext OCCViewer::GetAISContext(void)
{
	return myAISContext;
}

int OCCViewer::CharToInt(char symbol)
{
	TCollection_AsciiString msg = symbol;
	return msg.IntegerValue();
}
Commit	Line	Data
7fd59977	1	#include "StdAfx.h"
	2	#include ".\occviewer.h"
	3
	4	#pragma warning( disable : 4800 )
	5	OCCViewer::OCCViewer(void)
	6	{
dc3fe572	7	myGraphicDriver=NULL;
7fd59977	8	myViewer=NULL;
	9	myView=NULL;
	10	myAISContext=NULL;
	11	}
	12
	13	OCCViewer::~OCCViewer(void)
	14	{
	15	myView->Remove();
	16	}
	17
	18	bool OCCViewer::InitViewer(void* wnd)
	19	{
dc3fe572	20	try
	21	{
	22	Handle(Aspect_DisplayConnection) aDisplayConnection;
	23	myGraphicDriver = Graphic3d::InitGraphicDriver (aDisplayConnection);
	24	}
	25	catch (Standard_Failure)
	26	{
	27	return false;
	28	}
	29
	30	TCollection_ExtendedString a3DName("Visu3D");
	31	myViewer = new V3d_Viewer (myGraphicDriver, a3DName.ToExtString(),"", 1000.0,
	32	V3d_XposYnegZpos, Quantity_NOC_GRAY30,
	33	V3d_ZBUFFER,V3d_GOURAUD,V3d_WAIT,
	34	Standard_True, Standard_False);
7fd59977	35
	36	myViewer->Init();
	37	myViewer->SetDefaultLights();
	38	myViewer->SetLightOn();
	39	myView = myViewer->CreateView();
dc3fe572	40	Handle(WNT_Window) aWNTWindow = new WNT_Window (reinterpret_cast<HWND> (wnd));
7fd59977	41	myView->SetWindow(aWNTWindow);
	42	if (!aWNTWindow->IsMapped())
	43	aWNTWindow->Map();
	44	myAISContext = new AIS_InteractiveContext(myViewer);
	45	myAISContext->UpdateCurrentViewer();
	46	myView->Redraw();
	47	myView->MustBeResized();
	48	return true;
	49
	50	}
	51
	52	bool OCCViewer::ImportBRep(char* filename)
	53	{
	54	Standard_CString aFileName = (Standard_CString) filename;
	55	TopoDS_Shape aShape;
	56	BRep_Builder aBuilder;
	57	Standard_Boolean result = BRepTools::Read(aShape,aFileName,aBuilder);
	58	if (!result)
	59	return false;
	60	if(myAISContext->HasOpenedContext())
	61	myAISContext->CloseLocalContext();
	62	myAISContext->Display(new AIS_Shape(aShape));
	63	return true;
	64	}
	65
	66	void OCCViewer::UpdateView(void)
	67	{
	68	if (!myView.IsNull())
	69	myView->MustBeResized();
	70	}
	71
	72	void OCCViewer::RedrawView(void)
	73	{
	74	if (!myView.IsNull())
	75	myView->Redraw();
	76	}
	77
	78	void OCCViewer::SetDegenerateModeOn(void)
	79	{
	80	if (!myView.IsNull())
de75ed09	81	myView->SetComputedMode (Standard_False);
7fd59977	82	}
	83
	84	void OCCViewer::SetDegenerateModeOff(void)
	85	{
	86	if (!myView.IsNull())
de75ed09	87	myView->SetComputedMode (Standard_True);
7fd59977	88	}
	89
	90	void OCCViewer::WindowFitAll(int Xmin, int Ymin, int Xmax, int Ymax)
	91	{
	92	if (!myView.IsNull())
	93	myView->WindowFitAll(Xmin, Ymin, Xmax, Ymax);
	94	}
	95
	96	void OCCViewer::Place(int x, int y, float zoomFactor)
	97	{
	98	Quantity_Factor aZoomFactor = zoomFactor;
	99	if (!myView.IsNull())
	100	myView->Place(x, y, aZoomFactor);
	101	}
	102
	103	void OCCViewer::Zoom(int x1, int y1, int x2, int y2)
	104	{
	105	if (!myView.IsNull())
	106	myView->Zoom(x1, y1, x2, y2);
	107	}
	108
	109	void OCCViewer::Pan(int x, int y)
	110	{
	111	if (!myView.IsNull())
	112	myView->Pan(x, y);
	113	}
	114
	115	void OCCViewer::Rotation(int x, int y)
	116	{
	117	if (!myView.IsNull())
	118	myView->Rotation(x, y);
	119	}
	120
	121	void OCCViewer::StartRotation(int x, int y)
	122	{
	123	if (!myView.IsNull())
	124	myView->StartRotation(x, y);
	125	}
	126
	127	void OCCViewer::Select(int x1, int y1, int x2, int y2)
	128	{
	129	if (!myAISContext.IsNull())
	130	myAISContext->Select(x1, y1, x2, y2, myView);
	131	}
	132
	133	void OCCViewer::Select(void)
	134	{
	135	if (!myAISContext.IsNull())
	136	myAISContext->Select();
	137	}
	138
	139	void OCCViewer::MoveTo(int x, int y)
	140	{
	141	if ((!myAISContext.IsNull()) && (!myView.IsNull()))
	142	myAISContext->MoveTo(x, y, myView);
	143	}
	144
	145	void OCCViewer::ShiftSelect(int x1, int y1, int x2, int y2)
	146	{
	147	if ((!myAISContext.IsNull()) && (!myView.IsNull()))
	148	myAISContext->ShiftSelect(x1, y1, x2, y2, myView);
	149	}
	150
	151	void OCCViewer::ShiftSelect(void)
152	{
153	if (!myAISContext.IsNull())
154	myAISContext->ShiftSelect();
155	}
156
157	void OCCViewer::BackgroundColor(int& r, int& g, int& b)
158	{
159	Standard_Real R1;
160	Standard_Real G1;
161	Standard_Real B1;
162	if (!myView.IsNull())
163	myView->BackgroundColor(Quantity_TOC_RGB,R1,G1,B1);
164	r = (int)R1*255;
165	g = (int)G1*255;
166	b = (int)B1*255;
167	}
168
169
170
171	void OCCViewer::UpdateCurrentViewer(void)
172	{
173	if (!myAISContext.IsNull())
174	myAISContext->UpdateCurrentViewer();
175	}
176
177	void OCCViewer::FrontView(void)
178	{
179	if (!myView.IsNull())
180	myView->SetProj(V3d_Xpos);
181	}
182
183	void OCCViewer::TopView(void)
184	{
185	if (!myView.IsNull())
186	myView->SetProj(V3d_Zpos);
187	}
188
189	void OCCViewer::LeftView(void)
190	{
191	if (!myView.IsNull())
192	myView->SetProj(V3d_Ypos);
193	}
194
195	void OCCViewer::BackView(void)
196	{
197	if (!myView.IsNull())
198	myView->SetProj(V3d_Xneg);
199	}
200
201	void OCCViewer::RightView(void)
202	{
203	if (!myView.IsNull())
204	myView->SetProj(V3d_Yneg);
205	}
206
207	void OCCViewer::BottomView(void)
208	{
209	if (!myView.IsNull())
210	myView->SetProj(V3d_Zneg);
211	}
212
213	void OCCViewer::AxoView(void)
214	{
215	if (!myView.IsNull())
216	myView->SetProj(V3d_XposYnegZpos);
217	}
218
219	void OCCViewer::ZoomAllView(void)
220	{
221	if (!myView.IsNull())
222	{
223	myView->FitAll();
224	myView->ZFitAll();
225	}
226	}
227
228	float OCCViewer::Scale(void)
229	{
230	if (myView.IsNull())
231	return -1;
232	else
233	return (float)myView->Scale();
234	}
235
236	void OCCViewer::ResetView(void)
237	{
238	if (!myView.IsNull())
239	myView->Reset();
240	}
241
242	void OCCViewer::SetDisplayMode(int aMode)
243	{
244	if (myAISContext.IsNull())
245	return;
246	AIS_DisplayMode CurrentMode;
247	if (aMode == 0)
248	CurrentMode=AIS_WireFrame;
249	else
250	CurrentMode=AIS_Shaded;
251	if(myAISContext->NbCurrents()==0 \|\| myAISContext->NbSelected()==0)
252	myAISContext->SetDisplayMode(CurrentMode);
253	else
254	{
255	for(myAISContext->InitCurrent();myAISContext->MoreCurrent();myAISContext->NextCurrent())
256	myAISContext->SetDisplayMode(myAISContext->Current(),aMode,Standard_False);
257
258	}
259	myAISContext->UpdateCurrentViewer();
260	}
261
262	void OCCViewer::SetColor(int r, int g, int b)
263	{
264	if (myAISContext.IsNull())
265	return;
266	Quantity_Color col = Quantity_Color(r/255.,g/255.,b/255.,Quantity_TOC_RGB);
267	for (;myAISContext->MoreCurrent ();myAISContext->NextCurrent ())
268	myAISContext->SetColor (myAISContext->Current(),col.Name());
269	}
270
271	void OCCViewer::ObjectColor(int& r, int& g, int& b)
272	{
273	if (myAISContext.IsNull())
274	return;
275	r=255;
276	g=255;
277	b=255;
278	Handle_AIS_InteractiveObject Current ;
279	Quantity_Color ObjCol;
280	myAISContext->InitCurrent();
281	if (!myAISContext->MoreCurrent())
282	return;
283	Current = myAISContext->Current();
284	if ( Current->HasColor () ) {
285	ObjCol = myAISContext->Color(myAISContext->Current());
286	Quantity_Parameter r1, r2, r3;
287	ObjCol.Values(r1, r2, r3, Quantity_TOC_RGB);
288	r=(int)r1*255;
289	g=(int)r2*255;
290	b=(int)r3*255;
291	}
292	}
293
294	void OCCViewer::SetBackgroundColor(int r, int g, int b)
295	{
296	if (!myView.IsNull())
297	myView->SetBackgroundColor(Quantity_TOC_RGB,r/255.,g/255.,b/255.);
298	}
299
300	void OCCViewer::EraseObjects(void)
301	{
302	if (myAISContext.IsNull())
303	return;
304	for(myAISContext->InitCurrent();myAISContext->MoreCurrent();myAISContext->NextCurrent())
bcb1adc5	305	myAISContext->Erase(myAISContext->Current(),Standard_True);
7fd59977	306	myAISContext->ClearCurrents();
	307	}
	308
	309	float OCCViewer::GetVersion(void)
	310	{
	311	return (float)OCC_VERSION;
	312	}
	313
	314	void OCCViewer::SetMaterial(int theMaterial)
	315	{
	316	if (myAISContext.IsNull())
	317	return;
	318	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent (); myAISContext->NextCurrent () )
	319	myAISContext->SetMaterial( myAISContext->Current(), (Graphic3d_NameOfMaterial)theMaterial );
	320	myAISContext->UpdateCurrentViewer();
	321	}
	322
	323	void OCCViewer::SetTransparency(int theTrans)
	324	{
	325	if (myAISContext.IsNull())
	326	return;
	327	for( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextSelected() )
	328	myAISContext->SetTransparency( myAISContext->Current(), ((Standard_Real)theTrans) / 10.0 );
	329	}
	330
	331	bool OCCViewer::ImportCsfdb(char* filename)
	332	{
	333	Standard_CString aFileName = (Standard_CString) filename;
	334	if ( FSD_File::IsGoodFileType(aFileName) != Storage_VSOk )
	335	return false;
	336
	337	static FSD_File fileDriver;
	338	TCollection_AsciiString aName( aFileName );
	339	if ( fileDriver.Open( aName, Storage_VSRead ) != Storage_VSOk )
	340	return false;
	341
	342	Handle(ShapeSchema) schema = new ShapeSchema();
	343	Handle(Storage_Data) data = schema->Read( fileDriver );
	344	if ( data->ErrorStatus() != Storage_VSOk )
	345	return false;
	346	fileDriver.Close();
	347
	348	Handle(Storage_HSeqOfRoot) roots = data->Roots();
	349	for ( int i = 1; i <= roots->Length() ; i++ )
	350	{
	351	Handle(Storage_Root) r = roots->Value( i );
	352	Handle(Standard_Persistent) p = r->Object();
	353	Handle(PTopoDS_HShape) aPShape = Handle(PTopoDS_HShape)::DownCast(p);
	354	if ( !aPShape.IsNull() )
	355	{
	356	PTColStd_PersistentTransientMap aMap;
	357	TopoDS_Shape aTShape;
	358	MgtBRep::Translate( aPShape, aMap, aTShape, MgtBRep_WithTriangle );
	359	myAISContext->Display(new AIS_Shape(aTShape));
	360	}
	361	}
	362
	363	return true;
	364	}
	365
	366	bool OCCViewer::ImportIges(char* filename)
	367	{
	368	Standard_CString aFileName = (Standard_CString) filename;
	369	IGESControl_Reader Reader;
370	int status = Reader.ReadFile( aFileName );
371
372	if ( status == IFSelect_RetDone )
373	{
374	Reader.TransferRoots();
375	TopoDS_Shape aShape = Reader.OneShape();
376	myAISContext->Display(new AIS_Shape(aShape));
377	} else
378	return false;
379	myAISContext->UpdateCurrentViewer();
380	return true;
381	}
382
383	bool OCCViewer::ImportStep(char* filename)
384	{
385	Standard_CString aFileName = (Standard_CString) filename;
386	STEPControl_Reader aReader;
387	IFSelect_ReturnStatus status = aReader.ReadFile(aFileName);
388	if ( status == IFSelect_RetDone )
389	{
390	bool failsonly = false;
391	aReader.PrintCheckLoad( failsonly, IFSelect_ItemsByEntity );
392
393	int nbr = aReader.NbRootsForTransfer();
394	aReader.PrintCheckTransfer( failsonly, IFSelect_ItemsByEntity );
395	for ( Standard_Integer n = 1; n <= nbr; n++ )
396	{
397	Standard_Boolean ok = aReader.TransferRoot( n );
398	int nbs = aReader.NbShapes();
399	if ( nbs > 0 )
400	{
401	for ( int i = 1; i <= nbs; i++ )
402	{
403	TopoDS_Shape shape = aReader.Shape( i );
404	myAISContext->Display(new AIS_Shape(shape));
405	}
406	}
407	}
408	} else
409	return false;
410	return true;
411	}
412
413	bool OCCViewer::ExportBRep(char* filename)
414	{
415	myAISContext->InitCurrent();
416	if (!myAISContext->MoreCurrent())
417	return false;
418	Handle_AIS_InteractiveObject anIO = myAISContext->Current();
419	Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
420	return (bool)BRepTools::Write( anIS->Shape(), (Standard_CString)filename ); ;
421	}
422
423
424	bool OCCViewer::ExportIges(char* filename)
425	{
426	IGESControl_Controller::Init();
427	IGESControl_Writer writer( Interface_Static::CVal( "XSTEP.iges.unit" ),
428	Interface_Static::IVal( "XSTEP.iges.writebrep.mode" ) );
429
430
431	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent() )
432	{
433	Handle_AIS_InteractiveObject anIO = myAISContext->Current();
434	Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
435	TopoDS_Shape shape = anIS->Shape();
436	writer.AddShape ( shape );
437	}
438	writer.ComputeModel();
439	return (bool)writer.Write( (Standard_CString)filename );
440	}
441
442	bool OCCViewer::ExpotStep(char* filename)
443	{
444	STEPControl_StepModelType type = STEPControl_AsIs;
445	IFSelect_ReturnStatus status;
446	STEPControl_Writer writer;
447	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent() )
448	{
449	Handle_AIS_InteractiveObject anIO = myAISContext->Current();
450	Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
451	TopoDS_Shape shape = anIS->Shape();
452	status = writer.Transfer( shape , type );
453	if ( status != IFSelect_RetDone )
454	return false;
455	}
456
457	status = writer.Write( (Standard_CString)filename );
458	if ( status != IFSelect_RetDone )
459	return false;
460	return true;
461	}
462
463	bool OCCViewer::ExportStl(char* filename)
464	{
465	TopoDS_Compound comp;
466	BRep_Builder builder;
467	builder.MakeCompound( comp );
468
469	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent() )
470	{
471	Handle_AIS_InteractiveObject anIO = myAISContext->Current();
472	Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
473	TopoDS_Shape shape = anIS->Shape();
474	if ( shape.IsNull() )
475	return false;
476	builder.Add( comp, shape );
477	}
478
479	StlAPI_Writer writer;
480	writer.Write( comp, (Standard_CString)filename );
481	return true;
482	}
483
484	bool OCCViewer::ExportVrml(char* filename)
485	{
486	TopoDS_Compound res;
487	BRep_Builder builder;
488	builder.MakeCompound( res );
489
490	for ( myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent() )
491	{
492	Handle_AIS_InteractiveObject anIO = myAISContext->Current();
493	Handle_AIS_Shape anIS=Handle_AIS_Shape::DownCast(anIO);
494	TopoDS_Shape shape = anIS->Shape();
495	if ( shape.IsNull() )
496	return false;
497
498	builder.Add( res, shape );
499	}
500
501	VrmlAPI_Writer writer;
502	writer.Write( res, (Standard_CString)filename );
503
504	return true;
505	}
506
507	bool OCCViewer::Dump(char *filename)
508	{
509	if (myView.IsNull())
510	return false;
511	myView->Redraw();
512	return (bool)myView->Dump(filename);
513	}
514
515	bool OCCViewer::IsObjectSelected(void)
516	{
517	if (myAISContext.IsNull())
518	return false;
519	myAISContext->InitCurrent();
520	return (bool)myAISContext->MoreCurrent();
521	}
522
523	int OCCViewer::DisplayMode(void)
524	{
525	if (myAISContext.IsNull())
526	return -1;
527	int mode = -1;
528	bool OneOrMoreInShading=false;
529	bool OneOrMoreInWireframe=false;
530	for (myAISContext->InitCurrent(); myAISContext->MoreCurrent(); myAISContext->NextCurrent())
531	{
532	if ( myAISContext->IsDisplayed( myAISContext->Current(), 1 ) )
533	OneOrMoreInShading = true;
534	if ( myAISContext->IsDisplayed( myAISContext->Current(), 0 ) )
535	OneOrMoreInWireframe = true;
536	}
537	if (OneOrMoreInShading&&OneOrMoreInWireframe)
538	mode=10;
539	else if(OneOrMoreInShading)
540	mode=1;
541	else if (OneOrMoreInWireframe)
542	mode=0;
543	return mode;
544	}
545
546	void OCCViewer::CreateNewView(void* wnd)
547	{
548	if (myAISContext.IsNull())
549	return;
550	myView = myAISContext->CurrentViewer()->CreateView();
dc3fe572	551	if (myGraphicDriver.IsNull())
	552	{
	553	Handle(Aspect_DisplayConnection) aDisplayConnection;
	554	myGraphicDriver = Graphic3d::InitGraphicDriver (aDisplayConnection);
	555	}
	556	Handle(WNT_Window) aWNTWindow = new WNT_Window (reinterpret_cast<HWND> (wnd));
7fd59977	557	myView->SetWindow(aWNTWindow);
	558	Standard_Integer w=100, h=100;
	559	aWNTWindow->Size(w,h);
	560	if (!aWNTWindow->IsMapped())
	561	aWNTWindow->Map();
	562	}
	563
	564	bool OCCViewer::SetAISContext(OCCViewer* Viewer)
	565	{
	566	this->myAISContext=Viewer->GetAISContext();
	567	if (myAISContext.IsNull())
	568	return false;
	569	return true;
	570	}
	571
	572	Handle_AIS_InteractiveContext OCCViewer::GetAISContext(void)
	573	{
	574	return myAISContext;
	575	}
	576
	577	int OCCViewer::CharToInt(char symbol)
	578	{
	579	TCollection_AsciiString msg = symbol;
	580	return msg.IntegerValue();
	581	}