Session.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
 
#include "Core/CADKernelArchive.h"
#include "Core/Database.h"
#include "Core/CADEntity.h"
#include "Core/Types.h"
#include "Math/Geometry.h"
 
class FArchive;
 
namespace UE::CADKernel
{
class FEntity;
class FModel;
 
class CADKERNEL_API FSession
{
    friend FEntity;
    friend FCADKernelArchive;
    template<typename, ESPMode> friend class SharedPointerInternals::TIntrusiveReferenceController;
 
protected:
 
    double GeometricTolerance;
    FDatabase Database;
    int32 LastHostId;
 
public:
#if defined(CADKERNEL_DEV) || defined(CADKERNEL_STDA)
    static FSession Session;
#endif
 
    FSession(double InGeometricTolerance)
        : GeometricTolerance(InGeometricTolerance)
    {
        IntersectionTool::SetTolerance(InGeometricTolerance);
    }
 
    FModel& GetModel() { return *GetModelAsShared(); }
    TSharedPtr<FModel> GetModelAsShared();
 
    void Serialize(FCADKernelArchive& Ar)
    {
        Ar << GeometricTolerance;
        IntersectionTool::SetTolerance(GeometricTolerance);
    }
 
    FDatabase& GetDatabase()
    {
        return Database;
    }
 
    void SetGeometricTolerance(double NewTolerance);
 
    double GetGeometricTolerance() const
    {
        return GeometricTolerance;
    }
 
    bool SaveDatabase(const TCHAR* FilePath);
 
    void SaveDatabase(const TCHAR* FileName, const TArray<TSharedPtr<FEntity>>& Entities);
 
    void SaveDatabase(const TCHAR* FileName, const TArray<FEntity*>& Entities);
 
    void SaveDatabase(const TCHAR* FileName, FEntity& Entity)
    {
        TArray<FEntity*> Entities;
        Entities.Emplace(&Entity);
        SaveDatabase(FileName, Entities);
    }
 
    void SaveDatabase(const TCHAR* FileName, const TSharedPtr<FEntity> Entity)
    {
        TArray<TSharedPtr<FEntity>> Entities;
        Entities.Emplace(Entity);
        SaveDatabase(FileName, Entities);
    }
 
    bool LoadDatabase(const TCHAR* FilePath);
 
    bool SaveDatabase(TArray<uint8>& Bytes);
    bool LoadDatabase(const TArray<uint8>& Bytes);
 
    void AddDatabase(const TArray<uint8>& InRawData);
 
    void Clear()
    {
        Database.Empty();
    }
 
    uint32 SpawnEntityIdent(FEntity& SelectedEntity, bool bForceSpawning = false)
    {
        return Database.SpawnEntityIdent(SelectedEntity, bForceSpawning);
    }
 
    uint32 SpawnEntityIdents(const TArray<TSharedPtr<FEntity>>& SelectedEntities, bool bForceSpawning = false)
    {
        return Database.SpawnEntityIdents(SelectedEntities, bForceSpawning);
    }
 
    uint32 SpawnEntityIdents(const TArray<FEntity*>& SelectedEntities, bool bForceSpawning = false)
    {
        return Database.SpawnEntityIdents(SelectedEntities, bForceSpawning);
    }
 
    int32 GetLastHostId() const
    {
        return LastHostId;
    }
 
    int32 NewHostId()
    {
        return ++LastHostId;
    }
 
    void SetFirstNewHostId(int32 StartHostId)
    {
        LastHostId = StartHostId;
    }
};
 
} // namespace UE::CADKernel