[2] | 1 | ;------------------------------------------------------------ |
---|
| 2 | ;------------------------------------------------------------ |
---|
| 3 | ;------------------------------------------------------------ |
---|
| 4 | ;+ |
---|
| 5 | ; NAME:colle |
---|
| 6 | ; |
---|
| 7 | ; PURPOSE: Cette fonction de concatenation existe ds IDL (avec [] cf |
---|
| 8 | ; ds le programme ds le case pour direc egale 1,2,3) tant que l''on ne |
---|
| 9 | ; cherche pas a coller suivant une dimensionsuperieure ou egale a 4. |
---|
| 10 | ; |
---|
| 11 | ; CATEGORY:bidouillage de matrice |
---|
| 12 | ; |
---|
| 13 | ; CALLING SEQUENCE:res=colle(bableau_de_pointeur,direc) ou bien |
---|
| 14 | ; res=colle(tab1,tab2,tab3,tab4,.....,direc) |
---|
| 15 | ; |
---|
| 16 | ; INPUTS: |
---|
| 17 | ; |
---|
| 18 | ; CAS 1: |
---|
| 19 | ; tableau_de_pointeur:comme son nom l''indique c'est un tableau |
---|
| 20 | ; de pointeur, dont chaque elements pointe sur tableau a coller |
---|
| 21 | ; |
---|
| 22 | ; par ex ds un programme, on veut coller n tableaux entre eux |
---|
| 23 | ; |
---|
| 24 | ; tab=ptrarr(n,/allocate_heap) |
---|
| 25 | ; for i=0,n-1 do begin |
---|
| 26 | ; *tab[n]=replicate(n,2,3) |
---|
| 27 | ; endfor |
---|
| 28 | ; res=colle(tab,1) |
---|
| 29 | ; |
---|
| 30 | ; CAS 2: |
---|
| 31 | ; on donne directement les tableaux a coller |
---|
| 32 | ; rq: ds ce cas on peut au plus donner 20 tableaux en entree. |
---|
| 33 | ; |
---|
| 34 | ; ATTENTION : sans le mot cle /SAUVE les arguments en entree |
---|
| 35 | ; sont detruits lorsque l''on construit res. ds le cas 1 on |
---|
| 36 | ; detruit le tableau de pointeurs et les variables sur |
---|
| 37 | ; lesquelles on pointe. |
---|
| 38 | ; |
---|
| 39 | ; direc: la direction suivant laquelle les coller, 1,2,3... |
---|
| 40 | ; |
---|
| 41 | ; KEYWORD PARAMETERS: |
---|
| 42 | ; |
---|
| 43 | ; /SAUVE: mot cle qui force a sauvegarder le tableau de |
---|
| 44 | ; pointeur et les tableaux a coller. |
---|
| 45 | ; |
---|
| 46 | ; OUTPUTS:res=matrice resultat |
---|
| 47 | ; |
---|
| 48 | ; RESTRICTIONS: |
---|
| 49 | ; |
---|
| 50 | ; EXAMPLE: |
---|
| 51 | ; |
---|
| 52 | ; IDL> print, colle(replicate(1,2,2,2),indgen(2,2,2),2) |
---|
| 53 | ; 1 1 |
---|
| 54 | ; 1 1 |
---|
| 55 | ; 0 1 |
---|
| 56 | ; 2 3 |
---|
| 57 | ; |
---|
| 58 | ; 1 1 |
---|
| 59 | ; 1 1 |
---|
| 60 | ; 4 5 |
---|
| 61 | ; 6 7 |
---|
| 62 | ; |
---|
| 63 | ; MODIFICATION HISTORY: Sebastien Masson (smasson@lodyc.jussieu.fr) |
---|
| 64 | ; 13/1/98 |
---|
| 65 | ;- |
---|
| 66 | ;------------------------------------------------------------ |
---|
| 67 | ;------------------------------------------------------------ |
---|
| 68 | ;------------------------------------------------------------ |
---|
| 69 | ; pour suprimer une variable. |
---|
| 70 | ;------------------------------------------------------------ |
---|
| 71 | PRO UNDEFINE, varname |
---|
| 72 | tempvar = SIZE(TEMPORARY(varname)) |
---|
| 73 | END |
---|
| 74 | ;------------------------------------------------------------ |
---|
| 75 | ;------------------------------------------------------------ |
---|
| 76 | FUNCTION colle, a0, a1, a2, a3, a4, a5, a6, a7, a8, a9, a10, a11, a12, a13, a14, a15, a16, a17, a18, a19, a20, SAUVE = sauve |
---|
| 77 | res = -1 |
---|
| 78 | ;------------------------------------------------------------ |
---|
| 79 | ; on met en place ptrtab et direc en fonction des arguments en entree |
---|
| 80 | ;------------------------------------------------------------ |
---|
| 81 | case 1 of |
---|
| 82 | n_params() EQ 2:BEGIN ; cas ou l'on donne directement le tableau de pointeurs |
---|
| 83 | ptrtab = a0 |
---|
| 84 | direc = a1 |
---|
| 85 | if NOT keyword_set(sauve) then undefine, a0 |
---|
| 86 | ; on recupere le nombre de tableaux a coller |
---|
| 87 | nbretab = (size(ptrtab))[1] |
---|
| 88 | end |
---|
| 89 | n_params() GT 2:BEGIN |
---|
| 90 | ; on recupere le nombre de tableaux a coller |
---|
| 91 | nbretab = n_params()-1 |
---|
| 92 | bidon = execute('direc = a'+strtrim(n_params()-1, 2)) |
---|
| 93 | ; on ecrit le tableau de pointeur dont chaque element pointe sur un tableau |
---|
| 94 | ptrtab=ptrarr(nbretab,/allocate_heap) |
---|
| 95 | for n = 0,nbretab-1 do begin |
---|
| 96 | bidon = execute('*ptrtab[n]=a'+strtrim(n, 2)) |
---|
| 97 | if NOT keyword_set(sauve) then bidon = execute('undefine, a'+strtrim(n, 2)) |
---|
| 98 | endfor |
---|
| 99 | sauve = 0 |
---|
| 100 | end |
---|
| 101 | ELSE: |
---|
| 102 | endcase |
---|
| 103 | ;------------------------------------------------------------ |
---|
| 104 | ; case sur la valeure de direc |
---|
| 105 | ;------------------------------------------------------------ |
---|
| 106 | case direc of |
---|
[7] | 107 | 1:BEGIN ; on colle suivant la dimension 1 |
---|
[2] | 108 | res = *ptrtab[0] |
---|
| 109 | if NOT keyword_set(sauve) then ptr_free, ptrtab[0] |
---|
| 110 | FOR n = 1,nbretab-1 DO BEGIN |
---|
| 111 | res = [temporary(res), *ptrtab[n]] |
---|
| 112 | if NOT keyword_set(sauve) then ptr_free, ptrtab[n] |
---|
| 113 | ENDFOR |
---|
| 114 | END |
---|
[7] | 115 | 2:BEGIN ; on colle suivant la dimension 2 |
---|
[2] | 116 | res = *ptrtab[0] |
---|
| 117 | if NOT keyword_set(sauve) then ptr_free, ptrtab[0] |
---|
| 118 | FOR n = 1,nbretab-1 DO BEGIN |
---|
| 119 | res = [[temporary(res)], [*ptrtab[n]]] |
---|
| 120 | if NOT keyword_set(sauve) then ptr_free, ptrtab[n] |
---|
| 121 | ENDFOR |
---|
| 122 | END |
---|
[7] | 123 | 3:BEGIN ; on colle suivant la dimension 3 |
---|
[2] | 124 | res = *ptrtab[0] |
---|
| 125 | if NOT keyword_set(sauve) then ptr_free, ptrtab[0] |
---|
| 126 | FOR n = 1,nbretab-1 DO BEGIN |
---|
| 127 | res = [[[temporary(res)]], [[*ptrtab[n]]]] |
---|
| 128 | if NOT keyword_set(sauve) then ptr_free, ptrtab[n] |
---|
| 129 | ENDFOR |
---|
| 130 | END |
---|
| 131 | ELSE:BEGIN |
---|
| 132 | ; on transpose res de facon a mettre la dimension a coller numero 1. |
---|
| 133 | ; pour cela on contient le vecteur permute qui donne la place que |
---|
| 134 | ; doivent prendre les dimensions ds la matrice transposee |
---|
[31] | 135 | siz = (size(*ptrtab[0]))[0] |
---|
| 136 | if siz LT direc then $ |
---|
| 137 | *ptrtab[0] = reform(*ptrtab[0], [(size(*ptrtab[0]))[1:siz], replicate(1, direc-siz)], /over) |
---|
[2] | 138 | permute = indgen((size(*ptrtab[0]))[0]) |
---|
| 139 | permute[0] = direc-1 |
---|
| 140 | permute[direc-1] = 0 |
---|
| 141 | res = transpose(*ptrtab[0], permute) |
---|
| 142 | if NOT keyword_set(sauve) then ptr_free, ptrtab[0] |
---|
| 143 | FOR n = 1,nbretab-1 DO BEGIN ; on colle suivant la dimension 1 |
---|
[31] | 144 | if (size(*ptrtab[n]))[0] LT direc then $ |
---|
| 145 | *ptrtab[n] = reform(*ptrtab[n], [(size(*ptrtab[n]))[1:siz], replicate(1, direc-siz)]) |
---|
[2] | 146 | res = [temporary(res), transpose(*ptrtab[n], permute)] |
---|
| 147 | if NOT keyword_set(sauve) then ptr_free, ptrtab[n] |
---|
| 148 | ENDFOR |
---|
| 149 | res = transpose(temporary(res), permute) |
---|
| 150 | END |
---|
| 151 | ENDCASE |
---|
| 152 | ;------------------------------------------------------------ |
---|
| 153 | if NOT keyword_set(sauve) then undefine, ptrtab |
---|
| 154 | sortie: |
---|
| 155 | return, res |
---|
| 156 | |
---|
| 157 | |
---|
| 158 | END |
---|
| 159 | |
---|
| 160 | ;------------------------------------------------------------ |
---|