Source.Python forums

# =============================================================================
# >> IMPORTS
# =============================================================================
# Python
import math
import os

# Source.Python
import players
import entities
import core
import memory_c

from memory_c  import Convention
from events    import Event
from vecmath_c import CVector


# =============================================================================
# >> GLOBAL VARIABLES
# =============================================================================
MAX_COORD_RANGE = 16384

server = memory_c.find_binary(core.GAME_NAME + '/bin/server')
if os.name == 'nt':
    identifier = b'\x53\x8B\xDC\x83\xEC\x08\x83\xE4\xF0\x83\xC4\x04\x55\x8B\x6B\x04\x89\x6C\x24\x04\x8B\xEC\x83\xEC\x6C\x56'
else:
    identifier = '_Z14UTIL_TraceLineRK6VectorS1_jPK13IHandleEntityiP10CGameTrace'

UTIL_TraceLine = server[identifier].make_function(Convention.CDECL, 'ppIpip)v')


# =============================================================================
# >> GAME EVENTS
# =============================================================================
@Event
def player_say(ev):
    userid = ev.get_int('userid')
    edict  = players.helpers.edict_from_userid(userid)

    index = get_viewed_entity(edict)
    print(entities.helpers.edict_from_index(index).get_class_name())

def get_view_coords(edict, mask=0xFFFFFFFF, collision_group=0):
    '''
    Returns the edict's view coordinates as a tuple.
    '''

    # Create the start and end vectors
    startvec = get_eye_location(edict)
    endvec = startvec + get_eye_angle(edict) * MAX_COORD_RANGE

    # "Convert" them to pointers
    startvec_ptr = create_vector(*startvec)
    endvec_ptr   = create_vector(*endvec)

    # Allocate space for the CGameTrace pointer
    ptr = memory_c.CPointer()
    ptr.alloc(84)

    # Call UTIL_TraceLine
    UTIL_TraceLine(startvec_ptr, endvec_ptr, mask,
        entities.helpers.pointer_from_edict(edict), collision_group, ptr)

    # Get the end vector of the ray
    result = (ptr.get_float(12), ptr.get_float(16), ptr.get_float(20))

    # Deallocate the pointers to avoid memory leaks
    startvec_ptr.dealloc()
    endvec_ptr.dealloc()
    ptr.dealloc()

    return result

def get_viewed_entity(edict, mask=0xFFFFFFFF, collision_group=0):
    '''
    Returns the index of the entity which is the edict looking at.
    '''

    # Create the start and end vectors
    startvec = get_eye_location(edict)
    endvec = startvec + get_eye_angle(edict) * MAX_COORD_RANGE

    # "Convert" them to pointers
    startvec_ptr = create_vector(*startvec)
    endvec_ptr   = create_vector(*endvec)

    # Allocate space for the CGameTrace pointer
    ptr = memory_c.CPointer()
    ptr.alloc(84)

    # Call UTIL_TraceLine
    UTIL_TraceLine(startvec_ptr, endvec_ptr, mask,
        entities.helpers.pointer_from_edict(edict), collision_group, ptr)

    # Get the entity that was hit
    result = ptr.get_ptr(76).addr

    # Deallocate the pointers to avoid memory leaks
    startvec_ptr.dealloc()
    endvec_ptr.dealloc()
    ptr.dealloc()

    # Return either None or the entity's index
    return entities.helpers.index_from_pointer(result) if result else None


# =============================================================================
# >> HELPER FUNCTIONS
# =============================================================================
def create_vector(x, y, z):
    ptr = memory_c.CPointer()
    ptr.alloc(12)
    ptr.set_float(x)
    ptr.set_float(y, 4)
    ptr.set_float(z, 8)
    return ptr

def get_eye_location(edict):
    info = players.helpers.playerinfo_from_edict(edict)
    return CVector(*tuple(edict.get_prop('m_vecViewOffset[%s]'% x).get_float() + y for x, y in enumerate(info.get_abs_origin())))

def get_eye_angle(edict):
    myEyeAngle0 = edict.get_prop('m_angEyeAngles[0]').get_float()
    myEyeAngle1 = edict.get_prop('m_angEyeAngles[1]').get_float()
    myX = math.cos(math.radians(myEyeAngle1))
    myY = math.sin(math.radians(myEyeAngle1))
    myZ = -1 * math.sin(math.radians(myEyeAngle0))
    return CVector(myX, myY, myZ)