Mac/Python/macmain.c

   1 /***********************************************************
   2 Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam,
   3 The Netherlands.
   4
   5                         All Rights Reserved
   6
   7 Permission to use, copy, modify, and distribute this software and its
   8 documentation for any purpose and without fee is hereby granted,
   9 provided that the above copyright notice appear in all copies and that
  10 both that copyright notice and this permission notice appear in
  11 supporting documentation, and that the names of Stichting Mathematisch
  12 Centrum or CWI not be used in advertising or publicity pertaining to
  13 distribution of the software without specific, written prior permission.
  14
  15 STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
  16 THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
  17 FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
  18 FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  19 WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  20 ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
  21 OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  22
  23 ******************************************************************/
  24
  25 /* Python interpreter main program */
  26
  27 #include "Python.h"
  28 #include "pythonresources.h"
  29 #include "import.h"
  30 #include "marshal.h"
  31
  32 #include <Memory.h>
  33 #include <Resources.h>
  34 #include <stdio.h>
  35 #include <Events.h>
  36 #include <Windows.h>
  37 #include <Desk.h>
  38
  39 #ifdef __MWERKS__
  40 #include <SIOUX.h>
  41 #endif
  42
  43 #define STARTUP "PythonStartup"
  44
  45 extern int Py_DebugFlag; /* For parser.c, declared in pythonrun.c */
  46 extern int Py_VerboseFlag; /* For import.c, declared in pythonrun.c */
  47 extern int Py_SuppressPrintingFlag; /* For ceval.c, declared in pythonrun.c */
  48
  49
  50 /* Subroutines that live in their own file */
  51 extern char *getversion();
  52 extern char *getcopyright();
  53
  54
  55 /* For getprogramname(); set by main() */
  56 static char *argv0;
  57
  58 /* For getargcargv(); set by main() */
  59 static char **orig_argv;
  60 static int  orig_argc;
  61
  62 #ifdef USE_MAC_APPLET_SUPPORT
  63 /* Applet support */
  64
  65 /* Run a compiled Python Python script from 'PYC ' resource __main__ */
  66 static int
  67 run_main_resource()
  68 {
  69         Handle h;
  70         long size;
  71         PyObject *code;
  72         PyObject *result;
  73
  74         h = GetNamedResource('PYC ', "\p__main__");
  75         if (h == NULL) {
  76                 Alert(NOPYC_ALERT, NULL);
  77                 return 1;
  78         }
  79         size = GetResourceSizeOnDisk(h);
  80         HLock(h);
  81         code = PyMarshal_ReadObjectFromString(*h + 8, (int)(size - 8));
  82         HUnlock(h);
  83         ReleaseResource(h);
  84         if (code == NULL) {
  85                 PyErr_Print();
  86                 return 1;
  87         }
  88         result = PyImport_ExecCodeModule("__main__", code);
  89         Py_DECREF(code);
  90         if (result == NULL) {
  91                 PyErr_Print();
  92                 return 1;
  93         }
  94         Py_DECREF(result);
  95         return 0;
  96 }
  97
  98 /* Initialization sequence for applets */
  99 void
 100 PyMac_InitApplet()
 101 {
 102         int argc;
 103         char **argv;
 104         int err;
 105
 106 #ifdef USE_MAC_SHARED_LIBRARY
 107         PyMac_AddLibResources();
 108 #endif
 109 #ifdef __MWERKS__
 110         SIOUXSettings.asktosaveonclose = 0;
 111         SIOUXSettings.showstatusline = 0;
 112         SIOUXSettings.tabspaces = 4;
 113 #endif
 114         argc = PyMac_GetArgv(&argv);
 115         Py_Initialize();
 116         PySys_SetArgv(argc, argv);
 117         err = run_main_resource();
 118         fflush(stderr);
 119         fflush(stdout);
 120 #ifdef __MWERKS__
 121         if (!err)
 122                 SIOUXSettings.autocloseonquit = 1;
 123         else
 124                 printf("\n[Terminated]\n");
 125 #endif
 126         /* XXX Should we bother to Py_Exit(sts)? */
 127 }
 128
 129 #endif /* USE_MAC_APPLET_SUPPORT */
 130
 131 /* For normal application */
 132 void
 133 PyMac_InitApplication()
 134 {
 135         int argc;
 136         char **argv;
 137
 138 #ifdef USE_MAC_SHARED_LIBRARY
 139         PyMac_AddLibResources();
 140 #endif
 141 #ifdef __MWERKS__
 142         SIOUXSettings.asktosaveonclose = 0;
 143         SIOUXSettings.showstatusline = 0;
 144         SIOUXSettings.tabspaces = 4;
 145 #endif
 146         argc = PyMac_GetArgv(&argv);
 147         if ( argc > 1 ) {
 148                 /* We're running a script. Attempt to change current directory */
 149                 char curwd[256], *endp;
 150
 151                 strcpy(curwd, argv[1]);
 152                 endp = strrchr(curwd, ':');
 153                 if ( endp && endp > curwd ) {
 154                         *endp = '\0';
 155
 156                         chdir(curwd);
 157                 }
 158         }
 159         Py_Main(argc, argv);
 160 }
 161
 162 /*
 163 ** PyMac_InteractiveOptions - Allow user to set options if option key is pressed
 164 */
 165 void
 166 PyMac_InteractiveOptions(int *inspect, int *verbose, int *suppress_print,
 167                                                  int *unbuffered, int *debugging)
 168 {
 169         KeyMap rmap;
 170         unsigned char *map;
 171         short item, type;
 172         ControlHandle handle;
 173         DialogPtr dialog;
 174         Rect rect;
 175
 176         GetKeys(rmap);
 177         map = (unsigned char *)rmap;
 178         if ( ( map[0x3a>>3] & (1<<(0x3a&7)) ) == 0 )    /* option key is 3a */
 179                 return;
 180
 181         dialog = GetNewDialog(OPT_DIALOG, NULL, (WindowPtr)-1);
 182         if ( dialog == NULL ) {
 183                 printf("Option dialog not found - cannot set options\n");
 184                 return;
 185         }
 186         while (1) {
 187                 handle = NULL;
 188                 ModalDialog(NULL, &item);
 189                 if ( item == OPT_OK )
 190                         break;
 191                 if ( item == OPT_CANCEL ) {
 192                         DisposDialog(dialog);
 193                         exit(0);
 194                 }
 195 #define OPT_ITEM(num, var) \
 196                 if ( item == (num) ) { \
 197                         *(var) = !*(var); \
 198                         GetDialogItem(dialog, (num), &type, (Handle *)&handle, &rect); \
 199                         SetCtlValue(handle, (short)*(var)); \
 200                 }
 201
 202                 OPT_ITEM(OPT_INSPECT, inspect);
 203                 OPT_ITEM(OPT_VERBOSE, verbose);
 204                 OPT_ITEM(OPT_SUPPRESS, suppress_print);
 205                 OPT_ITEM(OPT_UNBUFFERED, unbuffered);
 206                 OPT_ITEM(OPT_DEBUGGING, debugging);
 207
 208 #undef OPT_ITEM
 209         }
 210         DisposDialog(dialog);
 211 }
 212 /* Main program */
 213
 214 int
 215 Py_Main(argc, argv)
 216         int argc;
 217         char **argv;
 218 {
 219         int c;
 220         int sts;
 221         char *command = NULL;
 222         char *filename = NULL;
 223         FILE *fp = stdin;
 224         char *p;
 225         int inspect = 0;
 226         int unbuffered = 0;
 227
 228         orig_argc = argc;       /* For getargcargv() */
 229         orig_argv = argv;
 230         argv0 = argv[0];        /* For getprogramname() */
 231
 232         PyMac_InteractiveOptions(&inspect, &Py_VerboseFlag, &Py_SuppressPrintingFlag,
 233                         &unbuffered, &Py_DebugFlag);
 234
 235
 236         if (unbuffered) {
 237 #ifndef MPW
 238                 setbuf(stdout, (char *)NULL);
 239                 setbuf(stderr, (char *)NULL);
 240 #else
 241                 /* On MPW (3.2) unbuffered seems to hang */
 242                 setvbuf(stdout, (char *)NULL, _IOLBF, BUFSIZ);
 243                 setvbuf(stderr, (char *)NULL, _IOLBF, BUFSIZ);
 244 #endif
 245         }
 246
 247         filename = argv[1];
 248
 249         if (Py_VerboseFlag ||
 250             command == NULL && filename == NULL && isatty((int)fileno(fp)))
 251                 fprintf(stderr, "Python %s\n%s\n",
 252                         getversion(), getcopyright());
 253
 254         if (filename != NULL) {
 255                 if ((fp = fopen(filename, "r")) == NULL) {
 256                         fprintf(stderr, "%s: can't open file '%s'\n",
 257                                 argv[0], filename);
 258                         exit(2);
 259                 }
 260         }
 261
 262         Py_Initialize();
 263
 264         PySys_SetArgv(argc-1, argv+1);
 265
 266         if (filename == NULL && isatty((int)fileno(fp))) {
 267                 FILE *fp = fopen(STARTUP, "r");
 268                 if (fp != NULL) {
 269                         (void) PyRun_SimpleFile(fp, STARTUP);
 270                         PyErr_Clear();
 271                         fclose(fp);
 272                 }
 273         }
 274         sts = PyRun_AnyFile(
 275                         fp, filename == NULL ? "<stdin>" : filename) != 0;
 276         if (filename != NULL)
 277                 fclose(fp);
 278
 279         if (inspect && isatty((int)fileno(stdin)) &&
 280             (filename != NULL || command != NULL))
 281                 sts = PyRun_AnyFile(stdin, "<stdin>") != 0;
 282
 283         Py_Exit(sts);
 284         /*NOTREACHED*/
 285 }
 286
 287
 288 /* Return the program name -- some code out there needs this. */
 289
 290 char *
 291 getprogramname()
 292 {
 293         return argv0;
 294 }
 295
 296
 297 /* Make the *original* argc/argv available to other modules.
 298    This is rare, but it is needed by the secureware extension. */
 299
 300 void
 301 getargcargv(argc,argv)
 302         int *argc;
 303         char ***argv;
 304 {
 305         *argc = orig_argc;
 306         *argv = orig_argv;
 307 }