internal.h

Go to the documentation of this file.

/*
**  Command & Conquer Generals Zero Hour(tm)
**  Copyright 2025 Electronic Arts Inc.
**
**  This program is free software: you can redistribute it and/or modify
**  it under the terms of the GNU General Public License as published by
**  the Free Software Foundation, either version 3 of the License, or
**  (at your option) any later version.
**
**  This program is distributed in the hope that it will be useful,
**  but WITHOUT ANY WARRANTY; without even the implied warranty of
**  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
**  GNU General Public License for more details.
**
**  You should have received a copy of the GNU General Public License
**  along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

// $File: //depot/GeneralsMD/Staging/code/Libraries/Source/profile/internal.h $
// $Author: mhoffe $
// $Revision: #3 $
// $DateTime: 2003/07/09 10:57:23 $
//
// ©2003 Electronic Arts
//
// Internal header
#ifdef _MSC_VER
#  pragma once
#endif
#ifndef INTERNAL_H // Include guard
#define INTERNAL_H
 
#include "../debug/debug.h"
#include "internal_funclevel.h"
#include "internal_highlevel.h"
#include "internal_cmd.h"
#include "internal_result.h"
 
class ProfileFastCS
{
  ProfileFastCS(const ProfileFastCS&);
  ProfileFastCS& operator=(const ProfileFastCS&);
 
    volatile unsigned m_Flag;
  static HANDLE testEvent;
 
    void ThreadSafeSetFlag()
    {
        volatile unsigned& nFlag=m_Flag;
 
        #define ts_lock _emit 0xF0
        DASSERT(((unsigned)&nFlag % 4) == 0);
 
        __asm mov ebx, [nFlag]
        __asm ts_lock
        __asm bts dword ptr [ebx], 0
        __asm jc The_Bit_Was_Previously_Set_So_Try_Again
        return;
 
    The_Bit_Was_Previously_Set_So_Try_Again:
    // can't use SwitchToThread() here because Win9X doesn't have it!
    if (testEvent)
          ::WaitForSingleObject(testEvent,1);
        __asm mov ebx, [nFlag]
        __asm ts_lock
        __asm bts dword ptr [ebx], 0
        __asm jc  The_Bit_Was_Previously_Set_So_Try_Again
    }
 
    void ThreadSafeClearFlag()
    {
        m_Flag=0;
    }
 
public:
    ProfileFastCS(void):
    m_Flag(0) 
  {
  }
 
    class Lock
    {
    Lock(const Lock&);
    Lock& operator=(const Lock&);
 
        ProfileFastCS& CriticalSection;
 
    public:
        Lock(ProfileFastCS& cs): 
      CriticalSection(cs)
        {
            CriticalSection.ThreadSafeSetFlag();
        }
 
        ~Lock()
        {
            CriticalSection.ThreadSafeClearFlag();
        }
    };
 
    friend class Lock;
};
 
void *ProfileAllocMemory(unsigned numBytes);
void *ProfileReAllocMemory(void *oldPtr, unsigned newSize);
void ProfileFreeMemory(void *ptr);
 
__forceinline void ProfileGetTime(__int64 &t)
{
  _asm
  {
    mov ecx,[t]
    push eax
    push edx
    rdtsc
    mov [ecx],eax
    mov [ecx+4],edx
    pop edx
    pop eax
  };
}
 
#endif // INTERNAL_H