workbench/tools/PrintFiles.c

   1 /*
   2     Copyright © 2012, The AROS Development Team. All rights reserved.
   3     $Id$
   4
   5     Desc: Print multiple files with optional formfeed
   6     Lang: English
   7 */
   8
   9 #include <proto/exec.h>
  10 #include <proto/dos.h>
  11 #include <proto/alib.h>
  12 #include <proto/datatypes.h>
  13 #include <proto/icon.h>
  14
  15 #include <workbench/startup.h>
  16 #include <datatypes/datatypesclass.h>
  17 #include <datatypes/pictureclass.h>
  18
  19 //#define DEBUG 1
  20 #include <aros/debug.h>
  21
  22 #define USAGE "Usage: PrintFiles [-f] [-u N] file [file] [file...] (-f=formfeed -u=unit number)\n"
  23
  24 const char *vers = "$VER: PrintFiles 1.1 (12.03.2012)";
  25
  26 static struct MsgPort *mp;
  27 static union printerIO *io;
  28
  29 char __stdiowin[]="CON:/30/400/100/PrintFiles/AUTO/CLOSE/WAIT";
  30
  31
  32 static BOOL initdevice(ULONG unit)
  33 {
  34     D(bug("[PrintFiles] init unit %d\n", unit));
  35
  36     if ((mp = CreateMsgPort()))
  37     {
  38         if ((io = CreateIORequest(mp, sizeof(union printerIO))))
  39         {
  40             if (0 == OpenDevice("printer.device", unit, (struct IORequest *)io, 0))
  41             {
  42                 return TRUE;
  43             }
  44             else
  45             {
  46                 Printf("Can't open printer.device %ld\n", unit);
  47             }
  48         }
  49         else
  50         {
  51             PutStr("Can't create IO request\n");
  52         }
  53     }
  54     else
  55     {
  56         PutStr("Can't create message port\n");
  57     }
  58     return FALSE;
  59 }
  60
  61
  62 static void cleanupdevice(void)
  63 {
  64     if (io)
  65     {
  66         CloseDevice((struct IORequest *)io);
  67         DeleteIORequest(io);
  68         io = NULL;
  69     }
  70     if (mp)
  71     {
  72         DeleteMsgPort(mp);
  73         mp = NULL;
  74     }
  75 }
  76
  77
  78 static void printfile(STRPTR filename, BOOL formfeed)
  79 {
  80     if (filename != NULL)
  81     {
  82         Object *o;
  83         struct dtPrint msg;
  84         if ((o = NewDTObject(filename, PDTA_Remap, FALSE, TAG_END)))
  85         {
  86             struct TagItem tags[] = {
  87                 { DTA_Special, SPECIAL_ASPECT | SPECIAL_CENTER },
  88                 { TAG_END }
  89             };
  90             msg.MethodID          = DTM_PRINT;
  91             msg.dtp_GInfo         = NULL;
  92             msg.dtp_PIO           = (union printerIO *)io;
  93             msg.dtp_AttrList      = tags;
  94             D(bug("[PrintFiles] Trying to print %s\n", filename));
  95             if (0 == DoDTMethodA(o, NULL, NULL, (Msg)&msg))
  96             {
  97                 if (formfeed)
  98                 {
  99                     D(bug("[PrintFiles] Sending formfeed\n"));
 100                     io->ios.io_Length  = 1;
 101                     io->ios.io_Data = "\x0C";
 102                     io->ios.io_Command = CMD_WRITE;
 103                     DoIO((struct IORequest *)io);
 104                 }
 105             }
 106             else
 107             {
 108                 Printf("Failed to print %s\n", filename);
 109             }
 110             DisposeDTObject(o);
 111         }
 112     }
 113 }
 114
 115
 116 static void read_icon(struct WBArg *wbarg, BOOL *formfeed, ULONG *unit)
 117 {
 118     struct DiskObject *dobj;
 119     STRPTR *toolarray;
 120     STRPTR result;
 121
 122     *formfeed = FALSE;
 123     *unit = 0;
 124
 125     dobj = GetDiskObject(wbarg->wa_Name);
 126     if (dobj)
 127     {
 128         toolarray = dobj->do_ToolTypes;
 129
 130         if (FindToolType(toolarray, "FORMFEED"))
 131         {
 132             *formfeed = TRUE;
 133         }
 134         result = FindToolType(toolarray, "UNIT");
 135         if (result)
 136         {
 137             StrToLong(result, unit);
 138         }
 139         FreeDiskObject(dobj);
 140     }
 141 }
 142
 143
 144 int main(int argc, char **argv)
 145 {
 146     ULONG unit = 0;
 147     BOOL formfeed = FALSE;
 148     ULONG i;
 149
 150     if (argc == 0)
 151     {
 152         // started from Workbench
 153         struct WBStartup *wbmsg = (struct WBStartup *)argv;
 154         struct WBArg *wbarg = wbmsg->sm_ArgList;
 155         BPTR olddir = (BPTR)-1;
 156
 157         D(bug("[PrintFiles] numargs %d wa_lock %lx wa_name %s\n", wbmsg->sm_NumArgs, wbarg[0].wa_Lock, wbarg[0].wa_Name));
 158         if (wbmsg->sm_NumArgs > 1 && wbarg[0].wa_Lock && *wbarg[0].wa_Name)
 159         {
 160             // handle program's icon
 161             olddir = CurrentDir(wbarg->wa_Lock);
 162             read_icon(wbarg, &formfeed, &unit);
 163             if (olddir != (BPTR)-1)
 164                 CurrentDir(olddir);
 165             if (initdevice(unit))
 166             {
 167                 // handle project icons
 168                 for (i = 1; i < wbmsg->sm_NumArgs; i++)
 169                 {
 170                     D(bug("[PrintFiles] i %d wa_lock %lx wa_name %s\n", i, wbarg[i].wa_Lock, wbarg[i].wa_Name));
 171                     olddir = (BPTR)-1;
 172                     if ((wbarg[i].wa_Lock) && (*wbarg[i].wa_Name) )
 173                     {
 174                         olddir = CurrentDir(wbarg[i].wa_Lock);
 175
 176                         printfile(wbarg[i].wa_Name, formfeed);
 177
 178                         if (olddir != (BPTR)-1)
 179                             CurrentDir(olddir);
 180                     }
 181                 }
 182             }
 183         }
 184     }
 185     else
 186     {
 187         // started from CLI
 188
 189         if (argc == 1 || argv[1][0] == '?')
 190         {
 191             PutStr(USAGE);
 192             return RETURN_ERROR;
 193         }
 194
 195         // read options
 196         i = 1;
 197         while (i < argc && argv[i][0] == '-')
 198         {
 199             if (argv[i][1] == 'f')
 200             {
 201                 formfeed = TRUE;
 202             }
 203             else if (argv[i][1] == 'u' && (i + 1 < argc))
 204             {
 205                 i++;
 206                 StrToLong(argv[i], &unit);
 207             }
 208             else
 209             {
 210                 PutStr(USAGE);
 211                 return RETURN_ERROR;
 212             }
 213             i++;
 214         }
 215
 216         // print files
 217         if (initdevice(unit))
 218         {
 219             while (i < argc)
 220             {
 221                 printfile(argv[i], formfeed);
 222                 i++;
 223             }
 224         }
 225     }
 226
 227     cleanupdevice();
 228
 229     return 0;
 230 }