lang/LangPrimSource/PyrBitPrim.cpp

   1 /*
   2         SuperCollider real time audio synthesis system
   3     Copyright (c) 2002 James McCartney. All rights reserved.
   4         http://www.audiosynth.com
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with this program; if not, write to the Free Software
  18     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA
  19 */
  20 /*
  21
  22 Primitives for some bit operations.
  23
  24 */
  25
  26 #include "PyrPrimitive.h"
  27 #include "VMGlobals.h"
  28 #include "clz.h"
  29
  30
  31 int prNumBits(VMGlobals *g, int numArgsPushed);
  32 int prNumBits(VMGlobals *g, int numArgsPushed)
  33 {
  34         PyrSlot *a = g->sp;
  35         SetRaw(a, NUMBITS(slotRawInt(a)));
  36         return errNone;
  37 }
  38
  39 int prLog2Ceil(VMGlobals *g, int numArgsPushed);
  40 int prLog2Ceil(VMGlobals *g, int numArgsPushed)
  41 {
  42         PyrSlot *a = g->sp;
  43         SetRaw(a, LOG2CEIL(slotRawInt(a)));
  44         return errNone;
  45 }
  46
  47 int prCLZ(VMGlobals *g, int numArgsPushed);
  48 int prCLZ(VMGlobals *g, int numArgsPushed)
  49 {
  50         PyrSlot *a = g->sp;
  51         SetRaw(a, CLZ(slotRawInt(a)));
  52         return errNone;
  53 }
  54
  55 int prCTZ(VMGlobals *g, int numArgsPushed);
  56 int prCTZ(VMGlobals *g, int numArgsPushed)
  57 {
  58         PyrSlot *a = g->sp;
  59         SetRaw(a, CTZ(slotRawInt(a)));
  60         return errNone;
  61 }
  62
  63 int prNextPowerOfTwo(VMGlobals *g, int numArgsPushed);
  64 int prNextPowerOfTwo(VMGlobals *g, int numArgsPushed)
  65 {
  66         PyrSlot *a;
  67
  68         a = g->sp;
  69
  70         SetRaw(a, NEXTPOWEROFTWO(slotRawInt(a)));
  71         return errNone;
  72 }
  73
  74 int prIsPowerOfTwo(VMGlobals *g, int numArgsPushed);
  75 int prIsPowerOfTwo(VMGlobals *g, int numArgsPushed)
  76 {
  77         PyrSlot *a = g->sp;
  78         SetBool(a, ISPOWEROFTWO(slotRawInt(a)));
  79         return errNone;
  80 }
  81
  82 int prBinaryGrayCode(VMGlobals *g, int numArgsPushed);
  83 int prBinaryGrayCode(VMGlobals *g, int numArgsPushed)
  84 {
  85         PyrSlot *a;
  86
  87         a = g->sp;
  88         SetRaw(a, GRAYCODE(slotRawInt(a)));
  89         return errNone;
  90 }
  91
  92 int prSetBit(VMGlobals *g, int numArgsPushed);
  93 int prSetBit(VMGlobals *g, int numArgsPushed)
  94 {
  95         PyrSlot *a = g->sp - 2;
  96         PyrSlot *b = g->sp - 1;
  97         PyrSlot *c = g->sp;
  98
  99         int32 bit, mask;
 100         int err = slotIntVal(b, &bit);
 101         if (err) return err;
 102         bit = bit & 31;
 103         mask = 1L << bit;
 104         if (IsFalse(c)) {
 105                 SetRaw(a, slotRawInt(a) & ~mask);
 106         } else {
 107                 SetRaw(a, slotRawInt(a) |  mask);
 108         }
 109         return errNone;
 110 }
 111
 112 int prHammingDistance(VMGlobals *g, int numArgsPushed);
 113 int prHammingDistance(VMGlobals *g, int numArgsPushed)
 114 {
 115         PyrSlot *a = g->sp - 1;
 116         PyrSlot *b = g->sp;
 117
 118         if (NotInt(a) || NotInt(b))
 119                 return errWrongType;
 120
 121         int aInt = slotRawInt(a);
 122         int bInt = slotRawInt(b);
 123
 124         int count = 0, mask = 1;
 125         for(int i = 0; i < 32; i++) {
 126                 if((aInt & mask) != (bInt & mask)) count = count + 1;
 127                 mask = mask << 1;
 128         }
 129         SetRaw(a, count);
 130
 131         return errNone;
 132 }
 133
 134 void initBitPrimitives();
 135 void initBitPrimitives()
 136 {
 137         int base, index = 0;
 138
 139         base = nextPrimitiveIndex();
 140
 141         definePrimitive(base, index++, "_NextPowerOfTwo", prNextPowerOfTwo, 1, 0);
 142         definePrimitive(base, index++, "_IsPowerOfTwo", prIsPowerOfTwo, 1, 0);
 143         definePrimitive(base, index++, "_CLZ", prCLZ, 1, 0);
 144         definePrimitive(base, index++, "_CTZ", prCTZ, 1, 0);
 145         definePrimitive(base, index++, "_NumBits", prNumBits, 1, 0);
 146         definePrimitive(base, index++, "_Log2Ceil", prLog2Ceil, 1, 0);
 147         definePrimitive(base, index++, "_SetBit", prSetBit, 3, 0);
 148         definePrimitive(base, index++, "_BinaryGrayCode", prBinaryGrayCode, 1, 0);
 149         definePrimitive(base, index++, "_HammingDistance", prHammingDistance, 2, 0);
 150
 151 }
 152
 153
 154 #if _SC_PLUGINS_
 155
 156 #include "SCPlugin.h"
 157
 158 // export the function that SC will call to load the plug in.
 159 #pragma export on
 160 extern "C" { SCPlugIn* loadPlugIn(void); }
 161 #pragma export off
 162
 163
 164 // define plug in object
 165 class APlugIn : public SCPlugIn
 166 {
 167 public:
 168         APlugIn();
 169         virtual ~APlugIn();
 170
 171         virtual void AboutToCompile();
 172 };
 173
 174 APlugIn::APlugIn()
 175 {
 176         // constructor for plug in
 177 }
 178
 179 APlugIn::~APlugIn()
 180 {
 181         // destructor for plug in
 182 }
 183
 184 void APlugIn::AboutToCompile()
 185 {
 186         // this is called each time the class library is compiled.
 187         initBitPrimitives();
 188 }
 189
 190 // This function is called when the plug in is loaded into SC.
 191 // It returns an instance of APlugIn.
 192 SCPlugIn* loadPlugIn()
 193 {
 194         return new APlugIn();
 195 }
 196
 197 #endif