ArtemisWrapper.cs

Go to the documentation of this file.

using UnityEngine;
using System.Collections;
using System.Runtime.InteropServices;

using ArtemisClientPointer = System.IntPtr;

public class ArtemisWrapper {

    // Header types
    public enum artemis_imd_type_e {
        IMD_DISCONNECT = 0, //Client disconnect from server
        IMD_ENERGIES = 1, // Energies structures
        IMD_FCOORDS = 2, // Atoms coordinates as floating point numbers
        IMD_GO = 3, // Client wants the simulation to run
        IMD_HANDSHAKE = 4, // Handshake for endian consistency
        IMD_KILL = 5, // Client wants to shutdown the simulation
        IMD_MDCOMM = 6, // Client sends forces to the simulation
        IMD_PAUSE = 7, // Client wants to pause the simulation for a while
        IMD_TRATE = 8, // Client sets the IMD transmission rate
        IMD_IOERROR = 9 // An I/O error occured
    };

    public struct ArtemisHeader
    {
        public artemis_imd_type_e type;
        public int length;
    };

    public struct ArtemisImdEnergies
    {
        public int tstep;  
        public float T;          
        public float Etot;       
        public float Epot;       
        public float Evdw;       
        public float Eelec;      
        public float Ebond;      
        public float Eangle;     
        public float Edihe;      
        public float Eimpr;
    };

    [DllImport ("artemis")]
    public static extern ArtemisClientPointer artemis_client_create(int nb_atoms);
    
    [DllImport ("artemis")]
    public static extern int artemis_client_destroy(ArtemisClientPointer client);
    
    [DllImport ("artemis")]
    public static extern int artemis_client_connect(ArtemisClientPointer client, [In] string hostname, [In] int port);
    
    [DllImport ("artemis")]
    public static extern int artemis_client_disconnect(ArtemisClientPointer client);

    [DllImport ("artemis")]
    public static extern int artemis_client_receive(ArtemisClientPointer client);

    [DllImport ("artemis")]
    public static extern int artemis_client_read_header(ArtemisClientPointer client, ref ArtemisHeader header);

    [DllImport ("artemis")]
    public static extern int artemis_client_read_energies(ArtemisClientPointer client, ref ArtemisImdEnergies energies);

    [DllImport ("artemis")]
    public static extern int artemis_client_read_coords(ArtemisClientPointer client, [In, Out] float[] coords, int nb_atoms);
    
    [DllImport ("artemis")]
    public static extern int artemis_client_send_forces(ArtemisClientPointer client, int nb_forces, [In] int[] indexes, [In] float[] forces);

    [DllImport ("artemis")]
    public static extern int artemis_client_send_go(ArtemisClientPointer client);

    [DllImport ("artemis")]
    public static extern int artemis_client_send_kill(ArtemisClientPointer client);

    [DllImport ("artemis")]
    public static extern int artemis_client_send_pause(ArtemisClientPointer client);

    [DllImport ("artemis")]
    public static extern int artemis_client_send_trate(ArtemisClientPointer client, int rate);
}